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
57 changes: 46 additions & 11 deletions user/user.go
Original file line number Diff line number Diff line change
Expand Up @@ -56,13 +56,17 @@ type IDMap struct {
Count int64
}

func parseLine(line []byte, v ...any) {
parseParts(bytes.Split(line, []byte(":")), v...)
func parseLine(line []byte, v ...any) (ok bool, _ error) {
line = bytes.TrimSpace(line)
if len(line) == 0 {
return false, nil
}
return parseParts(bytes.Split(line, []byte(":")), v...)
}

func parseParts(parts [][]byte, v ...any) {
func parseParts(parts [][]byte, v ...any) (ok bool, _ error) {
if len(parts) == 0 {
return
return false, nil
}

for i, p := range parts {
Expand All @@ -78,10 +82,24 @@ func parseParts(parts [][]byte, v ...any) {
case *string:
*e = string(p)
case *int:
// "numbers", with conversion errors ignored because of some misbehaving configuration files.
*e, _ = strconv.Atoi(string(p))
if len(p) != 0 {
n, ok, err := parseNumeric(string(p))
if err != nil {
return true, fmt.Errorf("parsing integer field %d: %w", i, err)
}
if !ok {
return true, fmt.Errorf("parsing integer field %d: %q is not numeric", i, p)
}
*e = n
}
case *int64:
*e, _ = strconv.ParseInt(string(p), 10, 64)
if len(p) != 0 {
n, err := strconv.ParseInt(string(p), 10, 64)
if err != nil {
return true, fmt.Errorf("parsing integer field %d: %w", i, err)
}
*e = n
}
case *[]string:
// Comma-separated lists.
if len(p) != 0 {
Expand All @@ -94,6 +112,7 @@ func parseParts(parts [][]byte, v ...any) {
panic(fmt.Sprintf("parseLine only accepts {*string, *int, *int64, *[]string} as arguments! %#v is not a pointer!", e))
}
}
return true, nil
}

func ParsePasswdFile(path string) ([]User, error) {
Expand Down Expand Up @@ -135,7 +154,11 @@ func ParsePasswdFilter(r io.Reader, filter func(User) bool) ([]User, error) {
// root:x:0:0:root:/root:/bin/bash
// adm:x:3:4:adm:/var/adm:/bin/false
p := User{}
parseLine(line, &p.Name, &p.Pass, &p.Uid, &p.Gid, &p.Gecos, &p.Home, &p.Shell)
ok, err := parseLine(line, &p.Name, &p.Pass, &p.Uid, &p.Gid, &p.Gecos, &p.Home, &p.Shell)
if err != nil || !ok {
// Skip malformed and empty lines, and don't consider them.
continue
}
Comment on lines +157 to +161

Copy link
Copy Markdown
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Possibly also skip zero-records;

if p == (User{}) {
    continue
}


if filter == nil || filter(p) {
out = append(out, p)
Expand Down Expand Up @@ -194,7 +217,11 @@ func ParseGroupFilter(r io.Reader, filter func(Group) bool) ([]Group, error) {
// root:x:0:root
// adm:x:4:root,adm,daemon
p := Group{}
parseLine(line, &p.Name, &p.Pass, &p.Gid, &p.List)
ok, err := parseLine(line, &p.Name, &p.Pass, &p.Gid, &p.List)
if err != nil || !ok {
// Skip malformed and empty lines, and don't consider them.
continue
}

if filter == nil || filter(p) {
out = append(out, p)
Comment on lines +220 to 227

Copy link
Copy Markdown
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Here as well, but Group is not comparable (due to the List []string)

Expand Down Expand Up @@ -508,7 +535,11 @@ func ParseSubIDFilter(r io.Reader, filter func(SubID) bool) ([]SubID, error) {

// see: man 5 subuid
p := SubID{}
parseLine(line, &p.Name, &p.SubID, &p.Count)
ok, err := parseLine(line, &p.Name, &p.SubID, &p.Count)
if err != nil || !ok {
// Skip malformed and empty lines, and don't consider them.
continue
}

if filter == nil || filter(p) {
out = append(out, p)
Expand Down Expand Up @@ -556,7 +587,11 @@ func ParseIDMapFilter(r io.Reader, filter func(IDMap) bool) ([]IDMap, error) {

// see: man 7 user_namespaces
p := IDMap{}
parseParts(bytes.Fields(line), &p.ID, &p.ParentID, &p.Count)
ok, err := parseParts(bytes.Fields(line), &p.ID, &p.ParentID, &p.Count)
if err != nil || !ok {
// Skip malformed and empty lines, and don't consider them.
continue
}

if filter == nil || filter(p) {
out = append(out, p)
Expand Down
43 changes: 35 additions & 8 deletions user/user_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -21,42 +21,69 @@ func TestParseLine(t *testing.T) {
d int
)

parseLine([]byte(""), &a, &b)
_, err := parseLine([]byte(""), &a, &b)
if err != nil {
t.Fatal(err)
}
if a != "" || b != "" {
t.Fatalf("a and b should be empty ('%v', '%v')", a, b)
}

parseLine([]byte("a"), &a, &b)
_, err = parseLine([]byte("a"), &a, &b)
if err != nil {
t.Fatal(err)
}
if a != "a" || b != "" {
t.Fatalf("a should be 'a' and b should be empty ('%v', '%v')", a, b)
}

parseLine([]byte("bad boys:corny cows"), &a, &b)
_, err = parseLine([]byte("bad boys:corny cows"), &a, &b)
if err != nil {
t.Fatal(err)
}
if a != "bad boys" || b != "corny cows" {
t.Fatalf("a should be 'bad boys' and b should be 'corny cows' ('%v', '%v')", a, b)
}

parseLine([]byte(""), &c)
ok, err := parseLine([]byte(""), &c)
if err != nil {
t.Fatal(err)
}
if ok {
t.Fatalf("ok should be false for empty lines")
}
if len(c) != 0 {
t.Fatalf("c should be empty (%#v)", c)
}

parseLine([]byte("d,e,f:g:h:i,j,k"), &c, &a, &b, &c)
_, err = parseLine([]byte("d,e,f:g:h:i,j,k"), &c, &a, &b, &c)
if err != nil {
t.Fatal(err)
}
if a != "g" || b != "h" || len(c) != 3 || c[0] != "i" || c[1] != "j" || c[2] != "k" {
t.Fatalf("a should be 'g', b should be 'h', and c should be ['i','j','k'] ('%v', '%v', '%#v')", a, b, c)
}

parseLine([]byte("::::::::::"), &a, &b, &c)
_, err = parseLine([]byte("::::::::::"), &a, &b, &c)
if err != nil {
t.Fatal(err)
}
if a != "" || b != "" || len(c) != 0 {
t.Fatalf("a, b, and c should all be empty ('%v', '%v', '%#v')", a, b, c)
}

parseLine([]byte("not a number"), &d)
_, err = parseLine([]byte("not a number"), &d)
if err == nil {
t.Fatal("expected an error")
}
if d != 0 {
t.Fatalf("d should be 0 (%v)", d)
}

parseLine([]byte("b:12:c"), &a, &d, &b)
_, err = parseLine([]byte("b:12:c"), &a, &d, &b)
if err != nil {
t.Fatal(err)
}
if a != "b" || b != "c" || d != 12 {
t.Fatalf("a should be 'b' and b should be 'c', and d should be 12 ('%v', '%v', %v)", a, b, d)
}
Expand Down
Loading