本文介绍了如何在 Go 语言中实现用户身份切换，通过 go-sysuser 包获取用户信息，并使用 syscall 包提供的函数设置用户 ID (UID)，从而以指定用户的身份执行命令。本文提供代码示例，并着重强调了安全性和跨平台兼容性方面需要注意的问题。
在某些场景下，我们需要在 Go 程序中以特定用户的身份执行系统命令。例如，需要以 devrim 用户的身份创建文件 miki，相当于执行 su devrim -c "touch miki" 命令。 直接使用 os/exec 包执行命令，默认是以运行该 Go 程序的用户的身份执行。要切换用户身份，需要先获取目标用户的 UID，然后使用 syscall 包提供的函数设置当前进程的 UID。
获取用户 UID
可以使用第三方库 go-sysuser 来获取指定用户的 UID。该库可以解析 /etc/passwd 文件，并提供便捷的 API 来查找用户信息。
首先，安装 go-sysuser 包：
go get github.com/kless/go-sysuser
然后，可以使用以下代码获取用户的 UID：
系统特色及功能简介，主要包括以下方面： 合一：包括语言、模板风格、用户群；此版本内订简体、繁体、英文于一体；可另增设其它语言选项；模板风格指可以存在多界面的情况下进行界面互换；用户群指可写于单用户版本，也可用于多用户商城版本，具体设置可通过会员组权限修改 会员组定制：系统初安装时，内订６级会员分组，即 游客组、管理员组、ＶＩＰ用户组、柜台用户组、柜台ＶＩＰ用户组；此６级会员组不可以删除。另管理
package main
import (
"fmt"
"log"
"github.com/kless/go-sysuser"
)
func getUid(username string) (int, error) {
u, err := user.Lookup(username)
if err != nil {
return -1, err
}
return u.Uid, nil
}
func main() {
username := "devrim"
uid, err := getUid(username)
if err != nil {
log.Fatalf("Failed to get UID for user %s: %v", username, err)
}
fmt.Printf("UID for user %s: %s\n", username, uid)
}切换用户身份并执行命令
获取到用户的 UID 后，可以使用 syscall.Setuid 函数切换当前进程的用户身份。注意，该操作需要 root 权限。
package main
import (
"fmt"
"log"
"os"
"os/exec"
"strconv"
"syscall"
"github.com/kless/go-sysuser"
)
func getUid(username string) (int, error) {
u, err := user.Lookup(username)
if err != nil {
return -1, err
}
uidInt, err := strconv.Atoi(u.Uid)
if err != nil {
return -1, err
}
return uidInt, nil
}
func executeAsUser(username string, command string, args ...string) error {
uid, err := getUid(username)
if err != nil {
return fmt.Errorf("failed to get UID for user %s: %w", username, err)
}
// 只有 root 用户才能切换用户身份
if os.Getuid() != 0 {
return fmt.Errorf("requires root privileges to switch user")
}
// 创建命令
cmd := exec.Command(command, args...)
// 设置用户 ID
cmd.SysProcAttr = &syscall.SysProcAttr{}
cmd.SysProcAttr.Credential = &syscall.Credential{Uid: uint32(uid), Gid: uint32(uid)} // 假设 GID 和 UID 相同
// 执行命令
output, err := cmd.CombinedOutput()
if err != nil {
return fmt.Errorf("failed to execute command: %w, output: %s", err, string(output))
}
fmt.Println(string(output))
return nil
}
func main() {
username := "devrim"
command := "touch"
filename := "miki"
err := executeAsUser(username, command, filename)
if err != nil {
log.Fatalf("Failed to execute command as user %s: %v", username, err)
}
fmt.Printf("Successfully executed 'touch miki' as user %s\n", username)
}注意事项
- 安全性： 切换用户身份是一项敏感操作，需要谨慎处理。确保只允许可信的用户执行此操作，并对输入进行严格的验证，以防止潜在的安全漏洞。
- root 权限： 只有 root 用户才能使用 syscall.Setuid 函数切换用户身份。
- 错误处理： 在代码中添加完善的错误处理机制，以便在出现问题时能够及时发现并进行处理。
- 跨平台兼容性： 上述代码在 Unix-like 系统上有效。在 Windows 系统上，用户身份切换的实现方式有所不同，需要使用 Windows 相关的 API。
- GID: 上面的代码中假设 GID 和 UID 相同，在实际应用中，可能需要根据用户的实际 GID 进行设置。可以通过 user.Lookup 获取用户的 GID 信息。
总结
本文介绍了如何在 Go 语言中切换用户身份并执行命令。通过使用 go-sysuser 包获取用户信息，并使用 syscall.Setuid 函数设置用户 ID，可以实现以指定用户的身份执行系统命令。需要注意的是，切换用户身份是一项敏感操作，需要谨慎处理，并确保代码的安全性、健壮性和跨平台兼容性。
