sbin/devfs/rule.c

   1 /*-
   2  * Copyright (c) 2002 Dima Dorfman.
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  24  * SUCH DAMAGE.
  25  */
  26
  27 /*
  28  * Rule subsystem manipulation.
  29  */
  30
  31 #include <sys/cdefs.h>
  32 __FBSDID("$FreeBSD$");
  33
  34 #include <sys/param.h>
  35 #include <sys/conf.h>
  36 #include <sys/ioctl.h>
  37
  38 #include <assert.h>
  39 #include <err.h>
  40 #include <errno.h>
  41 #include <grp.h>
  42 #include <pwd.h>
  43 #include <stdio.h>
  44 #include <stdlib.h>
  45 #include <string.h>
  46 #include <unistd.h>
  47
  48 #include "extern.h"
  49
  50 static void rulespec_infp(FILE *fp, unsigned long request, devfs_rsnum rsnum);
  51 static void rulespec_instr(struct devfs_rule *dr, const char *str,
  52     devfs_rsnum rsnum);
  53 static void rulespec_intok(struct devfs_rule *dr, int ac, char **av,
  54     devfs_rsnum rsnum);
  55 static void rulespec_outfp(FILE *fp, struct devfs_rule *dr);
  56
  57 static command_t rule_add, rule_apply, rule_applyset;
  58 static command_t rule_del, rule_delset, rule_show, rule_showsets;
  59
  60 static ctbl_t ctbl_rule = {
  61         { "add",                rule_add },
  62         { "apply",              rule_apply },
  63         { "applyset",           rule_applyset },
  64         { "del",                rule_del },
  65         { "delset",             rule_delset },
  66         { "show",               rule_show },
  67         { "showsets",           rule_showsets },
  68         { NULL,                 NULL }
  69 };
  70
  71 static struct intstr ist_type[] = {
  72         { "disk",               D_DISK },
  73         { "mem",                D_MEM },
  74         { "tape",               D_TAPE },
  75         { "tty",                D_TTY },
  76         { NULL,                 -1 }
  77 };
  78
  79 static devfs_rsnum in_rsnum;
  80
  81 int
  82 rule_main(int ac, char **av)
  83 {
  84         struct cmd *c;
  85         int ch;
  86
  87         setprogname("devfs rule");
  88         optreset = optind = 1;
  89         while ((ch = getopt(ac, av, "s:")) != -1)
  90                 switch (ch) {
  91                 case 's':
  92                         in_rsnum = eatonum(optarg);
  93                         break;
  94                 default:
  95                         usage();
  96                 }
  97         ac -= optind;
  98         av += optind;
  99         if (ac < 1)
 100                 usage();
 101
 102         for (c = ctbl_rule; c->name != NULL; ++c)
 103                 if (strcmp(c->name, av[0]) == 0)
 104                         exit((*c->handler)(ac, av));
 105         errx(1, "unknown command: %s", av[0]);
 106 }
 107
 108 static int
 109 rule_add(int ac, char **av)
 110 {
 111         struct devfs_rule dr;
 112         int rv;
 113
 114         if (ac < 2)
 115                 usage();
 116         if (strcmp(av[1], "-") == 0)
 117                 rulespec_infp(stdin, DEVFSIO_RADD, in_rsnum);
 118         else {
 119                 rulespec_intok(&dr, ac - 1, av + 1, in_rsnum);
 120                 rv = ioctl(mpfd, DEVFSIO_RADD, &dr);
 121                 if (rv == -1)
 122                         err(1, "ioctl DEVFSIO_RADD");
 123         }
 124         return (0);
 125 }
 126
 127 static int
 128 rule_apply(int ac __unused, char **av __unused)
 129 {
 130         struct devfs_rule dr;
 131         devfs_rnum rnum;
 132         devfs_rid rid;
 133         int rv;
 134
 135         if (ac < 2)
 136                 usage();
 137         if (!atonum(av[1], &rnum)) {
 138                 if (strcmp(av[1], "-") == 0)
 139                         rulespec_infp(stdin, DEVFSIO_RAPPLY, in_rsnum);
 140                 else {
 141                         rulespec_intok(&dr, ac - 1, av + 1, in_rsnum);
 142                         rv = ioctl(mpfd, DEVFSIO_RAPPLY, &dr);
 143                         if (rv == -1)
 144                                 err(1, "ioctl DEVFSIO_RAPPLY");
 145                 }
 146         } else {
 147                 rid = mkrid(in_rsnum, rnum);
 148                 rv = ioctl(mpfd, DEVFSIO_RAPPLYID, &rid);
 149                 if (rv == -1)
 150                         err(1, "ioctl DEVFSIO_RAPPLYID");
 151         }
 152         return (0);
 153 }
 154
 155 static int
 156 rule_applyset(int ac, char **av __unused)
 157 {
 158         int rv;
 159
 160         if (ac != 1)
 161                 usage();
 162         rv = ioctl(mpfd, DEVFSIO_SAPPLY, &in_rsnum);
 163         if (rv == -1)
 164                 err(1, "ioctl DEVFSIO_SAPPLY");
 165         return (0);
 166 }
 167
 168 static int
 169 rule_del(int ac __unused, char **av)
 170 {
 171         devfs_rid rid;
 172         int rv;
 173
 174         if (av[1] == NULL)
 175                 usage();
 176         rid = mkrid(in_rsnum, eatoi(av[1]));
 177         rv = ioctl(mpfd, DEVFSIO_RDEL, &rid);
 178         if (rv == -1)
 179                 err(1, "ioctl DEVFSIO_RDEL");
 180         return (0);
 181 }
 182
 183 static int
 184 rule_delset(int ac, char **av __unused)
 185 {
 186         struct devfs_rule dr;
 187         int rv;
 188
 189         if (ac != 1)
 190                 usage();
 191         memset(&dr, '\0', sizeof(dr));
 192         dr.dr_magic = DEVFS_MAGIC;
 193         dr.dr_id = mkrid(in_rsnum, 0);
 194         while (ioctl(mpfd, DEVFSIO_RGETNEXT, &dr) != -1) {
 195                 rv = ioctl(mpfd, DEVFSIO_RDEL, &dr.dr_id);
 196                 if (rv == -1)
 197                         err(1, "ioctl DEVFSIO_RDEL");
 198         }
 199         if (errno != ENOENT)
 200                 err(1, "ioctl DEVFSIO_RGETNEXT");
 201         return (0);
 202 }
 203
 204 static int
 205 rule_show(int ac __unused, char **av)
 206 {
 207         struct devfs_rule dr;
 208         devfs_rnum rnum;
 209         int rv;
 210
 211         memset(&dr, '\0', sizeof(dr));
 212         dr.dr_magic = DEVFS_MAGIC;
 213         if (av[1] != NULL) {
 214                 rnum = eatoi(av[1]);
 215                 dr.dr_id = mkrid(in_rsnum, rnum - 1);
 216                 rv = ioctl(mpfd, DEVFSIO_RGETNEXT, &dr);
 217                 if (rv == -1)
 218                         err(1, "ioctl DEVFSIO_RGETNEXT");
 219                 if (rid2rn(dr.dr_id) == rnum)
 220                         rulespec_outfp(stdout, &dr);
 221         } else {
 222                 dr.dr_id = mkrid(in_rsnum, 0);
 223                 while (ioctl(mpfd, DEVFSIO_RGETNEXT, &dr) != -1)
 224                         rulespec_outfp(stdout, &dr);
 225                 if (errno != ENOENT)
 226                         err(1, "ioctl DEVFSIO_RGETNEXT");
 227         }
 228         return (0);
 229 }
 230
 231 static int
 232 rule_showsets(int ac, char **av __unused)
 233 {
 234         devfs_rsnum rsnum;
 235
 236         if (ac != 1)
 237                 usage();
 238         rsnum = 0;
 239         while (ioctl(mpfd, DEVFSIO_SGETNEXT, &rsnum) != -1)
 240                 printf("%d\n", rsnum);
 241         if (errno != ENOENT)
 242                 err(1, "ioctl DEVFSIO_SGETNEXT");
 243         return (0);
 244 }
 245
 246 int
 247 ruleset_main(int ac, char **av)
 248 {
 249         devfs_rsnum rsnum;
 250         int rv;
 251
 252         setprogname("devfs ruleset");
 253         if (ac < 2)
 254                 usage();
 255         rsnum = eatonum(av[1]);
 256         rv = ioctl(mpfd, DEVFSIO_SUSE, &rsnum);
 257         if (rv == -1)
 258                 err(1, "ioctl DEVFSIO_SUSE");
 259         return (0);
 260 }
 261
 262
 263 /*
 264  * Input rules from a file (probably the standard input).  This
 265  * differs from the other rulespec_in*() routines in that it also
 266  * calls ioctl() for the rules, since it is impractical (and not very
 267  * useful) to return a list (or array) of rules, just so the caller
 268  * can call call ioctl() for each of them.
 269  */
 270 static void
 271 rulespec_infp(FILE *fp, unsigned long request, devfs_rsnum rsnum)
 272 {
 273         struct devfs_rule dr;
 274         char *line;
 275         int rv;
 276
 277         assert(fp == stdin);    /* XXX: De-hardcode "stdin" from error msg. */
 278         while (efgetln(fp, &line)) {
 279                 rulespec_instr(&dr, line, rsnum);
 280                 rv = ioctl(mpfd, request, &dr);
 281                 if (rv == -1)
 282                         err(1, "ioctl");
 283                 free(line);     /* efgetln() always malloc()s. */
 284         }
 285         if (ferror(stdin))
 286                 err(1, "stdin");
 287 }
 288
 289 /*
 290  * Construct a /struct devfs_rule/ from a string.
 291  */
 292 static void
 293 rulespec_instr(struct devfs_rule *dr, const char *str, devfs_rsnum rsnum)
 294 {
 295         char **av;
 296         int ac;
 297
 298         tokenize(str, &ac, &av);
 299         if (ac == 0)
 300                 errx(1, "unexpected end of rulespec");
 301         rulespec_intok(dr, ac, av, rsnum);
 302         free(av[0]);
 303         free(av);
 304 }
 305
 306 /*
 307  * Construct a /struct devfs_rule/ from ac and av.
 308  */
 309 static void
 310 rulespec_intok(struct devfs_rule *dr, int ac __unused, char **av,
 311     devfs_rsnum rsnum)
 312 {
 313         struct intstr *is;
 314         struct passwd *pw;
 315         struct group *gr;
 316         devfs_rnum rnum;
 317         void *set;
 318
 319         memset(dr, '\0', sizeof(*dr));
 320
 321         /*
 322          * We don't maintain ac hereinafter.
 323          */
 324         if (av[0] == NULL)
 325                 errx(1, "unexpected end of rulespec");
 326
 327         /* If the first argument is an integer, treat it as a rule number. */
 328         if (!atonum(av[0], &rnum))
 329                 rnum = 0;               /* auto-number */
 330         else
 331                 ++av;
 332
 333         /*
 334          * These aren't table-driven since that would result in more
 335          * tiny functions than I care to deal with.
 336          */
 337         for (;;) {
 338                 if (av[0] == NULL)
 339                         break;
 340                 else if (strcmp(av[0], "type") == 0) {
 341                         if (av[1] == NULL)
 342                                 errx(1, "expecting argument for type");
 343                         for (is = ist_type; is->s != NULL; ++is)
 344                                 if (strcmp(av[1], is->s) == 0) {
 345                                         dr->dr_dswflags |= is->i;
 346                                         break;
 347                                 }
 348                         if (is->s == NULL)
 349                                 errx(1, "unknown type: %s", av[1]);
 350                         dr->dr_icond |= DRC_DSWFLAGS;
 351                         av += 2;
 352                 } else if (strcmp(av[0], "path") == 0) {
 353                         if (av[1] == NULL)
 354                                 errx(1, "expecting argument for path");
 355                         if (strlcpy(dr->dr_pathptrn, av[1], DEVFS_MAXPTRNLEN)
 356                             >= DEVFS_MAXPTRNLEN)
 357                                 warnx("pattern specified too long; truncated");
 358                         dr->dr_icond |= DRC_PATHPTRN;
 359                         av += 2;
 360                 } else
 361                         break;
 362         }
 363         while (av[0] != NULL) {
 364                 if (strcmp(av[0], "hide") == 0) {
 365                         dr->dr_iacts |= DRA_BACTS;
 366                         dr->dr_bacts |= DRB_HIDE;
 367                         ++av;
 368                 } else if (strcmp(av[0], "unhide") == 0) {
 369                         dr->dr_iacts |= DRA_BACTS;
 370                         dr->dr_bacts |= DRB_UNHIDE;
 371                         ++av;
 372                 } else if (strcmp(av[0], "user") == 0) {
 373                         if (av[1] == NULL)
 374                                 errx(1, "expecting argument for user");
 375                         dr->dr_iacts |= DRA_UID;
 376                         pw = getpwnam(av[1]);
 377                         if (pw != NULL)
 378                                 dr->dr_uid = pw->pw_uid;
 379                         else
 380                                 dr->dr_uid = eatoi(av[1]); /* XXX overflow */
 381                         av += 2;
 382                 } else if (strcmp(av[0], "group") == 0) {
 383                         if (av[1] == NULL)
 384                                 errx(1, "expecting argument for group");
 385                         dr->dr_iacts |= DRA_GID;
 386                         gr = getgrnam(av[1]);
 387                         if (gr != NULL)
 388                                 dr->dr_gid = gr->gr_gid;
 389                         else
 390                                 dr->dr_gid = eatoi(av[1]); /* XXX overflow */
 391                         av += 2;
 392                 } else if (strcmp(av[0], "mode") == 0) {
 393                         if (av[1] == NULL)
 394                                 errx(1, "expecting argument for mode");
 395                         dr->dr_iacts |= DRA_MODE;
 396                         set = setmode(av[1]);
 397                         if (set == NULL)
 398                                 errx(1, "invalid mode: %s", av[1]);
 399                         dr->dr_mode = getmode(set, 0);
 400                         av += 2;
 401                 } else if (strcmp(av[0], "include") == 0) {
 402                         if (av[1] == NULL)
 403                                 errx(1, "expecting argument for include");
 404                         dr->dr_iacts |= DRA_INCSET;
 405                         dr->dr_incset = eatonum(av[1]);
 406                         av += 2;
 407                 } else
 408                         errx(1, "unknown argument: %s", av[0]);
 409         }
 410
 411         dr->dr_id = mkrid(rsnum, rnum);
 412         dr->dr_magic = DEVFS_MAGIC;
 413 }
 414
 415 /*
 416  * Write a human-readable (and machine-parsable, by rulespec_in*())
 417  * representation of dr to bufp.  *bufp should be free(3)'d when the
 418  * caller is finished with it.
 419  */
 420 static void
 421 rulespec_outfp(FILE *fp, struct devfs_rule *dr)
 422 {
 423         struct intstr *is;
 424         struct passwd *pw;
 425         struct group *gr;
 426
 427         fprintf(fp, "%d", rid2rn(dr->dr_id));
 428
 429         if (dr->dr_icond & DRC_DSWFLAGS)
 430                 for (is = ist_type; is->s != NULL; ++is)
 431                         if (dr->dr_dswflags & is->i)
 432                                 fprintf(fp, " type %s", is->s);
 433         if (dr->dr_icond & DRC_PATHPTRN)
 434                 fprintf(fp, " path %s", dr->dr_pathptrn);
 435
 436         if (dr->dr_iacts & DRA_BACTS) {
 437                 if (dr->dr_bacts & DRB_HIDE)
 438                         fprintf(fp, " hide");
 439                 if (dr->dr_bacts & DRB_UNHIDE)
 440                         fprintf(fp, " unhide");
 441         }
 442         if (dr->dr_iacts & DRA_UID) {
 443                 pw = getpwuid(dr->dr_uid);
 444                 if (pw == NULL)
 445                         fprintf(fp, " user %d", dr->dr_uid);
 446                 else
 447                         fprintf(fp, " user %s", pw->pw_name);
 448         }
 449         if (dr->dr_iacts & DRA_GID) {
 450                 gr = getgrgid(dr->dr_gid);
 451                 if (gr == NULL)
 452                         fprintf(fp, " group %d", dr->dr_gid);
 453                 else
 454                         fprintf(fp, " group %s", gr->gr_name);
 455         }
 456         if (dr->dr_iacts & DRA_MODE)
 457                 fprintf(fp, " mode %o", dr->dr_mode);
 458         if (dr->dr_iacts & DRA_INCSET)
 459                 fprintf(fp, " include %d", dr->dr_incset);
 460
 461         fprintf(fp, "\n");
 462 }