IE Drop-Down Button

// ArrowButton.cpp : Implementation of CArrowButton

#include "stdafx.h"
#include "ArrowButton.h"


// CArrowButton

STDMETHODIMP CArrowButton::QueryStatus(const GUID* pguidCmdGroup, ULONG cCmds, OLECMD prgCmds[], OLECMDTEXT* pCmdText)
{
	if (cCmds == 0) 
	{
		return E_INVALIDARG;
	}

	if (prgCmds == 0)
	{
		return E_POINTER;
	}

	prgCmds[0].cmdf = OLECMDF_ENABLED;

	return S_OK;
}

STDMETHODIMP CArrowButton::SetSite(IUnknown *pUnkSite)
{
	if (!pUnkSite)
	{
		ATLTRACE(_T("SetSite(): pUnkSite is NULL\n"));
	}
	else
	{
		HRESULT hRes = S_OK;
		CComPtr<IServiceProvider> spSP;
		hRes = pUnkSite->QueryInterface(&spSP);

		if(SUCCEEDED(hRes) && spSP)
		{
			hRes = spSP->QueryService(IID_IWebBrowserApp, &m_pWebBrowser2); 
		}
	}
	return S_OK;
}


STDMETHODIMP CArrowButton::Exec(const GUID*, DWORD nCmdID, DWORD, VARIANTARG*, VARIANTARG*)
{
	MessageBox(NULL, L"Miss Inject", L"Error", MB_OK);
	return S_OK;
}