Refactor out userLookup to helper package

Also replaces user.Lookup in exec driver
This commit is contained in:
Carlos Diaz-Padron 2015-12-01 11:59:55 -08:00
parent ffe67d8910
commit 55e49506f0
3 changed files with 37 additions and 29 deletions

View File

@ -5,11 +5,10 @@ package allocdir
import (
"fmt"
"io/ioutil"
"github.com/hashicorp/nomad/helper/user-lookup"
"os"
"os/user"
"strconv"
"strings"
"syscall"
)
@ -28,7 +27,7 @@ func (d *AllocDir) dropDirPermissions(path string) error {
return nil
}
u, err := userLookup("nobody")
u, err := userlookup.Lookup("nobody")
if err != nil {
return err
}
@ -71,28 +70,3 @@ func getGid(u *user.User) (int, error) {
return gid, nil
}
// userLookup checks if the given username or uid is present in /etc/passwd
// and returns the user struct.
// If the username is not found, an error is returned.
// Credit to @creak, https://github.com/docker/docker/pull/1096
func userLookup(uid string) (*user.User, error) {
file, err := ioutil.ReadFile("/etc/passwd")
if err != nil {
return nil, err
}
for _, line := range strings.Split(string(file), "\n") {
data := strings.Split(line, ":")
if len(data) > 5 && (data[0] == uid || data[2] == uid) {
return &user.User{
Uid: data[2],
Gid: data[3],
Username: data[0],
Name: data[4],
HomeDir: data[5],
}, nil
}
}
return nil, fmt.Errorf("User not found in /etc/passwd")
}

View File

@ -18,6 +18,7 @@ import (
"github.com/hashicorp/nomad/client/driver/args"
"github.com/hashicorp/nomad/client/driver/environment"
"github.com/hashicorp/nomad/client/driver/spawn"
"github.com/hashicorp/nomad/helper/user-lookup"
"github.com/hashicorp/nomad/nomad/structs"
"github.com/opencontainers/runc/libcontainer/cgroups"
@ -122,7 +123,7 @@ func (e *LinuxExecutor) ID() (string, error) {
// runAs takes a user id as a string and looks up the user, and sets the command
// to execute as that user.
func (e *LinuxExecutor) runAs(userid string) error {
u, err := user.Lookup(userid)
u, err := userlookup.Lookup(userid)
if err != nil {
return fmt.Errorf("Failed to identify user %v: %v", userid, err)
}

View File

@ -0,0 +1,33 @@
package userlookup
import (
"fmt"
"io/ioutil"
"os/user"
"strings"
)
// Lookup checks if the given username or uid is present in /etc/passwd
// and returns the user struct.
// If the username is not found, an error is returned.
// Credit to @creak, https://github.com/docker/docker/pull/1096
func Lookup(uid string) (*user.User, error) {
file, err := ioutil.ReadFile("/etc/passwd")
if err != nil {
return nil, err
}
for _, line := range strings.Split(string(file), "\n") {
data := strings.Split(line, ":")
if len(data) > 5 && (data[0] == uid || data[2] == uid) {
return &user.User{
Uid: data[2],
Gid: data[3],
Username: data[0],
Name: data[4],
HomeDir: data[5],
}, nil
}
}
return nil, fmt.Errorf("User not found in /etc/passwd")
}