// THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF
// ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED TO
// THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A
// PARTICULAR PURPOSE.
//
// Copyright (c) Microsoft Corporation. All rights reserved.

#ifndef __METHODSUB_H__
#define __METHODSUB_H__

#include "resource.h"       // main symbols

typedef list<LONGLONG> TimeStack;
typedef map<ULONG64, TimeStack *> TimeMap;

/*C+C+++C+++C+++C+++C+++C+++C+++C+++C+++C+++C+++C+++C+++C+++C+++C+++C+++C+++C
Class:   CMethodSub
Summary: Method Events Subscriber 
C---C---C---C---C---C---C---C---C---C---C---C---C---C---C---C---C---C---C-C*/ 
class ATL_NO_VTABLE CMethodSub : 
	public CSysLCESub,
	public IComMethodEvents
{
public:

	TimeMap m_map;
	CMethodSub(CComSpy * pSpy)
	{
		m_pSpy = pSpy;
	}
	CMethodSub()
	{
		m_pSpy = NULL;
	}

	DECLARE_NOT_AGGREGATABLE(CMethodSub)
	DECLARE_GET_CONTROLLING_UNKNOWN()

	BEGIN_COM_MAP(CMethodSub)
		COM_INTERFACE_ENTRY(IComMethodEvents)
		COM_INTERFACE_ENTRY_CHAIN(CSysLCESub)
	END_COM_MAP()

    virtual EventEnum EventType() { return Method; }
	virtual REFCLSID EventCLSID() { return CLSID_ComServiceEvents; }
    virtual REFIID EventIID() { return IID_IComMethodEvents; }

	STDMETHOD(OnMethodCall)(COMSVCSEVENTINFO * pInfo, ULONG64 oid, REFCLSID cid, REFIID rid, ULONG iMeth);
	STDMETHOD(OnMethodReturn)(COMSVCSEVENTINFO * pInfo, ULONG64 oid, REFCLSID cid, REFIID rid, ULONG iMeth, HRESULT hr);
	STDMETHOD(OnMethodException)(COMSVCSEVENTINFO * pInfo, ULONG64 oid, REFCLSID cid, REFIID rid, ULONG iMeth);

	HRESULT GetClsidOfTypeLib2 (IID * piid, UUID * puuidClsid);
	HRESULT GetMethodName (REFIID riid, int iMeth, __deref_out LPWSTR* ppwszMethodName);

};
#endif //__METHODSUB_H__