add a test for tricking the server into allocating huge buffers

1 files changed, 19 insertions, 0 deletions

diff --git a/nslcd_systemd/util_test.go b/nslcd_systemd/util_test.go
index 15147b7..89cf7cb 100644
--- a/nslcd_systemd/util_test.go
+++ b/nslcd_systemd/util_test.go
@@ -191,3 +191,22 @@ func sdNotifyHandle(sock *net.UnixConn, fn func(dat []byte, oob []byte) error) e
 		}
 	}
 }
+
+func humanizeU64(n uint64) string {
+	str := fmt.Sprintf("%d", n)
+	bts := make([]byte, len(str)+(len(str)-1)/3)
+
+	s := 0
+	b := 0
+	for s < len(str) && b < len(bts) {
+		if (s % 3 == 0 && s > 0) {
+			bts[len(bts)-1-b] = ','
+			b++
+		}
+		bts[len(bts)-1-b] = str[len(str)-1-s]
+		b++
+		s++
+	}
+
+	return string(bts)
+}