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
67   // TODO(raymes): Currently we're freeing per-isolate data everytime a
68   // document is destroyed even though it may be in use by other documents. For
69   // now we leak the per-isolate data (when m_isolateManaged is false) until
70   // crbug.com/531339 is fixed.
71   if (m_isolateManaged)
72     FXJS_ReleaseRuntime(GetIsolate(), m_context);
73
74   m_pApp = NULL;
75   m_pDocument = NULL;
76   m_context.Reset();
77
78   if (m_isolateManaged)
79     m_isolate->Dispose();
80 }
81
82 void CJS_Runtime::DefineJSObjects() {
83   v8::Isolate::Scope isolate_scope(GetIsolate());
84   v8::HandleScope handle_scope(GetIsolate());
85   v8::Local<v8::Context> context = v8::Context::New(GetIsolate());
86   v8::Context::Scope context_scope(context);
87
88   // The call order determines the "ObjDefID" assigned to each class.
89   // ObjDefIDs 0 - 2
90   CJS_Border::DefineJSObjects(GetIsolate(), FXJS_STATIC);
91   CJS_Display::DefineJSObjects(GetIsolate(), FXJS_STATIC);
92   CJS_Font::DefineJSObjects(GetIsolate(), FXJS_STATIC);
93
94   // ObjDefIDs 3 - 5
95   CJS_Highlight::DefineJSObjects(GetIsolate(), FXJS_STATIC);
96   CJS_Position::DefineJSObjects(GetIsolate(), FXJS_STATIC);
97   CJS_ScaleHow::DefineJSObjects(GetIsolate(), FXJS_STATIC);
98
99   // ObjDefIDs 6 - 8
100   CJS_ScaleWhen::DefineJSObjects(GetIsolate(), FXJS_STATIC);
101   CJS_Style::DefineJSObjects(GetIsolate(), FXJS_STATIC);
102   CJS_Zoomtype::DefineJSObjects(GetIsolate(), FXJS_STATIC);
103
104   // ObjDefIDs 9 - 11
105   CJS_App::DefineJSObjects(GetIsolate(), FXJS_STATIC);
106   CJS_Color::DefineJSObjects(GetIsolate(), FXJS_STATIC);
107   CJS_Console::DefineJSObjects(GetIsolate(), FXJS_STATIC);
108
109   // ObjDefIDs 12 - 14
110   CJS_Document::DefineJSObjects(GetIsolate(), FXJS_DYNAMIC);
111   CJS_Event::DefineJSObjects(GetIsolate(), FXJS_STATIC);
112   CJS_Field::DefineJSObjects(GetIsolate(), FXJS_DYNAMIC);
113
114   // ObjDefIDs 15 - 17
115   CJS_Global::DefineJSObjects(GetIsolate(), FXJS_STATIC);
116   CJS_Icon::DefineJSObjects(GetIsolate(), FXJS_DYNAMIC);
117   CJS_Util::DefineJSObjects(GetIsolate(), FXJS_STATIC);
118
119   // ObjDefIDs 18 - 20 (these can't fail, return void).
120   CJS_PublicMethods::DefineJSObjects(GetIsolate());
121   CJS_GlobalConsts::DefineJSObjects(GetIsolate());
122   CJS_GlobalArrays::DefineJSObjects(GetIsolate());
123
124   // ObjDefIDs 21 - 22.
125   CJS_TimerObj::DefineJSObjects(GetIsolate(), FXJS_DYNAMIC);
126   CJS_PrintParamsObj::DefineJSObjects(GetIsolate(), FXJS_DYNAMIC);
127 }
128
129 IFXJS_Context* CJS_Runtime::NewContext() {
130   CJS_Context* p = new CJS_Context(this);
131   m_ContextArray.Add(p);
132   return p;
133 }
134
135 void CJS_Runtime::ReleaseContext(IFXJS_Context* pContext) {
136   CJS_Context* pJSContext = (CJS_Context*)pContext;
137
138   for (int i = 0, sz = m_ContextArray.GetSize(); i < sz; i++) {
139     if (pJSContext == m_ContextArray.GetAt(i)) {
140       delete pJSContext;
141       m_ContextArray.RemoveAt(i);
142       break;
143     }
144   }
145 }
146
147 IFXJS_Context* CJS_Runtime::GetCurrentContext() {
148   if (!m_ContextArray.GetSize())
149     return NULL;
150   return m_ContextArray.GetAt(m_ContextArray.GetSize() - 1);
151 }
152
153 void CJS_Runtime::SetReaderDocument(CPDFSDK_Document* pReaderDoc) {
154   if (m_pDocument != pReaderDoc) {
155     v8::Isolate::Scope isolate_scope(m_isolate);
156     v8::HandleScope handle_scope(m_isolate);
157     v8::Local<v8::Context> context =
158         v8::Local<v8::Context>::New(m_isolate, m_context);
159     v8::Context::Scope context_scope(context);
160
161     m_pDocument = pReaderDoc;
162     if (pReaderDoc) {
163       v8::Local<v8::Object> pThis = FXJS_GetThisObj(GetIsolate());
164       if (!pThis.IsEmpty()) {
165         if (FXJS_GetObjDefnID(pThis) ==
166             FXJS_GetObjDefnID(GetIsolate(), L"Document")) {
167           if (CJS_Document* pJSDocument =
168                   (CJS_Document*)FXJS_GetPrivate(GetIsolate(), pThis)) {
169             if (Document* pDocument = (Document*)pJSDocument->GetEmbedObject())
170               pDocument->AttachDoc(pReaderDoc);
171           }
172         }
173       }
174     }
175   }
176 }
177
178 bool CJS_Runtime::AddEventToSet(const FieldEvent& event) {
179   return m_FieldEventSet.insert(event).second;
180 }
181
182 void CJS_Runtime::RemoveEventFromSet(const FieldEvent& event) {
183   m_FieldEventSet.erase(event);
184 }
185
186 v8::Local<v8::Context> CJS_Runtime::NewJSContext() {
187   return v8::Local<v8::Context>::New(m_isolate, m_context);
188 }
189
190 CFX_WideString ChangeObjName(const CFX_WideString& str) {
191   CFX_WideString sRet = str;
192   sRet.Replace(L"_", L".");
193   return sRet;
194 }