Don't pass null isolates to FXJS_ when we have a real isolate.
[pdfium.git] / fpdfsdk / src / javascript / JS_Runtime.cpp
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 "../../include/javascript/JavaScript.h"
8 #include "../../include/javascript/IJavaScript.h"
9 #include "../../include/javascript/JS_EventHandler.h"
10 #include "../../include/javascript/JS_Runtime.h"
11 #include "../../include/javascript/JS_Context.h"
12 #include "../../include/javascript/JS_Define.h"
13 #include "../../include/javascript/JS_Object.h"
14 #include "../../include/javascript/JS_Value.h"
15 #include "../../include/javascript/Document.h"
16 #include "../../include/javascript/app.h"
17 #include "../../include/javascript/color.h"
18 #include "../../include/javascript/Consts.h"
19 #include "../../include/javascript/Document.h"
20 #include "../../include/javascript/event.h"
21 #include "../../include/javascript/Field.h"
22 #include "../../include/javascript/Icon.h"
23 #include "../../include/javascript/PublicMethods.h"
24 #include "../../include/javascript/report.h"
25 #include "../../include/javascript/util.h"
26 #include "../../include/javascript/JS_GlobalData.h"
27 #include "../../include/javascript/global.h"
28 #include "../../include/javascript/console.h"
29
30 CJS_RuntimeFactory::~CJS_RuntimeFactory() {}
31
32 IFXJS_Runtime* CJS_RuntimeFactory::NewJSRuntime(CPDFDoc_Environment* pApp) {
33   m_bInit = true;
34   return new CJS_Runtime(pApp);
35 }
36 void CJS_RuntimeFactory::AddRef() {
37   m_nRef++;
38 }
39 void CJS_RuntimeFactory::Release() {
40   if (m_bInit) {
41     if (--m_nRef == 0) {
42       FXJS_Release();
43       m_bInit = FALSE;
44     }
45   }
46 }
47
48 void CJS_RuntimeFactory::DeleteJSRuntime(IFXJS_Runtime* pRuntime) {
49   delete (CJS_Runtime*)pRuntime;
50 }
51
52 /* ------------------------------ CJS_Runtime ------------------------------ */
53
54 CJS_Runtime::CJS_Runtime(CPDFDoc_Environment* pApp)
55     : m_pApp(pApp),
56       m_pDocument(NULL),
57       m_bBlocking(FALSE),
58       m_pFieldEventPath(NULL),
59       m_isolate(NULL),
60       m_isolateManaged(false) {
61   unsigned int embedderDataSlot = 0;
62   if (m_pApp->GetFormFillInfo()->m_pJsPlatform->version >= 2) {
63     m_isolate = reinterpret_cast<v8::Isolate*>(
64         m_pApp->GetFormFillInfo()->m_pJsPlatform->m_isolate);
65     embedderDataSlot = pApp->GetFormFillInfo()->m_pJsPlatform->m_v8EmbedderSlot;
66   }
67   if (!m_isolate) {
68     m_pArrayBufferAllocator.reset(new FXJS_ArrayBufferAllocator());
69
70     v8::Isolate::CreateParams params;
71     params.array_buffer_allocator = m_pArrayBufferAllocator.get();
72     m_isolate = v8::Isolate::New(params);
73     m_isolateManaged = true;
74   }
75
76   FXJS_Initialize(embedderDataSlot);
77   DefineJSObjects();
78
79   CJS_Context* pContext = (CJS_Context*)NewContext();
80   FXJS_InitializeRuntime(GetIsolate(), this, pContext, m_context);
81   ReleaseContext(pContext);
82 }
83
84 CJS_Runtime::~CJS_Runtime() {
85   for (int i = 0, sz = m_ContextArray.GetSize(); i < sz; i++)
86     delete m_ContextArray.GetAt(i);
87
88   m_ContextArray.RemoveAll();
89   FXJS_ReleaseRuntime(GetIsolate(), m_context);
90   RemoveEventsInLoop(m_pFieldEventPath);
91
92   m_pApp = NULL;
93   m_pDocument = NULL;
94   m_pFieldEventPath = NULL;
95   m_context.Reset();
96
97   if (m_isolateManaged)
98     m_isolate->Dispose();
99 }
100
101 void CJS_Runtime::DefineJSObjects() {
102   v8::Isolate::Scope isolate_scope(GetIsolate());
103   v8::HandleScope handle_scope(GetIsolate());
104   v8::Local<v8::Context> context = v8::Context::New(GetIsolate());
105   v8::Context::Scope context_scope(context);
106
107   // The call order determines the "ObjDefID" assigned to each class.
108   // ObjDefIDs 0 - 2
109   CJS_Border::DefineJSObjects(GetIsolate(), FXJS_STATIC);
110   CJS_Display::DefineJSObjects(GetIsolate(), FXJS_STATIC);
111   CJS_Font::DefineJSObjects(GetIsolate(), FXJS_STATIC);
112
113   // ObjDefIDs 3 - 5
114   CJS_Highlight::DefineJSObjects(GetIsolate(), FXJS_STATIC);
115   CJS_Position::DefineJSObjects(GetIsolate(), FXJS_STATIC);
116   CJS_ScaleHow::DefineJSObjects(GetIsolate(), FXJS_STATIC);
117
118   // ObjDefIDs 6 - 8
119   CJS_ScaleWhen::DefineJSObjects(GetIsolate(), FXJS_STATIC);
120   CJS_Style::DefineJSObjects(GetIsolate(), FXJS_STATIC);
121   CJS_Zoomtype::DefineJSObjects(GetIsolate(), FXJS_STATIC);
122
123   // ObjDefIDs 9 - 11
124   CJS_App::DefineJSObjects(GetIsolate(), FXJS_STATIC);
125   CJS_Color::DefineJSObjects(GetIsolate(), FXJS_STATIC);
126   CJS_Console::DefineJSObjects(GetIsolate(), FXJS_STATIC);
127
128   // ObjDefIDs 12 - 14
129   CJS_Document::DefineJSObjects(GetIsolate(), FXJS_DYNAMIC);
130   CJS_Event::DefineJSObjects(GetIsolate(), FXJS_STATIC);
131   CJS_Field::DefineJSObjects(GetIsolate(), FXJS_DYNAMIC);
132
133   // ObjDefIDs 15 - 17
134   CJS_Global::DefineJSObjects(GetIsolate(), FXJS_STATIC);
135   CJS_Icon::DefineJSObjects(GetIsolate(), FXJS_DYNAMIC);
136   CJS_Util::DefineJSObjects(GetIsolate(), FXJS_STATIC);
137
138   // ObjDefIDs 18 - 20 (these can't fail, return void).
139   CJS_PublicMethods::DefineJSObjects(GetIsolate());
140   CJS_GlobalConsts::DefineJSObjects(GetIsolate());
141   CJS_GlobalArrays::DefineJSObjects(GetIsolate());
142
143   // ObjDefIDs 21 - 22.
144   CJS_TimerObj::DefineJSObjects(GetIsolate(), FXJS_DYNAMIC);
145   CJS_PrintParamsObj::DefineJSObjects(GetIsolate(), FXJS_DYNAMIC);
146 }
147
148 IFXJS_Context* CJS_Runtime::NewContext() {
149   CJS_Context* p = new CJS_Context(this);
150   m_ContextArray.Add(p);
151   return p;
152 }
153
154 void CJS_Runtime::ReleaseContext(IFXJS_Context* pContext) {
155   CJS_Context* pJSContext = (CJS_Context*)pContext;
156
157   for (int i = 0, sz = m_ContextArray.GetSize(); i < sz; i++) {
158     if (pJSContext == m_ContextArray.GetAt(i)) {
159       delete pJSContext;
160       m_ContextArray.RemoveAt(i);
161       break;
162     }
163   }
164 }
165
166 IFXJS_Context* CJS_Runtime::GetCurrentContext() {
167   if (!m_ContextArray.GetSize())
168     return NULL;
169   return m_ContextArray.GetAt(m_ContextArray.GetSize() - 1);
170 }
171
172 void CJS_Runtime::SetReaderDocument(CPDFSDK_Document* pReaderDoc) {
173   if (m_pDocument != pReaderDoc) {
174     v8::Isolate::Scope isolate_scope(m_isolate);
175     v8::HandleScope handle_scope(m_isolate);
176     v8::Local<v8::Context> context =
177         v8::Local<v8::Context>::New(m_isolate, m_context);
178     v8::Context::Scope context_scope(context);
179
180     m_pDocument = pReaderDoc;
181     if (pReaderDoc) {
182       v8::Local<v8::Object> pThis = FXJS_GetThisObj(GetIsolate());
183       if (!pThis.IsEmpty()) {
184         if (FXJS_GetObjDefnID(pThis) ==
185             FXJS_GetObjDefnID(GetIsolate(), L"Document")) {
186           if (CJS_Document* pJSDocument =
187                   (CJS_Document*)FXJS_GetPrivate(GetIsolate(), pThis)) {
188             if (Document* pDocument = (Document*)pJSDocument->GetEmbedObject())
189               pDocument->AttachDoc(pReaderDoc);
190           }
191         }
192       }
193     }
194   }
195 }
196
197 FX_BOOL CJS_Runtime::AddEventToLoop(const CFX_WideString& sTargetName,
198                                     JS_EVENT_T eEventType) {
199   if (m_pFieldEventPath == NULL) {
200     m_pFieldEventPath = new CJS_FieldEvent;
201     m_pFieldEventPath->sTargetName = sTargetName;
202     m_pFieldEventPath->eEventType = eEventType;
203     m_pFieldEventPath->pNext = NULL;
204
205     return TRUE;
206   }
207
208   // to search
209   CJS_FieldEvent* p = m_pFieldEventPath;
210   CJS_FieldEvent* pLast = m_pFieldEventPath;
211   while (p) {
212     if (p->eEventType == eEventType && p->sTargetName == sTargetName)
213       return FALSE;
214
215     pLast = p;
216     p = p->pNext;
217   }
218
219   // to add
220   CJS_FieldEvent* pNew = new CJS_FieldEvent;
221   pNew->sTargetName = sTargetName;
222   pNew->eEventType = eEventType;
223   pNew->pNext = NULL;
224
225   pLast->pNext = pNew;
226
227   return TRUE;
228 }
229
230 void CJS_Runtime::RemoveEventInLoop(const CFX_WideString& sTargetName,
231                                     JS_EVENT_T eEventType) {
232   FX_BOOL bFind = FALSE;
233
234   CJS_FieldEvent* p = m_pFieldEventPath;
235   CJS_FieldEvent* pLast = NULL;
236   while (p) {
237     if (p->eEventType == eEventType && p->sTargetName == sTargetName) {
238       bFind = TRUE;
239       break;
240     }
241
242     pLast = p;
243     p = p->pNext;
244   }
245
246   if (bFind) {
247     RemoveEventsInLoop(p);
248
249     if (p == m_pFieldEventPath)
250       m_pFieldEventPath = NULL;
251
252     if (pLast)
253       pLast->pNext = NULL;
254   }
255 }
256
257 void CJS_Runtime::RemoveEventsInLoop(CJS_FieldEvent* pStart) {
258   CJS_FieldEvent* p = pStart;
259
260   while (p) {
261     CJS_FieldEvent* pOld = p;
262     p = pOld->pNext;
263
264     delete pOld;
265   }
266 }
267
268 v8::Local<v8::Context> CJS_Runtime::NewJSContext() {
269   return v8::Local<v8::Context>::New(m_isolate, m_context);
270 }
271
272 CFX_WideString ChangeObjName(const CFX_WideString& str) {
273   CFX_WideString sRet = str;
274   sRet.Replace(L"_", L".");
275   return sRet;
276 }