Go语言中unsafe.Pointer与函数指针的转换及注意事项

func myFunc(arg int) string {
    return fmt.Sprintf("Hello, %d", arg)
}

var fnPtr func(int) string = myFunc // 获取函数指针变量

genericPtr := unsafe.Pointer(&fnPtr) // genericPtr现在存储了fnPtr变量的内存地址

// 假设 genericPtr 实际上指向一个 func(int) int 类型的函数
// 但是我们将其重塑为 func(int) bool
reinterpretedFn := (*func(int) bool)(genericPtr)

package main

import (
    "fmt"
    "unsafe"
)

// 定义一个接受字符串参数的函数
func function1(s string) {
    fmt.Printf("Function1 called with: %s\n", s)
}

// 定义一个接受整数参数并返回整数的函数
func function2(i int) int {
    fmt.Printf("Function2 called with: %d\n", i)
    return i + 1
}

func main() {
    // 1. 获取函数指针变量
    // 注意：这里是获取存储函数地址的变量的地址，而不是函数本身的地址
    f1Var := function1
    f2Var := function2

    // 2. 将函数指针变量的地址转换为 unsafe.Pointer 并存储
    pointers := []unsafe.Pointer{
        unsafe.Pointer(&f1Var), // 存储 function1 的指针变量的地址
        unsafe.Pointer(&f2Var), // 存储 function2 的指针变量的地址
    }

    fmt.Println("--- 原始函数调用示例 ---")
    f1Var("Hello Go")
    result2 := f2Var(10)
    fmt.Printf("Function2 original result: %d\n", result2)

    fmt.Println("\n--- 通过 unsafe.Pointer 重塑和调用 ---")

    // 3. 从 unsafe.Pointer 恢复并重塑函数签名
    // 尝试将 function2 (func(int) int) 的指针重塑为 func(int) bool
    // 这意味着：
    // - 调用时传入 int 类型参数
    // - 期望返回 bool 类型
    // 实际上 function2 返回 int，这里会发生类型解释错误
    reinterpretedF2 := (*func(int) bool)(pointers[1])

    // 4. 调用重塑后的函数
    // 传入整数 5。function2会执行，并返回 5+1=6。
    // 但由于我们将其重塑为 func(int) bool，Go会尝试将返回的整数 6 解释为布尔值。
    // 在Go中，非零整数在某些上下文中可能被视为true（虽然这里是直接解释，不是条件判断）。
    // 具体行为取决于Go的ABI和内存布局，可能导致不可预测的结果。
    // 在大多数情况下，非零值解释为true，零值解释为false。
    resultReinterpreted := (*reinterpretedF2)(5)
    fmt.Printf("Reinterpreted function called with 5, result: %v (type: %T)\n", resultReinterpreted, resultReinterpreted)

    // 另一个例子：尝试将 function1 (func(string)) 重塑为 func(int, int) int
    // 这将导致更严重的问题，因为参数数量和类型完全不匹配
    // reinterpretedF1 := (*func(int, int) int)(pointers[0])
    // fmt.Println((*reinterpretedF1)(1, 2)) // 运行时很可能崩溃或产生垃圾数据
}

--- 原始函数调用示例 ---
Function1 called with: Hello Go
Function2 called with: 10
Function2 original result: 11

--- 通过 unsafe.Pointer 重塑和调用 ---
Function2 called with: 5
Reinterpreted function called with 5, result: true (type: bool)