350 lines
10 KiB
C
350 lines
10 KiB
C
#include <windows.h>
|
|
#include <userenv.h>
|
|
#include <userenvp.h>
|
|
#include <shlwapi.h>
|
|
#include "shmgdefs.h"
|
|
|
|
EXTERN_C void UpgradeDesktopWallpaper(void);
|
|
|
|
enum {
|
|
CMD_W95_TO_SUR,
|
|
CMD_DAYTONA_CURSORS,
|
|
CMD_FIX_CURSOR_PATHS,
|
|
CMD_FIX_SPECIAL_FOLDERS,
|
|
CMD_FIX_WINDOWS_PROFILE_SECURITY,
|
|
CMD_FIX_USER_PROFILE_SECURITY,
|
|
CMD_FIX_POLICIES_SECURITY,
|
|
CMD_CVT_SCHEMES_TO_MULTIUSER,
|
|
CMD_UPGRADE_NT4_PROFILE_TO_NT5,
|
|
CMD_UPGRADE_SCHEMES_AND_NCMETRICS_TO_WIN2000,
|
|
CMD_UPGRADE_SCHEMES_AND_NCMETRICS_FROM_WIN9X_TO_WIN2000,
|
|
CMD_FIX_CAPI_DIR_ATTRIBUTES,
|
|
CMD_MERGE_DESKTOP_AND_NORMAL_STREAMS,
|
|
CMD_MOVE_AND_ADJUST_ICON_METRICS,
|
|
CMD_FIX_HTML_HELP,
|
|
CMD_SET_SCREENSAVER_FRIENDLYUI,
|
|
CMD_ADD_CONFIGURE_PROGRAMS,
|
|
CMD_OCINSTALL_FIXUP,
|
|
|
|
// enums from here down will run within OleInitialize/OleUnitialize
|
|
CMD_OLE_REQUIRED_START = 10000,
|
|
CMD_OCINSTALL_SHOW_IE,
|
|
CMD_OCINSTALL_REINSTALL_IE,
|
|
CMD_OCINSTALL_HIDE_IE,
|
|
CMD_OCINSTALL_USER_CONFIG_IE,
|
|
CMD_OCINSTALL_SHOW_OE,
|
|
CMD_OCINSTALL_REINSTALL_OE,
|
|
CMD_OCINSTALL_HIDE_OE,
|
|
CMD_OCINSTALL_USER_CONFIG_OE,
|
|
CMD_OCINSTALL_SHOW_VM,
|
|
CMD_OCINSTALL_REINSTALL_VM,
|
|
CMD_OCINSTALL_HIDE_VM,
|
|
CMD_OCINSTALL_UPDATE,
|
|
CMD_OCINSTALL_CLEANUP_INITIALLY_CLEAR,
|
|
} CMD_VALS;
|
|
|
|
int mystrcpy( LPTSTR pszOut, LPTSTR pszIn, TCHAR chTerm ) {
|
|
BOOL fInQuote = FALSE;
|
|
LPTSTR pszStrt = pszOut;
|
|
|
|
while( *pszIn && !fInQuote && *pszIn != chTerm ) {
|
|
if (*pszIn == TEXT('"')) {
|
|
fInQuote = !fInQuote;
|
|
}
|
|
*pszOut++ = *pszIn++;
|
|
}
|
|
|
|
*pszOut = TEXT('\0');
|
|
return (int)(pszOut - pszStrt);
|
|
}
|
|
|
|
BOOL HasPath( LPTSTR pszFilename ) {
|
|
//
|
|
// Special case null string so it won't get changed
|
|
//
|
|
if (*pszFilename == TEXT('\0'))
|
|
return TRUE;
|
|
|
|
for(; *pszFilename; pszFilename++ ) {
|
|
if (*pszFilename == TEXT(':') || *pszFilename == TEXT('\\') || *pszFilename == TEXT('/')) {
|
|
return TRUE;
|
|
}
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
void HandOffToShell32 (LPCSTR pcszCommand, LPCSTR pcszOptionalArgument)
|
|
|
|
{
|
|
TCHAR szShell32Path[MAX_PATH];
|
|
|
|
if (GetSystemDirectory(szShell32Path, ARRAYSIZE(szShell32Path)) != 0)
|
|
{
|
|
HINSTANCE hInstShell32;
|
|
|
|
PathAppend(szShell32Path, TEXT("shell32.dll"));
|
|
hInstShell32 = LoadLibrary(szShell32Path);
|
|
if (hInstShell32 != NULL)
|
|
{
|
|
typedef HRESULT (*PFNFirstUserLogon) (LPCSTR pcszCommand, LPCSTR pcszOptionalArgument);
|
|
|
|
PFNFirstUserLogon pfnFirstUserLogon;
|
|
|
|
// Call shell32!FirstUserLogon which is Ordinal230 defined in shell32.src.
|
|
|
|
pfnFirstUserLogon = (PFNFirstUserLogon)GetProcAddress(hInstShell32, (LPCSTR)MAKEINTRESOURCE(230));
|
|
if (pfnFirstUserLogon != NULL)
|
|
pfnFirstUserLogon(pcszCommand, pcszOptionalArgument);
|
|
FreeLibrary(hInstShell32);
|
|
}
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Command Parser
|
|
*/
|
|
typedef struct {
|
|
LPSTR pszCmd;
|
|
int idCmd;
|
|
} CMD;
|
|
|
|
CMD aCmds[] = {
|
|
{ "W", CMD_W95_TO_SUR },
|
|
{ "Cvt-Curs", CMD_DAYTONA_CURSORS },
|
|
{ "Fix-Curs", CMD_FIX_CURSOR_PATHS },
|
|
{ "Fix-Folders", CMD_FIX_SPECIAL_FOLDERS},
|
|
{ "Fix-Win-Security", CMD_FIX_WINDOWS_PROFILE_SECURITY},
|
|
{ "Fix-User-Security", CMD_FIX_USER_PROFILE_SECURITY},
|
|
{ "Fix-Policies-Security", CMD_FIX_POLICIES_SECURITY},
|
|
{ "Cvt-Mouse-Schemes", CMD_CVT_SCHEMES_TO_MULTIUSER},
|
|
{ "UpgradeProfileNT4ToNT5", CMD_UPGRADE_NT4_PROFILE_TO_NT5},
|
|
{ "UpgradeSchemesAndNcMetricsToWin2000", CMD_UPGRADE_SCHEMES_AND_NCMETRICS_TO_WIN2000},
|
|
{ "UpgradeSchemesAndNcMetricsFromWin9xToWin2000", CMD_UPGRADE_SCHEMES_AND_NCMETRICS_FROM_WIN9X_TO_WIN2000},
|
|
{ "FixCAPIDirAttrib", CMD_FIX_CAPI_DIR_ATTRIBUTES},
|
|
{ "MergeDesktopAndNormalStreams", CMD_MERGE_DESKTOP_AND_NORMAL_STREAMS },
|
|
{ "MoveAndAdjustIconMetrics", CMD_MOVE_AND_ADJUST_ICON_METRICS },
|
|
{ "Fix-HTML-Help", CMD_FIX_HTML_HELP },
|
|
{ "Set-Screensaver-On-FriendlyUI", CMD_SET_SCREENSAVER_FRIENDLYUI },
|
|
{ "AddConfigurePrograms", CMD_ADD_CONFIGURE_PROGRAMS },
|
|
{ "OCInstallShowIE", CMD_OCINSTALL_SHOW_IE },
|
|
{ "OCInstallReinstallIE", CMD_OCINSTALL_REINSTALL_IE },
|
|
{ "OCInstallHideIE", CMD_OCINSTALL_HIDE_IE },
|
|
{ "OCInstallUserConfigIE", CMD_OCINSTALL_USER_CONFIG_IE },
|
|
{ "OCInstallShowOE", CMD_OCINSTALL_SHOW_OE },
|
|
{ "OCInstallReinstallOE", CMD_OCINSTALL_REINSTALL_OE },
|
|
{ "OCInstallHideOE", CMD_OCINSTALL_HIDE_OE },
|
|
{ "OCInstallUserConfigOE", CMD_OCINSTALL_USER_CONFIG_OE },
|
|
// { "OCInstallShowVM", CMD_OCINSTALL_SHOW_VM },
|
|
{ "OCInstallReinstallVM", CMD_OCINSTALL_REINSTALL_VM },
|
|
// { "OCInstallHideVM", CMD_OCINSTALL_HIDE_VM },
|
|
{ "OCInstallFixup", CMD_OCINSTALL_FIXUP },
|
|
{ "OCInstallUpdate", CMD_OCINSTALL_UPDATE },
|
|
{ "OCInstallCleanupInitiallyClear", CMD_OCINSTALL_CLEANUP_INITIALLY_CLEAR },
|
|
};
|
|
|
|
#define C_CMDS ARRAYSIZE(aCmds)
|
|
|
|
void __cdecl main( int cArgs, char **szArg) {
|
|
int i;
|
|
HRESULT hrOle = E_FAIL;
|
|
|
|
SHMGLogErrMsg("main called with args", (DWORD)cArgs);
|
|
|
|
if (cArgs < 2 || cArgs > 3)
|
|
ExitProcess(1);
|
|
|
|
#ifdef SHMG_DBG
|
|
for(i=1; i<cArgs; i++)
|
|
SHMGLogErrMsg(szArg[i], (DWORD)i);
|
|
#endif
|
|
|
|
for( i = 0; i < C_CMDS && lstrcmpA( szArg[1], aCmds[i].pszCmd ) != 0; i++ );
|
|
|
|
if (i >= C_CMDS)
|
|
ExitProcess(1);
|
|
|
|
if (aCmds[i].idCmd >= CMD_OLE_REQUIRED_START)
|
|
{
|
|
hrOle = OleInitialize(NULL);
|
|
|
|
if (FAILED(hrOle))
|
|
{
|
|
ExitProcess(1);
|
|
}
|
|
}
|
|
|
|
switch( aCmds[i].idCmd ) {
|
|
|
|
case CMD_W95_TO_SUR:
|
|
CvtDeskCPL_Win95ToSUR();
|
|
break;
|
|
|
|
case CMD_DAYTONA_CURSORS:
|
|
CvtCursorsCPL_DaytonaToSUR();
|
|
break;
|
|
|
|
case CMD_FIX_CURSOR_PATHS:
|
|
FixupCursorSchemePaths();
|
|
break;
|
|
|
|
case CMD_FIX_SPECIAL_FOLDERS:
|
|
ResetUserSpecialFolderPaths();
|
|
break;
|
|
|
|
case CMD_FIX_WINDOWS_PROFILE_SECURITY:
|
|
FixWindowsProfileSecurity();
|
|
break;
|
|
|
|
case CMD_FIX_USER_PROFILE_SECURITY:
|
|
FixUserProfileSecurity();
|
|
break;
|
|
|
|
case CMD_FIX_POLICIES_SECURITY:
|
|
FixPoliciesSecurity();
|
|
break;
|
|
|
|
case CMD_CVT_SCHEMES_TO_MULTIUSER:
|
|
CvtCursorSchemesToMultiuser();
|
|
break;
|
|
|
|
case CMD_UPGRADE_NT4_PROFILE_TO_NT5:
|
|
MigrateNT4ToNT5();
|
|
break;
|
|
|
|
case CMD_UPGRADE_SCHEMES_AND_NCMETRICS_TO_WIN2000:
|
|
UpgradeSchemesAndNcMetricsToWin2000();
|
|
break;
|
|
|
|
case CMD_UPGRADE_SCHEMES_AND_NCMETRICS_FROM_WIN9X_TO_WIN2000:
|
|
UpgradeSchemesAndNcMetricsFromWin9xToWin2000(cArgs > 2 ? szArg[2] : NULL);
|
|
break;
|
|
|
|
case CMD_FIX_CAPI_DIR_ATTRIBUTES:
|
|
SetSystemBitOnCAPIDir();
|
|
break;
|
|
|
|
case CMD_MERGE_DESKTOP_AND_NORMAL_STREAMS:
|
|
case CMD_MOVE_AND_ADJUST_ICON_METRICS:
|
|
HandOffToShell32(szArg[1], cArgs > 2 ? szArg[2] : NULL);
|
|
break;
|
|
|
|
case CMD_FIX_HTML_HELP:
|
|
FixHtmlHelp();
|
|
break;
|
|
|
|
case CMD_SET_SCREENSAVER_FRIENDLYUI:
|
|
SetScreensaverOnFriendlyUI();
|
|
break;
|
|
|
|
case CMD_ADD_CONFIGURE_PROGRAMS:
|
|
AddConfigurePrograms();
|
|
break;
|
|
|
|
case CMD_OCINSTALL_FIXUP:
|
|
FixupOptionalComponents();
|
|
break;
|
|
|
|
case CMD_OCINSTALL_UPDATE:
|
|
OCInstallUpdate();
|
|
break;
|
|
|
|
case CMD_OCINSTALL_CLEANUP_INITIALLY_CLEAR:
|
|
OCInstallCleanupInitiallyClear();
|
|
break;
|
|
|
|
case CMD_OCINSTALL_SHOW_IE:
|
|
ShowHideIE(TRUE, FALSE, TRUE);
|
|
break;
|
|
|
|
case CMD_OCINSTALL_REINSTALL_IE:
|
|
ShowHideIE(TRUE, TRUE, TRUE);
|
|
break;
|
|
|
|
case CMD_OCINSTALL_HIDE_IE:
|
|
ShowHideIE(FALSE, FALSE, TRUE);
|
|
break;
|
|
|
|
case CMD_OCINSTALL_USER_CONFIG_IE:
|
|
UserConfigIE();
|
|
break;
|
|
|
|
case CMD_OCINSTALL_SHOW_OE:
|
|
ShowHideOE(TRUE, FALSE, TRUE);
|
|
break;
|
|
|
|
case CMD_OCINSTALL_REINSTALL_OE:
|
|
ShowHideOE(TRUE, TRUE, TRUE);
|
|
break;
|
|
|
|
case CMD_OCINSTALL_HIDE_OE:
|
|
ShowHideOE(FALSE, FALSE, TRUE);
|
|
break;
|
|
|
|
case CMD_OCINSTALL_USER_CONFIG_OE:
|
|
UserConfigOE();
|
|
break;
|
|
|
|
case CMD_OCINSTALL_SHOW_VM:
|
|
// Do nothing...
|
|
break;
|
|
|
|
case CMD_OCINSTALL_REINSTALL_VM:
|
|
ReinstallVM();
|
|
break;
|
|
|
|
case CMD_OCINSTALL_HIDE_VM:
|
|
// Do nothing...
|
|
break;
|
|
|
|
default:
|
|
ExitProcess(2);
|
|
}
|
|
|
|
if (SUCCEEDED(hrOle))
|
|
{
|
|
OleUninitialize();
|
|
}
|
|
|
|
|
|
ExitProcess(0);
|
|
}
|
|
|
|
|
|
|
|
#ifdef SHMG_DBG
|
|
/***************************************************************************\
|
|
*
|
|
* FUNCTION: FmtSprintf( DWORD id, ... );
|
|
*
|
|
* PURPOSE: sprintf but it gets the pattern string from the message rc.
|
|
*
|
|
* History:
|
|
* 03-May-1993 JonPa Created it.
|
|
\***************************************************************************/
|
|
TCHAR g_szDbgBuffer[16384];
|
|
char g_szDbgBufA[16384];
|
|
void Dprintf( LPTSTR pszFmt, ... ) {
|
|
DWORD cb;
|
|
LPVOID psz = g_szDbgBuffer;
|
|
va_list marker;
|
|
|
|
va_start( marker, pszFmt );
|
|
|
|
wvsprintf( g_szDbgBuffer, pszFmt, marker );
|
|
OutputDebugString(g_szDbgBuffer);
|
|
|
|
|
|
#ifdef UNICODE
|
|
cb = WideCharToMultiByte(CP_ACP, 0, g_szDbgBuffer, -1, g_szDbgBufA, sizeof(g_szDbgBufA), NULL, NULL);
|
|
psz = g_szDbgBufA;
|
|
#else
|
|
cb = lstrlen(g_szDbgBuffer) * sizeof(TCHAR);
|
|
#endif
|
|
|
|
WriteFile(GetStdHandle(STD_OUTPUT_HANDLE), psz, cb, &cb, NULL);
|
|
|
|
va_end( marker );
|
|
}
|
|
#endif
|