Add parseUidGid, formatUidGid functions

Signed-off-by: Oleksandr Redko <[email protected]>

Author: alexandear

.golangci.yml

@@ -151,3 +151,8 @@ linters-settings:
     - typeUnparen
     - unnamedResult
     - unnecessaryBlock
+issues:
+  exclude-rules:
+  # Allow using Uid, Gid in pkg/osutil.
+  - path: "pkg/osutil/"
+    text: "uid"

pkg/osutil/osutil_linux.go

@@ -10,7 +10,7 @@ const UnixPathMax = 108
 
 // Stat is a selection of syscall.Stat_t
 type Stat struct {
-	Uid uint32 //nolint:revive
+	Uid uint32
 	Gid uint32
 }
 

pkg/osutil/osutil_others.go

@@ -13,7 +13,7 @@ const UnixPathMax = 104
 
 // Stat is a selection of syscall.Stat_t
 type Stat struct {
-	Uid uint32 //nolint:revive
+	Uid uint32
 	Gid uint32
 }
 

pkg/osutil/osutil_windows.go

@@ -13,7 +13,7 @@ const UnixPathMax = 108
 
 // Stat is a selection of syscall.Stat_t
 type Stat struct {
-	Uid uint32 //nolint:revive
+	Uid uint32
 	Gid uint32
 }
 

pkg/osutil/user.go

@@ -16,7 +16,7 @@ import (
 
 type User struct {
 	User  string
-	Uid   uint32 //nolint:revive
+	Uid   uint32
 	Group string
 	Gid   uint32
 	Home  string
@@ -30,11 +30,9 @@ type Group struct {
 var users map[string]User
 var groups map[string]Group
 
-// regexUidGid detects valid Linux uid or gid.
-var regexUidGid = regexp.MustCompile("^[0-9]+$") //nolint:revive
-
 // regexUsername matches user and group names to be valid for `useradd`.
-// It allows a trailing '$', but it feels prudent to map those to the fallback user as well.
+// `useradd` allows names with a trailing '$', but it feels prudent to map those
+// names to the fallback user as well, so the regex does not allow them.
 var regexUsername = regexp.MustCompile("^[a-z_][a-z0-9_-]*$")
 
 // regexPath detects valid Linux path.
@@ -53,15 +51,15 @@ func LookupUser(name string) (User, error) {
 		if err != nil {
 			return User{}, err
 		}
-		uid, err := strconv.ParseUint(u.Uid, 10, 32)
+		uid, err := parseUidGid(u.Uid)
 		if err != nil {
 			return User{}, err
 		}
-		gid, err := strconv.ParseUint(u.Gid, 10, 32)
+		gid, err := parseUidGid(u.Gid)
 		if err != nil {
 			return User{}, err
 		}
-		users[name] = User{User: u.Username, Uid: uint32(uid), Group: g.Name, Gid: uint32(gid), Home: u.HomeDir}
+		users[name] = User{User: u.Username, Uid: uid, Group: g.Name, Gid: gid, Home: u.HomeDir}
 	}
 	return users[name], nil
 }
@@ -75,18 +73,18 @@ func LookupGroup(name string) (Group, error) {
 		if err != nil {
 			return Group{}, err
 		}
-		gid, err := strconv.ParseUint(g.Gid, 10, 32)
+		gid, err := parseUidGid(g.Gid)
 		if err != nil {
 			return Group{}, err
 		}
-		groups[name] = Group{Name: g.Name, Gid: uint32(gid)}
+		groups[name] = Group{Name: g.Name, Gid: gid}
 	}
 	return groups[name], nil
 }
 
 const (
 	fallbackUser = "lima"
-	fallbackUid  = 1000 //nolint:revive
+	fallbackUid  = 1000
 	fallbackGid  = 1000
 )
 
@@ -122,29 +120,29 @@ func LimaUser(warn bool) (*user.User, error) {
 				if err != nil {
 					logrus.Debug(err)
 				}
-				uid, err := strconv.ParseUint(idu, 10, 32)
+				uid, err := parseUidGid(idu)
 				if err != nil {
 					uid = fallbackUid
 				}
-				if !regexUidGid.MatchString(cache.u.Uid) {
+				if _, err := parseUidGid(cache.u.Uid); err != nil {
 					warning := fmt.Sprintf("local uid %q is not a valid Linux uid (must be integer); using %d uid instead",
 						cache.u.Uid, uid)
 					cache.warnings = append(cache.warnings, warning)
-					cache.u.Uid = fmt.Sprintf("%d", uid)
+					cache.u.Uid = formatUidGid(uid)
 				}
 				idg, err := call([]string{"id", "-g"})
 				if err != nil {
 					logrus.Debug(err)
 				}
-				gid, err := strconv.ParseUint(idg, 10, 32)
+				gid, err := parseUidGid(idg)
 				if err != nil {
 					gid = fallbackGid
 				}
-				if !regexUidGid.MatchString(cache.u.Gid) {
+				if _, err := parseUidGid(cache.u.Gid); err != nil {
 					warning := fmt.Sprintf("local gid %q is not a valid Linux gid (must be integer); using %d gid instead",
 						cache.u.Gid, gid)
 					cache.warnings = append(cache.warnings, warning)
-					cache.u.Gid = fmt.Sprintf("%d", gid)
+					cache.u.Gid = formatUidGid(gid)
 				}
 				home, err := call([]string{"cygpath", cache.u.HomeDir})
 				if err != nil {
@@ -173,3 +171,17 @@ func LimaUser(warn bool) (*user.User, error) {
 	}
 	return cache.u, cache.err
 }
+
+// parseUidGid converts string value to Linux uid or gid.
+func parseUidGid(uidOrGid string) (uint32, error) {
+	res, err := strconv.ParseUint(uidOrGid, 10, 32)
+	if err != nil {
+		return 0, err
+	}
+	return uint32(res), nil
+}
+
+// formatUidGid converts uid or gid to string value.
+func formatUidGid(uidOrGid uint32) string {
+	return strconv.FormatUint(uint64(uidOrGid), 10)
+}