815 lines
24 KiB
C++
815 lines
24 KiB
C++
//
|
|
// shint.cpp: client shell utitilies
|
|
// internal functions
|
|
//
|
|
|
|
#include "stdafx.h"
|
|
|
|
#define TRC_GROUP TRC_GROUP_UI
|
|
#define TRC_FILE "shint.cpp"
|
|
#include <atrcapi.h>
|
|
|
|
|
|
#include "sh.h"
|
|
#include "rmigrate.h"
|
|
|
|
//
|
|
// Fix union names, because some headers redifine this
|
|
// and break STRRET to str field access
|
|
//
|
|
#undef DUMMYUNIONNAME
|
|
#define NONAMELESSUNION
|
|
|
|
#define CLXSERVER TEXT("CLXSERVER")
|
|
#define CLXCMDLINE TEXT("CLXCMDLINE")
|
|
#define FULLSCREEN TEXT("FULLSCREEN")
|
|
#define SWITCH_EDIT TEXT("EDIT")
|
|
#define SWITCH_MIGRATE TEXT("MIGRATE")
|
|
#define SWITCH_CONSOLE TEXT("CONSOLE")
|
|
|
|
|
|
#ifdef NONAMELESSUNION
|
|
#define NAMELESS_MEMBER(member) DUMMYUNIONNAME.##member
|
|
#else
|
|
#define NAMELESS_MEMBER(member) member
|
|
#endif
|
|
|
|
#define STRRET_OLESTR STRRET_WSTR // same as STRRET_WSTR
|
|
#define STRRET_OFFPTR(pidl,lpstrret) \
|
|
((LPSTR)((LPBYTE)(pidl)+(lpstrret)->NAMELESS_MEMBER(uOffset)))
|
|
|
|
|
|
|
|
|
|
/****************************************************************************/
|
|
/* Name: SHValidateParsedCmdParam */
|
|
/* */
|
|
/* Purpose: validates settings in _SH that were read from the cmd line */
|
|
/* */
|
|
/* Returns: Nothing */
|
|
/* */
|
|
/* Params: IN - lpszCmdParam */
|
|
/* */
|
|
/****************************************************************************/
|
|
DCBOOL CSH::SHValidateParsedCmdParam()
|
|
{
|
|
DC_BEGIN_FN("SHValidateParsedCmdParam");
|
|
//
|
|
//["<session>"] [-v:<server>] [-f[ullscreen]] [-w[idth]:<wd> -h[eight]:<ht>]
|
|
//
|
|
|
|
//
|
|
// if one of width/height is specified but not the other..
|
|
// then fill with defaults
|
|
//
|
|
if (_SH.commandLineHeight != _SH.commandLineWidth)
|
|
{
|
|
if (!_SH.commandLineHeight)
|
|
{
|
|
_SH.commandLineHeight = DEFAULT_DESKTOP_HEIGHT;
|
|
}
|
|
if (!_SH.commandLineWidth)
|
|
{
|
|
_SH.commandLineWidth = DEFAULT_DESKTOP_WIDTH;
|
|
}
|
|
}
|
|
|
|
//
|
|
// clamp to min max sizes. 0 means we are not set
|
|
//
|
|
if (_SH.commandLineHeight != 0)
|
|
{
|
|
if (_SH.commandLineHeight < MIN_DESKTOP_HEIGHT)
|
|
{
|
|
_SH.commandLineHeight = MIN_DESKTOP_HEIGHT;
|
|
}
|
|
else if (_SH.commandLineHeight > MAX_DESKTOP_HEIGHT)
|
|
{
|
|
_SH.commandLineHeight = MAX_DESKTOP_HEIGHT;
|
|
}
|
|
}
|
|
if (_SH.commandLineWidth != 0)
|
|
{
|
|
if (_SH.commandLineWidth < MIN_DESKTOP_WIDTH)
|
|
{
|
|
_SH.commandLineWidth = MIN_DESKTOP_WIDTH;
|
|
}
|
|
else if (_SH.commandLineWidth > MAX_DESKTOP_WIDTH)
|
|
{
|
|
_SH.commandLineWidth = MAX_DESKTOP_WIDTH;
|
|
}
|
|
}
|
|
|
|
DC_END_FN();
|
|
return TRUE;
|
|
}
|
|
|
|
//*************************************************************
|
|
//
|
|
// CLX_SkipWhite()
|
|
//
|
|
// Purpose: Skips whitespace characters
|
|
//
|
|
// Parameters: IN [lpszCmdParam] - Ptr to string
|
|
//
|
|
// Return: Ptr string past whitespace
|
|
//
|
|
// History: 09-30-97 BrianTa Created
|
|
//
|
|
//*************************************************************
|
|
|
|
LPTSTR
|
|
CLX_SkipWhite(IN LPTSTR lpszCmdParam)
|
|
{
|
|
while (*lpszCmdParam)
|
|
{
|
|
if (*lpszCmdParam != ' ')
|
|
break;
|
|
|
|
lpszCmdParam++;
|
|
}
|
|
|
|
return(lpszCmdParam);
|
|
}
|
|
|
|
|
|
//*************************************************************
|
|
//
|
|
// CLX_SkipNonWhite()
|
|
//
|
|
// Purpose: Skips non-whitespace characters
|
|
//
|
|
// Parameters: IN [lpszCmdParam] - Ptr to string
|
|
//
|
|
// Return: Ptr string past non-whitespace
|
|
//
|
|
// History: 09-30-97 BrianTa Created
|
|
//
|
|
//*************************************************************
|
|
|
|
LPTSTR CLX_SkipNonWhite(LPTSTR lpszCmdParam)
|
|
{
|
|
char Delim;
|
|
|
|
Delim = ' ';
|
|
|
|
if (*lpszCmdParam == '"')
|
|
{
|
|
Delim = '"';
|
|
lpszCmdParam++;
|
|
}
|
|
|
|
while (*lpszCmdParam)
|
|
{
|
|
if (*lpszCmdParam == Delim)
|
|
break;
|
|
|
|
lpszCmdParam++;
|
|
}
|
|
|
|
if (*lpszCmdParam == Delim)
|
|
lpszCmdParam++;
|
|
|
|
return(lpszCmdParam);
|
|
}
|
|
|
|
//*************************************************************
|
|
//
|
|
// CLX_GetSwitch_CLXSERVER()
|
|
//
|
|
// Purpose: Processes /CLXSERVER cmdline switch
|
|
//
|
|
// Parameters: IN [lpszCmdParam] - Ptr to cmdline
|
|
//
|
|
// Return: Number of characters consumed
|
|
//
|
|
// History: 09-30-97 BrianTa Created
|
|
//
|
|
//*************************************************************
|
|
|
|
UINT
|
|
CLX_GetSwitch_CLXSERVER(IN LPTSTR lpszCmdParam)
|
|
{
|
|
DC_BEGIN_FN("CLX_GetSwitch_CLXSERVER");
|
|
|
|
int len;
|
|
LPTSTR pszEnd;
|
|
LPTSTR pszStart;
|
|
|
|
pszStart = CLX_SkipWhite(lpszCmdParam);
|
|
|
|
TRC_ASSERT(*pszStart == _T('='),
|
|
(TB,_T("Invalid /clxserver syntax - expected '='\n")));
|
|
pszStart++;
|
|
|
|
pszStart = CLX_SkipWhite(pszStart);
|
|
pszEnd = CLX_SkipNonWhite(pszStart);
|
|
|
|
len = (INT) (pszEnd - pszStart);
|
|
|
|
DC_END_FN();
|
|
return(UINT) (pszEnd - lpszCmdParam);
|
|
}
|
|
|
|
//*************************************************************
|
|
//
|
|
// CLX_GetSwitch_CLXCMDLINE()
|
|
//
|
|
// Purpose: Processes /CLXCMDLINE cmdline switch
|
|
//
|
|
// Parameters: IN [lpszCmdParam] - Ptr to cmdline
|
|
//
|
|
// Return: Number of characters consumed
|
|
//
|
|
// History: 09-30-97 BrianTa Created
|
|
//
|
|
//*************************************************************
|
|
|
|
UINT
|
|
CSH::CLX_GetSwitch_CLXCMDLINE(IN LPTSTR lpszCmdParam)
|
|
{
|
|
int len;
|
|
LPTSTR pszEnd;
|
|
LPTSTR pszStart;
|
|
DC_BEGIN_FN("CLX_GetSwitch_CLXCMDLINE");
|
|
|
|
pszStart = CLX_SkipWhite(lpszCmdParam);
|
|
|
|
TRC_ASSERT(*pszStart == _T('='),
|
|
(TB,_T("Invalid /clxserver syntax - expected '='\n")));
|
|
|
|
pszStart++;
|
|
|
|
pszStart = CLX_SkipWhite(pszStart);
|
|
pszEnd = CLX_SkipNonWhite(pszStart);
|
|
|
|
len = (INT) (pszEnd - pszStart);
|
|
|
|
if (len > 0)
|
|
{
|
|
memmove(_SH.szCLXCmdLine, pszStart, len*sizeof(TCHAR));
|
|
_SH.szCLXCmdLine[len] = 0;
|
|
}
|
|
|
|
DC_END_FN();
|
|
return(UINT) (pszEnd - lpszCmdParam);
|
|
}
|
|
|
|
/****************************************************************************/
|
|
/* Name: SHGetSwitch */
|
|
/* */
|
|
/* Purpose: Retrieves cmdline switches */
|
|
/* */
|
|
/* Returns: Nothing */
|
|
/* */
|
|
/* Params: IN - lpszCmdParam */
|
|
/* */
|
|
/****************************************************************************/
|
|
LPTSTR CSH::SHGetSwitch(LPTSTR lpszCmdParam)
|
|
{
|
|
DCINT i;
|
|
DCTCHAR szParam[100];
|
|
|
|
DC_BEGIN_FN("SHGetSwitch");
|
|
|
|
/************************************************************************/
|
|
/* Retrieve the switch (case insensitive) */
|
|
/************************************************************************/
|
|
i=0;
|
|
|
|
while (*lpszCmdParam)
|
|
{
|
|
if (*lpszCmdParam == _T(' ') || *lpszCmdParam == _T('=') ||
|
|
*lpszCmdParam == _T(':'))
|
|
break;
|
|
|
|
if (i < sizeof(szParam) / sizeof(DCTCHAR) - 1)
|
|
{
|
|
#ifdef UNICODE
|
|
szParam[i] = (DCTCHAR) towupper(*lpszCmdParam);
|
|
#else // UNICODE
|
|
szParam[i] = (DCTCHAR) toupper(*lpszCmdParam);
|
|
#endif // UNICODE
|
|
|
|
i++;
|
|
}
|
|
|
|
lpszCmdParam++;
|
|
}
|
|
|
|
szParam[i] = 0;
|
|
|
|
#ifndef OS_WINCE
|
|
|
|
// Are we seeing the "/f" from /F[ullscreen]
|
|
if (szParam[0] == _T('F'))
|
|
{
|
|
_SH.fCommandStartFullScreen = TRUE;
|
|
}
|
|
|
|
|
|
// Are we seeing the "/W" from /w[idth]
|
|
else if (szParam[0] == _T('W'))
|
|
{
|
|
lpszCmdParam = SHGetCmdLineInt(lpszCmdParam, &_SH.commandLineWidth);
|
|
}
|
|
|
|
// Are we seeing the "/H" from /h[eight]
|
|
else if (szParam[0] == _T('H'))
|
|
{
|
|
lpszCmdParam = SHGetCmdLineInt(lpszCmdParam, &_SH.commandLineHeight);
|
|
}
|
|
|
|
// Are we seeing the "/V" for server
|
|
else
|
|
|
|
#endif
|
|
|
|
if (szParam[0] == _T('V'))
|
|
{
|
|
lpszCmdParam = SHGetServer(lpszCmdParam);
|
|
}
|
|
|
|
// Are we seeing the "/S"
|
|
else if (memcmp(szParam, "S", i) == 0)
|
|
{
|
|
lpszCmdParam = SHGetSession(lpszCmdParam);
|
|
}
|
|
|
|
// Are we seeing the "/C"
|
|
else if (memcmp(szParam, "C", i) == 0)
|
|
{
|
|
lpszCmdParam = SHGetCacheToClear(lpszCmdParam);
|
|
}
|
|
|
|
// Are we seeing the "/CLXCMDLINE=xyzzy"
|
|
else if (memcmp(szParam, CLXCMDLINE, i) == 0)
|
|
{
|
|
lpszCmdParam += CLX_GetSwitch_CLXCMDLINE(lpszCmdParam);
|
|
}
|
|
|
|
else if (memcmp(szParam, FULLSCREEN, i) == 0)
|
|
{
|
|
lpszCmdParam += SIZECHAR(FULLSCREEN);
|
|
}
|
|
|
|
else if (memcmp(szParam, SWITCH_EDIT,i) == 0)
|
|
{
|
|
lpszCmdParam = SHGetFileName(lpszCmdParam);
|
|
_fFileForEdit = TRUE;
|
|
_SH.autoConnectEnabled = FALSE;
|
|
}
|
|
|
|
else if (memcmp(szParam, SWITCH_MIGRATE,i) == 0)
|
|
{
|
|
_fMigrateOnly = TRUE;
|
|
}
|
|
|
|
else if (memcmp(szParam, SWITCH_CONSOLE,i) == 0)
|
|
{
|
|
SH_SetCmdConnectToConsole(TRUE);
|
|
}
|
|
|
|
|
|
|
|
/************************************************************************/
|
|
/* Not a recognized switch. Bring up usage */
|
|
/************************************************************************/
|
|
else
|
|
{
|
|
TRC_NRM((TB,_T("Invalid CmdLine switch - Display Usage AND EXIT %s"),
|
|
szParam));
|
|
DCTCHAR szCmdLineUsage[4096]; //Long string here.
|
|
DCTCHAR szUsageTitle[256];
|
|
if (!LoadString(_hInstance,
|
|
UI_IDS_USAGE_TITLE,
|
|
szUsageTitle,
|
|
SIZECHAR(szUsageTitle)))
|
|
{
|
|
TRC_ERR((TB,_T("Error loading UI_IDS_USAGE_TITLE")));
|
|
return NULL;
|
|
}
|
|
|
|
if (!LoadString(_hInstance,
|
|
UI_IDS_CMD_LINE_USAGE,
|
|
szCmdLineUsage,
|
|
SIZECHAR(szCmdLineUsage)))
|
|
{
|
|
TRC_NRM((TB,_T("Error loading UI_IDS_CMD_LINE_USAGE")));
|
|
return NULL;
|
|
}
|
|
|
|
MessageBox(NULL, szCmdLineUsage, szUsageTitle,
|
|
MB_ICONINFORMATION | MB_OK);
|
|
|
|
return NULL;
|
|
}
|
|
|
|
DC_END_FN();
|
|
|
|
return(lpszCmdParam);
|
|
}
|
|
|
|
/****************************************************************************/
|
|
/* Name: SHGetServer */
|
|
/* */
|
|
/* Purpose: Retrieves the server name (if specified) */
|
|
/* */
|
|
/* Returns: Nothing */
|
|
/* */
|
|
/* Params: IN - lpszCmdParam */
|
|
/* */
|
|
/****************************************************************************/
|
|
LPTSTR CSH::SHGetServer(LPTSTR lpszCmdParam)
|
|
{
|
|
DC_BEGIN_FN("SHGetServer");
|
|
if (!lpszCmdParam)
|
|
{
|
|
return NULL;
|
|
}
|
|
|
|
/************************************************************************/
|
|
/* Retrieve the server */
|
|
/************************************************************************/
|
|
lpszCmdParam = SHGetCmdLineString(lpszCmdParam, _SH.szCommandLineServer,
|
|
SIZECHAR(_SH.szCommandLineServer) -1);
|
|
|
|
DC_END_FN();
|
|
return lpszCmdParam;
|
|
}
|
|
|
|
|
|
/****************************************************************************/
|
|
/* Name: SHGetSession */
|
|
/* */
|
|
/* Purpose: Retrieves the session name (if specified) */
|
|
/* */
|
|
/* Returns: Nothing */
|
|
/* */
|
|
/* Params: IN - lpszCmdParam */
|
|
/* */
|
|
/****************************************************************************/
|
|
LPTSTR CSH::SHGetSession(LPTSTR lpszCmdParam)
|
|
{
|
|
BOOL fQuote = FALSE;
|
|
|
|
DC_BEGIN_FN("SHGetSession");
|
|
|
|
TRC_ASSERT((_SH.fRegDefault == TRUE),
|
|
(TB,_T("Invalid CmdLine syntax - session respecified.")));
|
|
|
|
|
|
// Retrieve the reg session
|
|
lpszCmdParam = SHGetCmdLineString(lpszCmdParam, _SH.regSession,
|
|
SIZECHAR(_SH.regSession) -1);
|
|
|
|
// In the non-default session, display the session name. Choose the
|
|
// appropriate connected/disconnected strings.
|
|
TRC_DBG((TB, _T("Named session")));
|
|
|
|
_SH.fRegDefault = FALSE;
|
|
_SH.connectedStringID = UI_IDS_FRAME_TITLE_CONNECTED;
|
|
_SH.disconnectedStringID = UI_IDS_FRAME_TITLE_DISCONNECTED;
|
|
|
|
DC_END_FN();
|
|
|
|
return(lpszCmdParam);
|
|
}
|
|
|
|
LPTSTR CSH::SHGetFileName(LPTSTR lpszCmdParam)
|
|
{
|
|
BOOL fQuote = FALSE;
|
|
|
|
DC_BEGIN_FN("SHGetSession");
|
|
|
|
TRC_ASSERT((_SH.fRegDefault == TRUE),
|
|
(TB,_T("Invalid CmdLine syntax - session respecified.")));
|
|
|
|
|
|
// Retrieve the filename
|
|
lpszCmdParam = SHGetCmdLineString(lpszCmdParam, _szFileName,
|
|
SIZECHAR(_szFileName) -1);
|
|
|
|
_SH.fRegDefault = FALSE;
|
|
_SH.connectedStringID = UI_IDS_FRAME_TITLE_CONNECTED;
|
|
_SH.disconnectedStringID = UI_IDS_FRAME_TITLE_DISCONNECTED;
|
|
|
|
DC_END_FN();
|
|
|
|
return(lpszCmdParam);
|
|
}
|
|
|
|
/****************************************************************************/
|
|
/* Name: SHGetCmdLineString */
|
|
/* */
|
|
/* Purpose: Retrieve a string parameter */
|
|
/* */
|
|
/* Returns: Nothing */
|
|
/* */
|
|
/* Params: IN - lpszCmdParam */
|
|
/* */
|
|
/****************************************************************************/
|
|
LPTSTR CSH::SHGetCmdLineString(LPTSTR lpszCmdParam, LPTSTR lpszDest,
|
|
DCINT cbDestLen)
|
|
{
|
|
DCINT i;
|
|
BOOL fQuote = FALSE;
|
|
|
|
DC_BEGIN_FN("SHGetCmdLineString");
|
|
|
|
TRC_ASSERT(lpszCmdParam && lpszDest && cbDestLen,
|
|
(TB, _T("SHGetCmdLineString. Invalid param(s)\n")));
|
|
if (!lpszCmdParam || !lpszDest || !cbDestLen)
|
|
{
|
|
return NULL;
|
|
}
|
|
|
|
/************************************************************************/
|
|
/* Retrieve a command line string parameter */
|
|
/************************************************************************/
|
|
while (*lpszCmdParam == _T(' '))
|
|
lpszCmdParam++;
|
|
|
|
if (*lpszCmdParam == _T('=') || *lpszCmdParam == _T(':'))
|
|
lpszCmdParam++;
|
|
|
|
while (*lpszCmdParam == _T(' '))
|
|
lpszCmdParam++;
|
|
|
|
i=0;
|
|
|
|
while (*lpszCmdParam)
|
|
{
|
|
switch (*lpszCmdParam)
|
|
{
|
|
case _T('"'):
|
|
fQuote = !fQuote;
|
|
lpszCmdParam++;
|
|
break;
|
|
|
|
case _T(' '):
|
|
if (!fQuote)
|
|
{
|
|
lpszCmdParam++;
|
|
DC_QUIT;
|
|
}
|
|
// else fall through
|
|
|
|
default:
|
|
if (i < cbDestLen)
|
|
lpszDest[i++] = *lpszCmdParam;
|
|
lpszCmdParam++;
|
|
}
|
|
}
|
|
|
|
|
|
DC_EXIT_POINT:
|
|
DC_END_FN();
|
|
lpszDest[i] = 0;
|
|
return lpszCmdParam;
|
|
}
|
|
|
|
/****************************************************************************/
|
|
/* Name: SHGetCmdLineInt */
|
|
/* */
|
|
/* Purpose: Retrieves an integer parameter */
|
|
/* */
|
|
/* Returns: Nothing */
|
|
/* */
|
|
/* Params: IN - lpszCmdParam, OUT- PInt */
|
|
/* */
|
|
/****************************************************************************/
|
|
LPTSTR CSH::SHGetCmdLineInt(LPTSTR lpszCmdParam, PDCUINT pInt)
|
|
{
|
|
DC_BEGIN_FN("SHGetCmdLineInt");
|
|
if (!pInt)
|
|
{
|
|
return NULL;
|
|
}
|
|
if (!lpszCmdParam)
|
|
{
|
|
return NULL;
|
|
}
|
|
|
|
/************************************************************************/
|
|
/* Retrieve an integer parameter */
|
|
/************************************************************************/
|
|
while (*lpszCmdParam == _T(' '))
|
|
lpszCmdParam++;
|
|
|
|
if (*lpszCmdParam == _T('=') || *lpszCmdParam == _T(':'))
|
|
lpszCmdParam++;
|
|
|
|
while (*lpszCmdParam == _T(' '))
|
|
lpszCmdParam++;
|
|
|
|
DCUINT readInt = 0;
|
|
while (*lpszCmdParam)
|
|
{
|
|
if (*lpszCmdParam == _T(' '))
|
|
break;
|
|
|
|
if (_istdigit(*lpszCmdParam))
|
|
{
|
|
DCINT digit = *lpszCmdParam - _T('0');
|
|
TRC_ASSERT(digit >=0 && digit <=9, (TB,_T("digit read error\n")));
|
|
if (digit <0 || digit >9)
|
|
{
|
|
break;
|
|
}
|
|
readInt = readInt * 10 + digit;
|
|
}
|
|
else
|
|
{
|
|
break;
|
|
}
|
|
lpszCmdParam++;
|
|
}
|
|
|
|
*pInt = readInt;
|
|
|
|
DC_END_FN();
|
|
return lpszCmdParam;
|
|
}
|
|
|
|
/****************************************************************************/
|
|
/* Name: SHGetCacheToClear */
|
|
/* */
|
|
/* Purpose: Retrieves the cache type (e.g. bitmap) to be cleared */
|
|
/* */
|
|
/* Returns: Nothing */
|
|
/* */
|
|
/* Params: IN - lpszCmdParam */
|
|
/* */
|
|
/****************************************************************************/
|
|
LPTSTR CSH::SHGetCacheToClear(LPTSTR lpszCmdParam)
|
|
{
|
|
DCINT i;
|
|
TCHAR cacheType[10];
|
|
|
|
DC_BEGIN_FN("SHGetCacheToClear");
|
|
|
|
/************************************************************************/
|
|
/* Retrieve the cache type */
|
|
/************************************************************************/
|
|
while (*lpszCmdParam == _T(' '))
|
|
lpszCmdParam++;
|
|
|
|
if (*lpszCmdParam == _T('=') || *lpszCmdParam == _T(':'))
|
|
lpszCmdParam++;
|
|
|
|
while (*lpszCmdParam == _T(' '))
|
|
lpszCmdParam++;
|
|
|
|
i=0;
|
|
|
|
while (*lpszCmdParam)
|
|
{
|
|
if (*lpszCmdParam == _T(' '))
|
|
break;
|
|
|
|
if (i < sizeof(cacheType) / sizeof(DCTCHAR) -1)
|
|
#ifdef UNICODE
|
|
cacheType[i++] = (DCTCHAR) towupper(*lpszCmdParam);
|
|
#else // UNICODE
|
|
cacheType[i++] = (DCTCHAR) toupper(*lpszCmdParam);
|
|
#endif // UNICODE
|
|
|
|
lpszCmdParam++;
|
|
}
|
|
|
|
cacheType[i] = 0;
|
|
|
|
if (memcmp(cacheType, "BITMAP", i) == 0)
|
|
{
|
|
_SH.fClearPersistBitmapCache = TRUE;
|
|
}
|
|
else
|
|
{
|
|
TRC_NRM((TB,_T("Invalid Cache Type - %s"), cacheType));
|
|
}
|
|
|
|
DC_END_FN();
|
|
|
|
return(lpszCmdParam);
|
|
}
|
|
|
|
//
|
|
// Take the session in _SH.regSession and figure out
|
|
// if it is a file or a registry session
|
|
//
|
|
// We need to do this because for compatability reasons
|
|
// the client has to be able to support both a file name
|
|
// and the registry session name as default command line
|
|
// params (enter this logic to determine which is which).
|
|
//
|
|
//
|
|
// Return TRUE if it's a valid reg or connection param
|
|
// or FALSE otherwise
|
|
//
|
|
BOOL CSH::ParseFileOrRegConnectionParam()
|
|
{
|
|
BOOL fRet = TRUE;
|
|
DC_BEGIN_FN("ParseFileOrRegConnectionParam");
|
|
|
|
//
|
|
// If a connection parameter is specified that is
|
|
// different from the default
|
|
//
|
|
|
|
if(_tcscmp(_SH.regSession, SH_DEFAULT_REG_SESSION)) {
|
|
|
|
//
|
|
// A connection parameter is specified
|
|
// check for the three possible cases
|
|
// 1) it's an RDP file
|
|
// 2) it's a registry connection
|
|
// 3) it's INVALID!
|
|
//
|
|
|
|
//a) check if the session is really a file
|
|
if (SH_FileExists(_SH.regSession)) {
|
|
_tcsncpy(_szFileName, _SH.regSession,
|
|
SIZECHAR(_szFileName));
|
|
_fFileForConnect = TRUE;
|
|
_SH.autoConnectEnabled = TRUE;
|
|
SetRegSessionSpecified(FALSE);
|
|
}
|
|
else if (SH_TSSettingsRegKeyExists(_SH.regSession)) {
|
|
|
|
//Assume it's an old registry style session name
|
|
SetRegSessionSpecified(TRUE);
|
|
}
|
|
else {
|
|
TRC_ERR((TB,_T("Reg session is neither file nore reg key: %s"),
|
|
_SH.regSession));
|
|
fRet = FALSE;
|
|
}
|
|
}
|
|
|
|
DC_END_FN();
|
|
return fRet;
|
|
}
|
|
|
|
#ifndef OS_WINCE
|
|
//
|
|
// Copy of StrRetToStrW from \shell\shlwapi\strings.c
|
|
//
|
|
//
|
|
|
|
// dupe a string using the task allocator for returing from a COM interface
|
|
//
|
|
HRESULT XSHStrDupA(LPCSTR psz, WCHAR **ppwsz)
|
|
{
|
|
WCHAR *pwsz;
|
|
DWORD cch;
|
|
|
|
//RIPMSG(psz && IS_VALID_STRING_PTRA(psz, -1), "SHStrDupA: Caller passed invalid psz");
|
|
|
|
if (psz)
|
|
{
|
|
cch = MultiByteToWideChar(CP_ACP, 0, psz, -1, NULL, 0);
|
|
pwsz = (WCHAR *)CoTaskMemAlloc((cch + 1) * sizeof(WCHAR));
|
|
}
|
|
else
|
|
pwsz = NULL;
|
|
|
|
*((PVOID UNALIGNED64 *) ppwsz) = pwsz;
|
|
|
|
if (pwsz)
|
|
{
|
|
MultiByteToWideChar(CP_ACP, 0, psz, -1, *ppwsz, cch);
|
|
return S_OK;
|
|
}
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
|
|
HRESULT XStrRetToStrW(LPSTRRET psr, LPCITEMIDLIST pidl, WCHAR **ppsz)
|
|
{
|
|
HRESULT hres = S_OK;
|
|
|
|
switch (psr->uType)
|
|
{
|
|
case STRRET_WSTR:
|
|
*ppsz = psr->DUMMYUNIONNAME.pOleStr;
|
|
psr->DUMMYUNIONNAME.pOleStr = NULL; // avoid alias
|
|
hres = *ppsz ? S_OK : E_FAIL;
|
|
break;
|
|
|
|
case STRRET_OFFSET:
|
|
hres = XSHStrDupA(STRRET_OFFPTR(pidl, psr), ppsz);
|
|
break;
|
|
|
|
case STRRET_CSTR:
|
|
hres = XSHStrDupA(psr->DUMMYUNIONNAME.cStr, ppsz);
|
|
break;
|
|
|
|
default:
|
|
*ppsz = NULL;
|
|
hres = E_FAIL;
|
|
}
|
|
return hres;
|
|
}
|
|
|
|
#endif
|