| /* | |
| * Copyright 2008 Google Inc. | |
| * | |
| * Licensed under the Apache License, Version 2.0 (the "License"); you may not | |
| * use this file except in compliance with the License. You may obtain a copy of | |
| * the License at | |
| * | |
| * http://www.apache.org/licenses/LICENSE-2.0 | |
| * | |
| * Unless required by applicable law or agreed to in writing, software | |
| * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT | |
| * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the | |
| * License for the specific language governing permissions and limitations under | |
| * the License. | |
| */ | |
| #include <iostream> | |
| #include <map> | |
| #include <JavaScriptCore/JavaScriptCore.h> | |
| #include "trace.h" | |
| #ifdef ENABLE_JSVALUE_PROTECTION_CHECKING | |
| static std::map<JSValueRef, int> _protectMap; | |
| #endif | |
| /* | |
| * | |
| */ | |
| void JSValueProtectChecked(JSContextRef jsContext, JSValueRef jsValue) { | |
| JSValueProtect(jsContext, jsValue); | |
| #ifdef ENABLE_JSVALUE_PROTECTION_CHECKING | |
| _protectMap[jsValue]++; | |
| #endif | |
| } | |
| /* | |
| * | |
| */ | |
| void JSValueUnprotectChecked(JSContextRef jsContext, JSValueRef jsValue) { | |
| #ifdef ENABLE_JSVALUE_PROTECTION_CHECKING | |
| // Unrecord in a hash_map | |
| unsigned count = _protectMap[jsValue]; | |
| if (count == 0) { | |
| std::cerr << "[WARNING] JSValueUnprotect called on unprotected JSValueRef (0x" | |
| << std::hex << ((unsigned)jsValue) << ")" << std::endl; | |
| return; | |
| } | |
| _protectMap[jsValue] = count - 1; | |
| #else | |
| JSValueUnprotect(jsContext, jsValue); | |
| #endif | |
| } | |
| bool JSValueIsProtected(JSValueRef jsValue) { | |
| #ifdef ENABLE_JSVALUE_PROTECTION_CHECKING | |
| return _protectMap[jsValue] > 0; | |
| #else | |
| return true; | |
| #endif | |
| } | |
| bool JSValueProtectCheckingIsEnabled() { | |
| #ifdef ENABLE_JSVALUE_PROTECTION_CHECKING | |
| return true; | |
| #else | |
| return false; | |
| #endif | |
| } |