add some rlimit adjustments

3 files changed, 79 insertions, 0 deletions

diff --git a/client/go/script-utils/startcbinary/setrlimit.go b/client/go/script-utils/startcbinary/setrlimit.go
new file mode 100644
index 00000000000..5d7f33d5a0e
--- /dev/null
+++ b/client/go/script-utils/startcbinary/setrlimit.go
@@ -0,0 +1,58 @@
+// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+
+//go:build !windows
+
+package startcbinary
+
+import (
+	"github.com/vespa-engine/vespa/client/go/trace"
+	"golang.org/x/sys/unix"
+	"os"
+)
+
+type ResourceId int
+
+const (
+	RLIMIT_CORE   ResourceId = unix.RLIMIT_AS
+	RLIMIT_NOFILE ResourceId = unix.RLIMIT_NOFILE
+	RLIMIT_NPROC  ResourceId = unix.RLIMIT_NPROC
+	NO_RLIMIT     uint64     = ^uint64(0)
+)
+
+func (rid ResourceId) String() string {
+	switch rid {
+	case RLIMIT_CORE:
+		return "core file size"
+	case RLIMIT_NOFILE:
+		return "open files"
+	case RLIMIT_NPROC:
+		return "max user processes"
+	}
+	return "unknown resource id"
+}
+
+func setResourceLimit(resource ResourceId, newVal uint64) {
+	var current unix.Rlimit
+	err := unix.Getrlimit(int(resource), &current)
+	if err != nil {
+		trace.Warning("Could not get current resource limit:", err)
+		return
+	}
+	wanted := current
+	if current.Max < newVal {
+		if os.Getuid() == 0 {
+			wanted.Max = newVal
+		} else {
+			newVal = current.Max
+		}
+	}
+	if current.Cur < newVal {
+		wanted.Cur = newVal
+	}
+	err = unix.Setrlimit(int(resource), &wanted)
+	if err != nil {
+		trace.Trace("Failed setting resource limit:", err)
+	} else {
+		trace.Trace("Resource limit", resource, "adjusted OK:", wanted.Cur, "/", wanted.Max)
+	}
+}
diff --git a/client/go/script-utils/startcbinary/setrlimit_windows.go b/client/go/script-utils/startcbinary/setrlimit_windows.go
new file mode 100644
index 00000000000..5bed1916246
--- /dev/null
+++ b/client/go/script-utils/startcbinary/setrlimit_windows.go
@@ -0,0 +1,18 @@
+// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+
+//go:build windows
+
+package startcbinary
+
+type ResourceId int
+
+const (
+	RLIMIT_CORE ResourceId = iota
+	RLIMIT_NOFILE
+	RLIMIT_NPROC
+	NO_RLIMIT uint64 = ^uint64(0)
+)
+
+func setResourceLimit(resource ResourceId, max uint64) {
+	// nop
+}
diff --git a/client/go/script-utils/startcbinary/tuning.go b/client/go/script-utils/startcbinary/tuning.go
index e528d44e2d0..80fec7cc6c2 100644
--- a/client/go/script-utils/startcbinary/tuning.go
+++ b/client/go/script-utils/startcbinary/tuning.go
@@ -44,4 +44,7 @@ func getThpSizeMb() int {
 
 func (spec *ProgSpec) configureTuning() {
 	spec.optionallyReduceBaseFrequency()
+	setResourceLimit(RLIMIT_CORE, NO_RLIMIT)
+	setResourceLimit(RLIMIT_NOFILE, 262144)
+	setResourceLimit(RLIMIT_NPROC, 409600)
 }