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