reactos/rosapps/applications/sysutils/fontsub/fontsub.cpp

1604 lines
44 KiB
C++

// FontSub by Katayama Hirofumi MZ
//
// To the extent possible under law, the person who associated CC0 with
// FontSub has waived all copyright and related or neighboring rights
// to FontSub.
//
// You should have received a copy of the CC0 legalcode along with this
// work. If not, see <http://creativecommons.org/publicdomain/zero/1.0/>.
#include <windows.h>
#include <windowsx.h>
#include <commctrl.h>
#include <tchar.h>
#include <vector> // for std::vector
#include <set> // for std::set
#include <string> // for std::basic_string
#include <algorithm> // for std::sort
#include <cstdio>
#include <cstring>
#include <cassert>
#include "resource.h"
#define NAME_COLUMN_WIDTH 250
#define SUB_COLUMN_WIDTH 250
#define MAX_STRING 120
#ifndef _countof
#define _countof(array) (sizeof(array) / sizeof(array[0]))
#endif
typedef std::wstring STRING;
struct ITEM
{
STRING m_Name, m_Substitute;
BYTE m_CharSet1, m_CharSet2;
ITEM(const STRING& Name, const STRING& Substitute,
BYTE CharSet1, BYTE CharSet2)
: m_Name(Name), m_Substitute(Substitute),
m_CharSet1(CharSet1), m_CharSet2(CharSet2) { }
};
/* global variables */
HINSTANCE g_hInstance = NULL;
HWND g_hMainWnd = NULL;
HICON g_hIcon = NULL;
HWND g_hListView = NULL;
BOOL g_bModified = FALSE;
BOOL g_bNeedsReboot = FALSE;
INT g_iItem = 0;
LPCWSTR g_pszClassName = L"ReactOS Font Substitutes Editor";
LPCWSTR g_pszFileHeader = L"Windows Registry Editor Version 5.00";
WCHAR g_szTitle[MAX_STRING];
WCHAR g_szNameHead[MAX_STRING];
WCHAR g_szSubstituteHead[MAX_STRING];
INT g_iSortColumn = 0;
BOOL g_bSortAscendant = TRUE;
LPCWSTR g_pszKey =
L"SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\FontSubstitutes";
typedef std::set<STRING> FONTNAMESET;
typedef std::vector<ITEM> ITEMVECTOR;
FONTNAMESET g_Names;
ITEMVECTOR g_Items;
STRING g_strFontName;
STRING g_strSubstitute;
BYTE g_CharSet1 = DEFAULT_CHARSET;
BYTE g_CharSet2 = DEFAULT_CHARSET;
typedef struct CHARSET_ENTRY
{
BYTE CharSet;
LPCWSTR DisplayName;
} CHARSET_ENTRY;
CHARSET_ENTRY g_CharSetList[] =
{
{ DEFAULT_CHARSET, L"DEFAULT_CHARSET (1)" },
{ ANSI_CHARSET, L"ANSI_CHARSET (0)" },
{ SYMBOL_CHARSET, L"SYMBOL_CHARSET (2)" },
{ SHIFTJIS_CHARSET, L"SHIFTJIS_CHARSET (128)" },
{ HANGUL_CHARSET, L"HANGUL_CHARSET (129)" },
{ GB2312_CHARSET, L"GB2312_CHARSET (134)" },
{ CHINESEBIG5_CHARSET, L"CHINESEBIG5_CHARSET (136)" },
{ OEM_CHARSET, L"OEM_CHARSET (255)" },
{ JOHAB_CHARSET, L"JOHAB_CHARSET (130)" },
{ HEBREW_CHARSET, L"HEBREW_CHARSET (177)" },
{ ARABIC_CHARSET, L"ARABIC_CHARSET (178)" },
{ GREEK_CHARSET, L"GREEK_CHARSET (161)" },
{ TURKISH_CHARSET, L"TURKISH_CHARSET (162)" },
{ VIETNAMESE_CHARSET, L"VIETNAMESE_CHARSET (163)" },
{ THAI_CHARSET, L"THAI_CHARSET (222)" },
{ EASTEUROPE_CHARSET, L"EASTEUROPE_CHARSET (238)" },
{ RUSSIAN_CHARSET, L"RUSSIAN_CHARSET (204)" },
{ MAC_CHARSET, L"MAC_CHARSET (77)" },
{ BALTIC_CHARSET, L"BALTIC_CHARSET (186)" }
};
const WCHAR g_LongestName[] = L"CHINESEBIG5_CHARSET (136)";
static void trim(STRING& str)
{
static const WCHAR Spaces[] = L" \t\r\n";
size_t i = str.find_first_not_of(Spaces);
size_t j = str.find_last_not_of(Spaces);
if (i == STRING::npos || j == STRING::npos)
{
str.clear();
}
else
{
str = str.substr(i, j - i + 1);
}
}
static int CALLBACK
EnumFontFamExProc(const ENUMLOGFONTW *pelf,
const NEWTEXTMETRICW *pntm,
int FontType,
LPARAM lParam)
{
switch (pelf->elfFullName[0])
{
case UNICODE_NULL: case L'@':
break;
default:
g_Names.insert((const WCHAR *)pelf->elfFullName);
}
switch (pelf->elfLogFont.lfFaceName[0])
{
case UNICODE_NULL: case L'@':
break;
default:
g_Names.insert(pelf->elfLogFont.lfFaceName);
}
return 1;
}
BOOL DoLoadNames(void)
{
g_Names.clear();
LOGFONTW lf;
ZeroMemory(&lf, sizeof(lf));
lf.lfCharSet = DEFAULT_CHARSET;
HDC hDC = CreateCompatibleDC(NULL);
EnumFontFamiliesExW(hDC, &lf, (FONTENUMPROCW)EnumFontFamExProc, 0, 0);
DeleteDC(hDC);
return !g_Names.empty();
}
inline bool ItemCompareByNameAscend(const ITEM& Item1, const ITEM& Item2)
{
return Item1.m_Name < Item2.m_Name;
}
inline bool ItemCompareByNameDescend(const ITEM& Item1, const ITEM& Item2)
{
return Item1.m_Name > Item2.m_Name;
}
inline bool ItemCompareBySubAscend(const ITEM& Item1, const ITEM& Item2)
{
return Item1.m_Substitute < Item2.m_Substitute;
}
inline bool ItemCompareBySubDescend(const ITEM& Item1, const ITEM& Item2)
{
return Item1.m_Substitute > Item2.m_Substitute;
}
void DoSort(INT iColumn, BOOL bAscendant = TRUE)
{
LV_COLUMN Column;
ZeroMemory(&Column, sizeof(Column));
Column.mask = LVCF_IMAGE | LVCF_SUBITEM;
Column.iImage = 2;
Column.iSubItem = 0;
ListView_SetColumn(g_hListView, 0, &Column);
Column.iSubItem = 1;
ListView_SetColumn(g_hListView, 1, &Column);
switch (iColumn)
{
case 0:
Column.iSubItem = 0;
if (bAscendant)
{
std::sort(g_Items.begin(), g_Items.end(),
ItemCompareByNameAscend);
Column.iImage = 0;
ListView_SetColumn(g_hListView, 0, &Column);
}
else
{
std::sort(g_Items.begin(), g_Items.end(),
ItemCompareByNameDescend);
Column.iImage = 1;
ListView_SetColumn(g_hListView, 0, &Column);
}
break;
case 1:
Column.iSubItem = 1;
if (bAscendant)
{
std::sort(g_Items.begin(), g_Items.end(),
ItemCompareBySubAscend);
Column.iImage = 0;
ListView_SetColumn(g_hListView, 1, &Column);
}
else
{
std::sort(g_Items.begin(), g_Items.end(),
ItemCompareBySubDescend);
Column.iImage = 1;
ListView_SetColumn(g_hListView, 1, &Column);
}
break;
}
g_iSortColumn = iColumn;
g_bSortAscendant = bAscendant;
InvalidateRect(g_hListView, NULL, TRUE);
}
void LV_AddItems(HWND hwnd)
{
ListView_DeleteAllItems(hwnd);
LV_ITEM Item;
ZeroMemory(&Item, sizeof(Item));
Item.mask = LVIF_PARAM;
const INT Count = INT(g_Items.size());
for (INT i = 0; i < Count; ++i)
{
Item.iItem = i;
Item.iSubItem = 0;
Item.lParam = i;
ListView_InsertItem(hwnd, &Item);
Item.iItem = i;
Item.iSubItem = 1;
Item.lParam = i;
ListView_InsertItem(hwnd, &Item);
}
}
BOOL DoLoadItems(void)
{
ITEMVECTOR Items;
HKEY hKey = NULL;
RegOpenKeyExW(HKEY_LOCAL_MACHINE, g_pszKey, 0, KEY_READ, &hKey);
if (hKey == NULL)
return FALSE;
WCHAR szName[MAX_STRING], szValue[MAX_STRING];
DWORD cbName, cbValue;
for (DWORD dwIndex = 0; ; ++dwIndex)
{
cbName = sizeof(szName);
cbValue = sizeof(szValue);
LONG Error = RegEnumValueW(hKey, dwIndex, szName, &cbName,
NULL, NULL, (LPBYTE)szValue, &cbValue);
if (Error != ERROR_SUCCESS)
break;
BYTE CharSet1 = DEFAULT_CHARSET, CharSet2 = DEFAULT_CHARSET;
LPWSTR pch;
pch = wcsrchr(szName, L',');
if (pch)
{
*pch = 0;
CharSet1 = (BYTE)_wtoi(pch + 1);
}
pch = wcsrchr(szValue, L',');
if (pch)
{
*pch = 0;
CharSet2 = (BYTE)_wtoi(pch + 1);
}
ITEM Item(szName, szValue, CharSet1, CharSet2);
trim(Item.m_Name);
trim(Item.m_Substitute);
Items.push_back(Item);
}
RegCloseKey(hKey);
g_Items = Items;
LV_AddItems(g_hListView);
DoSort(0, TRUE);
g_bModified = FALSE;
g_bNeedsReboot = FALSE;
return !g_Items.empty();
}
BOOL DoLoad(void)
{
return DoLoadNames() && DoLoadItems();
}
void LV_InvalidateRow(HWND hwnd, INT iRow = -1)
{
if (iRow == -1)
iRow = ListView_GetNextItem(hwnd, -1, LVNI_SELECTED);
if (iRow == -1)
return;
RECT Rect;
LPRECT GccIsWhining = &Rect;
ListView_GetItemRect(hwnd, iRow, GccIsWhining, LVIR_BOUNDS);
InvalidateRect(hwnd, &Rect, FALSE);
}
BOOL LV_Init(HWND hwnd)
{
ListView_SetExtendedListViewStyle(hwnd,
LVS_EX_FULLROWSELECT | LVS_EX_GRIDLINES);
HIMAGELIST hImageList;
hImageList = ImageList_Create(12, 12, ILC_COLOR8 | ILC_MASK, 2, 2);
HBITMAP hbm;
hbm = (HBITMAP)LoadImage(g_hInstance, MAKEINTRESOURCE(2), IMAGE_BITMAP,
12, 12, LR_CREATEDIBSECTION | LR_LOADMAP3DCOLORS);
assert(hbm);
ImageList_AddMasked(hImageList, hbm, RGB(192, 192, 192));
DeleteObject(hbm);
hbm = (HBITMAP)LoadImage(g_hInstance, MAKEINTRESOURCE(3), IMAGE_BITMAP,
12, 12, LR_CREATEDIBSECTION | LR_LOADMAP3DCOLORS);
assert(hbm);
ImageList_AddMasked(hImageList, hbm, RGB(192, 192, 192));
DeleteObject(hbm);
hbm = (HBITMAP)LoadImage(g_hInstance, MAKEINTRESOURCE(4), IMAGE_BITMAP,
12, 12, LR_CREATEDIBSECTION | LR_LOADMAP3DCOLORS);
assert(hbm);
ImageList_AddMasked(hImageList, hbm, RGB(192, 192, 192));
DeleteObject(hbm);
ListView_SetImageList(hwnd, hImageList, LVSIL_SMALL);
LV_COLUMNW Column;
ZeroMemory(&Column, sizeof(Column));
Column.mask = LVCF_FMT | LVCF_TEXT | LVCF_SUBITEM | LVCF_WIDTH | LVCF_IMAGE;
Column.fmt = LVCFMT_LEFT;
Column.cx = NAME_COLUMN_WIDTH;
Column.pszText = g_szNameHead;
Column.iSubItem = 0;
Column.iImage = 0;
ListView_InsertColumn(hwnd, 0, &Column);
Column.cx = SUB_COLUMN_WIDTH;
Column.pszText = g_szSubstituteHead;
Column.iSubItem = 1;
Column.iImage = 2;
ListView_InsertColumn(hwnd, 1, &Column);
UINT State = LVIS_SELECTED | LVIS_FOCUSED;
ListView_SetItemState(hwnd, 0, State, State);
return TRUE;
}
BOOL EditDlg_OnInitDialog(HWND hwnd, HWND hwndFocus, LPARAM lParam)
{
COMBOBOXEXITEMW Item;
ZeroMemory(&Item, sizeof(Item));
Item.mask = CBEIF_TEXT;
FONTNAMESET::iterator it, end = g_Names.end();
for (it = g_Names.begin(); it != end; ++it)
{
Item.pszText = const_cast<LPWSTR>(it->c_str());
Item.iItem = ComboBox_GetCount(GetDlgItem(hwnd, cmb2));
SendDlgItemMessageW(hwnd, cmb2, CBEM_INSERTITEM, 0, (LPARAM)&Item);
}
SetDlgItemTextW(hwnd, edt1, g_strFontName.c_str());
SetDlgItemTextW(hwnd, cmb2, g_strSubstitute.c_str());
const INT Count = _countof(g_CharSetList);
for (INT i = 0; i < Count; ++i)
{
Item.pszText = const_cast<LPWSTR>(g_CharSetList[i].DisplayName);
Item.iItem = ComboBox_GetCount(GetDlgItem(hwnd, cmb3));
SendDlgItemMessageW(hwnd, cmb3, CBEM_INSERTITEM, 0, (LPARAM)&Item);
Item.iItem = ComboBox_GetCount(GetDlgItem(hwnd, cmb4));
SendDlgItemMessageW(hwnd, cmb4, CBEM_INSERTITEM, 0, (LPARAM)&Item);
}
SendDlgItemMessageW(hwnd, cmb3, CB_SETCURSEL, 0, 0);
SendDlgItemMessageW(hwnd, cmb4, CB_SETCURSEL, 0, 0);
for (INT i = 0; i < Count; ++i)
{
if (g_CharSet1 == g_CharSetList[i].CharSet)
{
SendDlgItemMessageW(hwnd, cmb3, CB_SETCURSEL, i, 0);
}
if (g_CharSet2 == g_CharSetList[i].CharSet)
{
SendDlgItemMessageW(hwnd, cmb4, CB_SETCURSEL, i, 0);
}
}
SIZE siz;
HDC hDC = CreateCompatibleDC(NULL);
SelectObject(hDC, GetStockObject(DEFAULT_GUI_FONT));
GetTextExtentPoint32W(hDC, g_LongestName, lstrlenW(g_LongestName), &siz);
DeleteDC(hDC);
SendDlgItemMessageW(hwnd, cmb3, CB_SETHORIZONTALEXTENT, siz.cx + 16, 0);
SendDlgItemMessageW(hwnd, cmb4, CB_SETHORIZONTALEXTENT, siz.cx + 16, 0);
EnableWindow(GetDlgItem(hwnd, cmb3), FALSE);
return TRUE;
}
void LV_OnDelete(HWND hwnd, INT iRow = -1)
{
if (iRow == -1)
iRow = ListView_GetNextItem(hwnd, -1, LVNI_SELECTED);
if (iRow == -1)
return;
UINT State = LVIS_SELECTED | LVIS_FOCUSED;
ListView_SetItemState(g_hListView, iRow, State, State);
WCHAR sz[MAX_STRING];
LoadStringW(g_hInstance, IDS_QUERYDELETE, sz, _countof(sz));
if (IDYES != MessageBoxW(g_hMainWnd, sz, g_szTitle,
MB_ICONINFORMATION | MB_YESNO))
{
return;
}
ListView_DeleteItem(hwnd, iRow);
g_Items.erase(g_Items.begin() + iRow);
g_bModified = TRUE;
ListView_SetItemState(g_hListView, iRow, State, State);
InvalidateRect(hwnd, NULL, TRUE);
}
void EditDlg_OnCommand(HWND hwnd, int id, HWND hwndCtl, UINT codeNotify)
{
WCHAR szValue[MAX_STRING];
STRING str;
INT i;
switch (id)
{
case IDOK:
GetDlgItemTextW(hwnd, cmb2, szValue, _countof(szValue));
str = szValue;
trim(str);
if (str.empty())
{
WCHAR sz[MAX_STRING];
SendDlgItemMessageW(hwnd, cmb2, CB_SETEDITSEL, 0, MAKELPARAM(0, -1));
SetFocus(GetDlgItem(hwnd, cmb2));
LoadStringW(g_hInstance, IDS_ENTERNAME2, sz, _countof(sz));
MessageBoxW(hwnd, sz, NULL, MB_ICONERROR);
return;
}
g_Items[g_iItem].m_CharSet2 = DEFAULT_CHARSET;
i = SendDlgItemMessageW(hwnd, cmb4, CB_GETCURSEL, 0, 0);
if (i != CB_ERR)
{
g_Items[g_iItem].m_CharSet2 = g_CharSetList[i].CharSet;
}
g_Items[g_iItem].m_Substitute = str;
g_bModified = TRUE;
EndDialog(hwnd, IDOK);
break;
case IDCANCEL:
EndDialog(hwnd, IDCANCEL);
break;
case psh1:
LV_OnDelete(g_hListView, g_iItem);
EndDialog(hwnd, psh1);
break;
}
}
INT_PTR CALLBACK
EditDlg_DlgProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
switch (uMsg)
{
HANDLE_MSG(hwnd, WM_INITDIALOG, EditDlg_OnInitDialog);
HANDLE_MSG(hwnd, WM_COMMAND, EditDlg_OnCommand);
}
return 0;
}
void LV_OnDblClk(HWND hwnd)
{
g_iItem = ListView_GetNextItem(hwnd, -1, LVNI_SELECTED);
if (g_iItem == -1)
return;
g_strFontName = g_Items[g_iItem].m_Name;
g_strSubstitute = g_Items[g_iItem].m_Substitute;
g_CharSet1 = g_Items[g_iItem].m_CharSet1;
g_CharSet2 = g_Items[g_iItem].m_CharSet2;
DialogBox(g_hInstance, MAKEINTRESOURCE(IDD_EDIT), g_hMainWnd,
EditDlg_DlgProc);
InvalidateRect(g_hListView, NULL, TRUE);
}
BOOL MainWnd_OnCreate(HWND hwnd, LPCREATESTRUCT lpCreateStruct)
{
DWORD dwStyle = WS_CHILD | WS_VISIBLE | WS_VSCROLL |
LVS_SINGLESEL | LVS_REPORT | LVS_OWNERDRAWFIXED;
DWORD dwExStyle = WS_EX_CLIENTEDGE;
g_hListView = CreateWindowEx(dwExStyle, WC_LISTVIEW, NULL, dwStyle,
0, 0, 0, 0,
hwnd, (HMENU)1, g_hInstance, NULL);
if (g_hListView == NULL)
return FALSE;
if (!LV_Init(g_hListView))
return FALSE;
if (!DoLoad())
return FALSE;
UINT State = LVIS_SELECTED | LVIS_FOCUSED;
ListView_SetItemState(g_hListView, 0, State, State);
SetFocus(g_hListView);
return TRUE;
}
BOOL AddDlg_OnInitDialog(HWND hwnd, HWND hwndFocus, LPARAM lParam)
{
COMBOBOXEXITEMW Item;
ZeroMemory(&Item, sizeof(Item));
Item.iItem = -1;
Item.mask = CBEIF_TEXT;
FONTNAMESET::iterator it, end = g_Names.end();
for (it = g_Names.begin(); it != end; ++it)
{
Item.pszText = const_cast<LPWSTR>(it->c_str());
Item.iItem = ComboBox_GetCount(GetDlgItem(hwnd, cmb1));
SendDlgItemMessageW(hwnd, cmb1, CBEM_INSERTITEM, 0, (LPARAM)&Item);
Item.iItem = ComboBox_GetCount(GetDlgItem(hwnd, cmb2));
SendDlgItemMessageW(hwnd, cmb2, CBEM_INSERTITEM, 0, (LPARAM)&Item);
}
WCHAR szEnterName[MAX_STRING];
LoadStringW(g_hInstance, IDS_ENTERNAME, szEnterName, _countof(szEnterName));
SetDlgItemTextW(hwnd, cmb1, szEnterName);
SendDlgItemMessageW(hwnd, cmb1, CB_SETEDITSEL, 0, MAKELPARAM(0, -1));
const INT Count = _countof(g_CharSetList);
for (INT i = 0; i < Count; ++i)
{
Item.pszText = const_cast<LPWSTR>(g_CharSetList[i].DisplayName);
Item.iItem = ComboBox_GetCount(GetDlgItem(hwnd, cmb3));
SendDlgItemMessageW(hwnd, cmb3, CBEM_INSERTITEM, 0, (LPARAM)&Item);
Item.iItem = ComboBox_GetCount(GetDlgItem(hwnd, cmb4));
SendDlgItemMessageW(hwnd, cmb4, CBEM_INSERTITEM, 0, (LPARAM)&Item);
}
SendDlgItemMessageW(hwnd, cmb3, CB_SETCURSEL, 0, 0);
SendDlgItemMessageW(hwnd, cmb4, CB_SETCURSEL, 0, 0);
for (INT i = 0; i < Count; ++i)
{
if (g_CharSet1 == g_CharSetList[i].CharSet)
{
SendDlgItemMessageW(hwnd, cmb3, CB_SETCURSEL, i, 0);
}
if (g_CharSet2 == g_CharSetList[i].CharSet)
{
SendDlgItemMessageW(hwnd, cmb4, CB_SETCURSEL, i, 0);
}
}
SIZE siz;
HDC hDC = CreateCompatibleDC(NULL);
SelectObject(hDC, GetStockObject(DEFAULT_GUI_FONT));
GetTextExtentPoint32W(hDC, g_LongestName, lstrlenW(g_LongestName), &siz);
DeleteDC(hDC);
SendDlgItemMessageW(hwnd, cmb3, CB_SETHORIZONTALEXTENT, siz.cx + 16, 0);
SendDlgItemMessageW(hwnd, cmb4, CB_SETHORIZONTALEXTENT, siz.cx + 16, 0);
return TRUE;
}
void AddDlg_OnCommand(HWND hwnd, int id, HWND hwndCtl, UINT codeNotify)
{
WCHAR szKey[MAX_STRING], szValue[MAX_STRING], sz[MAX_STRING];
INT i, iCharSet1, iCharSet2;
BYTE CharSet1, CharSet2;
STRING key, value;
switch (id)
{
case IDOK:
GetDlgItemTextW(hwnd, cmb1, szKey, _countof(szKey));
key = szKey;
trim(key);
LoadStringW(g_hInstance, IDS_ENTERNAME, sz, _countof(sz));
if (key.empty() || key == sz)
{
SendDlgItemMessageW(hwnd, cmb1, CB_SETEDITSEL, 0, MAKELPARAM(0, -1));
SetFocus(GetDlgItem(hwnd, cmb1));
LoadStringW(g_hInstance, IDS_ENTERNAME2, sz, _countof(sz));
MessageBoxW(hwnd, sz, NULL, MB_ICONERROR);
return;
}
GetDlgItemTextW(hwnd, cmb2, szValue, _countof(szValue));
value = szValue;
trim(value);
if (value.empty())
{
SendDlgItemMessageW(hwnd, cmb2, CB_SETEDITSEL, 0, MAKELPARAM(0, -1));
SetFocus(GetDlgItem(hwnd, cmb2));
LoadStringW(g_hInstance, IDS_ENTERNAME2, sz, _countof(sz));
MessageBoxW(hwnd, sz, NULL, MB_ICONERROR);
return;
}
iCharSet1 = SendDlgItemMessageW(hwnd, cmb3, CB_GETCURSEL, 0, 0);
if (iCharSet1 == CB_ERR)
iCharSet1 = 0;
iCharSet2 = SendDlgItemMessageW(hwnd, cmb4, CB_GETCURSEL, 0, 0);
if (iCharSet2 == CB_ERR)
iCharSet2 = 0;
CharSet1 = g_CharSetList[iCharSet1].CharSet;
CharSet2 = g_CharSetList[iCharSet2].CharSet;
for (i = 0; i < (INT)g_Items.size(); ++i)
{
if (g_Items[i].m_Name == key &&
g_Items[i].m_CharSet1 == CharSet1)
{
WCHAR sz[MAX_STRING];
SendDlgItemMessageW(hwnd, cmb1, CB_SETEDITSEL, 0, MAKELPARAM(0, -1));
SetFocus(GetDlgItem(hwnd, cmb1));
LoadStringW(g_hInstance, IDS_ALREADYEXISTS, sz, _countof(sz));
MessageBoxW(hwnd, sz, NULL, MB_ICONERROR);
return;
}
}
{
ITEM Item(key, value, CharSet1, CharSet2);
g_Items.push_back(Item);
g_bModified = TRUE;
i = (INT)g_Items.size();
LV_ITEM LvItem;
ZeroMemory(&LvItem, sizeof(LvItem));
LvItem.mask = LVIF_PARAM;
LvItem.iItem = i;
LvItem.lParam = i;
LvItem.iSubItem = 0;
ListView_InsertItem(g_hListView, &LvItem);
LvItem.iSubItem = 1;
ListView_InsertItem(g_hListView, &LvItem);
}
g_bModified = TRUE;
EndDialog(hwnd, IDOK);
break;
case IDCANCEL:
EndDialog(hwnd, IDCANCEL);
break;
}
}
INT_PTR CALLBACK
AddDlg_DlgProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
switch (uMsg)
{
HANDLE_MSG(hwnd, WM_INITDIALOG, AddDlg_OnInitDialog);
HANDLE_MSG(hwnd, WM_COMMAND, AddDlg_OnCommand);
}
return 0;
}
void MainWnd_OnNew(HWND hwnd)
{
g_iItem = ListView_GetNextItem(hwnd, -1, LVNI_SELECTED);
if (g_iItem == -1)
return;
g_strFontName = g_Items[g_iItem].m_Name;
g_strSubstitute = g_Items[g_iItem].m_Substitute;
g_CharSet1 = g_Items[g_iItem].m_CharSet1;
g_CharSet2 = g_Items[g_iItem].m_CharSet2;
if (IDOK == DialogBox(g_hInstance, MAKEINTRESOURCE(IDD_ADD), g_hMainWnd,
AddDlg_DlgProc))
{
INT i = ListView_GetItemCount(g_hListView) - 1;
UINT State = LVIS_SELECTED | LVIS_FOCUSED;
ListView_SetItemState(g_hListView, i, State, State);
ListView_EnsureVisible(g_hListView, i, FALSE);
}
}
BOOL MainWnd_OnUpdateRegistry(HWND hwnd)
{
// open the key
HKEY hKey = NULL;
RegOpenKeyExW(HKEY_LOCAL_MACHINE, g_pszKey, 0, KEY_ALL_ACCESS, &hKey);
if (hKey == NULL)
return FALSE;
// clear all values
WCHAR szName[MAX_STRING], szValue[MAX_STRING];
DWORD cbName, cbValue;
for (;;)
{
cbName = sizeof(szName);
cbValue = sizeof(szValue);
LONG Error = RegEnumValueW(hKey, 0, szName, &cbName,
NULL, NULL, (LPBYTE)szValue, &cbValue);
if (Error != ERROR_SUCCESS)
break;
RegDeleteValueW(hKey, szName);
}
// set values
size_t Count = g_Items.size();
for (size_t i = 0; i < Count; ++i)
{
DWORD cbData = (g_Items[i].m_Substitute.size() + 1) * sizeof(WCHAR);
RegSetValueExW(hKey, g_Items[i].m_Name.c_str(), 0,
REG_SZ, (LPBYTE)g_Items[i].m_Substitute.c_str(), cbData);
}
// close now
RegCloseKey(hKey);
g_bModified = FALSE;
g_bNeedsReboot = TRUE;
return TRUE;
}
LPWSTR SkipSpace(LPCWSTR pch)
{
while (*pch && wcschr(L" \t\r\n", *pch) != NULL)
{
++pch;
}
return const_cast<LPWSTR>(pch);
}
LPWSTR SkipQuoted(LPWSTR pch)
{
++pch; // L'"'
while (*pch)
{
if (*pch == L'"')
{
++pch;
break;
}
if (*pch == L'\\')
{
++pch;
}
++pch;
}
return pch;
}
void UnescapeHex(const STRING& str, size_t& i, STRING& Ret, BOOL Unicode)
{
STRING Num;
// hexadecimal
if (iswxdigit(str[i]))
{
Num += str[i];
++i;
if (iswxdigit(str[i]))
{
Num += str[i];
++i;
if (Unicode)
{
if (iswxdigit(str[i]))
{
Num += str[i];
++i;
if (iswxdigit(str[i]))
{
Num += str[i];
++i;
}
}
}
}
}
if (!Num.empty())
{
Ret += (WCHAR)wcstoul(&Num[0], NULL, 16);
}
}
void UnescapeOther(const STRING& str, size_t& i, STRING& Ret)
{
STRING Num;
// check octal
if (L'0' <= str[i] && str[i] < L'8')
{
Num += str[i];
++i;
if (L'0' <= str[i] && str[i] < L'8')
{
Num += str[i];
++i;
if (L'0' <= str[i] && str[i] < L'8')
{
Num += str[i];
++i;
}
}
}
if (Num.empty())
{
Ret += str[i];
++i;
}
else
{
// octal
Ret += (WCHAR)wcstoul(&Num[0], NULL, 8);
}
}
// process escape sequence
void UnescapeChar(const STRING& str, size_t& i, STRING& Ret)
{
if (str[i] != L'\\')
{
Ret += str[i];
++i;
return;
}
++i;
switch (str[i])
{
case L'a': Ret += L'\a'; ++i; break;
case L'b': Ret += L'\b'; ++i; break;
case L'f': Ret += L'\f'; ++i; break;
case L'n': Ret += L'\n'; ++i; break;
case L'r': Ret += L'\r'; ++i; break;
case L't': Ret += L'\t'; ++i; break;
case L'v': Ret += L'\v'; ++i; break;
case L'x':
// hexidemical
++i;
UnescapeHex(str, i, Ret, FALSE);
break;
case L'u':
// Unicode hexidemical
++i;
UnescapeHex(str, i, Ret, TRUE);
break;
default:
// other case
UnescapeOther(str, i, Ret);
break;
}
}
STRING Unquote(const STRING& str)
{
if (str[0] != L'"')
return str;
STRING Ret;
size_t i = 1;
while (i < str.size())
{
if (str[i] == L'"' || str[i] == UNICODE_NULL)
break;
UnescapeChar(str, i, Ret);
}
return Ret;
}
BOOL DoParseFile(LPVOID pvContents, DWORD dwSize)
{
ITEMVECTOR Items;
LPWSTR pch, pchSep, pchStart = (LPWSTR)pvContents;
pchStart[dwSize / sizeof(WCHAR)] = UNICODE_NULL;
// check header
const DWORD cbHeader = lstrlenW(g_pszFileHeader) * sizeof(WCHAR);
if (memcmp(pchStart, g_pszFileHeader, cbHeader) != 0)
return FALSE;
pchStart += cbHeader / sizeof(WCHAR);
// find the key
WCHAR szKey[MAX_STRING];
wsprintfW(szKey, L"[HKEY_LOCAL_MACHINE\\%s]", g_pszKey);
pch = wcsstr(pchStart, szKey);
if (pch == NULL)
return FALSE;
pchStart = pch + lstrlenW(szKey);
for (;;)
{
pchStart = SkipSpace(pchStart);
if (*pchStart == UNICODE_NULL || *pchStart == L'[')
break;
pch = wcschr(pchStart, L'\n');
if (pch)
*pch = UNICODE_NULL;
pchSep = SkipQuoted(pchStart);
if (*pchSep == L'=')
{
*pchSep = UNICODE_NULL;
STRING key = pchStart;
trim(key);
key = Unquote(key);
STRING value = pchSep + 1;
trim(value);
value = Unquote(value);
BYTE CharSet1 = DEFAULT_CHARSET, CharSet2 = DEFAULT_CHARSET;
size_t pos;
pos = key.find(L',');
if (pos != STRING::npos)
{
CharSet1 = (BYTE)_wtoi(&key[pos + 1]);
key.resize(pos);
trim(key);
}
pos = value.find(L',');
if (pos != STRING::npos)
{
CharSet2 = (BYTE)_wtoi(&value[pos + 1]);
value.resize(pos);
trim(value);
}
ITEM Item(key, value, CharSet1, CharSet2);
Items.push_back(Item);
}
if (pch == NULL)
break;
pchStart = pch + 1;
}
g_Items = Items;
g_bModified = TRUE;
LV_AddItems(g_hListView);
return TRUE;
}
BOOL DoImport(HWND hwnd, LPCWSTR pszFile)
{
HANDLE hFile = CreateFileW(pszFile, GENERIC_READ,
FILE_SHARE_READ | FILE_SHARE_DELETE, NULL,
OPEN_EXISTING, FILE_FLAG_SEQUENTIAL_SCAN, NULL);
if (hFile == INVALID_HANDLE_VALUE)
return FALSE;
BOOL bSuccess = FALSE;
DWORD dwSize = GetFileSize(hFile, NULL);
if (dwSize != 0xFFFFFFFF)
{
std::vector<BYTE> Contents(dwSize + 2);
DWORD cbRead;
if (ReadFile(hFile, &Contents[0], dwSize, &cbRead, NULL) &&
cbRead == dwSize)
{
/* check BOM */
if (memcmp(&Contents[0], "\xFF\xFE", 2) == 0)
{
bSuccess = DoParseFile(&Contents[2], dwSize - 2);
}
else
{
bSuccess = DoParseFile(&Contents[0], dwSize);
}
}
}
CloseHandle(hFile);
return bSuccess;
}
STRING Escape(const STRING& str)
{
STRING Ret;
for (size_t i = 0; i < str.size(); ++i)
{
switch (str[i])
{
case L'"': case L'\\':
Ret += L'\\';
Ret += str[i];
break;
default:
Ret += str[i];
}
}
return Ret;
}
BOOL DoExport(HWND hwnd, LPCWSTR pszFile)
{
HANDLE hFile = CreateFileW(pszFile, GENERIC_WRITE, FILE_SHARE_READ,
NULL, CREATE_ALWAYS,
FILE_ATTRIBUTE_NORMAL | FILE_FLAG_WRITE_THROUGH, NULL);
if (hFile == INVALID_HANDLE_VALUE)
return FALSE;
BOOL bSuccess;
DWORD dwSize, cbWritten;
WCHAR szCharSet1[MAX_STRING], szCharSet2[MAX_STRING];
WCHAR szLine[MAX_STRING * 2 + 4];
/* write header */
dwSize = lstrlenW(g_pszFileHeader) * sizeof(WCHAR);
bSuccess =
WriteFile(hFile, "\xFF\xFE", 2, &cbWritten, NULL) &&
WriteFile(hFile, g_pszFileHeader, dwSize, &cbWritten, NULL);
if (bSuccess)
{
wsprintfW(szLine, L"\r\n\r\n[HKEY_LOCAL_MACHINE\\%s]\r\n", g_pszKey);
dwSize = lstrlenW(szLine) * sizeof(WCHAR);
bSuccess = WriteFile(hFile, szLine, dwSize, &cbWritten, NULL);
}
if (bSuccess)
{
size_t i, Count = g_Items.size();
for (i = 0; i < Count; ++i)
{
if (g_Items[i].m_CharSet1 != DEFAULT_CHARSET)
wsprintfW(szCharSet1, L",%u", g_Items[i].m_CharSet1);
else
szCharSet1[0] = UNICODE_NULL;
if (g_Items[i].m_CharSet2 != DEFAULT_CHARSET)
wsprintfW(szCharSet2, L",%u", g_Items[i].m_CharSet2);
else
szCharSet2[0] = UNICODE_NULL;
STRING Name = Escape(g_Items[i].m_Name);
STRING Substitute = Escape(g_Items[i].m_Substitute);
wsprintfW(szLine, L"\"%s%s\"=\"%s%s\"\r\n",
Name.c_str(), szCharSet1,
Substitute.c_str(), szCharSet2);
dwSize = lstrlenW(szLine) * sizeof(WCHAR);
if (!WriteFile(hFile, szLine, dwSize, &cbWritten, NULL))
{
bSuccess = FALSE;
break;
}
}
WriteFile(hFile, L"\r\n", 2 * sizeof(WCHAR), &cbWritten, NULL);
}
CloseHandle(hFile);
if (!bSuccess)
{
DeleteFileW(pszFile);
}
return bSuccess;
}
void MakeFilter(LPWSTR pszFilter)
{
while (*pszFilter)
{
if (*pszFilter == L'|')
*pszFilter = 0;
++pszFilter;
}
}
void MainWnd_OnImport(HWND hwnd)
{
OPENFILENAMEW ofn = {0};
WCHAR szFile[MAX_PATH] = L"";
WCHAR szImportTitle[MAX_STRING];
WCHAR szCannotImport[MAX_STRING];
WCHAR szImportFilter[MAX_STRING];
LoadStringW(g_hInstance, IDS_IMPORT, szImportTitle, _countof(szImportTitle));
LoadStringW(g_hInstance, IDS_CANTIMPORT, szCannotImport, _countof(szCannotImport));
LoadStringW(g_hInstance, IDS_INPFILTER, szImportFilter, _countof(szImportFilter));
MakeFilter(szImportFilter);
ofn.lStructSize = OPENFILENAME_SIZE_VERSION_400;
ofn.hwndOwner = hwnd;
ofn.lpstrFilter = szImportFilter;
ofn.lpstrFile = szFile;
ofn.nMaxFile = _countof(szFile);
ofn.lpstrTitle = szImportTitle;
ofn.Flags = OFN_DONTADDTORECENT | OFN_ENABLESIZING |
OFN_EXPLORER | OFN_FILEMUSTEXIST |
OFN_HIDEREADONLY | OFN_LONGNAMES |
OFN_PATHMUSTEXIST;
ofn.lpstrDefExt = L"reg";
if (GetOpenFileNameW(&ofn))
{
if (!DoImport(hwnd, szFile))
{
MessageBoxW(hwnd, szCannotImport, g_szTitle, MB_ICONERROR);
}
}
}
void MainWnd_OnExport(HWND hwnd)
{
OPENFILENAMEW ofn = {0};
WCHAR szFile[MAX_PATH] = L"";
WCHAR szExportTitle[MAX_STRING];
WCHAR szCannotExport[MAX_STRING];
WCHAR szExportFilter[MAX_STRING];
LoadStringW(g_hInstance, IDS_EXPORT, szExportTitle, _countof(szExportTitle));
LoadStringW(g_hInstance, IDS_CANTEXPORT, szCannotExport, _countof(szCannotExport));
LoadStringW(g_hInstance, IDS_OUTFILTER, szExportFilter, _countof(szExportFilter));
MakeFilter(szExportFilter);
ofn.lStructSize = OPENFILENAME_SIZE_VERSION_400;
ofn.hwndOwner = hwnd;
ofn.lpstrFilter = szExportFilter;
ofn.lpstrFile = szFile;
ofn.nMaxFile = _countof(szFile);
ofn.lpstrTitle = szExportTitle;
ofn.Flags = OFN_DONTADDTORECENT | OFN_ENABLESIZING |
OFN_EXPLORER | OFN_HIDEREADONLY | OFN_LONGNAMES |
OFN_PATHMUSTEXIST | OFN_OVERWRITEPROMPT;
ofn.lpstrDefExt = L"reg";
if (GetSaveFileNameW(&ofn))
{
if (!DoExport(hwnd, szFile))
{
MessageBoxW(hwnd, szCannotExport, g_szTitle, MB_ICONERROR);
}
}
}
void MainWnd_OnReload(HWND hwnd)
{
DoLoad();
}
void MainWnd_OnEdit(HWND hwnd)
{
LV_OnDblClk(g_hListView);
}
void MainWnd_OnDelete(HWND hwnd)
{
LV_OnDelete(g_hListView);
}
void MainWnd_OnOpenRegKey(HWND hwnd)
{
static const WCHAR s_szRegeditKey[] =
L"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Applets\\Regedit";
WCHAR sz[MAX_STRING];
// open regedit key
HKEY hKey = NULL;
LSTATUS Result = RegCreateKeyExW(HKEY_CURRENT_USER, s_szRegeditKey, 0,
NULL, 0, KEY_WRITE, NULL, &hKey, NULL);
if (Result != ERROR_SUCCESS)
{
LoadStringW(g_hInstance, IDS_CANTOPENKEY, sz, _countof(sz));
MessageBoxW(hwnd, sz, NULL, MB_ICONERROR);
return;
}
// set LastKey value
wsprintfW(sz, L"HKEY_LOCAL_MACHINE\\%s", g_pszKey);
DWORD dwSize = sizeof(sz);
Result = RegSetValueExW(hKey, L"LastKey", 0, REG_SZ,
(LPBYTE)sz, dwSize);
// close now
RegCloseKey(hKey);
if (Result != ERROR_SUCCESS)
{
LoadStringW(g_hInstance, IDS_CANTOPENKEY, sz, _countof(sz));
MessageBoxW(hwnd, sz, NULL, MB_ICONERROR);
return;
}
// open by regedit
ShellExecuteW(hwnd, NULL, L"regedit.exe", NULL, NULL, SW_SHOWNORMAL);
}
void MainWnd_OnAbout(HWND hwnd)
{
WCHAR szAbout[MAX_PATH];
LoadStringW(g_hInstance, IDS_ABOUT, szAbout, _countof(szAbout));
MSGBOXPARAMS Params;
ZeroMemory(&Params, sizeof(Params));
Params.cbSize = sizeof(Params);
Params.hwndOwner = hwnd;
Params.hInstance = g_hInstance;
Params.lpszText = szAbout;
Params.lpszCaption = g_szTitle;
Params.dwStyle = MB_OK | MB_USERICON;
Params.lpszIcon = MAKEINTRESOURCEW(1);
Params.dwLanguageId = LANG_USER_DEFAULT;
MessageBoxIndirectW(&Params);
}
void MainWnd_OnCommand(HWND hwnd, int id, HWND hwndCtl, UINT codeNotify)
{
switch (id)
{
case ID_NEW:
MainWnd_OnNew(hwnd);
break;
case ID_EDIT:
MainWnd_OnEdit(hwnd);
break;
case ID_EXIT:
PostMessage(hwnd, WM_CLOSE, 0, 0);
break;
case ID_RELOAD:
MainWnd_OnReload(hwnd);
break;
case ID_UPDATE_REGISTRY:
MainWnd_OnUpdateRegistry(hwnd);
break;
case ID_DELETE:
MainWnd_OnDelete(hwnd);
break;
case ID_IMPORT:
MainWnd_OnImport(hwnd);
break;
case ID_EXPORT:
MainWnd_OnExport(hwnd);
break;
case ID_OPEN_REGKEY:
MainWnd_OnOpenRegKey(hwnd);
break;
case ID_ABOUT:
MainWnd_OnAbout(hwnd);
break;
}
}
void MainWnd_OnDestroy(HWND hwnd)
{
PostQuitMessage(0);
}
void MainWnd_OnSize(HWND hwnd, UINT state, int cx, int cy)
{
MoveWindow(g_hListView, 0, 0, cx, cy, TRUE);
}
void MainWnd_OnDrawItem(HWND hwnd, const DRAWITEMSTRUCT *lpDrawItem)
{
if (lpDrawItem->CtlType != ODT_LISTVIEW)
return;
HDC hDC = lpDrawItem->hDC;
SetBkMode(hDC, TRANSPARENT);
INT iColumn = 0, x, cx;
RECT rcItem, rcSubItem, rcText;
STRING Str;
x = -GetScrollPos(g_hListView, SB_HORZ);
rcItem = lpDrawItem->rcItem;
if (lpDrawItem->itemState & ODS_SELECTED)
{
FillRect(hDC, &rcItem, (HBRUSH)(COLOR_HIGHLIGHT + 1));
SetTextColor(hDC, GetSysColor(COLOR_HIGHLIGHTTEXT));
}
else
{
FillRect(hDC, &rcItem, (HBRUSH)(COLOR_WINDOW + 1));
SetTextColor(hDC, GetSysColor(COLOR_WINDOWTEXT));
}
cx = ListView_GetColumnWidth(g_hListView, iColumn);
rcSubItem = rcItem;
rcSubItem.left = x;
rcSubItem.right = x + cx;
WCHAR sz[MAX_STRING];
rcText = rcSubItem;
InflateRect(&rcText, -1, -1);
Str = g_Items[lpDrawItem->itemID].m_Name;
BYTE CharSet1 = g_Items[lpDrawItem->itemID].m_CharSet1;
if (CharSet1 != DEFAULT_CHARSET)
wsprintfW(sz, L"%s,%u", Str.c_str(), CharSet1);
else
wsprintfW(sz, L"%s", Str.c_str());
DrawTextW(hDC, sz, lstrlenW(sz), &rcText,
DT_SINGLELINE | DT_LEFT | DT_VCENTER | DT_END_ELLIPSIS |
DT_NOPREFIX);
x += cx;
++iColumn;
cx = ListView_GetColumnWidth(g_hListView, iColumn);
rcSubItem = rcItem;
rcSubItem.left = x;
rcSubItem.right = x + cx;
rcText = rcSubItem;
InflateRect(&rcText, -1, -1);
Str = g_Items[lpDrawItem->itemID].m_Substitute;
BYTE CharSet2 = g_Items[lpDrawItem->itemID].m_CharSet2;
if (CharSet2 != DEFAULT_CHARSET)
wsprintfW(sz, L"%s,%u", Str.c_str(), CharSet2);
else
wsprintfW(sz, L"%s", Str.c_str());
DrawTextW(hDC, sz, lstrlenW(sz), &rcText,
DT_SINGLELINE | DT_LEFT | DT_VCENTER | DT_END_ELLIPSIS |
DT_NOPREFIX);
}
void MainWnd_OnMeasureItem(HWND hwnd, MEASUREITEMSTRUCT *lpMeasureItem)
{
if (lpMeasureItem->CtlType != ODT_LISTVIEW)
return;
TEXTMETRIC tm;
HDC hDC = GetDC(hwnd);
GetTextMetrics(hDC, &tm);
ReleaseDC(hwnd, hDC);
lpMeasureItem->itemHeight = tm.tmHeight * 4 / 3;
}
LRESULT MainWnd_OnNotify(HWND hwnd, int idFrom, NMHDR *pnmhdr)
{
NM_LISTVIEW *pNMLV = (NM_LISTVIEW *)pnmhdr;
LV_KEYDOWN *pLVKD = (LV_KEYDOWN *)pnmhdr;
switch (pnmhdr->code)
{
case LVN_COLUMNCLICK:
if (pNMLV->iSubItem == g_iSortColumn)
DoSort(pNMLV->iSubItem, !g_bSortAscendant);
else
DoSort(pNMLV->iSubItem, TRUE);
break;
case NM_DBLCLK:
LV_OnDblClk(g_hListView);
break;
case LVN_KEYDOWN:
if (pLVKD->wVKey == VK_RETURN) // [Enter] key
{
LV_OnDblClk(g_hListView);
}
if (pLVKD->wVKey == VK_DELETE) // [Del] key
{
LV_OnDelete(g_hListView);
}
break;
}
return 0;
}
LRESULT MainWnd_OnContextMenu(HWND hwnd, HWND hwndContext, UINT xPos, UINT yPos)
{
POINT pt = {(INT)xPos, (INT)yPos};
ScreenToClient(g_hListView, &pt);
SendMessageW(g_hListView, WM_LBUTTONDOWN, 0, MAKELPARAM(pt.x, pt.y));
HMENU hMenu = LoadMenu(g_hInstance, MAKEINTRESOURCE(2));
if (hMenu == NULL)
return 0;
HMENU hSubMenu = GetSubMenu(hMenu, 0);
if (hSubMenu == NULL)
return 0;
SetForegroundWindow(hwnd);
TrackPopupMenu(hSubMenu, TPM_LEFTALIGN | TPM_RIGHTBUTTON,
xPos, yPos, 0, g_hMainWnd, NULL);
PostMessage(g_hMainWnd, WM_NULL, 0, 0);
return 0;
}
void MainWnd_OnActivate(HWND hwnd, UINT state, HWND hwndActDeact, BOOL fMinimized)
{
if (state != WA_INACTIVE)
{
SetFocus(g_hListView);
}
}
BOOL EnableProcessPrivileges(LPCWSTR lpPrivilegeName, BOOL bEnable = TRUE)
{
HANDLE hToken;
LUID luid;
TOKEN_PRIVILEGES tokenPrivileges;
BOOL Ret;
Ret = ::OpenProcessToken(::GetCurrentProcess(),
TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY,
&hToken);
if (!Ret)
return Ret; // failure
Ret = ::LookupPrivilegeValueW(NULL, lpPrivilegeName, &luid);
if (Ret)
{
tokenPrivileges.PrivilegeCount = 1;
tokenPrivileges.Privileges[0].Luid = luid;
tokenPrivileges.Privileges[0].Attributes = bEnable ? SE_PRIVILEGE_ENABLED : 0;
Ret = ::AdjustTokenPrivileges(hToken, FALSE, &tokenPrivileges, 0, 0, 0);
}
::CloseHandle(hToken);
return Ret;
}
void MainWnd_OnClose(HWND hwnd)
{
if (!g_bNeedsReboot && !g_bModified)
{
DestroyWindow(hwnd);
return;
}
if (g_bModified)
{
WCHAR szUpdateNow[MAX_STRING];
LoadStringW(g_hInstance, IDS_QUERYUPDATE, szUpdateNow, _countof(szUpdateNow));
INT id = MessageBoxW(hwnd, szUpdateNow, g_szTitle,
MB_ICONINFORMATION | MB_YESNOCANCEL);
switch (id)
{
case IDYES:
MainWnd_OnUpdateRegistry(hwnd);
break;
case IDNO:
break;
case IDCANCEL:
return;
}
}
if (g_bNeedsReboot)
{
WCHAR szRebootNow[MAX_STRING];
LoadStringW(g_hInstance, IDS_REBOOTNOW, szRebootNow, _countof(szRebootNow));
INT id = MessageBoxW(hwnd, szRebootNow, g_szTitle,
MB_ICONINFORMATION | MB_YESNOCANCEL);
switch (id)
{
case IDYES:
EnableProcessPrivileges(SE_SHUTDOWN_NAME, TRUE);
::ExitWindowsEx(EWX_REBOOT | EWX_FORCE, 0);
break;
case IDNO:
break;
case IDCANCEL:
return;
}
}
::DestroyWindow(hwnd);
}
LRESULT CALLBACK
WindowProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
switch (uMsg)
{
HANDLE_MSG(hwnd, WM_CREATE, MainWnd_OnCreate);
HANDLE_MSG(hwnd, WM_COMMAND, MainWnd_OnCommand);
HANDLE_MSG(hwnd, WM_DESTROY, MainWnd_OnDestroy);
HANDLE_MSG(hwnd, WM_SIZE, MainWnd_OnSize);
HANDLE_MSG(hwnd, WM_DRAWITEM, MainWnd_OnDrawItem);
HANDLE_MSG(hwnd, WM_MEASUREITEM, MainWnd_OnMeasureItem);
HANDLE_MSG(hwnd, WM_NOTIFY, MainWnd_OnNotify);
HANDLE_MSG(hwnd, WM_CONTEXTMENU, MainWnd_OnContextMenu);
HANDLE_MSG(hwnd, WM_ACTIVATE, MainWnd_OnActivate);
HANDLE_MSG(hwnd, WM_CLOSE, MainWnd_OnClose);
default:
return DefWindowProc(hwnd, uMsg, wParam, lParam);
}
}
INT WINAPI wWinMain(
HINSTANCE hInstance,
HINSTANCE hPrevInstance,
LPWSTR lpCmdLine,
INT nCmdShow)
{
g_hInstance = hInstance;
InitCommonControls();
HACCEL hAccel = LoadAcceleratorsW(hInstance, MAKEINTRESOURCEW(1));
LoadStringW(hInstance, IDS_TITLE, g_szTitle, _countof(g_szTitle));
LoadStringW(hInstance, IDS_FONTNAME, g_szNameHead, _countof(g_szNameHead));
LoadStringW(hInstance, IDS_SUBSTITUTE, g_szSubstituteHead, _countof(g_szSubstituteHead));
WNDCLASSW wc = {0};
wc.style = 0;
wc.lpfnWndProc = WindowProc;
wc.hInstance = hInstance;
g_hIcon = LoadIcon(hInstance, MAKEINTRESOURCE(1));
wc.hIcon = g_hIcon;
wc.hCursor = LoadCursor(NULL, IDC_ARROW);
wc.hbrBackground = (HBRUSH)(COLOR_3DFACE + 1);
wc.lpszMenuName = MAKEINTRESOURCEW(1);
wc.lpszClassName = g_pszClassName;
if (!RegisterClassW(&wc))
{
MessageBoxA(NULL, "ERROR: RegisterClass failed.", NULL, MB_ICONERROR);
return 1;
}
const DWORD dwStyle = WS_OVERLAPPEDWINDOW;
INT Width = NAME_COLUMN_WIDTH + SUB_COLUMN_WIDTH +
GetSystemMetrics(SM_CXVSCROLL) +
GetSystemMetrics(SM_CXSIZEFRAME);
INT Height = 320;
RECT Rect = { 0, 0, Width, Height };
AdjustWindowRect(&Rect, dwStyle, TRUE);
Width = Rect.right - Rect.left;
Height = Rect.bottom - Rect.top;
g_hMainWnd = CreateWindowW(g_pszClassName, g_szTitle, dwStyle,
CW_USEDEFAULT, CW_USEDEFAULT, Width, Height,
NULL, NULL, hInstance, NULL);
if (g_hMainWnd == NULL)
{
MessageBoxA(NULL, "ERROR: CreateWindow failed.", NULL, MB_ICONERROR);
return 2;
}
ShowWindow(g_hMainWnd, nCmdShow);
UpdateWindow(g_hMainWnd);
MSG msg;
while (GetMessage(&msg, NULL, 0, 0))
{
if (TranslateAccelerator(g_hMainWnd, hAccel, &msg))
continue;
TranslateMessage(&msg);
DispatchMessage(&msg);
}
return (INT)msg.wParam;
}