LunaHook-mirror/LunaHost/GUI/QtLoader_inline.cpp

#include <Windows.h>
#include <thread>
#include <mutex>
#include <Shlwapi.h>
#include <filesystem>
#include <queue>
#include "pluginmanager.h"
#include "lockedqueue.hpp"
#ifndef _WIN64
#define THISCALL __thiscall
#define _CDECL __cdecl
#define fnQString_fromStdWString "?fromStdWString@QString@@SA?AV1@ABV?$basic_string@_WU?$char_traits@_W@std@@V?$allocator@_W@2@@std@@@Z"
#define fnQCoreApplication_addLibraryPath "?addLibraryPath@QCoreApplication@@SAXABVQString@@@Z"
#define fnQString_dtor "??1QString@@QAE@XZ"
#define fnQApplication_ctor "??0QApplication@@QAE@AAHPAPADH@Z"
#define fnQFont_ctor "??0QFont@@QAE@ABVQString@@HH_N@Z"
#define fnQApplication_setFont "?setFont@QApplication@@SAXABVQFont@@PBD@Z"
#define fnQFont_dtor "??1QFont@@QAE@XZ"
#define fnQApplication_exec "?exec@QApplication@@SAHXZ"
#define fnQApplication_dtor "??1QApplication@@UAE@XZ"
#else
#define THISCALL __fastcall
#define _CDECL __fastcall
#define fnQString_fromStdWString "?fromStdWString@QString@@SA?AV1@AEBV?$basic_string@_WU?$char_traits@_W@std@@V?$allocator@_W@2@@std@@@Z"
#define fnQCoreApplication_addLibraryPath "?addLibraryPath@QCoreApplication@@SAXAEBVQString@@@Z"
#define fnQString_dtor "??1QString@@QEAA@XZ"
#define fnQApplication_ctor "??0QApplication@@QEAA@AEAHPEAPEADH@Z"
#define fnQFont_ctor "??0QFont@@QEAA@AEBVQString@@HH_N@Z"
#define fnQApplication_setFont "?setFont@QApplication@@SAXAEBVQFont@@PEBD@Z"
#define fnQFont_dtor "??1QFont@@QEAA@XZ"
#define fnQApplication_exec "?exec@QApplication@@SAHXZ"
#define fnQApplication_dtor "??1QApplication@@UEAA@XZ"
#endif
#define fnQApplication_processEvents "?processEvents@QCoreApplication@@SAXV?$QFlags@W4ProcessEventsFlag@QEventLoop@@@@@Z"

FARPROC QString_fromStdWString, QCoreApplication_addLibraryPath, QString_dtor, QApplication_ctor, QFont_ctor, QFont_dtor, QApplication_setFont, QApplication_exec, QApplication_dtor, QApplication_processEvents;

bool checkqterror()
{
    return QString_fromStdWString == 0 || QCoreApplication_addLibraryPath == 0 || QString_dtor == 0 || QApplication_ctor == 0 || QFont_ctor == 0 || QFont_dtor == 0 || QApplication_setFont == 0 || QApplication_exec == 0 || QApplication_dtor == 0 || QApplication_processEvents == 0;
}
void loadqtdlls()
{
    QString_fromStdWString = QCoreApplication_addLibraryPath = QString_dtor = QApplication_ctor = QFont_ctor = QFont_dtor = QApplication_setFont = QApplication_exec = QApplication_dtor = QApplication_processEvents = 0;

    auto Qt5Widgets = LoadLibrary(L"Qt5Widgets.dll");
    auto Qt5Gui = LoadLibrary(L"Qt5Gui.dll");
    auto Qt5Core = LoadLibrary(L"Qt5Core.dll");
    if (Qt5Core == 0 || Qt5Gui == 0 || Qt5Widgets == 0)
        return;

    QString_fromStdWString = GetProcAddress(Qt5Core, fnQString_fromStdWString);
    QCoreApplication_addLibraryPath = GetProcAddress(Qt5Core, fnQCoreApplication_addLibraryPath);
    QString_dtor = GetProcAddress(Qt5Core, fnQString_dtor);
    QApplication_ctor = GetProcAddress(Qt5Widgets, fnQApplication_ctor);
    QFont_ctor = GetProcAddress(Qt5Gui, fnQFont_ctor);
    QFont_dtor = GetProcAddress(Qt5Gui, fnQFont_dtor);
    QApplication_setFont = GetProcAddress(Qt5Widgets, fnQApplication_setFont);
    QApplication_exec = GetProcAddress(Qt5Widgets, fnQApplication_exec);
    QApplication_dtor = GetProcAddress(Qt5Widgets, fnQApplication_dtor);
    QApplication_processEvents = GetProcAddress(Qt5Core, fnQApplication_processEvents);
}
struct info
{
    int type;
    std::wstring dll;
    HMODULE hdll;
};
lockedqueue<info> waitingtask;
lockedqueue<HMODULE> waitingresult;

extern "C" __declspec(dllexport) void QtStartUp(std::vector<std::wstring> *dlls)
{

    static bool once = false;
    if (once)
        return;
    loadqtdlls();
    once = !checkqterror();
    if (!once)
        return;
    std::thread([=]()
                {
                    static void *qapp; // 必须static
                    void *qstring;
                    void *qfont;
                    for (int i = 0; i < dlls->size(); i++)
                    {
                        auto dirname = std::filesystem::path(dlls->at(i)).parent_path().wstring();
                        ((void *(_CDECL *)(void *, void *))QString_fromStdWString)(&qstring, &dirname);
                        ((void(_CDECL *)(void *))QCoreApplication_addLibraryPath)(&qstring);
                        ((void(THISCALL *)(void *))QString_dtor)(&qstring);
                        // QCoreApplication_addLibraryPath(QString_fromStdWString(std::filesystem::path(collectQtplugs[i]).parent_path()));
                    }

                    int _ = 0;
                    ((void *(THISCALL *)(void *, int *, char **, int))QApplication_ctor)(&qapp, &_, 0, 331266);

                    std::wstring font = L"MS Shell Dlg 2";
                    ((void *(_CDECL *)(void *, void *))QString_fromStdWString)(&qstring, &font);
                    ((void *(THISCALL *)(void *, void *, int, int, bool))QFont_ctor)(&qfont, &qstring, 10, -1, 0);
                    ((void(_CDECL *)(void *, void *))QApplication_setFont)(&qfont, 0);
                    ((void(THISCALL *)(void *))QFont_dtor)(&qfont);
                    ((void(THISCALL *)(void *))QString_dtor)(&qstring);

                    while (true)
                    {
                        if (!waitingtask.empty())
                        {
                            auto top = waitingtask.pop();
                            if (top.type == 1)
                            {
                                waitingresult.push(LoadLibraryW(top.dll.c_str()));
                            }
                            else if (top.type == 2)
                            {
                                FreeLibrary(top.hdll);
                            }
                        }
                        ((void(_CDECL *)(DWORD))QApplication_processEvents)(0);
                        Sleep(1);
                    }

                    // ((void(*)())QApplication_exec)();

                    // ((void(THISCALL*)(void*))QApplication_dtor)(&qapp);
                })
        .detach();
}
std::mutex loadmutex;

extern "C" __declspec(dllexport) std::vector<HMODULE> *QtLoadLibraryBatch(std::vector<std::wstring> *dlls)
{
    std::lock_guard _(loadmutex);
    QtStartUp(dlls);
    auto hdlls = new std::vector<HMODULE>;
    for (int i = 0; i < dlls->size(); i++)
    {
        if (checkqterror())
        {
            hdlls->push_back(0);
        }
        else
        {
            waitingtask.push({1, dlls->at(i)});
            hdlls->push_back(waitingresult.pop());
        }
    }
    return hdlls;
}

extern "C" __declspec(dllexport) void QtFreeLibrary(HMODULE hd)
{
    std::lock_guard _(loadmutex);
    waitingtask.push({2, L"", hd});
}