• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // Copyright 2014 The PDFium Authors
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4 
5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
6 
7 #include "fxjs/cfxjs_engine.h"
8 
9 #include <memory>
10 #include <utility>
11 
12 #include "core/fxcrt/stl_util.h"
13 #include "core/fxcrt/unowned_ptr.h"
14 #include "fxjs/cfx_v8_array_buffer_allocator.h"
15 #include "fxjs/cjs_object.h"
16 #include "fxjs/fxv8.h"
17 #include "fxjs/xfa/cfxjse_runtimedata.h"
18 #include "third_party/base/check.h"
19 #include "third_party/base/check_op.h"
20 #include "v8/include/v8-context.h"
21 #include "v8/include/v8-exception.h"
22 #include "v8/include/v8-isolate.h"
23 #include "v8/include/v8-message.h"
24 #include "v8/include/v8-primitive.h"
25 #include "v8/include/v8-script.h"
26 #include "v8/include/v8-util.h"
27 
28 namespace {
29 
30 unsigned int g_embedderDataSlot = 1u;
31 v8::Isolate* g_isolate = nullptr;
32 size_t g_isolate_ref_count = 0;
33 CFX_V8ArrayBufferAllocator* g_arrayBufferAllocator = nullptr;
34 v8::Global<v8::ObjectTemplate>* g_DefaultGlobalObjectTemplate = nullptr;
35 
36 // Only the address matters, values are for humans debugging. ASLR should
37 // ensure that these values are unlikely to arise otherwise. Keep these
38 // wchar_t to prevent the compiler from doing something clever, like
39 // aligning them on a byte boundary to save space, which would make them
40 // incompatible for use as V8 aligned pointers.
41 const wchar_t kPerObjectDataTag[] = L"CFXJS_PerObjectData";
42 const wchar_t kPerIsolateDataTag[] = L"FXJS_PerIsolateData";
43 
GetAlignedPointerForPerObjectDataTag()44 void* GetAlignedPointerForPerObjectDataTag() {
45   return const_cast<void*>(static_cast<const void*>(kPerObjectDataTag));
46 }
47 
GetLineAndColumnFromError(v8::Local<v8::Message> message,v8::Local<v8::Context> context)48 std::pair<int, int> GetLineAndColumnFromError(v8::Local<v8::Message> message,
49                                               v8::Local<v8::Context> context) {
50   if (message.IsEmpty())
51     return std::make_pair(-1, -1);
52   return std::make_pair(message->GetLineNumber(context).FromMaybe(-1),
53                         message->GetStartColumn());
54 }
55 
56 }  // namespace
57 
58 class CFXJS_PerObjectData {
59  public:
60   ~CFXJS_PerObjectData() = default;
61 
SetNewDataInObject(uint32_t nObjDefnID,v8::Local<v8::Object> pObj)62   static void SetNewDataInObject(uint32_t nObjDefnID,
63                                  v8::Local<v8::Object> pObj) {
64     if (pObj->InternalFieldCount() == 2) {
65       pObj->SetAlignedPointerInInternalField(
66           0, GetAlignedPointerForPerObjectDataTag());
67       pObj->SetAlignedPointerInInternalField(
68           1, new CFXJS_PerObjectData(nObjDefnID));
69     }
70   }
71 
GetFromObject(v8::Local<v8::Object> pObj)72   static CFXJS_PerObjectData* GetFromObject(v8::Local<v8::Object> pObj) {
73     if (pObj.IsEmpty() || pObj->InternalFieldCount() != 2 ||
74         pObj->GetAlignedPointerFromInternalField(0) !=
75             GetAlignedPointerForPerObjectDataTag()) {
76       return nullptr;
77     }
78     return static_cast<CFXJS_PerObjectData*>(
79         pObj->GetAlignedPointerFromInternalField(1));
80   }
81 
GetObjDefnID() const82   uint32_t GetObjDefnID() const { return m_ObjDefnID; }
GetPrivate()83   CJS_Object* GetPrivate() { return m_pPrivate.get(); }
SetPrivate(std::unique_ptr<CJS_Object> p)84   void SetPrivate(std::unique_ptr<CJS_Object> p) { m_pPrivate = std::move(p); }
85 
86  private:
CFXJS_PerObjectData(uint32_t nObjDefnID)87   explicit CFXJS_PerObjectData(uint32_t nObjDefnID) : m_ObjDefnID(nObjDefnID) {}
88 
89   const uint32_t m_ObjDefnID;
90   std::unique_ptr<CJS_Object> m_pPrivate;
91 };
92 
93 // Global weak map to save dynamic objects.
94 class V8TemplateMapTraits final
95     : public v8::StdMapTraits<CFXJS_PerObjectData*, v8::Object> {
96  public:
97   using WeakCallbackDataType = CFXJS_PerObjectData;
98   using MapType = v8::
99       GlobalValueMap<WeakCallbackDataType*, v8::Object, V8TemplateMapTraits>;
100 
101   static const v8::PersistentContainerCallbackType kCallbackType =
102       v8::kWeakWithInternalFields;
103 
WeakCallbackParameter(MapType * map,WeakCallbackDataType * key,v8::Local<v8::Object> value)104   static WeakCallbackDataType* WeakCallbackParameter(
105       MapType* map,
106       WeakCallbackDataType* key,
107       v8::Local<v8::Object> value) {
108     return key;
109   }
110   static MapType* MapFromWeakCallbackInfo(
111       const v8::WeakCallbackInfo<WeakCallbackDataType>&);
KeyFromWeakCallbackInfo(const v8::WeakCallbackInfo<WeakCallbackDataType> & data)112   static WeakCallbackDataType* KeyFromWeakCallbackInfo(
113       const v8::WeakCallbackInfo<WeakCallbackDataType>& data) {
114     return data.GetParameter();
115   }
OnWeakCallback(const v8::WeakCallbackInfo<WeakCallbackDataType> & data)116   static void OnWeakCallback(
117       const v8::WeakCallbackInfo<WeakCallbackDataType>& data) {}
118   static void DisposeWeak(
119       const v8::WeakCallbackInfo<WeakCallbackDataType>& data);
120   static void Dispose(v8::Isolate* isolate,
121                       v8::Global<v8::Object> value,
122                       WeakCallbackDataType* key);
DisposeCallbackData(WeakCallbackDataType * callbackData)123   static void DisposeCallbackData(WeakCallbackDataType* callbackData) {}
124 };
125 
126 class V8TemplateMap {
127  public:
128   using WeakCallbackDataType = CFXJS_PerObjectData;
129   using MapType = v8::
130       GlobalValueMap<WeakCallbackDataType*, v8::Object, V8TemplateMapTraits>;
131 
V8TemplateMap(v8::Isolate * isolate)132   explicit V8TemplateMap(v8::Isolate* isolate) : m_map(isolate) {}
133   ~V8TemplateMap() = default;
134 
SetAndMakeWeak(v8::Local<v8::Object> handle)135   void SetAndMakeWeak(v8::Local<v8::Object> handle) {
136     WeakCallbackDataType* key = CFXJS_PerObjectData::GetFromObject(handle);
137     DCHECK(!m_map.Contains(key));
138 
139     // Inserting an object into a GlobalValueMap with the appropriate traits
140     // has the side-effect of making the object weak deep in the guts of V8,
141     // and arranges for it to be cleaned up by the methods in the traits.
142     m_map.Set(key, handle);
143   }
144 
GetMap()145   MapType* GetMap() { return &m_map; }
146 
147  private:
148   MapType m_map;
149 };
150 
151 class CFXJS_ObjDefinition {
152  public:
CFXJS_ObjDefinition(v8::Isolate * isolate,const char * sObjName,FXJSOBJTYPE eObjType,CFXJS_Engine::Constructor pConstructor,CFXJS_Engine::Destructor pDestructor)153   CFXJS_ObjDefinition(v8::Isolate* isolate,
154                       const char* sObjName,
155                       FXJSOBJTYPE eObjType,
156                       CFXJS_Engine::Constructor pConstructor,
157                       CFXJS_Engine::Destructor pDestructor)
158       : m_ObjName(sObjName),
159         m_ObjType(eObjType),
160         m_pConstructor(pConstructor),
161         m_pDestructor(pDestructor),
162         m_pIsolate(isolate) {
163     v8::Isolate::Scope isolate_scope(isolate);
164     v8::HandleScope handle_scope(isolate);
165     v8::Local<v8::FunctionTemplate> fn = v8::FunctionTemplate::New(isolate);
166     fn->InstanceTemplate()->SetInternalFieldCount(2);
167     fn->InstanceTemplate()->SetImmutableProto();
168     fn->SetCallHandler(CallHandler, v8::Number::New(isolate, eObjType));
169     if (eObjType == FXJSOBJTYPE_GLOBAL) {
170       fn->InstanceTemplate()->Set(v8::Symbol::GetToStringTag(isolate),
171                                   fxv8::NewStringHelper(isolate, "global"));
172     }
173     m_FunctionTemplate.Reset(isolate, fn);
174     m_Signature.Reset(isolate, v8::Signature::New(isolate, fn));
175   }
176 
CallHandler(const v8::FunctionCallbackInfo<v8::Value> & info)177   static void CallHandler(const v8::FunctionCallbackInfo<v8::Value>& info) {
178     v8::Isolate* isolate = info.GetIsolate();
179     if (!info.IsConstructCall()) {
180       fxv8::ThrowExceptionHelper(isolate, "illegal constructor");
181       return;
182     }
183     if (info.Data().As<v8::Int32>()->Value() != FXJSOBJTYPE_DYNAMIC) {
184       fxv8::ThrowExceptionHelper(isolate, "not a dynamic object");
185       return;
186     }
187     v8::Local<v8::Object> holder = info.Holder();
188     DCHECK_EQ(holder->InternalFieldCount(), 2);
189     holder->SetAlignedPointerInInternalField(0, nullptr);
190     holder->SetAlignedPointerInInternalField(1, nullptr);
191   }
192 
GetObjType() const193   FXJSOBJTYPE GetObjType() const { return m_ObjType; }
GetObjName() const194   const char* GetObjName() const { return m_ObjName; }
GetIsolate() const195   v8::Isolate* GetIsolate() const { return m_pIsolate; }
196 
DefineConst(const char * sConstName,v8::Local<v8::Value> pDefault)197   void DefineConst(const char* sConstName, v8::Local<v8::Value> pDefault) {
198     GetInstanceTemplate()->Set(GetIsolate(), sConstName, pDefault);
199   }
200 
DefineProperty(v8::Local<v8::String> sPropName,v8::AccessorGetterCallback pPropGet,v8::AccessorSetterCallback pPropPut)201   void DefineProperty(v8::Local<v8::String> sPropName,
202                       v8::AccessorGetterCallback pPropGet,
203                       v8::AccessorSetterCallback pPropPut) {
204     GetInstanceTemplate()->SetAccessor(sPropName, pPropGet, pPropPut);
205   }
206 
DefineMethod(v8::Local<v8::String> sMethodName,v8::FunctionCallback pMethodCall)207   void DefineMethod(v8::Local<v8::String> sMethodName,
208                     v8::FunctionCallback pMethodCall) {
209     v8::Local<v8::FunctionTemplate> fun = v8::FunctionTemplate::New(
210         GetIsolate(), pMethodCall, v8::Local<v8::Value>(), GetSignature());
211     fun->RemovePrototype();
212     GetInstanceTemplate()->Set(sMethodName, fun, v8::ReadOnly);
213   }
214 
DefineAllProperties(v8::GenericNamedPropertyQueryCallback pPropQurey,v8::GenericNamedPropertyGetterCallback pPropGet,v8::GenericNamedPropertySetterCallback pPropPut,v8::GenericNamedPropertyDeleterCallback pPropDel,v8::GenericNamedPropertyEnumeratorCallback pPropEnum)215   void DefineAllProperties(
216       v8::GenericNamedPropertyQueryCallback pPropQurey,
217       v8::GenericNamedPropertyGetterCallback pPropGet,
218       v8::GenericNamedPropertySetterCallback pPropPut,
219       v8::GenericNamedPropertyDeleterCallback pPropDel,
220       v8::GenericNamedPropertyEnumeratorCallback pPropEnum) {
221     GetInstanceTemplate()->SetHandler(v8::NamedPropertyHandlerConfiguration(
222         pPropGet, pPropPut, pPropQurey, pPropDel, pPropEnum,
223         v8::Local<v8::Value>(),
224         v8::PropertyHandlerFlags::kOnlyInterceptStrings));
225   }
226 
GetInstanceTemplate()227   v8::Local<v8::ObjectTemplate> GetInstanceTemplate() {
228     v8::EscapableHandleScope scope(GetIsolate());
229     v8::Local<v8::FunctionTemplate> function =
230         m_FunctionTemplate.Get(GetIsolate());
231     return scope.Escape(function->InstanceTemplate());
232   }
233 
GetSignature()234   v8::Local<v8::Signature> GetSignature() {
235     v8::EscapableHandleScope scope(GetIsolate());
236     return scope.Escape(m_Signature.Get(GetIsolate()));
237   }
238 
RunConstructor(CFXJS_Engine * pEngine,v8::Local<v8::Object> obj,v8::Local<v8::Object> proxy)239   void RunConstructor(CFXJS_Engine* pEngine,
240                       v8::Local<v8::Object> obj,
241                       v8::Local<v8::Object> proxy) {
242     if (m_pConstructor)
243       m_pConstructor(pEngine, obj, proxy);
244   }
245 
RunDestructor(v8::Local<v8::Object> obj)246   void RunDestructor(v8::Local<v8::Object> obj) {
247     if (m_pDestructor)
248       m_pDestructor(obj);
249   }
250 
251  private:
252   UnownedPtr<const char> const m_ObjName;
253   const FXJSOBJTYPE m_ObjType;
254   const CFXJS_Engine::Constructor m_pConstructor;
255   const CFXJS_Engine::Destructor m_pDestructor;
256   UnownedPtr<v8::Isolate> m_pIsolate;
257   v8::Global<v8::FunctionTemplate> m_FunctionTemplate;
258   v8::Global<v8::Signature> m_Signature;
259 };
260 
GetGlobalObjectTemplate(v8::Isolate * pIsolate)261 static v8::Local<v8::ObjectTemplate> GetGlobalObjectTemplate(
262     v8::Isolate* pIsolate) {
263   FXJS_PerIsolateData* pIsolateData = FXJS_PerIsolateData::Get(pIsolate);
264   for (uint32_t i = 1; i <= pIsolateData->CurrentMaxObjDefinitionID(); ++i) {
265     CFXJS_ObjDefinition* pObjDef = pIsolateData->ObjDefinitionForID(i);
266     if (pObjDef->GetObjType() == FXJSOBJTYPE_GLOBAL)
267       return pObjDef->GetInstanceTemplate();
268   }
269   if (!g_DefaultGlobalObjectTemplate) {
270     v8::Local<v8::ObjectTemplate> hGlobalTemplate =
271         v8::ObjectTemplate::New(pIsolate);
272     hGlobalTemplate->Set(v8::Symbol::GetToStringTag(pIsolate),
273                          fxv8::NewStringHelper(pIsolate, "global"));
274     g_DefaultGlobalObjectTemplate =
275         new v8::Global<v8::ObjectTemplate>(pIsolate, hGlobalTemplate);
276   }
277   return g_DefaultGlobalObjectTemplate->Get(pIsolate);
278 }
279 
Dispose(v8::Isolate * isolate,v8::Global<v8::Object> value,WeakCallbackDataType * key)280 void V8TemplateMapTraits::Dispose(v8::Isolate* isolate,
281                                   v8::Global<v8::Object> value,
282                                   WeakCallbackDataType* key) {
283   v8::Local<v8::Object> obj = value.Get(isolate);
284   if (obj.IsEmpty())
285     return;
286   uint32_t id = CFXJS_Engine::GetObjDefnID(obj);
287   if (id == 0)
288     return;
289   FXJS_PerIsolateData* pIsolateData = FXJS_PerIsolateData::Get(isolate);
290   CFXJS_ObjDefinition* pObjDef = pIsolateData->ObjDefinitionForID(id);
291   if (!pObjDef)
292     return;
293   pObjDef->RunDestructor(obj);
294   CFXJS_Engine::FreeObjectPrivate(obj);
295 }
296 
DisposeWeak(const v8::WeakCallbackInfo<WeakCallbackDataType> & data)297 void V8TemplateMapTraits::DisposeWeak(
298     const v8::WeakCallbackInfo<WeakCallbackDataType>& data) {
299   // TODO(tsepez): this is expected be called during GC.
300 }
301 
MapFromWeakCallbackInfo(const v8::WeakCallbackInfo<WeakCallbackDataType> & info)302 V8TemplateMapTraits::MapType* V8TemplateMapTraits::MapFromWeakCallbackInfo(
303     const v8::WeakCallbackInfo<WeakCallbackDataType>& info) {
304   auto* pIsolateData = FXJS_PerIsolateData::Get(info.GetIsolate());
305   V8TemplateMap* pObjsMap = pIsolateData->GetDynamicObjsMap();
306   return pObjsMap ? pObjsMap->GetMap() : nullptr;
307 }
308 
FXJS_Initialize(unsigned int embedderDataSlot,v8::Isolate * pIsolate)309 void FXJS_Initialize(unsigned int embedderDataSlot, v8::Isolate* pIsolate) {
310   if (g_isolate) {
311     DCHECK_EQ(g_embedderDataSlot, embedderDataSlot);
312     DCHECK_EQ(g_isolate, pIsolate);
313     return;
314   }
315   g_embedderDataSlot = embedderDataSlot;
316   g_isolate = pIsolate;
317 }
318 
FXJS_Release()319 void FXJS_Release() {
320   DCHECK(!g_isolate || g_isolate_ref_count == 0);
321   delete g_DefaultGlobalObjectTemplate;
322   g_DefaultGlobalObjectTemplate = nullptr;
323   g_isolate = nullptr;
324 
325   delete g_arrayBufferAllocator;
326   g_arrayBufferAllocator = nullptr;
327 }
328 
FXJS_GetIsolate(v8::Isolate ** pResultIsolate)329 bool FXJS_GetIsolate(v8::Isolate** pResultIsolate) {
330   if (g_isolate) {
331     *pResultIsolate = g_isolate;
332     return false;
333   }
334   // Provide backwards compatibility when no external isolate.
335   if (!g_arrayBufferAllocator)
336     g_arrayBufferAllocator = new CFX_V8ArrayBufferAllocator();
337   v8::Isolate::CreateParams params;
338   params.array_buffer_allocator = g_arrayBufferAllocator;
339   *pResultIsolate = v8::Isolate::New(params);
340   return true;
341 }
342 
FXJS_GlobalIsolateRefCount()343 size_t FXJS_GlobalIsolateRefCount() {
344   return g_isolate_ref_count;
345 }
346 
347 // static
SetUp(v8::Isolate * pIsolate)348 void FXJS_PerIsolateData::SetUp(v8::Isolate* pIsolate) {
349   if (!pIsolate->GetData(g_embedderDataSlot))
350     pIsolate->SetData(g_embedderDataSlot, new FXJS_PerIsolateData(pIsolate));
351 }
352 
353 // static
Get(v8::Isolate * pIsolate)354 FXJS_PerIsolateData* FXJS_PerIsolateData::Get(v8::Isolate* pIsolate) {
355   auto* result =
356       static_cast<FXJS_PerIsolateData*>(pIsolate->GetData(g_embedderDataSlot));
357   CHECK(result->m_Tag == kPerIsolateDataTag);
358   return result;
359 }
360 
FXJS_PerIsolateData(v8::Isolate * pIsolate)361 FXJS_PerIsolateData::FXJS_PerIsolateData(v8::Isolate* pIsolate)
362     : m_Tag(kPerIsolateDataTag),
363       m_pDynamicObjsMap(std::make_unique<V8TemplateMap>(pIsolate)) {}
364 
365 FXJS_PerIsolateData::~FXJS_PerIsolateData() = default;
366 
CurrentMaxObjDefinitionID() const367 uint32_t FXJS_PerIsolateData::CurrentMaxObjDefinitionID() const {
368   return fxcrt::CollectionSize<uint32_t>(m_ObjectDefnArray);
369 }
370 
ObjDefinitionForID(uint32_t id) const371 CFXJS_ObjDefinition* FXJS_PerIsolateData::ObjDefinitionForID(
372     uint32_t id) const {
373   return id > 0 && id <= CurrentMaxObjDefinitionID()
374              ? m_ObjectDefnArray[id - 1].get()
375              : nullptr;
376 }
377 
AssignIDForObjDefinition(std::unique_ptr<CFXJS_ObjDefinition> pDefn)378 uint32_t FXJS_PerIsolateData::AssignIDForObjDefinition(
379     std::unique_ptr<CFXJS_ObjDefinition> pDefn) {
380   m_ObjectDefnArray.push_back(std::move(pDefn));
381   return CurrentMaxObjDefinitionID();
382 }
383 
CFXJS_Engine()384 CFXJS_Engine::CFXJS_Engine() : CFX_V8(nullptr) {}
385 
CFXJS_Engine(v8::Isolate * pIsolate)386 CFXJS_Engine::CFXJS_Engine(v8::Isolate* pIsolate) : CFX_V8(pIsolate) {}
387 
388 CFXJS_Engine::~CFXJS_Engine() = default;
389 
390 // static
GetObjDefnID(v8::Local<v8::Object> pObj)391 uint32_t CFXJS_Engine::GetObjDefnID(v8::Local<v8::Object> pObj) {
392   CFXJS_PerObjectData* pData = CFXJS_PerObjectData::GetFromObject(pObj);
393   return pData ? pData->GetObjDefnID() : 0;
394 }
395 
396 // static
SetObjectPrivate(v8::Local<v8::Object> pObj,std::unique_ptr<CJS_Object> p)397 void CFXJS_Engine::SetObjectPrivate(v8::Local<v8::Object> pObj,
398                                     std::unique_ptr<CJS_Object> p) {
399   CFXJS_PerObjectData* pPerObjectData =
400       CFXJS_PerObjectData::GetFromObject(pObj);
401   if (pPerObjectData)
402     pPerObjectData->SetPrivate(std::move(p));
403 }
404 
405 // static
FreeObjectPrivate(v8::Local<v8::Object> pObj)406 void CFXJS_Engine::FreeObjectPrivate(v8::Local<v8::Object> pObj) {
407   CFXJS_PerObjectData* pData = CFXJS_PerObjectData::GetFromObject(pObj);
408   pObj->SetAlignedPointerInInternalField(0, nullptr);
409   pObj->SetAlignedPointerInInternalField(1, nullptr);
410   delete pData;
411 }
412 
DefineObj(const char * sObjName,FXJSOBJTYPE eObjType,CFXJS_Engine::Constructor pConstructor,CFXJS_Engine::Destructor pDestructor)413 uint32_t CFXJS_Engine::DefineObj(const char* sObjName,
414                                  FXJSOBJTYPE eObjType,
415                                  CFXJS_Engine::Constructor pConstructor,
416                                  CFXJS_Engine::Destructor pDestructor) {
417   v8::Isolate::Scope isolate_scope(GetIsolate());
418   v8::HandleScope handle_scope(GetIsolate());
419   FXJS_PerIsolateData::SetUp(GetIsolate());
420   FXJS_PerIsolateData* pIsolateData = FXJS_PerIsolateData::Get(GetIsolate());
421   return pIsolateData->AssignIDForObjDefinition(
422       std::make_unique<CFXJS_ObjDefinition>(GetIsolate(), sObjName, eObjType,
423                                             pConstructor, pDestructor));
424 }
425 
DefineObjMethod(uint32_t nObjDefnID,const char * sMethodName,v8::FunctionCallback pMethodCall)426 void CFXJS_Engine::DefineObjMethod(uint32_t nObjDefnID,
427                                    const char* sMethodName,
428                                    v8::FunctionCallback pMethodCall) {
429   v8::Isolate::Scope isolate_scope(GetIsolate());
430   v8::HandleScope handle_scope(GetIsolate());
431   FXJS_PerIsolateData* pIsolateData = FXJS_PerIsolateData::Get(GetIsolate());
432   CFXJS_ObjDefinition* pObjDef = pIsolateData->ObjDefinitionForID(nObjDefnID);
433   pObjDef->DefineMethod(NewString(sMethodName), pMethodCall);
434 }
435 
DefineObjProperty(uint32_t nObjDefnID,const char * sPropName,v8::AccessorGetterCallback pPropGet,v8::AccessorSetterCallback pPropPut)436 void CFXJS_Engine::DefineObjProperty(uint32_t nObjDefnID,
437                                      const char* sPropName,
438                                      v8::AccessorGetterCallback pPropGet,
439                                      v8::AccessorSetterCallback pPropPut) {
440   v8::Isolate::Scope isolate_scope(GetIsolate());
441   v8::HandleScope handle_scope(GetIsolate());
442   FXJS_PerIsolateData* pIsolateData = FXJS_PerIsolateData::Get(GetIsolate());
443   CFXJS_ObjDefinition* pObjDef = pIsolateData->ObjDefinitionForID(nObjDefnID);
444   pObjDef->DefineProperty(NewString(sPropName), pPropGet, pPropPut);
445 }
446 
DefineObjAllProperties(uint32_t nObjDefnID,v8::GenericNamedPropertyQueryCallback pPropQurey,v8::GenericNamedPropertyGetterCallback pPropGet,v8::GenericNamedPropertySetterCallback pPropPut,v8::GenericNamedPropertyDeleterCallback pPropDel,v8::GenericNamedPropertyEnumeratorCallback pPropEnum)447 void CFXJS_Engine::DefineObjAllProperties(
448     uint32_t nObjDefnID,
449     v8::GenericNamedPropertyQueryCallback pPropQurey,
450     v8::GenericNamedPropertyGetterCallback pPropGet,
451     v8::GenericNamedPropertySetterCallback pPropPut,
452     v8::GenericNamedPropertyDeleterCallback pPropDel,
453     v8::GenericNamedPropertyEnumeratorCallback pPropEnum) {
454   v8::Isolate::Scope isolate_scope(GetIsolate());
455   v8::HandleScope handle_scope(GetIsolate());
456   FXJS_PerIsolateData* pIsolateData = FXJS_PerIsolateData::Get(GetIsolate());
457   CFXJS_ObjDefinition* pObjDef = pIsolateData->ObjDefinitionForID(nObjDefnID);
458   pObjDef->DefineAllProperties(pPropQurey, pPropGet, pPropPut, pPropDel,
459                                pPropEnum);
460 }
461 
DefineObjConst(uint32_t nObjDefnID,const char * sConstName,v8::Local<v8::Value> pDefault)462 void CFXJS_Engine::DefineObjConst(uint32_t nObjDefnID,
463                                   const char* sConstName,
464                                   v8::Local<v8::Value> pDefault) {
465   v8::Isolate::Scope isolate_scope(GetIsolate());
466   v8::HandleScope handle_scope(GetIsolate());
467   FXJS_PerIsolateData* pIsolateData = FXJS_PerIsolateData::Get(GetIsolate());
468   CFXJS_ObjDefinition* pObjDef = pIsolateData->ObjDefinitionForID(nObjDefnID);
469   pObjDef->DefineConst(sConstName, pDefault);
470 }
471 
DefineGlobalMethod(const char * sMethodName,v8::FunctionCallback pMethodCall)472 void CFXJS_Engine::DefineGlobalMethod(const char* sMethodName,
473                                       v8::FunctionCallback pMethodCall) {
474   v8::Isolate::Scope isolate_scope(GetIsolate());
475   v8::HandleScope handle_scope(GetIsolate());
476   v8::Local<v8::FunctionTemplate> fun =
477       v8::FunctionTemplate::New(GetIsolate(), pMethodCall);
478   fun->RemovePrototype();
479   GetGlobalObjectTemplate(GetIsolate())
480       ->Set(NewString(sMethodName), fun, v8::ReadOnly);
481 }
482 
DefineGlobalConst(const wchar_t * sConstName,v8::FunctionCallback pConstGetter)483 void CFXJS_Engine::DefineGlobalConst(const wchar_t* sConstName,
484                                      v8::FunctionCallback pConstGetter) {
485   v8::Isolate::Scope isolate_scope(GetIsolate());
486   v8::HandleScope handle_scope(GetIsolate());
487   v8::Local<v8::FunctionTemplate> fun =
488       v8::FunctionTemplate::New(GetIsolate(), pConstGetter);
489   fun->RemovePrototype();
490   GetGlobalObjectTemplate(GetIsolate())
491       ->SetAccessorProperty(NewString(sConstName), fun);
492 }
493 
InitializeEngine()494 void CFXJS_Engine::InitializeEngine() {
495   if (GetIsolate() == g_isolate)
496     ++g_isolate_ref_count;
497 
498   v8::Isolate::Scope isolate_scope(GetIsolate());
499   v8::HandleScope handle_scope(GetIsolate());
500 
501   // This has to happen before we call GetGlobalObjectTemplate because that
502   // method gets the PerIsolateData from GetIsolate().
503   FXJS_PerIsolateData::SetUp(GetIsolate());
504 
505   v8::Local<v8::Context> v8Context = v8::Context::New(
506       GetIsolate(), nullptr, GetGlobalObjectTemplate(GetIsolate()));
507 
508   // May not have the internal fields when called from tests.
509   v8::Local<v8::Object> pThisProxy = v8Context->Global();
510   if (pThisProxy->InternalFieldCount() == 2) {
511     pThisProxy->SetAlignedPointerInInternalField(0, nullptr);
512     pThisProxy->SetAlignedPointerInInternalField(1, nullptr);
513   }
514   v8::Local<v8::Object> pThis = pThisProxy->GetPrototype().As<v8::Object>();
515   if (pThis->InternalFieldCount() == 2) {
516     pThis->SetAlignedPointerInInternalField(0, nullptr);
517     pThis->SetAlignedPointerInInternalField(1, nullptr);
518   }
519 
520   v8::Context::Scope context_scope(v8Context);
521   FXJS_PerIsolateData* pIsolateData = FXJS_PerIsolateData::Get(GetIsolate());
522   uint32_t maxID = pIsolateData->CurrentMaxObjDefinitionID();
523   m_StaticObjects.resize(maxID + 1);
524   for (uint32_t i = 1; i <= maxID; ++i) {
525     CFXJS_ObjDefinition* pObjDef = pIsolateData->ObjDefinitionForID(i);
526     if (pObjDef->GetObjType() == FXJSOBJTYPE_GLOBAL) {
527       CFXJS_PerObjectData::SetNewDataInObject(i, pThis);
528       pObjDef->RunConstructor(this, pThis, pThisProxy);
529     } else if (pObjDef->GetObjType() == FXJSOBJTYPE_STATIC) {
530       v8::Local<v8::String> pObjName = NewString(pObjDef->GetObjName());
531       v8::Local<v8::Object> obj = NewFXJSBoundObject(i, FXJSOBJTYPE_STATIC);
532       if (!obj.IsEmpty()) {
533         v8Context->Global()->Set(v8Context, pObjName, obj).FromJust();
534         m_StaticObjects[i] = v8::Global<v8::Object>(GetIsolate(), obj);
535       }
536     }
537   }
538   m_V8Context.Reset(GetIsolate(), v8Context);
539 }
540 
ReleaseEngine()541 void CFXJS_Engine::ReleaseEngine() {
542   v8::Isolate::Scope isolate_scope(GetIsolate());
543   v8::HandleScope handle_scope(GetIsolate());
544   v8::Local<v8::Context> context = GetV8Context();
545   v8::Context::Scope context_scope(context);
546   FXJS_PerIsolateData* pIsolateData = FXJS_PerIsolateData::Get(GetIsolate());
547   if (!pIsolateData)
548     return;
549 
550   m_ConstArrays.clear();
551 
552   for (uint32_t i = 1; i <= pIsolateData->CurrentMaxObjDefinitionID(); ++i) {
553     CFXJS_ObjDefinition* pObjDef = pIsolateData->ObjDefinitionForID(i);
554     v8::Local<v8::Object> pObj;
555     if (pObjDef->GetObjType() == FXJSOBJTYPE_GLOBAL) {
556       pObj =
557           context->Global()->GetPrototype()->ToObject(context).ToLocalChecked();
558     } else if (!m_StaticObjects[i].IsEmpty()) {
559       pObj = v8::Local<v8::Object>::New(GetIsolate(), m_StaticObjects[i]);
560       m_StaticObjects[i].Reset();
561     }
562     if (!pObj.IsEmpty()) {
563       pObjDef->RunDestructor(pObj);
564       FreeObjectPrivate(pObj);
565     }
566   }
567 
568   m_V8Context.Reset();
569 
570   if (GetIsolate() == g_isolate && --g_isolate_ref_count > 0)
571     return;
572 
573   delete pIsolateData;
574   GetIsolate()->SetData(g_embedderDataSlot, nullptr);
575 }
576 
Execute(const WideString & script)577 absl::optional<IJS_Runtime::JS_Error> CFXJS_Engine::Execute(
578     const WideString& script) {
579   v8::Isolate::Scope isolate_scope(GetIsolate());
580   v8::TryCatch try_catch(GetIsolate());
581   v8::Local<v8::Context> context = GetIsolate()->GetCurrentContext();
582   v8::Local<v8::Script> compiled_script;
583   if (!v8::Script::Compile(context, NewString(script.AsStringView()))
584            .ToLocal(&compiled_script)) {
585     v8::String::Utf8Value error(GetIsolate(), try_catch.Exception());
586     v8::Local<v8::Message> msg = try_catch.Message();
587     int line = -1;
588     int column = -1;
589     std::tie(line, column) = GetLineAndColumnFromError(msg, context);
590     return IJS_Runtime::JS_Error(line, column, WideString::FromUTF8(*error));
591   }
592 
593   v8::Local<v8::Value> result;
594   if (!compiled_script->Run(context).ToLocal(&result)) {
595     v8::String::Utf8Value error(GetIsolate(), try_catch.Exception());
596     auto msg = try_catch.Message();
597     int line = -1;
598     int column = -1;
599     std::tie(line, column) = GetLineAndColumnFromError(msg, context);
600     return IJS_Runtime::JS_Error(line, column, WideString::FromUTF8(*error));
601   }
602   return absl::nullopt;
603 }
604 
NewFXJSBoundObject(uint32_t nObjDefnID,FXJSOBJTYPE type)605 v8::Local<v8::Object> CFXJS_Engine::NewFXJSBoundObject(uint32_t nObjDefnID,
606                                                        FXJSOBJTYPE type) {
607   v8::Isolate::Scope isolate_scope(GetIsolate());
608   v8::Local<v8::Context> context = GetIsolate()->GetCurrentContext();
609   FXJS_PerIsolateData* pData = FXJS_PerIsolateData::Get(GetIsolate());
610   if (!pData)
611     return v8::Local<v8::Object>();
612 
613   CFXJS_ObjDefinition* pObjDef = pData->ObjDefinitionForID(nObjDefnID);
614   if (!pObjDef)
615     return v8::Local<v8::Object>();
616 
617   v8::Local<v8::Object> obj;
618   if (!pObjDef->GetInstanceTemplate()->NewInstance(context).ToLocal(&obj))
619     return v8::Local<v8::Object>();
620 
621   CFXJS_PerObjectData::SetNewDataInObject(nObjDefnID, obj);
622   pObjDef->RunConstructor(this, obj, obj);
623   if (type == FXJSOBJTYPE_DYNAMIC) {
624     auto* pIsolateData = FXJS_PerIsolateData::Get(GetIsolate());
625     V8TemplateMap* pObjsMap = pIsolateData->GetDynamicObjsMap();
626     if (pObjsMap)
627       pObjsMap->SetAndMakeWeak(obj);
628   }
629   return obj;
630 }
631 
GetThisObj()632 v8::Local<v8::Object> CFXJS_Engine::GetThisObj() {
633   v8::Isolate::Scope isolate_scope(GetIsolate());
634   if (!FXJS_PerIsolateData::Get(GetIsolate()))
635     return v8::Local<v8::Object>();
636 
637   // Return the global object.
638   v8::Local<v8::Context> context = GetIsolate()->GetCurrentContext();
639   return context->Global()->GetPrototype()->ToObject(context).ToLocalChecked();
640 }
641 
Error(const WideString & message)642 void CFXJS_Engine::Error(const WideString& message) {
643   fxv8::ThrowExceptionHelper(GetIsolate(), message.AsStringView());
644 }
645 
GetV8Context()646 v8::Local<v8::Context> CFXJS_Engine::GetV8Context() {
647   return v8::Local<v8::Context>::New(GetIsolate(), m_V8Context);
648 }
649 
650 // static
GetObjectPrivate(v8::Isolate * pIsolate,v8::Local<v8::Object> pObj)651 CJS_Object* CFXJS_Engine::GetObjectPrivate(v8::Isolate* pIsolate,
652                                            v8::Local<v8::Object> pObj) {
653   auto* pData = CFXJS_PerObjectData::GetFromObject(pObj);
654   if (pData)
655     return pData->GetPrivate();
656 
657   if (pObj.IsEmpty())
658     return nullptr;
659 
660   // It could be a global proxy object, in which case the prototype holds
661   // the actual bound object.
662   v8::Local<v8::Value> val = pObj->GetPrototype();
663   if (!val->IsObject())
664     return nullptr;
665 
666   auto* pProtoData = CFXJS_PerObjectData::GetFromObject(val.As<v8::Object>());
667   if (!pProtoData)
668     return nullptr;
669 
670   auto* pIsolateData = FXJS_PerIsolateData::Get(pIsolate);
671   if (!pIsolateData)
672     return nullptr;
673 
674   CFXJS_ObjDefinition* pObjDef =
675       pIsolateData->ObjDefinitionForID(pProtoData->GetObjDefnID());
676   if (!pObjDef || pObjDef->GetObjType() != FXJSOBJTYPE_GLOBAL)
677     return nullptr;
678 
679   return pProtoData->GetPrivate();
680 }
681 
GetConstArray(const WideString & name)682 v8::Local<v8::Array> CFXJS_Engine::GetConstArray(const WideString& name) {
683   return v8::Local<v8::Array>::New(GetIsolate(), m_ConstArrays[name]);
684 }
685 
SetConstArray(const WideString & name,v8::Local<v8::Array> array)686 void CFXJS_Engine::SetConstArray(const WideString& name,
687                                  v8::Local<v8::Array> array) {
688   m_ConstArrays[name] = v8::Global<v8::Array>(GetIsolate(), array);
689 }
690