在 Go 中以指定用户名切换用户身份
本文介绍了如何在 Go 语言中以指定用户名切换用户身份，并执行相应的命令。通过使用 go-sysuser 库获取用户 UID，并结合文章 syscall 包中的函数设置用户身份，我们可以实现在 Go 程序中模拟 su 命令的功能。提供了详细的代码示例和注意事项，有助于开发者理解和应用该技术。
在某些情况下，我们需要在Go程序中以不同的用户身份执行命令，例如模拟su命令。这可以通过以下步骤实现：1. 获取用户UID
首先，我们需要根据用户名获取对应的用户ID（UID）。可以使用第三方库go-sysuser来实现此功能。该库提供了一种跨平台的方式来访问系统用户数据库。
package mainimport ( "；fmt"； "；log"； "；strconv"； "；strings"； "；github.com/kless/go-sysuser"；)func getUid(username string) (int， error) { u， err ：= user.Lookup(username) if err != nil { return -1， err } uid， err ：= strconv.Atoi(u.Uid) if err != nil { return -1， fmt.Errorf("；无效的 UID q： s"；， u.Uid， err) } return uid， nil}func getUidFromPasswd(su string) int { passwdFile ：= "；/etc/passwd"； passwd， err ：= os.Open(passwdFile) if err != nil { log.Fatalf(quot；打开文件错误： vquot；， passwdFile， err) return -1 } defer passwd.Close() reader ：= bufio.NewReader(passwd) for { line， err ：= reader.ReadString('\n') if err != nil { if err != io.EOF { log.Printf(quot；读取第 1 行错误： vquot；， err) } break } parsed ：= strings.Split(line， quot；：quot；， 4) if parsed[0] == su { value， err ：= strconv.Atoi(parsed[2]) if err != nil { log.Printf(quot；将 UID 转换为 int 时出错： vquot；， err) return -1 } return value } } return -1}func main() { username ：= quot；devrimquot； // 替换为你要切换的用户名 uid， err ：= getUid(username) if err != nil { fmt.Printf(quot；获取用户 s 的 UID 时出错： v\nquot；， user
name， err) uid = getUidFromPasswd(username) if uid == -1 { log.Fatalf(quot；两种方法均无法获取用户 s 的 UID。quot；， username) return } fmt.Printf(quot；从 /etc/passwd\nquot 获取 UID d；， uid) } else { fmt.Printf(quot；用户 s 的 UID 为： d\nquot；， username，uid) }}登录后复制
注意事项：确保已经安装 go-sysuser 库。可以使用 go get github.com/kless/go-sysuser 命令安装。user.Lookup(username) 函数可能会返回错误，例如用户不存在。需要适当地处理这些错误。如果 go-sysuser 无法工作，可以考虑直接解析 /etc/passwd 文件。但是，请注意不同 *nix 系统的 /etc/passwd格式可能存在差异，因此需要字符串处理。2. 设置用户身份
获取到UID后，可以使用syscall包中的Setuid函数来设置进程的用户身份。
package mainimport ( "；fmt"； "；log"； "；os"； "；os/exec"； "；strconv"； "；strings"； "；syscall"； "；github.com/kless/go-sysuser"；)func getUid(username string) (int， error) { u， err ：= user.Lookup(username) if err != nil { return -1， err } uid， err ：= strconv.Atoi(u.Uid) if err != nil { return -1， fmt.Errorf("；无效的 UID q： s"；， u.Uid， err) } return uid， nil}func switchUserAndExecute(username string， command string， args ...string) error { uid， err ：= getUid(username) if err != nil { return fmt.Errorf(quot；获取用户 s 的 UID 失败： wquot；， username， err) } // 设置用户身份 if err ：= syscall.Setuid(uid)； err != nil { return fmt.Errorf(quot；获取用户 s 的 UID 失败： wquot；， uid， err) } // 执行命令 cmd ：= exec.Command(command， args...) cmd.Stdout = os.Stdout cmd.Stderr = os.Stderr if err ：= cmd.Run()； err != nil { return fmt.Errorf(quot；执行命令失败： wquot；， err) } return nil}func main() { username ：= quot；devrimquot； //替换为你要切换的用户名命令：= quot；touchquot； args ：= []string{quot；mikiquot；} if err ：= switchUserAndExecute(username， command， args...)； err != nil { log.Fatalf(quot；切换用户并执行命令时出错： vquot；， err) } fmt.Println(quot；命令执行成功。quot；
)}登录后复制登录后复制
需要注意事项：Setuid函数root权限才能成功设置用户身份。因此，程序需要以root用户身份运行。在设置用户身份后，后续执行的命令都按照该用户的身份运行。在完成操作后，可以考虑恢复原始用户身份，可能会出现安全问题。
3. 完整示例
以下是一个完整的示例，演示了如何指定用户身份执行 touch 命令：package mainimport ( quot；fmtquot； quot；logquot； quot；osquot； quot；os/execquot； quot；strconvquot； quot；stringsquot； quot；syscallquot； quot；github.com/kless/go-sysuserquot；)func getUid(username string) (int， error) { u， err ：= user.Lookup(用户名) if err != nil { return -1， err } uid， err ：= strconv.Atoi(u.Uid) if err != nil { return -1， fmt.Errorf(quot；无效 UID q： squot；， u.Uid， err) } return uid， nil}func switchUserAndExecute(username字符串，命令string， args ...string) error { uid， err ：= getUid(username) if err != nil { return fmt.Errorf(quot；无法获取用户 s 的 UID： wquot；， username， err) } // 设置用户身份 if err ：= syscall.Setuid(uid)； err != nil { return fmt.Errorf(quot；无法将 UID 设置为 d： wquot；， uid， err) } // 执行命令 cmd ：= exec.Command(command， args...) cmd.Stdout = os.Stdout cmd.Stderr = os.Stderr if err ：= cmd.Run()； err != nil { return fmt.Errorf(quot；无法执行命令： wquot；， err) } return nil}func main() { username ：= quot；devrimquot； // 替换为你要切换的用户名 command ：= quot；touchquot； args ：= []string{quot；mikiquot；} if err ：= switchUserAndExecute(username， command， args...)； err != nil { log. Fatalf(quot；切换用户和执行命令时出错： vquot；， err) } fmt
.Println(quot；命令执行成功。quot；)}登录后复制 登录后复制
总结：
通过使用 go-sysuser 库和 syscall 包，我们在 Go 程序中可以模拟 su 命令的功能，以指定用户身份执行命令。请务必注意权限问题和安全性，并进程处理错误。
以上就是在 Go中以指定用户名切换用户身份的详细内容，更多请关注乐哥常识网其他相关文章！