/*
- * udevcontrol.c
- *
- * Copyright (C) 2005 Kay Sievers <kay.sievers@vrfy.org>
+ * Copyright (C) 2005-2006 Kay Sievers <kay.sievers@vrfy.org>
*
* This program is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License as published by the
*
* You should have received a copy of the GNU General Public License along
* with this program; if not, write to the Free Software Foundation, Inc.,
- * 675 Mass Ave, Cambridge, MA 02139, USA.
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*
*/
static int sock = -1;
static int udev_log = 0;
-#ifdef USE_LOG
-void log_message (int priority, const char *format, ...)
-{
- va_list args;
-
- if (priority > udev_log)
- return;
-
- va_start(args, format);
- vsyslog(priority, format, args);
- va_end(args);
-}
-#endif
-
-static void usage(void)
-{
- printf("Usage: udevcontrol COMMAND\n"
- " log_priority=<level> set the udev log level for the daemon\n"
- " stop_exec_queue keep udevd from executing events, queue only\n"
- " start_exec_queue execute events, flush queue\n"
- " reload_rules reloads the rules files\n"
- " max_childs=<N> maximum number of childs running at the same time\n"
- " --help print this help text\n\n");
-}
-
-int main(int argc, char *argv[], char *envp[])
+int udevcontrol(int argc, char *argv[], char *envp[])
{
- static struct udevd_msg usend_msg;
+ static struct udevd_ctrl_msg ctrl_msg;
struct sockaddr_un saddr;
socklen_t addrlen;
const char *env;
+ const char *arg;
const char *val;
int *intval;
- int i;
int retval = 1;
env = getenv("UDEV_LOG");
udev_log = log_priority(env);
logging_init("udevcontrol");
- dbg("version %s", UDEV_VERSION);
+ dbg("version %s\n", UDEV_VERSION);
if (argc < 2) {
fprintf(stderr, "missing command\n\n");
- usage();
goto exit;
}
-
- memset(&usend_msg, 0x00, sizeof(struct udevd_msg));
- strcpy(usend_msg.magic, UDEV_MAGIC);
-
- for (i = 1 ; i < argc; i++) {
- char *arg = argv[i];
-
- if (!strcmp(arg, "stop_exec_queue"))
- usend_msg.type = UDEVD_STOP_EXEC_QUEUE;
- else if (!strcmp(arg, "start_exec_queue"))
- usend_msg.type = UDEVD_START_EXEC_QUEUE;
- else if (!strcmp(arg, "reload_rules"))
- usend_msg.type = UDEVD_RELOAD_RULES;
- else if (!strncmp(arg, "log_priority=", strlen("log_priority="))) {
- intval = (int *) usend_msg.envbuf;
- val = &arg[strlen("log_priority=")];
- usend_msg.type = UDEVD_SET_LOG_LEVEL;
- *intval = log_priority(val);
- info("send log_priority=%i", *intval);
- } else if (!strncmp(arg, "max_childs=", strlen("max_childs="))) {
- intval = (int *) usend_msg.envbuf;
- val = &arg[strlen("max_childs=")];
- usend_msg.type = UDEVD_SET_MAX_CHILDS;
- *intval = atoi(val);
- info("send max_childs=%i", *intval);
- } else if (strcmp(arg, "help") == 0 || strcmp(arg, "--help") == 0 || strcmp(arg, "-h") == 0) {
- usage();
+ memset(&ctrl_msg, 0x00, sizeof(struct udevd_ctrl_msg));
+ strcpy(ctrl_msg.magic, UDEVD_CTRL_MAGIC);
+ arg = argv[1];
+
+ /* allow instructions passed as options */
+ if (strncmp(arg, "--", 2) == 0)
+ arg += 2;
+
+ if (!strcmp(arg, "stop_exec_queue"))
+ ctrl_msg.type = UDEVD_CTRL_STOP_EXEC_QUEUE;
+ else if (!strcmp(arg, "start_exec_queue"))
+ ctrl_msg.type = UDEVD_CTRL_START_EXEC_QUEUE;
+ else if (!strcmp(arg, "reload_rules"))
+ ctrl_msg.type = UDEVD_CTRL_RELOAD_RULES;
+ else if (!strncmp(arg, "log_priority=", strlen("log_priority="))) {
+ intval = (int *) ctrl_msg.buf;
+ val = &arg[strlen("log_priority=")];
+ ctrl_msg.type = UDEVD_CTRL_SET_LOG_LEVEL;
+ *intval = log_priority(val);
+ info("send log_priority=%i\n", *intval);
+ } else if (!strncmp(arg, "max_childs=", strlen("max_childs="))) {
+ char *endp;
+ int count;
+
+ intval = (int *) ctrl_msg.buf;
+ val = &arg[strlen("max_childs=")];
+ ctrl_msg.type = UDEVD_CTRL_SET_MAX_CHILDS;
+ count = strtoul(val, &endp, 0);
+ if (endp[0] != '\0' || count < 1) {
+ fprintf(stderr, "invalid number\n");
+ goto exit;
+ }
+ *intval = count;
+ info("send max_childs=%i\n", *intval);
+ } else if (!strncmp(arg, "max_childs_running=", strlen("max_childs_running="))) {
+ char *endp;
+ int count;
+
+ intval = (int *) ctrl_msg.buf;
+ val = &arg[strlen("max_childs_running=")];
+ ctrl_msg.type = UDEVD_CTRL_SET_MAX_CHILDS_RUNNING;
+ count = strtoul(val, &endp, 0);
+ if (endp[0] != '\0' || count < 1) {
+ fprintf(stderr, "invalid number\n");
goto exit;
- } else {
- fprintf(stderr, "unknown option\n\n");
- usage();
+ }
+ *intval = count;
+ info("send max_childs_running=%i\n", *intval);
+ } else if (!strncmp(arg, "env", strlen("env"))) {
+ if (!strncmp(arg, "env=", strlen("env=")))
+ val = &arg[strlen("env=")];
+ else
+ val = argv[2];
+ if (val == NULL) {
+ fprintf(stderr, "missing key\n");
goto exit;
}
+ ctrl_msg.type = UDEVD_CTRL_ENV;
+ strlcpy(ctrl_msg.buf, val, sizeof(ctrl_msg.buf));
+ info("send env '%s'\n", val);
+ } else if (strcmp(arg, "help") == 0 || strcmp(arg, "-h") == 0) {
+ printf("Usage: udevadm control COMMAND\n"
+ " --log_priority=<level> set the udev log level for the daemon\n"
+ " --stop_exec_queue keep udevd from executing events, queue only\n"
+ " --start_exec_queue execute events, flush queue\n"
+ " --reload_rules reloads the rules files\n"
+ " --env=<KEY>=<value> set a global environment variable\n"
+ " --max_childs=<N> maximum number of childs\n"
+ " --max_childs_running=<N> maximum number of childs running at the same time\n"
+ " --help print this help text\n\n");
+ goto exit;
+ } else {
+ fprintf(stderr, "unrecognized command '%s'\n", arg);
+ goto exit;
}
if (getuid() != 0) {
- fprintf(stderr, "need to be root, exit\n\n");
+ fprintf(stderr, "root privileges required\n");
goto exit;
}
sock = socket(AF_LOCAL, SOCK_DGRAM, 0);
if (sock == -1) {
- err("error getting socket: %s", strerror(errno));
+ err("error getting socket: %s\n", strerror(errno));
goto exit;
}
memset(&saddr, 0x00, sizeof(struct sockaddr_un));
saddr.sun_family = AF_LOCAL;
/* use abstract namespace for socket path */
- strcpy(&saddr.sun_path[1], UDEVD_SOCK_PATH);
- addrlen = offsetof(struct sockaddr_un, sun_path) + strlen(saddr.sun_path+1) + 1;
+ strcpy(&saddr.sun_path[1], UDEVD_CTRL_SOCK_PATH);
+ addrlen = offsetof(struct sockaddr_un, sun_path) + 1 + strlen(&saddr.sun_path[1]);
- retval = sendto(sock, &usend_msg, sizeof(usend_msg), 0, (struct sockaddr *)&saddr, addrlen);
+ retval = sendto(sock, &ctrl_msg, sizeof(ctrl_msg), 0, (struct sockaddr *)&saddr, addrlen);
if (retval == -1) {
- err("error sending message: %s", strerror(errno));
+ err("error sending message: %s\n", strerror(errno));
retval = 1;
} else {
- dbg("sent message type=0x%02x, %u bytes sent", usend_msg.type, retval);
+ dbg("sent message type=0x%02x, %u bytes sent\n", ctrl_msg.type, retval);
retval = 0;
}