Windows Kernel

Development Platform:
Visual C++

repair.c：Code Content
							/*
 * repair - Dialog box property sheet for "Repair"
 */
#include "tweakui.h"
#pragma BEGIN_CONST_DATA
KL const c_klRegView = { &g_hkCUSMWCV, c_tszRegedit, c_tszView };
const static DWORD CODESEG rgdwHelp[] = {
#if 0
	IDC_REBUILDCACHEICON,	IDH_REPAIRICON,
	IDC_REBUILDCACHE,	IDH_REPAIRICON,
	IDC_REPAIRFONTFLDICON,	IDH_REPAIRFONTFLD,
	IDC_REPAIRFONTFLD,	IDH_REPAIRFONTFLD,
	IDC_REPAIRREGEDITICON,	IDH_REPAIRREGEDIT,
	IDC_REPAIRREGEDIT,	IDH_REPAIRREGEDIT,
	IDC_REPAIRASSOCICON,	IDH_REPAIRASSOC,
	IDC_REPAIRASSOC,	IDH_REPAIRASSOC,
	IDC_REPAIRDLLSICON,	IDH_REPAIRDLLS,
	IDC_REPAIRDLLS,		IDH_REPAIRDLLS,
#endif
	0,		    	0,
};
typedef UINT (PASCAL *REPAIRPATHPROC)(HWND hdlg, LPTSTR ptszBuf);
/*****************************************************************************
 *
 *  _Repair_FileFromResource
 *
 *	Given a resource ID and a file name, extract the c_tszExe resource
 *	into that file.
 *
 *****************************************************************************/
BOOL PASCAL
_Repair_FileFromResource(UINT idx, LPCTSTR ptszFileName)
{
    BOOL fRc;
    HRSRC hrsrc = FindResource(hinstCur, (LPVOID)idx, c_tszExe);
    if (hrsrc) {
	HGLOBAL hglob = LoadResource(hinstCur, hrsrc);
	if (hglob) {
	    PV pv = LockResource(hglob);
	    if (pv) {
		HFILE hf = _lcreat(ptszFileName, 0);
		if (hf != HFILE_ERROR) {
		    UINT cb = SizeofResource(hinstCur, hrsrc);
		    fRc = _lwrite(hf, pv, cb) == cb;
		    _lclose(hf);
		    if (fRc) {
		    } else {
			DeleteFile(ptszFileName);
		    }
		}
	    } else {
		fRc = 0;
	    }
	} else {
	    fRc = 0;
	}
    } else {
	fRc = 0;
    }
    return fRc;
}
/*****************************************************************************
 *
 *  Repair_RunSetupCqn
 *
 *	Put the Setup program into the specified directory and run it.
 *	WithTempDirectory will clean up the file when we return.
 *
 *****************************************************************************/
BOOL PASCAL
Repair_RunSetupCqn(LPCTSTR cqn, LPVOID pv)
{
    BOOL fRc;
    fRc = _Repair_FileFromResource(IDX_SETUP, c_tszSetupExe);
    if (fRc) {
	STARTUPINFO si;
	PROCESS_INFORMATION pi;
	ZeroMemory(&si, sizeof(si));
	si.cb = sizeof(si);
	fRc = CreateProcess(0, c_tszSetupExe, 0, 0, 0, 0, 0, 0, &si, &pi);
	if (fRc) {
	    CloseHandle(pi.hThread);
	    WaitForSingleObject(pi.hProcess, INFINITE);
	    CloseHandle(pi.hProcess);
	}
    }
    return fRc;
}
#ifndef WINNT
/*****************************************************************************
 *
 *  Repair_MoveFileExWininit
 *
 *	Rename a file via wininit.ini.
 *
 *****************************************************************************/
#pragma BEGIN_CONST_DATA
TCHAR c_tszPercentSEqualsPercentS[] = TEXT("%s=%s");
#pragma END_CONST_DATA
void PASCAL
Repair_MoveFileExWininit(LPCTSTR ptszDst, LPCTSTR ptszSrc)
{
    TCHAR tszWininit[MAX_PATH];
    DWORD cb;
    LPSTR pszBuf;
    WIN32_FIND_DATA wfd;
    HANDLE h;
    GetWindowsDirectory(tszWininit, cA(tszWininit));
    lstrcatnBs(tszWininit, c_tszWininit, MAX_PATH);
    /* INI files are always in ANSI */
    h = FindFirstFile(ptszSrc, &wfd);
    if (h != INVALID_HANDLE_VALUE) {
	FindClose(h);
    } else {
	wfd.nFileSizeLow = 0;
    }
    cb = wfd.nFileSizeLow + MAX_PATH + 1 + MAX_PATH + 3;
    pszBuf = lAlloc(cb);
    if (pszBuf) {
	LPSTR psz;
	if (!GetPrivateProfileSection(c_tszRename, pszBuf, cb, tszWininit)) {
	    /* No such section; create one */
	    /* Already done by LocalAlloc (zero-init) */
	}
	for (psz = pszBuf; psz[0]; psz += lstrlenA(psz) + 1);
	psz += wsprintf(psz, c_tszPercentSEqualsPercentS, ptszDst, ptszSrc);
	psz[1] = '';
	WritePrivateProfileSection(c_tszRename, pszBuf, tszWininit);
	lFree(pszBuf);
    }
}
#endif
/*****************************************************************************
 *
 *  Repair_MoveFileEx
 *
 *	Try to rename the file from existing.dll to existing.bak,
 *	using MoveFileEx or wininit.ini as necessary.
 *
 *	Note that we must use the short name because that's all that
 *	wininit.ini understands.
 *
 *****************************************************************************/
void PASCAL
Repair_MoveFileEx(LPTSTR ptszSrc)
{
    TCHAR tszDst[MAX_PATH];
    GetShortPathName(ptszSrc, ptszSrc, MAX_PATH);
    lstrcpy(tszDst, ptszSrc);
    lstrcpy(tszDst + lstrlen(tszDst) - 4, c_tszDotBak);
    DeleteFile(tszDst);
    if (MoveFile(ptszSrc, tszDst)) {
	/* All done */
    } else if (MoveFileEx(ptszSrc, tszDst, MOVEFILE_DELAY_UNTIL_REBOOT)) {
	/* All done */
#ifndef WINNT
    } else if (g_fNT) {
	/* I did my best */
    } else {		/* Stupid wininit.ini for Windows 95 */
	Repair_MoveFileExWininit(tszDst, ptszSrc);
#endif
    }
}
/*****************************************************************************
 *
 *  _Repair_RunSetup
 *
 *	Check if there are any hidden DLLs on the desktop.
 *
 *	Extract the tiny little "setup" program attached via our resources,
 *	run it, delete it, then tell the user they she may need to restart
 *	the computer for the full effect to kick in.
 *
 *****************************************************************************/
#pragma BEGIN_CONST_DATA
TCHAR c_tszStarDotDll[] = TEXT("*.DLL");
#pragma END_CONST_DATA
BOOL PASCAL
_Repair_RunSetup(HWND hdlg, REPAIRPATHPROC GetRepairPath)
{
    PIDL pidl;
    if (SUCCEEDED(SHGetSpecialFolderLocation(hdlg, CSIDL_DESKTOPDIRECTORY,
					     &pidl))) {
	HANDLE h;
	TCHAR tszDesktop[MAX_PATH];
	TCHAR tszSrc[MAX_PATH];
	WIN32_FIND_DATA wfd;
	SHGetPathFromIDList(pidl, tszDesktop);
	lstrcpy(tszSrc, tszDesktop);
	lstrcatnBsA(tszSrc, c_tszStarDotDll);
	h = FindFirstFile(tszSrc, &wfd);
	if (h != INVALID_HANDLE_VALUE) {
	    do {
		lstrcpy(tszSrc, tszDesktop);
		lstrcatnBsA(tszSrc, wfd.cFileName);
		Repair_MoveFileEx(tszSrc);
	    } while (FindNextFile(h, &wfd));
	    FindClose(h);
	}
	Ole_Free(pidl);
    }
    WithTempDirectory(Repair_RunSetupCqn, 0);
    return IDS_MAYBEREBOOT;
}
/*****************************************************************************
 *
 *  _Repair_GetFontPath
 *
 *      Obtain the path to the Fonts folder.
 *
 *****************************************************************************/
BOOL PASCAL
_Repair_GetFontPath(HWND hdlg, LPTSTR ptszBuf)
{
    PIDL pidlFonts;
    SHGetSpecialFolderLocation(hdlg, CSIDL_FONTS, &pidlFonts);
    SHGetPathFromIDListA(pidlFonts, ptszBuf);
    Ole_Free(pidlFonts);
    return IDX_FONTFOLDER;
}
/*****************************************************************************
 *
 *  _Repair_GetIEPath
 *
 *      Obtain the path to some IE special folder.
 *
 *      If we can't get it from the registry, then it's just
 *      "%windir%ptszSubdir".
 *
 *****************************************************************************/
void PASCAL
_Repair_GetIEPath(LPTSTR ptszBuf, LPCTSTR ptszKey, LPCTSTR ptszSubdir)
{
    HKEY hk;
    ptszBuf[0] = TEXT('');
    if (RegOpenKeyEx(g_hkLMSMWCV, ptszKey, 0, KEY_QUERY_VALUE, &hk)
        == ERROR_SUCCESS) {
        DWORD cb = cbCtch(MAX_PATH);
        RegQueryValueEx(hk, c_tszDirectory, 0, 0, (LPBYTE)ptszBuf,
                        &cb);
        RegCloseKey(hk);
    }
    if (ptszBuf[0] == TEXT('')) {
        GetWindowsDirectory(ptszBuf, MAX_PATH - lstrlen(ptszSubdir) - 1);
        ptszBuf = TweakUi_TrimTrailingBs(ptszBuf);
        *ptszBuf++ = TEXT('\');
        lstrcpy(ptszBuf, ptszSubdir);
    }
}
/*****************************************************************************
 *
 *  _Repair_GetHistoryPath
 *
 *      Obtain the path to the URL History folder.
 *
 *      If we can't get it from the registry, then it's just
 *      "%windir%History".
 *
 *****************************************************************************/
UINT PASCAL
_Repair_GetHistoryPath(HWND hdlg, LPTSTR ptszBuf)
{
    _Repair_GetIEPath(ptszBuf, c_tszUrlHist, c_tszHistory);
    return IDX_HISTORY;
}
/*****************************************************************************
 *
 *  _Repair_GetCachePath
 *
 *      Obtain the path to the Temporary Internet Files folder.
 *
 *      If we can't get it from the registry, then it's just
 *      "%windir%Temporary Internet Files".
 *
 *****************************************************************************/
UINT PASCAL
_Repair_GetCachePath(HWND hdlg, LPTSTR ptszBuf)
{
    _Repair_GetIEPath(ptszBuf, c_tszIECache, c_tszTempInet);
    return IDX_TEMPINET;
}
/*****************************************************************************
 *
 *  _Repair_RepairJunction
 *
 *      Hack at a junction to make it magic again.
 *
 *****************************************************************************/
UINT PASCAL
_Repair_RepairJunction(HWND hdlg, REPAIRPATHPROC GetRepairPath)
{
    PIDL pidlFonts;
    TCHAR tsz[MAX_PATH];
    UINT ids;
    tsz[0] = TEXT('');
    ids = GetRepairPath(hdlg, tsz);
    if (ids && tsz[0]) {
        /* Ignore error; might already exist */
        CreateDirectory(tsz, 0);
        /* Ignore error; might not have permission to change attributes */
        SetFileAttributes(tsz, FILE_ATTRIBUTE_SYSTEM);
        lstrcpy(TweakUi_TrimTrailingBs(tsz), c_tszBSDesktopIni);
        _Repair_FileFromResource(ids, tsz);
    }
    return IDS_MAYBEREBOOT;
}
/*****************************************************************************
 *
 *  _Repair_RepairIconCache
 *
 *      Rebuild the icon cache.
 *
 *****************************************************************************/
UINT PASCAL
_Repair_RepairIconCache(HWND hdlg, REPAIRPATHPROC GetRepairPath)
{
    Misc_RebuildIcoCache();
    return 0;
}
/*****************************************************************************
 *
 *  _Repair_RepairRegedit
 *
 *      Nuke the saved view goo so Regedit won't be hosed.
 *
 *****************************************************************************/
UINT PASCAL
_Repair_RepairRegedit(HWND hdlg, REPAIRPATHPROC GetRepairPath)
{
    DelPkl(&c_klRegView);
    return 0;
}
/*****************************************************************************
 *
 *  _Repair_RepairAssociations
 *
 *      Rebuild the associations.
 *
 *****************************************************************************/
UINT PASCAL
_Repair_RepairAssociations(HWND hdlg, REPAIRPATHPROC GetRepairPath)
{
    if (MessageBoxId(hdlg, IDS_DESKTOPRESETOK,
                     g_tszName, MB_YESNO + MB_DEFBUTTON2) == IDYES) {
        pcdii->fRunShellInf = 1;
        Common_NeedLogoff(hdlg);
        PropSheet_Apply(GetParent(hdlg));
    }
    return 0;
}
/*****************************************************************************
 *
 *  Repair_IsWin95
 *
 *      Nonzero if we are on Windows 95.
 *
 *****************************************************************************/
BOOL PASCAL
Repair_IsWin95(void)
{
    return !g_fNT;
}
/*****************************************************************************
 *
 *  Repair_IsIE3
 *
 *      Nonzero if we are on Windows 95.
 *
 *****************************************************************************/
BOOL PASCAL
Repair_IsIE3(void)
{
    return (BOOL)g_hkCUSMIE;
}
/*****************************************************************************
 *
 *  REPAIRINFO
 *
 *****************************************************************************/
typedef struct REPAIRINFO {
    BOOL (PASCAL *CanRepair)(void);
    UINT (PASCAL *Repair)(HWND hdlg, REPAIRPATHPROC GetRepairPath);
    REPAIRPATHPROC GetRepairPath;
} REPAIRINFO, *PREPAIRINFO;
/*
 *  Note that this needs to be in sync with the IDS_REPAIR strings.
 */
struct REPAIRINFO c_rgri[] = {
    {   0,
        _Repair_RepairIconCache,
        0,
    },                                  /* Rebuild Icons */
    {   0,
        _Repair_RepairJunction,
        _Repair_GetFontPath,
    },                                  /* Repair Font Folder */
    {   0,
        _Repair_RunSetup,
        0,
    },                                  /* Repair System Files */
    {   0,
        _Repair_RepairRegedit,
        0,
    },                                  /* Rebuild Regedit */
    {   Repair_IsWin95,
        _Repair_RepairAssociations,
        0,
    },                                  /* Repair Associations */
    {   Repair_IsIE3,
        _Repair_RepairJunction,
        _Repair_GetHistoryPath,
    },                                  /* Repair URL History */
    {   Repair_IsIE3,
        _Repair_RepairJunction,
        _Repair_GetCachePath,
    },                                  /* Repair Temporary Internet Files */
};
#pragma END_CONST_DATA
/*****************************************************************************
 *
 *  Repair_GetCurSelIndex
 *
 *      Get the index of the selected item.
 *
 *****************************************************************************/
int PASCAL
Repair_GetCurSelIndex(HWND hdlg)
{
    HWND hwnd = GetDlgItem(hdlg, IDC_REPAIRCOMBO);
    int iItem = ComboBox_GetCurSel(hwnd);
    if (iItem >= 0) {
        return ComboBox_GetItemData(hwnd, iItem);
    } else {
        return iItem;
    }
}
/*****************************************************************************
 *
 *  Repair_OnSelChange
 *
 *      Ooh, the selection changed.
 *
 *****************************************************************************/
void PASCAL
Repair_OnSelChange(HWND hdlg)
{
    HWND hwnd = GetDlgItem(hdlg, IDC_REPAIRCOMBO);
    int dids = Repair_GetCurSelIndex(hdlg);
    if (dids >= 0) {
        TCHAR tsz[1024];
        LoadString(hinstCur, IDS_REPAIRHELP+dids, tsz, cA(tsz));
        SetDlgItemText(hdlg, IDC_REPAIRHELP, tsz);
    }
}
/*****************************************************************************
 *
 *  Repair_OnInitDialog
 *
 *	Disable the shell.inf thing on NT, because NT doesn't have one.
 *
 *****************************************************************************/
BOOL PASCAL
Repair_OnInitDialog(HWND hdlg)
{
    HWND hwnd = GetDlgItem(hdlg, IDC_REPAIRCOMBO);
    int dids;
    for (dids = 0; dids < cA(c_rgri); dids++) {
        if (c_rgri[dids].CanRepair == 0 ||
            c_rgri[dids].CanRepair()) {
            TCHAR tsz[MAX_PATH];
            int iItem;
            LoadString(hinstCur, IDS_REPAIR+dids, tsz, cA(tsz));
            iItem = ComboBox_AddString(hwnd, tsz);
            ComboBox_SetItemData(hwnd, iItem, dids);
        }
    }
    ComboBox_SetCurSel(hwnd, 0);
    Repair_OnSelChange(hdlg);
    return 1;
}
/*****************************************************************************
 *
 *  Repair_OnRepairNow
 *
 *      Ooh, go repair something.
 *
 *****************************************************************************/
void PASCAL
Repair_OnRepairNow(HWND hdlg)
{
    HWND hwnd = GetDlgItem(hdlg, IDC_REPAIRCOMBO);
    int dids = Repair_GetCurSelIndex(hdlg);
    if (dids >= 0) {
        UINT ids = c_rgri[dids].Repair(hdlg, c_rgri[dids].GetRepairPath);
        if (ids) {
            MessageBoxId(hdlg, ids, g_tszName, MB_OK);
        }
    }
}
/*****************************************************************************
 *
 *  Repair_OnCommand
 *
 *	Ooh, we got a command.
 *
 *****************************************************************************/
BOOL PASCAL
Repair_OnCommand(HWND hdlg, int id, UINT codeNotify)
{
    switch (id) {
    case IDC_REPAIRCOMBO:
        if (codeNotify == CBN_SELCHANGE) {
            Repair_OnSelChange(hdlg);
        }
        break;
    case IDC_REPAIRNOW:
        if (codeNotify == BN_CLICKED) {
            Repair_OnRepairNow(hdlg);
        }
    }
#if 0
    switch (id) {
    case IDC_REBUILDCACHE:
	if (codeNotify == BN_CLICKED) {
	    Misc_RebuildIcoCache();
	    MessageBoxId(hdlg, IDS_ICONSREBUILT, g_tszName, MB_OK);
	}
	break;
    case IDC_REPAIRFONTFLD:
	if (codeNotify == BN_CLICKED) {
	    if (_Repair_RepairFontFolder(hdlg)) {
		MessageBoxId(hdlg, IDS_MAYBEREBOOT, g_tszName, MB_OK);
	    }
	}
	break;
    case IDC_REPAIRREGEDIT:
	if (codeNotify == BN_CLICKED) {
	    DelPkl(&c_klRegView);
	}
	break;
    case IDC_REPAIRASSOC:
	if (codeNotify == BN_CLICKED) {
	    if (MessageBoxId(hdlg, IDS_DESKTOPRESETOK,
			     g_tszName, MB_YESNO + MB_DEFBUTTON2) == IDYES) {
		pcdii->fRunShellInf = 1;
		Common_NeedLogoff(hdlg);
		PropSheet_Apply(GetParent(hdlg));
	    }
	}
	break;
    case IDC_REPAIRDLLS:
	if (codeNotify == BN_CLICKED) {
	    if (_Repair_RunSetup(hdlg)) {
		MessageBoxId(hdlg, IDS_MAYBEREBOOT, g_tszName, MB_OK);
	    }
	}
	break;
    }
#endif
    return 0;
}
#if 0
/*****************************************************************************
 *
 *  Repair_OnNotify
 *
 *	Ooh, we got a notification.
 *
 *****************************************************************************/
BOOL PASCAL
Repair_OnNotify(HWND hdlg, NMHDR FAR *pnm)
{
    switch (pnm->code) {
    case PSN_APPLY:
	Repair_Apply(hdlg);
	break;
    }
    return 0;
}
#endif
/*****************************************************************************
 *
 *  Our window procedure.
 *
 *****************************************************************************/
/*
 * The HANDLE_WM_* macros weren't designed to be used from a dialog
 * proc, so we need to handle the messages manually.  (But carefully.)
 */
BOOL EXPORT
Repair_DlgProc(HWND hdlg, UINT wm, WPARAM wParam, LPARAM lParam)
{
    switch (wm) {
    case WM_INITDIALOG: return Repair_OnInitDialog(hdlg);
    case WM_COMMAND:
	return Repair_OnCommand(hdlg,
			       (int)GET_WM_COMMAND_ID(wParam, lParam),
			       (UINT)GET_WM_COMMAND_CMD(wParam, lParam));
#if 0
    case WM_NOTIFY:
	return Repair_OnNotify(hdlg, (NMHDR FAR *)lParam);
#endif
    case WM_HELP: Common_OnHelp(lParam, &rgdwHelp[0]); break;
    case WM_CONTEXTMENU: Common_OnContextMenu(wParam, &rgdwHelp[0]); break;
    default: return 0;	/* Unhandled */
    }
    return 1;		/* Handled */
}

						