shadowsocks: set GOMEMLIMIT env

golang/go#39174 https://pkg.go.dev/runtime#hdr-Environment_Variables
tsl0922 · Sep 25, 2022 · b18ea97 · b18ea97
1 parent bafa6b6
commit b18ea97
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/trunk/user/shadowsocks/scripts/shadowsocks.sh b/trunk/user/shadowsocks/scripts/shadowsocks.sh
@@ -67,11 +67,22 @@ find_bin() {
 	echo $ret
 }
 
+go_mem_limit() {
+	mem_limit=$(nvram get ss_cgoups_mem_s)
+	if [ -n "$mem_limit" ]; then
+		echo "${mem_limit}iB"
+	else
+		mem_total=$(grep MemTotal /proc/meminfo | awk '{print $2}')
+		echo "$(($mem_total / 2))KiB"
+	fi
+}
+
 run_bin() {
 	(if [ "$(nvram get ss_cgroups)" = "1" ]; then
 	 	echo 0 > /sys/fs/cgroup/cpu/$NAME/tasks
 	 	echo 0 > /sys/fs/cgroup/memory/$NAME/tasks
 	 fi
+	 export GOMEMLIMIT=$(go_mem_limit)
 	 "$@" > /dev/null 2>&1
 	) &
 }

diff --git a/trunk/user/shadowsocks/scripts/ssr-monitor b/trunk/user/shadowsocks/scripts/ssr-monitor
@@ -28,11 +28,22 @@ log() {
 	echo "$(date "+%Y-%m-%d %H:%M:%S") $@" >> "/tmp/ssrplus.log"
 }
 
+go_mem_limit() {
+	mem_limit=$(nvram get ss_cgoups_mem_s)
+	if [ -n "$mem_limit" ]; then
+		echo "${mem_limit}iB"
+	else
+		mem_total=$(grep MemTotal /proc/meminfo | awk '{print $2}')
+		echo "$(($mem_total / 2))KiB"
+	fi
+}
+
 run_bin() {
 	(if [ "$(nvram get ss_cgroups)" = "1" ]; then
 	 	echo 0 > /sys/fs/cgroup/cpu/$NAME/tasks
 	 	echo 0 > /sys/fs/cgroup/memory/$NAME/tasks
 	 fi
+	 export GOMEMLIMIT=$(go_mem_limit)
 	 "$@" > /dev/null 2>&1
 	) &
 }