chiark / gitweb /
networkd: unref tunnel
[elogind.git] / src / machine / machinectl.c
1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
2
3 /***
4   This file is part of systemd.
5
6   Copyright 2013 Lennart Poettering
7
8   systemd is free software; you can redistribute it and/or modify it
9   under the terms of the GNU Lesser General Public License as published by
10   the Free Software Foundation; either version 2.1 of the License, or
11   (at your option) any later version.
12
13   systemd is distributed in the hope that it will be useful, but
14   WITHOUT ANY WARRANTY; without even the implied warranty of
15   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16   Lesser General Public License for more details.
17
18   You should have received a copy of the GNU Lesser General Public License
19   along with systemd; If not, see <http://www.gnu.org/licenses/>.
20 ***/
21
22 #include <sys/socket.h>
23 #include <unistd.h>
24 #include <errno.h>
25 #include <string.h>
26 #include <getopt.h>
27 #include <pwd.h>
28 #include <locale.h>
29 #include <fcntl.h>
30 #include <netinet/in.h>
31 #include <arpa/inet.h>
32
33 #include "sd-bus.h"
34 #include "log.h"
35 #include "util.h"
36 #include "macro.h"
37 #include "pager.h"
38 #include "bus-util.h"
39 #include "bus-error.h"
40 #include "build.h"
41 #include "strv.h"
42 #include "unit-name.h"
43 #include "cgroup-show.h"
44 #include "cgroup-util.h"
45 #include "ptyfwd.h"
46
47 static char **arg_property = NULL;
48 static bool arg_all = false;
49 static bool arg_full = false;
50 static bool arg_no_pager = false;
51 static bool arg_legend = true;
52 static const char *arg_kill_who = NULL;
53 static int arg_signal = SIGTERM;
54 static BusTransport arg_transport = BUS_TRANSPORT_LOCAL;
55 static char *arg_host = NULL;
56
57 static void pager_open_if_enabled(void) {
58
59         /* Cache result before we open the pager */
60         if (arg_no_pager)
61                 return;
62
63         pager_open(false);
64 }
65
66 static int list_machines(sd_bus *bus, char **args, unsigned n) {
67         _cleanup_bus_message_unref_ sd_bus_message *reply = NULL;
68         _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;
69         const char *name, *class, *service, *object;
70         unsigned k = 0;
71         int r;
72
73         pager_open_if_enabled();
74
75         r = sd_bus_call_method(
76                                 bus,
77                                 "org.freedesktop.machine1",
78                                 "/org/freedesktop/machine1",
79                                 "org.freedesktop.machine1.Manager",
80                                 "ListMachines",
81                                 &error,
82                                 &reply,
83                                 "");
84         if (r < 0) {
85                 log_error("Could not get machines: %s", bus_error_message(&error, -r));
86                 return r;
87         }
88
89         if (arg_legend)
90                 printf("%-32s %-9s %-16s\n", "MACHINE", "CONTAINER", "SERVICE");
91
92         r = sd_bus_message_enter_container(reply, SD_BUS_TYPE_ARRAY, "(ssso)");
93         if (r < 0)
94                 return bus_log_parse_error(r);
95
96         while ((r = sd_bus_message_read(reply, "(ssso)", &name, &class, &service, &object)) > 0) {
97                 printf("%-32s %-9s %-16s\n", name, class, service);
98
99                 k++;
100         }
101         if (r < 0)
102                 return bus_log_parse_error(r);
103
104         r = sd_bus_message_exit_container(reply);
105         if (r < 0)
106                 return bus_log_parse_error(r);
107
108         if (arg_legend)
109                 printf("\n%u machines listed.\n", k);
110
111         return 0;
112 }
113
114 static int show_unit_cgroup(sd_bus *bus, const char *unit, pid_t leader) {
115         _cleanup_bus_message_unref_ sd_bus_message *reply = NULL;
116         _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;
117         _cleanup_free_ char *path = NULL;
118         const char *cgroup;
119         int r, output_flags;
120         unsigned c;
121
122         assert(bus);
123         assert(unit);
124
125         if (arg_transport == BUS_TRANSPORT_REMOTE)
126                 return 0;
127
128         path = unit_dbus_path_from_name(unit);
129         if (!path)
130                 return log_oom();
131
132         r = sd_bus_get_property(
133                         bus,
134                         "org.freedesktop.systemd1",
135                         path,
136                         endswith(unit, ".scope") ? "org.freedesktop.systemd1.Scope" : "org.freedesktop.systemd1.Service",
137                         "ControlGroup",
138                         &error,
139                         &reply,
140                         "s");
141         if (r < 0) {
142                 log_error("Failed to query ControlGroup: %s", bus_error_message(&error, -r));
143                 return r;
144         }
145
146         r = sd_bus_message_read(reply, "s", &cgroup);
147         if (r < 0)
148                 return bus_log_parse_error(r);
149
150         if (isempty(cgroup))
151                 return 0;
152
153         if (cg_is_empty_recursive(SYSTEMD_CGROUP_CONTROLLER, cgroup, false) != 0 && leader <= 0)
154                 return 0;
155
156         output_flags =
157                 arg_all * OUTPUT_SHOW_ALL |
158                 arg_full * OUTPUT_FULL_WIDTH;
159
160         c = columns();
161         if (c > 18)
162                 c -= 18;
163         else
164                 c = 0;
165
166         show_cgroup_and_extra(SYSTEMD_CGROUP_CONTROLLER, cgroup, "\t\t  ", c, false, &leader, leader > 0, output_flags);
167         return 0;
168 }
169
170 static int print_addresses(sd_bus *bus, const char *name, const char *prefix, const char *prefix2) {
171         _cleanup_bus_message_unref_ sd_bus_message *reply = NULL;
172         int r;
173
174         assert(bus);
175         assert(name);
176         assert(prefix);
177         assert(prefix2);
178
179         r = sd_bus_call_method(bus,
180                                "org.freedesktop.machine1",
181                                "/org/freedesktop/machine1",
182                                "org.freedesktop.machine1.Manager",
183                                "GetMachineAddresses",
184                                NULL,
185                                &reply,
186                                "s", name);
187         if (r < 0)
188                 return r;
189
190         r = sd_bus_message_enter_container(reply, 'a', "(yay)");
191         if (r < 0)
192                 return bus_log_parse_error(r);
193
194         while ((r = sd_bus_message_enter_container(reply, 'r', "yay")) > 0) {
195                 unsigned char family;
196                 const void *a;
197                 size_t sz;
198                 char buffer[MAX(INET6_ADDRSTRLEN, INET_ADDRSTRLEN)];
199
200                 r = sd_bus_message_read(reply, "y", &family);
201                 if (r < 0)
202                         return bus_log_parse_error(r);
203
204                 r = sd_bus_message_read_array(reply, 'y', &a, &sz);
205                 if (r < 0)
206                         return bus_log_parse_error(r);
207
208                 printf("%s%s\n", prefix, inet_ntop(family, a, buffer, sizeof(buffer)));
209
210                 r = sd_bus_message_exit_container(reply);
211                 if (r < 0)
212                         return bus_log_parse_error(r);
213
214                 if (prefix != prefix2)
215                         prefix = prefix2;
216         }
217         if (r < 0)
218                 return bus_log_parse_error(r);
219
220         r = sd_bus_message_exit_container(reply);
221         if (r < 0)
222                 return bus_log_parse_error(r);
223
224         return 0;
225 }
226
227 typedef struct MachineStatusInfo {
228         char *name;
229         sd_id128_t id;
230         char *class;
231         char *service;
232         char *unit;
233         char *root_directory;
234         pid_t leader;
235         usec_t timestamp;
236 } MachineStatusInfo;
237
238 static void print_machine_status_info(sd_bus *bus, MachineStatusInfo *i) {
239         char since1[FORMAT_TIMESTAMP_RELATIVE_MAX], *s1;
240         char since2[FORMAT_TIMESTAMP_MAX], *s2;
241         assert(i);
242
243         fputs(strna(i->name), stdout);
244
245         if (!sd_id128_equal(i->id, SD_ID128_NULL))
246                 printf("(" SD_ID128_FORMAT_STR ")\n", SD_ID128_FORMAT_VAL(i->id));
247         else
248                 putchar('\n');
249
250         s1 = format_timestamp_relative(since1, sizeof(since1), i->timestamp);
251         s2 = format_timestamp(since2, sizeof(since2), i->timestamp);
252
253         if (s1)
254                 printf("\t   Since: %s; %s\n", s2, s1);
255         else if (s2)
256                 printf("\t   Since: %s\n", s2);
257
258         if (i->leader > 0) {
259                 _cleanup_free_ char *t = NULL;
260
261                 printf("\t  Leader: %u", (unsigned) i->leader);
262
263                 get_process_comm(i->leader, &t);
264                 if (t)
265                         printf(" (%s)", t);
266
267                 putchar('\n');
268         }
269
270         if (i->service) {
271                 printf("\t Service: %s", i->service);
272
273                 if (i->class)
274                         printf("; class %s", i->class);
275
276                 putchar('\n');
277         } else if (i->class)
278                 printf("\t   Class: %s\n", i->class);
279
280         if (i->root_directory)
281                 printf("\t    Root: %s\n", i->root_directory);
282
283         print_addresses(bus, i->name,
284                        "\t Address: ",
285                        "\t          ");
286
287         if (i->unit) {
288                 printf("\t    Unit: %s\n", i->unit);
289                 show_unit_cgroup(bus, i->unit, i->leader);
290         }
291 }
292
293 static int show_info(const char *verb, sd_bus *bus, const char *path, bool *new_line) {
294
295         static const struct bus_properties_map map[]  = {
296                 { "Name",          "s",  NULL,          offsetof(MachineStatusInfo, name) },
297                 { "Class",         "s",  NULL,          offsetof(MachineStatusInfo, class) },
298                 { "Service",       "s",  NULL,          offsetof(MachineStatusInfo, service) },
299                 { "Unit",          "s",  NULL,          offsetof(MachineStatusInfo, unit) },
300                 { "RootDirectory", "s",  NULL,          offsetof(MachineStatusInfo, root_directory) },
301                 { "Leader",        "u",  NULL,          offsetof(MachineStatusInfo, leader) },
302                 { "Timestamp",     "t",  NULL,          offsetof(MachineStatusInfo, timestamp) },
303                 { "Id",            "ay", bus_map_id128, offsetof(MachineStatusInfo, id) },
304                 {}
305         };
306
307         MachineStatusInfo info = {};
308         int r;
309
310         assert(path);
311         assert(new_line);
312
313         r = bus_map_all_properties(bus,
314                                    "org.freedesktop.machine1",
315                                    path,
316                                    map,
317                                    &info);
318         if (r < 0) {
319                 log_error("Could not get properties: %s", strerror(-r));
320                 return r;
321         }
322
323         if (*new_line)
324                 printf("\n");
325         *new_line = true;
326
327         print_machine_status_info(bus, &info);
328
329         free(info.name);
330         free(info.class);
331         free(info.service);
332         free(info.unit);
333         free(info.root_directory);
334
335         return r;
336 }
337
338 static int show_properties(sd_bus *bus, const char *path, bool *new_line) {
339         int r;
340
341         if (*new_line)
342                 printf("\n");
343
344         *new_line = true;
345
346         r = bus_print_all_properties(bus, "org.freedesktop.machine1", path, arg_property, arg_all);
347         if (r < 0)
348                 log_error("Could not get properties: %s", strerror(-r));
349
350         return r;
351 }
352
353 static int show(sd_bus *bus, char **args, unsigned n) {
354         _cleanup_bus_message_unref_ sd_bus_message *reply = NULL;
355         _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;
356         int r = 0;
357         unsigned i;
358         bool properties, new_line = false;
359
360         assert(bus);
361         assert(args);
362
363         properties = !strstr(args[0], "status");
364
365         pager_open_if_enabled();
366
367         if (properties && n <= 1) {
368
369                 /* If no argument is specified, inspect the manager
370                  * itself */
371                 r = show_properties(bus, "/org/freedesktop/machine1", &new_line);
372                 if (r < 0)
373                         return r;
374         }
375
376         for (i = 1; i < n; i++) {
377                 const char *path = NULL;
378
379                 r = sd_bus_call_method(
380                                         bus,
381                                         "org.freedesktop.machine1",
382                                         "/org/freedesktop/machine1",
383                                         "org.freedesktop.machine1.Manager",
384                                         "GetMachine",
385                                         &error,
386                                         &reply,
387                                         "s", args[i]);
388                 if (r < 0) {
389                         log_error("Could not get path to machine: %s", bus_error_message(&error, -r));
390                         return r;
391                 }
392
393                 r = sd_bus_message_read(reply, "o", &path);
394                 if (r < 0)
395                         return bus_log_parse_error(r);
396
397                 if (properties)
398                         r = show_properties(bus, path, &new_line);
399                 else
400                         r = show_info(args[0], bus, path, &new_line);
401         }
402
403         return r;
404 }
405
406 static int kill_machine(sd_bus *bus, char **args, unsigned n) {
407         _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;
408         unsigned i;
409
410         assert(args);
411
412         if (!arg_kill_who)
413                 arg_kill_who = "all";
414
415         for (i = 1; i < n; i++) {
416                 int r;
417
418                 r = sd_bus_call_method(
419                                         bus,
420                                         "org.freedesktop.machine1",
421                                         "/org/freedesktop/machine1",
422                                         "org.freedesktop.machine1.Manager",
423                                         "KillMachine",
424                                         &error,
425                                         NULL,
426                                         "ssi", args[i], arg_kill_who, arg_signal);
427                 if (r < 0) {
428                         log_error("Could not kill machine: %s", bus_error_message(&error, -r));
429                         return r;
430                 }
431         }
432
433         return 0;
434 }
435
436 static int reboot_machine(sd_bus *bus, char **args, unsigned n) {
437         arg_kill_who = "leader";
438         arg_signal = SIGINT; /* sysvinit + systemd */
439
440         return kill_machine(bus, args, n);
441 }
442
443 static int poweroff_machine(sd_bus *bus, char **args, unsigned n) {
444         arg_kill_who = "leader";
445         arg_signal = SIGRTMIN+4; /* only systemd */
446
447         return kill_machine(bus, args, n);
448 }
449
450 static int terminate_machine(sd_bus *bus, char **args, unsigned n) {
451         _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;
452         unsigned i;
453
454         assert(args);
455
456         for (i = 1; i < n; i++) {
457                 int r;
458
459                 r = sd_bus_call_method(
460                                 bus,
461                                 "org.freedesktop.machine1",
462                                 "/org/freedesktop/machine1",
463                                 "org.freedesktop.machine1.Manager",
464                                 "TerminateMachine",
465                                 &error,
466                                 NULL,
467                                 "s", args[i]);
468                 if (r < 0) {
469                         log_error("Could not terminate machine: %s", bus_error_message(&error, -r));
470                         return r;
471                 }
472         }
473
474         return 0;
475 }
476
477 static int openpt_in_namespace(pid_t pid, int flags) {
478         _cleanup_close_pair_ int pair[2] = { -1, -1 };
479         _cleanup_close_ int pidnsfd = -1, mntnsfd = -1, rootfd = -1;
480         union {
481                 struct cmsghdr cmsghdr;
482                 uint8_t buf[CMSG_SPACE(sizeof(int))];
483         } control = {};
484         struct msghdr mh = {
485                 .msg_control = &control,
486                 .msg_controllen = sizeof(control),
487         };
488         struct cmsghdr *cmsg;
489         int master = -1, r;
490         pid_t child;
491         siginfo_t si;
492
493         r = namespace_open(pid, &pidnsfd, &mntnsfd, NULL, &rootfd);
494         if (r < 0)
495                 return r;
496
497         if (socketpair(AF_UNIX, SOCK_DGRAM, 0, pair) < 0)
498                 return -errno;
499
500         child = fork();
501         if (child < 0)
502                 return -errno;
503
504         if (child == 0) {
505                 pair[0] = safe_close(pair[0]);
506
507                 r = namespace_enter(pidnsfd, mntnsfd, -1, rootfd);
508                 if (r < 0)
509                         _exit(EXIT_FAILURE);
510
511                 master = posix_openpt(flags);
512                 if (master < 0)
513                         _exit(EXIT_FAILURE);
514
515                 cmsg = CMSG_FIRSTHDR(&mh);
516                 cmsg->cmsg_level = SOL_SOCKET;
517                 cmsg->cmsg_type = SCM_RIGHTS;
518                 cmsg->cmsg_len = CMSG_LEN(sizeof(int));
519                 memcpy(CMSG_DATA(cmsg), &master, sizeof(int));
520
521                 mh.msg_controllen = cmsg->cmsg_len;
522
523                 if (sendmsg(pair[1], &mh, MSG_NOSIGNAL) < 0)
524                         _exit(EXIT_FAILURE);
525
526                 _exit(EXIT_SUCCESS);
527         }
528
529         pair[1] = safe_close(pair[1]);
530
531         r = wait_for_terminate(child, &si);
532         if (r < 0)
533                 return r;
534         if (si.si_code != CLD_EXITED || si.si_status != EXIT_SUCCESS)
535                 return -EIO;
536
537         if (recvmsg(pair[0], &mh, MSG_NOSIGNAL|MSG_CMSG_CLOEXEC) < 0)
538                 return -errno;
539
540         for (cmsg = CMSG_FIRSTHDR(&mh); cmsg; cmsg = CMSG_NXTHDR(&mh, cmsg))
541                 if (cmsg->cmsg_level == SOL_SOCKET && cmsg->cmsg_type == SCM_RIGHTS) {
542                         int *fds;
543                         unsigned n_fds;
544
545                         fds = (int*) CMSG_DATA(cmsg);
546                         n_fds = (cmsg->cmsg_len - CMSG_LEN(0)) / sizeof(int);
547
548                         if (n_fds != 1) {
549                                 close_many(fds, n_fds);
550                                 return -EIO;
551                         }
552
553                         master = fds[0];
554                 }
555
556         if (master < 0)
557                 return -EIO;
558
559         return master;
560 }
561
562 static int login_machine(sd_bus *bus, char **args, unsigned n) {
563         _cleanup_bus_message_unref_ sd_bus_message *reply = NULL, *reply2 = NULL, *reply3 = NULL;
564         _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;
565         _cleanup_bus_unref_ sd_bus *container_bus = NULL;
566         _cleanup_close_ int master = -1;
567         _cleanup_free_ char *getty = NULL;
568         const char *path, *pty, *p;
569         uint32_t leader;
570         sigset_t mask;
571         int r;
572
573         assert(bus);
574         assert(args);
575
576         if (arg_transport != BUS_TRANSPORT_LOCAL) {
577                 log_error("Login only supported on local machines.");
578                 return -ENOTSUP;
579         }
580
581         r = sd_bus_call_method(
582                         bus,
583                         "org.freedesktop.machine1",
584                         "/org/freedesktop/machine1",
585                         "org.freedesktop.machine1.Manager",
586                         "GetMachine",
587                         &error,
588                         &reply,
589                         "s", args[1]);
590         if (r < 0) {
591                 log_error("Could not get path to machine: %s", bus_error_message(&error, -r));
592                 return r;
593         }
594
595         r = sd_bus_message_read(reply, "o", &path);
596         if (r < 0)
597                 return bus_log_parse_error(r);
598
599         r = sd_bus_get_property(
600                         bus,
601                         "org.freedesktop.machine1",
602                         path,
603                         "org.freedesktop.machine1.Machine",
604                         "Leader",
605                         &error,
606                         &reply2,
607                         "u");
608         if (r < 0) {
609                 log_error("Failed to retrieve PID of leader: %s", strerror(-r));
610                 return r;
611         }
612
613         r = sd_bus_message_read(reply2, "u", &leader);
614         if (r < 0)
615                 return bus_log_parse_error(r);
616
617         master = openpt_in_namespace(leader, O_RDWR|O_NOCTTY|O_CLOEXEC|O_NDELAY);
618         if (master < 0) {
619                 log_error("Failed to acquire pseudo tty: %s", strerror(-master));
620                 return master;
621         }
622
623         pty = ptsname(master);
624         if (!pty) {
625                 log_error("Failed to get pty name: %m");
626                 return -errno;
627         }
628
629         p = startswith(pty, "/dev/pts/");
630         if (!p) {
631                 log_error("Invalid pty name %s.", pty);
632                 return -EIO;
633         }
634
635         r = sd_bus_open_system_container(&container_bus, args[1]);
636         if (r < 0) {
637                 log_error("Failed to get container bus: %s", strerror(-r));
638                 return r;
639         }
640
641         getty = strjoin("container-getty@", p, ".service", NULL);
642         if (!getty)
643                 return log_oom();
644
645         if (unlockpt(master) < 0) {
646                 log_error("Failed to unlock tty: %m");
647                 return -errno;
648         }
649
650         r = sd_bus_call_method(container_bus,
651                                "org.freedesktop.systemd1",
652                                "/org/freedesktop/systemd1",
653                                "org.freedesktop.systemd1.Manager",
654                                "StartUnit",
655                                &error, &reply3,
656                                "ss", getty, "replace");
657         if (r < 0) {
658                 log_error("Failed to start getty service: %s", bus_error_message(&error, r));
659                 return r;
660         }
661
662         container_bus = sd_bus_unref(container_bus);
663
664         assert_se(sigemptyset(&mask) == 0);
665         sigset_add_many(&mask, SIGWINCH, SIGTERM, SIGINT, -1);
666         assert_se(sigprocmask(SIG_BLOCK, &mask, NULL) == 0);
667
668         log_info("Connected to container %s. Press ^] three times within 1s to exit session.", args[1]);
669
670         r = process_pty(master, &mask, 0, 0);
671         if (r < 0) {
672                 log_error("Failed to process pseudo tty: %s", strerror(-r));
673                 return r;
674         }
675
676         fputc('\n', stdout);
677
678         log_info("Connection to container %s terminated.", args[1]);
679
680         return 0;
681 }
682
683 static int help(void) {
684
685         printf("%s [OPTIONS...] {COMMAND} ...\n\n"
686                "Send control commands to or query the virtual machine and container registration manager.\n\n"
687                "  -h --help              Show this help\n"
688                "     --version           Show package version\n"
689                "     --no-pager          Do not pipe output into a pager\n"
690                "     --no-legend         Do not show the headers and footers\n"
691                "  -H --host=[USER@]HOST  Operate on remote host\n"
692                "  -M --machine=CONTAINER Operate on local container\n"
693                "  -p --property=NAME     Show only properties by this name\n"
694                "  -a --all               Show all properties, including empty ones\n"
695                "  -l --full              Do not ellipsize output\n"
696                "     --kill-who=WHO      Who to send signal to\n"
697                "  -s --signal=SIGNAL     Which signal to send\n\n"
698                "Commands:\n"
699                "  list                   List running VMs and containers\n"
700                "  status NAME...         Show VM/container status\n"
701                "  show NAME...           Show properties of one or more VMs/containers\n"
702                "  login NAME             Get a login prompt on a container\n"
703                "  poweroff NAME...       Power off one or more containers\n"
704                "  reboot NAME...         Reboot one or more containers\n"
705                "  kill NAME...           Send signal to processes of a VM/container\n"
706                "  terminate NAME...      Terminate one or more VMs/containers\n",
707                program_invocation_short_name);
708
709         return 0;
710 }
711
712 static int parse_argv(int argc, char *argv[]) {
713
714         enum {
715                 ARG_VERSION = 0x100,
716                 ARG_NO_PAGER,
717                 ARG_NO_LEGEND,
718                 ARG_KILL_WHO,
719         };
720
721         static const struct option options[] = {
722                 { "help",            no_argument,       NULL, 'h'                 },
723                 { "version",         no_argument,       NULL, ARG_VERSION         },
724                 { "property",        required_argument, NULL, 'p'                 },
725                 { "all",             no_argument,       NULL, 'a'                 },
726                 { "full",            no_argument,       NULL, 'l'                 },
727                 { "no-pager",        no_argument,       NULL, ARG_NO_PAGER        },
728                 { "no-legend",       no_argument,       NULL, ARG_NO_LEGEND       },
729                 { "kill-who",        required_argument, NULL, ARG_KILL_WHO        },
730                 { "signal",          required_argument, NULL, 's'                 },
731                 { "host",            required_argument, NULL, 'H'                 },
732                 { "machine",         required_argument, NULL, 'M'                 },
733                 {}
734         };
735
736         int c, r;
737
738         assert(argc >= 0);
739         assert(argv);
740
741         while ((c = getopt_long(argc, argv, "hp:als:H:M:", options, NULL)) >= 0) {
742
743                 switch (c) {
744
745                 case 'h':
746                         return help();
747
748                 case ARG_VERSION:
749                         puts(PACKAGE_STRING);
750                         puts(SYSTEMD_FEATURES);
751                         return 0;
752
753                 case 'p':
754                         r = strv_extend(&arg_property, optarg);
755                         if (r < 0)
756                                 return log_oom();
757
758                         /* If the user asked for a particular
759                          * property, show it to him, even if it is
760                          * empty. */
761                         arg_all = true;
762                         break;
763
764                 case 'a':
765                         arg_all = true;
766                         break;
767
768                 case 'l':
769                         arg_full = true;
770                         break;
771
772                 case ARG_NO_PAGER:
773                         arg_no_pager = true;
774                         break;
775
776                 case ARG_NO_LEGEND:
777                         arg_legend = false;
778                         break;
779
780                 case ARG_KILL_WHO:
781                         arg_kill_who = optarg;
782                         break;
783
784                 case 's':
785                         arg_signal = signal_from_string_try_harder(optarg);
786                         if (arg_signal < 0) {
787                                 log_error("Failed to parse signal string %s.", optarg);
788                                 return -EINVAL;
789                         }
790                         break;
791
792                 case 'H':
793                         arg_transport = BUS_TRANSPORT_REMOTE;
794                         arg_host = optarg;
795                         break;
796
797                 case 'M':
798                         arg_transport = BUS_TRANSPORT_CONTAINER;
799                         arg_host = optarg;
800                         break;
801
802                 case '?':
803                         return -EINVAL;
804
805                 default:
806                         assert_not_reached("Unhandled option");
807                 }
808         }
809
810         return 1;
811 }
812
813 static int machinectl_main(sd_bus *bus, int argc, char *argv[]) {
814
815         static const struct {
816                 const char* verb;
817                 const enum {
818                         MORE,
819                         LESS,
820                         EQUAL
821                 } argc_cmp;
822                 const int argc;
823                 int (* const dispatch)(sd_bus *bus, char **args, unsigned n);
824         } verbs[] = {
825                 { "list",                  LESS,   1, list_machines     },
826                 { "status",                MORE,   2, show              },
827                 { "show",                  MORE,   1, show              },
828                 { "terminate",             MORE,   2, terminate_machine },
829                 { "reboot",                MORE,   2, reboot_machine    },
830                 { "poweroff",              MORE,   2, poweroff_machine  },
831                 { "kill",                  MORE,   2, kill_machine      },
832                 { "login",                 MORE,   2, login_machine     },
833         };
834
835         int left;
836         unsigned i;
837
838         assert(argc >= 0);
839         assert(argv);
840
841         left = argc - optind;
842
843         if (left <= 0)
844                 /* Special rule: no arguments means "list" */
845                 i = 0;
846         else {
847                 if (streq(argv[optind], "help")) {
848                         help();
849                         return 0;
850                 }
851
852                 for (i = 0; i < ELEMENTSOF(verbs); i++)
853                         if (streq(argv[optind], verbs[i].verb))
854                                 break;
855
856                 if (i >= ELEMENTSOF(verbs)) {
857                         log_error("Unknown operation %s", argv[optind]);
858                         return -EINVAL;
859                 }
860         }
861
862         switch (verbs[i].argc_cmp) {
863
864         case EQUAL:
865                 if (left != verbs[i].argc) {
866                         log_error("Invalid number of arguments.");
867                         return -EINVAL;
868                 }
869
870                 break;
871
872         case MORE:
873                 if (left < verbs[i].argc) {
874                         log_error("Too few arguments.");
875                         return -EINVAL;
876                 }
877
878                 break;
879
880         case LESS:
881                 if (left > verbs[i].argc) {
882                         log_error("Too many arguments.");
883                         return -EINVAL;
884                 }
885
886                 break;
887
888         default:
889                 assert_not_reached("Unknown comparison operator.");
890         }
891
892         return verbs[i].dispatch(bus, argv + optind, left);
893 }
894
895 int main(int argc, char*argv[]) {
896         _cleanup_bus_unref_ sd_bus *bus = NULL;
897         int r;
898
899         setlocale(LC_ALL, "");
900         log_parse_environment();
901         log_open();
902
903         r = parse_argv(argc, argv);
904         if (r <= 0)
905                 goto finish;
906
907         r = bus_open_transport(arg_transport, arg_host, false, &bus);
908         if (r < 0) {
909                 log_error("Failed to create bus connection: %s", strerror(-r));
910                 goto finish;
911         }
912
913         r = machinectl_main(bus, argc, argv);
914
915 finish:
916         pager_close();
917
918         strv_free(arg_property);
919
920         return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS;
921 }