sdbinst.c File Reference

#include <tchar.h>
#include <windef.h>
#include <winbase.h>
#include <winreg.h>
#include <strsafe.h>
#include <objbase.h>
#include <apphelp.h>
#include <shlwapi.h>

Include dependency graph for sdbinst.c:

Go to the source code of this file.

Macros
#define	APPCOMPAT_CUSTOM_REG_PATH   L"Software\\Microsoft\\Windows NT\\CurrentVersion\\AppCompatFlags\\Custom"
#define	APPCOMPAT_LAYERS_REG_PATH   L"Software\\Microsoft\\Windows NT\\CurrentVersion\\AppCompatFlags\\Custom\\Layers"
#define	APPCOMPAT_INSTALLEDSDB_REG_PATH   L"Software\\Microsoft\\Windows NT\\CurrentVersion\\AppCompatFlags\\InstalledSDB"
#define	UNINSTALL_REG_PATH   L"Software\\Microsoft\\Windows\\CurrentVersion\\Uninstall"
#define	DBPATH_KEY_NAME   L"DatabasePath"
#define	SDB_EXT   L".sdb"
#define	GUID_STR_LENGTH   38
#define	GUID_SBD_NAME_LENGTH   (GUID_STR_LENGTH + ARRAYSIZE(SDB_EXT))

Functions
BOOL	SdbUninstallByGuid (_In_ LPWSTR guidSdbStr)
HRESULT	RegisterSdbEntry (_In_ PWCHAR sdbEntryName, _In_ LPCWSTR dbGuid, _In_ ULONGLONG time, _In_ BOOL isInstall, _In_ BOOL isExe)
HRESULT	AddUninstallKey (_In_ LPCWSTR dbName, _In_ LPCWSTR sdbInstalledPath, _In_ LPCWSTR guidDbStr)
BOOL	GetSdbGuid (_In_ PDB pdb, _In_ TAGID tagDb, _Out_ GUID *guid)
HRESULT	ProcessLayers (_In_ PDB pdb, _In_ TAGID tagDb, _In_opt_ LPCWSTR guidDbStr, _In_opt_ ULONGLONG time, _In_ BOOL isInstall)
HRESULT	ProcessExe (_In_ PDB pdb, _In_ TAGID tagDb, _In_opt_ LPCWSTR guidDbStr, _In_opt_ ULONGLONG time, _In_ BOOL isInstall)
HRESULT	CopySdbToAppPatch (_In_ LPCWSTR sourceSdbPath, _In_ LPCWSTR destSdbPath)
HRESULT	BuildPathToSdb (_In_ PWCHAR buffer, _In_ SIZE_T bufLen, _In_ LPCWSTR guidDbStr)
BOOL	SdbInstall (_In_ LPCWSTR sdbPath)
HRESULT	DeleteUninstallKey (_In_ LPCWSTR keyName)
BOOL	SdbUninstall (_In_ LPWSTR sdbPath)
BOOL	ValidateGuidString (_In_ PWCHAR guidStr)
BOOL	SdbUninstallByName (_In_ LPWSTR nameSdbStr)
void	ShowHelp ()
int	_tmain (int argc, LPWSTR argv[])

Macro Definition Documentation

APPCOMPAT_CUSTOM_REG_PATH

#define APPCOMPAT_CUSTOM_REG_PATH   L"Software\\Microsoft\\Windows NT\\CurrentVersion\\AppCompatFlags\\Custom"

Definition at line 18 of file sdbinst.c.

APPCOMPAT_INSTALLEDSDB_REG_PATH

#define APPCOMPAT_INSTALLEDSDB_REG_PATH   L"Software\\Microsoft\\Windows NT\\CurrentVersion\\AppCompatFlags\\InstalledSDB"

Definition at line 20 of file sdbinst.c.

APPCOMPAT_LAYERS_REG_PATH

#define APPCOMPAT_LAYERS_REG_PATH   L"Software\\Microsoft\\Windows NT\\CurrentVersion\\AppCompatFlags\\Custom\\Layers"

Definition at line 19 of file sdbinst.c.

DBPATH_KEY_NAME

#define DBPATH_KEY_NAME   L"DatabasePath"

Definition at line 22 of file sdbinst.c.

GUID_SBD_NAME_LENGTH

#define GUID_SBD_NAME_LENGTH   (GUID_STR_LENGTH + ARRAYSIZE(SDB_EXT))

Definition at line 25 of file sdbinst.c.

GUID_STR_LENGTH

#define GUID_STR_LENGTH   38

Definition at line 24 of file sdbinst.c.

SDB_EXT

#define SDB_EXT   L".sdb"

Definition at line 23 of file sdbinst.c.

UNINSTALL_REG_PATH

#define UNINSTALL_REG_PATH   L"Software\\Microsoft\\Windows\\CurrentVersion\\Uninstall"

Definition at line 21 of file sdbinst.c.

Function Documentation

_tmain()

int _tmain	(	int	argc,
		LPWSTR	argv[]
	)

Definition at line 819 of file sdbinst.c.

{
    LPWSTR sdbPath = NULL;
    BOOL isInstall = TRUE;
    BOOL isUninstByGuid = FALSE;
    BOOL isUninstByName = FALSE;
    BOOL success = FALSE;
    LPWSTR guidSdbStr = NULL;
    LPWSTR nameSdbStr = NULL;
 
    if (argc < 2)
    {
        ShowHelp();
    }
 
    for (int i = 1; i < argc; ++i)
    {
        if (argv[i][0] != L'-')
        {
            sdbPath = argv[i];
            break;
        }
 
        switch (argv[i][1])
        {
            case L'?':
                ShowHelp();
                return ERROR_SUCCESS;
            break;
 
            case L'u':
                isInstall = FALSE;
            break;
 
            case L'g':
                ++i;
 
                if (i >= argc)
                {
                    return ERROR_INVALID_PARAMETER;
                }
 
                guidSdbStr = argv[i];
                wprintf(L"guidSdbStr %ls\n", guidSdbStr);
 
                isUninstByGuid = TRUE;
                isInstall = FALSE;
            break;
 
            case L'n':
                ++i;
 
                if (i >= argc)
                {
                    return ERROR_INVALID_PARAMETER;
                }
 
                nameSdbStr = argv[i];
                wprintf(L"nameSdbStr %ls\n", nameSdbStr);
 
                isUninstByName = TRUE;
                isInstall = FALSE;
            break;
        }
    }
 
    if (isInstall)
    {
        wprintf(L"install\n");
        success = SdbInstall(sdbPath);
    }
    else if (isUninstByGuid)
    {
        wprintf(L"uninstall by GUID\n");
        success = SdbUninstallByGuid(guidSdbStr);
    }
    else if (isUninstByName)
    {
        wprintf(L"uninstall by name\n");
        success = SdbUninstallByName(nameSdbStr);
    }
    else
    {
        wprintf(L"uninstall\n");
        success = SdbUninstall(sdbPath);
    }
 
    if (!success)
    {
        wprintf(isInstall ? L"Sdb install failed\n" : L"Sdb uninstall failed\n");
        return -1;
    }
 
    return ERROR_SUCCESS;
}

AddUninstallKey()

HRESULT AddUninstallKey	(	_In_ LPCWSTR	dbName,
		_In_ LPCWSTR	sdbInstalledPath,
		_In_ LPCWSTR	guidDbStr
	)

Definition at line 104 of file sdbinst.c.

{
    WCHAR sdbinstPath[MAX_PATH];
    WCHAR regName[MAX_PATH];
    WCHAR uninstString[MAX_PATH];
    HKEY hKey = NULL;
    HRESULT hres;
 
    UINT count = GetSystemWindowsDirectory(sdbinstPath, MAX_PATH);
    if (sdbinstPath[count - 1] != L'\\')
    {
        hres = StringCchCatW(sdbinstPath, MAX_PATH, L"\\");
        if (FAILED(hres))
        {
            wprintf(L"StringCchCatW error: 0x%08X", hres);
            goto end;
        }
    }
 
    // Full path to sdbinst.exe
    hres = StringCchCatW(sdbinstPath, MAX_PATH, L"System32\\sdbinst.exe");
    if (FAILED(hres))
    {
        wprintf(L"StringCchCatW error: 0x%08X", hres);
        goto end;
    }
 
    // Sdb GUID registry key
    hres = StringCchPrintfW(regName, MAX_PATH, L"%ls\\%ls", UNINSTALL_REG_PATH, guidDbStr);
    if (FAILED(hres))
    {
        wprintf(L"StringCchPrintfW error: 0x%08X", hres);
        goto end;
    }
 
    // Create main key
    LSTATUS status = RegCreateKeyExW(HKEY_LOCAL_MACHINE,
                                    regName,
                                    0,
                                    NULL,
                                    REG_OPTION_NON_VOLATILE,
                                    KEY_ALL_ACCESS,
                                    NULL,
                                    &hKey,
                                    NULL);
 
    if (status != ERROR_SUCCESS)
    {
        wprintf(L"RegKeyCreateEx error: 0x%08X", status);
        hres = HRESULT_FROM_WIN32(status);
        goto end;
    }
 
    // Set Display name
    DWORD length = wcslen(dbName) * sizeof(WCHAR);
    status = RegSetValueExW(hKey,
                           L"DisplayName",
                           0,
                           REG_SZ,
                           (PBYTE)dbName,
                           length + sizeof(WCHAR));
    if (status != ERROR_SUCCESS)
    {
        wprintf(L"RegSetValueExW error: 0x%08X", status);
        hres = HRESULT_FROM_WIN32(status);
        goto end;
    }
 
    // Uninstall full string
    hres = StringCchPrintfW(uninstString, MAX_PATH, L"%ls -u \"%ls\"", sdbinstPath, sdbInstalledPath);
    if (FAILED(hres))
    {
        wprintf(L"StringCchPrintfW error: 0x%08X", hres);
        goto end;
    }
 
    // Set uninstall string
    length = wcslen(uninstString) * sizeof(WCHAR);
    status = RegSetValueExW(hKey,
                           L"UninstallString",
                           0,
                           REG_SZ,
                           (PBYTE)uninstString,
                           length + sizeof(WCHAR));
    if (status != ERROR_SUCCESS)
    {
        wprintf(L"RegSetValueExW error: 0x%08X", status);
        hres = HRESULT_FROM_WIN32(status);
        goto end;
    }
 
end:
    if (hKey)
    {
        RegCloseKey(hKey);
    }
 
    return hres;
}

Referenced by SdbInstall().

BuildPathToSdb()

HRESULT BuildPathToSdb	(	_In_ PWCHAR	buffer,
		_In_ SIZE_T	bufLen,
		_In_ LPCWSTR	guidDbStr
	)

Definition at line 373 of file sdbinst.c.

{
    ZeroMemory(buffer, bufLen * sizeof(WCHAR));
 
    // Can't use here SdbGetAppPatchDir, because Windows XP haven't this function
    UINT count = GetSystemWindowsDirectory(buffer, bufLen);
    if (buffer[count - 1] != L'\\')
    {
        buffer[count] = L'\\';
    }
 
    HRESULT res = StringCchCatW(buffer, bufLen, L"AppPatch\\Custom\\");
    if (FAILED(res))
    {
        goto end;
    }
 
    res = StringCchCatW(buffer, bufLen, guidDbStr);
 
end:
    return res;
}

Referenced by SdbInstall().

CopySdbToAppPatch()

HRESULT CopySdbToAppPatch	(	_In_ LPCWSTR	sourceSdbPath,
		_In_ LPCWSTR	destSdbPath
	)

Definition at line 316 of file sdbinst.c.

{
    DWORD error = ERROR_SUCCESS;
    PWCHAR pTmpSysdir = NULL;
    SIZE_T destLen = wcslen(destSdbPath);
    PWCHAR sysdirPath = (PWCHAR)HeapAlloc(GetProcessHeap(), 0, destLen * sizeof(WCHAR));
 
    if (sysdirPath == NULL)
    {
        error = ERROR_NOT_ENOUGH_MEMORY;
        goto end;
    }
 
    // Get parent folder fo sdb file
    CopyMemory(sysdirPath, destSdbPath, destLen * sizeof(WCHAR));
    pTmpSysdir = StrRChrW(sysdirPath, sysdirPath + destLen, L'\\');
    if (pTmpSysdir == NULL)
    {
        wprintf(L"Can't find directory separator\n");
        goto end;
    }
    else
    {
        *pTmpSysdir = UNICODE_NULL;
    }
 
    // Create directory if need
    if (!CreateDirectory(sysdirPath, NULL))
    {
        error = GetLastError();
        if (error != ERROR_ALREADY_EXISTS)
        {
            wprintf(L"Can't create folder %ls\n Error: 0x%08X\n", sysdirPath, error);
            goto end;
        }
        error = ERROR_SUCCESS;
    }
 
    // Copy file
    if (!CopyFile(sourceSdbPath, destSdbPath, TRUE))
    {
        error = GetLastError();
        wprintf(L"Can't copy sdb file");
    }
 
end:
    if (sysdirPath)
    {
        HeapFree(GetProcessHeap(), 0, sysdirPath);
    }
 
    return HRESULT_FROM_WIN32(error);
}

Referenced by SdbInstall().

DeleteUninstallKey()

HRESULT DeleteUninstallKey

(

_In_ LPCWSTR

keyName

)

Definition at line 513 of file sdbinst.c.

{
    HKEY hKey = NULL;
    HRESULT hres = HRESULT_FROM_WIN32(ERROR_SUCCESS);
 
    LSTATUS status = RegCreateKeyExW(HKEY_LOCAL_MACHINE,
                                    UNINSTALL_REG_PATH,
                                    0,
                                    NULL,
                                    REG_OPTION_NON_VOLATILE,
                                    KEY_ALL_ACCESS,
                                    NULL,
                                    &hKey,
                                    NULL);
 
    if (status != ERROR_SUCCESS)
    {
        hres = HRESULT_FROM_WIN32(status);
        goto end;
    }
 
    status = RegDeleteKeyW(hKey, keyName);
    if (status != ERROR_SUCCESS)
    {
        hres = HRESULT_FROM_WIN32(status);
    }
 
end:
    if (hKey)
    {
        RegCloseKey(hKey);
    }
 
    return hres;
}

Referenced by SdbUninstall().

GetSdbGuid()

BOOL GetSdbGuid	(	_In_ PDB	pdb,
		_In_ TAGID	tagDb,
		_Out_ GUID *	guid
	)

Definition at line 211 of file sdbinst.c.

{
    TAGID tagDbId;
 
    tagDbId = SdbFindFirstTag(pdb, tagDb, TAG_DATABASE_ID);
    if (!tagDbId)
    {
        wprintf(L"Can't find database id tag");
        return FALSE;
    }
 
    if (!SdbReadBinaryTag(pdb, tagDbId, (PBYTE)guid, sizeof(GUID)))
    {
        wprintf(L"Can't read database id");
        return FALSE;
    }
 
    return TRUE;
}

Referenced by SdbInstall(), and SdbUninstall().

ProcessExe()

HRESULT ProcessExe	(	_In_ PDB	pdb,
		_In_ TAGID	tagDb,
		_In_opt_ LPCWSTR	guidDbStr,
		_In_opt_ ULONGLONG	time,
		_In_ BOOL	isInstall
	)

Definition at line 275 of file sdbinst.c.

{
    HRESULT res = ERROR_SUCCESS;
    TAGID tagExeName;
    TAGID tagExePrev = 0;
 
    TAGID tagExe = SdbFindFirstTag(pdb, tagDb, TAG_EXE);
 
    // Add all exe to registry (AppCompatFlags)
    while (tagExe != 0 && (tagExe != tagExePrev))
    {
        tagExeName = SdbFindFirstTag(pdb, tagExe, TAG_NAME);
        if (!tagExeName)
        {
            wprintf(L"Can't find exe tag\n");
            res = ERROR_NOT_FOUND;
            break;
        }
 
        LPWSTR name = SdbGetStringTagPtr(pdb, tagExeName);
 
        res = RegisterSdbEntry(name, guidDbStr, time, isInstall, TRUE);
        if (FAILED(res))
        {
            wprintf(L"Can't register exe 0x%08X\n", res);
            break;
        }
 
        tagExePrev = tagExe;
        tagExe = SdbFindNextTag(pdb, tagDb, tagExe);
    }
 
    return res;
}

Referenced by SdbInstall(), and SdbUninstall().

ProcessLayers()

HRESULT ProcessLayers	(	_In_ PDB	pdb,
		_In_ TAGID	tagDb,
		_In_opt_ LPCWSTR	guidDbStr,
		_In_opt_ ULONGLONG	time,
		_In_ BOOL	isInstall
	)

Definition at line 235 of file sdbinst.c.

{
    HRESULT res = ERROR_SUCCESS;
    TAGID tagLayerName;
    TAGID prevTagLayer = 0;
 
    TAGID tagLayer = SdbFindFirstTag(pdb, tagDb, TAG_LAYER);
 
    // Add all layers to registry (AppCompatFlags)
    while (tagLayer && (tagLayer != prevTagLayer))
    {
        tagLayerName = SdbFindFirstTag(pdb, tagLayer, TAG_NAME);
        if (!tagLayerName)
        {
            res = ERROR_NOT_FOUND;
            break;
        }
 
        LPWSTR name = SdbGetStringTagPtr(pdb, tagLayerName);
 
        res = RegisterSdbEntry(name, guidDbStr, time, isInstall, FALSE);
        if (FAILED(res))
        {
            wprintf(L"Can't register layer\n");
            break;
        }
 
        prevTagLayer = tagLayer;
        tagLayer = SdbFindNextTag(pdb, tagDb, tagLayer);
    }
 
    return res;
}

Referenced by SdbInstall(), and SdbUninstall().

RegisterSdbEntry()

HRESULT RegisterSdbEntry	(	_In_ PWCHAR	sdbEntryName,
		_In_ LPCWSTR	dbGuid,
		_In_ ULONGLONG	time,
		_In_ BOOL	isInstall,
		_In_ BOOL	isExe
	)

Definition at line 30 of file sdbinst.c.

{
    WCHAR regName[MAX_PATH];
    HKEY hKey = NULL;
    LSTATUS status;
    HRESULT hres;
 
    hres = StringCchPrintfW(regName, MAX_PATH, L"%ls\\%ls",
                            isExe ? APPCOMPAT_CUSTOM_REG_PATH : APPCOMPAT_LAYERS_REG_PATH,
                            sdbEntryName);
    if (FAILED(hres))
    {
        wprintf(L"StringCchPrintfW error: 0x%08X\n", hres);
        goto end;
    }
 
    // Remove key
    if (!isInstall)
    {
        status = RegDeleteKeyW(HKEY_LOCAL_MACHINE, regName);
        if (status == ERROR_FILE_NOT_FOUND)
        {
            status = ERROR_SUCCESS;
        }
 
        return HRESULT_FROM_WIN32(status);
    }
 
    // Create main key
    status = RegCreateKeyExW(HKEY_LOCAL_MACHINE,
                            regName,
                            0,
                            NULL,
                            REG_OPTION_NON_VOLATILE,
                            KEY_ALL_ACCESS,
                            NULL,
                            &hKey,
                            NULL);
    if (status != ERROR_SUCCESS)
    {
        wprintf(L"RegKeyCreateEx error: 0x%08X", status);
        hres = HRESULT_FROM_WIN32(status);
        goto end;
    }
 
    // Set installed time
    status = RegSetValueExW(hKey,
                           dbGuid,
                           0,
                           REG_QWORD,
                           (PBYTE)&time,
                           sizeof(time));
    if (status != ERROR_SUCCESS)
    {
        wprintf(L"RegSetValueExW error: 0x%08X", status);
        hres = HRESULT_FROM_WIN32(status);
        goto end;
    }
 
end:
    if (hKey)
    {
        RegCloseKey(hKey);
    }
 
    return hres;
}

Referenced by ProcessExe(), and ProcessLayers().

SdbInstall()

BOOL SdbInstall

(

_In_ LPCWSTR

sdbPath

)

Definition at line 400 of file sdbinst.c.

{
    BOOL res = FALSE;
    PDB pdb = NULL;
    TAGID tagDb;
    TAGID tagDbName;
    GUID dbGuid = {0};
    FILETIME systemTime = {0};
    ULARGE_INTEGER currentTime = {0};
    WCHAR sysdirPatchPath[MAX_PATH];
    WCHAR guidDbStr[GUID_SBD_NAME_LENGTH];
 
    GetSystemTimeAsFileTime(&systemTime);
    currentTime.LowPart  = systemTime.dwLowDateTime;
    currentTime.HighPart = systemTime.dwHighDateTime;
 
    // Open database
    pdb = SdbOpenDatabase(sdbPath, DOS_PATH);
    if (pdb == NULL)
    {
        wprintf(L"Can't open database %ls\n", sdbPath);
        goto end;
    }
 
    tagDb = SdbFindFirstTag(pdb, TAGID_ROOT, TAG_DATABASE);
    if (!tagDb)
    {
        wprintf(L"Can't find database tag\n");
        goto end;
    }
 
    // Get database GUID
    if (!GetSdbGuid(pdb, tagDb, &dbGuid))
    {
        wprintf(L"GetSdbGuid error\n");
        goto end;
    }
 
    StringFromGUID2(&dbGuid, guidDbStr, GUID_SBD_NAME_LENGTH);
    HRESULT hres = StringCchCatW(guidDbStr, GUID_SBD_NAME_LENGTH, SDB_EXT);
    if (FAILED(hres))
    {
        wprintf(L"StringCchCatW error 0x%08X\n", hres);
        goto end;
    }
 
    wprintf(L"Database guid %ls\n", guidDbStr);
 
    tagDbName = SdbFindFirstTag(pdb, tagDb, TAG_NAME);
    if (!tagDbName)
    {
        wprintf(L"Can't get tag name\n");
        goto end;
    }
 
    LPWSTR dbName = SdbGetStringTagPtr(pdb, tagDbName);
    wprintf(L"Database name %ls\n", dbName);
 
    // Process exe tags
    hres = ProcessExe(pdb, tagDb, guidDbStr, currentTime.QuadPart, TRUE);
    if (FAILED(hres))
    {
        wprintf(L"Process exe failed. Status: 0x%08X", res);
        goto end;
    }
 
    // Proess layer tags
    hres = ProcessLayers(pdb, tagDb, guidDbStr, currentTime.QuadPart, TRUE);
    if (FAILED(hres))
    {
        wprintf(L"Process layers failed. Status: 0x%08X", res);
        goto end;
    }
 
    // Create full path to sdb in system folder
    hres = BuildPathToSdb(sysdirPatchPath, ARRAYSIZE(sysdirPatchPath), guidDbStr);
    if (FAILED(hres))
    {
        wprintf(L"Build path error\n");
        goto end;
    }
 
    wprintf(L"file path %ls\n", sysdirPatchPath);
 
    res = CopySdbToAppPatch(sdbPath, sysdirPatchPath);
    if (FAILED(res))
    {
        wprintf(L"Copy sdb error. Status: 0x%08X\n", res);
        goto end;
    }
 
    AddUninstallKey(dbName, sysdirPatchPath, guidDbStr);
 
    // Registration
    if (!SdbRegisterDatabaseEx(sysdirPatchPath, SDB_DATABASE_SHIM, &currentTime.QuadPart))
    {
        wprintf(L"SdbRegisterDatabaseEx failed");
        goto end;
    }
 
    res = TRUE;
 
end:
    if (pdb)
    {
        SdbCloseDatabase(pdb);
    }
 
    return res;
}

Referenced by _tmain().

SdbUninstall()

BOOL SdbUninstall

(

_In_ LPWSTR

sdbPath

)

Definition at line 551 of file sdbinst.c.

{
    BOOL res = FALSE;
    PWCHAR sdbName = NULL;
    PDB pdb;
    TAGID tagDb;
    GUID dbGuid = {0};
    WCHAR guidDbStr[GUID_SBD_NAME_LENGTH];
 
    SIZE_T sdbPathLen = wcslen(sdbPath);
    sdbName = sdbPath + sdbPathLen;
 
    wprintf(L"uninstall path %ls\n", sdbPath);
    sdbName = StrRChrW(sdbPath, sdbPath + sdbPathLen, L'\\');
    if (sdbName == NULL)
    {
        sdbName = sdbPath;
    }
    else
    {
        sdbName++;
    }
 
    wprintf(L"uninstall name %ls\n", sdbName);
 
    // open sdb
    pdb = SdbOpenDatabase(sdbPath, DOS_PATH);
    if (pdb == NULL)
    {
        wprintf(L"Can't open database %ls\n", sdbPath);
        return FALSE;
    }
 
    tagDb = SdbFindFirstTag(pdb, TAGID_ROOT, TAG_DATABASE);
    if (!tagDb)
    {
        wprintf(L"Can't find database tag\n");
        goto end;
    }
 
    if (!GetSdbGuid(pdb, tagDb, &dbGuid))
    {
        wprintf(L"GetSdbGuid error\n");
        goto end;
    }
 
    // Database name must be GUID string
    if (wcslen(sdbName) + 1 != GUID_SBD_NAME_LENGTH)
    {
        StringFromGUID2(&dbGuid, guidDbStr, GUID_SBD_NAME_LENGTH);
        SdbCloseDatabase(pdb);
        return SdbUninstallByGuid(guidDbStr);
    }
 
    //remove regkey in appatch/custom
    HRESULT hres = ProcessExe(pdb, tagDb, NULL, 0, FALSE);
    if (FAILED(hres))
    {
        wprintf(L"Process exe fail\n");
        goto end;
    }
 
    hres = ProcessLayers(pdb, tagDb, NULL, 0, FALSE);
    if (FAILED(hres))
    {
        wprintf(L"Process layers fail\n");
        goto end;
    }
 
    SdbCloseDatabase(pdb);
    pdb = NULL;
 
    hres = DeleteUninstallKey(sdbName);
    if (FAILED(hres))
    {
        wprintf(L"Remove uninstall key fail\n");
    }
 
    if (!SdbUnregisterDatabase(&dbGuid))
    {
        wprintf(L"SdbUnregisterDatabase\n");
        return FALSE;
    }
 
    SetFileAttributesW(sdbPath, FILE_ATTRIBUTE_NORMAL);
    if (!DeleteFileW(sdbPath))
    {
        wprintf(L"Remove file fail 0x%08X\n", GetLastError());
        return FALSE;
    }
 
    res = TRUE;
 
end:
    if (pdb)
        SdbCloseDatabase(pdb);
    return res;
}

Referenced by _tmain(), SdbUninstallByGuid(), and SdbUninstallByName().

SdbUninstallByGuid()

BOOL SdbUninstallByGuid

(

_In_ LPWSTR

guidSdbStr

)

Definition at line 672 of file sdbinst.c.

{
    BOOL res = FALSE;
    HKEY hKey = NULL;
    HKEY guidKey = NULL;
    LSTATUS status;
    WCHAR dbPath[MAX_PATH];
    DWORD keyValSize = sizeof(dbPath);
 
    if (!ValidateGuidString(guidSdbStr))
    {
        wprintf(L"Invalid GUID: %ls\n", guidSdbStr);
        return res;
    }
 
    status = RegOpenKeyExW(HKEY_LOCAL_MACHINE, APPCOMPAT_INSTALLEDSDB_REG_PATH, 0, KEY_READ | KEY_QUERY_VALUE, &hKey);
 
    if (status != ERROR_SUCCESS)
    {
        wprintf(L"RegOpenKeyW error: 0x%08X", status);
        goto end;
    }
 
    status = RegOpenKeyExW(hKey, guidSdbStr, 0, KEY_READ | KEY_QUERY_VALUE, &guidKey);
 
    if (status != ERROR_SUCCESS)
    {
        wprintf(L"Cant open key: 0x%08X %ls\n", status, guidSdbStr);
        goto end;
    }
 
    status = RegQueryValueExW(guidKey, DBPATH_KEY_NAME, NULL, NULL, (LPBYTE)dbPath, &keyValSize);
    if (status != ERROR_SUCCESS)
    {
        wprintf(L"RegQueryValueExW: 0x%08X\n", status);
        goto end;
    }
 
    res = SdbUninstall(dbPath);
 
end:
    if (hKey)
    {
        RegCloseKey(hKey);
    }
 
    if (guidKey)
    {
        RegCloseKey(guidKey);
    }
 
    return res;
}

Referenced by _tmain(), and SdbUninstall().

SdbUninstallByName()

BOOL SdbUninstallByName

(

_In_ LPWSTR

nameSdbStr

)

Definition at line 728 of file sdbinst.c.

{
    BOOL res = FALSE;
    LSTATUS status;
    HKEY hKey = NULL;
    HKEY subKey = NULL;
    DWORD index = 0;
    WCHAR keyName[MAX_PATH];
    DWORD keyNameLen = ARRAYSIZE(keyName);
    DWORD keyValSize;
    WCHAR dbDescript[MAX_PATH];
    WCHAR dbPath[MAX_PATH];
 
    status = RegOpenKeyExW(HKEY_LOCAL_MACHINE, APPCOMPAT_INSTALLEDSDB_REG_PATH, 0, KEY_READ | KEY_QUERY_VALUE, &hKey);
 
    if (status != ERROR_SUCCESS)
    {
        wprintf(L"RegOpenKeyW error: 0x%08X", status);
        goto end;
    }
 
    status = RegEnumKeyEx(hKey, index, keyName, &keyNameLen, NULL, NULL, NULL, NULL);
    wprintf(L"0x%08X %d %ls \n", status, keyNameLen, keyName);
 
    // Search database GUID by name
    while (status == ERROR_SUCCESS)
    {
        status = RegOpenKeyExW(hKey, keyName, 0, KEY_READ | KEY_QUERY_VALUE, &subKey);
        if (status != ERROR_SUCCESS)
        {
            break;
        }
 
        keyValSize = sizeof(dbDescript);
        status = RegQueryValueExW(subKey, L"DatabaseDescription", NULL, NULL, (LPBYTE)dbDescript, &keyValSize);
        if (status != ERROR_SUCCESS)
        {
            break;
        }
 
        wprintf(L"dbdescript: %ls \n", dbDescript);
 
        if (_wcsnicmp(dbDescript, nameSdbStr, keyNameLen) == 0)
        {
            // Take db full path
            keyValSize = sizeof(dbPath);
            status = RegQueryValueExW(subKey, DBPATH_KEY_NAME, NULL, NULL, (LPBYTE)dbPath, &keyValSize);
            if (status != ERROR_SUCCESS)
            {
                dbPath[0] = UNICODE_NULL;
                break;
            }
 
            wprintf(L"dbpath: 0x%08X %ls \n", status, dbPath);
            RegCloseKey(subKey);
            break;
        }
 
        RegCloseKey(subKey);
 
        keyName[0] = UNICODE_NULL;
 
        ++index;
        keyNameLen = ARRAYSIZE(keyName);
        status = RegEnumKeyExW(hKey, index, keyName, &keyNameLen, NULL, NULL, NULL, NULL);
    }
 
    RegCloseKey(hKey);
 
    if (dbPath[0] != UNICODE_NULL)
    {
        res = SdbUninstall(dbPath);
    }
 
end:
    return res;
}

Referenced by _tmain().

ShowHelp()

void ShowHelp

(

)

Definition at line 809 of file sdbinst.c.

{
    /* FIXME: to be localized */
    wprintf(L"Using: sdbinst [-?][-q][-u][-g][-n] foo.sdb | {guid} | \"name\" \n"
            L"-? - show help\n"
            L"-u - uninstall\n"
            L"-g - {guid} file GUID (only uninstall)\n"
            L"-n - \"name\" - file name (only uninstall)\n");
}

Referenced by _tmain(), and DECLARE_INTERFACE_().

ValidateGuidString()

BOOL ValidateGuidString

(

_In_ PWCHAR

guidStr

)

Definition at line 652 of file sdbinst.c.

{
    ULONG length = wcslen(guidStr);
 
    if (length == GUID_STR_LENGTH &&
        guidStr[0] == L'{' &&
        guidStr[GUID_STR_LENGTH - 1] == L'}' &&
        guidStr[9] == L'-' &&
        guidStr[14] == L'-' &&
        guidStr[19] == L'-' &&
        guidStr[24] == L'-')
    {
        return TRUE;
    }
 
    return FALSE;
}

Referenced by SdbUninstallByGuid().