Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 3 additions & 5 deletions pkg/commands/run.go
Original file line number Diff line number Diff line change
Expand Up @@ -94,11 +94,9 @@ func runCommandInExec(config *v1.Config, buildArgs *dockerfile.BuildArgs, cmdRun
replacementEnvs := buildArgs.ReplacementEnvs(config.Env)
cmd.SysProcAttr = &syscall.SysProcAttr{Setpgid: true}

u := config.User
userAndGroup := strings.Split(u, ":")
userStr, err := util.ResolveEnvironmentReplacement(userAndGroup[0], replacementEnvs, false)
userStr, err := util.ResolveEnvironmentReplacement(config.User, replacementEnvs, false)
if err != nil {
return errors.Wrapf(err, "resolving user %s", userAndGroup[0])
return errors.Wrapf(err, "resolving user %s", config.User)
}

// If specified, run the command as a specific user
Expand All @@ -109,7 +107,7 @@ func runCommandInExec(config *v1.Config, buildArgs *dockerfile.BuildArgs, cmdRun
}
}

env, err := addDefaultHOME(userStr, replacementEnvs)
env, err := addDefaultHOME(strings.Split(userStr, ":")[0], replacementEnvs)
if err != nil {
return errors.Wrap(err, "adding default HOME variable")
}
Expand Down
12 changes: 9 additions & 3 deletions pkg/commands/user.go
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@ package commands

import (
"fmt"
"os/user"
"strings"

"github.com/GoogleContainerTools/kaniko/pkg/dockerfile"
Expand All @@ -43,15 +44,20 @@ func (r *UserCommand) ExecuteCommand(config *v1.Config, buildArgs *dockerfile.Bu
return errors.Wrap(err, fmt.Sprintf("resolving user %s", userAndGroup[0]))
}

user, err := user.Current()
if err != nil {
return errors.Wrapf(err, "failed to lookup current user")
}

var groupStr = user.Gid
if len(userAndGroup) > 1 {
groupStr, err := util.ResolveEnvironmentReplacement(userAndGroup[1], replacementEnvs, false)
groupStr, err = util.ResolveEnvironmentReplacement(userAndGroup[1], replacementEnvs, false)
if err != nil {
return errors.Wrap(err, fmt.Sprintf("resolving group %s", userAndGroup[1]))
}
userStr = userStr + ":" + groupStr
}

config.User = userStr
config.User = userStr + ":" + groupStr
return nil
}

Expand Down
25 changes: 23 additions & 2 deletions pkg/commands/user_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@ limitations under the License.
package commands

import (
"fmt"
"os/user"
"testing"

Expand All @@ -37,34 +38,40 @@ var userTests = []struct {
user: "root",
userObj: &user.User{Uid: "root", Gid: "root"},
expectedUID: "root",
expectedGID: "",
},
{
user: "root-add",
userObj: &user.User{Uid: "root-add", Gid: "root"},
expectedUID: "root-add",
expectedGID: "",
},
{
user: "0",
userObj: &user.User{Uid: "0", Gid: "0"},
expectedUID: "0",
expectedGID: "",
},
{
user: "fakeUser",
userObj: &user.User{Uid: "fakeUser", Gid: "fakeUser"},
expectedUID: "fakeUser",
expectedGID: "",
},
{
user: "root",
userObj: &user.User{Uid: "root", Gid: "some"},
expectedUID: "root",
expectedGID: "",
},
{
user: "0",
userObj: &user.User{Uid: "0"},
expectedUID: "0",
expectedGID: "",
},
{
user: "root",
user: "root:f0",
userObj: &user.User{Uid: "root"},
expectedUID: "root",
expectedGID: "f0",
Expand All @@ -73,29 +80,38 @@ var userTests = []struct {
user: "0",
userObj: &user.User{Uid: "0"},
expectedUID: "0",
expectedGID: "",
},
{
user: "$envuser",
userObj: &user.User{Uid: "root", Gid: "root"},
expectedUID: "root",
expectedGID: "",
},
{
user: "root",
userObj: &user.User{Uid: "root"},
expectedUID: "root",
expectedGID: "",
},
{
user: "some",
userObj: &user.User{Uid: "some"},
expectedUID: "some",
expectedGID: "",
},
{
user: "some",
expectedUID: "some",
expectedGID: "",
},
}

func TestUpdateUser(t *testing.T) {
user, err := user.Current()
if err != nil {
t.Errorf("failed to get user %v", err)
}
for _, test := range userTests {
cfg := &v1.Config{
Env: []string{
Expand All @@ -110,6 +126,11 @@ func TestUpdateUser(t *testing.T) {
}
buildArgs := dockerfile.NewBuildArgs([]string{})
err := cmd.ExecuteCommand(cfg, buildArgs)
testutil.CheckErrorAndDeepEqual(t, false, err, test.expectedUID, cfg.User)
uid := test.expectedUID
gid := test.expectedGID
if gid == "" {
gid = user.Gid
}
testutil.CheckErrorAndDeepEqual(t, false, err, fmt.Sprintf("%s:%s", uid, gid), cfg.User)
}
}
3 changes: 3 additions & 0 deletions pkg/util/fs_util.go
Original file line number Diff line number Diff line change
Expand Up @@ -289,6 +289,9 @@ func UnTar(r io.Reader, dest string) ([]string, error) {
return nil, err
}
cleanedName := filepath.Clean(hdr.Name)
if cleanedName == "." {
continue
}
if err := ExtractFile(dest, hdr, cleanedName, tr); err != nil {
return nil, err
}
Expand Down
Loading