Go中如何切换用户身份方法详解

IT行业相对于一般传统行业，发展更新速度更快，一旦停止了学习，很快就会被行业所淘汰。所以我们需要踏踏实实的不断学习，精进自己的技术，尤其是初学者。今天golang学习网给大家整理了《Go 中如何切换指定用户身份》，聊聊，我们一起来看看吧！

本文介绍了如何在 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 函数来设置进程的用户身份。

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 命令的功能，以指定用户身份执行命令。 请务必注意权限问题和安全性，并谨慎处理错误。

以上就是《Go中如何切换用户身份方法详解》的详细内容，更多关于的资料请关注golang学习网公众号！