chiark / gitweb /
udevd: implement a more efficient queue file format
[elogind.git] / udev / udevadm-settle.c
index 0cb9ece472ebc2e05a78cf888ef5adf0fa411bbd..f1052aa14896f11d698736c7b0fd2b2f06dce2bf 100644 (file)
@@ -1,19 +1,20 @@
 /*
- * Copyright (C) 2006 Kay Sievers <kay@vrfy.org>
+ * Copyright (C) 2006-2008 Kay Sievers <kay@vrfy.org>
+ * Copyright (C) 2009 Canonical Ltd.
+ * Copyright (C) 2009 Scott James Remnant <scott@netsplit.com>
  *
- *     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
- *     Free Software Foundation version 2 of the License.
- * 
- *     This program is distributed in the hope that it will be useful, but
- *     WITHOUT ANY WARRANTY; without even the implied warranty of
- *     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
- *     General Public License for more details.
- * 
- *     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.,
- *     51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
+ * 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 Free Software Foundation, either version 2 of the License, or
+ * (at your option) any later version.
  *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
 
 #include <stdlib.h>
 #include <fcntl.h>
 #include <syslog.h>
 #include <getopt.h>
+#include <signal.h>
 #include <sys/stat.h>
 #include <sys/types.h>
 
 #include "udev.h"
-#include "udevd.h"
 
 #define DEFAULT_TIMEOUT                        180
 #define LOOP_PER_SECOND                        20
 
-static void print_queue(const char *dir)
-{
-       LIST_HEAD(files);
-       struct name_entry *item;
-
-       if (add_matching_files(&files, dir, NULL) < 0)
-               return;
-
-       printf("\n\nAfter the udevadm settle timeout, the events queue contains:\n\n");
-
-       list_for_each_entry(item, &files, node) {
-               char target[NAME_SIZE];
-               size_t len;
-               const char *filename = strrchr(item->name, '/');
+static volatile sig_atomic_t is_timeout;
 
-               if (filename == NULL)
-                       continue;
-               filename++;
-               if (*filename == '\0')
-                       continue;
-
-               len = readlink(item->name, target, sizeof(target));
-               if (len < 0)
-                       continue;
-               target[len] = '\0';
-
-               printf("%s: %s\n", filename, target);
+static void sig_handler(int signum)
+{
+       switch (signum) {
+               case SIGALRM:
+                       is_timeout = 1;
+               case SIGUSR1:
+                       ;
        }
-
-       printf("\n\n");
 }
 
-int udevadm_settle(int argc, char *argv[])
+int udevadm_settle(struct udev *udev, int argc, char *argv[])
 {
-       char queuename[PATH_SIZE];
-       char filename[PATH_SIZE];
-       unsigned long long seq_kernel;
-       unsigned long long seq_udev;
-       char seqnum[32];
-       int fd;
-       ssize_t len;
-       int timeout = DEFAULT_TIMEOUT;
-       int loop;
        static const struct option options[] = {
-               { "timeout", 1, NULL, 't' },
-               { "help", 0, NULL, 'h' },
+               { "seq-start", required_argument, NULL, 's' },
+               { "seq-end", required_argument, NULL, 'e' },
+               { "timeout", required_argument, NULL, 't' },
+               { "quiet", no_argument, NULL, 'q' },
+               { "help", no_argument, NULL, 'h' },
                {}
        };
-       int option;
-       int rc = 1;
-       int seconds;
-
-       logging_init("udevsettle");
-       udev_config_init();
-       dbg("version %s\n", VERSION);
-       sysfs_init();
+       unsigned long long start = 0;
+       unsigned long long end = 0;
+       int quiet = 0;
+       int timeout = 0;
+       struct sigaction act;
+       struct udev_queue *udev_queue = NULL;
+       int rc = 0;
+
+       dbg(udev, "version %s\n", VERSION);
+
+       /* set signal handlers */
+       memset(&act, 0x00, sizeof(act));
+       act.sa_handler = sig_handler;
+       sigemptyset (&act.sa_mask);
+       act.sa_flags = 0;
+       sigaction(SIGALRM, &act, NULL);
+       sigaction(SIGUSR1, &act, NULL);
 
        while (1) {
-               option = getopt_long(argc, argv, "t:h", options, NULL);
+               int option;
+               int seconds;
+
+               option = getopt_long(argc, argv, "s:e:t:qh", options, NULL);
                if (option == -1)
                        break;
 
                switch (option) {
+               case 's':
+                       start = strtoull(optarg, NULL, 0);
+                       break;
+               case 'e':
+                       end = strtoull(optarg, NULL, 0);
+                       break;
                case 't':
                        seconds = atoi(optarg);
-                       if (seconds > 0)
+                       if (seconds >= 0)
                                timeout = seconds;
                        else
                                fprintf(stderr, "invalid timeout value\n");
-                       dbg("timeout=%i\n", timeout);
+                       dbg(udev, "timeout=%i\n", timeout);
+                       break;
+               case 'q':
+                       quiet = 1;
                        break;
                case 'h':
-                       printf("Usage: udevadm settle [--help] [--timeout=<seconds>]\n\n");
+                       printf("Usage: udevadm settle OPTIONS\n"
+                              "  --timeout=<seconds>   maximum time to wait for events\n"
+                              "  --seq-start=<seqnum>  first seqnum to wait for\n"
+                              "  --seq-end=<seqnum>    last seqnum to wait for\n"
+                              "  --quiet               do not print list after timeout\n"
+                              "  --help\n\n");
                        goto exit;
                }
        }
 
-       strlcpy(queuename, udev_root, sizeof(queuename));
-       strlcat(queuename, "/.udev/queue", sizeof(queuename));
+       if (timeout > 0)
+               alarm(timeout);
+       else
+               alarm(DEFAULT_TIMEOUT);
 
-       loop = timeout * LOOP_PER_SECOND;
-       while (loop--) {
-               /* wait for events in queue to finish */
-               while (loop--) {
-                       struct stat statbuf;
+       udev_queue = udev_queue_new(udev);
+       if (udev_queue == NULL)
+               goto exit;
 
-                       if (stat(queuename, &statbuf) < 0) {
-                               info("queue is empty\n");
-                               break;
-                       }
-                       usleep(1000 * 1000 / LOOP_PER_SECOND);
+       if (start > 0) {
+               unsigned long long kernel_seq;
+
+               kernel_seq = udev_queue_get_kernel_seqnum(udev_queue);
+
+               /* unless specified, the last event is the current kernel seqnum */
+               if (end == 0)
+                       end = udev_queue_get_kernel_seqnum(udev_queue);
+
+               if (start > end) {
+                       err(udev, "seq-start larger than seq-end, ignoring\n");
+                       fprintf(stderr, "seq-start larger than seq-end, ignoring\n");
+                       start = 0;
+                       end = 0;
                }
-               if (loop <= 0) {
-                       info("timeout waiting for queue\n");
-                       print_queue(queuename);
-                       goto exit;
+
+               if (start > kernel_seq || end > kernel_seq) {
+                       err(udev, "seq-start or seq-end larger than current kernel value, ignoring\n");
+                       fprintf(stderr, "seq-start or seq-end larger than current kernel value, ignoring\n");
+                       start = 0;
+                       end = 0;
+               }
+               info(udev, "start=%llu end=%llu current=%llu\n", start, end, kernel_seq);
+       } else {
+               if (end > 0) {
+                       err(udev, "seq-end needs seq-start parameter, ignoring\n");
+                       fprintf(stderr, "seq-end needs seq-start parameter, ignoring\n");
+                       end = 0;
                }
+       }
 
-               /* read current udev seqnum */
-               strlcpy(filename, udev_root, sizeof(filename));
-               strlcat(filename, "/.udev/uevent_seqnum", sizeof(filename));
-               fd = open(filename, O_RDONLY);
-               if (fd < 0)
-                       goto exit;
-               len = read(fd, seqnum, sizeof(seqnum)-1);
-               close(fd);
-               if (len <= 0)
-                       goto exit;
-               seqnum[len] = '\0';
-               seq_udev = strtoull(seqnum, NULL, 10);
-               info("udev seqnum = %llu\n", seq_udev);
-
-               /* read current kernel seqnum */
-               strlcpy(filename, sysfs_path, sizeof(filename));
-               strlcat(filename, "/kernel/uevent_seqnum", sizeof(filename));
-               fd = open(filename, O_RDONLY);
-               if (fd < 0)
-                       goto exit;
-               len = read(fd, seqnum, sizeof(seqnum)-1);
-               close(fd);
-               if (len <= 0)
-                       goto exit;
-               seqnum[len] = '\0';
-               seq_kernel = strtoull(seqnum, NULL, 10);
-               info("kernel seqnum = %llu\n", seq_kernel);
-
-               /* make sure all kernel events have arrived in the queue */
-               if (seq_udev >= seq_kernel) {
-                       info("queue is empty and no pending events left\n");
-                       rc = 0;
-                       goto exit;
+       /* guarantee that the udev daemon isn't pre-processing */
+       if (getuid() == 0) {
+               struct udev_ctrl *uctrl;
+
+               uctrl = udev_ctrl_new_from_socket(udev, UDEV_CTRL_SOCK_PATH);
+               if (uctrl != NULL) {
+                       sigset_t mask, oldmask;
+
+                       sigemptyset(&mask);
+                       sigaddset(&mask, SIGUSR1);
+                       sigaddset(&mask, SIGALRM);
+                       sigprocmask(SIG_BLOCK, &mask, &oldmask);
+                       if (udev_ctrl_send_settle(uctrl) > 0)
+                               sigsuspend(&oldmask);
+                       sigprocmask(SIG_SETMASK, &oldmask, NULL);
+                       udev_ctrl_unref(uctrl);
                }
+       }
+
+       while (!is_timeout) {
+               if (start > 0) {
+                       /* if asked for, wait for a specific sequence of events */
+                       if (udev_queue_get_seqnum_sequence_is_finished(udev_queue, start, end) == 1)
+                               break;
+               } else {
+                       /* exit if queue is empty */
+                       if (udev_queue_get_queue_is_empty(udev_queue))
+                               break;
+               }
+
                usleep(1000 * 1000 / LOOP_PER_SECOND);
-               info("queue is empty, but events still pending\n");
        }
 
+       /* if we reached the timeout, print the list of remaining events */
+       if (is_timeout) {
+               struct udev_list_entry *list_entry;
+
+               if (!quiet) {
+                       info(udev, "timeout waiting for udev queue\n");
+                       printf("\nudevadm settle - timeout of %i seconds reached, the event queue contains:\n", timeout);
+                       udev_list_entry_foreach(list_entry, udev_queue_get_queued_list_entry(udev_queue))
+                               printf("  %s (%s)\n",
+                                      udev_list_entry_get_name(list_entry),
+                                      udev_list_entry_get_value(list_entry));
+               }
+               rc = 1;
+       }
 exit:
-       sysfs_cleanup();
-       logging_close();
+       udev_queue_unref(udev_queue);
        return rc;
 }