From 0738305d0db813155ff2a0faca698b6cb9f767cc Mon Sep 17 00:00:00 2001
From: Claus <312590259@qq.com>
Date: Fri, 10 Oct 2025 20:07:01 +0800
Subject: [PATCH] feat: update stub ext update stub ext for
 stub-cpp(2ec4837af78eb07648d03e2e0e8162cac636e995)

---
 stub-ext/stubext.h | 16 ++++++----------
 1 file changed, 6 insertions(+), 10 deletions(-)

diff --git a/stub-ext/stubext.h b/stub-ext/stubext.h
index 071070c..51b930f 100644
--- a/stub-ext/stubext.h
+++ b/stub-ext/stubext.h
@@ -44,7 +44,7 @@ public:
     template<typename T, class Lamda>
     bool set_lamda(T addr, Lamda lamda)
     {
-        char *fn = reinterpret_cast<char*>(addrof(addr));
+        unsigned char *fn = addrof(addr);
         if (m_result.find(fn) != m_result.end()) {
             assert(m_result.find(fn) == m_result.end());
             return false;
@@ -52,20 +52,16 @@ public:
 
         Wrapper *wrapper = nullptr;
         auto addr_stub = depictShadow(&wrapper,addr,lamda);
-        if (set(addr, addr_stub)) {
-            m_wrappers.insert(std::make_pair(fn, wrapper));
-            return true;
-        } else {
-            freeWrapper(wrapper);
-        }
-        return false;
+        set(addr, addr_stub);
+        m_wrappers.insert(std::make_pair(fn, wrapper));
+        return true;
     }
 
     template<typename T>
     void reset(T addr)
     {
         Stub::reset(addr);
-        char *fn = reinterpret_cast<char*>(addrof(addr));
+        unsigned char *fn = addrof(addr);
         auto iter = m_wrappers.find(fn);
         if (iter != m_wrappers.end()) {
             freeWrapper(iter->second);
@@ -80,7 +76,7 @@ public:
         }
     }
 protected:
-    std::map<char*, Wrapper*> m_wrappers;
+    std::map<unsigned char*, Wrapper*> m_wrappers;
 };
 
 }
-- 
Gitee