From ecea41a0abd95ff25aae007675266f2cad19396e Mon Sep 17 00:00:00 2001
From: Michael Vines <mvines@gmail.com>
Date: Tue, 7 Aug 2018 16:16:11 -0700
Subject: [PATCH] Install EarlyOOM on testnet nodes

---
 ci/install-earlyoom.sh | 32 ++++++++++++++++++++++++++++++++
 ci/testnet-deploy.sh   |  6 ++++++
 2 files changed, 38 insertions(+)
 create mode 100755 ci/install-earlyoom.sh

diff --git a/ci/install-earlyoom.sh b/ci/install-earlyoom.sh
new file mode 100755
index 0000000000..01c27c7542
--- /dev/null
+++ b/ci/install-earlyoom.sh
@@ -0,0 +1,32 @@
+#!/bin/bash -x
+#
+# Install EarlyOOM
+#
+
+[[ $(uname) = Linux ]] || exit 1
+
+# 64 - enable signalling of processes (term, kill, oom-kill)
+# TODO: This setting will not persist across reboots
+sysrq=$(( $(cat /proc/sys/kernel/sysrq) | 64 ))
+sudo sysctl -w kernel.sysrq=$sysrq
+
+if command -v earlyoom; then
+  sudo systemctl status earlyoom
+  exit 0
+fi
+
+wget http://ftp.us.debian.org/debian/pool/main/e/earlyoom/earlyoom_1.1-2_amd64.deb
+sudo apt install --quiet --yes ./earlyoom_1.1-2_amd64.deb
+
+cat > earlyoom <<OOM
+# use the kernel OOM killer, trigger at 20% available RAM,
+EARLYOOM_ARGS="-k -m 20"
+OOM
+sudo cp earlyoom /etc/default/
+rm earlyoom
+
+sudo systemctl stop earlyoom
+sudo systemctl enable earlyoom
+sudo systemctl start earlyoom
+
+exit 0
diff --git a/ci/testnet-deploy.sh b/ci/testnet-deploy.sh
index c28caac56c..c6972a9a50 100755
--- a/ci/testnet-deploy.sh
+++ b/ci/testnet-deploy.sh
@@ -52,6 +52,12 @@ if [[ -n $LOCAL_SNAP ]]; then
 fi
 SNAP_INSTALL_CMD="sudo snap remove solana; $SNAP_INSTALL_CMD"
 
+EARLYOOM_INSTALL_CMD="\
+  wget -O install-earlyoom.sh https://raw.githubusercontent.com/solana-labs/solana/master/ci/install-earlyoom.sh; \
+  bash install-earlyoom.sh \
+"
+SNAP_INSTALL_CMD="$EARLYOOM_INSTALL_CMD; $SNAP_INSTALL_CMD"
+
 # `export SKIP_INSTALL=1` to reset the network without reinstalling the snap
 if [[ -n $SKIP_INSTALL ]]; then
   SNAP_INSTALL_CMD="echo Install skipped"