#ifndef ____H_GUESTIMPL

#define ____H_GUESTIMPL

#include "VirtualBoxBase.h"

#include <VBox/ostypes.h>

#ifdef VBOX_WITH_GUEST_CONTROL

# include <VBox/HostServices/GuestControlSvc.h>

# include <hgcm/HGCM.h>

using namespace guestControl;

#endif

typedef enum

{

GUESTSTATTYPE_CPUUSER = 0,

GUESTSTATTYPE_CPUKERNEL = 1,

GUESTSTATTYPE_CPUIDLE = 2,

GUESTSTATTYPE_MEMTOTAL = 3,

GUESTSTATTYPE_MEMFREE = 4,

GUESTSTATTYPE_MEMBALLOON = 5,

GUESTSTATTYPE_MEMCACHE = 6,

GUESTSTATTYPE_PAGETOTAL = 7,

GUESTSTATTYPE_PAGEFREE = 8,

GUESTSTATTYPE_MAX = 9

} GUESTSTATTYPE;

class Console;

#ifdef VBOX_WITH_GUEST_CONTROL

class Progress;

#endif

class ATL_NO_VTABLE Guest :

public VirtualBoxBase,

VBOX_SCRIPTABLE_IMPL(IGuest)

{

public:

VIRTUALBOXBASE_ADD_ERRORINFO_SUPPORT(Guest, IGuest)

DECLARE_NOT_AGGREGATABLE(Guest)

DECLARE_PROTECT_FINAL_CONSTRUCT()

BEGIN_COM_MAP(Guest)

COM_INTERFACE_ENTRY(ISupportErrorInfo)

COM_INTERFACE_ENTRY(IGuest)

COM_INTERFACE_ENTRY(IDispatch)

END_COM_MAP()

DECLARE_EMPTY_CTOR_DTOR (Guest)

HRESULT FinalConstruct();

void FinalRelease();

// Public initializer/uninitializer for internal purposes only

HRESULT init (Console *aParent);

void uninit();

// IGuest properties

STDMETHOD(COMGETTER(OSTypeId)) (BSTR *aOSTypeId);

#if 0

/** @todo Will replace old AdditionsActive call. */

STDMETHOD(COMGETTER(AdditionsActive)) (ULONG aLevel, BOOL *aAdditionsActive);

#endif

STDMETHOD(COMGETTER(AdditionsActive)) (BOOL *aAdditionsActive);

STDMETHOD(COMGETTER(AdditionsVersion)) (BSTR *aAdditionsVersion);

/** @todo Remove later by replacing it by AdditionsFeatureAvailable(). */

STDMETHOD(COMGETTER(SupportsSeamless)) (BOOL *aSupportsSeamless);

STDMETHOD(COMGETTER(SupportsGraphics)) (BOOL *aSupportsGraphics);

#if 0

/** @todo Will replace SupportsSeamless, SupportsGraphics, ... */

STDMETHOD(COMGETTER(AdditionsFeatureAvailable)) (ULONG64 aFeature, BOOL *aActive, BOOL *aAvailable);

#endif

STDMETHOD(COMGETTER(MemoryBalloonSize)) (ULONG *aMemoryBalloonSize);

STDMETHOD(COMSETTER(MemoryBalloonSize)) (ULONG aMemoryBalloonSize);

STDMETHOD(COMGETTER(PageFusionEnabled)) (BOOL *aPageFusionEnabled);

STDMETHOD(COMSETTER(PageFusionEnabled)) (BOOL aPageFusionEnabled);

STDMETHOD(COMGETTER(StatisticsUpdateInterval)) (ULONG *aUpdateInterval);

STDMETHOD(COMSETTER(StatisticsUpdateInterval)) (ULONG aUpdateInterval);

// IGuest methods

STDMETHOD(SetCredentials)(IN_BSTR aUserName, IN_BSTR aPassword,

IN_BSTR aDomain, BOOL aAllowInteractiveLogon);

STDMETHOD(ExecuteProcess)(IN_BSTR aCommand, ULONG aFlags,

ComSafeArrayIn(IN_BSTR, aArguments), ComSafeArrayIn(IN_BSTR, aEnvironment),

IN_BSTR aUserName, IN_BSTR aPassword,

ULONG aTimeoutMS, ULONG *aPID, IProgress **aProgress);

STDMETHOD(GetProcessOutput)(ULONG aPID, ULONG aFlags, ULONG aTimeoutMS, ULONG64 aSize, ComSafeArrayOut(BYTE, aData));

STDMETHOD(GetProcessStatus)(ULONG aPID, ULONG *aExitCode, ULONG *aFlags, ULONG *aStatus);

STDMETHOD(InternalGetStatistics)(ULONG *aCpuUser, ULONG *aCpuKernel, ULONG *aCpuIdle,

ULONG *aMemTotal, ULONG *aMemFree, ULONG *aMemBalloon, ULONG *aMemShared, ULONG *aMemCache,

ULONG *aPageTotal, ULONG *aMemAllocTotal, ULONG *aMemFreeTotal, ULONG *aMemBalloonTotal, ULONG *aMemSharedTotal);

// Public methods that are not in IDL (only called internally).

void setAdditionsInfo(Bstr aInterfaceVersion, VBOXOSTYPE aOsType);

void setAdditionsInfo2(Bstr aAdditionsVersion, Bstr aVersionName);

void setAdditionsStatus(VBoxGuestStatusFacility Facility, VBoxGuestStatusCurrent Status, ULONG ulFlags);

void setSupportedFeatures(ULONG64 ulCaps, ULONG64 ulActive);

HRESULT setStatistic(ULONG aCpuId, GUESTSTATTYPE enmType, ULONG aVal);

# ifdef VBOX_WITH_GUEST_CONTROL

/** Static callback for handling guest notifications. */

static DECLCALLBACK(int) doGuestCtrlNotification(void *pvExtension, uint32_t u32Function, void *pvParms, uint32_t cbParms);

# endif

private:

# ifdef VBOX_WITH_GUEST_CONTROL

struct CallbackContext

{

eVBoxGuestCtrlCallbackType mType;

/** Pointer to user-supplied data. */

void *pvData;

/** Size of user-supplied data. */

uint32_t cbData;

/** Pointer to user-supplied IProgress. */

ComObjPtr<Progress> pProgress;

};

/*

typedef std::map< uint32_t, CallbackContext > CallbackMap;

typedef std::map< uint32_t, CallbackContext >::iterator CallbackMapIter;

typedef std::map< uint32_t, CallbackContext >::const_iterator CallbackMapIterConst;

struct GuestProcess

{

uint32_t mStatus;

uint32_t mFlags;

uint32_t mExitCode;

};

/*

typedef std::map< uint32_t, GuestProcess > GuestProcessMap;

typedef std::map< uint32_t, GuestProcess >::iterator GuestProcessMapIter;

typedef std::map< uint32_t, GuestProcess >::const_iterator GuestProcessMapIterConst;

int prepareExecuteEnv(const char *pszEnv, void **ppvList, uint32_t *pcbList, uint32_t *pcEnv);

/** Handler for guest execution control notifications. */

int notifyCtrlClientDisconnected(uint32_t u32Function, PCALLBACKDATACLIENTDISCONNECTED pData);

int notifyCtrlExecStatus(uint32_t u32Function, PCALLBACKDATAEXECSTATUS pData);

int notifyCtrlExecOut(uint32_t u32Function, PCALLBACKDATAEXECOUT pData);

CallbackMapIter getCtrlCallbackContextByID(uint32_t u32ContextID);

GuestProcessMapIter getProcessByPID(uint32_t u32PID);

void destroyCtrlCallbackContext(CallbackMapIter it);

uint32_t addCtrlCallbackContext(eVBoxGuestCtrlCallbackType enmType, void *pvData, uint32_t cbData, Progress* pProgress);

# endif

struct Data

{

Data() : mAdditionsActive (FALSE), mSupportsSeamless (FALSE),

mSupportsGraphics (FALSE) {}

Bstr mOSTypeId;

BOOL mAdditionsActive;

Bstr mAdditionsVersion;

BOOL mSupportsSeamless;

BOOL mSupportsGraphics;

};

ULONG mMemoryBalloonSize;

ULONG mStatUpdateInterval;

ULONG mCurrentGuestStat[GUESTSTATTYPE_MAX];

BOOL mfPageFusionEnabled;

Console *mParent;

Data mData;

# ifdef VBOX_WITH_GUEST_CONTROL

/** General extension callback for guest control. */

HGCMSVCEXTHANDLE mhExtCtrl;

volatile uint32_t mNextContextID;

CallbackMap mCallbackMap;

GuestProcessMap mGuestProcessMap;

# endif

};

#endif // ____H_GUESTIMPL