168 lines
4.3 KiB
C++
168 lines
4.3 KiB
C++
// SessionResolver.h : Declaration of the CSessionResolver
|
|
|
|
#ifndef __SESSIONRESOLVER_H_
|
|
#define __SESSIONRESOLVER_H_
|
|
|
|
#include "resource.h" // main symbols
|
|
#include <io.h>
|
|
#include <fcntl.h>
|
|
#include <stdio.h>
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
|
|
#define DBG_MSG_DEST_FILE 0x020
|
|
|
|
extern int gDbgFlag, iDbgFileHandle;
|
|
|
|
typedef struct _SPLASHLIST
|
|
{
|
|
void *next;
|
|
int refcount;
|
|
SID Sid;
|
|
} SPLASHLIST, *PSPLASHLIST;
|
|
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CSessionResolver
|
|
class ATL_NO_VTABLE CSessionResolver :
|
|
public CComObjectRootEx<CComSingleThreadModel>,
|
|
public CComCoClass<CSessionResolver, &CLSID_SessionResolver>,
|
|
public IDispatchImpl<ISAFRemoteDesktopCallback, &IID_ISAFRemoteDesktopCallback, &LIBID_SAFSESSIONRESOLVERLib>
|
|
{
|
|
public:
|
|
CSessionResolver()
|
|
{
|
|
CRegKey hkSpew;
|
|
|
|
// initialize our defaults
|
|
m_iWaitDuration = 300 * 1000; // 5 minute timeout
|
|
|
|
#ifdef _PERF_OPTIMIZATIONS
|
|
m_bstrResolveURL = OLESTR("hcp://system/Remote%20Assistance/RAStartPage.htm");
|
|
m_bstrConnectURL = OLESTR("hcp://system/Remote%20Assistance/RAStartPage.htm");
|
|
m_bstrRemoteURL = OLESTR("hcp://system/Remote%20Assistance/RAStartPage.htm");
|
|
#else
|
|
m_bstrResolveURL = OLESTR("hcp://CN=Microsoft%20Corporation,L=Redmond,S=Washington,C=US/Remote%20Assistance/RAStartPage.htm");
|
|
m_bstrConnectURL = OLESTR("hcp://CN=Microsoft%20Corporation,L=Redmond,S=Washington,C=US/Remote%20Assistance/RAStartPage.htm");
|
|
m_bstrRemoteURL = OLESTR("hcp://CN=Microsoft%20Corporation,L=Redmond,S=Washington,C=US/Remote%20Assistance/RAStartPage.htm");
|
|
#endif
|
|
|
|
m_lSessionTag = 0;
|
|
|
|
gDbgFlag = 3;
|
|
iDbgFileHandle = 0;
|
|
|
|
if (ERROR_SUCCESS == hkSpew.Open(HKEY_LOCAL_MACHINE, L"SOFTWARE\\Microsoft\\PCHealth\\HelpCtr\\SAFSessionResolver", KEY_READ))
|
|
{
|
|
|
|
DWORD dwLen = sizeof(gDbgFlag);
|
|
|
|
hkSpew.QueryValue((LPTSTR)&gDbgFlag, L"DebugSpew", &dwLen);
|
|
}
|
|
|
|
if (gDbgFlag & DBG_MSG_DEST_FILE)
|
|
{
|
|
WCHAR szLogfileName[MAX_PATH];
|
|
|
|
GetSystemDirectory(szLogfileName, sizeof(szLogfileName)/sizeof(szLogfileName[0]));
|
|
wcscat(szLogfileName, L"\\safrslv.log");
|
|
|
|
iDbgFileHandle = _wopen(szLogfileName, _O_APPEND | _O_BINARY | _O_RDWR, 0);
|
|
if (-1 != iDbgFileHandle)
|
|
{
|
|
OutputDebugStringA("opened debug log file\n");
|
|
}
|
|
else
|
|
{
|
|
unsigned char UniCode[2] = {0xff, 0xfe};
|
|
|
|
// we must create the file
|
|
OutputDebugStringA("must create debug log file");
|
|
iDbgFileHandle = _wopen(szLogfileName, _O_BINARY | _O_CREAT | _O_RDWR, _S_IREAD | _S_IWRITE);
|
|
if (-1 != iDbgFileHandle)
|
|
_write(iDbgFileHandle, &UniCode, sizeof(UniCode));
|
|
else
|
|
{
|
|
OutputDebugStringA("ERROR: failed to create debug log file");
|
|
iDbgFileHandle = 0;
|
|
}
|
|
}
|
|
}
|
|
|
|
InitializeCriticalSection(&m_CritSec);
|
|
m_pSplash = (SPLASHLIST *)LocalAlloc(LMEM_FIXED, sizeof(SPLASHLIST));
|
|
if (m_pSplash)
|
|
{
|
|
// make certain that we start the splash list with an empty
|
|
// SID.
|
|
ZeroMemory(m_pSplash, sizeof(SPLASHLIST));
|
|
}
|
|
}
|
|
|
|
~CSessionResolver()
|
|
{
|
|
m_bstrResolveURL.Empty();
|
|
m_bstrConnectURL.Empty();
|
|
m_bstrRemoteURL.Empty();
|
|
DeleteCriticalSection(&m_CritSec);
|
|
PSPLASHLIST lpWalker, lpNext;
|
|
|
|
lpWalker = m_pSplash;
|
|
|
|
while (lpWalker)
|
|
{
|
|
lpNext = (SPLASHLIST *)lpWalker->next;
|
|
LocalFree(lpWalker);
|
|
lpWalker = lpNext;
|
|
}
|
|
if (iDbgFileHandle)
|
|
_close(iDbgFileHandle);
|
|
iDbgFileHandle = 0;
|
|
}
|
|
|
|
DECLARE_REGISTRY_RESOURCEID(IDR_SESSIONRESOLVER)
|
|
DECLARE_NOT_AGGREGATABLE(CSessionResolver)
|
|
|
|
DECLARE_PROTECT_FINAL_CONSTRUCT()
|
|
|
|
BEGIN_COM_MAP(CSessionResolver)
|
|
COM_INTERFACE_ENTRY(ISAFRemoteDesktopCallback)
|
|
COM_INTERFACE_ENTRY(IDispatch)
|
|
END_COM_MAP()
|
|
|
|
public:
|
|
|
|
//
|
|
// ISAFRemoteDesktopCallback Methods
|
|
//
|
|
STDMETHOD(OnDisconnect)(
|
|
/*[in]*/BSTR connectParms,
|
|
/*[in]*/BSTR userSID,
|
|
/*[in]*/long sessionID
|
|
);
|
|
STDMETHOD(ResolveUserSessionID)(
|
|
/*[in]*/BSTR connectParms,
|
|
/*[in]*/BSTR userSID,
|
|
/*[in]*/ BSTR expertHelpBlob,
|
|
/*[in]*/ BSTR userHelpBlob,
|
|
/*[in, out, retval]*/long *sessionID,
|
|
/*[in]*/DWORD dwPID
|
|
,/*[out]*/ULONG_PTR* hHelpCtr
|
|
,/*[out, retval]*/int *result
|
|
);
|
|
|
|
private:
|
|
int m_iWaitDuration;
|
|
int m_iDbgFlag;
|
|
long m_lSessionTag;
|
|
|
|
CRITICAL_SECTION m_CritSec;
|
|
PSPLASHLIST m_pSplash;
|
|
|
|
CComBSTR m_bstrResolveURL;
|
|
CComBSTR m_bstrConnectURL;
|
|
CComBSTR m_bstrRemoteURL;
|
|
};
|
|
|
|
#endif //__SESSIONRESOLVER_H_
|