1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <getopt.h>
#include <sys/swap.h>
static void usage(char *name)
{
fprintf(stderr, "Usage: %s [-p prio] <filename>\n"
" prio must be between 0 and %d\n", name, SWAP_FLAG_PRIO_MASK);
}
static int parse_prio(char *prio_str)
{
unsigned long p = strtoul(prio_str, NULL, 10);
return (p > SWAP_FLAG_PRIO_MASK)? -1 : (int)p;
}
int swapon_main(int argc, char **argv)
{
int err = 0;
int flags = 0;
int prio;
opterr = 0;
do {
int c = getopt(argc, argv, "hp:");
if (c == -1)
break;
switch (c) {
case 'p':
if (optarg != NULL)
prio = parse_prio(optarg);
else
prio = -1;
if (prio < 0) {
usage(argv[0]);
return -EINVAL;
}
flags |= SWAP_FLAG_PREFER;
flags |= (prio << SWAP_FLAG_PRIO_SHIFT) & SWAP_FLAG_PRIO_MASK;
break;
case 'h':
usage(argv[0]);
return 0;
case '?':
fprintf(stderr, "unknown option: %c\n", optopt);
return -EINVAL;
}
} while (1);
if (optind != argc - 1) {
usage(argv[0]);
return -EINVAL;
}
err = swapon(argv[argc - 1], flags);
if (err) {
fprintf(stderr, "swapon failed for %s\n", argv[argc - 1]);
}
return err;
}
|