1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "third_party/jni_zero/java_refs.h"
#include "third_party/jni_zero/jni_methods.h"
namespace jni_zero {
namespace {
const int kDefaultLocalFrameCapacity = 16;
}
ScopedJavaLocalFrame::ScopedJavaLocalFrame(JNIEnv* env) : env_(env) {
int failed = env_->PushLocalFrame(kDefaultLocalFrameCapacity);
JNI_ZERO_DCHECK(!failed);
}
ScopedJavaLocalFrame::ScopedJavaLocalFrame(JNIEnv* env, int capacity)
: env_(env) {
int failed = env_->PushLocalFrame(capacity);
JNI_ZERO_DCHECK(!failed);
}
ScopedJavaLocalFrame::~ScopedJavaLocalFrame() {
env_->PopLocalFrame(nullptr);
}
#if JNI_ZERO_DCHECK_IS_ON()
// This constructor is inlined when DCHECKs are disabled; don't add anything
// else here.
JavaRef<jobject>::JavaRef(JNIEnv* env, jobject obj) : obj_(obj) {
if (obj) {
JNI_ZERO_DCHECK(env && env->GetObjectRefType(obj) == JNILocalRefType);
}
}
#endif
JNIEnv* JavaRef<jobject>::SetNewLocalRef(JNIEnv* env, jobject obj) {
if (!env) {
env = AttachCurrentThread();
} else {
JNI_ZERO_DCHECK(env ==
AttachCurrentThread()); // Is |env| on correct thread.
}
if (obj) {
obj = env->NewLocalRef(obj);
}
if (obj_) {
env->DeleteLocalRef(obj_);
}
obj_ = obj;
return env;
}
void JavaRef<jobject>::SetNewGlobalRef(JNIEnv* env, jobject obj) {
if (!env) {
env = AttachCurrentThread();
} else {
JNI_ZERO_DCHECK(env ==
AttachCurrentThread()); // Is |env| on correct thread.
}
if (obj) {
obj = env->NewGlobalRef(obj);
}
if (obj_) {
env->DeleteGlobalRef(obj_);
}
obj_ = obj;
}
void JavaRef<jobject>::ResetLocalRef(JNIEnv* env) {
if (obj_) {
JNI_ZERO_DCHECK(env ==
AttachCurrentThread()); // Is |env| on correct thread.
env->DeleteLocalRef(obj_);
obj_ = nullptr;
}
}
void JavaRef<jobject>::ResetGlobalRef() {
if (obj_) {
AttachCurrentThread()->DeleteGlobalRef(obj_);
obj_ = nullptr;
}
}
ScopedJavaGlobalWeakRef::ScopedJavaGlobalWeakRef(
const ScopedJavaGlobalWeakRef& orig) {
Assign(orig);
}
ScopedJavaGlobalWeakRef::ScopedJavaGlobalWeakRef(JNIEnv* env,
const JavaRef<jobject>& obj)
: obj_(env->NewWeakGlobalRef(obj.obj())) {}
void ScopedJavaGlobalWeakRef::reset() {
if (obj_) {
AttachCurrentThread()->DeleteWeakGlobalRef(obj_);
obj_ = nullptr;
}
}
ScopedJavaLocalRef<jobject> ScopedJavaGlobalWeakRef::get(JNIEnv* env) const {
jobject real = obj_ ? real = env->NewLocalRef(obj_) : nullptr;
return ScopedJavaLocalRef<jobject>(env, real);
}
void ScopedJavaGlobalWeakRef::Assign(const ScopedJavaGlobalWeakRef& other) {
if (&other == this) {
return;
}
JNIEnv* env = AttachCurrentThread();
if (obj_) {
env->DeleteWeakGlobalRef(obj_);
}
obj_ = other.obj_ ? env->NewWeakGlobalRef(other.obj_) : nullptr;
}
} // namespace jni_zero
|
