1 files changed, 130 insertions, 0 deletions
diff --git a/src/inc/fusionsink.h b/src/inc/fusionsink.h
new file mode 100644
index 0000000000..d2b9679e87
--- /dev/null
+++ b/src/inc/fusionsink.h
@@ -0,0 +1,130 @@
+//
+// Copyright (c) Microsoft. All rights reserved.
+// Licensed under the MIT license. See LICENSE file in the project root for full license information.
+//
+
+
+
+/*============================================================
+**
+** Header:  FusionSink.hpp
+**
+** Purpose: Implements FusionSink
+**
+**
+===========================================================*/
+#ifndef _FUSIONSINK_H
+#define _FUSIONSINK_H
+
+#include <fusion.h>
+#include <fusionpriv.h>
+#include "corhlpr.h"
+#include "corpriv.h"
+
+class FusionSink : public IAssemblyBindSink, public INativeImageEvaluate
+{
+public:
+    
+    FusionSink() :
+        m_punk(NULL),
+        m_pNIunk(NULL),
+        m_pAbortUnk(NULL),
+        m_pFusionLog(NULL),
+        m_cRef(1),
+        m_hEvent(NULL),
+        m_LastResult(S_OK)
+    {
+        WRAPPER_NO_CONTRACT;
+    }
+
+    virtual void Reset()
+    {
+        CONTRACTL
+        {
+            INSTANCE_CHECK;
+            NOTHROW;
+        }
+        CONTRACTL_END;
+
+        if(m_pAbortUnk) {
+            m_pAbortUnk->Release();
+            m_pAbortUnk = NULL;
+        }
+
+        if(m_punk) {
+            m_punk->Release();
+            m_punk = NULL;
+        }
+
+        if(m_pNIunk) {
+            m_pNIunk->Release();
+            m_pNIunk = NULL;
+        }
+        
+        if(m_pFusionLog) {
+            m_pFusionLog->Release();
+            m_pFusionLog = NULL;
+        }
+
+        m_LastResult = S_OK;
+    }
+
+    ~FusionSink()
+    {
+        CONTRACTL
+        {
+            DESTRUCTOR_CHECK;
+            NOTHROW;
+        }
+        CONTRACTL_END;
+ 
+        if(m_hEvent) {
+            delete m_hEvent;
+            m_hEvent = NULL;
+        }
+
+        Reset();
+    }
+
+    HRESULT AssemblyResetEvent();
+    HRESULT LastResult()
+    {
+        LIMITED_METHOD_CONTRACT;
+        return m_LastResult;
+    }
+
+    STDMETHODIMP QueryInterface(REFIID riid, void **ppInterface);
+    ULONG STDMETHODCALLTYPE AddRef(void); 
+    ULONG STDMETHODCALLTYPE Release(void);
+    
+    STDMETHODIMP OnProgress(DWORD dwNotification,
+                            HRESULT hrNotification,
+                            LPCWSTR szNotification,
+                            DWORD dwProgress,
+                            DWORD dwProgressMax,
+                            LPVOID pvBindInfo,
+                            IUnknown* punk);
+
+    // Wait on the event.
+    virtual HRESULT Wait();
+
+    STDMETHODIMP Evaluate(
+        IAssembly *pILAssembly, 
+        IAssembly *pNativeAssembly,
+        BYTE * pbCachedData,
+        DWORD dwDataSize);
+    
+    IUnknown*    m_punk;      // Getting an assembly
+    IUnknown*    m_pNIunk;      // Getting an assembly    
+    IUnknown*    m_pAbortUnk; // pUnk for aborting a bind
+    IFusionBindLog *m_pFusionLog;
+
+protected:
+    HRESULT AssemblyCreateEvent();
+
+    LONG        m_cRef;    // Ref count.
+    Event      *m_hEvent;  // Event to block thread.
+    HRESULT     m_LastResult; // Last notification result
+};
+
+#endif  // _FUSIONSINK_H