]> CyberLeo.Net >> Repos - FreeBSD/stable/10.git/blob - usr.sbin/jls/jls.c
MFC r279081:
[FreeBSD/stable/10.git] / usr.sbin / jls / jls.c
1 /*-
2  * Copyright (c) 2003 Mike Barcroft <mike@FreeBSD.org>
3  * Copyright (c) 2008 Bjoern A. Zeeb <bz@FreeBSD.org>
4  * Copyright (c) 2009 James Gritton <jamie@FreeBSD.org>
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26  * SUCH DAMAGE.
27  */
28
29 #include <sys/cdefs.h>
30 __FBSDID("$FreeBSD$");
31
32 #include <sys/param.h>
33 #include <sys/jail.h>
34 #include <sys/socket.h>
35 #include <sys/sysctl.h>
36
37 #include <arpa/inet.h>
38 #include <netinet/in.h>
39
40 #include <err.h>
41 #include <errno.h>
42 #include <jail.h>
43 #include <limits.h>
44 #include <stdio.h>
45 #include <stdlib.h>
46 #include <string.h>
47 #include <unistd.h>
48
49 #define JP_USER         0x01000000
50 #define JP_OPT          0x02000000
51
52 #define PRINT_DEFAULT   0x01
53 #define PRINT_HEADER    0x02
54 #define PRINT_NAMEVAL   0x04
55 #define PRINT_QUOTED    0x08
56 #define PRINT_SKIP      0x10
57 #define PRINT_VERBOSE   0x20
58 #define PRINT_JAIL_NAME 0x40
59
60 static struct jailparam *params;
61 static int *param_parent;
62 static int nparams;
63 #ifdef INET6
64 static int ip6_ok;
65 #endif
66 #ifdef INET
67 static int ip4_ok;
68 #endif
69
70 static int add_param(const char *name, void *value, size_t valuelen,
71                 struct jailparam *source, unsigned flags);
72 static int sort_param(const void *a, const void *b);
73 static char *noname(const char *name);
74 static char *nononame(const char *name);
75 static int print_jail(int pflags, int jflags);
76 static void quoted_print(char *str);
77
78 int
79 main(int argc, char **argv)
80 {
81         char *dot, *ep, *jname;
82         int c, i, jflags, jid, lastjid, pflags, spc;
83
84         jname = NULL;
85         pflags = jflags = jid = 0;
86         while ((c = getopt(argc, argv, "adj:hNnqsv")) >= 0)
87                 switch (c) {
88                 case 'a':
89                 case 'd':
90                         jflags |= JAIL_DYING;
91                         break;
92                 case 'j':
93                         jid = strtoul(optarg, &ep, 10);
94                         if (!jid || *ep) {
95                                 jid = 0;
96                                 jname = optarg;
97                         }
98                         break;
99                 case 'h':
100                         pflags = (pflags & ~(PRINT_SKIP | PRINT_VERBOSE)) |
101                             PRINT_HEADER;
102                         break;
103                 case 'N':
104                         pflags |= PRINT_JAIL_NAME;
105                         break;
106                 case 'n':
107                         pflags = (pflags & ~PRINT_VERBOSE) | PRINT_NAMEVAL;
108                         break;
109                 case 'q':
110                         pflags |= PRINT_QUOTED;
111                         break;
112                 case 's':
113                         pflags = (pflags & ~(PRINT_HEADER | PRINT_VERBOSE)) |
114                             PRINT_NAMEVAL | PRINT_QUOTED | PRINT_SKIP;
115                         break;
116                 case 'v':
117                         pflags = (pflags &
118                             ~(PRINT_HEADER | PRINT_NAMEVAL | PRINT_SKIP)) |
119                             PRINT_VERBOSE;
120                         break;
121                 default:
122                         errx(1, "usage: jls [-dhNnqv] [-j jail] [param ...]");
123                 }
124
125 #ifdef INET6
126         ip6_ok = feature_present("inet6");
127 #endif
128 #ifdef INET
129         ip4_ok = feature_present("inet");
130 #endif
131
132         /* Add the parameters to print. */
133         if (optind == argc) {
134                 if (pflags & (PRINT_HEADER | PRINT_NAMEVAL))
135                         add_param("all", NULL, (size_t)0, NULL, JP_USER);
136                 else if (pflags & PRINT_VERBOSE) {
137                         add_param("jid", NULL, (size_t)0, NULL, JP_USER);
138                         add_param("host.hostname", NULL, (size_t)0, NULL,
139                             JP_USER);
140                         add_param("path", NULL, (size_t)0, NULL, JP_USER);
141                         add_param("name", NULL, (size_t)0, NULL, JP_USER);
142                         add_param("dying", NULL, (size_t)0, NULL, JP_USER);
143                         add_param("cpuset.id", NULL, (size_t)0, NULL, JP_USER);
144 #ifdef INET
145                         if (ip4_ok)
146                                 add_param("ip4.addr", NULL, (size_t)0, NULL,
147                                     JP_USER);
148 #endif
149 #ifdef INET6
150                         if (ip6_ok)
151                                 add_param("ip6.addr", NULL, (size_t)0, NULL,
152                                     JP_USER | JP_OPT);
153 #endif
154                 } else {
155                         pflags |= PRINT_DEFAULT;
156                         if (pflags & PRINT_JAIL_NAME)
157                                 add_param("name", NULL, (size_t)0, NULL, JP_USER);
158                         else
159                                 add_param("jid", NULL, (size_t)0, NULL, JP_USER);
160 #ifdef INET
161                         if (ip4_ok)
162                                 add_param("ip4.addr", NULL, (size_t)0, NULL,
163                                     JP_USER);
164 #endif
165                         add_param("host.hostname", NULL, (size_t)0, NULL,
166                             JP_USER);
167                         add_param("path", NULL, (size_t)0, NULL, JP_USER);
168                 }
169         } else {
170                 pflags &= ~PRINT_VERBOSE;
171                 while (optind < argc)
172                         add_param(argv[optind++], NULL, (size_t)0, NULL,
173                             JP_USER);
174         }
175
176         if (pflags & PRINT_SKIP) {
177                 /* Check for parameters with jailsys parents. */
178                 for (i = 0; i < nparams; i++) {
179                         if ((params[i].jp_flags & JP_USER) &&
180                             (dot = strchr(params[i].jp_name, '.'))) {
181                                 *dot = 0;
182                                 param_parent[i] = add_param(params[i].jp_name,
183                                     NULL, (size_t)0, NULL, JP_OPT);
184                                 *dot = '.';
185                         }
186                 }
187         }
188
189         /* Add the index key parameters. */
190         if (jid != 0)
191                 add_param("jid", &jid, sizeof(jid), NULL, 0);
192         else if (jname != NULL)
193                 add_param("name", jname, strlen(jname), NULL, 0);
194         else
195                 add_param("lastjid", &lastjid, sizeof(lastjid), NULL, 0);
196
197         /* Print a header line if requested. */
198         if (pflags & PRINT_VERBOSE)
199                 printf("   JID  Hostname                      Path\n"
200                        "        Name                          State\n"
201                        "        CPUSetID\n"
202                        "        IP Address(es)\n");
203         else if (pflags & PRINT_DEFAULT)
204                 if (pflags & PRINT_JAIL_NAME)
205                         printf(" JID             IP Address      "
206                             "Hostname                      Path\n");
207                 else
208                         printf("   JID  IP Address      "
209                             "Hostname                      Path\n");
210         else if (pflags & PRINT_HEADER) {
211                 for (i = spc = 0; i < nparams; i++)
212                         if (params[i].jp_flags & JP_USER) {
213                                 if (spc)
214                                         putchar(' ');
215                                 else
216                                         spc = 1;
217                                 fputs(params[i].jp_name, stdout);
218                         }
219                 putchar('\n');
220         }
221
222         /* Fetch the jail(s) and print the paramters. */
223         if (jid != 0 || jname != NULL) {
224                 if (print_jail(pflags, jflags) < 0)
225                         errx(1, "%s", jail_errmsg);
226         } else {
227                 for (lastjid = 0;
228                      (lastjid = print_jail(pflags, jflags)) >= 0; )
229                         ;
230                 if (errno != 0 && errno != ENOENT)
231                         errx(1, "%s", jail_errmsg);
232         }
233
234         return (0);
235 }
236
237 static int
238 add_param(const char *name, void *value, size_t valuelen,
239     struct jailparam *source, unsigned flags)
240 {
241         struct jailparam *param, *tparams;
242         int i, tnparams;
243
244         static int paramlistsize;
245
246         /* The pseudo-parameter "all" scans the list of available parameters. */
247         if (!strcmp(name, "all")) {
248                 tnparams = jailparam_all(&tparams);
249                 if (tnparams < 0)
250                         errx(1, "%s", jail_errmsg);
251                 qsort(tparams, (size_t)tnparams, sizeof(struct jailparam),
252                     sort_param);
253                 for (i = 0; i < tnparams; i++)
254                         add_param(tparams[i].jp_name, NULL, (size_t)0,
255                             tparams + i, flags);
256                 free(tparams);
257                 return -1;
258         }
259
260         /* Check for repeat parameters. */
261         for (i = 0; i < nparams; i++)
262                 if (!strcmp(name, params[i].jp_name)) {
263                         if (value != NULL && jailparam_import_raw(params + i,
264                             value, valuelen) < 0)
265                                 errx(1, "%s", jail_errmsg);
266                         params[i].jp_flags |= flags;
267                         if (source != NULL)
268                                 jailparam_free(source, 1);
269                         return i;
270                 }
271
272         /* Make sure there is room for the new param record. */
273         if (!nparams) {
274                 paramlistsize = 32;
275                 params = malloc(paramlistsize * sizeof(*params));
276                 param_parent = malloc(paramlistsize * sizeof(*param_parent));
277                 if (params == NULL || param_parent == NULL)
278                         err(1, "malloc");
279         } else if (nparams >= paramlistsize) {
280                 paramlistsize *= 2;
281                 params = realloc(params, paramlistsize * sizeof(*params));
282                 param_parent = realloc(param_parent,
283                     paramlistsize * sizeof(*param_parent));
284                 if (params == NULL || param_parent == NULL)
285                         err(1, "realloc");
286         }
287
288         /* Look up the parameter. */
289         param_parent[nparams] = -1;
290         param = params + nparams++;
291         if (source != NULL) {
292                 *param = *source;
293                 param->jp_flags |= flags;
294                 return param - params;
295         }
296         if (jailparam_init(param, name) < 0)
297                 errx(1, "%s", jail_errmsg);
298         param->jp_flags = flags;
299         if ((value != NULL ? jailparam_import_raw(param, value, valuelen)
300              : jailparam_import(param, value)) < 0) {
301                 if (flags & JP_OPT) {
302                         nparams--;
303                         return (-1);
304                 }
305                 errx(1, "%s", jail_errmsg);
306         }
307         return param - params;
308 }
309
310 static int
311 sort_param(const void *a, const void *b)
312 {
313         const struct jailparam *parama, *paramb;
314         char *ap, *bp;
315
316         /* Put top-level parameters first. */
317         parama = a;
318         paramb = b;
319         ap = strchr(parama->jp_name, '.');
320         bp = strchr(paramb->jp_name, '.');
321         if (ap && !bp)
322                 return (1);
323         if (bp && !ap)
324                 return (-1);
325         return (strcmp(parama->jp_name, paramb->jp_name));
326 }
327
328 static char *
329 noname(const char *name)
330 {
331         char *nname, *p;
332
333         nname = malloc(strlen(name) + 3);
334         if (nname == NULL)
335                 err(1, "malloc");
336         p = strrchr(name, '.');
337         if (p != NULL)
338                 sprintf(nname, "%.*s.no%s", (int)(p - name), name, p + 1);
339         else
340                 sprintf(nname, "no%s", name);
341         return nname;
342 }
343
344 static char *
345 nononame(const char *name)
346 {
347         char *nname, *p;
348
349         p = strrchr(name, '.');
350         if (strncmp(p ? p + 1 : name, "no", 2))
351                 return NULL;
352         nname = malloc(strlen(name) - 1);
353         if (nname == NULL)
354                 err(1, "malloc");
355         if (p != NULL)
356                 sprintf(nname, "%.*s.%s", (int)(p - name), name, p + 3);
357         else
358                 strcpy(nname, name + 2);
359         return nname;
360 }
361
362 static int
363 print_jail(int pflags, int jflags)
364 {
365         char *nname;
366         char **param_values;
367         int i, ai, jid, count, n, spc;
368         char ipbuf[INET6_ADDRSTRLEN];
369
370         jid = jailparam_get(params, nparams, jflags);
371         if (jid < 0)
372                 return jid;
373         if (pflags & PRINT_VERBOSE) {
374                 printf("%6d  %-29.29s %.74s\n"
375                        "%6s  %-29.29s %.74s\n"
376                        "%6s  %-6d\n",
377                     *(int *)params[0].jp_value,
378                     (char *)params[1].jp_value,
379                     (char *)params[2].jp_value,
380                     "",
381                     (char *)params[3].jp_value,
382                     *(int *)params[4].jp_value ? "DYING" : "ACTIVE",
383                     "",
384                     *(int *)params[5].jp_value);
385                 n = 6;
386 #ifdef INET
387                 if (ip4_ok && !strcmp(params[n].jp_name, "ip4.addr")) {
388                         count = params[n].jp_valuelen / sizeof(struct in_addr);
389                         for (ai = 0; ai < count; ai++)
390                                 if (inet_ntop(AF_INET,
391                                     &((struct in_addr *)params[n].jp_value)[ai],
392                                     ipbuf, sizeof(ipbuf)) == NULL)
393                                         err(1, "inet_ntop");
394                                 else
395                                         printf("%6s  %-15.15s\n", "", ipbuf);
396                         n++;
397                 }
398 #endif
399 #ifdef INET6
400                 if (ip6_ok && !strcmp(params[n].jp_name, "ip6.addr")) {
401                         count = params[n].jp_valuelen / sizeof(struct in6_addr);
402                         for (ai = 0; ai < count; ai++)
403                                 if (inet_ntop(AF_INET6,
404                                     &((struct in6_addr *)
405                                         params[n].jp_value)[ai],
406                                     ipbuf, sizeof(ipbuf)) == NULL)
407                                         err(1, "inet_ntop");
408                                 else
409                                         printf("%6s  %s\n", "", ipbuf);
410                         n++;
411                 }
412 #endif
413         } else if (pflags & PRINT_DEFAULT) {
414                 if (pflags & PRINT_JAIL_NAME)
415                         printf(" %-15s ", (char *)params[0].jp_value);
416                 else
417                         printf("%6d  ", *(int *)params[0].jp_value);
418                 printf("%-15.15s %-29.29s %.74s\n",
419 #ifdef INET
420                     (!ip4_ok || params[1].jp_valuelen == 0) ? "-"
421                     : inet_ntoa(*(struct in_addr *)params[1].jp_value),
422                     (char *)params[2-!ip4_ok].jp_value,
423                     (char *)params[3-!ip4_ok].jp_value);
424 #else
425                     "-",
426                     (char *)params[1].jp_value,
427                     (char *)params[2].jp_value);
428 #endif
429         } else {
430                 param_values = alloca(nparams * sizeof(*param_values));
431                 for (i = 0; i < nparams; i++) {
432                         if (!(params[i].jp_flags & JP_USER))
433                                 continue;
434                         param_values[i] = jailparam_export(params + i);
435                         if (param_values[i] == NULL)
436                                 errx(1, "%s", jail_errmsg);
437                 }
438                 for (i = spc = 0; i < nparams; i++) {
439                         if (!(params[i].jp_flags & JP_USER))
440                                 continue;
441                         if ((pflags & PRINT_SKIP) &&
442                             ((!(params[i].jp_ctltype &
443                                 (CTLFLAG_WR | CTLFLAG_TUN))) ||
444                              (param_parent[i] >= 0 &&
445                               *(int *)params[param_parent[i]].jp_value !=
446                               JAIL_SYS_NEW)))
447                                 continue;
448                         if (spc)
449                                 putchar(' ');
450                         else
451                                 spc = 1;
452                         if (pflags & PRINT_NAMEVAL) {
453                                 /*
454                                  * Generally "name=value", but for booleans
455                                  * either "name" or "noname".
456                                  */
457                                 if (params[i].jp_flags &
458                                     (JP_BOOL | JP_NOBOOL)) {
459                                         if (*(int *)params[i].jp_value)
460                                                 printf("%s", params[i].jp_name);
461                                         else {
462                                                 nname = (params[i].jp_flags &
463                                                     JP_NOBOOL) ?
464                                                     nononame(params[i].jp_name)
465                                                     : noname(params[i].jp_name);
466                                                 printf("%s", nname);
467                                                 free(nname);
468                                         }
469                                         continue;
470                                 }
471                                 printf("%s=", params[i].jp_name);
472                         }
473                         if (params[i].jp_valuelen == 0) {
474                                 if (pflags & PRINT_QUOTED)
475                                         printf("\"\"");
476                                 else if (!(pflags & PRINT_NAMEVAL))
477                                         putchar('-');
478                         } else
479                                 quoted_print(param_values[i]);
480                 }
481                 putchar('\n');
482                 for (i = 0; i < nparams; i++)
483                         if (params[i].jp_flags & JP_USER)
484                                 free(param_values[i]);
485         }
486         return (jid);
487 }
488
489 static void
490 quoted_print(char *str)
491 {
492         int c, qc;
493         char *p = str;
494
495         /* An empty string needs quoting. */
496         if (!*p) {
497                 fputs("\"\"", stdout);
498                 return;
499         }
500
501         /*
502          * The value will be surrounded by quotes if it contains spaces
503          * or quotes.
504          */
505         qc = strchr(p, '\'') ? '"'
506             : strchr(p, '"') ? '\''
507             : strchr(p, ' ') || strchr(p, '\t') ? '"'
508             : 0;
509         if (qc)
510                 putchar(qc);
511         while ((c = *p++)) {
512                 if (c == '\\' || c == qc)
513                         putchar('\\');
514                 putchar(c);
515         }
516         if (qc)
517                 putchar(qc);
518 }