feat: 添加简单的xml解析代码[只能解析2层]，来自TrafficMonitor拷贝的简单逻辑

Author: sdttttt

simple_xml.cpp

@@ -0,0 +1,90 @@
+//
+// Created by kms on 2025/9/11.
+//
+
+#include "simple_xml.h"
+
+BOOL HasUtf8Bom(const std::string_view str)
+{
+    return str.size() >= 3 &&
+        static_cast<unsigned char>(str[0]) == 0xEF &&
+        static_cast<unsigned char>(str[1]) == 0xBB &&
+        static_cast<unsigned char>(str[2]) == 0xBF;
+}
+
+std::wstring UTF82UTF16(const std::string& str)
+{
+    if (str.empty())
+    {
+        return L"";
+    }
+
+    auto size_needed = MultiByteToWideChar(CP_UTF8, 0, str.data(), str.size(), nullptr, 0);
+
+    if (size_needed <= 0)
+    {
+        return L"";
+    }
+
+    std::wstring ret(size_needed, 0);
+    MultiByteToWideChar(CP_UTF8, 0, str.data(), str.size(), ret.data(), size_needed);
+    return ret;
+}
+
+
+CSimpleXML::CSimpleXML(const wstring& xml_path)
+{
+    std::ifstream file_stream(xml_path, std::ios::binary);
+
+    if (!file_stream.is_open())
+    {
+        return;
+    }
+
+    //读取文件内容
+
+    std::stringstream buffer;
+    buffer << file_stream.rdbuf();
+    std::string xml_str = buffer.str();
+
+    // UTF-8 BOM（EF BB BF → -17, -69, -65）
+    if (HasUtf8Bom(xml_str))
+    {
+        xml_str.erase(0, 3);
+    }
+
+    // 默认全按照UTF8处理
+    this->m_xml_content = UTF82UTF16(xml_str);
+}
+
+std::wstring CSimpleXML::GetNode(LPCWSTR node, LPCWSTR parent) const
+{
+    wstring node_content = _GetNode(parent, this->m_xml_content);
+    return _GetNode(node, node_content);
+}
+
+std::wstring CSimpleXML::GetNode(LPCWSTR node) const
+{
+    return _GetNode(node, this->m_xml_content);
+}
+
+std::wstring CSimpleXML::_GetNode(LPCWSTR node, const wstring& content)
+{
+    std::wstring node_start{L'<'};
+    std::wstring node_end{L'<'};
+    node_start += node;
+    node_start += L'>';
+    node_end += L'/';
+    node_end += node;
+    node_end += L'>';
+
+    size_t index_start, index_end;
+
+    index_start = content.find(node_start);
+    index_end = content.find(node_end);
+    if (index_start == wstring::npos || index_end == wstring::npos)
+        return wstring();
+
+    auto result = content.substr(index_start + node_start.size(), index_end - index_start - node_start.size());
+    return result;
+}

simple_xml.h

@@ -0,0 +1,27 @@
+//
+// Created by kms on 2025/9/11.
+//
+
+#include <fstream>
+#include <sstream>
+
+using std::wstring;
+
+class CSimpleXML
+{
+public:
+    CSimpleXML(const wstring& xml_path);
+    CSimpleXML() = default;
+    ~CSimpleXML() = default;
+
+    void LoadXMLContentDirect(const wstring& xml_content) { m_xml_content = xml_content; }
+
+    wstring GetNode(const wchar_t* node, const wchar_t* parent) const;
+    wstring GetNode(const wchar_t* node) const;
+
+    static wstring _GetNode(const wchar_t* node, const wstring& content);
+
+protected:
+    wstring m_xml_content;
+
+};

task_sched.cpp

@@ -25,6 +25,28 @@ BOOL TryInvoke(F&& func, LPCWSTR msg)
 #define ErrThrow(msg) \
     do { TryInvoke([&] { return E_FAIL; }, msg); return FALSE; } while (0);
 
+
+std::wstring defaultTaskName()
+{
+    // 获取所需缓冲区大小
+    const auto required_size = GetEnvironmentVariableW(L"USERNAME", nullptr, 0);
+    if (required_size == 0) {
+        return L"timer-devourer-unknown";
+    }
+
+    std::wstring username(required_size, L'\0');
+
+    const auto copied = GetEnvironmentVariableW(L"USERNAME", username.data(), required_size);
+    if (copied == 0 || copied >= required_size) {
+        return L"timer-devourer-unknown";
+    }
+
+    // 调整大小（copied 是实际字符数）
+    username.resize(copied);
+
+    return L"timer-devourer-" + username;
+}
+
 /**
  *
  * @return BOOL
@@ -34,7 +56,7 @@ BOOL CreateStartupTask()
     // 错误位标志
     auto hr = S_OK;
 
-    std::wstring task_name;
+    std::wstring task_name = defaultTaskName();
 
     std::wstring username_domain;
     username_domain.resize(USERNAME_DOMAIN_LEN);
@@ -68,12 +90,12 @@ BOOL CreateStartupTask()
         ErrThrow(L"环境变量:USERDOMAIN获取失败");
     }
 
+    username.resize(username_len);
+    username_domain.resize(userdomain_len);
+
     username_domain += L"\\";
     username_domain += username;
 
-    task_name += L"timer-";
-    task_name += username;
-
     TryStep(com_scope.init(), L"COM组件:初始化错误");
 
 
@@ -226,3 +248,117 @@ BOOL CreateStartupTask()
 
     return SUCCEEDED(hr);
 }
+
+
+BOOL DeleteStartupTask()
+{
+    HRESULT hr = S_OK;
+
+    WCHAR username[USERNAME_LEN];
+    std::wstring task_name = defaultTaskName();
+
+    COMScope com_scope;
+    COMPtr<ITaskService> p_service(nullptr);
+    COMPtr<ITaskFolder> p_task_folder(nullptr);
+
+
+    TryStep(com_scope.init(), L"COM组件:初始化错误");
+
+    TryStep(CoCreateInstance(
+            __uuidof(TaskScheduler),
+            nullptr,
+            CLSCTX_INPROC_SERVER,
+            __uuidof(ITaskService),
+            reinterpret_cast<void**>(p_service.AsOutPtr()))
+            , L"COM组件TaskScheduler:创建失败");
+    TryStep(hr , L"COM组件:创建ITaskService失败");
+
+    hr = p_service->Connect(_variant_t(), _variant_t(), _variant_t(), _variant_t());
+    TryStep(hr, L"COM组件:连接ITaskService失败");
+
+    hr = p_service->GetFolder(_bstr_t(TASK_SCHED_FOLDER), p_task_folder.AsOutPtr());
+    if (FAILED(hr))
+    {
+        // 断言文件夹不存在，不需要删除任务
+        return TRUE;
+    }
+
+    {
+        COMPtr<IRegisteredTask> p_exist_reg_task(nullptr);
+        hr = p_task_folder->GetTask(_bstr_t(task_name.c_str()), p_exist_reg_task.AsOutPtr());
+        if (SUCCEEDED(hr))
+        {
+            hr = p_task_folder->DeleteTask(_bstr_t(task_name.c_str()), 0);
+        }
+    }
+
+    return (SUCCEEDED(hr));
+}
+
+
+BOOL IsActiveStartupTask(std::wstring* path)
+{
+    HRESULT hr = S_OK;
+
+    std::wstring wstrTaskName = defaultTaskName();
+
+    COMScope com_scope;
+    COMPtr<ITaskService> p_service(nullptr);
+    COMPtr<ITaskFolder> p_task_folder(nullptr);
+
+    BOOL command_path_match = FALSE;
+
+    TryStep(com_scope.init(), L"COM组件:初始化错误");
+    TryStep(CoCreateInstance(
+            __uuidof(TaskScheduler),
+            nullptr,
+            CLSCTX_INPROC_SERVER,
+            __uuidof(ITaskService),
+            p_service.AsOutVoidPtr())
+            , L"COM组件TaskScheduler:创建失败");
+
+    // Connect to the task service.
+    hr = p_service->Connect(_variant_t(), _variant_t(), _variant_t(), _variant_t());
+    TryStep(hr, L"COM组件:连接ITaskService失败");
+
+    // ------------------------------------------------------
+    // Get the TrafficMonitor task folder.
+    hr = p_service->GetFolder(_bstr_t(TASK_SCHED_FOLDER), p_task_folder.AsOutPtr());
+    TryStep(hr, L"COM组件:获取ITaskFolder失败");
+
+    // ------------------------------------------------------
+    // If the task exists, disable.
+    {
+        COMPtr<IRegisteredTask> p_exist_reg_task(nullptr);
+        hr = p_task_folder->GetTask(_bstr_t(wstrTaskName.c_str()), p_exist_reg_task.AsOutPtr());
+        if (SUCCEEDED(hr))
+        {
+            VARIANT_BOOL is_enabled;
+            hr = p_exist_reg_task->get_Enabled(&is_enabled);
+            //判断已存在的任务计划命令的exe文件路径是否为当前exe的路径
+            CComBSTR xml_buff{};
+            p_exist_reg_task->get_Xml(&xml_buff);
+            CSimpleXML xml;
+            xml.LoadXMLContentDirect(std::wstring(xml_buff));
+            // 获得配置中的执行路径
+            std::wstring command_path = xml.GetNode(L"Command", L"Exec");
+            if (path != nullptr)
+            {
+                *path = command_path;
+            }
+            // 获得当前exe的路径
+            std::wstring exec_path;
+            exec_path.resize(MAX_PATH);
+            GetModuleFileName(NULL, exec_path.data(), MAX_PATH);
+            command_path_match = (command_path == exec_path);
+            SysFreeString(xml_buff);
+            if (SUCCEEDED(hr))
+            {
+                // Got the value. Return it.
+                hr = (is_enabled == VARIANT_TRUE) ? S_OK : E_FAIL; // Fake success or fail to return the value.
+            }
+        }
+    }
+
+    return (SUCCEEDED(hr) && command_path_match);
+}

task_sched.h

@@ -7,6 +7,7 @@
 #include <comdef.h>
 #include <Lmcons.h>
 #include <type_traits>
+#include "simple_xml.h"
 
 
 constexpr auto ENV_USERNAME = L"USERNAME";
@@ -25,6 +26,8 @@ namespace TaskSched
 
 
 /**
+ * //--------------------- 后面发现官方有自动包装的CComPtr，努力全部木大
+ *
  * 自动包装COM对象指针，生命周期结束自动调用Release
  * 要求T必须能调用Release方法，且T必须是IUnknown的子类
  *