1 // Copyright 2014 PDFium Authors. All rights reserved.
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 "fpdfsdk/include/jsapi/fxjs_v8.h" // For per-isolate data.
8 #include "xfa/src/foxitlib.h"
9 #include "fxv8.h"
10 #include "runtime.h"
11 #include "scope_inline.h"
12
13 // Duplicates fpdfsdk's JS_Runtime.h, but keeps XFA from depending on it.
14 // TODO(tsepez): make a single version of this.
15 class FXJSE_ArrayBufferAllocator : public v8::ArrayBuffer::Allocator {
Allocate(size_t length)16 void* Allocate(size_t length) override { return calloc(1, length); }
AllocateUninitialized(size_t length)17 void* AllocateUninitialized(size_t length) override { return malloc(length); }
Free(void * data,size_t length)18 void Free(void* data, size_t length) override { free(data); }
19 };
20
FXJSE_KillV8()21 static void FXJSE_KillV8() {
22 v8::V8::Dispose();
23 }
FXJSE_Initialize()24 void FXJSE_Initialize() {
25 if (!CFXJSE_RuntimeData::g_RuntimeList) {
26 CFXJSE_RuntimeData::g_RuntimeList = new CFXJSE_RuntimeList;
27 }
28 static FX_BOOL bV8Initialized = FALSE;
29 if (bV8Initialized) {
30 return;
31 }
32 bV8Initialized = TRUE;
33 atexit(FXJSE_KillV8);
34 }
FXJSE_Runtime_DisposeCallback(v8::Isolate * pIsolate)35 static void FXJSE_Runtime_DisposeCallback(v8::Isolate* pIsolate) {
36 {
37 v8::Locker locker(pIsolate);
38 if (FXJS_PerIsolateData* pData = FXJS_PerIsolateData::Get(pIsolate)) {
39 delete pData->m_pFXJSERuntimeData;
40 pData->m_pFXJSERuntimeData = nullptr;
41 }
42 }
43 pIsolate->Dispose();
44 }
FXJSE_Finalize()45 void FXJSE_Finalize() {
46 if (CFXJSE_RuntimeData::g_RuntimeList) {
47 CFXJSE_RuntimeData::g_RuntimeList->RemoveAllRuntimes(
48 FXJSE_Runtime_DisposeCallback);
49 delete CFXJSE_RuntimeData::g_RuntimeList;
50 CFXJSE_RuntimeData::g_RuntimeList = NULL;
51 }
52 }
FXJSE_Runtime_Create()53 FXJSE_HRUNTIME FXJSE_Runtime_Create() {
54 v8::Isolate::CreateParams params;
55 params.array_buffer_allocator = new FXJSE_ArrayBufferAllocator();
56 v8::Isolate* pIsolate = v8::Isolate::New(params);
57 ASSERT(pIsolate && CFXJSE_RuntimeData::g_RuntimeList);
58 CFXJSE_RuntimeData::g_RuntimeList->AppendRuntime(pIsolate);
59 return reinterpret_cast<FXJSE_HRUNTIME>(pIsolate);
60 }
FXJSE_Runtime_Release(FXJSE_HRUNTIME hRuntime)61 void FXJSE_Runtime_Release(FXJSE_HRUNTIME hRuntime) {
62 v8::Isolate* pIsolate = reinterpret_cast<v8::Isolate*>(hRuntime);
63 if (pIsolate) {
64 ASSERT(CFXJSE_RuntimeData::g_RuntimeList);
65 CFXJSE_RuntimeData::g_RuntimeList->RemoveRuntime(
66 pIsolate, FXJSE_Runtime_DisposeCallback);
67 }
68 }
Create(v8::Isolate * pIsolate)69 CFXJSE_RuntimeData* CFXJSE_RuntimeData::Create(v8::Isolate* pIsolate) {
70 CFXJSE_RuntimeData* pRuntimeData = new CFXJSE_RuntimeData(pIsolate);
71 CFXJSE_ScopeUtil_IsolateHandle scope(pIsolate);
72 v8::Local<v8::FunctionTemplate> hFuncTemplate =
73 v8::FunctionTemplate::New(pIsolate);
74 v8::Local<v8::Context> hContext =
75 v8::Context::New(pIsolate, 0, hFuncTemplate->InstanceTemplate());
76 hContext->SetSecurityToken(v8::External::New(pIsolate, pIsolate));
77 pRuntimeData->m_hRootContextGlobalTemplate.Reset(pIsolate, hFuncTemplate);
78 pRuntimeData->m_hRootContext.Reset(pIsolate, hContext);
79 return pRuntimeData;
80 }
Get(v8::Isolate * pIsolate)81 CFXJSE_RuntimeData* CFXJSE_RuntimeData::Get(v8::Isolate* pIsolate) {
82 FXJS_PerIsolateData::SetUp(pIsolate);
83 FXJS_PerIsolateData* pData = FXJS_PerIsolateData::Get(pIsolate);
84 if (!pData->m_pFXJSERuntimeData)
85 pData->m_pFXJSERuntimeData = CFXJSE_RuntimeData::Create(pIsolate);
86 return pData->m_pFXJSERuntimeData;
87 }
88 CFXJSE_RuntimeList* CFXJSE_RuntimeData::g_RuntimeList = NULL;
AppendRuntime(v8::Isolate * pIsolate)89 void CFXJSE_RuntimeList::AppendRuntime(v8::Isolate* pIsolate) {
90 m_RuntimeList.Add(pIsolate);
91 }
RemoveRuntime(v8::Isolate * pIsolate,CFXJSE_RuntimeList::RuntimeDisposeCallback lpfnDisposeCallback)92 void CFXJSE_RuntimeList::RemoveRuntime(
93 v8::Isolate* pIsolate,
94 CFXJSE_RuntimeList::RuntimeDisposeCallback lpfnDisposeCallback) {
95 int32_t iIdx = m_RuntimeList.Find(pIsolate, 0);
96 if (iIdx >= 0) {
97 m_RuntimeList.RemoveAt(iIdx, 1);
98 }
99 if (lpfnDisposeCallback) {
100 lpfnDisposeCallback(pIsolate);
101 }
102 }
RemoveAllRuntimes(CFXJSE_RuntimeList::RuntimeDisposeCallback lpfnDisposeCallback)103 void CFXJSE_RuntimeList::RemoveAllRuntimes(
104 CFXJSE_RuntimeList::RuntimeDisposeCallback lpfnDisposeCallback) {
105 int32_t iSize = m_RuntimeList.GetSize();
106 if (lpfnDisposeCallback) {
107 for (int32_t iIdx = 0; iIdx < iSize; iIdx++) {
108 lpfnDisposeCallback(m_RuntimeList[iIdx]);
109 }
110 }
111 m_RuntimeList.RemoveAll();
112 }
113