yggdrasil-network/yggdrasil-go
1
0
Fork 0
mirror of https://github.com/yggdrasil-network/yggdrasil-go.git synced 2025-04-28 14:15:06 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

one chuser() to rule them all

Browse source
This commit is contained in:
Klemens Nanni 2024-11-11 23:14:29 +03:00
parent a0bfd9da44
commit 3fede90ae1
No known key found for this signature in database
2 changed files with 33 additions and 126 deletions
Download patch file Download diff file Expand all files Collapse all files

57
cmd/yggdrasil/chuser_openbsd.go
View file

@ -1,57 +0,0 @@
//go:build openbsd
// +build openbsd
package main
import (
"fmt"
"os/user"
"strconv"
"strings"
"golang.org/x/sys/unix"
)
func chuser(input string) error {
givenUser, givenGroup, _ := strings.Cut(input, ":")
var (
err error
usr *user.User
grp *user.Group
uid, gid int
)
if usr, err = user.Lookup(givenUser); err != nil {
if usr, err = user.LookupId(givenUser); err != nil {
return err
}
}
if uid, err = strconv.Atoi(usr.Uid); err != nil {
return err
}
if givenGroup != "" {
if grp, err = user.LookupGroup(givenGroup); err != nil {
if grp, err = user.LookupGroupId(givenGroup); err != nil {
return err
}
}
gid, _ = strconv.Atoi(grp.Gid)
} else {
gid, _ = strconv.Atoi(usr.Gid)
}
if err := unix.Setgroups([]int{gid}); err != nil {
return fmt.Errorf("setgroups: %d: %v", gid, err)
}
if err := unix.Setresgid(gid, gid, gid); err != nil {
return fmt.Errorf("setresgid: %d: %v", gid, err)
}
if err := unix.Setresuid(uid, uid, uid); err != nil {
return fmt.Errorf("setresuid: %d: %v", uid, err)
}
return nil
}

102
cmd/yggdrasil/chuser_unix.go
View file

@ -1,92 +1,56 @@
//go:build aix || darwin || dragonfly || freebsd || linux || netbsd || solaris //go:build aix || darwin || dragonfly || freebsd || linux || netbsd || openbsd || solaris
// +build aix darwin dragonfly freebsd linux netbsd solaris // +build aix darwin dragonfly freebsd linux netbsd openbsd solaris
package main package main
import ( import (
"errors"
"fmt" "fmt"
"math" "os/user"
osuser "os/user"
"strconv" "strconv"
"strings" "strings"
"syscall"
"golang.org/x/sys/unix"
) )
func chuser(user string) error { func chuser(input string) error {
group := "" givenUser, givenGroup, _ := strings.Cut(input, ":")
if i := strings.IndexByte(user, ':'); i >= 0 {
user, group = user[:i], user[i+1:]
}
u := (*osuser.User)(nil) var (
g := (*osuser.Group)(nil) err error
usr *user.User
grp *user.Group
uid, gid int
)
if user != "" { if usr, err = user.Lookup(givenUser); err != nil {
if _, err := strconv.ParseUint(user, 10, 32); err == nil { if usr, err = user.LookupId(givenUser); err != nil {
u, err = osuser.LookupId(user) return err
if err != nil {
return fmt.Errorf("failed to lookup user by id %q: %v", user, err)
}
} else {
u, err = osuser.Lookup(user)
if err != nil {
return fmt.Errorf("failed to lookup user by name %q: %v", user, err)
}
} }
} }
if group != "" { if uid, err = strconv.Atoi(usr.Uid); err != nil {
if _, err := strconv.ParseUint(group, 10, 32); err == nil { return err
g, err = osuser.LookupGroupId(group)
if err != nil {
return fmt.Errorf("failed to lookup group by id %q: %v", user, err)
}
} else {
g, err = osuser.LookupGroup(group)
if err != nil {
return fmt.Errorf("failed to lookup group by name %q: %v", user, err)
}
}
} }
if g != nil { if givenGroup != "" {
gid, _ := strconv.ParseUint(g.Gid, 10, 32) if grp, err = user.LookupGroup(givenGroup); err != nil {
var err error if grp, err = user.LookupGroupId(givenGroup); err != nil {
if gid < math.MaxInt { return err
if err := syscall.Setgroups([]int{int(gid)}); err != nil {
return fmt.Errorf("failed to setgroups %d: %v", gid, err)
} }
err = syscall.Setgid(int(gid))
} else {
err = errors.New("gid too big")
} }
if err != nil { gid, _ = strconv.Atoi(grp.Gid)
return fmt.Errorf("failed to setgid %d: %v", gid, err) } else {
} gid, _ = strconv.Atoi(usr.Gid)
} else if u != nil {
gid, _ := strconv.ParseUint(u.Gid, 10, 32)
if err := syscall.Setgroups([]int{int(uint32(gid))}); err != nil {
return fmt.Errorf("failed to setgroups %d: %v", gid, err)
}
err := syscall.Setgid(int(uint32(gid)))
if err != nil {
return fmt.Errorf("failed to setgid %d: %v", gid, err)
}
} }
if u != nil { if err := unix.Setgroups([]int{gid}); err != nil {
uid, _ := strconv.ParseUint(u.Uid, 10, 32) return fmt.Errorf("setgroups: %d: %v", gid, err)
var err error }
if uid < math.MaxInt { if err := unix.Setresgid(gid, gid, gid); err != nil {
err = syscall.Setuid(int(uid)) return fmt.Errorf("setresgid: %d: %v", gid, err)
} else { }
err = errors.New("uid too big") if err := unix.Setresuid(uid, uid, uid); err != nil {
} return fmt.Errorf("setresuid: %d: %v", uid, err)
if err != nil {
return fmt.Errorf("failed to setuid %d: %v", uid, err)
}
} }
return nil return nil