160 lines
3.8 KiB
C
160 lines
3.8 KiB
C
/* * * * * * * *\
|
|
REGISTRY.C -
|
|
Copyright (c) 2023 freedom7341, Freedom Desktop
|
|
DESCRIPTION -
|
|
Program Manager's registry and settings related functions.
|
|
LICENSE INFORMATION -
|
|
MIT License, see LICENSE.txt in the root folder
|
|
\* * * * * * * */
|
|
|
|
/* Headers */
|
|
#include "progmgr.h"
|
|
#include "registry.h"
|
|
// #define WIN32_LEAN_AND_MEAN
|
|
#include <Windows.h>
|
|
#include <shlwapi.h>
|
|
#include <strsafe.h>
|
|
|
|
/* Variables */
|
|
// Global HKEYs
|
|
HKEY hKeyProgramManager = NULL;
|
|
HKEY hKeyProgramGroups = NULL;
|
|
HKEY hKeySettings = NULL;
|
|
// Global Values
|
|
BOOL bSaveSettings;
|
|
BOOL bShowUsername;
|
|
BOOL bTopMost;
|
|
BOOL bMinOnRun;
|
|
BOOL bAutoArrange;
|
|
DWORD dwSettingsMask = {
|
|
PMS_SAVESETTINGS |
|
|
PMS_SHOWUSERNAME |
|
|
!PMS_TOPMOST |
|
|
!PMS_MINONRUN |
|
|
PMS_AUTOARRANGE
|
|
};
|
|
RECT rcMainWindow;
|
|
// Registry Subkeys
|
|
PWSTR pszProgramGroups = L"Program Groups";
|
|
PWSTR pszSettings = L"Settings";
|
|
// Settings Subkeys
|
|
PWSTR pszSettingsWindow = L"Window";
|
|
PWSTR pszSettingsMask = L"SettingsMask";
|
|
|
|
/* Functions */
|
|
|
|
/* * * *\
|
|
InitializeRegistryKeys -
|
|
Takes the relevant registry keys and turns them
|
|
into valid and usable handles.
|
|
RETURNS -
|
|
TRUE if successful, FALSE if unsuccessful.
|
|
\* * * */
|
|
BOOL InitializeRegistryKeys()
|
|
{
|
|
if (!RegCreateKeyEx(HKEY_CURRENT_USER, PROGMGR_KEY, 0, szProgMgr, 0,
|
|
KEY_READ | KEY_WRITE, NULL, &hKeyProgramManager, NULL))
|
|
{
|
|
// Create Program Groups and Settings keys
|
|
RegCreateKeyEx(hKeyProgramManager, pszProgramGroups, 0, szProgMgr, 0,
|
|
KEY_READ | KEY_WRITE, NULL, &hKeyProgramGroups, NULL);
|
|
RegCreateKeyEx(hKeyProgramManager, pszSettings, 0, szProgMgr, 0,
|
|
KEY_READ | KEY_WRITE, NULL, &hKeySettings, NULL);
|
|
|
|
return TRUE;
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
/* * * *\
|
|
IsProgMgrDefaultShell -
|
|
Detects if Program Manager is the default shell.
|
|
RETURNS -
|
|
TRUE if Program Manager is the default shell,
|
|
FALSE if otherwise or an error occurs.
|
|
\* * * */
|
|
BOOL IsProgMgrDefaultShell()
|
|
{
|
|
HKEY hKeyWinlogon;
|
|
WCHAR szShell[HKEYMAXLEN] = L"";
|
|
DWORD dwType;
|
|
DWORD dwBufferSize = 0;
|
|
|
|
if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, WINLOGON_KEY,
|
|
0, KEY_READ, &hKeyWinlogon) == ERROR_SUCCESS)
|
|
{
|
|
if (RegQueryValueEx(hKeyWinlogon, L"Shell", 0,
|
|
&dwType, (LPBYTE)szShell, &dwBufferSize) == ERROR_SUCCESS)
|
|
{
|
|
if (StrStr(szShell, szProgMgr))
|
|
{
|
|
// ProgMgr detected >:)
|
|
RegCloseKey(hKeyWinlogon);
|
|
return TRUE;
|
|
}
|
|
else
|
|
{
|
|
// Inferior shell detected.
|
|
RegCloseKey(hKeyWinlogon);
|
|
return FALSE;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// Assume that we're the shell... just incase.
|
|
RegCloseKey(hKeyWinlogon);
|
|
return TRUE;
|
|
}
|
|
// No registry access.
|
|
return FALSE;
|
|
}
|
|
|
|
/* * * *\
|
|
SaveConfig -
|
|
Finds and collapses all settings and saves them to the registry.
|
|
RETURNS -
|
|
TRUE if successful, FALSE if unsuccessful.
|
|
\* * * */
|
|
BOOL SaveConfig()
|
|
{
|
|
// Shrink the settings into the bitmask
|
|
dwSettingsMask =
|
|
(bAutoArrange * PMS_AUTOARRANGE) |
|
|
(bMinOnRun * PMS_MINONRUN) |
|
|
(bTopMost * PMS_TOPMOST) |
|
|
(bShowUsername * PMS_SHOWUSERNAME) |
|
|
(bSaveSettings * PMS_SAVESETTINGS);
|
|
|
|
// Save settings bitmask
|
|
if (RegSetValueEx(hKeySettings, pszSettingsMask, 0, REG_DWORD, &dwSettingsMask, sizeof(DWORD)) == ERROR_SUCCESS)
|
|
{
|
|
return TRUE;
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
/* * * *\
|
|
BOOL LoadConfig() -
|
|
Finds all settings and retrieves them from the registry.
|
|
RETURNS -
|
|
TRUE if successful, FALSE if unsuccessful.
|
|
\* * * */
|
|
BOOL LoadConfig()
|
|
{
|
|
DWORD dwBufferSize = 0;
|
|
|
|
// Load settings bitmask
|
|
if (RegGetValue(hKeySettings, L"", pszSettingsMask, RRF_RT_DWORD, NULL, &dwSettingsMask, &dwBufferSize))
|
|
{
|
|
bAutoArrange = ((dwSettingsMask & PMS_AUTOARRANGE) == PMS_AUTOARRANGE);
|
|
bMinOnRun = ((dwSettingsMask & PMS_MINONRUN) == PMS_MINONRUN);
|
|
bTopMost = ((dwSettingsMask & PMS_TOPMOST) == PMS_TOPMOST);
|
|
bSaveSettings = ((dwSettingsMask & PMS_SAVESETTINGS) == PMS_SAVESETTINGS);
|
|
bShowUsername = ((dwSettingsMask & PMS_SHOWUSERNAME) == PMS_SHOWUSERNAME);
|
|
return TRUE;
|
|
}
|
|
|
|
return FALSE;
|
|
} |