maxDcb-C2Core/modules/PwSh/AssemblyManager.cpp

#include "AssemblyManager.hpp"


MyAssemblyManager::MyAssemblyManager(void)
{
    count = 0;
    m_assemblyStore = new MyAssemblyStore();
};


MyAssemblyManager::~MyAssemblyManager(void)
{
    delete m_assemblyStore;
};


HRESULT STDMETHODCALLTYPE MyAssemblyManager::QueryInterface(REFIID vTableGuid, void** ppv) 
{
    if (!IsEqualIID(vTableGuid, IID_IUnknown) && !IsEqualIID(vTableGuid, IID_IHostAssemblyManager)) 
    {
        *ppv = 0;
        return E_NOINTERFACE;
    }
    *ppv = this;
    this->AddRef();
    return S_OK;
}


ULONG STDMETHODCALLTYPE MyAssemblyManager::AddRef() 
{
    return(++((MyAssemblyManager*)this)->count);
}


ULONG STDMETHODCALLTYPE MyAssemblyManager::Release() 
{
    if (--((MyAssemblyManager*)this)->count == 0) 
    {
        GlobalFree(this);
        return 0;
    }
    return ((MyAssemblyManager*)this)->count;
}


// This returns a list of assemblies that we are telling the CLR that we want it to handle loading (when/if a load is requested for them)
// We can just return NULL and we will always be asked to load the assembly, but we can tell the CLR to load it in our ProvideAssembly implementation
HRESULT STDMETHODCALLTYPE MyAssemblyManager::GetNonHostStoreAssemblies(ICLRAssemblyReferenceList** ppReferenceList) 
{
    *ppReferenceList = NULL;
    return S_OK;
}


//This is responsible for returning our IHostAssemblyStore implementation
HRESULT STDMETHODCALLTYPE MyAssemblyManager::GetAssemblyStore(IHostAssemblyStore** ppAssemblyStore) 
{
    *ppAssemblyStore = m_assemblyStore;
    return S_OK;
}