寻求os.Getenv()反向工具/库：从环境变量值反查变量名

我有一系列REG_EXPAND_SZ类型的十六进制字符串，示例如下：

reg,[HKEY_LOCAL_MACHINE\SOFTWAR\WOW6432Node], "Pathethic"=hex(2):43,00,3a,00,5c,00,57,00,49,00,4e,00,44,00,4f,00,57,00,53,00,\00,00

解析后得到的是C:\Windows，但我需要的是原始的环境变量形式**%systemroot%**。问题根源在于用户执行reg add命令时，终端自动展开了环境变量，导致注册表中存储的是展开后的实际路径，而非变量名。现在需要实现一个反向功能：输入展开后的路径值，返回对应的环境变量名（比如输入C:\Windows返回%systemroot%）。

现有Go解析代码 #

package main

import (
    "encoding/hex"
    "fmt"
    "log"
    "os"
)

func CToGoString(b []byte) string {
    var buf []byte
    for _, c := range b {
        if c != 0 {
            buf = append(buf, c)
        }
    }
    return string(buf)
}

func main() {
    str := "43003a005c00570049004e0044004f00570053000000" // 对应%systemroot%
    bs, err := hex.DecodeString(str)
    if err != nil {
        panic(err)
    }
    result := CToGoString(bs)
    fmt.Println(result)

    f, err := os.OpenFile("rollback.bat", os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0755)
    if err != nil {
        log.Fatal(err)
    }
    defer f.Close() // 补充关闭文件操作，避免资源泄漏

    f.WriteString(result + "\n")
}

尝试过的手动匹配方案 #

我曾手动实现匹配逻辑，但环境变量数量过多，局限性很大：

func reverseEnvVar(value string) string {
    // 展开输入字符串中的环境变量
    expanded := os.ExpandEnv(value)

    // 检查展开后的字符串是否匹配已知环境变量的值
    switch expanded {
    case os.Getenv("SystemRoot"):
        return "%systemroot%"
    case os.Getenv("ProgramFiles"):
        return "%programfiles%"
    case os.Getenv("ProgramFiles(x86)"):
        return "%programfiles(x86)%"
    case os.Getenv("AppData"):
        return "%appdata%"
    case os.Getenv("LocalAppData"):
        return "%localappdata%"
    case os.Getenv("UserProfile"):
        return "%userprofile%"
    case os.Getenv("TEMP"):
        return "%temp%"
    case os.Getenv("TMP"):
        return "%tmp%"
    default:
        // 若不匹配已知环境变量值，返回原始输入值
        return value
    }
}

优化解决方案：动态遍历所有环境变量并反向匹配 #

手动枚举环境变量显然不现实，正确的做法是遍历系统中所有环境变量，建立值到变量名的映射（处理值重复的情况）。以下是优化后的实现：

package main

import (
    "encoding/hex"
    "fmt"
    "log"
    "os"
    "strings"
)

func CToGoString(b []byte) string {
    var buf []byte
    for _, c := range b {
        if c != 0 {
            buf = append(buf, c)
        }
    }
    return string(buf)
}

// ReverseExpandEnv 从展开后的路径反向查找对应的环境变量名，优先匹配最长的变量值
func ReverseExpandEnv(value string) string {
    // 先展开输入值中的可能变量，确保匹配的是实际路径
    target := strings.ToLower(os.ExpandEnv(value))

    // 收集所有环境变量，按值的长度降序排序，优先替换最长的匹配项
    type envEntry struct {
        name  string
        value string
    }
    var envList []envEntry

    for _, e := range os.Environ() {
        parts := strings.SplitN(e, "=", 2)
        if len(parts) != 2 {
            continue
        }
        name := parts[0]
        val := strings.ToLower(parts[1])
        // 只保留非空且长度大于等于3的变量值，避免匹配太短的无意义值
        if val != "" && len(val) >= 3 {
            envList = append(envList, envEntry{name, val})
        }
    }

    // 按值的长度降序排序，确保最长的匹配项被优先替换
    for i := 0; i < len(envList); i++ {
        for j := i + 1; j < len(envList); j++ {
            if len(envList[j].value) > len(envList[i].value) {
                envList[i], envList[j] = envList[j], envList[i]
            }
        }
    }

    // 遍历查找匹配项
    for _, entry := range envList {
        if target == entry.value {
            return fmt.Sprintf("%%%s%%", strings.ToLower(entry.name))
        }
        // 可选：处理路径中包含环境变量的情况，比如C:\Windows\System32替换为%systemroot%\System32
        //if strings.HasPrefix(target, entry.value + "\\") {
        //    return fmt.Sprintf("%%%s%%%s", strings.ToLower(entry.name), target[len(entry.value):])
        //}
    }

    // 无匹配则返回原始值
    return value
}

func main() {
    str := "43003a005c00570049004e0044004f00570053000000" // 对应%systemroot%
    bs, err := hex.DecodeString(str)
    if err != nil {
        panic(err)
    }
    result := CToGoString(bs)
    fmt.Println("原始解析结果:", result)
    fmt.Println("反向还原结果:", ReverseExpandEnv(result))

    f, err := os.OpenFile("rollback.bat", os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0755)
    if err != nil {
        log.Fatal(err)
    }
    defer f.Close()

    f.WriteString(ReverseExpandEnv(result) + "\n")
}

关键说明 #

• 遍历所有环境变量：通过os.Environ()获取系统全部环境变量，避免手动枚举的局限性；
• 排序优先匹配最长值：防止短变量值被误匹配（比如避免"C:"优先于"C:\Windows"被匹配）；
• 大小写不敏感：Windows环境变量名不区分大小写，转换为小写后匹配更准确；
• 可扩展路径包含匹配：注释部分的代码可以处理路径中包含环境变量的场景，根据需求开启。

内容的提问来源于Stack Exchange，提问作者Aigerim