CAppDB Class Reference

#include <appdb.h>

Collaboration diagram for CAppDB:

Public Member Functions
	CAppDB (const CStringW &path)
VOID	GetApps (CAtlList< CAppInfo * > &List, AppsCategories Type) const
CAvailableApplicationInfo *	FindAvailableByPackageName (const CStringW &name)
CAppInfo *	FindByPackageName (const CStringW &name)
VOID	UpdateAvailable ()
VOID	UpdateInstalled ()
VOID	RemoveCached ()
size_t	GetAvailableCount () const

Static Public Member Functions
static DWORD	RemoveInstalledAppFromRegistry (const CAppInfo *Info)
static CInstalledApplicationInfo *	CreateInstalledAppByRegistryKey (LPCWSTR Name)
static CInstalledApplicationInfo *	CreateInstalledAppInstance (LPCWSTR KeyName, BOOL User, REGSAM WowSam)
static HKEY	EnumInstalledRootKey (UINT Index, REGSAM &RegSam)

Private Member Functions
BOOL	EnumerateFiles ()

Static Private Member Functions
static CInstalledApplicationInfo *	EnumerateRegistry (CAtlList< CAppInfo * > *List, LPCWSTR Name)
static CInstalledApplicationInfo *	CreateInstalledAppByRegistryKey (LPCWSTR KeyName, HKEY hKeyParent, UINT KeyIndex)

Private Attributes
CPathW	m_BasePath
CAtlList< CAppInfo * >	m_Available
CAtlList< CAppInfo * >	m_Installed

Detailed Description

Definition at line 8 of file appdb.h.

Constructor & Destructor Documentation

CAppDB()

CAppDB::CAppDB

(

const CStringW &

path

)

Definition at line 32 of file appdb.cpp.

                                   : m_BasePath(path)
{
    m_BasePath.Canonicalize();
}

Member Function Documentation

CreateInstalledAppByRegistryKey() [1/2]

CInstalledApplicationInfo * CAppDB::CreateInstalledAppByRegistryKey ( LPCWSTR  KeyName, HKEY  hKeyParent, UINT  KeyIndex  )

staticprivate

Definition at line 169 of file appdb.cpp.

{
    CRegKey hSubKey;
    if (hSubKey.Open(hKeyParent, KeyName, KEY_READ) != ERROR_SUCCESS)
        return NULL;
    DWORD value, size;
 
    size = sizeof(DWORD);
    if (!RegQueryValueExW(hSubKey, L"SystemComponent", NULL, NULL, (LPBYTE)&value, &size) && value == 1)
    {
        // Ignore system components
        return NULL;
    }
 
    size = 0;
    BOOL bIsUpdate = !RegQueryValueExW(hSubKey, L"ParentKeyName", NULL, NULL, NULL, &size);
 
    AppsCategories cat = bIsUpdate ? ENUM_UPDATES : ENUM_INSTALLED_APPLICATIONS;
    CInstalledApplicationInfo *pInfo;
    pInfo = new CInstalledApplicationInfo(hSubKey.Detach(), KeyName, cat, KeyIndex);
    if (pInfo && pInfo->Valid())
    {
        return pInfo;
    }
    delete pInfo;
    return NULL;
}

Referenced by CreateInstalledAppInstance(), EnumerateRegistry(), and HandleUninstallCommand().

CreateInstalledAppByRegistryKey() [2/2]

CInstalledApplicationInfo * CAppDB::CreateInstalledAppByRegistryKey ( LPCWSTR  Name )

static

Definition at line 245 of file appdb.cpp.

{
    return EnumerateRegistry(NULL, Name);
}

CreateInstalledAppInstance()

CInstalledApplicationInfo * CAppDB::CreateInstalledAppInstance ( LPCWSTR  KeyName, BOOL  User, REGSAM  WowSam  )

static

Definition at line 251 of file appdb.cpp.

{
    HKEY hRootKey = User ? HKEY_CURRENT_USER : HKEY_LOCAL_MACHINE;
    UINT KeyIndex = User ? (0) : ((WowSam & KEY_WOW64_64KEY) ? 2 : 1);
    CRegKey hKey;
    if (hKey.Open(hRootKey, UNINSTALL_SUBKEY, KEY_READ | WowSam) == ERROR_SUCCESS)
    {
        return CreateInstalledAppByRegistryKey(KeyName, hKey, KeyIndex);
    }
    return NULL;
}

Referenced by HandleUninstallCommand().

EnumerateFiles()

BOOL CAppDB::EnumerateFiles ( )

private

Definition at line 72 of file appdb.cpp.

{
    ClearList(m_Available);
 
    CPathW AppsPath = m_BasePath;
    AppsPath += RAPPS_DATABASE_SUBDIR;
    CPathW WildcardPath = AppsPath;
    WildcardPath += L"*.txt";
 
    WIN32_FIND_DATAW FindFileData;
    HANDLE hFind = FindFirstFileW(WildcardPath, &FindFileData);
    if (hFind == INVALID_HANDLE_VALUE)
    {
        return FALSE;
    }
 
    do
    {
        CStringW szPkgName = FindFileData.cFileName;
        PathRemoveExtensionW(szPkgName.GetBuffer(MAX_PATH));
        szPkgName.ReleaseBuffer();
 
        CAppInfo *Info = FindByPackageName(szPkgName);
        ATLASSERT(Info == NULL);
        if (!Info)
        {
            CConfigParser *Parser = new CConfigParser(CPathW(AppsPath) += FindFileData.cFileName);
            int Cat;
            if (!Parser->GetInt(DB_CATEGORY, Cat))
                Cat = ENUM_INVALID;
 
            Info = new CAvailableApplicationInfo(Parser, szPkgName, static_cast<AppsCategories>(Cat), AppsPath);
            if (Info->Valid())
            {
                m_Available.AddTail(Info);
            }
            else
            {
                delete Info;
            }
        }
 
    } while (FindNextFileW(hFind, &FindFileData));
 
    FindClose(hFind);
    return TRUE;
}

Referenced by UpdateAvailable().

EnumerateRegistry()

CInstalledApplicationInfo * CAppDB::EnumerateRegistry ( CAtlList< CAppInfo * > *  List, LPCWSTR  Name  )

staticprivate

Definition at line 198 of file appdb.cpp.

{
    ATLASSERT(List || SearchOnly);
    REGSAM wowsam;
    HKEY hRootKey;
    for (UINT rki = 0; (hRootKey = EnumInstalledRootKey(rki, wowsam)); ++rki)
    {
        CRegKey hKey;
        if (hKey.Open(hRootKey, UNINSTALL_SUBKEY, KEY_READ | wowsam) != ERROR_SUCCESS)
        {
            continue;
        }
        for (DWORD Index = 0;; ++Index)
        {
            WCHAR szKeyName[MAX_PATH];
            DWORD dwSize = _countof(szKeyName);
            if (hKey.EnumKey(Index, szKeyName, &dwSize) != ERROR_SUCCESS)
            {
                break;
            }
            if (List || !StrCmpIW(SearchOnly, szKeyName))
            {
                CInstalledApplicationInfo *Info;
                Info = CreateInstalledAppByRegistryKey(szKeyName, hKey, rki);
                if (Info)
                {
                    if (List)
                        List->AddTail(Info);
                    else
                        return Info;
                }
            }
        }
    }
    return NULL;
}

Referenced by CreateInstalledAppByRegistryKey(), and UpdateInstalled().

EnumInstalledRootKey()

HKEY CAppDB::EnumInstalledRootKey ( UINT  Index, REGSAM &  RegSam  )

static

Definition at line 157 of file appdb.cpp.

{
    // Loop for through all combinations.
    // Note that HKEY_CURRENT_USER\Software does not have a redirect
    // https://docs.microsoft.com/en-us/windows/win32/winprog64/shared-registry-keys#redirected-shared-and-reflected-keys-under-wow64
    if (Index < (IsSystem64Bit() ? 3 : 2))
        return GetRootKeyInfo(Index, RegSam);
    else
        return NULL;
}

Referenced by EnumerateRegistry().

FindAvailableByPackageName()

CAvailableApplicationInfo * CAppDB::FindAvailableByPackageName

(

const CStringW &

name

)

Definition at line 38 of file appdb.cpp.

{
    POSITION CurrentListPosition = m_Available.GetHeadPosition();
    while (CurrentListPosition)
    {
        CAppInfo *Info = m_Available.GetNext(CurrentListPosition);
        if (Info->szIdentifier == name)
        {
            return static_cast<CAvailableApplicationInfo *>(Info);
        }
    }
    return NULL;
}

Referenced by FindByPackageName(), HandleGenerateInstallerCommand(), and HandleUninstallCommand().

FindByPackageName()

CAppInfo * CAppDB::FindByPackageName ( const CStringW &  name )

inline

Definition at line 31 of file appdb.h.

{ return FindAvailableByPackageName(name); }

Referenced by EnumerateFiles(), HandleInfoCommand(), HandleInstallCommand(), and HandleSetupCommand().

GetApps()

void CAppDB::GetApps	(	CAtlList< CAppInfo * > &	List,
		AppsCategories	Type
	)		const

Definition at line 53 of file appdb.cpp.

{
    const BOOL UseInstalled = IsInstalledEnum(Type);
    const CAtlList<CAppInfo *> &list = UseInstalled ? m_Installed : m_Available;
    const BOOL IncludeAll = UseInstalled ? (Type == ENUM_ALL_INSTALLED) : (Type == ENUM_ALL_AVAILABLE);
 
    POSITION CurrentListPosition = list.GetHeadPosition();
    while (CurrentListPosition)
    {
        CAppInfo *Info = list.GetNext(CurrentListPosition);
 
        if (IncludeAll || Type == Info->iCategory)
        {
            List.AddTail(Info);
        }
    }
}

Referenced by HandleFindCommand(), SearchForAppWithDisplayName(), and CMainWindow::UpdateApplicationsList().

GetAvailableCount()

size_t CAppDB::GetAvailableCount ( ) const

inline

Definition at line 50 of file appdb.h.

    {
        return m_Available.GetCount();
    }

Referenced by CMainWindow::CheckAvailable().

RemoveCached()

VOID CAppDB::RemoveCached

(

)

Definition at line 291 of file appdb.cpp.

{
    // Delete icons
    CPathW AppsPath = m_BasePath;
    AppsPath += RAPPS_DATABASE_SUBDIR;
    CPathW IconPath = AppsPath;
    IconPath += L"icons";
    DeleteWithWildcard(IconPath, L"*.ico");
 
    // Delete leftover screenshots
    CPathW ScrnshotFolder = AppsPath;
    ScrnshotFolder += L"screenshots";
    DeleteWithWildcard(ScrnshotFolder, L"*.tmp");
 
    // Delete data base files (*.txt)
    DeleteWithWildcard(AppsPath, L"*.txt");
 
    RemoveDirectoryW(IconPath);
    RemoveDirectoryW(ScrnshotFolder);
    RemoveDirectoryW(AppsPath);
    RemoveDirectoryW(m_BasePath);
}

Referenced by CMainWindow::CheckAvailable(), CMainWindow::OnCommand(), and ParseCmdAndExecute().

RemoveInstalledAppFromRegistry()

DWORD CAppDB::RemoveInstalledAppFromRegistry ( const CAppInfo *  Info )

static

Definition at line 315 of file appdb.cpp.

{
    // Validate that this is actually an installed app / update
    ATLASSERT(Info->iCategory == ENUM_INSTALLED_APPLICATIONS || Info->iCategory == ENUM_UPDATES);
    if (Info->iCategory != ENUM_INSTALLED_APPLICATIONS && Info->iCategory != ENUM_UPDATES)
        return ERROR_INVALID_PARAMETER;
 
    const CInstalledApplicationInfo *InstalledInfo = static_cast<const CInstalledApplicationInfo *>(Info);
 
    CStringW Name = InstalledInfo->szIdentifier;
    REGSAM wowsam;
    HKEY hRoot = GetRootKeyInfo(InstalledInfo->m_KeyInfo, wowsam);
    ATLASSERT(hRoot);
    if (!hRoot)
        return ERROR_OPEN_FAILED;
 
    CRegKey Uninstall;
    LSTATUS err = Uninstall.Open(hRoot, UNINSTALL_SUBKEY, KEY_READ | KEY_WRITE | wowsam);
    if (err == ERROR_SUCCESS)
    {
        err = Uninstall.RecurseDeleteKey(Name);
    }
    return err;
}

Referenced by CMainWindow::RemoveSelectedAppFromRegistry(), and UninstallThread().

UpdateAvailable()

VOID CAppDB::UpdateAvailable

(

)

Definition at line 121 of file appdb.cpp.

{
    if (!CreateDirectoryW(m_BasePath, NULL) && GetLastError() != ERROR_ALREADY_EXISTS)
        return;
 
    if (EnumerateFiles())
        return;
 
    DownloadApplicationsDB(
        SettingsInfo.bUseSource ? SettingsInfo.szSourceURL : APPLICATION_DATABASE_URL, !SettingsInfo.bUseSource);
 
    CPathW AppsPath = m_BasePath;
    AppsPath += RAPPS_DATABASE_SUBDIR;
    if (!ExtractFilesFromCab(APPLICATION_DATABASE_NAME, m_BasePath, AppsPath))
        return;
 
    CPathW CabFile = m_BasePath;
    CabFile += APPLICATION_DATABASE_NAME;
    DeleteFileW(CabFile);
 
    EnumerateFiles();
}

Referenced by CMainWindow::CheckAvailable(), ParseCmdAndExecute(), and CMainWindow::UpdateApplicationsList().

UpdateInstalled()

VOID CAppDB::UpdateInstalled

(

)

Definition at line 236 of file appdb.cpp.

{
    // Remove all old entries
    ClearList(m_Installed);
 
    EnumerateRegistry(&m_Installed, NULL);
}

Referenced by ParseCmdAndExecute(), and CMainWindow::UpdateApplicationsList().

Member Data Documentation

m_Available

CAtlList<CAppInfo *> CAppDB::m_Available

private

Definition at line 12 of file appdb.h.

Referenced by EnumerateFiles(), FindAvailableByPackageName(), GetApps(), and GetAvailableCount().

m_BasePath

CPathW CAppDB::m_BasePath

private

Definition at line 11 of file appdb.h.

Referenced by CAppDB(), EnumerateFiles(), RemoveCached(), and UpdateAvailable().

m_Installed

CAtlList<CAppInfo *> CAppDB::m_Installed

private

Definition at line 13 of file appdb.h.

Referenced by GetApps(), and UpdateInstalled().

---

The documentation for this class was generated from the following files:

• base/applications/rapps/include/appdb.h
• base/applications/rapps/appdb.cpp