/*-- Copyright (c) 1999 Microsoft Corporation Module Name: sdbinst.cpp Abstract: installs custom SDB files into AppPatch\Custom, and adds registry entries to point to them Author: dmunsil 12/29/2000 Revision History: Many people contributed over time. (in alphabetical order: clupu, dmunsil, rparsons, vadimb) Notes: --*/ #define _UNICODE #define WIN #define FLAT_32 #define TRUE_IF_WIN32 1 #include #include #include #define _WINDOWS #include #include #include #include #include #include #include #include #include "resource.h" extern "C" { #include "shimdb.h" } // // DJM - turning off strsafe deprecation for now, but all of these deprecated functions // should be replaced. // #include BOOL g_bQuiet; BOOL g_bWin2K; WCHAR g_wszCustom[MAX_PATH]; BOOL g_bAllowPatches = FALSE; HINSTANCE g_hInst; HANDLE g_hLogFile = INVALID_HANDLE_VALUE; typedef enum _INSTALL_MODE { MODE_INSTALL, MODE_UNINSTALL, MODE_CLEANUP, MODE_CONVERT_FORMAT_NEW, MODE_CONVERT_FORMAT_OLD } INSTALL_MODE; #define UNINSTALL_KEY_PATH L"Software\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\" #define APPCOMPAT_KEY L"System\\CurrentControlSet\\Control\\Session Manager\\AppCompatibility" DWORD g_dwWow64Key = (DWORD)-1; UINT SdbGetWindowsDirectory( LPWSTR pwszBuffer, DWORD dwSize ) { pwszBuffer[0] = 0; UINT unRet = GetSystemWindowsDirectoryW(pwszBuffer, dwSize); if (!unRet) { goto out; } if (unRet >= dwSize) { unRet++; goto out; } // // throw a backslash on there if it's not already there // if (pwszBuffer[unRet - 1] != L'\\') { pwszBuffer[unRet] = L'\\'; unRet++; pwszBuffer[unRet] = 0; } out: return unRet; } void __cdecl vPrintError( UINT unRes, ... ) { WCHAR szT[1024]; WCHAR wszFormat[1024]; WCHAR wszCaption[1024]; va_list arglist; if (!g_bQuiet) { if (!LoadStringW(g_hInst, IDS_APP_ERROR_TITLE, wszCaption, ARRAYSIZE(wszCaption))) { return; } if (LoadStringW(g_hInst, unRes, wszFormat, ARRAYSIZE(wszFormat))) { va_start(arglist, unRes); StringCchVPrintfW(szT, ARRAYSIZE(szT), wszFormat, arglist); va_end(arglist); MessageBoxW(NULL, szT, wszCaption, MB_OK | MB_ICONWARNING); } } } void __cdecl vPrintMessage( UINT unRes, ... ) { WCHAR szT[1024]; WCHAR wszFormat[1024]; WCHAR wszCaption[1024]; va_list arglist; if (!g_bQuiet) { if (!LoadStringW(g_hInst, IDS_APP_TITLE, wszCaption, ARRAYSIZE(wszCaption))) { return; } if (LoadStringW(g_hInst, unRes, wszFormat, ARRAYSIZE(wszFormat))) { va_start(arglist, unRes); StringCchVPrintfW(szT, ARRAYSIZE(szT), wszFormat, arglist); va_end(arglist); MessageBoxW(NULL, szT, wszCaption, MB_OK | MB_ICONINFORMATION); } } } void __cdecl vLogMessage( LPCSTR pwszFormat, ... ) { CHAR szT[1024]; va_list arglist; int nLength; va_start(arglist, pwszFormat); StringCchVPrintfA(szT, CHARCOUNT(szT), pwszFormat, arglist); nLength = strlen(szT); va_end(arglist); if (g_hLogFile != INVALID_HANDLE_VALUE) { DWORD dwWritten; WriteFile(g_hLogFile, (LPVOID)szT, (DWORD)nLength, &dwWritten, NULL); } OutputDebugStringA(szT); } DWORD GetWow64Flag( void ) { if (g_dwWow64Key == (DWORD)-1) { if (g_bWin2K) { g_dwWow64Key = 0; // no flag since there is no wow64 on win2k } else { g_dwWow64Key = KEY_WOW64_64KEY; } } return g_dwWow64Key; } VOID OpenLogFile( VOID ) { WCHAR wszLogFile[MAX_PATH]; SdbGetWindowsDirectory(wszLogFile, ARRAYSIZE(wszLogFile)); StringCchCatW(wszLogFile, ARRAYSIZE(wszLogFile), L"AppPatch\\SdbInst.Log"); g_hLogFile = CreateFileW(wszLogFile, GENERIC_WRITE, FILE_SHARE_READ, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL); } VOID CloseLogFile( VOID ) { if (g_hLogFile != INVALID_HANDLE_VALUE) { CloseHandle(g_hLogFile); } g_hLogFile = INVALID_HANDLE_VALUE; } void vPrintHelp( WCHAR* szAppName ) { vPrintMessage(IDS_HELP_TEXT, szAppName); } typedef void (CALLBACK *pfn_ShimFlushCache)(HWND, HINSTANCE, LPSTR, int); void vFlushCache( void ) { HMODULE hAppHelp; pfn_ShimFlushCache pShimFlushCache; hAppHelp = LoadLibraryW(L"apphelp.dll"); if (hAppHelp) { pShimFlushCache = (pfn_ShimFlushCache)GetProcAddress(hAppHelp, "ShimFlushCache"); if (pShimFlushCache) { pShimFlushCache(NULL, NULL, NULL, 0); } } } BOOL bSearchGroupForSID( DWORD dwGroup, BOOL* pfIsMember ) { PSID pSID = NULL; SID_IDENTIFIER_AUTHORITY SIDAuth = SECURITY_NT_AUTHORITY; BOOL fRes = TRUE; if (!AllocateAndInitializeSid(&SIDAuth, 2, SECURITY_BUILTIN_DOMAIN_RID, dwGroup, 0, 0, 0, 0, 0, 0, &pSID)) { return FALSE; } if (!pSID) { return FALSE; } if (!CheckTokenMembership(NULL, pSID, pfIsMember)) { fRes = FALSE; } FreeSid(pSID); return fRes; } BOOL bCanRun( void ) { BOOL fIsAdmin; if (!bSearchGroupForSID(DOMAIN_ALIAS_RID_ADMINS, &fIsAdmin)) { return FALSE; } return fIsAdmin; } WCHAR* wszGetFileFromPath( WCHAR* wszPath ) { WCHAR* szTemp = wcsrchr(wszPath, L'\\'); if (szTemp) { return szTemp + 1; } return NULL; } BOOL bIsAlreadyInstalled( WCHAR* wszPath ) { DWORD dwCustomLen; DWORD dwInputLen; DWORD dwPos; dwCustomLen = wcslen(g_wszCustom); dwInputLen = wcslen(wszPath); if (_wcsnicmp(wszPath, g_wszCustom, dwCustomLen) != 0) { // // it's not in the custom directory // return FALSE; } for (dwPos = dwCustomLen; dwPos < dwInputLen; ++dwPos) { if (wszPath[dwPos] == L'\\') { // // it's in a subdirectory of Custom, // return FALSE; } } return TRUE; } BOOL bGuidToPath( GUID* pGuid, WCHAR* wszPath, DWORD dwPathSize ) { UNICODE_STRING ustrGuid; HRESULT hr1, hr2, hr3; if (!NT_SUCCESS(RtlStringFromGUID(*pGuid, &ustrGuid))) { return FALSE; } hr1 = StringCchCopyW(wszPath, dwPathSize, g_wszCustom); hr2 = StringCchCatW(wszPath, dwPathSize, ustrGuid.Buffer); hr3 = StringCchCatW(wszPath, dwPathSize, L".sdb"); RtlFreeUnicodeString(&ustrGuid); if (FAILED(hr1) || FAILED(hr2) || FAILED(hr3)) { return FALSE; } return TRUE; } BOOL bGetGuid( WCHAR* wszSDB, GUID* pGuid ) { PDB pdb = NULL; TAGID tiDatabase; TAGID tiID; BOOL bRet = FALSE; pdb = SdbOpenDatabase(wszSDB, DOS_PATH); if (!pdb) { vPrintError(IDS_UNABLE_TO_OPEN_FILE, wszSDB); bRet = FALSE; goto out; } tiDatabase = SdbFindFirstTag(pdb, TAGID_ROOT, TAG_DATABASE); if (!tiDatabase) { vPrintError(IDS_NO_DB_TAG, wszSDB); bRet = FALSE; goto out; } ZeroMemory(pGuid, sizeof(GUID)); tiID = SdbFindFirstTag(pdb, tiDatabase, TAG_DATABASE_ID); if (tiID) { if (SdbReadBinaryTag(pdb, tiID, (PBYTE)pGuid, sizeof(GUID))) { bRet = TRUE; } } if (!bRet) { vPrintError(IDS_NO_DB_ID, wszSDB); } out: if (pdb) { SdbCloseDatabase(pdb); pdb = NULL; } return bRet; } typedef enum _TIME_COMPARE { FILE_NEWER, FILE_SAME, FILE_OLDER } TIME_COMPARE; BOOL bOldSdbInstalled( WCHAR* wszPath, WCHAR* wszOldPath, DWORD dwOldPathSize ) { WIN32_FIND_DATAW FindData; GUID guidMain; BOOL bRet = FALSE; HANDLE hFind; // // get the guid from the DB we're installing // if (!bGetGuid(wszPath, &guidMain)) { // // there's no info in this DB, so no way to tell. // return FALSE; } // // get the path to the current file // if (!bGuidToPath(&guidMain, wszOldPath, dwOldPathSize)) { // // couldn't convert to path // return FALSE; } // // check to see if the file exists // hFind = FindFirstFileW(wszOldPath, &FindData); if (hFind != INVALID_HANDLE_VALUE) { // // yup // bRet = TRUE; FindClose(hFind); } return bRet; } BOOL IsKnownDatabaseGUID( GUID* pGuid ) { const GUID* rgpGUID[] = { &GUID_SYSMAIN_SDB, &GUID_APPHELP_SDB, &GUID_SYSTEST_SDB, &GUID_DRVMAIN_SDB, &GUID_MSIMAIN_SDB }; int i; for (i = 0; i < ARRAYSIZE(rgpGUID); ++i) { if (*rgpGUID[i] == *pGuid) { return TRUE; } } return FALSE; } BOOL DatabaseContainsPatch( WCHAR* wszSDB ) { PDB pdb = NULL; TAGID tiDatabase = TAGID_NULL; TAGID tiLibrary = TAGID_NULL; TAGID tiPatch = TAGID_NULL; BOOL bRet = FALSE; pdb = SdbOpenDatabase(wszSDB, DOS_PATH); if (!pdb) { vPrintError(IDS_UNABLE_TO_OPEN_FILE, wszSDB); bRet = FALSE; goto out; } tiDatabase = SdbFindFirstTag(pdb, TAGID_ROOT, TAG_DATABASE); if (!tiDatabase) { vPrintError(IDS_NO_DB_TAG, wszSDB); bRet = FALSE; goto out; } tiLibrary = SdbFindFirstTag(pdb, tiDatabase, TAG_LIBRARY); if (!tiLibrary) { // // this isn't an error -- no library just means no patches // bRet = FALSE; goto out; } tiPatch = SdbFindFirstTag(pdb, tiLibrary, TAG_PATCH); if (tiPatch) { bRet = TRUE; } else { bRet = FALSE; } out: if (pdb) { SdbCloseDatabase(pdb); pdb = NULL; } return bRet; } BOOL bGetInternalNameAndID( WCHAR* wszSDB, WCHAR* wszInternalName, DWORD dwInternalNameSize, GUID* pGuid ) { PDB pdb = NULL; TAGID tiDatabase; TAGID tiName; TAGID tiID; BOOL bRet = FALSE; WCHAR* wszTemp; pdb = SdbOpenDatabase(wszSDB, DOS_PATH); if (!pdb) { vPrintError(IDS_UNABLE_TO_OPEN_FILE, wszSDB); bRet = FALSE; goto out; } tiDatabase = SdbFindFirstTag(pdb, TAGID_ROOT, TAG_DATABASE); if (!tiDatabase) { vPrintError(IDS_NO_DB_TAG, wszSDB); bRet = FALSE; goto out; } tiName = SdbFindFirstTag(pdb, tiDatabase, TAG_NAME); if (tiName) { wszTemp = SdbGetStringTagPtr(pdb, tiName); } if (wszTemp) { StringCchCopyW(wszInternalName, dwInternalNameSize, wszTemp); } else { wszInternalName[0] = 0; } ZeroMemory(pGuid, sizeof(GUID)); tiID = SdbFindFirstTag(pdb, tiDatabase, TAG_DATABASE_ID); if (!tiID) { bRet = FALSE; goto out; } if (!SdbReadBinaryTag(pdb, tiID, (PBYTE)pGuid, sizeof(GUID))) { bRet = FALSE; goto out; } bRet = TRUE; out: if (pdb) { SdbCloseDatabase(pdb); pdb = NULL; } return bRet; } BOOL bFriendlyNameToFile( WCHAR* wszFriendlyName, WCHAR* wszFile, DWORD dwFileSize, WCHAR* wszPath, DWORD dwPathSize ) { WCHAR wszSearchPath[MAX_PATH]; WIN32_FIND_DATAW FindData; BOOL bRet = FALSE; WCHAR wszInternalTemp[256]; WCHAR wszFileTemp[MAX_PATH]; GUID guidTemp; HANDLE hFind; StringCchCopyW(wszSearchPath, ARRAYSIZE(wszSearchPath), g_wszCustom); StringCchCatW(wszSearchPath, ARRAYSIZE(wszSearchPath), L"*.sdb"); hFind = FindFirstFileW(wszSearchPath, &FindData); if (hFind == INVALID_HANDLE_VALUE) { return FALSE; } while (hFind != INVALID_HANDLE_VALUE) { StringCchCopyW(wszFileTemp, ARRAYSIZE(wszFileTemp), g_wszCustom); StringCchCatW(wszFileTemp, ARRAYSIZE(wszFileTemp), FindData.cFileName); if (!bGetInternalNameAndID(wszFileTemp, wszInternalTemp, ARRAYSIZE(wszInternalTemp), &guidTemp)) { goto nextFile; } if (_wcsicmp(wszInternalTemp, wszFriendlyName) == 0) { bRet = TRUE; StringCchCopyW(wszFile, dwFileSize, FindData.cFileName); StringCchCopyW(wszPath, dwPathSize, wszFileTemp); FindClose(hFind); break; } nextFile: if (!FindNextFileW(hFind, &FindData)) { FindClose(hFind); hFind = INVALID_HANDLE_VALUE; } } return bRet; } BOOL bFindInstallName( WCHAR* wszPath, WCHAR* wszInstallPath, DWORD dwInstallPathSize ) { GUID guidMain; // // get the guid from the DB we're installing // if (!bGetGuid(wszPath, &guidMain)) { // // there's no info in this DB, so no way to tell. // return FALSE; } // // get the path to the current file // if (!bGuidToPath(&guidMain, wszInstallPath, dwInstallPathSize)) { // // couldn't convert to path // return FALSE; } return TRUE; } // // this function is necessary because RegDeleteKey doesn't work right with // a 32-bit app deleting 64-bit reg keys // LONG LocalRegDeleteKeyW ( IN HKEY hKey, IN LPCWSTR lpSubKey ) { LONG lRes; HKEY hSubKey = NULL; lRes = RegOpenKeyExW(hKey, lpSubKey, 0, KEY_ALL_ACCESS|GetWow64Flag(), &hSubKey); if (lRes != ERROR_SUCCESS) { return lRes; } lRes = NtDeleteKey(hSubKey); RegCloseKey(hSubKey); return lRes; } VOID InstallW2KData( WCHAR* pszEntryName, LPCWSTR pszGuidDB ) { HKEY hKey; WCHAR wszRegPath[MAX_PATH * 2]; DWORD dwDisposition, cbData; LONG lResult = 0; BYTE data[16] = {0x0c, 0, 0, 0, 0, 0, 0, 0, 0x06, 0, 0, 0, 0, 0, 0, 0}; HRESULT hr; // // This is Windows 2000 - attempt to add custom SDB specific data. // hr = StringCchPrintfW(wszRegPath, ARRAYSIZE(wszRegPath), L"%s\\%s", APPCOMPAT_KEY, pszEntryName); if (FAILED(hr)) { vPrintError(IDS_BUFFER_TOO_SMALL); return; } lResult = RegCreateKeyExW(HKEY_LOCAL_MACHINE, wszRegPath, 0, NULL, 0, KEY_SET_VALUE, NULL, &hKey, &dwDisposition); if (ERROR_SUCCESS != lResult) { if (ERROR_ACCESS_DENIED == lResult) { vPrintError(IDS_NEED_INSTALL_PERMISSION); return; } else { vPrintError(IDS_CANT_CREATE_REG_KEY, pszEntryName); return; } } // // Set the registry values. // lResult = RegSetValueExW(hKey, pszGuidDB, 0, REG_BINARY, data, sizeof(data)); if (ERROR_SUCCESS != lResult) { RegCloseKey(hKey); if (ERROR_ACCESS_DENIED == lResult) { vPrintError(IDS_NEED_INSTALL_PERMISSION); } else { vPrintError(IDS_CANT_SET_REG_VALUE, pszEntryName); } return; } data[0] = 0; hr = StringCchPrintfW(wszRegPath, ARRAYSIZE(wszRegPath), L"DllPatch-%s", pszGuidDB); if (FAILED(hr)) { vPrintError(IDS_BUFFER_TOO_SMALL); return; } lResult = RegSetValueExW(hKey, wszRegPath, 0, REG_SZ, data, 2 * sizeof(WCHAR)); if (ERROR_SUCCESS != lResult) { RegCloseKey(hKey); if (ERROR_ACCESS_DENIED == lResult) { vPrintError(IDS_NEED_INSTALL_PERMISSION); } else { vPrintError(IDS_CANT_SET_REG_VALUE, pszEntryName); } return; } RegCloseKey(hKey); } VOID RemoveW2KData( WCHAR* pszEntryName, LPCWSTR pszGuidDB ) { HKEY hKey; WCHAR wszRegPath[MAX_PATH]; LONG lResult = 0; DWORD dwValues; HRESULT hr; // // This is Windows 2000 - attempt to remove custom SDB specific data. // hr = StringCchPrintfW(wszRegPath, ARRAYSIZE(wszRegPath), L"%s\\%s", APPCOMPAT_KEY, pszEntryName); if (FAILED(hr)) { vPrintError(IDS_BUFFER_TOO_SMALL); return; } lResult = RegOpenKeyExW(HKEY_LOCAL_MACHINE, wszRegPath, 0, KEY_ALL_ACCESS|GetWow64Flag(), &hKey); if (ERROR_SUCCESS != lResult) { if (ERROR_ACCESS_DENIED == lResult) { vPrintError(IDS_NEED_INSTALL_PERMISSION); return; } else { vPrintError(IDS_CANT_OPEN_REG_KEY, wszRegPath); return; } } RegDeleteValueW(hKey, pszGuidDB); hr = StringCchPrintfW(wszRegPath, ARRAYSIZE(wszRegPath), L"DllPatch-%s", pszGuidDB); if (FAILED(hr)) { vPrintError(IDS_BUFFER_TOO_SMALL); return; } RegDeleteValueW(hKey, wszRegPath); // // Figure out if we should delete the key, if there aren't any more values left // lResult = RegQueryInfoKey(hKey, NULL, NULL, NULL, NULL, NULL, NULL, &dwValues, NULL, NULL, NULL, NULL); RegCloseKey(hKey); hKey = NULL; if (dwValues != 0) { return; } lResult = RegOpenKeyExW(HKEY_LOCAL_MACHINE, APPCOMPAT_KEY, 0, KEY_ALL_ACCESS|GetWow64Flag(), &hKey); if (ERROR_SUCCESS == lResult) { lResult = LocalRegDeleteKeyW(hKey, pszEntryName); } if (lResult != ERROR_SUCCESS) { vPrintError(IDS_CANT_DELETE_REG_KEY, pszEntryName, APPCOMPAT_KEY); } RegCloseKey(hKey); } // Caller is responsible for freeing the memory using delete []. LPWSTR ExpandItem( LPCWSTR pwszItem ) { LPWSTR pwszItemExpand = NULL; // Get the required length. DWORD dwLenExpand = ExpandEnvironmentStringsW(pwszItem, NULL, 0); if (!dwLenExpand) { goto err; } // // Make room for "\\?\" // dwLenExpand += 4; pwszItemExpand = new WCHAR [dwLenExpand]; if (!pwszItemExpand) { goto err; } LPWSTR pwszTemp = pwszItemExpand; DWORD dwTemp = dwLenExpand; StringCchCopyW(pwszItemExpand, dwLenExpand, L"\\\\?\\"); pwszTemp += 4; dwTemp -= 4; if (!ExpandEnvironmentStringsW(pwszItem, pwszTemp, dwTemp)) { goto err; } return pwszItemExpand; err: if (pwszItemExpand) { delete [] pwszItemExpand; } return NULL; } DWORD GiveUsersWriteAccess( LPWSTR pwszDir ) { DWORD dwRes; EXPLICIT_ACCESS ea; PACL pOldDACL; PACL pNewDACL = NULL; PSECURITY_DESCRIPTOR pSD = NULL; SID_IDENTIFIER_AUTHORITY SIDAuth = SECURITY_NT_AUTHORITY; PSID pUsersSID = NULL; dwRes = GetNamedSecurityInfoW(pwszDir, SE_FILE_OBJECT, DACL_SECURITY_INFORMATION, NULL, NULL, &pOldDACL, NULL, &pSD); if (ERROR_SUCCESS != dwRes) { goto Cleanup; } if (!AllocateAndInitializeSid(&SIDAuth, 2, SECURITY_BUILTIN_DOMAIN_RID, DOMAIN_ALIAS_RID_USERS, 0, 0, 0, 0, 0, 0, &pUsersSID) ) { dwRes = ERROR_NOT_ENOUGH_MEMORY; goto Cleanup; } // // Initialize an EXPLICIT_ACCESS structure for the new ACE. // ZeroMemory(&ea, sizeof(EXPLICIT_ACCESS)); ea.grfAccessPermissions = FILE_GENERIC_WRITE | FILE_GENERIC_READ | DELETE; ea.grfAccessMode = GRANT_ACCESS; ea.grfInheritance = SUB_CONTAINERS_AND_OBJECTS_INHERIT; ea.Trustee.TrusteeForm = TRUSTEE_IS_SID; ea.Trustee.TrusteeType = TRUSTEE_IS_GROUP; ea.Trustee.ptstrName = (LPTSTR)pUsersSID; // // Create a new ACL that merges the new ACE // into the existing DACL. // dwRes = SetEntriesInAcl(1, &ea, pOldDACL, &pNewDACL); if (ERROR_SUCCESS != dwRes) { goto Cleanup; } dwRes = SetNamedSecurityInfoW(pwszDir, SE_FILE_OBJECT, DACL_SECURITY_INFORMATION, NULL, NULL, pNewDACL, NULL); if (ERROR_SUCCESS != dwRes) { goto Cleanup; } Cleanup: if (pSD) { LocalFree(pSD); } if (pUsersSID) { FreeSid(pUsersSID); } if (pNewDACL) { LocalFree(pNewDACL); } return dwRes; } BOOL SetupLUAAllUserDir( LPCWSTR pwszAllUserDir ) { BOOL bRes = FALSE; LPWSTR pwszExpandedDir = ExpandItem(pwszAllUserDir); if (!pwszExpandedDir) { vPrintError(IDS_CANT_EXPAND_DIR, pwszAllUserDir); return FALSE; } // // Create the directory if it doesn't already exist. // DWORD dwAttributes = GetFileAttributesW(pwszExpandedDir); if (dwAttributes != -1) { if (!(dwAttributes & FILE_ATTRIBUTE_DIRECTORY)) { vPrintError(IDS_OBJECT_ALREADY_EXISTS, pwszExpandedDir); goto Cleanup; } } else { if (!CreateDirectoryW(pwszExpandedDir, NULL)) { vPrintError(IDS_CANT_CREATE_DIRECTORY, pwszExpandedDir, GetLastError()); goto Cleanup; } } // // Give the Users group full control access (power users can already modify // files in this directory). // if (GiveUsersWriteAccess((LPWSTR)pwszExpandedDir) != ERROR_SUCCESS) { vPrintError(IDS_CANT_SET_ACLS, pwszExpandedDir); goto Cleanup; } bRes = TRUE; Cleanup: delete [] pwszExpandedDir; return bRes; } BOOL ProcessLUAAction( PDB pdb, TAGID tiExe ) { LPWSTR szAllUserDir = NULL; // // See if this EXE has an ACTION node. Currently only EXEs shimmed with the LUA // shims have ACTION nodes. // TAGID tiAction = SdbFindFirstTag(pdb, tiExe, TAG_ACTION); if (tiAction) { // // the ACTION node in the EXE shimmed with LUA looks like this: // // // // // TAGID tiName, tiType, tiData, tiValue; LPWSTR szName, szType, szData; if ((tiName = SdbFindFirstTag(pdb, tiAction, TAG_NAME)) && (szName = SdbGetStringTagPtr(pdb, tiName))) { if (!wcscmp(szName, L"REDIRECT")) { if ((tiType = SdbFindFirstTag(pdb, tiAction, TAG_ACTION_TYPE)) && (szType = SdbGetStringTagPtr(pdb, tiType))) { if (!wcscmp(szType, L"ChangeACLs")) { if ((tiData = SdbFindFirstTag(pdb, tiAction, TAG_DATA)) && (tiValue = SdbFindFirstTag(pdb, tiData, TAG_DATA_STRING)) && (szAllUserDir = SdbGetStringTagPtr(pdb, tiValue))) { if (!SetupLUAAllUserDir(szAllUserDir)) { return FALSE; } } } } } } } return TRUE; } // buffer size is in characters (unicode) BOOL InstallSdbEntry( WCHAR* szEntryName, // entry name (foo.exe or layer name) LPCWSTR pszGuidDB, // guid database id in string format ULONGLONG ullSdbTimeStamp, // representation of a timestamp BOOL bLayer // true if layer name ) { LONG lRes; WCHAR szRegPath[MAX_PATH * 2]; // MAX_PATH for the reg path, and MAX_PATH for the exe name WCHAR szDBName[MAX_PATH]; // this is used in older (win2k) versions HRESULT hr; BOOL bReturn = FALSE; HKEY hKey = NULL; StringCchCopyW(szDBName, ARRAYSIZE(szDBName), pszGuidDB); StringCchCatW(szDBName, ARRAYSIZE(szDBName), L".sdb"); pszGuidDB = szDBName; // // If this is Win2K, add data to the AppCompatibility key. // if (g_bWin2K) { InstallW2KData(szEntryName, pszGuidDB); } // else we have a string hr = StringCchPrintfW(szRegPath, ARRAYSIZE(szRegPath), (bLayer ? L"%s\\Layers\\%s": L"%s\\%s"), APPCOMPAT_KEY_PATH_CUSTOM_W, szEntryName); if (FAILED(hr)) { // error vPrintError(IDS_BUFFER_TOO_SMALL); goto HandleError; } lRes = RegCreateKeyExW(HKEY_LOCAL_MACHINE, szRegPath, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS|GetWow64Flag(), NULL, &hKey, NULL); // // on install, we want to quit if we hit an error. // BUGBUG - should we undo whatever we've already completed? // if (lRes != ERROR_SUCCESS) { vPrintError(IDS_CANT_CREATE_REG_KEY, szRegPath); goto HandleError; } lRes = RegSetValueExW(hKey, pszGuidDB, 0, REG_QWORD, (PBYTE)&ullSdbTimeStamp, sizeof(ullSdbTimeStamp)); if (lRes != ERROR_SUCCESS) { vPrintError(IDS_CANT_CREATE_VALUE, szRegPath); goto HandleError; } bReturn = TRUE; HandleError: if (hKey != NULL) { RegCloseKey(hKey); } return bReturn; } BOOL UninstallSdbEntry( WCHAR* szEntryName, // foo.exe or layer name LPCWSTR pszGuidDB, // guid (database id) in string format BOOL bLayer // true is layer ) { LONG lRes; WCHAR szRegPath[MAX_PATH * 2]; // MAX_PATH for the reg path, and MAX_PATH for the exe name WCHAR szDBName[MAX_PATH]; HRESULT hr; BOOL bReturn = FALSE; HKEY hKey = NULL; DWORD dwValues; WCHAR szOldInstallName[MAX_PATH]; StringCchCopyW(szDBName, ARRAYSIZE(szDBName), pszGuidDB); StringCchCatW(szDBName, ARRAYSIZE(szDBName), L".sdb"); pszGuidDB = szDBName; if (g_bWin2K) { RemoveW2KData(szEntryName, pszGuidDB); } hr = StringCchPrintfW(szRegPath, ARRAYSIZE(szRegPath), (bLayer ? L"%s\\Layers\\%s": L"%s\\%s"), APPCOMPAT_KEY_PATH_CUSTOM_W, szEntryName); if (FAILED(hr)) { // error vPrintError(IDS_BUFFER_TOO_SMALL); goto Out; } lRes = RegOpenKeyExW(HKEY_LOCAL_MACHINE, szRegPath, 0, KEY_ALL_ACCESS|GetWow64Flag(), &hKey); // // if we fail to open a key on uninstall, keep going, so // hopefully we can get as much uninstalled as possible. // if (lRes != ERROR_SUCCESS) { if (lRes == ERROR_ACCESS_DENIED) { vPrintError(IDS_NEED_UNINSTALL_PERMISSION); goto HandleError; } else { // // DO NOT report an error - this key might have been cleaned up during the // previous path, such as when identical exe names appear in the same db // for instance, two setup.exe's -- the first pass will clean up the key, // second path will fail to open them right here // // vPrintError(IDS_CANT_OPEN_REG_KEY, szRegPath); goto Out; } } lRes = RegDeleteValueW(hKey, pszGuidDB); if (lRes != ERROR_SUCCESS) { if (lRes == ERROR_ACCESS_DENIED) { vPrintError(IDS_NEED_UNINSTALL_PERMISSION); goto HandleError; // fatal error } else { // // bugbug - pszSdbInstallName // if (lRes == ERROR_FILE_NOT_FOUND) { WCHAR wszOldFormat[MAX_PATH]; // // aha, value's not there, try old format // StringCchCopyW(wszOldFormat, ARRAYSIZE(wszOldFormat), pszGuidDB); StringCchCatW(wszOldFormat, ARRAYSIZE(wszOldFormat), L".sdb"); lRes = RegDeleteValueW(hKey, wszOldFormat); } if (lRes != ERROR_SUCCESS) { vPrintError(IDS_CANT_DELETE_REG_VALUE, pszGuidDB, szRegPath); } } } // // figure out if we should delete the key, if there aren't any more values left // lRes = RegQueryInfoKey(hKey, NULL, NULL, NULL, NULL, NULL, NULL, &dwValues, NULL, NULL, NULL, NULL); if (dwValues == 0) { RegCloseKey(hKey); hKey = NULL; hr = StringCchPrintfW(szRegPath, sizeof(szRegPath)/sizeof(szRegPath[0]), (bLayer ? L"%s\\Layers": L"%s"), APPCOMPAT_KEY_PATH_CUSTOM_W); if (FAILED(hr)) { // error vPrintError(IDS_BUFFER_TOO_SMALL); goto Out; } lRes = RegOpenKeyExW(HKEY_LOCAL_MACHINE, szRegPath, 0, KEY_WRITE|GetWow64Flag(), &hKey); if (lRes != ERROR_SUCCESS) { vPrintError(IDS_CANT_OPEN_REG_KEY, szRegPath); goto Out; } lRes = LocalRegDeleteKeyW(hKey, szEntryName); if (lRes != ERROR_SUCCESS) { vPrintError(IDS_CANT_DELETE_REG_KEY, szEntryName, szRegPath); } } Out: bReturn = TRUE; HandleError: if (hKey != NULL) { RegCloseKey(hKey); } return bReturn; } NTSTATUS SDBAPI FindCharInUnicodeString( ULONG Flags, PCUNICODE_STRING StringToSearch, PCUNICODE_STRING CharSet, USHORT* NonInclusivePrefixLength ) { LPCWSTR pch; // // implement only the case when we move backward // if (Flags != RTL_FIND_CHAR_IN_UNICODE_STRING_START_AT_END) { return STATUS_NOT_IMPLEMENTED; } pch = StringToSearch->Buffer + StringToSearch->Length / sizeof(WCHAR); while (pch >= StringToSearch->Buffer) { if (_tcschr(CharSet->Buffer, *pch)) { // // got the char // if (NonInclusivePrefixLength) { *NonInclusivePrefixLength = (USHORT)(pch - StringToSearch->Buffer) * sizeof(WCHAR); } return STATUS_SUCCESS; } pch--; } // // We haven't found it. Return failure. // return STATUS_NOT_FOUND; } // // Database list entry // Used to represent a particular installed database // typedef struct tagSDBLISTENTRY { LIST_ENTRY ListEntry; // link list stuff ULONGLONG ullTimeStamp; // database install timestamp GUID guidDB; // database guid WCHAR szTimeStamp[32]; // time stamp in string form WCHAR szGuidDB[64]; // guid in string form WCHAR szDatabasePath[1]; // database path - we store only the name } SDBLISTENTRY, *PSDBLISTENTRY; /*++ AddSdbListEntry Adds a particular database to the list of installed sdbs (maintained internally) parses database path to retrieve database name [in out] pHeadList - pointer to the associated list head for the installed sdbs [in] guidDB - database guid [in] TimeStamp - database time stamp [in] pszDatabasePath - final database path returns true if success --*/ BOOL AddSdbListEntry( PLIST_ENTRY pHeadList, GUID& guidDB, ULONGLONG& TimeStamp, LPCWSTR pszDatabasePath ) { // // out of database path, recover the database name // UNICODE_STRING ustrPath = { 0 }; USHORT uPrefix; UNICODE_STRING ustrPathSep = RTL_CONSTANT_STRING(L"\\/"); NTSTATUS Status; UNICODE_STRING ustrGUID = { 0 }; if (pszDatabasePath != NULL) { RtlInitUnicodeString(&ustrPath, pszDatabasePath); Status = FindCharInUnicodeString(RTL_FIND_CHAR_IN_UNICODE_STRING_START_AT_END, &ustrPath, &ustrPathSep, &uPrefix); if (NT_SUCCESS(Status) && (uPrefix + sizeof(WCHAR)) < ustrPath.Length) { // // uPrefix is number of character preceding the one we found not including it // ustrPath.Buffer += uPrefix / sizeof(WCHAR) + 1; ustrPath.Length -= (uPrefix + sizeof(WCHAR)); ustrPath.MaximumLength -= (uPrefix + sizeof(WCHAR)); } // // at this point ustrPath has just the filename -- this is what we shall use // } PBYTE Buffer = new BYTE[sizeof(SDBLISTENTRY) + ustrPath.Length]; if (Buffer == NULL) { vLogMessage("[AddSdbListEntry] Failed to allocate 0x%lx bytes\n", sizeof(SDBLISTENTRY) + ustrPath.Length); return FALSE; } PSDBLISTENTRY pSdbEntry = (PSDBLISTENTRY)Buffer; pSdbEntry->guidDB = guidDB; pSdbEntry->ullTimeStamp = TimeStamp; Status = RtlStringFromGUID(guidDB, &ustrGUID); if (!NT_SUCCESS(Status)) { // // we can't convert guid to string? memory allocation failure // vLogMessage("[AddSdbListEntry] Failed to convert guid to string Status 0x%lx\n", Status); delete[] Buffer; return FALSE; } RtlCopyMemory(&pSdbEntry->szGuidDB[0], &ustrGUID.Buffer[0], ustrGUID.Length); pSdbEntry->szGuidDB[ustrGUID.Length/sizeof(WCHAR)] = L'\0'; RtlFreeUnicodeString(&ustrGUID); StringCchPrintfW(pSdbEntry->szTimeStamp, ARRAYSIZE(pSdbEntry->szTimeStamp), L"%.16I64X", TimeStamp); RtlCopyMemory(&pSdbEntry->szDatabasePath[0], &ustrPath.Buffer[0], ustrPath.Length); pSdbEntry->szDatabasePath[ustrPath.Length / sizeof(WCHAR)] = L'\0'; InsertHeadList(pHeadList, &pSdbEntry->ListEntry); return TRUE; } // // only pGuidDB OR pwszGuid is allowed // /*++ FindSdbListEntry Finds and returns an sdb list entry given a guid (in string or binary form) Whenever possible pwszGuid is used (if it's supplied). If pwszGuid happens to be an arbitrary filename -- it is assumed that it's the name of an installed sdb file as registered. [in] pHeadList - list of the installed sdbs [in] pwszGuid - guid or guid.sdb [out] ppSdbListEntry - if found, this receives a pointer to sdb list entry [in] pGuidDB - guid in binary form returns true if matching database has been located in the list --*/ BOOL FindSdbListEntry( PLIST_ENTRY pHeadList, LPCWSTR pwszGuid, // guid, possibly with trailing '.sdb' PSDBLISTENTRY* ppSdbListEntry, GUID* pGuidDB // guid ) { UNICODE_STRING ustrDot = RTL_CONSTANT_STRING(L"."); UNICODE_STRING ustrPath; USHORT uPrefix; NTSTATUS Status; PLIST_ENTRY pEntry; PSDBLISTENTRY pSdbEntry; GUID guidDB; BOOL bGuidSearch = TRUE; BOOL bFound = FALSE; LPCWSTR pch; if (pGuidDB == NULL) { RtlInitUnicodeString(&ustrPath, pwszGuid); Status = FindCharInUnicodeString(RTL_FIND_CHAR_IN_UNICODE_STRING_START_AT_END, &ustrPath, &ustrDot, &uPrefix); if (NT_SUCCESS(Status)) { // // uPrefix is number of character preceding the one we found not including it // ustrPath.Length = uPrefix; } // // convert to guid, but check first // pch = pwszGuid + wcsspn(pwszGuid, L" \t"); if (*pch != L'{') { // not a guid, why convert ? bGuidSearch = FALSE; } else { Status = RtlGUIDFromString(&ustrPath, &guidDB); if (!NT_SUCCESS(Status)) { // // failed, so use database path instead // bGuidSearch = FALSE; } } } else { guidDB = *pGuidDB; // guid search only } pEntry = pHeadList->Flink; while (pEntry != pHeadList && !bFound) { // // convert entry by subtracting the offset of the list entry // pSdbEntry = (PSDBLISTENTRY)((PBYTE)pEntry - OFFSETOF(SDBLISTENTRY, ListEntry)); // // compare db guids or paths // if (bGuidSearch) { bFound = RtlEqualMemory(&pSdbEntry->guidDB, &guidDB, sizeof(GUID)); } else { bFound = !_wcsicmp(pSdbEntry->szDatabasePath, pwszGuid); } pEntry = pEntry->Flink; } // // we have found an entry ? return it -- note that pEntry would have advanced while pSdbEntry // still points to the entry we have found // if (bFound) { *ppSdbListEntry = pSdbEntry; } return bFound; } /*++ CleanupSdbList Performs cleanup for the installed sdb list returns nothing --*/ VOID CleanupSdbList( PLIST_ENTRY pSdbListHead ) { PLIST_ENTRY pEntry; PSDBLISTENTRY pSdbEntry; PBYTE Buffer; pEntry = pSdbListHead->Flink; if (pEntry == NULL) { return; } while (pEntry != pSdbListHead) { pSdbEntry = (PSDBLISTENTRY)((PBYTE)pEntry - OFFSETOF(SDBLISTENTRY, ListEntry)); pEntry = pEntry->Flink; Buffer = (PBYTE)pSdbEntry; delete[] Buffer; } } /*++ ConvertInstalledSdbsToNewFormat Converts installed sdbs to new format, which involves storing (or verifying) the timestamp for each installed sdb file. This function also builds a list of sdbs used elsewhere [in] hKey - a key handle for hklm/..../InstalledSdb [in out] pSdbListHead - list head for the installed sdbs returns true if successful --*/ BOOL ConvertInstalledSdbsToNewFormat( HKEY hKey, // hklm/.../InstalledSdb PLIST_ENTRY pSdbListHead // we fill this list with our sdbs for later ) { DWORD dwIndex = 0; WCHAR szSubKeyName[MAX_PATH]; PWCHAR pwszKeyName; DWORD dwBufferSize; FILETIME ftLastWriteTime; HKEY hKeyEntry = NULL; LONG lResult; ULARGE_INTEGER liTimeStamp; UNICODE_STRING ustrGuid; GUID guidDB; NTSTATUS Status; WCHAR szDatabasePath[MAX_PATH]; PWCHAR pszDatabasePath; DWORD dwType; BOOL bSuccess = TRUE; while (TRUE) { dwBufferSize = sizeof(szSubKeyName)/sizeof(szSubKeyName[0]); lResult = RegEnumKeyExW(hKey, dwIndex, szSubKeyName, &dwBufferSize, NULL, NULL, NULL, &ftLastWriteTime); ++dwIndex; if (lResult != ERROR_SUCCESS) { // // done if no more keys, else some sort of error // bugbug // if (lResult == ERROR_NO_MORE_ITEMS) { // // we are done, clean // break; } // // this is unexpected // vLogMessage("[ConvertInstalledSdbsToNewFormat] RegEnumKeyExW for index 0x%lx returned unexpected error 0x%lx\n", dwIndex, lResult); break; } RtlInitUnicodeString(&ustrGuid, szSubKeyName); Status = RtlGUIDFromString(&ustrGuid, &guidDB); if (!NT_SUCCESS(Status)) { // // BUGBUG - failed to convert the guid (subkey name!) // extraneous entry, log warning // vLogMessage("[ConvertInstalledSdbsToNewFormat] Failed to convert string to guid for \"%ls\" status 0x%lx\n", szSubKeyName, Status); continue; } // // for this db entry we have to set the timestamp // lResult = RegOpenKeyExW(hKey, szSubKeyName, 0, KEY_READ|KEY_WRITE|GetWow64Flag(), &hKeyEntry); if (lResult != ERROR_SUCCESS) { // // bad error ? // BUGBUG vLogMessage("[ConvertInstalledSdbsToNewFormat] Failed to open subkey \"%ls\" error 0x%lx\n", szSubKeyName, lResult); continue; } // // now check the value // dwBufferSize = sizeof(liTimeStamp.QuadPart); lResult = RegQueryValueExW(hKeyEntry, L"DatabaseInstallTimeStamp", NULL, &dwType, (PBYTE)&liTimeStamp.QuadPart, &dwBufferSize); if (lResult != ERROR_SUCCESS || dwType != REG_BINARY) { // // we may either have this value already -- if not, set it up now // liTimeStamp.LowPart = ftLastWriteTime.dwLowDateTime; liTimeStamp.HighPart = ftLastWriteTime.dwHighDateTime; vLogMessage("[Info] Database \"%ls\" receives timestamp \"%.16I64X\"\n", szSubKeyName, liTimeStamp.QuadPart); lResult = RegSetValueExW(hKeyEntry, L"DatabaseInstallTimeStamp", 0, REG_BINARY, (PBYTE)&liTimeStamp.QuadPart, sizeof(liTimeStamp.QuadPart)); if (lResult != ERROR_SUCCESS) { // // error, ignore for now // vLogMessage("[ConvertInstalledSdbsToNewFormat] Failed to set timestamp value for database \"%ls\" value \"%.16I64X\" error 0x%lx\n", szSubKeyName, liTimeStamp.QuadPart, lResult); } } // // at this point we have : // sdb guid (in szSubKeyName) // time stamp in liTimeStamp // // // query also database path // pszDatabasePath = &szDatabasePath[0]; dwBufferSize = sizeof(szDatabasePath); lResult = RegQueryValueExW(hKeyEntry, L"DatabasePath", NULL, &dwType, (PBYTE)pszDatabasePath, &dwBufferSize); if (lResult != ERROR_SUCCESS || dwType != REG_SZ) { // // no database path // warn basically corrupt database path // vLogMessage("[ConvertInstalledSdbsToNewFormat] Failed to query database path for \"%s\" error 0x%lx\n", szSubKeyName, lResult); pszDatabasePath = NULL; } // // optional check: we can check here whether the sdb file does exist // // // add this sdb to our cache // if (!AddSdbListEntry(pSdbListHead, guidDB, liTimeStamp.QuadPart, pszDatabasePath)) { // // failed to add list entry - we cannot continue // bSuccess = FALSE; break; } RegCloseKey(hKeyEntry); hKeyEntry = NULL; } if (hKeyEntry != NULL) { RegCloseKey(hKeyEntry); } // // we are done converting entries -- and we have also collected cache of sdb info // return bSuccess; } // // this stucture is used to cache values associated with any particular entry (exe) // typedef struct tagSDBVALUEENTRY { LIST_ENTRY ListEntry; // link PSDBLISTENTRY pSdbEntry; // this entry belongs to this database WCHAR szValueName[1]; // value name as we got it from registry } SDBVALUEENTRY, *PSDBVALUEENTRY; /*++ AddValueEntry Adds an new link list element to the list of values [in out] pValueListHead - link list of values [in] pSdbEntry - pointer to a cached entry from sdb list [in] pwszValueName - value name as we got it from the db (something like {guid} or {guid}.sdb) returns true if successful --*/ BOOL AddValueEntry( PLIST_ENTRY pValueListHead, PSDBLISTENTRY pSdbEntry, LPCWSTR pwszValueName ) { PSDBVALUEENTRY pValueEntry; PBYTE Buffer; DWORD dwSize; dwSize = sizeof(SDBVALUEENTRY) + wcslen(pwszValueName) * sizeof(WCHAR); Buffer = new BYTE[dwSize]; if (Buffer == NULL) { // // out of memory // vLogMessage("[AddValueEntry] Failed to allocate buffer for %ls 0x%lx bytes\n", pwszValueName, dwSize); return FALSE; } pValueEntry = (PSDBVALUEENTRY)Buffer; pValueEntry->pSdbEntry = pSdbEntry; StringCchCopyW(pValueEntry->szValueName, (dwSize - sizeof(SDBVALUEENTRY) + sizeof(WCHAR)), pwszValueName); InsertHeadList(pValueListHead, &pValueEntry->ListEntry); return TRUE; } /*++ WriteEntryValue Writes value for a particular entry (exe or layer name), deletes old value associated with this particular database for this exe (or layer) [in] hKey - handle for an entry (for instance hklm/Software/Microsoft/Windows NT/CurrentVersion/AppcompatFlags/Custom/Notepad.exe) [in] pValueEntry - pointer to a value entry element from the value list [in] bWriteNewFormat - whether we are asked to write new or old format returns true if successful --*/ BOOL WriteEntryValue( HKEY hKey, PSDBVALUEENTRY pValueEntry, BOOL bWriteNewFormat // if true -- write new format else old format ) { LONG lResult; BOOL bSuccess = FALSE; LPCWSTR pValueName; if (bWriteNewFormat) { pValueName = pValueEntry->pSdbEntry->szGuidDB; lResult = RegSetValueExW(hKey, pValueName, 0, REG_QWORD, (PBYTE)&pValueEntry->pSdbEntry->ullTimeStamp, sizeof(pValueEntry->pSdbEntry->ullTimeStamp)); if (lResult != ERROR_SUCCESS) { // // we can't do this entry ? // vLogMessage("[WriteEntryValue] Failed to write qword value \"%ls\"=\"%.16I64X\" error 0x%lx\n", pValueEntry->pSdbEntry->szGuidDB, pValueEntry->pSdbEntry->ullTimeStamp, lResult); goto cleanup; } // // nuke old entry // } else { // // old style format please // pValueName = pValueEntry->pSdbEntry->szDatabasePath; lResult = RegSetValueExW(hKey, pValueName, 0, REG_SZ, (PBYTE)L"", sizeof(WCHAR)); if (lResult != ERROR_SUCCESS) { // // trouble -- error // vLogMessage("[WriteEntryValue] Failed to write string value \"%ls\" error 0x%lx\n", pValueEntry->pSdbEntry->szDatabasePath, lResult); goto cleanup; } } // // if we are here -- success, check to see if we can delete the old value // if (_wcsicmp(pValueEntry->szValueName, pValueName) != 0) { lResult = RegDeleteValueW(hKey, pValueEntry->szValueName); if (lResult != ERROR_SUCCESS) { vLogMessage("[WriteEntryValue] Failed to delete value \"%ls\" error 0x%lx\n", pValueEntry->szValueName, lResult); } } bSuccess = TRUE; cleanup: return bSuccess; } /*++ ConvertEntryToNewFormat Converts a particular entry (layer or exe) [in] hKeyParent - key handle for a parent key (for instance hklm/Software/Microsoft/Windows NT/CurrentVersion/AppcompatFlags/Custom when pwszEntryName == "Notepad.exe" or hklm/Software/Microsoft/Windows NT/CurrentVersion/AppcompatFlags/Custom/Layers when pwszEntryName == "RunLayer" [in] pwszEntryName - Either exe name or layer name [in] pSdbListHead - cached list of installed databases [in] bNewFormat - whether to use new or old format returns true if successful --*/ BOOL ConvertEntryToNewFormat( HKEY hKeyParent, LPCWSTR pwszEntryName, PLIST_ENTRY pSdbListHead, BOOL bConvertToNewFormat // true if converting to new format, false if reverting ) { LONG lResult; DWORD dwValues; DWORD dwMaxValueNameLen; DWORD dwMaxValueLen; DWORD dwType; DWORD dwValueNameSize; DWORD dwValueSize; LPWSTR pwszValueName = NULL; LPBYTE pValue = NULL; PSDBLISTENTRY pSdbEntry; DWORD dwIndex; LIST_ENTRY ValueList = { 0 }; PSDBVALUEENTRY pValueEntry; PLIST_ENTRY pValueList; PBYTE Buffer; BOOL bSuccess = FALSE; HKEY hKey = NULL; // // loop through values, for each value - find sdb and write out new entry // then delete old entry // lResult = RegOpenKeyExW(hKeyParent, pwszEntryName, 0, KEY_READ|KEY_WRITE|GetWow64Flag(), &hKey); if (lResult != ERROR_SUCCESS) { vLogMessage("[ConvertEntryToNewFormat] Failed to open key \"%ls\" error 0x%lx\n", pwszEntryName, lResult); goto cleanup; } lResult = RegQueryInfoKeyW(hKey, NULL, NULL, // class/class buffer NULL, // reserved NULL, NULL, // subkeys/max subkey length NULL, // max class len &dwValues, // value count &dwMaxValueNameLen, &dwMaxValueLen, NULL, NULL); if (lResult != ERROR_SUCCESS) { // // failed to query the key, very bad // bugbug vLogMessage("[ConvertEntryToNewFormat] Failed to query key information \"%ls\" error 0x%lx\n", pwszEntryName, lResult); goto cleanup; } // // allocate buffers // pwszValueName = new WCHAR[dwMaxValueNameLen + 1]; pValue = new BYTE[dwMaxValueLen]; if (pValue == NULL || pwszValueName == NULL) { // // bugbug // vLogMessage("[ConvertEntryToNewFormat] Failed to allocate memory buffer entry \"%ls\" (0x%lx, 0x%lx)\n", pwszEntryName, dwMaxValueNameLen, dwMaxValueLen); goto cleanup; } InitializeListHead(&ValueList); // // we have dwValues -- the count of values // for (dwIndex = 0; dwIndex < dwValues; ++dwIndex) { dwValueNameSize = dwMaxValueNameLen + 1; dwValueSize = dwMaxValueLen; lResult = RegEnumValueW(hKey, dwIndex, pwszValueName, &dwValueNameSize, NULL, &dwType, (PBYTE)pValue, &dwValueSize); // // check if we are successful // if (lResult != ERROR_SUCCESS) { if (lResult == ERROR_NO_MORE_ITEMS) { // // oops -- we ran out of values!!! Unexpected, but ok // vLogMessage("[ConvertEntryToNewFormat] RegEnumValue unexpectedly reports no more items for \"%ls\" index 0x%lx\n", pwszEntryName, dwIndex); break; } // // log error and continue // vLogMessage("[ConvertEntryToNewFormat] RegEnumValue failed for \"%ls\" index 0x%lx error 0x%lx\n", pwszEntryName, dwIndex, lResult); continue; } if (bConvertToNewFormat) { if (dwType != REG_SZ) { // // bad entry for sure -- this could be a new entry // log warning // if (dwType == REG_QWORD || (dwType == REG_BINARY && dwValueSize == sizeof(ULONGLONG))) { // // new style entry ? // if (wcsrchr(pwszValueName, L'.') == NULL && *pwszValueName == L'{' && *(pwszValueName + wcslen(pwszValueName) - 1) == L'}') { vLogMessage("[Info] Entry \"%ls\" value \"%ls\" already in new format.\n", pwszEntryName, pwszValueName); continue; } } // // very likely - some entry we do not understand // vLogMessage("[ConvertEntryToNewFormat] Bad value type (0x%lx) for entry \"%ls\" value \"%ls\" index 0x%lx\n", dwType, pwszEntryName, pwszValueName, dwIndex); continue; } // // search by pwszValueName (which happens to be the GUID.sdb) // this may be any kind of a string -- not nec. guid // if (!FindSdbListEntry(pSdbListHead, pwszValueName, &pSdbEntry, NULL)) { // // error - sdb not found! // vLogMessage("[ConvertEntryToNewFormat] Failed to find database \"%ls\" for entry \"%ls\" index 0x%lx\n", pwszValueName, pwszEntryName, dwIndex); continue; } } else { // // check the type first, if this is a new style entry - this will be bin // if (dwType == REG_SZ && wcsrchr(pwszValueName, L'.') != NULL && *(LPCWSTR)pValue == L'\0') { vLogMessage("[Info] Entry \"%ls\" value \"%ls\" is already in required (old) format.\n", pwszEntryName, pwszValueName); continue; } if (dwType != REG_QWORD && (dwType != REG_BINARY || dwValueSize < sizeof(ULONGLONG))) { // // error -- we don't know what this entry is, go to the next one // print warning actually // vLogMessage("[ConvertEntryToNewFormat] Bad value type (0x%lx) or size (0x%lx) for entry \"%ls\" value \"%ls\" index 0x%lx\n", dwType, dwValueSize, pwszEntryName, pwszValueName, dwIndex); continue; } if (!FindSdbListEntry(pSdbListHead, pwszValueName, &pSdbEntry, NULL)) { // // we're in trouble -- an entry has no registered database // vLogMessage("[ConvertEntryToNewFormat] Failed to find database for value \"%ls\" for entry \"%ls\" index 0x%lx\n", pwszValueName, pwszEntryName, dwIndex); continue; } } // // we have found entry and we're ready to write it out, queue it up // if (!AddValueEntry(&ValueList, pSdbEntry, pwszValueName)) { // // bugbug can't add value entry // vLogMessage("[ConvertEntryToNewFormat] Failed to add value \"%ls\" for entry \"%ls\" index 0x%lx\n", pwszValueName, pwszEntryName, dwIndex); goto cleanup; } } // // we have gone through all the values, write loop // bSuccess = TRUE; pValueList = ValueList.Flink; while (pValueList != &ValueList) { pValueEntry = (PSDBVALUEENTRY)((PBYTE)pValueList - OFFSETOF(SDBVALUEENTRY, ListEntry)); // // we can point to the next entry now // if (!WriteEntryValue(hKey, pValueEntry, bConvertToNewFormat)) { // // error, can't convert entry // continue though so that we cleanout the list vLogMessage("[ConvertEntryToNewFormat] Failed to write value for entry \"%ls\"\n", pwszEntryName); } pValueList = pValueList->Flink; } cleanup: if (ValueList.Flink) { pValueList = ValueList.Flink; while (pValueList != &ValueList) { Buffer = (PBYTE)pValueList - OFFSETOF(SDBVALUEENTRY, ListEntry); pValueList = pValueList->Flink; delete[] Buffer; } } if (hKey != NULL) { RegCloseKey(hKey); } if (pwszValueName != NULL) { delete[] pwszValueName; } if (pValue != NULL) { delete[] pValue; } return bSuccess; } /*++ ConvertFormat This function handles format conversions [in] bConvertToNewFormat - true if conversion old->new, false otherwise returns true if success --*/ BOOL ConvertFormat( BOOL bConvertToNewFormat ) { LIST_ENTRY SdbList = { 0 }; // installed sdbs cache HKEY hKey; LONG lResult; DWORD dwIndex; WCHAR szSubKeyName[MAX_PATH]; DWORD dwBufferSize; WCHAR szKeyPath[MAX_PATH]; BOOL bSuccess = FALSE; // // first convert installed sdbs // open installed sdb key // lResult = RegOpenKeyExW(HKEY_LOCAL_MACHINE, APPCOMPAT_KEY_PATH_INSTALLEDSDB_W, // path to InstalledSDB 0, KEY_READ|KEY_WRITE|GetWow64Flag(), &hKey); if (lResult != ERROR_SUCCESS) { // // perhaps no dbs are installed ? // if (lResult == ERROR_FILE_NOT_FOUND) { // // no installed sdbs -- no problem // vLogMessage("[ConvertFormat] No Installed sdbs found\n"); return TRUE; } // // some sort of error has occured // vLogMessage("[ConvertFormat] Failed to open key \"%ls\" Error 0x%lx\n", APPCOMPAT_KEY_PATH_INSTALLEDSDB_W, lResult); return FALSE; } // // note that ConvertInstalledSdbsToNewFormat works properly for both install and uninstall cases // InitializeListHead(&SdbList); if (!ConvertInstalledSdbsToNewFormat(hKey, &SdbList)) { goto cleanup; } // done with Installed sdbs RegCloseKey(hKey); hKey = NULL; // // next up is entry conversion -- first enum exes, then layers // lResult = RegOpenKeyExW(HKEY_LOCAL_MACHINE, APPCOMPAT_KEY_PATH_CUSTOM_W, 0, KEY_READ|KEY_WRITE|GetWow64Flag(), &hKey); if (lResult != ERROR_SUCCESS) { // // what is this? // if (lResult == ERROR_FILE_NOT_FOUND && !IsListEmpty(&SdbList)) { vLogMessage("[ConvertFormat] Failed to open \"%ls\" - check consistency\n", APPCOMPAT_KEY_PATH_CUSTOM_W); } else { vLogMessage("[ConvertFormat] Failed to open \"%ls\" error 0x%lx\n", APPCOMPAT_KEY_PATH_CUSTOM_W, lResult); } goto cleanup; } dwIndex = 0; while (TRUE) { dwBufferSize = sizeof(szSubKeyName)/sizeof(szSubKeyName[0]); lResult = RegEnumKeyExW(hKey, dwIndex, szSubKeyName, &dwBufferSize, NULL, NULL, NULL, NULL); ++dwIndex; if (lResult != ERROR_SUCCESS) { if (lResult == ERROR_NO_MORE_ITEMS) { break; } // // some sort of error, log and continue // vLogMessage("[ConvertFormat] RegEnumKey (entries) returned error for index 0x%lx error 0x%lx\n", dwIndex, lResult); break; } // // skip layers for now // if (!_wcsicmp(szSubKeyName, L"Layers")) { continue; } // for each of these -- call fixup function if (!ConvertEntryToNewFormat(hKey, szSubKeyName, &SdbList, bConvertToNewFormat)) { vLogMessage("[ConvertFormat] Failed to convert entry \"%ls\"\n", szSubKeyName); } } RegCloseKey(hKey); hKey = NULL; // // next up - layers // StringCchCopyW(szKeyPath, ARRAYSIZE(szKeyPath), APPCOMPAT_KEY_PATH_CUSTOM_W); StringCchCatW(szKeyPath, ARRAYSIZE(szKeyPath), L"\\Layers"); // // open and enum layers // lResult = RegOpenKeyExW(HKEY_LOCAL_MACHINE, szKeyPath, 0, KEY_READ|KEY_WRITE|GetWow64Flag(), &hKey); if (lResult != ERROR_SUCCESS) { // maybe dead ? if (lResult == ERROR_FILE_NOT_FOUND) { // // it's ok, maybe we have none of those ? // vLogMessage("[ConvertFormat] No layers found\n"); goto ConvertComplete; } vLogMessage("[ConvertFormat] Failed to open \"%ls\" error 0x%lx\n", szKeyPath, lResult); goto cleanup; } dwIndex = 0; while (TRUE) { dwBufferSize = sizeof(szSubKeyName)/sizeof(szSubKeyName[0]); lResult = RegEnumKeyExW(hKey, dwIndex, szSubKeyName, &dwBufferSize, NULL, NULL, NULL, NULL); ++dwIndex; if (lResult != ERROR_SUCCESS) { // check if this was the last entry if (lResult == ERROR_NO_MORE_ITEMS) { // clean break break; } // some sort of error, log and continue vLogMessage("[ConvertFormat] RegEnumKey (layers) returned error for index 0x%lx error 0x%lx\n", dwIndex, lResult); break; } // for each of these -- call fixup function if (!ConvertEntryToNewFormat(hKey, szSubKeyName, &SdbList, bConvertToNewFormat)) { vLogMessage("[ConvertFormat] Failed to convert entry \"%ls\"\n", szSubKeyName); } } RegCloseKey(hKey); hKey = NULL; ConvertComplete: bSuccess = TRUE; cleanup: if (hKey != NULL) { RegCloseKey(hKey); } // // free SdbList // CleanupSdbList(&SdbList); return bSuccess; } BOOL ProcessMSIPackages( PDB pdb, TAGID tiDatabase, LPCWSTR pszGuidDB, ULONGLONG ullSdbTimeStamp, INSTALL_MODE eMode) { TAGID tiMsiPackage; TAGID tiMsiPackageID; GUID* pGuidID; GUID GuidID; WCHAR szRegPath[MAX_PATH]; BOOL bReturn = TRUE; WCHAR wszGuid[64]; UNICODE_STRING ustrGuid = { 0 }; tiMsiPackage = SdbFindFirstTag(pdb, tiDatabase, TAG_MSI_PACKAGE); while (tiMsiPackage && bReturn) { // // we have a package, extract/find TAG_MSI_PACKAGE_ID // tiMsiPackageID = SdbFindFirstTag(pdb, tiMsiPackage, TAG_MSI_PACKAGE_ID); if (!tiMsiPackageID) { if (eMode == MODE_CLEANUP || eMode == MODE_UNINSTALL) { goto NextPackage; } else { vPrintError(IDS_MISSING_PACKAGE_ID); bReturn = FALSE; break; } } pGuidID = (GUID*)SdbGetBinaryTagData(pdb, tiMsiPackageID); if (pGuidID == NULL) { if (eMode == MODE_CLEANUP || eMode == MODE_UNINSTALL) { goto NextPackage; } else { vPrintError(IDS_MISSING_PACKAGE_ID); bReturn = FALSE; break; } } RtlCopyMemory(&GuidID, pGuidID, sizeof(GUID)); if (!NT_SUCCESS(RtlStringFromGUID(GuidID, &ustrGuid))) { vPrintError(IDS_GUID_BAD_FORMAT); bReturn = FALSE; break; } RtlCopyMemory(wszGuid, ustrGuid.Buffer, ustrGuid.Length); wszGuid[ustrGuid.Length / sizeof(WCHAR)] = TEXT('\0'); if (eMode == MODE_INSTALL) { bReturn = InstallSdbEntry(wszGuid, pszGuidDB, ullSdbTimeStamp, FALSE); } else { bReturn = UninstallSdbEntry(wszGuid, pszGuidDB, FALSE); } RtlFreeUnicodeString(&ustrGuid); NextPackage: tiMsiPackage = SdbFindNextTag(pdb, tiDatabase, tiMsiPackage); } return bReturn; } #define MAX_FRIENDLY_NAME_LEN 256 BOOL bHandleInstall( WCHAR* wszSdbPath, INSTALL_MODE eMode, WCHAR* wszSdbInstallPath, DWORD dwSdbInstallPathSize ) { PDB pdb = NULL; int i; WCHAR wszSdbName[MAX_PATH]; WCHAR wszSdbInstallName[MAX_PATH]; HKEY hKey = NULL; LONG lRes; TAGID tiDatabase, tiExe, tiLayer; TAGID tiDBName = TAGID_NULL; WCHAR* pszDBName = NULL; WCHAR wszFriendlyName[MAX_FRIENDLY_NAME_LEN]; WCHAR* wszTemp; GUID guidDB; NTSTATUS Status; FILETIME SystemTime; BOOL bRet = TRUE; UNICODE_STRING ustrGUID; ULARGE_INTEGER TimeStamp = { 0 }; // // determine the timestamp (for the install case) // if (eMode == MODE_INSTALL) { GetSystemTimeAsFileTime(&SystemTime); TimeStamp.LowPart = SystemTime.dwLowDateTime; TimeStamp.HighPart = SystemTime.dwHighDateTime; } assert(wszSdbPath && wszSdbInstallPath); if (!wszSdbPath || !wszSdbInstallPath) { bRet = FALSE; goto quickOut; } ZeroMemory(wszFriendlyName, sizeof(wszFriendlyName)); // // get the full path from the file name // wszTemp = wszGetFileFromPath(wszSdbPath); if (!wszTemp) { vPrintMessage(IDS_UNABLE_TO_GET_FILE); bRet = FALSE; goto quickOut; } StringCchCopyW(wszSdbName, ARRAYSIZE(wszSdbName), wszTemp); if (wcscmp(wszSdbName, L"sysmain.sdb") == 0) { vPrintError(IDS_CANT_INSTALL_SYS); bRet = FALSE; goto quickOut; } if (GetFileAttributesW(wszSdbPath) != -1 && bIsAlreadyInstalled(wszSdbPath)) { if (eMode == MODE_INSTALL) { // // they asked us to install, it's installed, so we're done // vPrintMessage(IDS_ALREADY_INSTALLED, wszSdbPath); goto quickOut; } } else { if (eMode == MODE_UNINSTALL) { // // they asked us to uninstall, it's not installed, so we're done // vPrintMessage(IDS_NOT_INSTALLED, wszSdbPath); goto quickOut; } } if (eMode == MODE_INSTALL) { // // find out what file name we're going to use for installing // if (!bFindInstallName(wszSdbPath, wszSdbInstallPath, dwSdbInstallPathSize)) { bRet = FALSE; goto quickOut; } } else if (eMode == MODE_CLEANUP) { // // we're cleaning up a bad install, so we need to get the install name from the // install path // wszTemp = wszGetFileFromPath(wszSdbInstallPath); if (!wszTemp) { vPrintMessage(IDS_UNABLE_TO_GET_FILE); bRet = FALSE; goto quickOut; } } else { // // we're uninstalling, so the install name is the given name // and the install path is the given path // StringCchCopyW(wszSdbInstallPath, dwSdbInstallPathSize, wszSdbPath); } // // try to get the guid for later // if (!bGetGuid(wszSdbPath, &guidDB)) { bRet = FALSE; goto out; } // // check whether the guid is coopted from one of the known databases // if (IsKnownDatabaseGUID(&guidDB)) { vPrintError(IDS_CANT_INSTALL_SYS); bRet = FALSE; goto quickOut; } // // in all cases, install name is the db GUID // Status = RtlStringFromGUID(guidDB, &ustrGUID); if (!NT_SUCCESS(Status)) { bRet = FALSE; goto out; } RtlCopyMemory(wszSdbInstallName, ustrGUID.Buffer, ustrGUID.Length); wszSdbInstallName[ustrGUID.Length/sizeof(WCHAR)] = L'\0'; RtlFreeUnicodeString(&ustrGUID); // // if we're installing, make sure the root tags are in place // if (eMode == MODE_INSTALL) { lRes = RegCreateKeyExW(HKEY_LOCAL_MACHINE, APPCOMPAT_KEY_PATH_W, 0, NULL, 0, KEY_ALL_ACCESS|GetWow64Flag(), NULL, &hKey, NULL); if (lRes != ERROR_SUCCESS) { if (lRes == ERROR_ACCESS_DENIED) { vPrintError(IDS_NEED_INSTALL_PERMISSION); } else { vPrintError(IDS_CANT_CREATE_REG_KEY, APPCOMPAT_KEY_PATH_W); } bRet = FALSE; goto out; } RegCloseKey(hKey); hKey = NULL; lRes = RegCreateKeyExW(HKEY_LOCAL_MACHINE, APPCOMPAT_KEY_PATH_CUSTOM_W, 0, NULL, 0, KEY_ALL_ACCESS|GetWow64Flag(), NULL, &hKey, NULL); if (lRes != ERROR_SUCCESS) { vPrintError(IDS_CANT_CREATE_REG_KEY, APPCOMPAT_KEY_PATH_CUSTOM_W); bRet = FALSE; goto out; } RegCloseKey(hKey); hKey = NULL; } // Open the DB. pdb = SdbOpenDatabase(wszSdbPath, DOS_PATH); if (pdb == NULL) { vPrintError(IDS_UNABLE_TO_OPEN_FILE, wszSdbPath); bRet = FALSE; goto out; } tiDatabase = SdbFindFirstTag(pdb, TAGID_ROOT, TAG_DATABASE); if (!tiDatabase) { vPrintError(IDS_NO_DB_TAG, wszSdbPath); bRet = FALSE; goto out; } // // get the friendly name of the database // tiDBName = SdbFindFirstTag(pdb, tiDatabase, TAG_NAME); if (tiDBName) { pszDBName = SdbGetStringTagPtr(pdb, tiDBName); } // // if we don't find a friendly name, use the SDB file name // if (pszDBName) { StringCchCopyW(wszFriendlyName, ARRAYSIZE(wszFriendlyName), pszDBName); } else { StringCchCopyW(wszFriendlyName, ARRAYSIZE(wszFriendlyName), wszSdbName); } tiExe = SdbFindFirstTag(pdb, tiDatabase, TAG_EXE); while (tiExe) { WCHAR szRegPath[MAX_PATH]; TAGID tiName; WCHAR *szName; TAGID tiTemp; tiName = SdbFindFirstTag(pdb, tiExe, TAG_NAME); if (!tiName) { bRet = FALSE; if (eMode == MODE_CLEANUP || eMode == MODE_UNINSTALL) { goto nextExe; } else { vPrintError(IDS_NO_EXE_NAME); goto quickOut; } } szName = SdbGetStringTagPtr(pdb, tiName); if (!szName) { bRet = FALSE; if (eMode == MODE_CLEANUP || eMode == MODE_UNINSTALL) { goto nextExe; } else { vPrintError(IDS_NO_EXE_NAME_PTR); goto quickOut; } } if (eMode == MODE_INSTALL) { if (!InstallSdbEntry(szName, wszSdbInstallName, TimeStamp.QuadPart, FALSE) || !ProcessLUAAction(pdb, tiExe)) { bRet = FALSE; goto out; } } else { if (!UninstallSdbEntry(szName, wszSdbInstallName, FALSE)) { goto quickOut; } } nextExe: tiTemp = tiExe; tiExe = SdbFindNextTag(pdb, tiDatabase, tiExe); // // fallback to ensure we don't get caught in a loop because of a quirk in // sdbapi and a corrupt database. If we get handed back the same TAGID as before, // get out. // if (tiExe == tiTemp) { break; } } // // Loop through the published layers // tiLayer = SdbFindFirstTag(pdb, tiDatabase, TAG_LAYER); while (tiLayer) { WCHAR szRegPath[MAX_PATH]; TAGID tiName; WCHAR* szName; TAGID tiTemp; tiName = SdbFindFirstTag(pdb, tiLayer, TAG_NAME); if (!tiName) { bRet = FALSE; if (eMode == MODE_CLEANUP || eMode == MODE_UNINSTALL) { goto nextLayer; } else { vPrintError(IDS_NO_EXE_NAME); goto quickOut; } } szName = SdbGetStringTagPtr(pdb, tiName); if (!szName) { bRet = FALSE; if (eMode == MODE_CLEANUP || eMode == MODE_UNINSTALL) { goto nextLayer; } else { vPrintError(IDS_NO_EXE_NAME_PTR); goto quickOut; } } if (eMode == MODE_INSTALL) { if (!InstallSdbEntry(szName, wszSdbInstallName, TimeStamp.QuadPart, TRUE)) { bRet = FALSE; goto out; } } else { if (!UninstallSdbEntry(szName, wszSdbInstallName, TRUE)) { goto quickOut; } } nextLayer: tiTemp = tiLayer; tiLayer = SdbFindNextTag(pdb, tiDatabase, tiLayer); // // fallback to ensure we don't get caught in a loop because of a quirk in // sdbapi and a corrupt database. If we get handed back the same TAGID as before, // get out. // if (tiLayer == tiTemp) { break; } } if (!ProcessMSIPackages(pdb, tiDatabase, wszSdbInstallName, TimeStamp.QuadPart, eMode)) { bRet = FALSE; goto quickOut; } if (pdb) { SdbCloseDatabase(pdb); pdb = NULL; } // // now that we've handled the registry keys, copy the file // if (eMode == MODE_INSTALL) { // // ensure the directory exists // CreateDirectoryW(g_wszCustom, NULL); if (!CopyFileW(wszSdbPath, wszSdbInstallPath, TRUE)) { vPrintError(IDS_CANT_COPY_FILE, wszSdbInstallPath); bRet = FALSE; goto out; } } else { // // ensure that we don't fail because of read-only files // SetFileAttributesW(wszSdbInstallPath, FILE_ATTRIBUTE_NORMAL); if (!DeleteFileW(wszSdbInstallPath)) { vPrintError(IDS_CANT_DELETE_FILE, wszSdbInstallPath); bRet = FALSE; } } // // set up or delete the uninstall registry keys // if (eMode == MODE_INSTALL) { WCHAR wszSDBInstPath[MAX_PATH]; WCHAR wszUninstallPath[MAX_PATH]; WCHAR wszUninstallString[MAX_PATH * 2 + 10]; HRESULT hr; // // goofball hack required because of crazy redirection strategy on IA64 // wszSDBInstPath[0] = 0; SdbGetWindowsDirectory(wszSDBInstPath, ARRAYSIZE(wszSDBInstPath)); #if defined(_WIN64) StringCchCatW(wszSDBInstPath, ARRAYSIZE(wszSDBInstPath), L"System32\\sdbinst.exe"); #else StringCchCatW(wszSDBInstPath, ARRAYSIZE(wszSDBInstPath), L"SysWow64\\sdbinst.exe"); #endif if (GetFileAttributesW(wszSDBInstPath) == -1) { // // there's no SysWow64 directory, so we'll just use system32 // wszSDBInstPath[0] = 0; SdbGetWindowsDirectory(wszSDBInstPath, ARRAYSIZE(wszSDBInstPath)); StringCchCatW(wszSDBInstPath, ARRAYSIZE(wszSDBInstPath), L"system32\\sdbinst.exe"); } StringCchCopyW(wszUninstallPath, ARRAYSIZE(wszUninstallPath), UNINSTALL_KEY_PATH); StringCchCatW(wszUninstallPath, ARRAYSIZE(wszUninstallPath), wszSdbInstallName); StringCchCatW(wszUninstallPath, ARRAYSIZE(wszUninstallPath), L".sdb"); lRes = RegCreateKeyExW(HKEY_LOCAL_MACHINE, wszUninstallPath, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS|GetWow64Flag(), NULL, &hKey, NULL); if (lRes != ERROR_SUCCESS) { vPrintError(IDS_CANT_CREATE_REG_KEY, wszUninstallPath); bRet = FALSE; goto out; } lRes = RegSetValueExW(hKey, L"DisplayName", 0, REG_SZ, (PBYTE)wszFriendlyName, (wcslen(wszFriendlyName) + 1) * sizeof(WCHAR)); if (lRes != ERROR_SUCCESS) { vPrintError(IDS_CANT_CREATE_VALUE, wszUninstallPath); bRet = FALSE; goto out; } hr = StringCchPrintfW(wszUninstallString, ARRAYSIZE(wszUninstallString), L"%s -u \"%s\"", wszSDBInstPath, wszSdbInstallPath); if (FAILED(hr)) { vPrintError(IDS_CANT_CREATE_VALUE, wszUninstallPath); bRet = FALSE; goto out; } lRes = RegSetValueExW(hKey, L"UninstallString", 0, REG_SZ, (PBYTE)wszUninstallString, (wcslen(wszUninstallString) + 1) * sizeof(WCHAR)); if (lRes != ERROR_SUCCESS) { vPrintError(IDS_CANT_CREATE_VALUE, wszUninstallPath); bRet = FALSE; goto out; } RegCloseKey(hKey); hKey = NULL; } else { WCHAR wszUninstallPath[MAX_PATH]; lRes = RegOpenKeyExW(HKEY_LOCAL_MACHINE, UNINSTALL_KEY_PATH, 0, KEY_ALL_ACCESS|GetWow64Flag(), &hKey); if (lRes != ERROR_SUCCESS) { vPrintError(IDS_CANT_OPEN_REG_KEY, UNINSTALL_KEY_PATH); bRet = FALSE; goto out; } // // create sdb path name // StringCchCopyW(wszUninstallPath, ARRAYSIZE(wszUninstallPath), wszSdbInstallName); StringCchCatW(wszUninstallPath, ARRAYSIZE(wszUninstallPath), L".sdb"); lRes = LocalRegDeleteKeyW(hKey, wszUninstallPath); if (lRes != ERROR_SUCCESS) { vPrintError(IDS_CANT_DELETE_REG_KEY, wszSdbInstallName, UNINSTALL_KEY_PATH); } RegCloseKey(hKey); hKey = NULL; } // // register or unregister the DB // if (eMode == MODE_INSTALL) { if (!SdbRegisterDatabaseEx(wszSdbInstallPath, SDB_DATABASE_SHIM, &TimeStamp.QuadPart)) { vPrintError(IDS_CANT_REGISTER_DB, wszFriendlyName); bRet = FALSE; goto out; } } else { if (!SdbUnregisterDatabase(&guidDB)) { vPrintError(IDS_CANT_UNREGISTER_DB, wszFriendlyName); } } if (eMode == MODE_INSTALL) { vPrintMessage(IDS_INSTALL_COMPLETE, wszFriendlyName); } else { vPrintMessage(IDS_UNINSTALL_COMPLETE, wszFriendlyName); } out: // // always silently delete the file on uninstall, whether we failed to remove the // registry entries or not. // if (eMode != MODE_INSTALL) { // // need to make sure the pdb is closed before deleting it. // if (pdb) { SdbCloseDatabase(pdb); pdb = NULL; } // // ensure that we don't fail because of read-only files // SetFileAttributesW(wszSdbInstallPath, FILE_ATTRIBUTE_NORMAL); DeleteFileW(wszSdbInstallPath); } quickOut: // // these cleanup steps are not strictly necessary, as they'll be cleaned up // on exit anyway. But what the heck. // if (pdb) { SdbCloseDatabase(pdb); pdb = NULL; } if (hKey) { RegCloseKey(hKey); hKey = NULL; } return bRet; } extern "C" int APIENTRY wWinMain( HINSTANCE hInstance, HINSTANCE hPrevInstance, LPWSTR lpCmdLine, int nCmdShow ) { int i; int nReturn = 0; WCHAR wszSdbName[MAX_PATH]; WCHAR wszSdbPath[MAX_PATH]; WCHAR wszSdbInstallPath[MAX_PATH]; WCHAR wszOldSdbPath[MAX_PATH]; TAGID tiDBName = TAGID_NULL; WCHAR* pszDBName = NULL; WCHAR wszFriendlyName[256]; WCHAR wszGuid[100]; OSVERSIONINFO osvi; LPWSTR szCommandLine; LPWSTR* argv; int argc; INSTALL_MODE eMode; HRESULT hr; DWORD dwLen; g_hInst = hInstance; // // check to make sure the user is an administrator // if (!bCanRun()) { vPrintError(IDS_NEED_INSTALL_PERMISSION); return 1; } // // init custom directory // g_wszCustom[0] = 0; SdbGetWindowsDirectory(g_wszCustom, ARRAYSIZE(g_wszCustom)); StringCchCatW(g_wszCustom, ARRAYSIZE(g_wszCustom), L"AppPatch\\Custom\\"); CreateDirectoryW(g_wszCustom, NULL); #if defined(_WIN64) StringCchCatW(g_wszCustom, ARRAYSIZE(g_wszCustom), L"IA64\\"); CreateDirectoryW(g_wszCustom, NULL); #endif // _WIN64 RtlZeroMemory(&osvi, sizeof(OSVERSIONINFO)); osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFO); GetVersionEx(&osvi); if (osvi.dwMajorVersion == 5 && osvi.dwMinorVersion == 0) { g_bWin2K = TRUE; } // // Note that this memory isn't freed because it will automatically // be freed on exit anyway, and there are a lot of exit cases from this application // szCommandLine = GetCommandLineW(); argv = CommandLineToArgvW(szCommandLine, &argc); if (!argv) { vPrintError(IDS_CANT_GET_ARGS); return 1; } if (argc < 2) { vPrintHelp(argv[0]); return 0; } g_bQuiet = FALSE; eMode = MODE_INSTALL; wszSdbName[0] = 0; wszGuid[0] = 0; wszFriendlyName[0] = 0; for (i = 1; i < argc; ++i) { if (argv[i][0] == L'-' || argv[i][0] == L'/') { switch (tolower(argv[i][1])) { case L'?': vPrintHelp(argv[0]); return 0; break; case L'c': // // convert entries to new format // eMode = MODE_CONVERT_FORMAT_NEW; break; case L'g': i++; if (i >= argc) { vPrintError(IDS_NEED_ARG, argv[i-1]); vPrintHelp(argv[0]); return 1; } eMode = MODE_UNINSTALL; hr = StringCchCopyW(wszGuid, ARRAYSIZE(wszGuid), argv[i]); if (FAILED(hr)) { vPrintError(IDS_ARG_TOO_LONG); vPrintHelp(argv[0]); return 1; } break; case L'n': i++; if (i >= argc) { vPrintError(IDS_NEED_ARG, argv[i-1]); vPrintHelp(argv[0]); return 1; } eMode = MODE_UNINSTALL; hr = StringCchCopyW(wszFriendlyName, ARRAYSIZE(wszFriendlyName), argv[i]); if (FAILED(hr)) { vPrintError(IDS_ARG_TOO_LONG); vPrintHelp(argv[0]); return 1; } break; case L'p': g_bAllowPatches = TRUE; break; case L'r': // // revert to old format // eMode = MODE_CONVERT_FORMAT_OLD; break; case L'q': g_bQuiet = TRUE; break; case L'u': eMode = MODE_UNINSTALL; break; default: vPrintError(IDS_INVALID_SWITCH, argv[i]); vPrintHelp(argv[0]); return 1; } } else { if (wszSdbName[0]) { vPrintError(IDS_TOO_MANY_ARGS); vPrintHelp(argv[0]); return 1; } hr = StringCchCopyW(wszSdbName, ARRAYSIZE(wszSdbName), argv[i]); if (FAILED(hr)) { vPrintError(IDS_ARG_TOO_LONG); vPrintHelp(argv[0]); return 1; } } } // // check if we are running in a special 'setup' mode (converting or reverting the entries) // if (eMode == MODE_CONVERT_FORMAT_NEW || eMode == MODE_CONVERT_FORMAT_OLD) { OpenLogFile(); if (!ConvertFormat(eMode == MODE_CONVERT_FORMAT_NEW)) { nReturn = 1; } CloseLogFile(); return nReturn; } if (eMode == MODE_INSTALL && !wszSdbName[0]) { vPrintError(IDS_MUST_SPECIFY_SDB); vPrintHelp(argv[0]); return 1; } if (eMode == MODE_UNINSTALL && !wszSdbName[0] && !wszGuid[0] && !wszFriendlyName[0]) { vPrintError(IDS_MUST_SPECIFY_SDB); vPrintHelp(argv[0]); return 1; } if (wszSdbName[0]) { if (wszSdbName[1] == L':' || wszSdbName[1] == L'\\') { // // this is a full path name, so just copy it // hr = StringCchCopyW(wszSdbPath, ARRAYSIZE(wszSdbPath), wszSdbName); if (FAILED(hr)) { vPrintError(IDS_ARG_TOO_LONG); vPrintHelp(argv[0]); return 1; } } else { DWORD dwRet; // // this is a relative path name, so get the full one // if (!_wfullpath(wszSdbPath, wszSdbName, ARRAYSIZE(wszSdbPath))) { vPrintError(IDS_CANT_GET_FULL_PATH); return 1; } } } // // First, get the real file name from other params, if necessary // if (eMode == MODE_UNINSTALL) { if (wszGuid[0]) { DWORD dwLen = wcslen(wszGuid); if (dwLen != 38 || wszGuid[0] != L'{' || wszGuid[dwLen - 1] != L'}' || wszGuid[9] != L'-' || wszGuid[14] != L'-' || wszGuid[19] != L'-' || wszGuid[24] != L'-') { vPrintError(IDS_GUID_BAD_FORMAT); return 1; } StringCchCopyW(wszSdbName, ARRAYSIZE(wszSdbName), wszGuid); StringCchCatW(wszSdbName, ARRAYSIZE(wszSdbName), L".sdb"); StringCchCopyW(wszSdbPath, ARRAYSIZE(wszSdbPath), g_wszCustom); StringCchCatW(wszSdbPath, ARRAYSIZE(wszSdbPath), wszSdbName); } else if (wszFriendlyName[0]) { if (!bFriendlyNameToFile(wszFriendlyName, wszSdbName, ARRAYSIZE(wszSdbName), wszSdbPath, ARRAYSIZE(wszSdbPath))) { vPrintError(IDS_NO_FRIENDLY_NAME, wszFriendlyName); return 1; } } else { if (!bIsAlreadyInstalled(wszSdbPath)) { WCHAR wszSdbPathTemp[MAX_PATH]; // // they're not giving us an installed file, so get the GUID and convert to a file // if (!bFindInstallName(wszSdbPath, wszSdbPathTemp, ARRAYSIZE(wszSdbPathTemp))) { return 1; } StringCchCopyW(wszSdbName, ARRAYSIZE(wszSdbName), wszSdbPathTemp); // name and path are the same StringCchCopyW(wszSdbPath, ARRAYSIZE(wszSdbPath), wszSdbPathTemp); } } } if (eMode == MODE_INSTALL && GetFileAttributesW(wszSdbPath) != -1 && bIsAlreadyInstalled(wszSdbPath)) { // // they asked us to install, it's installed, so we're done // vPrintMessage(IDS_ALREADY_INSTALLED, wszSdbPath); goto quickOut; } if (eMode == MODE_UNINSTALL && GetFileAttributesW(wszSdbPath) == -1) { // // they asked us to uninstall, it's not installed, so we're done // vPrintMessage(IDS_NOT_INSTALLED, wszSdbName); goto quickOut; } if (eMode == MODE_INSTALL && DatabaseContainsPatch(wszSdbPath) && !g_bAllowPatches) { // // we can't install because the SDB contains a patch and the user hasn't authorized it. // vPrintMessage(IDS_NO_PATCHES_ALLOWED); goto quickOut; } if (eMode == MODE_INSTALL && bOldSdbInstalled(wszSdbPath, wszOldSdbPath, ARRAYSIZE(wszOldSdbPath))) { // // we should ask if we're going to uninstall the old one, // unless we're in quiet mode. // int nRet; WCHAR wszCaption[1024]; WCHAR wszText[1024]; if (g_bQuiet) { nRet = IDYES; } else { if (!LoadStringW(g_hInst, IDS_APP_TITLE, wszCaption, ARRAYSIZE(wszCaption))) { return 1; } if (!LoadStringW(g_hInst, IDS_FOUND_SAME_ID, wszText, ARRAYSIZE(wszText))) { return 1; } nRet = MessageBoxW(NULL, wszText, wszCaption, MB_YESNO | MB_ICONQUESTION); } if (nRet == IDNO) { return 0; } else if (nRet == IDYES) { if (!bHandleInstall(wszOldSdbPath, MODE_UNINSTALL, wszSdbInstallPath, ARRAYSIZE(wszSdbInstallPath))) { vPrintError(IDS_FAILED_UNINSTALL); return 1; } } } wszSdbInstallPath[0] = 0; if (!bHandleInstall(wszSdbPath, eMode, wszSdbInstallPath, ARRAYSIZE(wszSdbInstallPath))) { if (eMode == MODE_INSTALL) { // // we need to clean up; the install failed. // g_bQuiet = TRUE; bHandleInstall(wszSdbPath, MODE_CLEANUP, wszSdbInstallPath, ARRAYSIZE(wszSdbInstallPath)); } nReturn = 1; } // // no matter what happens, flush the cache // vFlushCache(); quickOut: return nReturn; }