golang 执行命令行(二)

继续上文所述,有时候我们需要设置进程的启动用户,操作与设置进程组的方式类似,不多说直接上代码:

// 修改进程的执行用户

func withUserAttr(cmd *exec.Cmd, name string) error {

	// 检测用户是否存在

	user, err := user.Lookup(name)

	if err != nil {

		return errors.Wrapf(err, "invalid user %s", name)

	}

	// set process attr

	// 获取用户 id

	uid, err := strconv.ParseUint(user.Uid, 10, 32)

	if err != nil {

		return err

	}

	// 获取用户组 id

	gid, err := strconv.ParseUint(user.Gid, 10, 32)

	if err != nil {

		return err

	}

	attr := cmd.SysProcAttr

	//设置进程执行用户

	attr.Credential = &syscall.Credential{

		Uid:         uint32(uid),

		Gid:         uint32(gid),

		NoSetGroups: true,

	}

	cmd.SysProcAttr = attr

	return nil

}

1.首先检测 user name 是否合法

2.更新进程的启动用户属性