linux - 在 Go 中设置进程名称(如 `ps` 所示)

Question

以下(理所当然)不起作用:

package main

import (
        "os"
        "time"
)

func main() {
        os.Args[0] = "custom name"
        println("sleeping")
        time.Sleep(1000 * time.Second)
        println("done")
}

有些语言提供了 setting process name 的这个特性作为内置功能(例如，在 Ruby 中，这只是 assigning to $0 的问题)或作为第三方库 ( Python )。

我正在寻找至少可以在 Linux 上运行的解决方案。

Answer 1

有多种方法可以实现这一点，其中许多只在特定情况下有效。我真的不建议这样做，因为(一方面)它会导致您的流程在不同情况下以不同的名称出现。他们需要使用系统调用和/或不安全，所以你是在故意破坏 Go 语言的安全性。尽管如此，您的选择似乎是:

修改argv[0]

func SetProcessName(name string) error {
    argv0str := (*reflect.StringHeader)(unsafe.Pointer(&os.Args[0]))
    argv0 := (*[1 << 30]byte)(unsafe.Pointer(argv0str.Data))[:argv0str.Len]

    n := copy(argv0, name)
    if n < len(argv0) {
            argv0[n] = 0
    }

    return nil
}

在 Go 中，您无权访问实际的 argv 数组本身(不调用内部运行时函数)，因此您只能使用一个不超过当前进程名称长度的新名称。

这似乎主要适用于 Darwin 和 Linux。

调用 PR_SET_NAME

func SetProcessName(name string) error {
    bytes := append([]byte(name), 0)
    ptr := unsafe.Pointer(&bytes[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
}

新名称最多为 16 个字节。

这在 Darwin 上不起作用，而且在 Linux 上似乎也没什么作用，尽管它成功了并且 PR_GET_NAME 随后报告了正确的名称。不过，这可能是我的 Linux VM 的一些特殊之处。