1 files changed, 10 insertions, 4 deletions
diff --git a/Source/JavaScriptCore/dfg/DFGToFTLDeferredCompilationCallback.h b/Source/JavaScriptCore/dfg/DFGToFTLDeferredCompilationCallback.h
index 2a7931d2c..a4d840b20 100644
--- a/Source/JavaScriptCore/dfg/DFGToFTLDeferredCompilationCallback.h
+++ b/Source/JavaScriptCore/dfg/DFGToFTLDeferredCompilationCallback.h
@@ -26,6 +26,8 @@
 #ifndef DFGToFTLDeferredCompilationCallback_h
 #define DFGToFTLDeferredCompilationCallback_h
 
+#include <wtf/Platform.h>
+
 #if ENABLE(FTL_JIT)
 
 #include "DeferredCompilationCallback.h"
@@ -40,15 +42,19 @@ namespace DFG {
 
 class ToFTLDeferredCompilationCallback : public DeferredCompilationCallback {
 protected:
-    ToFTLDeferredCompilationCallback();
+    ToFTLDeferredCompilationCallback(PassRefPtr<CodeBlock> dfgCodeBlock);
 
 public:
     virtual ~ToFTLDeferredCompilationCallback();
 
-    static Ref<ToFTLDeferredCompilationCallback> create();
+    static PassRefPtr<ToFTLDeferredCompilationCallback> create(
+        PassRefPtr<CodeBlock> dfgCodeBlock);
     
-    virtual void compilationDidBecomeReadyAsynchronously(CodeBlock*, CodeBlock* profiledDFGCodeBlock);
-    virtual void compilationDidComplete(CodeBlock*, CodeBlock* profiledDFGCodeBlock, CompilationResult);
+    virtual void compilationDidBecomeReadyAsynchronously(CodeBlock*);
+    virtual void compilationDidComplete(CodeBlock*, CompilationResult);
+
+private:
+    RefPtr<CodeBlock> m_dfgCodeBlock;
 };
 
 } } // namespace JSC::DFG