kill_high_mem.sh mmx_node 8192

madMAx43v3r · Oct 31, 2022 · 10de04a · 10de04a
1 parent 345a9f5
commit 10de04a
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/run_gdb_node.sh b/run_gdb_node.sh
@@ -2,4 +2,8 @@
 
 source ./activate.sh
 
+./scripts/kill_high_mem.sh mmx_node 8192 &
+
 gdb -ex=run --args ./build/mmx_node -c config/${NETWORK}/ config/node/ "${MMX_HOME}config/local/" $@
+
+kill $(jobs -p)
diff --git a/scripts/kill_high_mem.sh b/scripts/kill_high_mem.sh
@@ -0,0 +1,16 @@
+#!/bin/sh
+
+while true;
+do
+    PID=$(pgrep $1)
+    if [ ! -e $PID ]; then
+	    SIZE=$(ps -q $PID -o rss=)
+	    SIZEMB=$((SIZE/1024))
+	    if [ $SIZEMB -gt $2 ]; then
+	        printf "Maximum memory of $2 MB exceeded: $SIZEMB MB\nTerminating process ..."
+	        kill $PID
+	        exit 0
+	    fi
+    fi
+    sleep 1
+done