Revert "Leak per-isolate data in pdfium"
[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_Runtime ------------------------------ */
31
32 CJS_Runtime::CJS_Runtime(CPDFDoc_Environment* pApp)
33     : m_pApp(pApp),
34       m_pDocument(NULL),
35       m_bBlocking(FALSE),
36       m_isolate(NULL),
37       m_isolateManaged(false) {
38   unsigned int embedderDataSlot = 0;
39   if (m_pApp->GetFormFillInfo()->m_pJsPlatform->version >= 2) {
40     m_isolate = reinterpret_cast<v8::Isolate*>(
41         m_pApp->GetFormFillInfo()->m_pJsPlatform->m_isolate);
42     embedderDataSlot = pApp->GetFormFillInfo()->m_pJsPlatform->m_v8EmbedderSlot;
43   }
44   if (!m_isolate) {
45     m_pArrayBufferAllocator.reset(new FXJS_ArrayBufferAllocator());
46
47     v8::Isolate::CreateParams params;
48     params.array_buffer_allocator = m_pArrayBufferAllocator.get();
49     m_isolate = v8::Isolate::New(params);
50     m_isolateManaged = true;
51   }
52
53   FXJS_Initialize(embedderDataSlot);
54   DefineJSObjects();
55
56   CJS_Context* pContext = (CJS_Context*)NewContext();
57   FXJS_InitializeRuntime(GetIsolate(), this, pContext, m_context);
58   ReleaseContext(pContext);
59 }
60
61 CJS_Runtime::~CJS_Runtime() {
62   for (int i = 0, sz = m_ContextArray.GetSize(); i < sz; i++)
63     delete m_ContextArray.GetAt(i);
64
65   m_ContextArray.RemoveAll();
66   FXJS_ReleaseRuntime(GetIsolate(), m_context);
67
68   m_pApp = NULL;
69   m_pDocument = NULL;
70   m_context.Reset();
71
72   if (m_isolateManaged)
73     m_isolate->Dispose();
74 }
75
76 void CJS_Runtime::DefineJSObjects() {
77   v8::Isolate::Scope isolate_scope(GetIsolate());
78   v8::HandleScope handle_scope(GetIsolate());
79   v8::Local<v8::Context> context = v8::Context::New(GetIsolate());
80   v8::Context::Scope context_scope(context);
81
82   // The call order determines the "ObjDefID" assigned to each class.
83   // ObjDefIDs 0 - 2
84   CJS_Border::DefineJSObjects(GetIsolate(), FXJS_STATIC);
85   CJS_Display::DefineJSObjects(GetIsolate(), FXJS_STATIC);
86   CJS_Font::DefineJSObjects(GetIsolate(), FXJS_STATIC);
87
88   // ObjDefIDs 3 - 5
89   CJS_Highlight::DefineJSObjects(GetIsolate(), FXJS_STATIC);
90   CJS_Position::DefineJSObjects(GetIsolate(), FXJS_STATIC);
91   CJS_ScaleHow::DefineJSObjects(GetIsolate(), FXJS_STATIC);
92
93   // ObjDefIDs 6 - 8
94   CJS_ScaleWhen::DefineJSObjects(GetIsolate(), FXJS_STATIC);
95   CJS_Style::DefineJSObjects(GetIsolate(), FXJS_STATIC);
96   CJS_Zoomtype::DefineJSObjects(GetIsolate(), FXJS_STATIC);
97
98   // ObjDefIDs 9 - 11
99   CJS_App::DefineJSObjects(GetIsolate(), FXJS_STATIC);
100   CJS_Color::DefineJSObjects(GetIsolate(), FXJS_STATIC);
101   CJS_Console::DefineJSObjects(GetIsolate(), FXJS_STATIC);
102
103   // ObjDefIDs 12 - 14
104   CJS_Document::DefineJSObjects(GetIsolate(), FXJS_DYNAMIC);
105   CJS_Event::DefineJSObjects(GetIsolate(), FXJS_STATIC);
106   CJS_Field::DefineJSObjects(GetIsolate(), FXJS_DYNAMIC);
107
108   // ObjDefIDs 15 - 17
109   CJS_Global::DefineJSObjects(GetIsolate(), FXJS_STATIC);
110   CJS_Icon::DefineJSObjects(GetIsolate(), FXJS_DYNAMIC);
111   CJS_Util::DefineJSObjects(GetIsolate(), FXJS_STATIC);
112
113   // ObjDefIDs 18 - 20 (these can't fail, return void).
114   CJS_PublicMethods::DefineJSObjects(GetIsolate());
115   CJS_GlobalConsts::DefineJSObjects(GetIsolate());
116   CJS_GlobalArrays::DefineJSObjects(GetIsolate());
117
118   // ObjDefIDs 21 - 22.
119   CJS_TimerObj::DefineJSObjects(GetIsolate(), FXJS_DYNAMIC);
120   CJS_PrintParamsObj::DefineJSObjects(GetIsolate(), FXJS_DYNAMIC);
121 }
122
123 IFXJS_Context* CJS_Runtime::NewContext() {
124   CJS_Context* p = new CJS_Context(this);
125   m_ContextArray.Add(p);
126   return p;
127 }
128
129 void CJS_Runtime::ReleaseContext(IFXJS_Context* pContext) {
130   CJS_Context* pJSContext = (CJS_Context*)pContext;
131
132   for (int i = 0, sz = m_ContextArray.GetSize(); i < sz; i++) {
133     if (pJSContext == m_ContextArray.GetAt(i)) {
134       delete pJSContext;
135       m_ContextArray.RemoveAt(i);
136       break;
137     }
138   }
139 }
140
141 IFXJS_Context* CJS_Runtime::GetCurrentContext() {
142   if (!m_ContextArray.GetSize())
143     return NULL;
144   return m_ContextArray.GetAt(m_ContextArray.GetSize() - 1);
145 }
146
147 void CJS_Runtime::SetReaderDocument(CPDFSDK_Document* pReaderDoc) {
148   if (m_pDocument != pReaderDoc) {
149     v8::Isolate::Scope isolate_scope(m_isolate);
150     v8::HandleScope handle_scope(m_isolate);
151     v8::Local<v8::Context> context =
152         v8::Local<v8::Context>::New(m_isolate, m_context);
153     v8::Context::Scope context_scope(context);
154
155     m_pDocument = pReaderDoc;
156     if (pReaderDoc) {
157       v8::Local<v8::Object> pThis = FXJS_GetThisObj(GetIsolate());
158       if (!pThis.IsEmpty()) {
159         if (FXJS_GetObjDefnID(pThis) ==
160             FXJS_GetObjDefnID(GetIsolate(), L"Document")) {
161           if (CJS_Document* pJSDocument =
162                   (CJS_Document*)FXJS_GetPrivate(GetIsolate(), pThis)) {
163             if (Document* pDocument = (Document*)pJSDocument->GetEmbedObject())
164               pDocument->AttachDoc(pReaderDoc);
165           }
166         }
167       }
168     }
169   }
170 }
171
172 bool CJS_Runtime::AddEventToSet(const FieldEvent& event) {
173   return m_FieldEventSet.insert(event).second;
174 }
175
176 void CJS_Runtime::RemoveEventFromSet(const FieldEvent& event) {
177   m_FieldEventSet.erase(event);
178 }
179
180 v8::Local<v8::Context> CJS_Runtime::NewJSContext() {
181   return v8::Local<v8::Context>::New(m_isolate, m_context);
182 }
183
184 CFX_WideString ChangeObjName(const CFX_WideString& str) {
185   CFX_WideString sRet = str;
186   sRet.Replace(L"_", L".");
187   return sRet;
188 }