pull

Author: nahida-mouo

.github/workflows/build.yml

@@ -78,14 +78,3 @@ jobs:
         path: src/Snap.Hutao.Remastered.Native/x64/${{ env.BUILD_CONFIGURATION }}/*.dll
         retention-days: 30
         if-no-files-found: error
-
-    - name: Upload Build logs
-      if: always()
-      uses: actions/upload-artifact@v4
-      with:
-        name: build-logs
-        path: |
-          **/*.log
-          **/bin/**/*.log
-          **/obj/**/*.log
-        retention-days: 7

src/Snap.Hutao.Remastered.Native/CustomImplements.h

@@ -47,7 +47,7 @@ class CustomImplements : public Interfaces... {
 
 public:
     // IUnknown methods
-    HRESULT STDMETHODCALLTYPE QueryInterface(REFIID riid, void** ppvObject) override {
+    HRESULT __stdcall QueryInterface(REFIID riid, void** ppvObject) override {
         if (ppvObject == nullptr) {
             return E_POINTER;
         }
@@ -72,11 +72,11 @@ class CustomImplements : public Interfaces... {
         return E_NOINTERFACE;
     }
 
-    ULONG STDMETHODCALLTYPE AddRef() override {
+    ULONG __stdcall AddRef() override {
         return InterlockedIncrement(&m_refCount);
     }
 
-    ULONG STDMETHODCALLTYPE Release() override {
+    ULONG __stdcall Release() override {
         ULONG refCount = InterlockedDecrement(&m_refCount);
         if (refCount == 0) {
             delete static_cast<Derived*>(this);

src/Snap.Hutao.Remastered.Native/DllInjectionUtils.cpp

@@ -1,13 +1,10 @@
+#include "pch.h"
 #include "DllInjectionUtils.h"
 #include "ProcessUtils.h"
 #include "PrivilegeUtils.h"
 #include "HookUtils.h"
 #include "MemoryUtils.h"
 #include "StringUtils.h"
-#include "Error.h"
-#include <Windows.h>
-#include <vector>
-#include <string>
 #include <tlhelp32.h>
 
 // Windows钩子注入方法1（使用WH_GETMESSAGE钩子）

src/Snap.Hutao.Remastered.Native/Error.cpp

@@ -1,3 +1,4 @@
+#include "pch.h"
 #include "Error.h"
 #include "WilCallbacksManager.h"
 #include "FailureInfo.h"

src/Snap.Hutao.Remastered.Native/FirewallRuleManager.cpp

@@ -1,12 +1,8 @@
-#pragma once
-
+#include "pch.h"
 #include "FirewallRuleManager.h"
 #include <sddl.h>
-#include <vector>
 #include <icftypes.h>
 #include <netfw.h>
-#include <Windows.h>
-#include <string>
 
 FirewallRuleManager::FirewallRuleManager()
     : m_policy(nullptr)
@@ -87,21 +83,21 @@ HRESULT FirewallRuleManager::AddLoopbackExempt(const std::wstring& familyName, c
 
     std::wstring ruleName = GetRuleName(familyName, sid);
 
-    // 检查规则是否已存在
+    // 检查规则是否已存在
     INetFwRule* existingRule = nullptr;
     hr = m_rules->Item(BSTR(ruleName.c_str()), &existingRule);
 
     if (SUCCEEDED(hr) && existingRule)
     {
-        // 规则已存在，启用它
+        // 规则已存在，启用它
         hr = existingRule->put_Enabled(VARIANT_TRUE);
         SafeRelease(existingRule);
         return hr;
     }
 
     SafeRelease(existingRule);
 
-    // 创建新规则
+    // 创建新规则
     return CreateFirewallRule(
         ruleName,
         L"loopback",
@@ -152,7 +148,7 @@ HRESULT FirewallRuleManager::CreateFirewallRule(const std::wstring& ruleName, co
     if (FAILED(hr))
         return hr;
 
-    // 设置规则属性
+    // 设置规则属性
     hr = rule->put_Name(BSTR(ruleName.c_str()));
     if (FAILED(hr)) goto cleanup;
 
@@ -186,7 +182,7 @@ HRESULT FirewallRuleManager::CreateFirewallRule(const std::wstring& ruleName, co
     }
     rule->put_Grouping(BSTR(groupName.c_str()));
 
-    // 将规则添加到集合中
+    // 将规则添加到集合中
     hr = m_rules->Add(rule);
 
 cleanup:
@@ -214,7 +210,7 @@ std::wstring FirewallRuleManager::GetCurrentUserSid()
     if (!OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &hToken))
         return sidString;
 
-    // 获取所需缓冲区大小
+    // 获取所需缓冲区大小
     DWORD tokenSize = 0;
     GetTokenInformation(hToken, TokenUser, nullptr, 0, &tokenSize);
 
@@ -236,4 +232,4 @@ std::wstring FirewallRuleManager::GetCurrentUserSid()
 
     CloseHandle(hToken);
     return sidString;
-}
+}

src/Snap.Hutao.Remastered.Native/FrameworkTheming.cpp

@@ -1,8 +1,5 @@
+#include "pch.h"
 #include "FrameworkTheming.h"
-#include "types.h"
-#include "Error.h"
-#include <Windows.h>
-#include <cstring>
 #include <atomic>
 
 // Global function pointers

src/Snap.Hutao.Remastered.Native/HookUtils.cpp

@@ -1,10 +1,8 @@
-#pragma once
-
+#include "pch.h"
 #include "HookUtils.h"
 #include "StringUtils.h"
-#include <Windows.h>
 
-// 使用Windows钩子注入DLL
+// 使用Windows钩子注入DLL
 HRESULT InjectUsingHook(LPCWSTR dllPath, LPCWSTR functionName, DWORD threadId, int hookType)
 {
     if (!dllPath || !functionName) {
@@ -25,14 +23,14 @@ HRESULT InjectUsingHook(LPCWSTR dllPath, LPCWSTR functionName, DWORD threadId, i
     HOOKPROC hookProc = NULL;
 
     if ((DWORD_PTR)ansiFunctionName <= 0xFFFF) {
-        // 是序号
+        // 是序号
         hookProc = (HOOKPROC)GetProcAddress(hHookDll, ansiFunctionName);
     }
     else {
-        // 是字符串指针
+        // 是字符串指针
         hookProc = (HOOKPROC)GetProcAddress(hHookDll, ansiFunctionName);
 
-        // 释放转换的字符串
+        // 释放转换的字符串
         FreeConvertedString((LPVOID)ansiFunctionName);
     }
 
@@ -41,12 +39,12 @@ HRESULT InjectUsingHook(LPCWSTR dllPath, LPCWSTR functionName, DWORD threadId, i
         return HRESULT_FROM_WIN32(ERROR_PROC_NOT_FOUND);
     }
 
-    // 设置Windows钩子
+    // 设置Windows钩子
     HHOOK hHook = SetWindowsHookExW(
-        hookType,           // 钩子类型
-        hookProc,           // 钩子过程
-        hHookDll,           // 包含钩子过程的DLL
-        threadId            // 目标线程ID（0表示全局钩子）
+        hookType,           // 钩子类型
+        hookProc,           // 钩子过程
+        hHookDll,           // 包含钩子过程的DLL
+        threadId            // 目标线程ID（0表示全局钩子）
     );
 
     if (!hHook) {
@@ -55,37 +53,37 @@ HRESULT InjectUsingHook(LPCWSTR dllPath, LPCWSTR functionName, DWORD threadId, i
         return HRESULT_FROM_WIN32(lastError);
     }
 
-    // 触发钩子 - 发送消息到目标线程的消息队列
+    // 触发钩子 - 发送消息到目标线程的消息队列
     if (threadId != 0) {
-        // 线程特定的钩子，发送线程消息
+        // 线程特定的钩子，发送线程消息
         PostThreadMessageW(threadId, WM_NULL, 0, 0);
     }
     else {
-        // 全局钩子，发送广播消息
+        // 全局钩子，发送广播消息
         PostMessageW(HWND_BROADCAST, WM_NULL, 0, 0);
     }
 
-    // 等待一段时间确保DLL被加载
+    // 等待一段时间确保DLL被加载
     Sleep(100);
 
-    // 卸载钩子（DLL已注入，不需要保持钩子）
+    // 卸载钩子（DLL已注入，不需要保持钩子）
     UnhookWindowsHookEx(hHook);
 
-    // 释放当前进程中的DLL（不影响已注入到目标进程的DLL）
+    // 释放当前进程中的DLL（不影响已注入到目标进程的DLL）
     FreeLibrary(hHookDll);
 
     return S_OK;
 }
 
-// 检查钩子是否可用
+// 检查钩子是否可用
 BOOL IsHookAvailable(int hookType)
 {
-    // 临时钩子过程
+    // 临时钩子过程
     HOOKPROC tempProc = [](int nCode, WPARAM wParam, LPARAM lParam) -> LRESULT {
         return CallNextHookEx(NULL, nCode, wParam, lParam);
         };
 
-    // 尝试设置一个临时钩子来检查是否可用
+    // 尝试设置一个临时钩子来检查是否可用
     HHOOK hHook = SetWindowsHookExW(hookType, tempProc, NULL, 0);
 
     if (hHook) {
@@ -96,7 +94,7 @@ BOOL IsHookAvailable(int hookType)
     return FALSE;
 }
 
-// 获取钩子类型名称
+// 获取钩子类型名称
 LPCWSTR GetHookTypeName(int hookType)
 {
     switch (hookType) {
@@ -129,4 +127,4 @@ LPCWSTR GetHookTypeName(int hookType)
     default:
         return L"UNKNOWN";
     }
-}
+}

src/Snap.Hutao.Remastered.Native/HotKeyCallbackManager.cpp

@@ -1,3 +1,4 @@
+#include "pch.h"
 #include "HotKeyCallbackManager.h"
 #include "HutaoNativeHotKeyBeforeSwitchCallback.h"
 
@@ -21,9 +22,9 @@ HutaoNativeHotKeyBeforeSwitchCallback HotKeyCallbackManager::GetCallback()
 
 BOOL HotKeyCallbackManager::InvokeCallback()
 {
-	if (callback_.value != nullptr)
+	if (callback_.has_value())
 	{
-		return callback_.value();
+		return callback_.value()();
 	}
 	return TRUE; // 如果没有回调，默认允许切换
 }
@@ -33,4 +34,4 @@ void HotKeyCallbackManager::ClearCallback()
 	callback_ = {};
 }
 
-HotKeyCallbackManager g_hotKeyCallbackManager = HotKeyCallbackManager();
+HotKeyCallbackManager gotKeyCallbackManager = HotKeyCallbackManager();

src/Snap.Hutao.Remastered.Native/HotKeyCallbackManager.h

@@ -18,4 +18,4 @@ class HotKeyCallbackManager
 	HutaoNativeHotKeyBeforeSwitchCallback callback_ = {};
 };
 
-extern HotKeyCallbackManager g_hotKeyCallbackManager;
+extern HotKeyCallbackManager gotKeyCallbackManager;