Go语言中设置进程名称的实用指南

package main

import (
    "fmt"
    "os"
    "reflect"
    "time"
    "unsafe"
)

// SetProcessName 修改进程名称，通过修改os.Args[0]的底层数据
// 新名称的长度不能超过原始进程名称的长度。
func SetProcessName(name string) error {
    // 获取os.Args[0]的字符串头信息
    argv0str := (*reflect.StringHeader)(unsafe.Pointer(&os.Args[0]))
    // 将字符串头的数据指针转换为可写的字节数组指针
    // 注意：这里创建了一个非常大的数组指针，然后切片到实际长度
    argv0 := (*[1 << 30]byte)(unsafe.Pointer(argv0str.Data))[:argv0str.Len]

    // 复制新名称到argv0的内存区域
    n := copy(argv0, name)
    // 如果新名称比原始名称短，用空字节填充剩余部分
    if n < len(argv0) {
        argv0[n] = 0 // 确保字符串正确终止
    }

    return nil
}

func main() {
    fmt.Printf("原始进程名称 (os.Args[0]): %s\n", os.Args[0])

    // 尝试修改进程名称
    newName := "my_custom_go_process"
    if len(newName) > len(os.Args[0]) {
        fmt.Printf("警告：新名称 '%s' 长度 (%d) 超过原始名称 '%s' 长度 (%d)，可能无法完全显示。\n",
            newName, len(newName), os.Args[0], len(os.Args[0]))
        // 截断新名称以适应长度限制
        newName = newName[:len(os.Args[0])]
    }

    err := SetProcessName(newName)
    if err != nil {
        fmt.Printf("设置进程名称失败: %v\n", err)
    } else {
        fmt.Printf("进程名称已尝试修改为: %s\n", newName)
        fmt.Println("程序将休眠60秒，请在此期间使用 `ps aux | grep my_custom_go_process` 或 `ps -p <PID> -o comm=` 查看效果。")
    }

    time.Sleep(60 * time.Second)
    fmt.Println("程序执行完毕。")
}

package main

import (
    "fmt"
    "os"
    "syscall"
    "time"
    "unsafe"
)

// SetProcessNameWithPrctl 使用PR_SET_NAME系统调用修改进程名称
// 此方法仅适用于Linux，且名称长度不能超过16字节（包括终止符）。
// 通常影响线程名称，而非主进程的命令行参数。
func SetProcessNameWithPrctl(name string) error {
    // PR_SET_NAME的名称长度限制为16字节（包括空终止符）
    if len(name) >= 16 {
        name = name[:15] // 截断以适应限制
    }
    bytes := append([]byte(name), 0) // 添加空终止符
    ptr := unsafe.Pointer(&bytes[0]) // 获取字节数组的指针

    // 调用prctl系统调用，PR_SET_NAME命令
    // 参数：syscall.SYS_PRCTL, PR_SET_NAME, 名称指针, 0, 0, 0
    if _, _, errno := syscall.RawSyscall6(syscall.SYS_PRCTL, syscall.PR_SET_NAME, uintptr(ptr), 0, 0, 0, 0); errno != 0 {
        return syscall.Errno(errno)
    }
    return nil
}

func main() {
    fmt.Printf("原始进程名称 (os.Args[0]): %s\n", os.Args[0])

    // 尝试修改进程名称
    newName := "go_prctl_proc" // 限制16字节
    err := SetProcessNameWithPrctl(newName)
    if err != nil {
        fmt.Printf("设置进程名称失败: %v\n", err)
    } else {
        fmt.Printf("进程名称已尝试通过PR_SET_NAME修改为: %s\n", newName)
        fmt.Println("程序将休眠60秒，请在此期间使用 `ps aux | grep go_prctl_proc` 或 `ps -L -p <PID> -o comm=` 查看效果。")
    }

    time.Sleep(60 * time.Second)
    fmt.Println("程序执行完毕。")
}