PowerToys/tools/HandlesExperiment/ContextMenuEntry/ClassFactory.cpp

#include "pch.h"

#include "ClassFactory.h"
#include "ExplorerCommand.h"
#include "dllmain.h"

// Class ctor/dtors

ClassFactory::ClassFactory(_In_ REFCLSID clsid) :
    m_ref_count(1),
    m_clsid(clsid)
{
    ++globals::ref_count;
}

ClassFactory::~ClassFactory()
{
    --globals::ref_count;
}

// Implementations of inherited IUnknown methods

IFACEMETHODIMP ClassFactory::QueryInterface(REFIID riid, void** ppv)
{
    static const QITAB qit[] = {
        QITABENT(ClassFactory, IClassFactory),
        { 0, 0 },
    };
    return QISearch(this, qit, riid, ppv);
}

IFACEMETHODIMP_(ULONG) ClassFactory::AddRef()
{
    return ++m_ref_count;
}

IFACEMETHODIMP_(ULONG) ClassFactory::Release()
{
    auto result = --m_ref_count;
    if (result == 0)
    {
        delete this;
    }
    return result;
}

// Implementations of inherited IClassFactory methods

IFACEMETHODIMP ClassFactory::CreateInstance(IUnknown* pUnkOuter, REFIID riid, void** ppvObject)
{
    *ppvObject = NULL;
    HRESULT hr;
    if (pUnkOuter)
    {
        hr = CLASS_E_NOAGGREGATION;
    }
    else if (m_clsid == __uuidof(ExplorerCommand))
    {
        hr = ExplorerCommand::s_CreateInstance(pUnkOuter, riid, ppvObject);
    }
    else
    {
        hr = CLASS_E_CLASSNOTAVAILABLE;
    }
    return hr;
}

IFACEMETHODIMP ClassFactory::LockServer(BOOL fLock)
{
    if (fLock)
    {
        ++globals::ref_count;
    }
    else
    {
        --globals::ref_count;
    }

    return S_OK;
}