libudev/libudev-util-private.c

   1 /*
   2  * libudev - interface to udev device information
   3  *
   4  * Copyright (C) 2004-2009 Kay Sievers <kay.sievers@vrfy.org>
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  */
  11
  12 #include <stdlib.h>
  13 #include <stdio.h>
  14 #include <stddef.h>
  15 #include <unistd.h>
  16 #include <string.h>
  17 #include <fcntl.h>
  18 #include <errno.h>
  19 #include <ctype.h>
  20 #include <pwd.h>
  21 #include <grp.h>
  22 #include <sys/wait.h>
  23 #include <sys/param.h>
  24
  25 #include "libudev.h"
  26 #include "libudev-private.h"
  27
  28 int util_create_path(struct udev *udev, const char *path)
  29 {
  30         char p[UTIL_PATH_SIZE];
  31         char *pos;
  32         struct stat stats;
  33         int ret;
  34
  35         util_strscpy(p, sizeof(p), path);
  36         pos = strrchr(p, '/');
  37         if (pos == p || pos == NULL)
  38                 return 0;
  39
  40         while (pos[-1] == '/')
  41                 pos--;
  42         pos[0] = '\0';
  43
  44         dbg(udev, "stat '%s'\n", p);
  45         if (stat(p, &stats) == 0 && (stats.st_mode & S_IFMT) == S_IFDIR)
  46                 return 0;
  47
  48         if (util_create_path(udev, p) != 0)
  49                 return -1;
  50
  51         dbg(udev, "mkdir '%s'\n", p);
  52         udev_selinux_setfscreatecon(udev, p, S_IFDIR|0755);
  53         ret = mkdir(p, 0755);
  54         udev_selinux_resetfscreatecon(udev);
  55         if (ret == 0)
  56                 return 0;
  57
  58         if (errno == EEXIST)
  59                 if (stat(p, &stats) == 0 && (stats.st_mode & S_IFMT) == S_IFDIR)
  60                         return 0;
  61         return -1;
  62 }
  63
  64 int util_delete_path(struct udev *udev, const char *path)
  65 {
  66         char p[UTIL_PATH_SIZE];
  67         char *pos;
  68         int retval;
  69
  70         strcpy (p, path);
  71         pos = strrchr(p, '/');
  72         if (pos == p || pos == NULL)
  73                 return 0;
  74
  75         while (1) {
  76                 *pos = '\0';
  77                 pos = strrchr(p, '/');
  78
  79                 /* don't remove the last one */
  80                 if ((pos == p) || (pos == NULL))
  81                         break;
  82
  83                 /* remove if empty */
  84                 retval = rmdir(p);
  85                 if (errno == ENOENT)
  86                         retval = 0;
  87                 if (retval) {
  88                         if (errno == ENOTEMPTY)
  89                                 return 0;
  90                         err(udev, "rmdir(%s) failed: %m\n", p);
  91                         break;
  92                 }
  93                 dbg(udev, "removed '%s'\n", p);
  94         }
  95         return 0;
  96 }
  97
  98 /* Reset permissions on the device node, before unlinking it to make sure,
  99  * that permisions of possible hard links will be removed too.
 100  */
 101 int util_unlink_secure(struct udev *udev, const char *filename)
 102 {
 103         int retval;
 104
 105         retval = chown(filename, 0, 0);
 106         if (retval)
 107                 err(udev, "chown(%s, 0, 0) failed: %m\n", filename);
 108
 109         retval = chmod(filename, 0000);
 110         if (retval)
 111                 err(udev, "chmod(%s, 0000) failed: %m\n", filename);
 112
 113         retval = unlink(filename);
 114         if (errno == ENOENT)
 115                 retval = 0;
 116
 117         if (retval)
 118                 err(udev, "unlink(%s) failed: %m\n", filename);
 119
 120         return retval;
 121 }
 122
 123 uid_t util_lookup_user(struct udev *udev, const char *user)
 124 {
 125         char *endptr;
 126         int buflen = sysconf(_SC_GETPW_R_SIZE_MAX);
 127         char buf[buflen];
 128         struct passwd pwbuf;
 129         struct passwd *pw;
 130         uid_t uid;
 131
 132         if (strcmp(user, "root") == 0)
 133                 return 0;
 134         uid = strtoul(user, &endptr, 10);
 135         if (endptr[0] == '\0')
 136                 return uid;
 137
 138         errno = 0;
 139         getpwnam_r(user, &pwbuf, buf, buflen, &pw);
 140         if (pw != NULL)
 141                 return pw->pw_uid;
 142         if (errno == 0 || errno == ENOENT || errno == ESRCH)
 143                 err(udev, "specified user '%s' unknown\n", user);
 144         else
 145                 err(udev, "error resolving user '%s': %m\n", user);
 146         return 0;
 147 }
 148
 149 gid_t util_lookup_group(struct udev *udev, const char *group)
 150 {
 151         char *endptr;
 152         int buflen;
 153         char *buf;
 154         struct group grbuf;
 155         struct group *gr;
 156         gid_t gid = 0;
 157
 158         if (strcmp(group, "root") == 0)
 159                 return 0;
 160         gid = strtoul(group, &endptr, 10);
 161         if (endptr[0] == '\0')
 162                 return gid;
 163
 164         buflen = sysconf(_SC_GETGR_R_SIZE_MAX);
 165         if (buflen < 0)
 166                 buflen = 1000;
 167         buf = NULL;
 168         gid = 0;
 169         for (;;) {
 170                 buf = realloc(buf, buflen);
 171                 if (!buf)
 172                         break;
 173                 errno = 0;
 174                 getgrnam_r(group, &grbuf, buf, buflen, &gr);
 175                 if (gr != NULL)
 176                         gid = gr->gr_gid;
 177                 else if (errno == ERANGE) {
 178                         buflen *= 2;
 179                         continue;
 180                 }
 181                 else if (errno == 0 || errno == ENOENT || errno == ESRCH)
 182                         err(udev, "specified group '%s' unknown\n", group);
 183                 else
 184                         err(udev, "error resolving group '%s': %m\n", group);
 185                 break;
 186         }
 187         free(buf);
 188         return gid;
 189 }
 190
 191 /* handle "[<SUBSYSTEM>/<KERNEL>]<attribute>" format */
 192 int util_resolve_subsys_kernel(struct udev *udev, const char *string,
 193                                char *result, size_t maxsize, int read_value)
 194 {
 195         char temp[UTIL_PATH_SIZE];
 196         char *subsys;
 197         char *sysname;
 198         struct udev_device *dev;
 199         char *attr;
 200
 201         if (string[0] != '[')
 202                 return -1;
 203
 204         util_strscpy(temp, sizeof(temp), string);
 205
 206         subsys = &temp[1];
 207
 208         sysname = strchr(subsys, '/');
 209         if (sysname == NULL)
 210                 return -1;
 211         sysname[0] = '\0';
 212         sysname = &sysname[1];
 213
 214         attr = strchr(sysname, ']');
 215         if (attr == NULL)
 216                 return -1;
 217         attr[0] = '\0';
 218         attr = &attr[1];
 219         if (attr[0] == '/')
 220                 attr = &attr[1];
 221         if (attr[0] == '\0')
 222                 attr = NULL;
 223
 224         if (read_value && attr == NULL)
 225                 return -1;
 226
 227         dev = udev_device_new_from_subsystem_sysname(udev, subsys, sysname);
 228         if (dev == NULL)
 229                 return -1;
 230
 231         if (read_value) {
 232                 const char *val;
 233
 234                 val = udev_device_get_sysattr_value(dev, attr);
 235                 if (val != NULL)
 236                         util_strscpy(result, maxsize, val);
 237                 else
 238                         result[0] = '\0';
 239                 info(udev, "value '[%s/%s]%s' is '%s'\n", subsys, sysname, attr, result);
 240         } else {
 241                 size_t l;
 242                 char *s;
 243
 244                 s = result;
 245                 l = util_strpcpyl(&s, maxsize, udev_device_get_syspath(dev), NULL);
 246                 if (attr != NULL)
 247                         util_strpcpyl(&s, l, "/", attr, NULL);
 248                 info(udev, "path '[%s/%s]%s' is '%s'\n", subsys, sysname, attr, result);
 249         }
 250         udev_device_unref(dev);
 251         return 0;
 252 }
 253
 254 int util_run_program(struct udev *udev, const char *command, char **envp,
 255                      char *result, size_t ressize, size_t *reslen)
 256 {
 257         int status;
 258         int outpipe[2] = {-1, -1};
 259         int errpipe[2] = {-1, -1};
 260         pid_t pid;
 261         char arg[UTIL_PATH_SIZE];
 262         char program[UTIL_PATH_SIZE];
 263         char *argv[(sizeof(arg) / 2) + 1];
 264         int devnull;
 265         int i;
 266         int err = 0;
 267
 268         info(udev, "'%s' started\n", command);
 269
 270         /* build argv from command */
 271         util_strscpy(arg, sizeof(arg), command);
 272         i = 0;
 273         if (strchr(arg, ' ') != NULL) {
 274                 char *pos = arg;
 275
 276                 while (pos != NULL && pos[0] != '\0') {
 277                         if (pos[0] == '\'') {
 278                                 /* do not separate quotes */
 279                                 pos++;
 280                                 argv[i] = strsep(&pos, "\'");
 281                                 while (pos != NULL && pos[0] == ' ')
 282                                         pos++;
 283                         } else {
 284                                 argv[i] = strsep(&pos, " ");
 285                         }
 286                         dbg(udev, "arg[%i] '%s'\n", i, argv[i]);
 287                         i++;
 288                 }
 289                 argv[i] = NULL;
 290         } else {
 291                 argv[0] = arg;
 292                 argv[1] = NULL;
 293         }
 294
 295         /* prepare pipes from child to parent */
 296         if (result != NULL || udev_get_log_priority(udev) >= LOG_INFO) {
 297                 if (pipe(outpipe) != 0) {
 298                         err(udev, "pipe failed: %m\n");
 299                         return -1;
 300                 }
 301         }
 302         if (udev_get_log_priority(udev) >= LOG_INFO) {
 303                 if (pipe(errpipe) != 0) {
 304                         err(udev, "pipe failed: %m\n");
 305                         return -1;
 306                 }
 307         }
 308
 309         /* allow programs in /lib/udev/ to be called without the path */
 310         if (argv[0][0] != '/') {
 311                 util_strscpyl(program, sizeof(program), LIBEXECDIR "/", argv[0], NULL);
 312                 argv[0] = program;
 313         }
 314
 315         pid = fork();
 316         switch(pid) {
 317         case 0:
 318                 /* child closes parent ends of pipes */
 319                 if (outpipe[READ_END] > 0)
 320                         close(outpipe[READ_END]);
 321                 if (errpipe[READ_END] > 0)
 322                         close(errpipe[READ_END]);
 323
 324                 /* discard child output or connect to pipe */
 325                 devnull = open("/dev/null", O_RDWR);
 326                 if (devnull > 0) {
 327                         dup2(devnull, STDIN_FILENO);
 328                         if (outpipe[WRITE_END] < 0)
 329                                 dup2(devnull, STDOUT_FILENO);
 330                         if (errpipe[WRITE_END] < 0)
 331                                 dup2(devnull, STDERR_FILENO);
 332                         close(devnull);
 333                 } else
 334                         err(udev, "open /dev/null failed: %m\n");
 335                 if (outpipe[WRITE_END] > 0) {
 336                         dup2(outpipe[WRITE_END], STDOUT_FILENO);
 337                         close(outpipe[WRITE_END]);
 338                 }
 339                 if (errpipe[WRITE_END] > 0) {
 340                         dup2(errpipe[WRITE_END], STDERR_FILENO);
 341                         close(errpipe[WRITE_END]);
 342                 }
 343                 execve(argv[0], argv, envp);
 344                 if (errno == ENOENT || errno == ENOTDIR) {
 345                         /* may be on a filesytem which is not mounted right now */
 346                         info(udev, "program '%s' not found\n", argv[0]);
 347                 } else {
 348                         /* other problems */
 349                         err(udev, "exec of program '%s' failed\n", argv[0]);
 350                 }
 351                 _exit(1);
 352         case -1:
 353                 err(udev, "fork of '%s' failed: %m\n", argv[0]);
 354                 return -1;
 355         default:
 356                 /* read from child if requested */
 357                 if (outpipe[READ_END] > 0 || errpipe[READ_END] > 0) {
 358                         ssize_t count;
 359                         size_t respos = 0;
 360
 361                         /* parent closes child ends of pipes */
 362                         if (outpipe[WRITE_END] > 0)
 363                                 close(outpipe[WRITE_END]);
 364                         if (errpipe[WRITE_END] > 0)
 365                                 close(errpipe[WRITE_END]);
 366
 367                         /* read child output */
 368                         while (outpipe[READ_END] > 0 || errpipe[READ_END] > 0) {
 369                                 int fdcount;
 370                                 fd_set readfds;
 371
 372                                 FD_ZERO(&readfds);
 373                                 if (outpipe[READ_END] > 0)
 374                                         FD_SET(outpipe[READ_END], &readfds);
 375                                 if (errpipe[READ_END] > 0)
 376                                         FD_SET(errpipe[READ_END], &readfds);
 377                                 fdcount = select(MAX(outpipe[READ_END], errpipe[READ_END])+1, &readfds, NULL, NULL, NULL);
 378                                 if (fdcount < 0) {
 379                                         if (errno == EINTR)
 380                                                 continue;
 381                                         err = -1;
 382                                         break;
 383                                 }
 384
 385                                 /* get stdout */
 386                                 if (outpipe[READ_END] > 0 && FD_ISSET(outpipe[READ_END], &readfds)) {
 387                                         char inbuf[1024];
 388                                         char *pos;
 389                                         char *line;
 390
 391                                         count = read(outpipe[READ_END], inbuf, sizeof(inbuf)-1);
 392                                         if (count <= 0) {
 393                                                 close(outpipe[READ_END]);
 394                                                 outpipe[READ_END] = -1;
 395                                                 if (count < 0) {
 396                                                         err(udev, "stdin read failed: %m\n");
 397                                                         err = -1;
 398                                                 }
 399                                                 continue;
 400                                         }
 401                                         inbuf[count] = '\0';
 402
 403                                         /* store result for rule processing */
 404                                         if (result) {
 405                                                 if (respos + count < ressize) {
 406                                                         memcpy(&result[respos], inbuf, count);
 407                                                         respos += count;
 408                                                 } else {
 409                                                         err(udev, "ressize %ld too short\n", (long)ressize);
 410                                                         err = -1;
 411                                                 }
 412                                         }
 413                                         pos = inbuf;
 414                                         while ((line = strsep(&pos, "\n")))
 415                                                 if (pos || line[0] != '\0')
 416                                                         info(udev, "'%s' (stdout) '%s'\n", argv[0], line);
 417                                 }
 418
 419                                 /* get stderr */
 420                                 if (errpipe[READ_END] > 0 && FD_ISSET(errpipe[READ_END], &readfds)) {
 421                                         char errbuf[1024];
 422                                         char *pos;
 423                                         char *line;
 424
 425                                         count = read(errpipe[READ_END], errbuf, sizeof(errbuf)-1);
 426                                         if (count <= 0) {
 427                                                 close(errpipe[READ_END]);
 428                                                 errpipe[READ_END] = -1;
 429                                                 if (count < 0)
 430                                                         err(udev, "stderr read failed: %m\n");
 431                                                 continue;
 432                                         }
 433                                         errbuf[count] = '\0';
 434                                         pos = errbuf;
 435                                         while ((line = strsep(&pos, "\n")))
 436                                                 if (pos || line[0] != '\0')
 437                                                         info(udev, "'%s' (stderr) '%s'\n", argv[0], line);
 438                                 }
 439                         }
 440                         if (outpipe[READ_END] > 0)
 441                                 close(outpipe[READ_END]);
 442                         if (errpipe[READ_END] > 0)
 443                                 close(errpipe[READ_END]);
 444
 445                         /* return the childs stdout string */
 446                         if (result) {
 447                                 result[respos] = '\0';
 448                                 dbg(udev, "result='%s'\n", result);
 449                                 if (reslen)
 450                                         *reslen = respos;
 451                         }
 452                 }
 453                 waitpid(pid, &status, 0);
 454                 if (WIFEXITED(status)) {
 455                         info(udev, "'%s' returned with exitcode %i\n", command, WEXITSTATUS(status));
 456                         if (WEXITSTATUS(status) != 0)
 457                                 err = -1;
 458                 } else {
 459                         err(udev, "'%s' unexpected exit with status 0x%04x\n", command, status);
 460                         err = -1;
 461                 }
 462         }
 463         return err;
 464 }