1 /**
2 * Copyright (c) 2023-2024 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16 #include "plugins/ets/runtime/interop_js/ets_proxy/shared_reference_storage.h"
17
18 #include "plugins/ets/runtime/interop_js/interop_context.h"
19
20 namespace ark::ets::interop::js::ets_proxy {
21
22 class SharedReferenceSanity {
23 public:
Kill(SharedReference * ref)24 static inline void Kill(SharedReference *ref)
25 {
26 ASSERT(ref->jsRef_ != nullptr);
27 ref->jsRef_ = nullptr;
28 }
29
CheckAlive(SharedReference * ref)30 static inline bool CheckAlive(SharedReference *ref)
31 {
32 return ref->jsRef_ != nullptr;
33 }
34 };
35
36 /*static*/
Create()37 std::unique_ptr<SharedReferenceStorage> SharedReferenceStorage::Create()
38 {
39 size_t realSize = SharedReferenceStorage::MAX_POOL_SIZE;
40
41 void *data = os::mem::MapRWAnonymousRaw(realSize);
42 if (data == nullptr) {
43 INTEROP_LOG(FATAL) << "Cannot allocate MemPool";
44 return nullptr;
45 }
46 return std::unique_ptr<SharedReferenceStorage>(new SharedReferenceStorage(data, realSize));
47 }
48
GetReference(EtsObject * etsObject)49 SharedReference *SharedReferenceStorage::GetReference(EtsObject *etsObject)
50 {
51 ASSERT(HasReference(etsObject));
52 return GetItemByIndex(SharedReference::ExtractMaybeIndex(etsObject));
53 }
54
GetReference(napi_env env,napi_value jsObject)55 SharedReference *SharedReferenceStorage::GetReference(napi_env env, napi_value jsObject)
56 {
57 void *data = SharedReference::ExtractMaybeReference(env, jsObject);
58 if (UNLIKELY(data == nullptr)) {
59 return nullptr;
60 }
61 return GetReference(data);
62 }
63
GetReference(void * data)64 SharedReference *SharedReferenceStorage::GetReference(void *data)
65 {
66 auto *sharedRef = reinterpret_cast<SharedReference *>(data);
67 if (UNLIKELY(!IsValidItem(sharedRef))) {
68 // We don't own that object
69 return nullptr;
70 }
71 ASSERT(SharedReferenceSanity::CheckAlive(sharedRef));
72 return sharedRef;
73 }
74
75 template <SharedReference::InitFn REF_INIT>
CreateReference(InteropCtx * ctx,EtsObject * etsObject,napi_value jsObject)76 inline SharedReference *SharedReferenceStorage::CreateReference(InteropCtx *ctx, EtsObject *etsObject,
77 napi_value jsObject)
78 {
79 ASSERT(!SharedReference::HasReference(etsObject));
80 SharedReference *sharedRef = AllocItem();
81 if (UNLIKELY(sharedRef == nullptr)) {
82 ctx->ThrowJSError(ctx->GetJSEnv(), "no free space for SharedReference");
83 return nullptr;
84 }
85 if (UNLIKELY(!(sharedRef->*REF_INIT)(ctx, etsObject, jsObject, GetIndexByItem(sharedRef)))) {
86 auto coro = EtsCoroutine::GetCurrent();
87 if (coro->HasPendingException()) {
88 ctx->ForwardEtsException(coro);
89 }
90 ASSERT(ctx->SanityJSExceptionPending());
91 return nullptr;
92 }
93 return sharedRef;
94 }
95
CreateETSObjectRef(InteropCtx * ctx,EtsObject * etsObject,napi_value jsObject)96 SharedReference *SharedReferenceStorage::CreateETSObjectRef(InteropCtx *ctx, EtsObject *etsObject, napi_value jsObject)
97 {
98 return CreateReference<&SharedReference::InitETSObject>(ctx, etsObject, jsObject);
99 }
100
CreateJSObjectRef(InteropCtx * ctx,EtsObject * etsObject,napi_value jsObject)101 SharedReference *SharedReferenceStorage::CreateJSObjectRef(InteropCtx *ctx, EtsObject *etsObject, napi_value jsObject)
102 {
103 return CreateReference<&SharedReference::InitJSObject>(ctx, etsObject, jsObject);
104 }
105
CreateHybridObjectRef(InteropCtx * ctx,EtsObject * etsObject,napi_value jsObject)106 SharedReference *SharedReferenceStorage::CreateHybridObjectRef(InteropCtx *ctx, EtsObject *etsObject,
107 napi_value jsObject)
108 {
109 return CreateReference<&SharedReference::InitHybridObject>(ctx, etsObject, jsObject);
110 }
111
RemoveReference(SharedReference * sharedRef)112 void SharedReferenceStorage::RemoveReference(SharedReference *sharedRef)
113 {
114 FreeItem(sharedRef);
115 SharedReferenceSanity::Kill(sharedRef);
116 }
117
CheckAlive(void * data)118 bool SharedReferenceStorage::CheckAlive(void *data)
119 {
120 auto *sharedRef = reinterpret_cast<SharedReference *>(data);
121 return IsValidItem(sharedRef) && SharedReferenceSanity::CheckAlive(sharedRef);
122 }
123
124 } // namespace ark::ets::interop::js::ets_proxy
125