Unicode与MultiByte之间的练习: Process Cleaner

#include <cstdio>
#include <cstring>
#include <windows.h>
#include <tlhelp32.h>

using namespace std;

const int MAX = 15;

WCHAR SysProcList[MAX][MAX_PATH] = {
    L"[system process]",
    L"system",
    L"smss.exe",
    L"csrss.exe",
    L"wininit.exe",
    L"winlogon.exe",
    L"services.exe",
    L"lsass.exe",
    L"lsm.exe",
    L"svchost.exe",
    L"audiodg.exe",
    L"explorer.exe",
    L"conhost.exe",
    L"dllhost.exe",
    L"ctfmon.exe"
};

bool CheckSysProcList(LPWSTR lpText);
void PrintfNormal(LPWSTR lpText);
void PrintfSucceed(LPWSTR lpText);
void PrintfError(LPWSTR lpText);

int main()
{
    HANDLE hToken;
    OpenProcessToken(GetCurrentProcess(), TOKEN_ALL_ACCESS, &hToken);
    
    LUID luid = {};
    LookupPrivilegeValue(NULL, L"SeDebugPrivilege", &luid);

    TOKEN_PRIVILEGES tp;
    tp.PrivilegeCount = 1;
    tp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
    tp.Privileges[0].Luid = luid;
    if (!AdjustTokenPrivileges(hToken, FALSE, &tp, sizeof(tp), NULL, NULL))
    {
        PrintfError(L"ERROR_01!\n");
        Sleep(2000);
        return 0;
    }

    HANDLE hSnapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, NULL);
    if (INVALID_HANDLE_VALUE == hSnapshot)
    {
        PrintfError(L"ERROR_02!\n");
        Sleep(2000);
        return 0;
    }

    PROCESSENTRY32 pe32 = {};
    pe32.dwSize = sizeof(pe32);
    
    BOOL ret = Process32First(hSnapshot, &pe32);
    while (ret)
    {
        Sleep(100);

        bool isSysProc = (pe32.th32ProcessID == GetCurrentProcessId()) || (CheckSysProcList(wcslwr(pe32.szExeFile)));

        WCHAR wstr[MAX_PATH];
        wsprintf(wstr, L"%d %s %s", pe32.th32ProcessID, wcslwr(pe32.szExeFile), isSysProc ? L"ignored\n" : L"");
        PrintfNormal(wstr);

        if (!isSysProc)
        {
            BOOL tpret = TerminateProcess(OpenProcess(PROCESS_ALL_ACCESS, FALSE, pe32.th32ProcessID), NULL);
            wsprintf(wstr, L"shutdown %s\n", tpret ? L"succeed" : L"faild");

            if (tpret)
                PrintfSucceed(wstr);
            else
                PrintfError(wstr);
        }

        ret = Process32Next(hSnapshot, &pe32);
    }

    PrintfNormal(L"\nProcess Clean Finish...\n");
    Sleep(2000);
    return 0;
}


bool CheckSysProcList(LPWSTR lpText)
{
    for (int i = 0; i < MAX; i++)
        if (0 == wcscmp(lpText, SysProcList[i]))
        return true;

    return false;
}

void PrintfNormal(LPWSTR lpText)
{
    SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), FOREGROUND_BLUE | FOREGROUND_GREEN | FOREGROUND_RED | FOREGROUND_INTENSITY);
    printf("%S", lpText);
}

void PrintfSucceed(LPWSTR lpText)
{
    SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), FOREGROUND_GREEN | FOREGROUND_INTENSITY);
    printf("%S", lpText);
}

void PrintfError(LPWSTR lpText)
{
    SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), FOREGROUND_RED | FOREGROUND_INTENSITY);
    printf("%S", lpText);
}