xref: /freebsd/sbin/swapon/swapon.c (revision ea906c4152774dff300bb26fbfc1e4188351c89a)
1 /*
2  * Copyright (c) 1980, 1993
3  *	The Regents of the University of California.  All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 4. Neither the name of the University nor the names of its contributors
14  *    may be used to endorse or promote products derived from this software
15  *    without specific prior written permission.
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27  * SUCH DAMAGE.
28  */
29 
30 #if 0
31 #ifndef lint
32 static const char copyright[] =
33 "@(#) Copyright (c) 1980, 1993\n\
34 	The Regents of the University of California.  All rights reserved.\n";
35 #endif /* not lint */
36 
37 #ifndef lint
38 static char sccsid[] = "@(#)swapon.c	8.1 (Berkeley) 6/5/93";
39 #endif /* not lint */
40 #endif
41 #include <sys/cdefs.h>
42 __FBSDID("$FreeBSD$");
43 
44 #include <sys/stat.h>
45 #include <sys/param.h>
46 #include <sys/sysctl.h>
47 #include <vm/vm_param.h>
48 
49 #include <err.h>
50 #include <errno.h>
51 #include <fstab.h>
52 #include <stdio.h>
53 #include <stdlib.h>
54 #include <string.h>
55 #include <unistd.h>
56 #include <fcntl.h>
57 #include <libutil.h>
58 
59 static void usage(void);
60 static int swap_on_off(char *name, int ignoreebusy);
61 static void swaplist(int, int, int);
62 
63 enum { SWAPON, SWAPOFF, SWAPCTL } orig_prog, which_prog = SWAPCTL;
64 
65 int
66 main(int argc, char **argv)
67 {
68 	struct fstab *fsp;
69 	char *ptr;
70 	int stat;
71 	int ch, doall;
72 	int sflag = 0, lflag = 0, hflag = 0, qflag = 0;
73 
74 	if ((ptr = strrchr(argv[0], '/')) == NULL)
75 		ptr = argv[0];
76 	if (strstr(ptr, "swapon"))
77 		which_prog = SWAPON;
78 	else if (strstr(ptr, "swapoff"))
79 		which_prog = SWAPOFF;
80 	orig_prog = which_prog;
81 
82 	doall = 0;
83 	while ((ch = getopt(argc, argv, "AadghklmqsU")) != -1) {
84 		switch(ch) {
85 		case 'A':
86 			if (which_prog == SWAPCTL) {
87 				doall = 1;
88 				which_prog = SWAPON;
89 			} else {
90 				usage();
91 			}
92 			break;
93 		case 'a':
94 			if (which_prog == SWAPON || which_prog == SWAPOFF)
95 				doall = 1;
96 			else
97 				which_prog = SWAPON;
98 			break;
99 		case 'd':
100 			if (which_prog == SWAPCTL)
101 				which_prog = SWAPOFF;
102 			else
103 				usage();
104 			break;
105 		case 'g':
106 			hflag = 'G';
107 			break;
108 		case 'h':
109 			hflag = 'H';
110 			break;
111 		case 'k':
112 			hflag = 'K';
113 			break;
114 		case 'l':
115 			lflag = 1;
116 			break;
117 		case 'm':
118 			hflag = 'M';
119 			break;
120 		case 'q':
121 			if (which_prog == SWAPON || which_prog == SWAPOFF)
122 				qflag = 1;
123 			break;
124 		case 's':
125 			sflag = 1;
126 			break;
127 		case 'U':
128 			if (which_prog == SWAPCTL) {
129 				doall = 1;
130 				which_prog = SWAPOFF;
131 			} else {
132 				usage();
133 			}
134 			break;
135 		case '?':
136 		default:
137 			usage();
138 		}
139 	}
140 	argv += optind;
141 
142 	stat = 0;
143 	if (which_prog == SWAPON || which_prog == SWAPOFF) {
144 		if (doall) {
145 			while ((fsp = getfsent()) != NULL) {
146 				if (strcmp(fsp->fs_type, FSTAB_SW))
147 					continue;
148 				if (strstr(fsp->fs_mntops, "noauto"))
149 					continue;
150 				if (swap_on_off(fsp->fs_spec, 1)) {
151 					stat = 1;
152 				} else {
153 					if (!qflag) {
154 						printf("%s: %sing %s as swap device\n",
155 						    getprogname(),
156 						    which_prog == SWAPOFF ? "remov" : "add",
157 						    fsp->fs_spec);
158 					}
159 				}
160 			}
161 		}
162 		else if (!*argv)
163 			usage();
164 		for (; *argv; ++argv) {
165 			if (swap_on_off(*argv, 0)) {
166 				stat = 1;
167 			} else if (orig_prog == SWAPCTL) {
168 				printf("%s: %sing %s as swap device\n",
169 				    getprogname(), which_prog == SWAPOFF ? "remov" : "add",
170 				    *argv);
171 			}
172 		}
173 	} else {
174 		if (lflag || sflag)
175 			swaplist(lflag, sflag, hflag);
176 		else
177 			usage();
178 	}
179 	exit(stat);
180 }
181 
182 static int
183 swap_on_off(char *name, int doingall)
184 {
185 	if ((which_prog == SWAPOFF ? swapoff(name) : swapon(name)) == -1) {
186 		switch (errno) {
187 		case EBUSY:
188 			if (!doingall)
189 				warnx("%s: device already in use", name);
190 			break;
191 		case EINVAL:
192 			if (which_prog == SWAPON)
193 				warnx("%s: NSWAPDEV limit reached", name);
194 			else if (!doingall)
195 				warn("%s", name);
196 			break;
197 		default:
198 			warn("%s", name);
199 			break;
200 		}
201 		return(1);
202 	}
203 	return(0);
204 }
205 
206 static void
207 usage(void)
208 {
209 	fprintf(stderr, "usage: %s ", getprogname());
210 	switch(orig_prog) {
211 	case SWAPON:
212 	case SWAPOFF:
213 	    fprintf(stderr, "-aq | file ...\n");
214 	    break;
215 	case SWAPCTL:
216 	    fprintf(stderr, "[-AghklmsU] [-a file ... | -d file ...]\n");
217 	    break;
218 	}
219 	exit(1);
220 }
221 
222 static void
223 sizetobuf(char *buf, size_t bufsize, int hflag, long long val, int hlen,
224     long blocksize)
225 {
226 
227 	if (hflag == 'H') {
228 		char tmp[16];
229 
230 		humanize_number(tmp, 5, (int64_t)val, "", HN_AUTOSCALE,
231 		    HN_B | HN_NOSPACE | HN_DECIMAL);
232 		snprintf(buf, bufsize, "%*s", hlen, tmp);
233 	} else {
234 		snprintf(buf, bufsize, "%*lld", hlen, val / blocksize);
235 	}
236 }
237 
238 static void
239 swaplist(int lflag, int sflag, int hflag)
240 {
241 	size_t mibsize, size;
242 	struct xswdev xsw;
243 	int hlen, mib[16], n, pagesize;
244 	long blocksize;
245 	long long total = 0;
246 	long long used = 0;
247 	long long tmp_total;
248 	long long tmp_used;
249 	char buf[32];
250 
251 	pagesize = getpagesize();
252 	switch(hflag) {
253 	case 'G':
254 	    blocksize = 1024 * 1024 * 1024;
255 	    strlcpy(buf, "1GB-blocks", sizeof(buf));
256 	    hlen = 10;
257 	    break;
258 	case 'H':
259 	    blocksize = -1;
260 	    strlcpy(buf, "Bytes", sizeof(buf));
261 	    hlen = 10;
262 	    break;
263 	case 'K':
264 	    blocksize = 1024;
265 	    strlcpy(buf, "1kB-blocks", sizeof(buf));
266 	    hlen = 10;
267 	    break;
268 	case 'M':
269 	    blocksize = 1024 * 1024;
270 	    strlcpy(buf, "1MB-blocks", sizeof(buf));
271 	    hlen = 10;
272 	    break;
273 	default:
274 	    getbsize(&hlen, &blocksize);
275 	    snprintf(buf, sizeof(buf), "%ld-blocks", blocksize);
276 	    break;
277 	}
278 
279 	mibsize = sizeof mib / sizeof mib[0];
280 	if (sysctlnametomib("vm.swap_info", mib, &mibsize) == -1)
281 		err(1, "sysctlnametomib()");
282 
283 	if (lflag) {
284 		printf("%-13s %*s %*s\n",
285 		    "Device:",
286 		    hlen, buf,
287 		    hlen, "Used:");
288 	}
289 
290 	for (n = 0; ; ++n) {
291 		mib[mibsize] = n;
292 		size = sizeof xsw;
293 		if (sysctl(mib, mibsize + 1, &xsw, &size, NULL, 0) == -1)
294 			break;
295 		if (xsw.xsw_version != XSWDEV_VERSION)
296 			errx(1, "xswdev version mismatch");
297 
298 		tmp_total = (long long)xsw.xsw_nblks * pagesize;
299 		tmp_used  = (long long)xsw.xsw_used * pagesize;
300 		total += tmp_total;
301 		used  += tmp_used;
302 		if (lflag) {
303 			sizetobuf(buf, sizeof(buf), hflag, tmp_total, hlen,
304 			    blocksize);
305 			printf("/dev/%-8s %s ", devname(xsw.xsw_dev, S_IFCHR),
306 			    buf);
307 			sizetobuf(buf, sizeof(buf), hflag, tmp_used, hlen,
308 			    blocksize);
309 			printf("%s\n", buf);
310 		}
311 	}
312 	if (errno != ENOENT)
313 		err(1, "sysctl()");
314 
315 	if (sflag) {
316 		sizetobuf(buf, sizeof(buf), hflag, total, hlen, blocksize);
317 		printf("Total:        %s ", buf);
318 		sizetobuf(buf, sizeof(buf), hflag, used, hlen, blocksize);
319 		printf("%s\n", buf);
320 	}
321 }
322 
323