blob: 0af2d361dda4b9fd167664b2ab0f7fc0982fd87e [file] [log] [blame]
// Copyright 2016 PDFium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
#include "fxjs/cfxjse_context.h"
#include <utility>
#include "fxjs/cfxjse_class.h"
#include "fxjs/cfxjse_value.h"
#include "third_party/base/ptr_util.h"
namespace {
const char szCompatibleModeScript[] =
"(function(global, list) {\n"
" 'use strict';\n"
" var objname;\n"
" for (objname in list) {\n"
" var globalobj = global[objname];\n"
" if (globalobj) {\n"
" list[objname].forEach(function(name) {\n"
" if (!globalobj[name]) {\n"
" Object.defineProperty(globalobj, name, {\n"
" writable: true,\n"
" enumerable: false,\n"
" value: (function(obj) {\n"
" if (arguments.length === 0) {\n"
" throw new TypeError('missing argument 0 when calling "
" function ' + objname + '.' + name);\n"
" }\n"
" return globalobj.prototype[name].apply(obj, "
" Array.prototype.slice.call(arguments, 1));\n"
" })\n"
" });\n"
" }\n"
" });\n"
" }\n"
" }\n"
"}(this, {String: ['substr', 'toUpperCase']}));";
wchar_t g_FXJSETagString[] = L"FXJSE_HostObject";
} // namespace
// Note, not in the anonymous namespace due to the friend call
// in cfxjse_context.h
// TODO(dsinclair): Remove the friending, use public methods.
class CFXJSE_ScopeUtil_IsolateHandleContext {
public:
explicit CFXJSE_ScopeUtil_IsolateHandleContext(CFXJSE_Context* pContext)
: m_context(pContext),
m_parent(pContext->m_pIsolate),
m_cscope(v8::Local<v8::Context>::New(pContext->m_pIsolate,
pContext->m_hContext)) {}
v8::Isolate* GetIsolate() { return m_context->m_pIsolate; }
v8::Local<v8::Context> GetLocalContext() {
return v8::Local<v8::Context>::New(m_context->m_pIsolate,
m_context->m_hContext);
}
private:
CFXJSE_ScopeUtil_IsolateHandleContext(
const CFXJSE_ScopeUtil_IsolateHandleContext&) = delete;
void operator=(const CFXJSE_ScopeUtil_IsolateHandleContext&) = delete;
void* operator new(size_t size) = delete;
void operator delete(void*, size_t) = delete;
UnownedPtr<CFXJSE_Context> m_context;
CFXJSE_ScopeUtil_IsolateHandle m_parent;
v8::Context::Scope m_cscope;
};
v8::Local<v8::Object> FXJSE_GetGlobalObjectFromContext(
v8::Local<v8::Context> hContext) {
return hContext->Global()->GetPrototype().As<v8::Object>();
}
void FXJSE_UpdateObjectBinding(v8::Local<v8::Object>& hObject,
CFXJSE_HostObject* lpNewBinding) {
ASSERT(!hObject.IsEmpty());
ASSERT(hObject->InternalFieldCount() == 2);
hObject->SetAlignedPointerInInternalField(0, g_FXJSETagString);
hObject->SetAlignedPointerInInternalField(1, lpNewBinding);
}
CFXJSE_HostObject* FXJSE_RetrieveObjectBinding(v8::Local<v8::Object> hJSObject,
CFXJSE_Class* lpClass) {
ASSERT(!hJSObject.IsEmpty());
if (!hJSObject->IsObject())
return nullptr;
v8::Local<v8::Object> hObject = hJSObject;
if (hObject->InternalFieldCount() != 2) {
v8::Local<v8::Value> hProtoObject = hObject->GetPrototype();
if (hProtoObject.IsEmpty() || !hProtoObject->IsObject())
return nullptr;
hObject = hProtoObject.As<v8::Object>();
if (hObject->InternalFieldCount() != 2)
return nullptr;
}
if (hObject->GetAlignedPointerFromInternalField(0) != g_FXJSETagString)
return nullptr;
if (lpClass) {
v8::Local<v8::FunctionTemplate> hClass =
v8::Local<v8::FunctionTemplate>::New(
lpClass->GetContext()->GetIsolate(), lpClass->GetTemplate());
if (!hClass->HasInstance(hObject))
return nullptr;
}
return static_cast<CFXJSE_HostObject*>(
hObject->GetAlignedPointerFromInternalField(1));
}
v8::Local<v8::Object> FXJSE_CreateReturnValue(v8::Isolate* pIsolate,
v8::TryCatch& trycatch) {
v8::Local<v8::Object> hReturnValue = v8::Object::New(pIsolate);
if (trycatch.HasCaught()) {
v8::Local<v8::Value> hException = trycatch.Exception();
v8::Local<v8::Message> hMessage = trycatch.Message();
if (hException->IsObject()) {
v8::Local<v8::Value> hValue;
hValue = hException.As<v8::Object>()->Get(
v8::String::NewFromUtf8(pIsolate, "name"));
if (hValue->IsString() || hValue->IsStringObject())
hReturnValue->Set(0, hValue);
else
hReturnValue->Set(0, v8::String::NewFromUtf8(pIsolate, "Error"));
hValue = hException.As<v8::Object>()->Get(
v8::String::NewFromUtf8(pIsolate, "message"));
if (hValue->IsString() || hValue->IsStringObject())
hReturnValue->Set(1, hValue);
else
hReturnValue->Set(1, hMessage->Get());
} else {
hReturnValue->Set(0, v8::String::NewFromUtf8(pIsolate, "Error"));
hReturnValue->Set(1, hMessage->Get());
}
hReturnValue->Set(2, hException);
hReturnValue->Set(3, v8::Integer::New(pIsolate, hMessage->GetLineNumber()));
hReturnValue->Set(4, hMessage->GetSourceLine());
v8::Maybe<int32_t> maybe_int =
hMessage->GetStartColumn(pIsolate->GetCurrentContext());
hReturnValue->Set(5, v8::Integer::New(pIsolate, maybe_int.FromMaybe(0)));
maybe_int = hMessage->GetEndColumn(pIsolate->GetCurrentContext());
hReturnValue->Set(6, v8::Integer::New(pIsolate, maybe_int.FromMaybe(0)));
}
return hReturnValue;
}
// static
std::unique_ptr<CFXJSE_Context> CFXJSE_Context::Create(
v8::Isolate* pIsolate,
const FXJSE_CLASS_DESCRIPTOR* pGlobalClass,
CFXJSE_HostObject* pGlobalObject) {
CFXJSE_ScopeUtil_IsolateHandle scope(pIsolate);
auto pContext = pdfium::MakeUnique<CFXJSE_Context>(pIsolate);
CFXJSE_Class* pGlobalClassObj = nullptr;
v8::Local<v8::ObjectTemplate> hObjectTemplate;
if (pGlobalClass) {
pGlobalClassObj = CFXJSE_Class::Create(pContext.get(), pGlobalClass, true);
ASSERT(pGlobalClassObj);
v8::Local<v8::FunctionTemplate> hFunctionTemplate =
v8::Local<v8::FunctionTemplate>::New(pIsolate,
pGlobalClassObj->m_hTemplate);
hObjectTemplate = hFunctionTemplate->InstanceTemplate();
} else {
hObjectTemplate = v8::ObjectTemplate::New(pIsolate);
hObjectTemplate->SetInternalFieldCount(2);
}
hObjectTemplate->Set(
v8::Symbol::GetToStringTag(pIsolate),
v8::String::NewFromUtf8(pIsolate, "global", v8::NewStringType::kNormal)
.ToLocalChecked());
v8::Local<v8::Context> hNewContext =
v8::Context::New(pIsolate, nullptr, hObjectTemplate);
v8::Local<v8::Context> hRootContext = v8::Local<v8::Context>::New(
pIsolate, CFXJSE_RuntimeData::Get(pIsolate)->m_hRootContext);
hNewContext->SetSecurityToken(hRootContext->GetSecurityToken());
v8::Local<v8::Object> hGlobalObject =
FXJSE_GetGlobalObjectFromContext(hNewContext);
FXJSE_UpdateObjectBinding(hGlobalObject, pGlobalObject);
pContext->m_hContext.Reset(pIsolate, hNewContext);
return pContext;
}
CFXJSE_Context::CFXJSE_Context(v8::Isolate* pIsolate) : m_pIsolate(pIsolate) {}
CFXJSE_Context::~CFXJSE_Context() {}
std::unique_ptr<CFXJSE_Value> CFXJSE_Context::GetGlobalObject() {
auto pValue = pdfium::MakeUnique<CFXJSE_Value>(m_pIsolate);
CFXJSE_ScopeUtil_IsolateHandleContext scope(this);
v8::Local<v8::Context> hContext =
v8::Local<v8::Context>::New(m_pIsolate, m_hContext);
v8::Local<v8::Object> hGlobalObject =
FXJSE_GetGlobalObjectFromContext(hContext);
pValue->ForceSetValue(hGlobalObject);
return pValue;
}
v8::Local<v8::Context> CFXJSE_Context::GetContext() {
return v8::Local<v8::Context>::New(m_pIsolate, m_hContext);
}
void CFXJSE_Context::AddClass(std::unique_ptr<CFXJSE_Class> pClass) {
m_rgClasses.push_back(std::move(pClass));
}
CFXJSE_Class* CFXJSE_Context::GetClassByName(
const ByteStringView& szName) const {
auto pClass =
std::find_if(m_rgClasses.begin(), m_rgClasses.end(),
[szName](const std::unique_ptr<CFXJSE_Class>& item) {
return szName == item->m_szClassName;
});
return pClass != m_rgClasses.end() ? pClass->get() : nullptr;
}
void CFXJSE_Context::EnableCompatibleMode() {
ExecuteScript(szCompatibleModeScript, nullptr, nullptr);
}
bool CFXJSE_Context::ExecuteScript(const char* szScript,
CFXJSE_Value* lpRetValue,
CFXJSE_Value* lpNewThisObject) {
CFXJSE_ScopeUtil_IsolateHandleContext scope(this);
v8::TryCatch trycatch(m_pIsolate);
v8::Local<v8::String> hScriptString =
v8::String::NewFromUtf8(m_pIsolate, szScript);
if (!lpNewThisObject) {
v8::Local<v8::Script> hScript = v8::Script::Compile(hScriptString);
if (!trycatch.HasCaught()) {
v8::Local<v8::Value> hValue = hScript->Run();
if (!trycatch.HasCaught()) {
if (lpRetValue)
lpRetValue->m_hValue.Reset(m_pIsolate, hValue);
return true;
}
}
if (lpRetValue) {
lpRetValue->m_hValue.Reset(m_pIsolate,
FXJSE_CreateReturnValue(m_pIsolate, trycatch));
}
return false;
}
v8::Local<v8::Value> hNewThis =
v8::Local<v8::Value>::New(m_pIsolate, lpNewThisObject->m_hValue);
ASSERT(!hNewThis.IsEmpty());
v8::Local<v8::Script> hWrapper = v8::Script::Compile(v8::String::NewFromUtf8(
m_pIsolate, "(function () { return eval(arguments[0]); })"));
v8::Local<v8::Value> hWrapperValue = hWrapper->Run();
ASSERT(hWrapperValue->IsFunction());
v8::Local<v8::Function> hWrapperFn = hWrapperValue.As<v8::Function>();
if (!trycatch.HasCaught()) {
v8::Local<v8::Value> rgArgs[] = {hScriptString};
v8::Local<v8::Value> hValue =
hWrapperFn->Call(hNewThis.As<v8::Object>(), 1, rgArgs);
if (!trycatch.HasCaught()) {
if (lpRetValue)
lpRetValue->m_hValue.Reset(m_pIsolate, hValue);
return true;
}
}
if (lpRetValue) {
lpRetValue->m_hValue.Reset(m_pIsolate,
FXJSE_CreateReturnValue(m_pIsolate, trycatch));
}
return false;
}