From ab2957d11001bc7a39c24c72ffc3869dc75f8019 Mon Sep 17 00:00:00 2001
From: Adrien Reslinger <adrien@reslinger.net>
Date: Tue, 14 Sep 2021 16:04:08 +0200
Subject: [PATCH] Update script

---
 files/usr/local/sbin/zram-swap.sh | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/files/usr/local/sbin/zram-swap.sh b/files/usr/local/sbin/zram-swap.sh
index 4f00aaa..aba4b47 100755
--- a/files/usr/local/sbin/zram-swap.sh
+++ b/files/usr/local/sbin/zram-swap.sh
@@ -36,18 +36,20 @@ case "$1" in
         else
             ZRAMID=$(cat /sys/class/zram-control/hot_add)
         fi
-        MEM=$(free -k | grep -e "^Mem:" | awk '{printf("%d\n",$2/1.2)}')
         # echo 1 > /sys/block/zram${ZRAMID}/reset
         if [ -e /sys/block/zram${ZRAMID}/comp_algorithm ]; then
             if [ $(grep -c lz4 /sys/block/zram${ZRAMID}/comp_algorithm) -eq 1 ]; then
                 echo lz4 > /sys/block/zram${ZRAMID}/comp_algorithm
             fi
         fi
+        if [ -e /sys/block/zram${ZRAMID}/backing_dev ]; then
+            if [ $(grep -v zram /proc/swaps | wc -l) -gt 1 ]; then
+                egrep -v '(zram|Filename)' /proc/swaps | cut -d\  -f1 > /sys/block/zram${ZRAMID}/backing_dev
+            fi
+        fi
+        echo $(free -k | grep -e "^Mem:" | awk '{printf("%d\n",$2*2.0)}')k > /sys/block/zram${ZRAMID}/disksize
         if [ -e /sys/block/zram${ZRAMID}/mem_limit ]; then
-            echo $[${MEM}*2]k > /sys/block/zram${ZRAMID}/disksize
-            echo ${MEM}k > /sys/block/zram${ZRAMID}/mem_limit
-        else
-            echo ${MEM}k > /sys/block/zram${ZRAMID}/disksize
+            echo $(free -k | grep -e "^Mem:" | awk '{printf("%d\n",$2*0.9)}')k > /sys/block/zram${ZRAMID}/mem_limit
         fi
         mkswap /dev/zram${ZRAMID}
         swapon -p 10 /dev/zram${ZRAMID}