本文介绍了如何在 Go 语言中以指定用户名切换用户身份，并执行相应的命令。通过使用 go-sysuser 库获取用户 UID，并结合 syscall 包中的函数设置用户身份，我们可以实现在 Go 程序中模拟 su 命令的功能。文章提供了详细的代码示例和注意事项，帮助开发者理解和应用该技术。
在某些情况下，我们需要在 Go 程序中以不同的用户身份执行命令，例如模拟 su 命令。 这可以通过以下步骤实现：
1. 获取用户 UID
首先，我们需要根据用户名获取对应的用户 ID (UID)。 可以使用第三方库 go-sysuser 来实现此功能。 该库提供了一种跨平台的方式来访问系统用户数据库。
package main
import (
"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("invalid 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("Error opening %s: %v", 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("Error reading line: %v", err)
}
break
}
parsed := strings.Split(line, ":", 4)
if parsed[0] == su {
value, err := strconv.Atoi(parsed[2])
if err != nil {
log.Printf("Error converting UID to int: %v", err)
return -1
}
return value
}
}
return -1
}
func main() {
username := "devrim" // 替换为你要切换的用户名
uid, err := getUid(username)
if err != nil {
fmt.Printf("Error getting UID for user %s: %v\n", username, err)
uid = getUidFromPasswd(username)
if uid == -1 {
log.Fatalf("Failed to get UID for user %s using both methods.", username)
return
}
fmt.Printf("Got UID %d from /etc/passwd\n", uid)
} else {
fmt.Printf("UID for user %s is: %d\n", 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 函数来设置进程的用户身份。
手机版企业智能品牌系统(手机网站)1.0
下载
该系统采用先进的HTML5+CSS3结构，既有手机APP的良好体验，又有智能建站系统的操作方便。在中国，企业网站建设在已有20年，但表现方式基本是一成不变，此产品进行了与众不同的偿试。一切以小微企业实际情况出发，注重核心产品的塑造以及企业文化展示。让小微企业及个人都能找准自身的细分化定位，服务好客户。
package main
import (
"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("invalid 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("failed to get UID for user %s: %w", username, err)
}
// 设置用户身份
if err := syscall.Setuid(uid); err != nil {
return fmt.Errorf("failed to set UID to %d: %w", uid, err)
}
// 执行命令
cmd := exec.Command(command, args...)
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
if err := cmd.Run(); err != nil {
return fmt.Errorf("failed to execute command: %w", err)
}
return nil
}
func main() {
username := "devrim" // 替换为你要切换的用户名
command := "touch"
args := []string{"miki"}
if err := switchUserAndExecute(username, command, args...); err != nil {
log.Fatalf("Error switching user and executing command: %v", err)
}
fmt.Println("Command executed successfully.")
}注意事项：
- Setuid 函数需要 root 权限才能成功设置用户身份。 因此，程序需要以 root 用户身份运行。
- 在设置用户身份后，后续执行的命令都将以该用户的身份运行。
- 在完成操作后，可以考虑恢复原始用户身份，以避免潜在的安全问题。
3. 完整示例
以下是一个完整的示例，演示了如何以指定用户身份执行 touch 命令：
package main
import (
"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("invalid 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("failed to get UID for user %s: %w", username, err)
}
// 设置用户身份
if err := syscall.Setuid(uid); err != nil {
return fmt.Errorf("failed to set UID to %d: %w", uid, err)
}
// 执行命令
cmd := exec.Command(command, args...)
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
if err := cmd.Run(); err != nil {
return fmt.Errorf("failed to execute command: %w", err)
}
return nil
}
func main() {
username := "devrim" // 替换为你要切换的用户名
command := "touch"
args := []string{"miki"}
if err := switchUserAndExecute(username, command, args...); err != nil {
log.Fatalf("Error switching user and executing command: %v", err)
}
fmt.Println("Command executed successfully.")
}总结：
通过使用 go-sysuser 库和 syscall 包，我们可以在 Go 程序中模拟 su 命令的功能，以指定用户身份执行命令。 请务必注意权限问题和安全性，并谨慎处理错误。
