native addon

2025-12-17 03:48:16 +08:00 · 2022-04-04 17:08:47 +08:00
parent 945a94222a
commit be5a457639
9 changed files with 223 additions and 0 deletions
--- a/native/utils.cc
+++ b/native/utils.cc
@@ -0,0 +1,83 @@
+#include "utils.h"
+#include "define.h"
+
+wstring StringToWString(const string &src) {
+    wstring result;
+    int len = MultiByteToWideChar(CP_ACP, 0, src.c_str(), src.size(), nullptr, 0); // NOLINT(cppcoreguidelines-narrowing-conversions)
+    auto *buffer = new WCHAR[len + 1];
+    MultiByteToWideChar(CP_ACP, 0, src.c_str(), src.size(), buffer, len); // NOLINT(cppcoreguidelines-narrowing-conversions)
+    buffer[len] = '\0';
+    result.append(buffer);
+    delete[] buffer;
+    return result;
+}
+
+string WStringToString(const wstring &src) {
+    string result;
+    int len = WideCharToMultiByte(CP_ACP, 0, src.c_str(), src.size(), nullptr, 0, nullptr, nullptr); // NOLINT(cppcoreguidelines-narrowing-conversions)
+    char *buffer = new char[len + 1];
+    WideCharToMultiByte(CP_ACP, 0, src.c_str(), src.size(), buffer, len, nullptr, nullptr); // NOLINT(cppcoreguidelines-narrowing-conversions)
+    buffer[len] = '\0';
+    result.append(buffer);
+    delete[] buffer;
+    return result;
+}
+
+LSTATUS OpenFile(Env env, Value &result, HWND parent) {
+    OPENFILENAME open;
+    ZeroMemory(&open, sizeof(open));
+    WCHAR file[32768];
+    file[0]=L'\0';
+    open.Flags       = OFN_FILEMUSTEXIST | OFN_NONETWORKBUTTON | OFN_PATHMUSTEXIST | OFN_EXPLORER;
+    open.hwndOwner   = parent;
+    open.nMaxFile    = 32768;
+    open.lpstrFile   = file;
+    open.lpstrTitle  = L"选择原神主程序";
+    open.lpstrFilter = L"国服/国际服主程序 (YuanShen/GenshinImpact.exe)\0YuanShen.exe;GenshinImpact.exe\0";
+    open.lStructSize = sizeof(open);
+    if(GetOpenFileName(&open)) {
+        string s = WStringToString(file);
+        if (CreateUTF8String(env, s.c_str(), NAPI_AUTO_LENGTH, &result) == napi_ok) {
+            return ERROR_SUCCESS;
+        } else {
+            return ERROR_ERRORS_ENCOUNTERED;
+        }
+    } else {
+        return (LSTATUS)CommDlgExtendedError();
+    }
+}
+
+Status RegisterFunction(Env env, Value exports, Callback cb, const string &name) {
+    Value fn;
+    Status status = CreateFunction(env, nullptr, 0, cb, nullptr, &fn);
+    if (status != napi_ok) return status;
+    status = SetNamedProperty(env, exports, name.c_str(), fn);
+    if (status != napi_ok) return status;
+    return napi_ok;
+}
+
+BOOL EnablePrivilege(const wstring &name) {
+    HANDLE hToken;
+    if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES, &hToken)) {
+        cout << "OpenProcessToken error: %lu\n" << GetLastError() << endl;
+        return FALSE;
+    }
+    TOKEN_PRIVILEGES tp;
+    ZeroMemory(&tp, sizeof(tp));
+    tp.PrivilegeCount = 1;
+    tp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
+    if (!LookupPrivilegeValue(nullptr, name.c_str(), &tp.Privileges[0].Luid)) {
+        cout << "LookupPrivilegeValue error: %lu\n" << GetLastError() << endl;
+        return FALSE;
+    }
+    if (!AdjustTokenPrivileges(hToken, FALSE, &tp, sizeof(tp), nullptr, nullptr)) {
+        cout << "AdjustTokenPrivileges error: %lu\n" << GetLastError() << endl;
+        return FALSE;
+    }
+    if (GetLastError() == ERROR_NOT_ALL_ASSIGNED) {
+        cout <<"The token does not have the specified privilege." << endl;
+        return FALSE;
+    }
+    CloseHandle(hToken);
+    return TRUE;
+}