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 // CC-OFFNXT(G.RES.09) private constructor
47 return std::unique_ptr<SharedReferenceStorage>(new SharedReferenceStorage(data, realSize));
48 }
49
GetReference(EtsObject * etsObject)50 SharedReference *SharedReferenceStorage::GetReference(EtsObject *etsObject)
51 {
52 ASSERT(HasReference(etsObject));
53 return GetItemByIndex(SharedReference::ExtractMaybeIndex(etsObject));
54 }
55
GetReference(napi_env env,napi_value jsObject)56 SharedReference *SharedReferenceStorage::GetReference(napi_env env, napi_value jsObject)
57 {
58 void *data = SharedReference::ExtractMaybeReference(env, jsObject);
59 if (UNLIKELY(data == nullptr)) {
60 return nullptr;
61 }
62 return GetReference(data);
63 }
64
GetReference(void * data)65 SharedReference *SharedReferenceStorage::GetReference(void *data)
66 {
67 auto *sharedRef = reinterpret_cast<SharedReference *>(data);
68 if (UNLIKELY(!IsValidItem(sharedRef))) {
69 // We don't own that object
70 return nullptr;
71 }
72 ASSERT(SharedReferenceSanity::CheckAlive(sharedRef));
73 return sharedRef;
74 }
75
76 template <SharedReference::InitFn REF_INIT>
CreateReference(InteropCtx * ctx,EtsObject * etsObject,napi_value jsObject)77 inline SharedReference *SharedReferenceStorage::CreateReference(InteropCtx *ctx, EtsObject *etsObject,
78 napi_value jsObject)
79 {
80 ASSERT(!SharedReference::HasReference(etsObject));
81 SharedReference *sharedRef = AllocItem();
82 if (UNLIKELY(sharedRef == nullptr)) {
83 ctx->ThrowJSError(ctx->GetJSEnv(), "no free space for SharedReference");
84 return nullptr;
85 }
86 if (UNLIKELY(!(sharedRef->*REF_INIT)(ctx, etsObject, jsObject, GetIndexByItem(sharedRef)))) {
87 auto coro = EtsCoroutine::GetCurrent();
88 if (coro->HasPendingException()) {
89 ctx->ForwardEtsException(coro);
90 }
91 ASSERT(ctx->SanityJSExceptionPending());
92 return nullptr;
93 }
94 return sharedRef;
95 }
96
CreateETSObjectRef(InteropCtx * ctx,EtsObject * etsObject,napi_value jsObject)97 SharedReference *SharedReferenceStorage::CreateETSObjectRef(InteropCtx *ctx, EtsObject *etsObject, napi_value jsObject)
98 {
99 return CreateReference<&SharedReference::InitETSObject>(ctx, etsObject, jsObject);
100 }
101
CreateJSObjectRef(InteropCtx * ctx,EtsObject * etsObject,napi_value jsObject)102 SharedReference *SharedReferenceStorage::CreateJSObjectRef(InteropCtx *ctx, EtsObject *etsObject, napi_value jsObject)
103 {
104 return CreateReference<&SharedReference::InitJSObject>(ctx, etsObject, jsObject);
105 }
106
CreateHybridObjectRef(InteropCtx * ctx,EtsObject * etsObject,napi_value jsObject)107 SharedReference *SharedReferenceStorage::CreateHybridObjectRef(InteropCtx *ctx, EtsObject *etsObject,
108 napi_value jsObject)
109 {
110 return CreateReference<&SharedReference::InitHybridObject>(ctx, etsObject, jsObject);
111 }
112
RemoveReference(SharedReference * sharedRef)113 void SharedReferenceStorage::RemoveReference(SharedReference *sharedRef)
114 {
115 FreeItem(sharedRef);
116 SharedReferenceSanity::Kill(sharedRef);
117 }
118
CheckAlive(void * data)119 bool SharedReferenceStorage::CheckAlive(void *data)
120 {
121 auto *sharedRef = reinterpret_cast<SharedReference *>(data);
122 return IsValidItem(sharedRef) && SharedReferenceSanity::CheckAlive(sharedRef);
123 }
124
125 } // namespace ark::ets::interop::js::ets_proxy
126