From 1c99297fd7b031f120324385c2be79a7306411d5 Mon Sep 17 00:00:00 2001
From: FUJI Goro <goro@fastly.com>
Date: Thu, 15 Apr 2021 16:49:02 +0900
Subject: [PATCH] support "infinity" in ulimit values, as an alias to -1

---
 ulimit.go      | 11 +++++++++--
 ulimit_test.go |  7 +++++++
 2 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/ulimit.go b/ulimit.go
index fca0400..ae04778 100644
--- a/ulimit.go
+++ b/ulimit.go
@@ -62,6 +62,13 @@ var ulimitNameMapping = map[string]int{
 	"stack":      rlimitStack,
 }
 
+func parseUlimitValue(s string) (int64, error) {
+	if s == "infinity" {
+		return -1, nil
+	}
+	return strconv.ParseInt(s, 10, 64)
+}
+
 // ParseUlimit parses and returns a Ulimit from the specified string.
 func ParseUlimit(val string) (*Ulimit, error) {
 	parts := strings.SplitN(val, "=", 2)
@@ -81,14 +88,14 @@ func ParseUlimit(val string) (*Ulimit, error) {
 	)
 	switch limitVals := strings.Split(parts[1], ":"); len(limitVals) {
 	case 2:
-		temp, err = strconv.ParseInt(limitVals[1], 10, 64)
+		temp, err = parseUlimitValue(limitVals[1])
 		if err != nil {
 			return nil, err
 		}
 		hard = &temp
 		fallthrough
 	case 1:
-		soft, err = strconv.ParseInt(limitVals[0], 10, 64)
+		soft, err = parseUlimitValue(limitVals[0])
 		if err != nil {
 			return nil, err
 		}
diff --git a/ulimit_test.go b/ulimit_test.go
index 0095855..46a1dd9 100644
--- a/ulimit_test.go
+++ b/ulimit_test.go
@@ -56,6 +56,13 @@ func TestParseUlimitHardLessThanSoft(t *testing.T) {
 	}
 }
 
+func TestParseUlimitInfinity(t *testing.T) {
+	u1 := &Ulimit{"memlock", -1, -1}
+	if u2, _ := ParseUlimit("memlock=infinity:infinity"); *u1 != *u2 {
+		t.Fatalf("expected %q, but got %q", u1, u2)
+	}
+}
+
 func TestParseUlimitUnlimited(t *testing.T) {
 	tt := []struct {
 		in       string