From 9d926a44d595f1a32cb87aba524fd044ac36416c Mon Sep 17 00:00:00 2001
From: Wenchao Hao <haowenchao22@gmail.com>
Date: Tue, 19 Nov 2024 21:52:41 +0800
Subject: [PATCH] update

Signed-off-by: Wenchao Hao <haowenchao22@gmail.com>
---
 run.sh      | 38 ++++++++++++++++++++++++++++++++++++++
 test_swap.c | 16 +++++++++++++++-
 2 files changed, 53 insertions(+), 1 deletion(-)
 create mode 100644 run.sh

diff --git a/run.sh b/run.sh
new file mode 100644
index 0000000..dfe563f
--- /dev/null
+++ b/run.sh
@@ -0,0 +1,38 @@
+echo always > /sys/kernel/mm/transparent_hugepage/hugepages-64kB/enabled
+
+tmpfile=$(mktemp)
+
+pgfault1=$(cat /proc/vmstat  | grep pgfault | awk '{print $2}')
+
+./test_swap > $tmpfile
+
+pgfault2=$(cat /proc/vmstat  | grep pgfault | awk '{print $2}')
+
+swapout=$(head -n 1 $tmpfile | awk '{print $4}')
+swapin=$(tail -n 1 $tmpfile | awk '{print $4}')
+
+cat /sys/block/zram0/mm_stat > $tmpfile
+
+total_size=$(cat $tmpfile | awk '{print $1}')
+used_size=$(cat $tmpfile | awk '{print $2}')
+comp_ratio=`echo "scale=4; $total_size / $used_size" | bc`
+
+
+anon_fault_alloc=$(cat /sys/kernel/mm/transparent_hugepage/hugepages-64kB/stats/anon_fault_alloc)
+anon_fault_fallback=$(cat /sys/kernel/mm/transparent_hugepage/hugepages-64kB/stats/anon_fault_fallback)
+anon_total=$(($anon_fault_fallback + $anon_fault_alloc))
+folio_alloc_success_ratio=`echo "scale=4; $anon_fault_alloc / $anon_total" | bc`
+
+swpout_alloc=$(cat /sys/kernel/mm/transparent_hugepage/hugepages-64kB/stats/swpout)
+swpout_fault_fallback=$(cat /sys/kernel/mm/transparent_hugepage/hugepages-64kB/stats/swpout_fallback)
+swpout_total=$(($swpout_fault_fallback + $swpout_alloc))
+swap_success_ratio=`echo "scale=4; $swpout_alloc / $swpout_total" | bc`
+
+echo swapout $swapout
+echo swapin $swapin
+echo comp_ratio $comp_ratio
+echo pagefault $((pgfault2-pgfault1))
+echo folio_alloc_success_ratio $folio_alloc_success_ratio
+echo swap_success_ratio $swap_success_ratio
+
+killall test_swap
diff --git a/test_swap.c b/test_swap.c
index d9f1694..7d650c2 100644
--- a/test_swap.c
+++ b/test_swap.c
@@ -22,6 +22,7 @@ int main()
 {
 	int i;
 	char *c;
+	pid_t pid;
 	unsigned long *p;
 	struct timeval tv_b, tv_e;;
 
@@ -35,7 +36,7 @@ int main()
 
 	/* fill data */
 	for (i = 0; i < SIZE / 8; i += 8)
-		*(p + i) = i;
+		*(p + i) = (unsigned long)rand() << 32 | rand();
 
 	/* trigger pageout and get time of pageout */
 	gettimeofday(&tv_b, NULL);
@@ -54,5 +55,18 @@ int main()
 	printf("swp in  bandwidth: %ld bytes/ms\n",
 			SIZE/(tv_to_ms(tv_e) - tv_to_ms(tv_b)));
 
+	pid = fork();
+
+	if (pid) {
+		munmap(p, SIZE);
+		return 0;
+	}
+
+	madvise(p, SIZE, MADV_PAGEOUT);
+
+	while (1) {
+		sleep(10);
+	}
+
 	return 0;
 }