PDBCallback.h

#pragma once
#include <dia2.h>       // IDia* interfaces

class PDBCallback
	: public IDiaLoadCallback2
{
	public:
		//
		// IUnknown
		//

		ULONG STDMETHODCALLTYPE AddRef() override
		{
			return m_RefCount++;
		}

		ULONG STDMETHODCALLTYPE Release() override
		{
			if ((--m_RefCount) == 0)
			{
				delete this;
			}

			return m_RefCount;
		}

		HRESULT STDMETHODCALLTYPE QueryInterface(REFIID Rid, void **Interface) override
		{
			if (Interface == nullptr)
			{
				return E_INVALIDARG;
			}

			if (Rid == __uuidof(IDiaLoadCallback2))
			{
				*Interface = (IDiaLoadCallback2 *)this;
			}
			else if (Rid == __uuidof(IDiaLoadCallback))
			{
				*Interface = (IDiaLoadCallback *)this;
			}
			else if (Rid == __uuidof(IUnknown))
			{
				*Interface = (IUnknown *)this;
			}
			else
			{
				*Interface = nullptr;
			}

			if (*Interface != nullptr)
			{
				AddRef();
				return S_OK;
			}

			return E_NOINTERFACE;
		}

		HRESULT STDMETHODCALLTYPE NotifyDebugDir(
			BOOL fExecutable,
			DWORD cbData,
			BYTE data[]) override // really a const struct _IMAGE_DEBUG_DIRECTORY *
		{
			return S_OK;
		}

		HRESULT STDMETHODCALLTYPE NotifyOpenDBG(
			LPCOLESTR dbgPath,
			HRESULT resultCode) override
		{
			// wprintf(L"opening %s...\n", dbgPath);
			return S_OK;
		}

		HRESULT STDMETHODCALLTYPE NotifyOpenPDB(
			LPCOLESTR pdbPath,
			HRESULT resultCode) override
		{
			// wprintf(L"opening %s...\n", pdbPath);
			return S_OK;
		}

		HRESULT STDMETHODCALLTYPE RestrictRegistryAccess() override
		{
			// return hr != S_OK to prevent querying the registry for symbol search paths
			return S_OK;
		}

		HRESULT STDMETHODCALLTYPE RestrictSymbolServerAccess() override
		{
		  // return hr != S_OK to prevent accessing a symbol server
			return S_OK;
		}

		HRESULT STDMETHODCALLTYPE RestrictOriginalPathAccess() override
		{
			// return hr != S_OK to prevent querying the registry for symbol search paths
			return S_OK;
		}

		HRESULT STDMETHODCALLTYPE RestrictReferencePathAccess() override
		{
			// return hr != S_OK to prevent accessing a symbol server
			return S_OK;
		}

		HRESULT STDMETHODCALLTYPE RestrictDBGAccess() override
		{
			return S_OK;
		}

		HRESULT STDMETHODCALLTYPE RestrictSystemRootAccess() override
		{
			return S_OK;
		}

	private:
		volatile unsigned long m_RefCount = 0;
};