/*gf25051小黄原创代码专属水印*/
#define WIN32_LEAN_AND_MEAN
#include <windows.h>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <shellapi.h>
#include <tlhelp32.h>
#include <ctime>
using namespace std;

#ifndef KEY_WOW64_64KEY
#define KEY_WOW64_64KEY 0x0100
#endif

const char* STUDENT_MAIN_PATH = "C:\\Program Files (x86)\\Mythware\\极域课堂管理系统软件V6.0 2016 豪华版\\StudentMain.exe";
const char* FROZEN_WND_CLASS = "ClassScreenFrozenWnd";

#ifdef __MINGW32__
#pragma comment(linker, "-ladvapi32 -lsetupapi -lgdi32 -luser32 -lshell32 -lole32 -lkernel32 -lpsapi")
#endif

int killCount = 0;
bool keyStates[256];
bool isScreenFrozen = false;
HWND g_hFrozenWnd = NULL;
HINSTANCE g_hInstance = NULL;
HBITMAP g_hScreenBmp = NULL;
int g_nScreenW = 0;
int g_nScreenH = 0;
HANDLE g_hCaptureThread = NULL;
bool g_bCapturing = false;
CRITICAL_SECTION g_csGlobal;

LRESULT CALLBACK FrozenWndProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam) {
    switch (uMsg) {
        case WM_PAINT: {
            PAINTSTRUCT ps;
            HDC hdc = BeginPaint(hWnd, &ps);
            EnterCriticalSection(&g_csGlobal);
            if (g_hScreenBmp != NULL) {
                HDC hMemDC = CreateCompatibleDC(hdc);
                HBITMAP hOldBmp = (HBITMAP)SelectObject(hMemDC, g_hScreenBmp);
                BitBlt(hdc, 0, 0, g_nScreenW, g_nScreenH, hMemDC, 0, 0, SRCCOPY);
                SelectObject(hMemDC, hOldBmp);
                DeleteDC(hMemDC);
            }
            LeaveCriticalSection(&g_csGlobal);
            EndPaint(hWnd, &ps);
            return 0;
        }
        case WM_ERASEBKGND: return 1;
        case WM_DESTROY: case WM_CLOSE: return 0;
        default: return DefWindowProcA(hWnd, uMsg, wParam, lParam);
    }
}

void UnlockUSBStorage() {
    HKEY hKey;
    DWORD dwValue = 0;

    system("taskkill /f /im StudentMain.exe > nul 2>&1");
    system("taskkill /f /im StudentService_Proxy.exe > nul 2>&1");
    system("taskkill /f /im JYService.exe > nul 2>&1");
    system("taskkill /f /im DispcapHelper.exe > nul 2>&1");
    system("taskkill /f /im MasterHelper.exe > nul 2>&1");
    system("taskkill /f /im ProcHelper64.exe > nul 2>&1");

    system("sc stop usbfilter > nul 2>&1");
    system("sc delete usbfilter > nul 2>&1");
    system("sc stop MythwareFilter > nul 2>&1");
    system("sc delete MythwareFilter > nul 2>&1");

    if (RegOpenKeyExA(HKEY_LOCAL_MACHINE, "SYSTEM\\CurrentControlSet\\Services\\USBSTOR",
        0, KEY_SET_VALUE | KEY_WOW64_64KEY, &hKey) == ERROR_SUCCESS) {
        dwValue = 3;
        RegSetValueExA(hKey, "Start", 0, REG_DWORD, (LPBYTE)&dwValue, sizeof(DWORD));
        RegCloseKey(hKey);
    }
    system("sc config USBSTOR start= demand > nul 2>&1");
    system("sc start USBSTOR > nul 2>&1");

    if (RegOpenKeyExA(HKEY_LOCAL_MACHINE, "SYSTEM\\CurrentControlSet\\Control\\StorageDevicePolicies",
        0, KEY_SET_VALUE | KEY_WOW64_64KEY, &hKey) == ERROR_SUCCESS) {
        dwValue = 0;
        RegSetValueExA(hKey, "WriteProtect", 0, REG_DWORD, (LPBYTE)&dwValue, sizeof(DWORD));
        RegCloseKey(hKey);
    }

    if (RegOpenKeyExA(HKEY_LOCAL_MACHINE, "SYSTEM\\CurrentControlSet\\Policies\\Microsoft\\Windows\\RemovableStorageDevices",
        0, KEY_ALL_ACCESS | KEY_WOW64_64KEY, &hKey) == ERROR_SUCCESS) {
        RegDeleteKeyA(hKey, "");
        RegCloseKey(hKey);
    }
    if (RegOpenKeyExA(HKEY_CURRENT_USER, "Software\\Policies\\Microsoft\\Windows\\RemovableStorageDevices",
        0, KEY_ALL_ACCESS | KEY_WOW64_64KEY, &hKey) == ERROR_SUCCESS) {
        RegDeleteKeyA(hKey, "");
        RegCloseKey(hKey);
    }

    if (RegCreateKeyExA(HKEY_LOCAL_MACHINE, "SOFTWARE\\Policies\\Microsoft\\Windows\\RemovableStorageAccess",
        0, NULL, REG_OPTION_NON_VOLATILE, KEY_SET_VALUE | KEY_WOW64_64KEY, NULL, &hKey, NULL) == ERROR_SUCCESS) {
        dwValue = 0;
        RegSetValueExA(hKey, "Deny_All", 0, REG_DWORD, (LPBYTE)&dwValue, sizeof(DWORD));
        RegSetValueExA(hKey, "Deny_USB", 0, REG_DWORD, (LPBYTE)&dwValue, sizeof(DWORD));
        RegCloseKey(hKey);
    }

    if (RegOpenKeyExA(HKEY_CURRENT_USER, "Software\\Microsoft\\Windows\\CurrentVersion\\Policies\\Explorer",
        0, KEY_SET_VALUE | KEY_WOW64_64KEY, &hKey) == ERROR_SUCCESS) {
        RegDeleteValueA(hKey, "NoDrives");
        RegDeleteValueA(hKey, "NoViewOnDrive");
        RegCloseKey(hKey);
    }
    if (RegOpenKeyExA(HKEY_LOCAL_MACHINE, "Software\\Microsoft\\Windows\\CurrentVersion\\Policies\\Explorer",
        0, KEY_SET_VALUE | KEY_WOW64_64KEY, &hKey) == ERROR_SUCCESS) {
        RegDeleteValueA(hKey, "NoDrives");
        RegDeleteValueA(hKey, "NoViewOnDrive");
        RegCloseKey(hKey);
    }

    if (RegOpenKeyExA(HKEY_CURRENT_USER, "Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\MountPoints2",
        0, KEY_ALL_ACCESS | KEY_WOW64_64KEY, &hKey) == ERROR_SUCCESS) {
        RegDeleteKeyA(hKey, "");
        RegCloseKey(hKey);
    }

    system("sc stop PlugPlay > nul 2>&1");
    Sleep(300);
    system("sc start PlugPlay > nul 2>&1");
    Sleep(800);
    system("devcon remove *USBSTOR* > nul 2>&1");
    Sleep(1000);
    system("devcon rescan > nul 2>&1");

    system("taskkill /f /im explorer.exe > nul 2>&1");
    Sleep(500);
    system("start explorer.exe > nul 2>&1");

    killCount++;
    cout << "\n===== 极域V6.0 2016豪华版 U盘解禁完成 =====" << endl;
    cout << "【必做】拔出U盘 → 等待3秒 → 重新插入！" << endl;
}

bool IsRunAsAdmin() {
    BOOL bIsAdmin = FALSE;
    SID_IDENTIFIER_AUTHORITY NtAuthority = SECURITY_NT_AUTHORITY;
    PSID pAdministratorSid = NULL;
    if (AllocateAndInitializeSid(&NtAuthority, 2,
        SECURITY_BUILTIN_DOMAIN_RID, DOMAIN_ALIAS_RID_ADMINS,
        0, 0, 0, 0, 0, 0, &pAdministratorSid)) {
        CheckTokenMembership(NULL, pAdministratorSid, &bIsAdmin);
        FreeSid(pAdministratorSid);
    }
    return bIsAdmin != FALSE;
}

void RequestAdminPrivileges() {
    char szPath[MAX_PATH] = {0};
    if (GetModuleFileNameA(NULL, szPath, MAX_PATH)) {
        HINSTANCE hResult = ShellExecuteA(NULL, "runas", szPath, NULL, NULL, SW_SHOWNORMAL);
        if ((INT_PTR)hResult <= 32) {
            DWORD err = GetLastError();
            if (err == ERROR_CANCELLED) {
                cout << "用户拒绝管理员权限，部分功能失效！" << endl;
                system("pause >nul");
            }
        } else {
            exit(0);
        }
    }
}

void SafeShowConsole(bool show) {
    HWND hwnd = GetConsoleWindow();
    if (hwnd != NULL) ShowWindow(hwnd, show ? SW_SHOW : SW_HIDE);
}

bool SystemCaptureScreenToMem() {
    EnterCriticalSection(&g_csGlobal);
    if (g_hScreenBmp != NULL) {
        DeleteObject(g_hScreenBmp);
        g_hScreenBmp = NULL;
    }
    LeaveCriticalSection(&g_csGlobal);

    HDC hScreenDC = GetDC(NULL);
    if (hScreenDC == NULL) {
        cout << "错误：获取屏幕DC失败！" << endl;
        return false;
    }
    
    g_nScreenW = GetDeviceCaps(hScreenDC, HORZRES);
    g_nScreenH = GetDeviceCaps(hScreenDC, VERTRES);
    
    HDC hMemDC = CreateCompatibleDC(hScreenDC);
    if (hMemDC == NULL) {
        ReleaseDC(NULL, hScreenDC);
        cout << "错误：创建内存DC失败！" << endl;
        return false;
    }
    
    BITMAPINFO bmi = {0};
    bmi.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
    bmi.bmiHeader.biWidth = g_nScreenW;
    bmi.bmiHeader.biHeight = -g_nScreenH;
    bmi.bmiHeader.biPlanes = 1;
    bmi.bmiHeader.biBitCount = 32;
    bmi.bmiHeader.biCompression = BI_RGB;
    
    void* pBits = NULL;
    HBITMAP hBmp = CreateDIBSection(hScreenDC, &bmi, DIB_RGB_COLORS, &pBits, NULL, 0);
    if (hBmp == NULL) {
        DeleteDC(hMemDC);
        ReleaseDC(NULL, hScreenDC);
        cout << "错误：创建DIB位图失败！" << endl;
        return false;
    }
    
    HBITMAP hOldBmp = (HBITMAP)SelectObject(hMemDC, hBmp);
    BitBlt(hMemDC, 0, 0, g_nScreenW, g_nScreenH, hScreenDC, 0, 0, SRCCOPY);
    SelectObject(hMemDC, hOldBmp);
    DeleteDC(hMemDC);
    ReleaseDC(NULL, hScreenDC);
    
    EnterCriticalSection(&g_csGlobal);
    g_hScreenBmp = hBmp;
    LeaveCriticalSection(&g_csGlobal);
    
    cout << "截屏成功！分辨率：" << g_nScreenW << "x" << g_nScreenH << endl;
    return true;
}

DWORD WINAPI AsyncCaptureScreen(LPVOID lpParam) {
    g_bCapturing = true;
    for (int i = 3; i > 0 && g_bCapturing; i--) {
        cout << "\r 即将截屏定格，倒计时：" << i << "秒" << flush;
        Sleep(1000);
    }
    
    if (g_bCapturing) {
        cout << "\r 开始截屏！" << endl;
        if (SystemCaptureScreenToMem()) {
            EnterCriticalSection(&g_csGlobal);
            WNDCLASSEXA wc = {0};
            wc.cbSize = sizeof(WNDCLASSEXA);
            wc.lpfnWndProc = FrozenWndProc;
            wc.hInstance = g_hInstance;
            wc.lpszClassName = FROZEN_WND_CLASS;
            wc.style = CS_HREDRAW | CS_VREDRAW;
            RegisterClassExA(&wc);
            
            g_hFrozenWnd = CreateWindowExA(
                WS_EX_TOPMOST | WS_EX_NOACTIVATE,
                FROZEN_WND_CLASS,
                "ScreenFrozen",
                WS_POPUP | WS_VISIBLE,
                0, 0, g_nScreenW, g_nScreenH,
                NULL, NULL, g_hInstance, NULL
            );
            LeaveCriticalSection(&g_csGlobal);
            
            if (g_hFrozenWnd != NULL) {
                isScreenFrozen = true;
                cout << "成功！画面已定格，按R键恢复！" << endl;
            } else {
                cout << "错误：定格窗口创建失败！" << endl;
                EnterCriticalSection(&g_csGlobal);
                if (g_hScreenBmp != NULL) DeleteObject(g_hScreenBmp);
                g_hScreenBmp = NULL;
                LeaveCriticalSection(&g_csGlobal);
            }
        } else {
            cout << "错误：截屏失败！" << endl;
        }
    }
    g_bCapturing = false;
    g_hCaptureThread = NULL;
    return 0;
}

void FreezeTeacherScreen() {
    EnterCriticalSection(&g_csGlobal);
    if (isScreenFrozen) {
        LeaveCriticalSection(&g_csGlobal);
        cout << "提示：画面已定格！" << endl;
        return;
    }
    LeaveCriticalSection(&g_csGlobal);
    
    if (g_bCapturing) {
        cout << "提示：截屏中，请稍候！" << endl;
        return;
    }
    
    g_hCaptureThread = CreateThread(NULL, 0, AsyncCaptureScreen, NULL, 0, NULL);
    if (g_hCaptureThread == NULL) cout << "错误：创建截屏线程失败！" << endl;
    else CloseHandle(g_hCaptureThread);
}

void RestoreTeacherScreen() {
    EnterCriticalSection(&g_csGlobal);
    if (g_bCapturing) {
        g_bCapturing = false;
        LeaveCriticalSection(&g_csGlobal);
        cout << "截屏已取消！" << endl;
        return;
    }
    
    if (!isScreenFrozen) {
        LeaveCriticalSection(&g_csGlobal);
        cout << "提示：画面未定格！" << endl;
        return;
    }
    
    if (g_hFrozenWnd != NULL) DestroyWindow(g_hFrozenWnd);
    if (g_hScreenBmp != NULL) DeleteObject(g_hScreenBmp);
    g_hFrozenWnd = NULL;
    g_hScreenBmp = NULL;
    g_nScreenW = 0;
    g_nScreenH = 0;
    isScreenFrozen = false;
    LeaveCriticalSection(&g_csGlobal);
    cout << "成功！画面已恢复！" << endl;
}

void StartTaskManager() {
    system("taskmgr");
    cout << "任务管理器已启动！" << endl;
}

void show() {
    cout << "=========================================================================================================" << endl;
    cout << "后台全局监听中，按以下按键执行操作：\n\n";
    cout << "B: 结束极域（仅控屏） | C: 结束极域（全部+禁用服务） | X: 恢复极域(已修复) | T: 解除U盘禁用"<<endl << endl;
    cout << "1-9: 打开对应网页/程序 | 按ESC退出程序 | 按E定格教师端此电脑的屏幕" <<endl<< endl;
    cout<<"1.打开bcoi                    7.gf25051小黄主页"<<endl<<endl;
    cout<<"2.显示当前系统时间            8.gf25152鼠鼠主页"<<endl<<endl;
    cout<<"3.我的世界                    9.游戏总结"<<endl<<endl;
    cout<<"4.抡棒作战                    Q.删掉极域!!!!!!!!!!! （需要管理员权限）"<<endl<<endl;
    cout<<"5.Slowroads                   P:后台运行杀手（后台运行更隐蔽） | 再次按P:恢复控制台显示"<<endl<<endl;
    cout<<"6.打开CMD命令提示符           ESC:退出杀手"<<endl<<endl;
	cout<<"D:任务管理器                  H:显示对应网页/程序"<<endl;
    cout << "=========================================================================================================" << endl;
}

void StartStudentMain() {
    ShellExecuteA(NULL, "open", STUDENT_MAIN_PATH, NULL, NULL, SW_SHOW);
    cout << "正在启动极域客户端..." << endl;
}

void OpenWebPage(const char* url) {
    ShellExecuteA(NULL, "open", url, NULL, NULL, SW_SHOWNORMAL);
}

void checkGlobalKeys() {
    for (int i = 0; i < 256; i++) {
        if (GetAsyncKeyState(i) & 0x8000) {
            if (!keyStates[i]) {
                keyStates[i] = true;
                switch (i) {
                    case 'E': FreezeTeacherScreen(); break;
                    case 'R': RestoreTeacherScreen(); break;
                    case 'B': 
                        system("taskkill /F /IM StudentMain.exe");
                        killCount++;
                        cout << "B键操作完成！" << endl;
                        break;
                    case 'C': 
                        cout <<endl<< "正在终止所有极域相关进程并禁用服务..." << endl;
                        system("sc stop JYService > nul 2>&1");
                        system("sc config JYService start= disabled > nul 2>&1");
                        system("taskkill /im MythwareService.exe /f > nul 2>&1");
                        system("taskkill /im StudentService_Proxy.exe /f > nul 2>&1");
                        system("taskkill /im StudentMain.exe /f > nul 2>&1");
                        system("taskkill /im XStudent.exe /f > nul 2>&1");
                        system("taskkill /im XTRudent.exe /f > nul 2>&1");
                        system("taskkill /im GATESRV.exe /f > nul 2>&1");
                        system("taskkill /im DispcapHelper.exe /f > nul 2>&1");
                        system("taskkill /im MasterHelper.exe /f > nul 2>&1");
                        system("taskkill /im ProcHelper64.exe /f > nul 2>&1");
                        system("taskkill /im Student.exe /f > nul 2>&1");
                        system("taskkill /im JYService.exe /f > nul 2>&1");
                        system("taskkill /im jyclass.exe /f > nul 2>&1");
                        killCount++;
                        cout <<endl<< "C键操作完成！（进程终止+服务禁用）" << endl;
                        break;
                    case 'Q': 
                        system("sc stop JYService > nul 2>&1 && sc delete JYService > nul 2>&1");
                        system("taskkill /f /im MythwareService.exe /im StudentMain.exe > nul 2>&1");
                        system("rmdir /s /q \"C:\\Program Files (x86)\\Mythware\" > nul 2>&1");
                        cout << "Q键操作完成！" << endl;
                        break;
                    case 'D': StartTaskManager(); break;
                    case 'H': show(); break;
                    case 'P': {
                        static bool consoleVisible = true;
                        consoleVisible = !consoleVisible;
                        SafeShowConsole(consoleVisible);
                    } break;
                    case 'X': StartStudentMain(); break;
                    case 'T':
                        cout << "正在执行U盘解禁..." << endl;
                        UnlockUSBStorage(); 
                        break;
                    case '2': {
                        time_t tNow = time(NULL);
                        tm* pLocalTm = localtime(&tNow);
                        char szTime[64] = {0};
                        sprintf(szTime, "%04d-%02d-%02d %02d:%02d:%02d",
                                pLocalTm->tm_year + 1900, pLocalTm->tm_mon + 1, pLocalTm->tm_mday,
                                pLocalTm->tm_hour, pLocalTm->tm_min, pLocalTm->tm_sec);
                        cout << "当前时间：" << szTime << endl;
                        break;
                    }
                    case '6': 
                        ShellExecuteA(NULL, "open", "cmd.exe", NULL, NULL, SW_SHOW);
                        cout << "已打开CMD！" << endl;
                        break;
                    case '1': OpenWebPage("http://www.bcoi.com"); cout << "打开bcoi！" << endl; break;
                    case '3': OpenWebPage("https://play.mc.js.cool/1.8/"); cout << "打开我的世界！" << endl; break;
                    case '4': OpenWebPage("https://www.dogod.io"); cout << "打开抡棒作战！" << endl; break;
                    case '5': OpenWebPage("https://slowroads.io/"); cout << "打开Slowroads！" << endl; break;
                    case '7': OpenWebPage("https://www.bcoi.cn/d/gfhd/user/507"); cout << "打开小黄主页！" << endl; break;
                    case '8': OpenWebPage("https://www.bcoi.cn/d/gfhd/user/616"); cout << "打开鼠鼠主页！" << endl; break;
                    case '9': OpenWebPage("https://xg9.cn/my/4qVWxG"); cout << "打开游戏总结！" << endl; break;
                    default: break;
                }
            }
        } else {
            keyStates[i] = false;
        }
    }
}

int main() {
    InitializeCriticalSection(&g_csGlobal);
    g_hInstance = GetModuleHandleA(NULL);
    memset(keyStates, 0, sizeof(keyStates));

    bool isAdmin = IsRunAsAdmin();
    if (!isAdmin) {
        cout << "检测到非管理员权限，正在请求..." << endl;
        Sleep(1000);
        RequestAdminPrivileges();
        system("cls");
    }

    system("title 极域V6.0 2016豪华版解禁工具v14.0Pro（C++98版）");
    system("chcp 936 > nul");
    system("cls");

    cout << "极域V6.0 2016豪华版解禁工具v14.0Pro（适配C++98/Dev-C++5.11）" << endl<<endl;
    cout << (isAdmin ? "[+] 管理员权限（功能全开）" : "[-] 普通权限（功能受限）") << endl<<endl;
    show();

    MSG msg;
    bool running = true;
    while (running) {
        while (PeekMessageA(&msg, NULL, 0, 0, PM_REMOVE)) {
            if (msg.message == WM_QUIT) running = false;
            TranslateMessage(&msg);
            DispatchMessage(&msg);
        }
        checkGlobalKeys();
        if (GetAsyncKeyState(VK_ESCAPE) & 0x8000) running = false;
        Sleep(10);
    }

    RestoreTeacherScreen();
    UnregisterClassA(FROZEN_WND_CLASS, g_hInstance);
    DeleteCriticalSection(&g_csGlobal);

    cout << "\n程序即将退出，共肘击极域 " << killCount << " 次！" << endl;
    system("pause > nul");
    return 0;
}
/*gf25051小黄原创代码专属水印*/