chiark / gitweb /
event: dynamically adjust size of events array instead of pre-allocating it possibly...
[elogind.git] / src / udev / udevadm-control.c
1 /*
2  * Copyright (C) 2005-2011 Kay Sievers <kay@vrfy.org>
3  *
4  * This program is free software: you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation, either version 2 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  */
14
15 #include <time.h>
16 #include <errno.h>
17 #include <stdio.h>
18 #include <stdlib.h>
19 #include <stddef.h>
20 #include <string.h>
21 #include <unistd.h>
22 #include <getopt.h>
23 #include <sys/types.h>
24 #include <sys/socket.h>
25 #include <sys/wait.h>
26 #include <sys/un.h>
27
28 #include "udev.h"
29
30 static void print_help(void)
31 {
32         printf("Usage: udevadm control COMMAND\n"
33                 "  --exit                   instruct the daemon to cleanup and exit\n"
34                 "  --log-priority=<level>   set the udev log level for the daemon\n"
35                 "  --stop-exec-queue        do not execute events, queue only\n"
36                 "  --start-exec-queue       execute events, flush queue\n"
37                 "  --reload                 reload rules and databases\n"
38                 "  --property=<KEY>=<value> set a global property for all events\n"
39                 "  --children-max=<N>       maximum number of children\n"
40                 "  --timeout=<seconds>      maximum time to block for a reply\n"
41                 "  --help                   print this help text\n\n");
42 }
43
44 static int adm_control(struct udev *udev, int argc, char *argv[])
45 {
46         struct udev_ctrl *uctrl = NULL;
47         int timeout = 60;
48         int rc = 1;
49
50         static const struct option options[] = {
51                 { "exit", no_argument, NULL, 'e' },
52                 { "log-priority", required_argument, NULL, 'l' },
53                 { "stop-exec-queue", no_argument, NULL, 's' },
54                 { "start-exec-queue", no_argument, NULL, 'S' },
55                 { "reload", no_argument, NULL, 'R' },
56                 { "reload-rules", no_argument, NULL, 'R' },
57                 { "property", required_argument, NULL, 'p' },
58                 { "env", required_argument, NULL, 'p' },
59                 { "children-max", required_argument, NULL, 'm' },
60                 { "timeout", required_argument, NULL, 't' },
61                 { "help", no_argument, NULL, 'h' },
62                 {}
63         };
64
65         if (getuid() != 0) {
66                 fprintf(stderr, "root privileges required\n");
67                 return 1;
68         }
69
70         uctrl = udev_ctrl_new(udev);
71         if (uctrl == NULL)
72                 return 2;
73
74         for (;;) {
75                 int option;
76
77                 option = getopt_long(argc, argv, "el:sSRp:m:h", options, NULL);
78                 if (option == -1)
79                         break;
80
81                 switch (option) {
82                 case 'e':
83                         if (udev_ctrl_send_exit(uctrl, timeout) < 0)
84                                 rc = 2;
85                         else
86                                 rc = 0;
87                         break;
88                 case 'l': {
89                         int i;
90
91                         i = util_log_priority(optarg);
92                         if (i < 0) {
93                                 fprintf(stderr, "invalid number '%s'\n", optarg);
94                                 goto out;
95                         }
96                         if (udev_ctrl_send_set_log_level(uctrl, util_log_priority(optarg), timeout) < 0)
97                                 rc = 2;
98                         else
99                                 rc = 0;
100                         break;
101                 }
102                 case 's':
103                         if (udev_ctrl_send_stop_exec_queue(uctrl, timeout) < 0)
104                                 rc = 2;
105                         else
106                                 rc = 0;
107                         break;
108                 case 'S':
109                         if (udev_ctrl_send_start_exec_queue(uctrl, timeout) < 0)
110                                 rc = 2;
111                         else
112                                 rc = 0;
113                         break;
114                 case 'R':
115                         if (udev_ctrl_send_reload(uctrl, timeout) < 0)
116                                 rc = 2;
117                         else
118                                 rc = 0;
119                         break;
120                 case 'p':
121                         if (strchr(optarg, '=') == NULL) {
122                                 fprintf(stderr, "expect <KEY>=<value> instead of '%s'\n", optarg);
123                                 goto out;
124                         }
125                         if (udev_ctrl_send_set_env(uctrl, optarg, timeout) < 0)
126                                 rc = 2;
127                         else
128                                 rc = 0;
129                         break;
130                 case 'm': {
131                         char *endp;
132                         int i;
133
134                         i = strtoul(optarg, &endp, 0);
135                         if (endp[0] != '\0' || i < 1) {
136                                 fprintf(stderr, "invalid number '%s'\n", optarg);
137                                 goto out;
138                         }
139                         if (udev_ctrl_send_set_children_max(uctrl, i, timeout) < 0)
140                                 rc = 2;
141                         else
142                                 rc = 0;
143                         break;
144                 }
145                 case 't': {
146                         int seconds;
147
148                         seconds = atoi(optarg);
149                         if (seconds >= 0)
150                                 timeout = seconds;
151                         else
152                                 fprintf(stderr, "invalid timeout value\n");
153                         break;
154                 }
155                 case 'h':
156                         print_help();
157                         rc = 0;
158                         break;
159                 }
160         }
161
162         if (argv[optind] != NULL)
163                 fprintf(stderr, "unknown option\n");
164         else if (optind == 1)
165                 fprintf(stderr, "missing option\n");
166 out:
167         udev_ctrl_unref(uctrl);
168         return rc;
169 }
170
171 const struct udevadm_cmd udevadm_control = {
172         .name = "control",
173         .cmd = adm_control,
174         .help = "control the udev daemon",
175 };