blob: 66963b1f673b47f8504e94b32d0761026d8bf3ab [file] [log] [blame]
// 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 "third_party/base/containers/span.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());
}
}
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,
pdfium::span<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;
}
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;
}
m_MapGlobal[propname] = std::move(pNewData);
return CJS_Result::Success();
}