1 files changed, 12 insertions, 7 deletions
diff --git a/Source/JavaScriptCore/dfg/DFGDesiredWeakReferences.h b/Source/JavaScriptCore/dfg/DFGDesiredWeakReferences.h
index 981e752ea..632735cd4 100644
--- a/Source/JavaScriptCore/dfg/DFGDesiredWeakReferences.h
+++ b/Source/JavaScriptCore/dfg/DFGDesiredWeakReferences.h
@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2013 Apple Inc. All rights reserved.
+ * Copyright (C) 2013-2015 Apple Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
@@ -23,10 +23,9 @@
  * THE POSSIBILITY OF SUCH DAMAGE.
  */
 
-#ifndef DFGDesiredWeakReferences_h
-#define DFGDesiredWeakReferences_h
+#pragma once
 
-#include <wtf/Vector.h>
+#include <wtf/HashSet.h>
 
 #if ENABLE(DFG_JIT)
 
@@ -34,6 +33,8 @@ namespace JSC {
 
 class CodeBlock;
 class JSCell;
+class JSValue;
+class SlotVisitor;
 class VM;
 
 namespace DFG {
@@ -42,19 +43,23 @@ class CommonData;
 
 class DesiredWeakReferences {
 public:
+    DesiredWeakReferences();
     DesiredWeakReferences(CodeBlock*);
     ~DesiredWeakReferences();
 
     void addLazily(JSCell*);
+    void addLazily(JSValue);
+    bool contains(JSCell*);
+    
     void reallyAdd(VM&, CommonData*);
+    
+    void visitChildren(SlotVisitor&);
 
 private:
     CodeBlock* m_codeBlock;
-    Vector<JSCell*> m_references;
+    HashSet<JSCell*> m_references;
 };
 
 } } // namespace JSC::DFG
 
 #endif // ENABLE(DFG_JIT)
-
-#endif // DFGDesiredWeakReferences_h