udevtrigger.c

   1 /*
   2  * Copyright (C) 2004-2006 Kay Sievers <kay@vrfy.org>
   3  * Copyright (C) 2006 Hannes Reinecke <hare@suse.de>
   4  *
   5  *      This program is free software; you can redistribute it and/or modify it
   6  *      under the terms of the GNU General Public License as published by the
   7  *      Free Software Foundation version 2 of the License.
   8  *
   9  *      This program is distributed in the hope that it will be useful, but
  10  *      WITHOUT ANY WARRANTY; without even the implied warranty of
  11  *      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  *      General Public License for more details.
  13  *
  14  *      You should have received a copy of the GNU General Public License along
  15  *      with this program; if not, write to the Free Software Foundation, Inc.,
  16  *      51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  17  *
  18  */
  19
  20 #include <stdlib.h>
  21 #include <stddef.h>
  22 #include <stdarg.h>
  23 #include <stdio.h>
  24 #include <stdbool.h>
  25 #include <unistd.h>
  26 #include <getopt.h>
  27 #include <errno.h>
  28 #include <dirent.h>
  29 #include <fcntl.h>
  30 #include <syslog.h>
  31 #include <sys/stat.h>
  32 #include <sys/types.h>
  33 #include <string.h>
  34
  35 #define PATH_SIZE 512
  36
  37 #ifndef strlcpy
  38 #define strlcpy(d,s,l) (strncpy(d,s,l), (d)[(l)-1] = '\0')
  39 #endif
  40
  41 #ifndef strlcat
  42 #define strlcat(d,s,l) strncat(d,s,(l)-strlen(d)-1)
  43 #endif
  44
  45 static int verbose;
  46 static int dry_run;
  47
  48 static void log_message(int priority, const char *format, ...)
  49 {
  50         va_list args;
  51
  52         va_start(args, format);
  53         vsyslog(priority, format, args);
  54         va_end(args);
  55 }
  56
  57 #undef err
  58 #define err(format, arg...)                         \
  59     do {                                    \
  60         log_message(LOG_ERR ,"%s: " format ,__FUNCTION__ ,## arg);  \
  61     } while (0)
  62
  63 #undef info
  64 #define info(format, arg...)                            \
  65     do {                                    \
  66         log_message(LOG_INFO ,"%s: " format ,__FUNCTION__ ,## arg); \
  67     } while (0)
  68
  69 #ifdef DEBUG
  70 #undef dbg
  71 #define dbg(format, arg...)                         \
  72     do {                                    \
  73         log_message(LOG_DEBUG ,"%s: " format ,__FUNCTION__ ,## arg);    \
  74     } while (0)
  75 #else
  76 #define dbg(...) do {} while(0)
  77 #endif
  78
  79
  80 static void trigger_uevent(const char *devpath)
  81 {
  82         char filename[PATH_SIZE];
  83         int fd;
  84
  85         strlcpy(filename, "/sys", sizeof(filename));
  86         strlcat(filename, devpath, sizeof(filename));
  87         strlcat(filename, "/uevent", sizeof(filename));
  88
  89         if (verbose)
  90                 printf("%s\n", devpath);
  91
  92         if (dry_run)
  93                 return;
  94
  95         fd = open(filename, O_WRONLY);
  96         if (fd < 0) {
  97                 dbg("error on opening %s: %s\n", filename, strerror(errno));
  98                 return;
  99         }
 100
 101         if (write(fd, "add", 3) < 0)
 102                 info("error on triggering %s: %s\n", filename, strerror(errno));
 103
 104         close(fd);
 105 }
 106
 107 static int sysfs_resolve_link(char *devpath, size_t size)
 108 {
 109         char link_path[PATH_SIZE];
 110         char link_target[PATH_SIZE];
 111         int len;
 112         int i;
 113         int back;
 114
 115         strlcpy(link_path, "/sys", sizeof(link_path));
 116         strlcat(link_path, devpath, sizeof(link_path));
 117         len = readlink(link_path, link_target, sizeof(link_target));
 118         if (len <= 0)
 119                 return -1;
 120         link_target[len] = '\0';
 121         dbg("path link '%s' points to '%s'", devpath, link_target);
 122
 123         for (back = 0; strncmp(&link_target[back * 3], "../", 3) == 0; back++)
 124                 ;
 125         dbg("base '%s', tail '%s', back %i", devpath, &link_target[back * 3], back);
 126         for (i = 0; i <= back; i++) {
 127                 char *pos = strrchr(devpath, '/');
 128
 129                 if (pos == NULL)
 130                         return -1;
 131                 pos[0] = '\0';
 132         }
 133         dbg("after moving back '%s'", devpath);
 134         strlcat(devpath, "/", size);
 135         strlcat(devpath, &link_target[back * 3], size);
 136         return 0;
 137 }
 138
 139 static bool device_has_attribute(const char *path, const char *attr,
 140                                  mode_t mode)
 141 {
 142         char filename[PATH_SIZE];
 143         struct stat statbuf;
 144
 145         strlcpy(filename, path, sizeof(filename));
 146         strlcat(filename, attr, sizeof(filename));
 147
 148         if (stat(filename, &statbuf) < 0)
 149                 return false;
 150
 151         if (!(statbuf.st_mode & mode))
 152                 return false;
 153
 154         return true;
 155 }
 156
 157 static int device_list_insert(const char *path)
 158 {
 159         char devpath[PATH_SIZE];
 160         struct stat statbuf;
 161
 162         dbg("add '%s'" , path);
 163
 164         /* we only have a device, if we have a dev and an uevent file */
 165         if (!device_has_attribute(path, "/dev", S_IRUSR) ||
 166             !device_has_attribute(path, "/uevent", S_IWUSR))
 167                 return -1;
 168
 169         strlcpy(devpath, &path[4], sizeof(devpath));
 170
 171         /* resolve possible link to real target */
 172         if (lstat(path, &statbuf) < 0)
 173                 return -1;
 174         if (S_ISLNK(statbuf.st_mode))
 175                 if (sysfs_resolve_link(devpath, sizeof(devpath)) != 0)
 176                         return -1;
 177
 178         trigger_uevent(devpath);
 179         return 0;
 180 }
 181
 182
 183 static void scan_subsystem(const char *subsys)
 184 {
 185         char base[PATH_SIZE];
 186         DIR *dir;
 187         struct dirent *dent;
 188
 189         strlcpy(base, "/sys/", sizeof(base));
 190         strlcat(base, subsys, sizeof(base));
 191
 192         dir = opendir(base);
 193         if (dir != NULL) {
 194                 for (dent = readdir(dir); dent != NULL; dent = readdir(dir)) {
 195                         char dirname[PATH_SIZE];
 196                         DIR *dir2;
 197                         struct dirent *dent2;
 198
 199                         if (dent->d_name[0] == '.')
 200                                 continue;
 201
 202                         strlcpy(dirname, base, sizeof(dirname));
 203                         strlcat(dirname, "/", sizeof(dirname));
 204                         strlcat(dirname, dent->d_name, sizeof(dirname));
 205                         strlcat(dirname, "/devices", sizeof(dirname));
 206
 207                         /* look for devices */
 208                         dir2 = opendir(dirname);
 209                         if (dir2 != NULL) {
 210                                 for (dent2 = readdir(dir2); dent2 != NULL; dent2 = readdir(dir2)) {
 211                                         char dirname2[PATH_SIZE];
 212
 213                                         if (dent2->d_name[0] == '.')
 214                                                 continue;
 215
 216                                         strlcpy(dirname2, dirname, sizeof(dirname2));
 217                                         strlcat(dirname2, "/", sizeof(dirname2));
 218                                         strlcat(dirname2, dent2->d_name, sizeof(dirname2));
 219                                         device_list_insert(dirname2);
 220                                 }
 221                                 closedir(dir2);
 222                         }
 223                 }
 224                 closedir(dir);
 225         }
 226 }
 227
 228 static void scan_block(void)
 229 {
 230         char base[PATH_SIZE];
 231         DIR *dir;
 232         struct dirent *dent;
 233
 234         strlcpy(base, "/sys/block", sizeof(base));
 235
 236         dir = opendir(base);
 237         if (dir != NULL) {
 238                 for (dent = readdir(dir); dent != NULL; dent = readdir(dir)) {
 239                         char dirname[PATH_SIZE];
 240                         DIR *dir2;
 241                         struct dirent *dent2;
 242
 243                         if (dent->d_name[0] == '.')
 244                                 continue;
 245
 246                         strlcpy(dirname, base, sizeof(dirname));
 247                         strlcat(dirname, "/", sizeof(dirname));
 248                         strlcat(dirname, dent->d_name, sizeof(dirname));
 249                         if (device_list_insert(dirname) != 0)
 250                                 continue;
 251
 252                         /* look for partitions */
 253                         dir2 = opendir(dirname);
 254                         if (dir2 != NULL) {
 255                                 for (dent2 = readdir(dir2); dent2 != NULL; dent2 = readdir(dir2)) {
 256                                         char dirname2[PATH_SIZE];
 257
 258                                         if (dent2->d_name[0] == '.')
 259                                                 continue;
 260
 261                                         if (!strcmp(dent2->d_name,"device"))
 262                                                 continue;
 263
 264                                         strlcpy(dirname2, dirname, sizeof(dirname2));
 265                                         strlcat(dirname2, "/", sizeof(dirname2));
 266                                         strlcat(dirname2, dent2->d_name, sizeof(dirname2));
 267                                         device_list_insert(dirname2);
 268                                 }
 269                                 closedir(dir2);
 270                         }
 271                 }
 272                 closedir(dir);
 273         }
 274 }
 275
 276 static void scan_class(void)
 277 {
 278         char base[PATH_SIZE];
 279         DIR *dir;
 280         struct dirent *dent;
 281
 282         strlcpy(base, "/sys/class", sizeof(base));
 283
 284         dir = opendir(base);
 285         if (dir != NULL) {
 286                 for (dent = readdir(dir); dent != NULL; dent = readdir(dir)) {
 287                         char dirname[PATH_SIZE];
 288                         DIR *dir2;
 289                         struct dirent *dent2;
 290
 291                         if (dent->d_name[0] == '.')
 292                                 continue;
 293
 294                         strlcpy(dirname, base, sizeof(dirname));
 295                         strlcat(dirname, "/", sizeof(dirname));
 296                         strlcat(dirname, dent->d_name, sizeof(dirname));
 297                         dir2 = opendir(dirname);
 298                         if (dir2 != NULL) {
 299                                 for (dent2 = readdir(dir2); dent2 != NULL; dent2 = readdir(dir2)) {
 300                                         char dirname2[PATH_SIZE];
 301
 302                                         if (dent2->d_name[0] == '.')
 303                                                 continue;
 304
 305                                         if (!strcmp(dent2->d_name, "device"))
 306                                                 continue;
 307
 308                                         strlcpy(dirname2, dirname, sizeof(dirname2));
 309                                         strlcat(dirname2, "/", sizeof(dirname2));
 310                                         strlcat(dirname2, dent2->d_name, sizeof(dirname2));
 311                                         device_list_insert(dirname2);
 312                                 }
 313                                 closedir(dir2);
 314                         }
 315                 }
 316                 closedir(dir);
 317         }
 318 }
 319
 320 int main(int argc, char *argv[], char *envp[])
 321 {
 322         char base[PATH_SIZE];
 323         struct stat statbuf;
 324         int option;
 325
 326         openlog("udevtrigger", LOG_PID | LOG_CONS, LOG_DAEMON);
 327
 328         while (1) {
 329                 option = getopt(argc, argv, "vnh");
 330                 if (option == -1)
 331                         break;
 332
 333                 switch (option) {
 334                 case 'v':
 335                         verbose = 1;
 336                         break;
 337                 case 'n':
 338                         dry_run = 1;
 339                         break;
 340                 case 'h':
 341                         printf("Usage: udevtrigger OPTIONS\n"
 342                                "  -v                     print the list of devices while running\n"
 343                                "  -n                     do not actually trigger the events\n"
 344                                "  -h                     print this text\n"
 345                                "\n");
 346                         goto exit;
 347                 default:
 348                         goto exit;
 349                 }
 350         }
 351
 352
 353         /* if we have /sys/subsystem, forget all the old stuff */
 354         scan_subsystem("bus");
 355         scan_class();
 356
 357         /* scan "block" if it isn't a "class" */
 358         strlcpy(base, "/sys/class/block", sizeof(base));
 359         if (stat(base, &statbuf) != 0)
 360                 scan_block();
 361
 362 exit:
 363
 364         closelog();
 365         return 0;
 366 }