Adding swapon and swapoff

3 files changed, 65 insertions, 0 deletions

diff --git a/toys.h b/toys.h
index 5ed8089f..504e0dcb 100644
--- a/toys.h
+++ b/toys.h
@@ -29,6 +29,7 @@
 #include <sys/stat.h>
 #include <sys/statvfs.h>
 #include <sys/sysinfo.h>
+#include <sys/swap.h>
 #include <sys/types.h>
 #include <sys/wait.h>
 #include <unistd.h>
diff --git a/toys/swapoff.c b/toys/swapoff.c
new file mode 100644
index 00000000..fba0de83
--- /dev/null
+++ b/toys/swapoff.c
@@ -0,0 +1,26 @@
+/* vi: set sw=4 ts=4:
+ *
+ * swapoff.c - Disable region for swapping
+ *
+ * Copyright 2012 Elie De Brauwer <eliedebrauwer@gmail.com>
+ *
+ * Not in SUSv4.
+
+USE_SWAPOFF(NEWTOY(swapoff, "<1>1", TOYFLAG_BIN|TOYFLAG_NEEDROOT))
+
+config SWAPOFF
+	bool "swapoff"
+	default y
+	help
+	  usage: swapoff swapregion
+
+	  Disable swapping on a given swapregion.
+*/
+
+#include "toys.h"
+
+void swapoff_main(void)
+{
+	if (swapoff(toys.optargs[0]))
+		perror_exit("failed to remove swaparea");
+}
diff --git a/toys/swapon.c b/toys/swapon.c
new file mode 100644
index 00000000..36e4507b
--- /dev/null
+++ b/toys/swapon.c
@@ -0,0 +1,38 @@
+/* vi: set sw=4 ts=4:
+ *
+ * swapon.c - Enable region for swapping
+ *
+ * Copyright 2012 Elie De Brauwer <eliedebrauwer@gmail.com>
+ *
+ * Not in SUSv4.
+
+USE_SWAPON(NEWTOY(swapon, "<1>1p#<0>32767", TOYFLAG_BIN|TOYFLAG_NEEDROOT))
+
+config SWAPON
+	bool "swapon"
+	default y
+	help
+	  usage: swapon swapregion
+
+	  Enable swapping on a given swapregion.
+*/
+
+#include "toys.h"
+
+DEFINE_GLOBALS(
+	long priority;
+)
+
+#define TT this.swapon
+
+void swapon_main(void)
+{
+	int flags = 0;
+
+	if (toys.optflags & 1)
+		flags = SWAP_FLAG_PREFER |
+			((TT.priority & SWAP_FLAG_PRIO_MASK) << SWAP_FLAG_PRIO_SHIFT);
+
+	if (swapon(toys.optargs[0], flags))
+		perror_exit("failed to enable swaparea");
+}