| OLD | NEW |
| 1 // Copyright 2014 PDFium Authors. All rights reserved. | 1 // Copyright 2014 PDFium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com | 5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com |
| 6 | 6 |
| 7 #include "JS_GlobalData.h" | 7 #include "JS_GlobalData.h" |
| 8 | 8 |
| 9 #include "core/include/fdrm/fx_crypt.h" | 9 #include "core/include/fdrm/fx_crypt.h" |
| 10 #include "fpdfsdk/include/javascript/IJavaScript.h" | 10 #include "fpdfsdk/include/javascript/IJavaScript.h" |
| 11 | 11 |
| 12 #define JS_MAXGLOBALDATA (1024 * 4 - 8) | 12 #define JS_MAXGLOBALDATA (1024 * 4 - 8) |
| 13 | 13 |
| 14 /* --------------------- CJS_GlobalVariableArray --------------------- */ | 14 /* --------------------- CJS_GlobalVariableArray --------------------- */ |
| 15 | 15 |
| 16 CJS_GlobalVariableArray::CJS_GlobalVariableArray() {} | 16 CJS_GlobalVariableArray::CJS_GlobalVariableArray() {} |
| 17 | 17 |
| 18 CJS_GlobalVariableArray::~CJS_GlobalVariableArray() { | 18 CJS_GlobalVariableArray::~CJS_GlobalVariableArray() { |
| 19 Empty(); | 19 Empty(); |
| 20 } | 20 } |
| 21 | 21 |
| 22 void CJS_GlobalVariableArray::Copy(const CJS_GlobalVariableArray& array) { | 22 void CJS_GlobalVariableArray::Copy(const CJS_GlobalVariableArray& array) { |
| 23 Empty(); | 23 Empty(); |
| 24 for (int i = 0, sz = array.Count(); i < sz; i++) { | 24 for (int i = 0, sz = array.Count(); i < sz; i++) { |
| 25 CJS_KeyValue* pOldObjData = array.GetAt(i); | 25 CJS_KeyValue* pOldObjData = array.GetAt(i); |
| 26 ASSERT(pOldObjData != NULL); | |
| 27 | |
| 28 switch (pOldObjData->nType) { | 26 switch (pOldObjData->nType) { |
| 29 case JS_GLOBALDATA_TYPE_NUMBER: { | 27 case JS_GLOBALDATA_TYPE_NUMBER: { |
| 30 CJS_KeyValue* pNewObjData = new CJS_KeyValue; | 28 CJS_KeyValue* pNewObjData = new CJS_KeyValue; |
| 31 pNewObjData->sKey = pOldObjData->sKey; | 29 pNewObjData->sKey = pOldObjData->sKey; |
| 32 pNewObjData->nType = pOldObjData->nType; | 30 pNewObjData->nType = pOldObjData->nType; |
| 33 pNewObjData->dData = pOldObjData->dData; | 31 pNewObjData->dData = pOldObjData->dData; |
| 34 Add(pNewObjData); | 32 Add(pNewObjData); |
| 35 } break; | 33 } break; |
| 36 case JS_GLOBALDATA_TYPE_BOOLEAN: { | 34 case JS_GLOBALDATA_TYPE_BOOLEAN: { |
| 37 CJS_KeyValue* pNewObjData = new CJS_KeyValue; | 35 CJS_KeyValue* pNewObjData = new CJS_KeyValue; |
| (...skipping 96 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 134 for (int i = 0, sz = m_arrayGlobalData.GetSize(); i < sz; i++) { | 132 for (int i = 0, sz = m_arrayGlobalData.GetSize(); i < sz; i++) { |
| 135 CJS_GlobalData_Element* pTemp = m_arrayGlobalData.GetAt(i); | 133 CJS_GlobalData_Element* pTemp = m_arrayGlobalData.GetAt(i); |
| 136 if (pTemp->data.sKey[0] == *propname && pTemp->data.sKey == propname) | 134 if (pTemp->data.sKey[0] == *propname && pTemp->data.sKey == propname) |
| 137 return i; | 135 return i; |
| 138 } | 136 } |
| 139 return -1; | 137 return -1; |
| 140 } | 138 } |
| 141 | 139 |
| 142 CJS_GlobalData_Element* CJS_GlobalData::GetGlobalVariable( | 140 CJS_GlobalData_Element* CJS_GlobalData::GetGlobalVariable( |
| 143 const FX_CHAR* propname) { | 141 const FX_CHAR* propname) { |
| 144 ASSERT(propname != NULL); | 142 ASSERT(propname); |
| 145 | 143 |
| 146 int nFind = FindGlobalVariable(propname); | 144 int nFind = FindGlobalVariable(propname); |
| 147 if (nFind >= 0) | 145 return nFind >= 0 ? m_arrayGlobalData.GetAt(nFind) : nullptr; |
| 148 return m_arrayGlobalData.GetAt(nFind); | |
| 149 | |
| 150 return NULL; | |
| 151 } | 146 } |
| 152 | 147 |
| 153 void CJS_GlobalData::SetGlobalVariableNumber(const FX_CHAR* propname, | 148 void CJS_GlobalData::SetGlobalVariableNumber(const FX_CHAR* propname, |
| 154 double dData) { | 149 double dData) { |
| 155 ASSERT(propname != NULL); | 150 ASSERT(propname); |
| 156 | |
| 157 CFX_ByteString sPropName = propname; | 151 CFX_ByteString sPropName = propname; |
| 158 sPropName.TrimLeft(); | 152 sPropName.TrimLeft(); |
| 159 sPropName.TrimRight(); | 153 sPropName.TrimRight(); |
| 160 if (sPropName.GetLength() == 0) | 154 if (sPropName.GetLength() == 0) |
| 161 return; | 155 return; |
| 162 | 156 |
| 163 if (CJS_GlobalData_Element* pData = GetGlobalVariable(sPropName)) { | 157 if (CJS_GlobalData_Element* pData = GetGlobalVariable(sPropName)) { |
| 164 pData->data.nType = JS_GLOBALDATA_TYPE_NUMBER; | 158 pData->data.nType = JS_GLOBALDATA_TYPE_NUMBER; |
| 165 pData->data.dData = dData; | 159 pData->data.dData = dData; |
| 166 } else { | 160 } else { |
| 167 CJS_GlobalData_Element* pNewData = new CJS_GlobalData_Element; | 161 CJS_GlobalData_Element* pNewData = new CJS_GlobalData_Element; |
| 168 pNewData->data.sKey = sPropName; | 162 pNewData->data.sKey = sPropName; |
| 169 pNewData->data.nType = JS_GLOBALDATA_TYPE_NUMBER; | 163 pNewData->data.nType = JS_GLOBALDATA_TYPE_NUMBER; |
| 170 pNewData->data.dData = dData; | 164 pNewData->data.dData = dData; |
| 171 m_arrayGlobalData.Add(pNewData); | 165 m_arrayGlobalData.Add(pNewData); |
| 172 } | 166 } |
| 173 } | 167 } |
| 174 | 168 |
| 175 void CJS_GlobalData::SetGlobalVariableBoolean(const FX_CHAR* propname, | 169 void CJS_GlobalData::SetGlobalVariableBoolean(const FX_CHAR* propname, |
| 176 bool bData) { | 170 bool bData) { |
| 177 ASSERT(propname != NULL); | 171 ASSERT(propname); |
| 178 CFX_ByteString sPropName = propname; | 172 CFX_ByteString sPropName = propname; |
| 179 | 173 |
| 180 sPropName.TrimLeft(); | 174 sPropName.TrimLeft(); |
| 181 sPropName.TrimRight(); | 175 sPropName.TrimRight(); |
| 182 | 176 |
| 183 if (sPropName.GetLength() == 0) | 177 if (sPropName.GetLength() == 0) |
| 184 return; | 178 return; |
| 185 | 179 |
| 186 if (CJS_GlobalData_Element* pData = GetGlobalVariable(sPropName)) { | 180 if (CJS_GlobalData_Element* pData = GetGlobalVariable(sPropName)) { |
| 187 pData->data.nType = JS_GLOBALDATA_TYPE_BOOLEAN; | 181 pData->data.nType = JS_GLOBALDATA_TYPE_BOOLEAN; |
| 188 pData->data.bData = bData; | 182 pData->data.bData = bData; |
| 189 } else { | 183 } else { |
| 190 CJS_GlobalData_Element* pNewData = new CJS_GlobalData_Element; | 184 CJS_GlobalData_Element* pNewData = new CJS_GlobalData_Element; |
| 191 pNewData->data.sKey = sPropName; | 185 pNewData->data.sKey = sPropName; |
| 192 pNewData->data.nType = JS_GLOBALDATA_TYPE_BOOLEAN; | 186 pNewData->data.nType = JS_GLOBALDATA_TYPE_BOOLEAN; |
| 193 pNewData->data.bData = bData; | 187 pNewData->data.bData = bData; |
| 194 | 188 |
| 195 m_arrayGlobalData.Add(pNewData); | 189 m_arrayGlobalData.Add(pNewData); |
| 196 } | 190 } |
| 197 } | 191 } |
| 198 | 192 |
| 199 void CJS_GlobalData::SetGlobalVariableString(const FX_CHAR* propname, | 193 void CJS_GlobalData::SetGlobalVariableString(const FX_CHAR* propname, |
| 200 const CFX_ByteString& sData) { | 194 const CFX_ByteString& sData) { |
| 201 ASSERT(propname != NULL); | 195 ASSERT(propname); |
| 202 CFX_ByteString sPropName = propname; | 196 CFX_ByteString sPropName = propname; |
| 203 | 197 |
| 204 sPropName.TrimLeft(); | 198 sPropName.TrimLeft(); |
| 205 sPropName.TrimRight(); | 199 sPropName.TrimRight(); |
| 206 | 200 |
| 207 if (sPropName.GetLength() == 0) | 201 if (sPropName.GetLength() == 0) |
| 208 return; | 202 return; |
| 209 | 203 |
| 210 if (CJS_GlobalData_Element* pData = GetGlobalVariable(sPropName)) { | 204 if (CJS_GlobalData_Element* pData = GetGlobalVariable(sPropName)) { |
| 211 pData->data.nType = JS_GLOBALDATA_TYPE_STRING; | 205 pData->data.nType = JS_GLOBALDATA_TYPE_STRING; |
| 212 pData->data.sData = sData; | 206 pData->data.sData = sData; |
| 213 } else { | 207 } else { |
| 214 CJS_GlobalData_Element* pNewData = new CJS_GlobalData_Element; | 208 CJS_GlobalData_Element* pNewData = new CJS_GlobalData_Element; |
| 215 pNewData->data.sKey = sPropName; | 209 pNewData->data.sKey = sPropName; |
| 216 pNewData->data.nType = JS_GLOBALDATA_TYPE_STRING; | 210 pNewData->data.nType = JS_GLOBALDATA_TYPE_STRING; |
| 217 pNewData->data.sData = sData; | 211 pNewData->data.sData = sData; |
| 218 | 212 |
| 219 m_arrayGlobalData.Add(pNewData); | 213 m_arrayGlobalData.Add(pNewData); |
| 220 } | 214 } |
| 221 } | 215 } |
| 222 | 216 |
| 223 void CJS_GlobalData::SetGlobalVariableObject( | 217 void CJS_GlobalData::SetGlobalVariableObject( |
| 224 const FX_CHAR* propname, | 218 const FX_CHAR* propname, |
| 225 const CJS_GlobalVariableArray& array) { | 219 const CJS_GlobalVariableArray& array) { |
| 226 ASSERT(propname != NULL); | 220 ASSERT(propname); |
| 227 CFX_ByteString sPropName = propname; | 221 CFX_ByteString sPropName = propname; |
| 228 | 222 |
| 229 sPropName.TrimLeft(); | 223 sPropName.TrimLeft(); |
| 230 sPropName.TrimRight(); | 224 sPropName.TrimRight(); |
| 231 | 225 |
| 232 if (sPropName.GetLength() == 0) | 226 if (sPropName.GetLength() == 0) |
| 233 return; | 227 return; |
| 234 | 228 |
| 235 if (CJS_GlobalData_Element* pData = GetGlobalVariable(sPropName)) { | 229 if (CJS_GlobalData_Element* pData = GetGlobalVariable(sPropName)) { |
| 236 pData->data.nType = JS_GLOBALDATA_TYPE_OBJECT; | 230 pData->data.nType = JS_GLOBALDATA_TYPE_OBJECT; |
| 237 pData->data.objData.Copy(array); | 231 pData->data.objData.Copy(array); |
| 238 } else { | 232 } else { |
| 239 CJS_GlobalData_Element* pNewData = new CJS_GlobalData_Element; | 233 CJS_GlobalData_Element* pNewData = new CJS_GlobalData_Element; |
| 240 pNewData->data.sKey = sPropName; | 234 pNewData->data.sKey = sPropName; |
| 241 pNewData->data.nType = JS_GLOBALDATA_TYPE_OBJECT; | 235 pNewData->data.nType = JS_GLOBALDATA_TYPE_OBJECT; |
| 242 pNewData->data.objData.Copy(array); | 236 pNewData->data.objData.Copy(array); |
| 243 | 237 |
| 244 m_arrayGlobalData.Add(pNewData); | 238 m_arrayGlobalData.Add(pNewData); |
| 245 } | 239 } |
| 246 } | 240 } |
| 247 | 241 |
| 248 void CJS_GlobalData::SetGlobalVariableNull(const FX_CHAR* propname) { | 242 void CJS_GlobalData::SetGlobalVariableNull(const FX_CHAR* propname) { |
| 249 ASSERT(propname != NULL); | 243 ASSERT(propname); |
| 250 CFX_ByteString sPropName = propname; | 244 CFX_ByteString sPropName = propname; |
| 251 | 245 |
| 252 sPropName.TrimLeft(); | 246 sPropName.TrimLeft(); |
| 253 sPropName.TrimRight(); | 247 sPropName.TrimRight(); |
| 254 | 248 |
| 255 if (sPropName.GetLength() == 0) | 249 if (sPropName.GetLength() == 0) |
| 256 return; | 250 return; |
| 257 | 251 |
| 258 if (CJS_GlobalData_Element* pData = GetGlobalVariable(sPropName)) { | 252 if (CJS_GlobalData_Element* pData = GetGlobalVariable(sPropName)) { |
| 259 pData->data.nType = JS_GLOBALDATA_TYPE_NULL; | 253 pData->data.nType = JS_GLOBALDATA_TYPE_NULL; |
| 260 } else { | 254 } else { |
| 261 CJS_GlobalData_Element* pNewData = new CJS_GlobalData_Element; | 255 CJS_GlobalData_Element* pNewData = new CJS_GlobalData_Element; |
| 262 pNewData->data.sKey = sPropName; | 256 pNewData->data.sKey = sPropName; |
| 263 pNewData->data.nType = JS_GLOBALDATA_TYPE_NULL; | 257 pNewData->data.nType = JS_GLOBALDATA_TYPE_NULL; |
| 264 | 258 |
| 265 m_arrayGlobalData.Add(pNewData); | 259 m_arrayGlobalData.Add(pNewData); |
| 266 } | 260 } |
| 267 } | 261 } |
| 268 | 262 |
| 269 FX_BOOL CJS_GlobalData::SetGlobalVariablePersistent(const FX_CHAR* propname, | 263 FX_BOOL CJS_GlobalData::SetGlobalVariablePersistent(const FX_CHAR* propname, |
| 270 FX_BOOL bPersistent) { | 264 FX_BOOL bPersistent) { |
| 271 ASSERT(propname != NULL); | 265 ASSERT(propname); |
| 272 CFX_ByteString sPropName = propname; | 266 CFX_ByteString sPropName = propname; |
| 273 | 267 |
| 274 sPropName.TrimLeft(); | 268 sPropName.TrimLeft(); |
| 275 sPropName.TrimRight(); | 269 sPropName.TrimRight(); |
| 276 | 270 |
| 277 if (sPropName.GetLength() == 0) | 271 if (sPropName.GetLength() == 0) |
| 278 return FALSE; | 272 return FALSE; |
| 279 | 273 |
| 280 if (CJS_GlobalData_Element* pData = GetGlobalVariable(sPropName)) { | 274 if (CJS_GlobalData_Element* pData = GetGlobalVariable(sPropName)) { |
| 281 pData->bPersistent = bPersistent; | 275 pData->bPersistent = bPersistent; |
| 282 return TRUE; | 276 return TRUE; |
| 283 } | 277 } |
| 284 | 278 |
| 285 return FALSE; | 279 return FALSE; |
| 286 } | 280 } |
| 287 | 281 |
| 288 FX_BOOL CJS_GlobalData::DeleteGlobalVariable(const FX_CHAR* propname) { | 282 FX_BOOL CJS_GlobalData::DeleteGlobalVariable(const FX_CHAR* propname) { |
| 289 ASSERT(propname != NULL); | 283 ASSERT(propname); |
| 290 CFX_ByteString sPropName = propname; | 284 CFX_ByteString sPropName = propname; |
| 291 | 285 |
| 292 sPropName.TrimLeft(); | 286 sPropName.TrimLeft(); |
| 293 sPropName.TrimRight(); | 287 sPropName.TrimRight(); |
| 294 | 288 |
| 295 if (sPropName.GetLength() == 0) | 289 if (sPropName.GetLength() == 0) |
| 296 return FALSE; | 290 return FALSE; |
| 297 | 291 |
| 298 int nFind = FindGlobalVariable(sPropName); | 292 int nFind = FindGlobalVariable(sPropName); |
| 299 | 293 |
| (...skipping 102 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 402 FX_Free(pBuffer); | 396 FX_Free(pBuffer); |
| 403 } | 397 } |
| 404 } | 398 } |
| 405 | 399 |
| 406 void CJS_GlobalData::SaveGlobalPersisitentVariables() { | 400 void CJS_GlobalData::SaveGlobalPersisitentVariables() { |
| 407 FX_DWORD nCount = 0; | 401 FX_DWORD nCount = 0; |
| 408 CFX_BinaryBuf sData; | 402 CFX_BinaryBuf sData; |
| 409 | 403 |
| 410 for (int i = 0, sz = m_arrayGlobalData.GetSize(); i < sz; i++) { | 404 for (int i = 0, sz = m_arrayGlobalData.GetSize(); i < sz; i++) { |
| 411 CJS_GlobalData_Element* pElement = m_arrayGlobalData.GetAt(i); | 405 CJS_GlobalData_Element* pElement = m_arrayGlobalData.GetAt(i); |
| 412 ASSERT(pElement != NULL); | |
| 413 | |
| 414 if (pElement->bPersistent) { | 406 if (pElement->bPersistent) { |
| 415 CFX_BinaryBuf sElement; | 407 CFX_BinaryBuf sElement; |
| 416 MakeByteString(pElement->data.sKey, &pElement->data, sElement); | 408 MakeByteString(pElement->data.sKey, &pElement->data, sElement); |
| 417 | 409 |
| 418 if (sData.GetSize() + sElement.GetSize() > JS_MAXGLOBALDATA) | 410 if (sData.GetSize() + sElement.GetSize() > JS_MAXGLOBALDATA) |
| 419 break; | 411 break; |
| 420 | 412 |
| 421 sData.AppendBlock(sElement.GetBuffer(), sElement.GetSize()); | 413 sData.AppendBlock(sElement.GetBuffer(), sElement.GetSize()); |
| 422 nCount++; | 414 nCount++; |
| 423 } | 415 } |
| (...skipping 65 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 489 case JS_GLOBALDATA_TYPE_NULL: { | 481 case JS_GLOBALDATA_TYPE_NULL: { |
| 490 FX_DWORD dwNameLen = (FX_DWORD)name.GetLength(); | 482 FX_DWORD dwNameLen = (FX_DWORD)name.GetLength(); |
| 491 sData.AppendBlock(&dwNameLen, sizeof(FX_DWORD)); | 483 sData.AppendBlock(&dwNameLen, sizeof(FX_DWORD)); |
| 492 sData.AppendString(name); | 484 sData.AppendString(name); |
| 493 sData.AppendBlock(&wType, sizeof(FX_DWORD)); | 485 sData.AppendBlock(&wType, sizeof(FX_DWORD)); |
| 494 } break; | 486 } break; |
| 495 default: | 487 default: |
| 496 break; | 488 break; |
| 497 } | 489 } |
| 498 } | 490 } |
| OLD | NEW |