windows-server-2003/windows/advcore/gdiplus/test/comtest/combase.hpp

#ifndef _COMBASE_HPP
#define _COMBASE_HPP


//
// Macros for incrementing or decrementing global component count
//

extern HINSTANCE globalInstanceHandle;
extern LONG globalComponentCount;

#define IncrementGlobalComponentCount() \
        InterlockedIncrement(&globalComponentCount)
    
#define DecrementGlobalComponentCount() \
        InterlockedDecrement(&globalComponentCount)


//
// Template for implementing IUnknown interface
//
// NOTES:
// 1. We can only handle objects that one interface other than IUnknown.
// 2. We do not support aggregation.
//

template <class I> class IUnknownBase : public I
{
public:

    // Constructor: notice that when an object is first
    // created, its reference count is set to 1.

    IUnknownBase<I>()
    {
        refCount = 1;
    }

    // Query interface: note that we can only handle
    // objects that one interface other than IUnknown.

    STDMETHOD(QueryInterface)(REFIID riid, VOID** ppv)
    {
        if (riid == IID_IUnknown)
            *ppv = static_cast<IUnknown*>(this);
        else if (riid == __uuidof(I))
            *ppv = static_cast<I*>(this);
        else
        {
            *ppv = NULL;
            return E_NOINTERFACE;
        }

        reinterpret_cast<IUnknown*>(*ppv)->AddRef();
        return S_OK;
    }

    // Increment reference count

    STDMETHOD_(ULONG, AddRef)(VOID)
    {
        return InterlockedIncrement(&refCount);
    }

    // Decrement reference count

    STDMETHOD_(ULONG, Release)(VOID)
    {
        ULONG count = InterlockedDecrement(&refCount);

        if (count == 0)
            delete this;

        return count;
    }

private:

    LONG refCount;
};


//
// Template for implementing IClassFactory interface
//

template <class T> class IFactoryBase
    : public IUnknownBase<IClassFactory>
{
public:

    // NOTE: We don't count class factory objects in globalComponentCount.
    // This means that the existence of a running class factory is not
    // guaranteed to keep a server loaded in memory.

    // Create a new instance of the component

    STDMETHOD(CreateInstance)(IUnknown* outer, REFIID riid, VOID** ppv)
    {
        // We don't support aggregation

        if (outer != NULL)
            return CLASS_E_NOAGGREGATION;

        // Instantiate a new object

        T* obj = new T;

        if (obj == NULL)
            return E_OUTOFMEMORY;

        // Get the requested interface

        HRESULT hr = obj->QueryInterface(riid, ppv);
        obj->Release();

        return hr;
    }

    // Lock/unlock the component server DLL

    STDMETHOD(LockServer)(BOOL lock)
    {
        if (lock)
        {
            IncrementGlobalComponentCount();
        }
        else
        {
            DecrementGlobalComponentCount();
        }

        return S_OK;
    }
};


//
// Helper function for registering and unregistering a component
//

struct ComponentRegData
{
    const CLSID* clsid;
    const WCHAR* compName;
    const WCHAR* progID;
    const WCHAR* progIDNoVer;
};

HRESULT
RegisterComponent(
    const ComponentRegData& regdata,
    BOOL registerIt
    );

#endif // !_COMBASE_HPP