JavaScriptCore-1218.tar.gz

[apple/javascriptcore.git] / API / JSWeakObjectMapRefPrivate.cpp

diff --git a/API/JSWeakObjectMapRefPrivate.cpp b/API/JSWeakObjectMapRefPrivate.cpp
index 13900cf8327b4f026e203f4a12cb58464c203a14..8cbe263a74d6ddbbd8a8aa5297a52ca713efe1ce 100644 (file)
--- a/API/JSWeakObjectMapRefPrivate.cpp
+++ b/API/JSWeakObjectMapRefPrivate.cpp
@@ -28,9 +28,11 @@
 
 #include "APICast.h"
 #include "APIShims.h"
 
 #include "APICast.h"
 #include "APIShims.h"

+#include "JSCJSValue.h"

 #include "JSCallbackObject.h"
 #include "JSCallbackObject.h"

-#include "JSValue.h"

 #include "JSWeakObjectMapRefInternal.h"
 #include "JSWeakObjectMapRefInternal.h"

+#include "Operations.h"
+#include "Weak.h"

 #include <wtf/HashMap.h>
 #include <wtf/text/StringHash.h>
 
 #include <wtf/HashMap.h>
 #include <wtf/text/StringHash.h>
 


@@ -52,27 +54,39 @@ JSWeakObjectMapRef JSWeakObjectMapCreate(JSContextRef context, void* privateData
 
 void JSWeakObjectMapSet(JSContextRef ctx, JSWeakObjectMapRef map, void* key, JSObjectRef object)
 {
 
 void JSWeakObjectMapSet(JSContextRef ctx, JSWeakObjectMapRef map, void* key, JSObjectRef object)
 {

+    if (!ctx) {
+        ASSERT_NOT_REACHED();
+        return;
+    }

     ExecState* exec = toJS(ctx);
     APIEntryShim entryShim(exec);
     JSObject* obj = toJS(object);
     if (!obj)
         return;
     ExecState* exec = toJS(ctx);
     APIEntryShim entryShim(exec);
     JSObject* obj = toJS(object);
     if (!obj)
         return;

-    ASSERT(obj->inherits(&JSCallbackObject<JSGlobalObject>::s_info) || obj->inherits(&JSCallbackObject<JSObjectWithGlobalObject>::s_info));
-    map->map().set(exec->globalData(), key, obj);
+    ASSERT(obj->inherits(&JSCallbackObject<JSGlobalObject>::s_info) || obj->inherits(&JSCallbackObject<JSDestructibleObject>::s_info));
+    map->map().set(key, obj);

 }
 
 JSObjectRef JSWeakObjectMapGet(JSContextRef ctx, JSWeakObjectMapRef map, void* key)
 {
 }
 
 JSObjectRef JSWeakObjectMapGet(JSContextRef ctx, JSWeakObjectMapRef map, void* key)
 {

+    if (!ctx) {
+        ASSERT_NOT_REACHED();
+        return 0;
+    }

     ExecState* exec = toJS(ctx);
     APIEntryShim entryShim(exec);
     ExecState* exec = toJS(ctx);
     APIEntryShim entryShim(exec);

-    return toRef(static_cast<JSObject*>(map->map().get(key)));
+    return toRef(jsCast<JSObject*>(map->map().get(key)));

 }
 
 void JSWeakObjectMapRemove(JSContextRef ctx, JSWeakObjectMapRef map, void* key)
 {
 }
 
 void JSWeakObjectMapRemove(JSContextRef ctx, JSWeakObjectMapRef map, void* key)
 {

+    if (!ctx) {
+        ASSERT_NOT_REACHED();
+        return;
+    }

     ExecState* exec = toJS(ctx);
     APIEntryShim entryShim(exec);
     ExecState* exec = toJS(ctx);
     APIEntryShim entryShim(exec);

-    map->map().take(key);
+    map->map().remove(key);

 }
 
 // We need to keep this function in the build to keep the nightlies running.
 }
 
 // We need to keep this function in the build to keep the nightlies running.