5f4a473c87107a4394c45087a0ac03388cf636e4
[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   if (!m_bInit) {
34     unsigned int embedderDataSlot = 0;
35     if (pApp->GetFormFillInfo()->m_pJsPlatform->version >= 2) {
36       embedderDataSlot =
37           pApp->GetFormFillInfo()->m_pJsPlatform->m_v8EmbedderSlot;
38     }
39     JS_Initial(embedderDataSlot);
40     m_bInit = TRUE;
41   }
42   return new CJS_Runtime(pApp);
43 }
44 void CJS_RuntimeFactory::AddRef() {
45   // to do.Should be implemented as atom manipulation.
46   m_nRef++;
47 }
48 void CJS_RuntimeFactory::Release() {
49   if (m_bInit) {
50     // to do.Should be implemented as atom manipulation.
51     if (--m_nRef == 0) {
52       JS_Release();
53       ReleaseGlobalData();
54       m_bInit = FALSE;
55     }
56   }
57 }
58
59 void CJS_RuntimeFactory::DeleteJSRuntime(IFXJS_Runtime* pRuntime) {
60   delete (CJS_Runtime*)pRuntime;
61 }
62
63 CJS_GlobalData* CJS_RuntimeFactory::NewGlobalData(CPDFDoc_Environment* pApp) {
64   if (m_pGlobalData) {
65     m_nGlobalDataCount++;
66     return m_pGlobalData;
67   }
68   m_nGlobalDataCount = 1;
69   m_pGlobalData = new CJS_GlobalData(pApp);
70   return m_pGlobalData;
71 }
72
73 void CJS_RuntimeFactory::ReleaseGlobalData() {
74   m_nGlobalDataCount--;
75
76   if (m_nGlobalDataCount <= 0) {
77     delete m_pGlobalData;
78     m_pGlobalData = NULL;
79   }
80 }
81
82 void* CJS_ArrayBufferAllocator::Allocate(size_t length) {
83   return calloc(1, length);
84 }
85
86 void* CJS_ArrayBufferAllocator::AllocateUninitialized(size_t length) {
87   return malloc(length);
88 }
89
90 void CJS_ArrayBufferAllocator::Free(void* data, size_t length) {
91   free(data);
92 }
93
94 /* ------------------------------ CJS_Runtime ------------------------------ */
95
96 CJS_Runtime::CJS_Runtime(CPDFDoc_Environment* pApp)
97     : m_pApp(pApp),
98       m_pDocument(NULL),
99       m_bBlocking(FALSE),
100       m_pFieldEventPath(NULL),
101       m_isolate(NULL),
102       m_isolateManaged(false) {
103   if (m_pApp->GetFormFillInfo()->m_pJsPlatform->version >= 2) {
104     m_isolate = reinterpret_cast<v8::Isolate*>(
105         m_pApp->GetFormFillInfo()->m_pJsPlatform->m_isolate);
106   }
107   if (!m_isolate) {
108     m_pArrayBufferAllocator.reset(new CJS_ArrayBufferAllocator());
109
110     v8::Isolate::CreateParams params;
111     params.array_buffer_allocator = m_pArrayBufferAllocator.get();
112     m_isolate = v8::Isolate::New(params);
113     m_isolateManaged = true;
114   }
115
116   DefineJSObjects();
117
118   CJS_Context* pContext = (CJS_Context*)NewContext();
119   JS_InitialRuntime(GetIsolate(), this, pContext, m_context);
120   ReleaseContext(pContext);
121 }
122
123 CJS_Runtime::~CJS_Runtime() {
124   for (int i = 0, sz = m_ContextArray.GetSize(); i < sz; i++)
125     delete m_ContextArray.GetAt(i);
126
127   m_ContextArray.RemoveAll();
128   JS_ReleaseRuntime(GetIsolate(), m_context);
129   RemoveEventsInLoop(m_pFieldEventPath);
130
131   m_pApp = NULL;
132   m_pDocument = NULL;
133   m_pFieldEventPath = NULL;
134   m_context.Reset();
135
136   if (m_isolateManaged)
137     m_isolate->Dispose();
138 }
139
140 FX_BOOL CJS_Runtime::DefineJSObjects() {
141   v8::Isolate::Scope isolate_scope(GetIsolate());
142   v8::HandleScope handle_scope(GetIsolate());
143   v8::Local<v8::Context> context = v8::Context::New(GetIsolate());
144   v8::Context::Scope context_scope(context);
145
146   // The call order determines the "ObjDefID" assigned to each class.
147   // ObjDefIDs 0 - 2
148   if (CJS_Border::DefineJSObjects(GetIsolate(), JS_STATIC) < 0)
149     return FALSE;
150   if (CJS_Display::DefineJSObjects(GetIsolate(), JS_STATIC) < 0)
151     return FALSE;
152   if (CJS_Font::DefineJSObjects(GetIsolate(), JS_STATIC) < 0)
153     return FALSE;
154
155   // ObjDefIDs 3 - 5
156   if (CJS_Highlight::DefineJSObjects(GetIsolate(), JS_STATIC) < 0)
157     return FALSE;
158   if (CJS_Position::DefineJSObjects(GetIsolate(), JS_STATIC) < 0)
159     return FALSE;
160   if (CJS_ScaleHow::DefineJSObjects(GetIsolate(), JS_STATIC) < 0)
161     return FALSE;
162
163   // ObjDefIDs 6 - 8
164   if (CJS_ScaleWhen::DefineJSObjects(GetIsolate(), JS_STATIC) < 0)
165     return FALSE;
166   if (CJS_Style::DefineJSObjects(GetIsolate(), JS_STATIC) < 0)
167     return FALSE;
168   if (CJS_Zoomtype::DefineJSObjects(GetIsolate(), JS_STATIC) < 0)
169     return FALSE;
170
171   // ObjDefIDs 9 - 11
172   if (CJS_App::DefineJSObjects(GetIsolate(), JS_STATIC) < 0)
173     return FALSE;
174   if (CJS_Color::DefineJSObjects(GetIsolate(), JS_STATIC) < 0)
175     return FALSE;
176   if (CJS_Console::DefineJSObjects(GetIsolate(), JS_STATIC) < 0)
177     return FALSE;
178
179   // ObjDefIDs 12 - 14
180   if (CJS_Document::DefineJSObjects(GetIsolate(), JS_DYNAMIC) < 0)
181     return FALSE;
182   if (CJS_Event::DefineJSObjects(GetIsolate(), JS_STATIC) < 0)
183     return FALSE;
184   if (CJS_Field::DefineJSObjects(GetIsolate(), JS_DYNAMIC) < 0)
185     return FALSE;
186
187   // ObjDefIDs 15 - 17
188   if (CJS_Global::DefineJSObjects(GetIsolate(), JS_STATIC) < 0)
189     return FALSE;
190   if (CJS_Icon::DefineJSObjects(GetIsolate(), JS_DYNAMIC) < 0)
191     return FALSE;
192   if (CJS_Util::DefineJSObjects(GetIsolate(), JS_STATIC) < 0)
193     return FALSE;
194
195   // ObjDefIDs 18 - 20
196   if (CJS_PublicMethods::DefineJSObjects(GetIsolate()) < 0)
197     return FALSE;
198   if (CJS_GlobalConsts::DefineJSObjects(GetIsolate()) < 0)
199     return FALSE;
200   if (CJS_GlobalArrays::DefineJSObjects(GetIsolate()) < 0)
201     return FALSE;
202
203   if (CJS_TimerObj::DefineJSObjects(GetIsolate(), JS_DYNAMIC) < 0)
204     return FALSE;
205   if (CJS_PrintParamsObj::DefineJSObjects(GetIsolate(), JS_DYNAMIC) < 0)
206     return FALSE;
207
208   return TRUE;
209 }
210
211 IFXJS_Context* CJS_Runtime::NewContext() {
212   CJS_Context* p = new CJS_Context(this);
213   m_ContextArray.Add(p);
214   return p;
215 }
216
217 void CJS_Runtime::ReleaseContext(IFXJS_Context* pContext) {
218   CJS_Context* pJSContext = (CJS_Context*)pContext;
219
220   for (int i = 0, sz = m_ContextArray.GetSize(); i < sz; i++) {
221     if (pJSContext == m_ContextArray.GetAt(i)) {
222       delete pJSContext;
223       m_ContextArray.RemoveAt(i);
224       break;
225     }
226   }
227 }
228
229 IFXJS_Context* CJS_Runtime::GetCurrentContext() {
230   if (!m_ContextArray.GetSize())
231     return NULL;
232   return m_ContextArray.GetAt(m_ContextArray.GetSize() - 1);
233 }
234
235 void CJS_Runtime::SetReaderDocument(CPDFSDK_Document* pReaderDoc) {
236   if (m_pDocument != pReaderDoc) {
237     v8::Isolate::Scope isolate_scope(m_isolate);
238     v8::HandleScope handle_scope(m_isolate);
239     v8::Local<v8::Context> context =
240         v8::Local<v8::Context>::New(m_isolate, m_context);
241     v8::Context::Scope context_scope(context);
242
243     m_pDocument = pReaderDoc;
244     if (pReaderDoc) {
245       v8::Local<v8::Object> pThis = JS_GetThisObj(GetIsolate());
246       if (!pThis.IsEmpty()) {
247         if (JS_GetObjDefnID(pThis) ==
248             JS_GetObjDefnID(GetIsolate(), L"Document")) {
249           if (CJS_Document* pJSDocument = (CJS_Document*)JS_GetPrivate(pThis)) {
250             if (Document* pDocument = (Document*)pJSDocument->GetEmbedObject())
251               pDocument->AttachDoc(pReaderDoc);
252           }
253         }
254       }
255     }
256   }
257 }
258
259 FX_BOOL CJS_Runtime::AddEventToLoop(const CFX_WideString& sTargetName,
260                                     JS_EVENT_T eEventType) {
261   if (m_pFieldEventPath == NULL) {
262     m_pFieldEventPath = new CJS_FieldEvent;
263     m_pFieldEventPath->sTargetName = sTargetName;
264     m_pFieldEventPath->eEventType = eEventType;
265     m_pFieldEventPath->pNext = NULL;
266
267     return TRUE;
268   }
269
270   // to search
271   CJS_FieldEvent* p = m_pFieldEventPath;
272   CJS_FieldEvent* pLast = m_pFieldEventPath;
273   while (p) {
274     if (p->eEventType == eEventType && p->sTargetName == sTargetName)
275       return FALSE;
276
277     pLast = p;
278     p = p->pNext;
279   }
280
281   // to add
282   CJS_FieldEvent* pNew = new CJS_FieldEvent;
283   pNew->sTargetName = sTargetName;
284   pNew->eEventType = eEventType;
285   pNew->pNext = NULL;
286
287   pLast->pNext = pNew;
288
289   return TRUE;
290 }
291
292 void CJS_Runtime::RemoveEventInLoop(const CFX_WideString& sTargetName,
293                                     JS_EVENT_T eEventType) {
294   FX_BOOL bFind = FALSE;
295
296   CJS_FieldEvent* p = m_pFieldEventPath;
297   CJS_FieldEvent* pLast = NULL;
298   while (p) {
299     if (p->eEventType == eEventType && p->sTargetName == sTargetName) {
300       bFind = TRUE;
301       break;
302     }
303
304     pLast = p;
305     p = p->pNext;
306   }
307
308   if (bFind) {
309     RemoveEventsInLoop(p);
310
311     if (p == m_pFieldEventPath)
312       m_pFieldEventPath = NULL;
313
314     if (pLast)
315       pLast->pNext = NULL;
316   }
317 }
318
319 void CJS_Runtime::RemoveEventsInLoop(CJS_FieldEvent* pStart) {
320   CJS_FieldEvent* p = pStart;
321
322   while (p) {
323     CJS_FieldEvent* pOld = p;
324     p = pOld->pNext;
325
326     delete pOld;
327   }
328 }
329
330 v8::Local<v8::Context> CJS_Runtime::NewJSContext() {
331   return v8::Local<v8::Context>::New(m_isolate, m_context);
332 }
333
334 CFX_WideString ChangeObjName(const CFX_WideString& str) {
335   CFX_WideString sRet = str;
336   sRet.Replace(L"_", L".");
337   return sRet;
338 }