Go语言中设置进程名称的实践与探讨

package main

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

// SetProcessName 修改进程名称
// 该方法通过unsafe包直接修改os.Args[0]的底层内存
// 新名称的长度不能超过原始进程名称的长度
func SetProcessName(name string) error {
    // 获取os.Args[0]的底层字符串头信息
    argv0str := (*reflect.StringHeader)(unsafe.Pointer(&os.Args[0]))
    // 将字符串头的数据指针转换为字节切片，表示原始argv[0]的内存区域
    argv0 := (*[1 << 30]byte)(unsafe.Pointer(argv0str.Data))[:argv0str.Len]

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

    return nil
}

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

    newName := "my_custom_go_process"
    err := SetProcessName(newName)
    if err != nil {
        fmt.Printf("设置进程名称失败: %v\n", err)
    } else {
        fmt.Printf("进程名称已尝试修改为: %s\n", newName)
    }

    fmt.Println("程序将休眠1000秒，请使用 `ps -ef | grep my_custom_go_process` 查看")
    time.Sleep(1000 * time.Second)
    fmt.Println("程序结束")
}

package main

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

// SetProcessNameBySyscall 通过PR_SET_NAME系统调用修改进程/线程名称
// 该方法仅适用于Linux系统，且新名称长度不能超过16字节（包括终止符）
func SetProcessNameBySyscall(name string) error {
    // 将名称转换为字节切片并添加终止符
    bytes := append([]byte(name), 0)
    // 获取字节切片的第一个元素的指针
    ptr := unsafe.Pointer(&bytes[0])

    // 调用Linux的prctl系统调用，PR_SET_NAME常量用于设置名称
    // syscall.SYS_PRCTL 是 prctl 系统调用的编号
    // PR_SET_NAME 是 prctl 的一个操作码，表示设置进程/线程名称
    // uintptr(ptr) 是新名称的指针
    // 其他参数为0，因为PR_SET_NAME只需要两个参数
    _, _, errno := syscall.RawSyscall6(syscall.SYS_PRCTL, syscall.PR_SET_NAME, uintptr(ptr), 0, 0, 0, 0)
    if errno != 0 {
        return syscall.Errno(errno)
    }
    return nil
}

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

    newName := "go_proc_16bytes_max" // 名称长度限制为15个字符 + 1个终止符
    err := SetProcessNameBySyscall(newName)
    if err != nil {
        fmt.Printf("通过syscall设置进程名称失败: %v\n", err)
    } else {
        fmt.Printf("进程名称已尝试通过syscall修改为: %s\n", newName)
    }

    fmt.Println("程序将休眠1000秒，请使用 `ps -ef | grep go_proc_16bytes_max` 查看")
    time.Sleep(1000 * time.Second)
    fmt.Println("程序结束")
}