mirror of git://git.suckless.org/ubase
68 lines
1.2 KiB
C
68 lines
1.2 KiB
C
/* See LICENSE file for copyright and license details. */
|
|
#include <sys/swap.h>
|
|
#include <mntent.h>
|
|
#include <errno.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include "util.h"
|
|
|
|
static void
|
|
usage(void)
|
|
{
|
|
eprintf("usage: %s [-dp] [-a] device\n", argv0);
|
|
}
|
|
|
|
int
|
|
main(int argc, char *argv[])
|
|
{
|
|
int i;
|
|
int ret = EXIT_SUCCESS;
|
|
int flags = 0;
|
|
int all = 0;
|
|
|
|
ARGBEGIN {
|
|
case 'a':
|
|
all = 1;
|
|
break;
|
|
case 'd':
|
|
flags |= SWAP_FLAG_DISCARD;
|
|
break;
|
|
case 'p':
|
|
flags |= SWAP_FLAG_PREFER;
|
|
break;
|
|
default:
|
|
usage();
|
|
} ARGEND;
|
|
|
|
if (!all && argc < 1)
|
|
usage();
|
|
|
|
if (all) {
|
|
struct mntent *me = NULL;
|
|
FILE *fp;
|
|
|
|
fp = setmntent("/etc/fstab", "r");
|
|
while ((me = getmntent(fp)) != NULL) {
|
|
if (strcmp(me->mnt_type, MNTTYPE_SWAP) == 0
|
|
&& (hasmntopt(me, MNTOPT_NOAUTO) == NULL)) {
|
|
if (swapon(me->mnt_fsname, flags) < 0) {
|
|
fprintf(stderr, "swapon %s: %s\n",
|
|
me->mnt_fsname, strerror(errno));
|
|
ret = EXIT_FAILURE;
|
|
}
|
|
}
|
|
}
|
|
endmntent(fp);
|
|
} else {
|
|
for (i = 0; i < argc; i++) {
|
|
if (swapon(argv[i], flags) < 0) {
|
|
fprintf(stderr, "swapon %s: %s\n",
|
|
argv[i], strerror(errno));
|
|
ret = EXIT_FAILURE;
|
|
}
|
|
}
|
|
}
|
|
return ret;
|
|
}
|