532 lines
18 KiB
C++
532 lines
18 KiB
C++
// Copyright 2014 The PDFium Authors
|
|
// 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/cjs_global.h"
|
|
|
|
#include <memory>
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
#include "core/fxcrt/fx_extension.h"
|
|
#include "fxjs/cfx_globaldata.h"
|
|
#include "fxjs/cfx_keyvalue.h"
|
|
#include "fxjs/cjs_event_context.h"
|
|
#include "fxjs/cjs_object.h"
|
|
#include "fxjs/fxv8.h"
|
|
#include "fxjs/js_define.h"
|
|
#include "fxjs/js_resources.h"
|
|
#include "third_party/base/check.h"
|
|
#include "third_party/base/containers/contains.h"
|
|
#include "v8/include/v8-isolate.h"
|
|
|
|
namespace {
|
|
|
|
ByteString ByteStringFromV8Name(v8::Isolate* pIsolate,
|
|
v8::Local<v8::Name> name) {
|
|
CHECK(name->IsString());
|
|
return fxv8::ToByteString(pIsolate, name.As<v8::String>());
|
|
}
|
|
|
|
} // namespace
|
|
|
|
CJS_Global::JSGlobalData::JSGlobalData() = default;
|
|
|
|
CJS_Global::JSGlobalData::~JSGlobalData() = default;
|
|
|
|
const JSMethodSpec CJS_Global::MethodSpecs[] = {
|
|
{"setPersistent", setPersistent_static}};
|
|
|
|
uint32_t CJS_Global::ObjDefnID = 0;
|
|
|
|
// static
|
|
void CJS_Global::setPersistent_static(
|
|
const v8::FunctionCallbackInfo<v8::Value>& info) {
|
|
JSMethod<CJS_Global, &CJS_Global::setPersistent>("setPersistent", "global",
|
|
info);
|
|
}
|
|
|
|
// static
|
|
void CJS_Global::queryprop_static(
|
|
v8::Local<v8::Name> property,
|
|
const v8::PropertyCallbackInfo<v8::Integer>& info) {
|
|
auto pObj = JSGetObject<CJS_Global>(info.GetIsolate(), info.Holder());
|
|
if (!pObj)
|
|
return;
|
|
|
|
ByteString bsProp = ByteStringFromV8Name(info.GetIsolate(), property);
|
|
if (pObj->HasProperty(bsProp))
|
|
info.GetReturnValue().Set(static_cast<int>(v8::PropertyAttribute::None));
|
|
}
|
|
|
|
// static
|
|
void CJS_Global::getprop_static(
|
|
v8::Local<v8::Name> property,
|
|
const v8::PropertyCallbackInfo<v8::Value>& info) {
|
|
auto pObj = JSGetObject<CJS_Global>(info.GetIsolate(), info.Holder());
|
|
if (!pObj)
|
|
return;
|
|
|
|
CJS_Runtime* pRuntime = pObj->GetRuntime();
|
|
if (!pRuntime)
|
|
return;
|
|
|
|
ByteString bsProp = ByteStringFromV8Name(info.GetIsolate(), property);
|
|
CJS_Result result = pObj->GetProperty(pRuntime, bsProp);
|
|
if (result.HasError()) {
|
|
pRuntime->Error(
|
|
JSFormatErrorString("global", "GetProperty", result.Error()));
|
|
return;
|
|
}
|
|
if (result.HasReturn())
|
|
info.GetReturnValue().Set(result.Return());
|
|
}
|
|
|
|
// static
|
|
void CJS_Global::putprop_static(
|
|
v8::Local<v8::Name> property,
|
|
v8::Local<v8::Value> value,
|
|
const v8::PropertyCallbackInfo<v8::Value>& info) {
|
|
auto pObj = JSGetObject<CJS_Global>(info.GetIsolate(), info.Holder());
|
|
if (!pObj)
|
|
return;
|
|
|
|
CJS_Runtime* pRuntime = pObj->GetRuntime();
|
|
if (!pRuntime)
|
|
return;
|
|
|
|
ByteString bsProp = ByteStringFromV8Name(info.GetIsolate(), property);
|
|
CJS_Result result = pObj->SetProperty(pRuntime, bsProp, value);
|
|
if (result.HasError()) {
|
|
pRuntime->Error(
|
|
JSFormatErrorString("global", "PutProperty", result.Error()));
|
|
return;
|
|
}
|
|
info.GetReturnValue().Set(value);
|
|
}
|
|
|
|
// static
|
|
void CJS_Global::delprop_static(
|
|
v8::Local<v8::Name> property,
|
|
const v8::PropertyCallbackInfo<v8::Boolean>& info) {
|
|
auto pObj = JSGetObject<CJS_Global>(info.GetIsolate(), info.Holder());
|
|
if (!pObj)
|
|
return;
|
|
|
|
ByteString bsProp = ByteStringFromV8Name(info.GetIsolate(), property);
|
|
if (pObj->DelProperty(bsProp))
|
|
info.GetReturnValue().Set(true);
|
|
}
|
|
|
|
void CJS_Global::enumprop_static(
|
|
const v8::PropertyCallbackInfo<v8::Array>& info) {
|
|
auto pObj = JSGetObject<CJS_Global>(info.GetIsolate(), info.Holder());
|
|
if (!pObj)
|
|
return;
|
|
|
|
CJS_Runtime* pRuntime = pObj->GetRuntime();
|
|
if (!pRuntime)
|
|
return;
|
|
|
|
pObj->EnumProperties(pRuntime, info);
|
|
}
|
|
|
|
// static
|
|
void CJS_Global::DefineAllProperties(CFXJS_Engine* pEngine) {
|
|
pEngine->DefineObjAllProperties(
|
|
ObjDefnID, CJS_Global::queryprop_static, CJS_Global::getprop_static,
|
|
CJS_Global::putprop_static, CJS_Global::delprop_static,
|
|
CJS_Global::enumprop_static);
|
|
}
|
|
|
|
// static
|
|
uint32_t CJS_Global::GetObjDefnID() {
|
|
return ObjDefnID;
|
|
}
|
|
|
|
// static
|
|
void CJS_Global::DefineJSObjects(CFXJS_Engine* pEngine) {
|
|
ObjDefnID = pEngine->DefineObj("global", FXJSOBJTYPE_STATIC,
|
|
JSConstructor<CJS_Global>, JSDestructor);
|
|
DefineMethods(pEngine, ObjDefnID, MethodSpecs);
|
|
DefineAllProperties(pEngine);
|
|
}
|
|
|
|
CJS_Global::CJS_Global(v8::Local<v8::Object> pObject, CJS_Runtime* pRuntime)
|
|
: CJS_Object(pObject, pRuntime),
|
|
m_pGlobalData(CFX_GlobalData::GetRetainedInstance(nullptr)) {
|
|
UpdateGlobalPersistentVariables();
|
|
}
|
|
|
|
CJS_Global::~CJS_Global() {
|
|
DestroyGlobalPersisitentVariables();
|
|
m_pGlobalData.ExtractAsDangling()->Release();
|
|
}
|
|
|
|
bool CJS_Global::HasProperty(const ByteString& propname) {
|
|
return pdfium::Contains(m_MapGlobal, propname);
|
|
}
|
|
|
|
bool CJS_Global::DelProperty(const ByteString& propname) {
|
|
auto it = m_MapGlobal.find(propname);
|
|
if (it == m_MapGlobal.end())
|
|
return false;
|
|
|
|
it->second->bDeleted = true;
|
|
return true;
|
|
}
|
|
|
|
CJS_Result CJS_Global::GetProperty(CJS_Runtime* pRuntime,
|
|
const ByteString& propname) {
|
|
auto it = m_MapGlobal.find(propname);
|
|
if (it == m_MapGlobal.end())
|
|
return CJS_Result::Success();
|
|
|
|
JSGlobalData* pData = it->second.get();
|
|
if (pData->bDeleted)
|
|
return CJS_Result::Success();
|
|
|
|
switch (pData->nType) {
|
|
case CFX_Value::DataType::kNumber:
|
|
return CJS_Result::Success(pRuntime->NewNumber(pData->dData));
|
|
case CFX_Value::DataType::kBoolean:
|
|
return CJS_Result::Success(pRuntime->NewBoolean(pData->bData));
|
|
case CFX_Value::DataType::kString:
|
|
return CJS_Result::Success(
|
|
pRuntime->NewString(pData->sData.AsStringView()));
|
|
case CFX_Value::DataType::kObject:
|
|
return CJS_Result::Success(
|
|
v8::Local<v8::Object>::New(pRuntime->GetIsolate(), pData->pData));
|
|
case CFX_Value::DataType::kNull:
|
|
return CJS_Result::Success(pRuntime->NewNull());
|
|
default:
|
|
break;
|
|
}
|
|
return CJS_Result::Failure(JSMessage::kObjectTypeError);
|
|
}
|
|
|
|
CJS_Result CJS_Global::SetProperty(CJS_Runtime* pRuntime,
|
|
const ByteString& propname,
|
|
v8::Local<v8::Value> vp) {
|
|
if (vp->IsNumber()) {
|
|
return SetGlobalVariables(propname, CFX_Value::DataType::kNumber,
|
|
pRuntime->ToDouble(vp), false, ByteString(),
|
|
v8::Local<v8::Object>(), false);
|
|
}
|
|
if (vp->IsBoolean()) {
|
|
return SetGlobalVariables(propname, CFX_Value::DataType::kBoolean, 0,
|
|
pRuntime->ToBoolean(vp), ByteString(),
|
|
v8::Local<v8::Object>(), false);
|
|
}
|
|
if (vp->IsString()) {
|
|
return SetGlobalVariables(propname, CFX_Value::DataType::kString, 0, false,
|
|
pRuntime->ToByteString(vp),
|
|
v8::Local<v8::Object>(), false);
|
|
}
|
|
if (vp->IsObject()) {
|
|
return SetGlobalVariables(propname, CFX_Value::DataType::kObject, 0, false,
|
|
ByteString(), pRuntime->ToObject(vp), false);
|
|
}
|
|
if (vp->IsNull()) {
|
|
return SetGlobalVariables(propname, CFX_Value::DataType::kNull, 0, false,
|
|
ByteString(), v8::Local<v8::Object>(), false);
|
|
}
|
|
if (vp->IsUndefined()) {
|
|
DelProperty(propname);
|
|
return CJS_Result::Success();
|
|
}
|
|
return CJS_Result::Failure(JSMessage::kObjectTypeError);
|
|
}
|
|
|
|
void CJS_Global::EnumProperties(
|
|
CJS_Runtime* pRuntime,
|
|
const v8::PropertyCallbackInfo<v8::Array>& info) {
|
|
v8::Local<v8::Array> result = pRuntime->NewArray();
|
|
int idx = 0;
|
|
for (const auto& it : m_MapGlobal) {
|
|
if (it.second->bDeleted)
|
|
continue;
|
|
v8::Local<v8::Name> name = pRuntime->NewString(it.first.AsStringView());
|
|
pRuntime->PutArrayElement(result, idx, name);
|
|
++idx;
|
|
}
|
|
info.GetReturnValue().Set(result);
|
|
}
|
|
|
|
CJS_Result CJS_Global::setPersistent(
|
|
CJS_Runtime* pRuntime,
|
|
const std::vector<v8::Local<v8::Value>>& params) {
|
|
if (params.size() != 2)
|
|
return CJS_Result::Failure(JSMessage::kParamError);
|
|
|
|
auto it = m_MapGlobal.find(pRuntime->ToByteString(params[0]));
|
|
if (it == m_MapGlobal.end() || it->second->bDeleted)
|
|
return CJS_Result::Failure(JSMessage::kGlobalNotFoundError);
|
|
|
|
it->second->bPersistent = pRuntime->ToBoolean(params[1]);
|
|
return CJS_Result::Success();
|
|
}
|
|
|
|
void CJS_Global::UpdateGlobalPersistentVariables() {
|
|
CJS_Runtime* pRuntime = GetRuntime();
|
|
if (!pRuntime)
|
|
return;
|
|
|
|
for (int i = 0, sz = m_pGlobalData->GetSize(); i < sz; i++) {
|
|
CFX_GlobalData::Element* pData = m_pGlobalData->GetAt(i);
|
|
switch (pData->data.nType) {
|
|
case CFX_Value::DataType::kNumber:
|
|
SetGlobalVariables(pData->data.sKey, CFX_Value::DataType::kNumber,
|
|
pData->data.dData, false, ByteString(),
|
|
v8::Local<v8::Object>(), pData->bPersistent);
|
|
pRuntime->PutObjectProperty(ToV8Object(),
|
|
pData->data.sKey.AsStringView(),
|
|
pRuntime->NewNumber(pData->data.dData));
|
|
break;
|
|
case CFX_Value::DataType::kBoolean:
|
|
SetGlobalVariables(pData->data.sKey, CFX_Value::DataType::kBoolean, 0,
|
|
pData->data.bData == 1, ByteString(),
|
|
v8::Local<v8::Object>(), pData->bPersistent);
|
|
pRuntime->PutObjectProperty(
|
|
ToV8Object(), pData->data.sKey.AsStringView(),
|
|
pRuntime->NewBoolean(pData->data.bData == 1));
|
|
break;
|
|
case CFX_Value::DataType::kString:
|
|
SetGlobalVariables(pData->data.sKey, CFX_Value::DataType::kString, 0,
|
|
false, pData->data.sData, v8::Local<v8::Object>(),
|
|
pData->bPersistent);
|
|
pRuntime->PutObjectProperty(
|
|
ToV8Object(), pData->data.sKey.AsStringView(),
|
|
pRuntime->NewString(pData->data.sData.AsStringView()));
|
|
break;
|
|
case CFX_Value::DataType::kObject: {
|
|
v8::Local<v8::Object> pObj = pRuntime->NewObject();
|
|
if (!pObj.IsEmpty()) {
|
|
PutObjectProperty(pObj, &pData->data);
|
|
SetGlobalVariables(pData->data.sKey, CFX_Value::DataType::kObject, 0,
|
|
false, ByteString(), pObj, pData->bPersistent);
|
|
pRuntime->PutObjectProperty(ToV8Object(),
|
|
pData->data.sKey.AsStringView(), pObj);
|
|
}
|
|
} break;
|
|
case CFX_Value::DataType::kNull:
|
|
SetGlobalVariables(pData->data.sKey, CFX_Value::DataType::kNull, 0,
|
|
false, ByteString(), v8::Local<v8::Object>(),
|
|
pData->bPersistent);
|
|
pRuntime->PutObjectProperty(
|
|
ToV8Object(), pData->data.sKey.AsStringView(), pRuntime->NewNull());
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
void CJS_Global::CommitGlobalPersisitentVariables() {
|
|
CJS_Runtime* pRuntime = GetRuntime();
|
|
if (!pRuntime)
|
|
return;
|
|
|
|
for (const auto& iter : m_MapGlobal) {
|
|
ByteString name = iter.first;
|
|
JSGlobalData* pData = iter.second.get();
|
|
if (pData->bDeleted) {
|
|
m_pGlobalData->DeleteGlobalVariable(name);
|
|
continue;
|
|
}
|
|
switch (pData->nType) {
|
|
case CFX_Value::DataType::kNumber:
|
|
m_pGlobalData->SetGlobalVariableNumber(name, pData->dData);
|
|
m_pGlobalData->SetGlobalVariablePersistent(name, pData->bPersistent);
|
|
break;
|
|
case CFX_Value::DataType::kBoolean:
|
|
m_pGlobalData->SetGlobalVariableBoolean(name, pData->bData);
|
|
m_pGlobalData->SetGlobalVariablePersistent(name, pData->bPersistent);
|
|
break;
|
|
case CFX_Value::DataType::kString:
|
|
m_pGlobalData->SetGlobalVariableString(name, pData->sData);
|
|
m_pGlobalData->SetGlobalVariablePersistent(name, pData->bPersistent);
|
|
break;
|
|
case CFX_Value::DataType::kObject: {
|
|
v8::Local<v8::Object> obj =
|
|
v8::Local<v8::Object>::New(pRuntime->GetIsolate(), pData->pData);
|
|
m_pGlobalData->SetGlobalVariableObject(name,
|
|
ObjectToArray(pRuntime, obj));
|
|
m_pGlobalData->SetGlobalVariablePersistent(name, pData->bPersistent);
|
|
} break;
|
|
case CFX_Value::DataType::kNull:
|
|
m_pGlobalData->SetGlobalVariableNull(name);
|
|
m_pGlobalData->SetGlobalVariablePersistent(name, pData->bPersistent);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
std::vector<std::unique_ptr<CFX_KeyValue>> CJS_Global::ObjectToArray(
|
|
CJS_Runtime* pRuntime,
|
|
v8::Local<v8::Object> pObj) {
|
|
std::vector<std::unique_ptr<CFX_KeyValue>> array;
|
|
std::vector<WideString> pKeyList = pRuntime->GetObjectPropertyNames(pObj);
|
|
for (const auto& ws : pKeyList) {
|
|
ByteString sKey = ws.ToUTF8();
|
|
v8::Local<v8::Value> v =
|
|
pRuntime->GetObjectProperty(pObj, sKey.AsStringView());
|
|
if (v->IsNumber()) {
|
|
auto pObjElement = std::make_unique<CFX_KeyValue>();
|
|
pObjElement->nType = CFX_Value::DataType::kNumber;
|
|
pObjElement->sKey = sKey;
|
|
pObjElement->dData = pRuntime->ToDouble(v);
|
|
array.push_back(std::move(pObjElement));
|
|
continue;
|
|
}
|
|
if (v->IsBoolean()) {
|
|
auto pObjElement = std::make_unique<CFX_KeyValue>();
|
|
pObjElement->nType = CFX_Value::DataType::kBoolean;
|
|
pObjElement->sKey = sKey;
|
|
pObjElement->dData = pRuntime->ToBoolean(v);
|
|
array.push_back(std::move(pObjElement));
|
|
continue;
|
|
}
|
|
if (v->IsString()) {
|
|
ByteString sValue = pRuntime->ToByteString(v);
|
|
auto pObjElement = std::make_unique<CFX_KeyValue>();
|
|
pObjElement->nType = CFX_Value::DataType::kString;
|
|
pObjElement->sKey = sKey;
|
|
pObjElement->sData = sValue;
|
|
array.push_back(std::move(pObjElement));
|
|
continue;
|
|
}
|
|
if (v->IsObject()) {
|
|
auto pObjElement = std::make_unique<CFX_KeyValue>();
|
|
pObjElement->nType = CFX_Value::DataType::kObject;
|
|
pObjElement->sKey = sKey;
|
|
pObjElement->objData = ObjectToArray(pRuntime, pRuntime->ToObject(v));
|
|
array.push_back(std::move(pObjElement));
|
|
continue;
|
|
}
|
|
if (v->IsNull()) {
|
|
auto pObjElement = std::make_unique<CFX_KeyValue>();
|
|
pObjElement->nType = CFX_Value::DataType::kNull;
|
|
pObjElement->sKey = sKey;
|
|
array.push_back(std::move(pObjElement));
|
|
}
|
|
}
|
|
return array;
|
|
}
|
|
|
|
void CJS_Global::PutObjectProperty(v8::Local<v8::Object> pObj,
|
|
CFX_KeyValue* pData) {
|
|
CJS_Runtime* pRuntime = GetRuntime();
|
|
if (pRuntime)
|
|
return;
|
|
|
|
for (size_t i = 0; i < pData->objData.size(); ++i) {
|
|
CFX_KeyValue* pObjData = pData->objData.at(i).get();
|
|
switch (pObjData->nType) {
|
|
case CFX_Value::DataType::kNumber:
|
|
pRuntime->PutObjectProperty(pObj, pObjData->sKey.AsStringView(),
|
|
pRuntime->NewNumber(pObjData->dData));
|
|
break;
|
|
case CFX_Value::DataType::kBoolean:
|
|
pRuntime->PutObjectProperty(pObj, pObjData->sKey.AsStringView(),
|
|
pRuntime->NewBoolean(pObjData->bData == 1));
|
|
break;
|
|
case CFX_Value::DataType::kString:
|
|
pRuntime->PutObjectProperty(
|
|
pObj, pObjData->sKey.AsStringView(),
|
|
pRuntime->NewString(pObjData->sData.AsStringView()));
|
|
break;
|
|
case CFX_Value::DataType::kObject: {
|
|
v8::Local<v8::Object> pNewObj = pRuntime->NewObject();
|
|
if (!pNewObj.IsEmpty()) {
|
|
PutObjectProperty(pNewObj, pObjData);
|
|
pRuntime->PutObjectProperty(pObj, pObjData->sKey.AsStringView(),
|
|
pNewObj);
|
|
}
|
|
} break;
|
|
case CFX_Value::DataType::kNull:
|
|
pRuntime->PutObjectProperty(pObj, pObjData->sKey.AsStringView(),
|
|
pRuntime->NewNull());
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
void CJS_Global::DestroyGlobalPersisitentVariables() {
|
|
m_MapGlobal.clear();
|
|
}
|
|
|
|
CJS_Result CJS_Global::SetGlobalVariables(const ByteString& propname,
|
|
CFX_Value::DataType nType,
|
|
double dData,
|
|
bool bData,
|
|
const ByteString& sData,
|
|
v8::Local<v8::Object> pData,
|
|
bool bDefaultPersistent) {
|
|
if (propname.IsEmpty())
|
|
return CJS_Result::Failure(JSMessage::kUnknownProperty);
|
|
|
|
auto it = m_MapGlobal.find(propname);
|
|
if (it != m_MapGlobal.end()) {
|
|
JSGlobalData* pTemp = it->second.get();
|
|
if (pTemp->bDeleted || pTemp->nType != nType) {
|
|
pTemp->dData = 0;
|
|
pTemp->bData = false;
|
|
pTemp->sData.clear();
|
|
pTemp->nType = nType;
|
|
}
|
|
pTemp->bDeleted = false;
|
|
switch (nType) {
|
|
case CFX_Value::DataType::kNumber:
|
|
pTemp->dData = dData;
|
|
break;
|
|
case CFX_Value::DataType::kBoolean:
|
|
pTemp->bData = bData;
|
|
break;
|
|
case CFX_Value::DataType::kString:
|
|
pTemp->sData = sData;
|
|
break;
|
|
case CFX_Value::DataType::kObject:
|
|
pTemp->pData.Reset(pData->GetIsolate(), pData);
|
|
break;
|
|
case CFX_Value::DataType::kNull:
|
|
break;
|
|
default:
|
|
return CJS_Result::Failure(JSMessage::kObjectTypeError);
|
|
}
|
|
return CJS_Result::Success();
|
|
}
|
|
|
|
auto pNewData = std::make_unique<JSGlobalData>();
|
|
switch (nType) {
|
|
case CFX_Value::DataType::kNumber:
|
|
pNewData->nType = CFX_Value::DataType::kNumber;
|
|
pNewData->dData = dData;
|
|
pNewData->bPersistent = bDefaultPersistent;
|
|
break;
|
|
case CFX_Value::DataType::kBoolean:
|
|
pNewData->nType = CFX_Value::DataType::kBoolean;
|
|
pNewData->bData = bData;
|
|
pNewData->bPersistent = bDefaultPersistent;
|
|
break;
|
|
case CFX_Value::DataType::kString:
|
|
pNewData->nType = CFX_Value::DataType::kString;
|
|
pNewData->sData = sData;
|
|
pNewData->bPersistent = bDefaultPersistent;
|
|
break;
|
|
case CFX_Value::DataType::kObject:
|
|
pNewData->nType = CFX_Value::DataType::kObject;
|
|
pNewData->pData.Reset(pData->GetIsolate(), pData);
|
|
pNewData->bPersistent = bDefaultPersistent;
|
|
break;
|
|
case CFX_Value::DataType::kNull:
|
|
pNewData->nType = CFX_Value::DataType::kNull;
|
|
pNewData->bPersistent = bDefaultPersistent;
|
|
break;
|
|
default:
|
|
return CJS_Result::Failure(JSMessage::kObjectTypeError);
|
|
}
|
|
m_MapGlobal[propname] = std::move(pNewData);
|
|
return CJS_Result::Success();
|
|
}
|