1 /*
   2  * CDDL HEADER START
   3  *
   4  * The contents of this file are subject to the terms of the
   5  * Common Development and Distribution License (the "License").
   6  * You may not use this file except in compliance with the License.
   7  *
   8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
   9  * or http://www.opensolaris.org/os/licensing.
  10  * See the License for the specific language governing permissions
  11  * and limitations under the License.
  12  *
  13  * When distributing Covered Code, include this CDDL HEADER in each
  14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
  15  * If applicable, add the following below this CDDL HEADER, with the
  16  * fields enclosed by brackets "[]" replaced with your own identifying
  17  * information: Portions Copyright [yyyy] [name of copyright owner]
  18  *
  19  * CDDL HEADER END
  20  */
  21 
  22 /*
  23  * Copyright (c) 1995, 2010, Oracle and/or its affiliates. All rights reserved.
  24  */
  25 
  26 #include "benv.h"
  27 #include "message.h"
  28 #include <ctype.h>
  29 #include <stdarg.h>
  30 #include <sys/mman.h>
  31 #include <unistd.h>
  32 #include <signal.h>
  33 #include <sys/wait.h>
  34 
  35 /*
  36  * Usage:  % eeprom [-v] [-f prom_dev] [-]
  37  *         % eeprom [-v] [-f prom_dev] field[=value] ...
  38  */
  39 
  40 extern void get_kbenv(void);
  41 extern void close_kbenv(void);
  42 extern caddr_t get_propval(char *name, char *node);
  43 extern void setpname(char *prog);
  44 extern char *getbootcmd(void);
  45 
  46 char *boottree;
  47 struct utsname uts_buf;
  48 
  49 static int test;
  50 int verbose;
  51 
  52 /*
  53  * Concatenate a NULL terminated list of strings into
  54  * a single string.
  55  */
  56 char *
  57 strcats(char *s, ...)
  58 {
  59         char *cp, *ret;
  60         size_t len;
  61         va_list ap;
  62 
  63         va_start(ap, s);
  64         for (ret = NULL, cp = s; cp; cp = va_arg(ap, char *)) {
  65                 if (ret == NULL) {
  66                         ret = strdup(s);
  67                         len = strlen(ret) + 1;
  68                 } else {
  69                         len += strlen(cp);
  70                         ret = realloc(ret, len);
  71                         (void) strcat(ret, cp);
  72                 }
  73         }
  74         va_end(ap);
  75 
  76         return (ret);
  77 }
  78 
  79 eplist_t *
  80 new_list(void)
  81 {
  82         eplist_t *list;
  83 
  84         list = (eplist_t *)malloc(sizeof (eplist_t));
  85         (void) memset(list, 0, sizeof (eplist_t));
  86 
  87         list->next = list;
  88         list->prev = list;
  89         list->item = NULL;
  90 
  91         return (list);
  92 }
  93 
  94 void
  95 add_item(void *item, eplist_t *list)
  96 {
  97         eplist_t *entry;
  98 
  99         entry = (eplist_t *)malloc(sizeof (eplist_t));
 100         (void) memset(entry, 0, sizeof (eplist_t));
 101         entry->item = item;
 102 
 103         entry->next = list;
 104         entry->prev = list->prev;
 105         list->prev->next = entry;
 106         list->prev = entry;
 107 }
 108 
 109 typedef struct benv_ent {
 110         char *cmd;
 111         char *name;
 112         char *val;
 113 } benv_ent_t;
 114 
 115 typedef struct benv_des {
 116         char *name;
 117         int fd;
 118         caddr_t adr;
 119         size_t len;
 120         eplist_t *elist;
 121 } benv_des_t;
 122 
 123 static benv_des_t *
 124 new_bd(void)
 125 {
 126 
 127         benv_des_t *bd;
 128 
 129         bd = (benv_des_t *)malloc(sizeof (benv_des_t));
 130         (void) memset(bd, 0, sizeof (benv_des_t));
 131 
 132         bd->elist = new_list();
 133 
 134         return (bd);
 135 }
 136 
 137 /*
 138  * Create a new entry.  Comment entries have NULL names.
 139  */
 140 static benv_ent_t *
 141 new_bent(char *comm, char *cmd, char *name, char *val)
 142 {
 143         benv_ent_t *bent;
 144 
 145         bent = (benv_ent_t *)malloc(sizeof (benv_ent_t));
 146         (void) memset(bent, 0, sizeof (benv_ent_t));
 147 
 148         if (comm) {
 149                 bent->cmd = strdup(comm);
 150                 comm = NULL;
 151         } else {
 152                 bent->cmd = strdup(cmd);
 153                 bent->name = strdup(name);
 154                 if (val)
 155                         bent->val = strdup(val);
 156         }
 157 
 158         return (bent);
 159 }
 160 
 161 /*
 162  * Add a new entry to the benv entry list.  Entries can be
 163  * comments or commands.
 164  */
 165 static void
 166 add_bent(eplist_t *list, char *comm, char *cmd, char *name, char *val)
 167 {
 168         benv_ent_t *bent;
 169 
 170         bent = new_bent(comm, cmd, name, val);
 171         add_item((void *)bent, list);
 172 }
 173 
 174 static benv_ent_t *
 175 get_var(char *name, eplist_t *list)
 176 {
 177         eplist_t *e;
 178         benv_ent_t *p;
 179 
 180         for (e = list->next; e != list; e = e->next) {
 181                 p = (benv_ent_t *)e->item;
 182                 if (p->name != NULL && strcmp(p->name, name) == 0)
 183                         return (p);
 184         }
 185 
 186         return (NULL);
 187 }
 188 
 189 /*PRINTFLIKE1*/
 190 static void
 191 eeprom_error(const char *format, ...)
 192 {
 193         va_list ap;
 194 
 195         va_start(ap, format);
 196         (void) fprintf(stderr, "eeprom: ");
 197         (void) vfprintf(stderr, format, ap);
 198         va_end(ap);
 199 }
 200 
 201 static int
 202 exec_cmd(char *cmdline, char *output, int64_t osize)
 203 {
 204         char buf[BUFSIZ];
 205         int ret;
 206         size_t len;
 207         FILE *ptr;
 208         sigset_t set;
 209         void (*disp)(int);
 210 
 211         if (output)
 212                 output[0] = '\0';
 213 
 214         /*
 215          * For security
 216          * - only absolute paths are allowed
 217          * - set IFS to space and tab
 218          */
 219         if (*cmdline != '/') {
 220                 eeprom_error(ABS_PATH_REQ, cmdline);
 221                 return (-1);
 222         }
 223         (void) putenv("IFS= \t");
 224 
 225         /*
 226          * We may have been exec'ed with SIGCHLD blocked
 227          * unblock it here
 228          */
 229         (void) sigemptyset(&set);
 230         (void) sigaddset(&set, SIGCHLD);
 231         if (sigprocmask(SIG_UNBLOCK, &set, NULL) != 0) {
 232                 eeprom_error(FAILED_SIG, strerror(errno));
 233                 return (-1);
 234         }
 235 
 236         /*
 237          * Set SIGCHLD disposition to SIG_DFL for popen/pclose
 238          */
 239         disp = sigset(SIGCHLD, SIG_DFL);
 240         if (disp == SIG_ERR) {
 241                 eeprom_error(FAILED_SIG, strerror(errno));
 242                 return (-1);
 243         }
 244         if (disp == SIG_HOLD) {
 245                 eeprom_error(BLOCKED_SIG, cmdline);
 246                 return (-1);
 247         }
 248 
 249         ptr = popen(cmdline, "r");
 250         if (ptr == NULL) {
 251                 eeprom_error(POPEN_FAIL, cmdline, strerror(errno));
 252                 return (-1);
 253         }
 254 
 255         /*
 256          * If we simply do a pclose() following a popen(), pclose()
 257          * will close the reader end of the pipe immediately even
 258          * if the child process has not started/exited. pclose()
 259          * does wait for cmd to terminate before returning though.
 260          * When the executed command writes its output to the pipe
 261          * there is no reader process and the command dies with
 262          * SIGPIPE. To avoid this we read repeatedly until read
 263          * terminates with EOF. This indicates that the command
 264          * (writer) has closed the pipe and we can safely do a
 265          * pclose().
 266          *
 267          * Since pclose() does wait for the command to exit,
 268          * we can safely reap the exit status of the command
 269          * from the value returned by pclose()
 270          */
 271         while (fgets(buf, sizeof (buf), ptr) != NULL) {
 272                 if (output && osize > 0) {
 273                         (void) snprintf(output, osize, "%s", buf);
 274                         len = strlen(buf);
 275                         output += len;
 276                         osize -= len;
 277                 }
 278         }
 279 
 280         /*
 281          * If there's a "\n" at the end, we want to chop it off
 282          */
 283         if (output) {
 284                 len = strlen(output) - 1;
 285                 if (output[len] == '\n')
 286                         output[len] = '\0';
 287         }
 288 
 289         ret = pclose(ptr);
 290         if (ret == -1) {
 291                 eeprom_error(PCLOSE_FAIL, cmdline, strerror(errno));
 292                 return (-1);
 293         }
 294 
 295         if (WIFEXITED(ret)) {
 296                 return (WEXITSTATUS(ret));
 297         } else {
 298                 eeprom_error(EXEC_FAIL, cmdline, ret);
 299                 return (-1);
 300         }
 301 }
 302 
 303 #define BOOTADM_STR     "bootadm: "
 304 
 305 /*
 306  * bootadm starts all error messages with "bootadm: ".
 307  * Add a note so users don't get confused on how they ran bootadm.
 308  */
 309 static void
 310 output_error_msg(const char *msg)
 311 {
 312         size_t len = sizeof (BOOTADM_STR) - 1;
 313 
 314         if (strncmp(msg, BOOTADM_STR, len) == 0) {
 315                 eeprom_error("error returned from %s\n", msg);
 316         } else if (msg[0] != '\0') {
 317                 eeprom_error("%s\n", msg);
 318         }
 319 }
 320 
 321 static char *
 322 get_bootadm_value(char *name, const int quiet)
 323 {
 324         char *ptr, *ret_str, *end_ptr, *orig_ptr;
 325         char output[BUFSIZ];
 326         int is_console, is_kernel = 0;
 327         size_t len;
 328 
 329         is_console = (strcmp(name, "console") == 0);
 330 
 331         if (strcmp(name, "boot-file") == 0) {
 332                 is_kernel = 1;
 333                 ptr = "/sbin/bootadm set-menu kernel 2>&1";
 334         } else if (is_console || (strcmp(name, "boot-args") == 0)) {
 335                 ptr = "/sbin/bootadm set-menu args 2>&1";
 336         } else {
 337                 eeprom_error("Unknown value in get_bootadm_value: %s\n", name);
 338                 return (NULL);
 339         }
 340 
 341         if (exec_cmd(ptr, output, BUFSIZ) != 0) {
 342                 if (quiet == 0) {
 343                         output_error_msg(output);
 344                 }
 345                 return (NULL);
 346         }
 347 
 348         if (is_console) {
 349                 if ((ptr = strstr(output, "console=")) == NULL) {
 350                         return (NULL);
 351                 }
 352                 ptr += strlen("console=");
 353 
 354                 /*
 355                  * -B may have comma-separated values.  It may also be
 356                  * followed by other flags.
 357                  */
 358                 len = strcspn(ptr, " \t,");
 359                 ret_str = calloc(len + 1, 1);
 360                 if (ret_str == NULL) {
 361                         eeprom_error(NO_MEM, len + 1);
 362                         return (NULL);
 363                 }
 364                 (void) strncpy(ret_str, ptr, len);
 365                 return (ret_str);
 366         } else if (is_kernel) {
 367                 ret_str = strdup(output);
 368                 if (ret_str == NULL)
 369                         eeprom_error(NO_MEM, strlen(output) + 1);
 370                 return (ret_str);
 371         } else {
 372                 /* If there's no console setting, we can return */
 373                 if ((orig_ptr = strstr(output, "console=")) == NULL) {
 374                         return (strdup(output));
 375                 }
 376                 len = strcspn(orig_ptr, " \t,");
 377                 ptr = orig_ptr;
 378                 end_ptr = orig_ptr + len + 1;
 379 
 380                 /* Eat up any white space */
 381                 while ((*end_ptr == ' ') || (*end_ptr == '\t'))
 382                         end_ptr++;
 383 
 384                 /*
 385                  * If there's data following the console string, copy it.
 386                  * If not, cut off the new string.
 387                  */
 388                 if (*end_ptr == '\0')
 389                         *ptr = '\0';
 390 
 391                 while (*end_ptr != '\0') {
 392                         *ptr = *end_ptr;
 393                         ptr++;
 394                         end_ptr++;
 395                 }
 396                 *ptr = '\0';
 397                 if ((strchr(output, '=') == NULL) &&
 398                     (strncmp(output, "-B ", 3) == 0)) {
 399                         /*
 400                          * Since we removed the console setting, we no
 401                          * longer need the initial "-B "
 402                          */
 403                         orig_ptr = output + 3;
 404                 } else {
 405                         orig_ptr = output;
 406                 }
 407 
 408                 ret_str = strdup(orig_ptr);
 409                 if (ret_str == NULL)
 410                         eeprom_error(NO_MEM, strlen(orig_ptr) + 1);
 411                 return (ret_str);
 412         }
 413 }
 414 
 415 /*
 416  * If quiet is 1, print nothing if there is no value.  If quiet is 0, print
 417  * a message.  Return 1 if the value is printed, 0 otherwise.
 418  */
 419 static int
 420 print_bootadm_value(char *name, const int quiet)
 421 {
 422         int rv = 0;
 423         char *value = get_bootadm_value(name, quiet);
 424 
 425         if ((value != NULL) && (value[0] != '\0')) {
 426                 (void) printf("%s=%s\n", name, value);
 427                 rv = 1;
 428         } else if (quiet == 0) {
 429                 (void) printf("%s: data not available.\n", name);
 430         }
 431 
 432         if (value != NULL)
 433                 free(value);
 434         return (rv);
 435 }
 436 
 437 static void
 438 print_var(char *name, eplist_t *list)
 439 {
 440         benv_ent_t *p;
 441         char *bootcmd;
 442 
 443         /*
 444          * The console property is kept in both menu.lst and bootenv.rc.  The
 445          * menu.lst value takes precedence.
 446          */
 447         if (strcmp(name, "console") == 0) {
 448                 if (print_bootadm_value(name, 1) == 0) {
 449                         if ((p = get_var(name, list)) != NULL) {
 450                                 (void) printf("%s=%s\n", name, p->val ?
 451                                     p->val : "");
 452                         } else {
 453                                 (void) printf("%s: data not available.\n",
 454                                     name);
 455                         }
 456                 }
 457         } else if (strcmp(name, "bootcmd") == 0) {
 458                 bootcmd = getbootcmd();
 459                 (void) printf("%s=%s\n", name, bootcmd ? bootcmd : "");
 460         } else if ((strcmp(name, "boot-file") == 0) ||
 461             (strcmp(name, "boot-args") == 0)) {
 462                 (void) print_bootadm_value(name, 0);
 463         } else if ((p = get_var(name, list)) == NULL) {
 464                 (void) printf("%s: data not available.\n", name);
 465         } else {
 466                 (void) printf("%s=%s\n", name, p->val ? p->val : "");
 467         }
 468 }
 469 
 470 static void
 471 print_vars(eplist_t *list)
 472 {
 473         eplist_t *e;
 474         benv_ent_t *p;
 475         int console_printed = 0;
 476 
 477         /*
 478          * The console property is kept both in menu.lst and bootenv.rc.
 479          * The menu.lst value takes precedence, so try printing that one
 480          * first.
 481          */
 482         console_printed = print_bootadm_value("console", 1);
 483 
 484         for (e = list->next; e != list; e = e->next) {
 485                 p = (benv_ent_t *)e->item;
 486                 if (p->name != NULL) {
 487                         if (((strcmp(p->name, "console") == 0) &&
 488                             (console_printed == 1)) ||
 489                             ((strcmp(p->name, "boot-file") == 0) ||
 490                             (strcmp(p->name, "boot-args") == 0))) {
 491                                 /* handle these separately */
 492                                 continue;
 493                         }
 494                         (void) printf("%s=%s\n", p->name, p->val ? p->val : "");
 495                 }
 496         }
 497         (void) print_bootadm_value("boot-file", 1);
 498         (void) print_bootadm_value("boot-args", 1);
 499 }
 500 
 501 /*
 502  * Write a string to a file, quoted appropriately.  We use single
 503  * quotes to prevent any variable expansion.  Of course, we backslash-quote
 504  * any single quotes or backslashes.
 505  */
 506 static void
 507 put_quoted(FILE *fp, char *val)
 508 {
 509         (void) putc('\'', fp);
 510         while (*val) {
 511                 switch (*val) {
 512                 case '\'':
 513                 case '\\':
 514                         (void) putc('\\', fp);
 515                         /* FALLTHROUGH */
 516                 default:
 517                         (void) putc(*val, fp);
 518                         break;
 519                 }
 520                 val++;
 521         }
 522         (void) putc('\'', fp);
 523 }
 524 
 525 static void
 526 set_bootadm_var(char *name, char *value)
 527 {
 528         char buf[BUFSIZ];
 529         char output[BUFSIZ] = "";
 530         char *console, *args;
 531         int is_console;
 532 
 533         if (verbose) {
 534                 (void) printf("old:");
 535                 (void) print_bootadm_value(name, 0);
 536         }
 537 
 538         /*
 539          * For security, we single-quote whatever we run on the command line,
 540          * and we don't allow single quotes in the string.
 541          */
 542         if (strchr(value, '\'') != NULL) {
 543                 eeprom_error("Single quotes are not allowed "
 544                     "in the %s property.\n", name);
 545                 return;
 546         }
 547 
 548         is_console = (strcmp(name, "console") == 0);
 549         if (strcmp(name, "boot-file") == 0) {
 550                 (void) snprintf(buf, BUFSIZ, "/sbin/bootadm set-menu "
 551                     "kernel='%s' 2>&1", value);
 552         } else if (is_console || (strcmp(name, "boot-args") == 0)) {
 553                 if (is_console) {
 554                         args = get_bootadm_value("boot-args", 1);
 555                         console = value;
 556                 } else {
 557                         args = value;
 558                         console = get_bootadm_value("console", 1);
 559                 }
 560                 if (((args == NULL) || (args[0] == '\0')) &&
 561                     ((console == NULL) || (console[0] == '\0'))) {
 562                         (void) snprintf(buf, BUFSIZ, "/sbin/bootadm set-menu "
 563                             "args= 2>&1");
 564                 } else if ((args == NULL) || (args[0] == '\0')) {
 565                         (void) snprintf(buf, BUFSIZ, "/sbin/bootadm "
 566                             "set-menu args='-B console=%s' 2>&1",
 567                             console);
 568                 } else if ((console == NULL) || (console[0] == '\0')) {
 569                         (void) snprintf(buf, BUFSIZ, "/sbin/bootadm "
 570                             "set-menu args='%s' 2>&1", args);
 571                 } else if (strncmp(args, "-B ", 3) != 0) {
 572                         (void) snprintf(buf, BUFSIZ, "/sbin/bootadm "
 573                             "set-menu args='-B console=%s %s' 2>&1",
 574                             console, args);
 575                 } else {
 576                         (void) snprintf(buf, BUFSIZ, "/sbin/bootadm "
 577                             "set-menu args='-B console=%s,%s' 2>&1",
 578                             console, args + 3);
 579                 }
 580         } else {
 581                 eeprom_error("Unknown value in set_bootadm_value: %s\n", name);
 582                 return;
 583         }
 584 
 585         if (exec_cmd(buf, output, BUFSIZ) != 0) {
 586                 output_error_msg(output);
 587                 return;
 588         }
 589 
 590         if (verbose) {
 591                 (void) printf("new:");
 592                 (void) print_bootadm_value(name, 0);
 593         }
 594 }
 595 
 596 /*
 597  * Returns 1 if bootenv.rc was modified, 0 otherwise.
 598  */
 599 static int
 600 set_var(char *name, char *val, eplist_t *list)
 601 {
 602         benv_ent_t *p;
 603         int old_verbose;
 604 
 605         if (strcmp(name, "bootcmd") == 0)
 606                 return (0);
 607 
 608         if ((strcmp(name, "boot-file") == 0) ||
 609             (strcmp(name, "boot-args") == 0)) {
 610                 set_bootadm_var(name, val);
 611                 return (0);
 612         }
 613 
 614         /*
 615          * The console property is kept in two places: menu.lst and bootenv.rc.
 616          * Update them both.  We clear verbose to prevent duplicate messages.
 617          */
 618         if (strcmp(name, "console") == 0) {
 619                 old_verbose = verbose;
 620                 verbose = 0;
 621                 set_bootadm_var(name, val);
 622                 verbose = old_verbose;
 623         }
 624 
 625         if (verbose) {
 626                 (void) printf("old:");
 627                 print_var(name, list);
 628         }
 629 
 630         if ((p = get_var(name, list)) != NULL) {
 631                 free(p->val);
 632                 p->val = strdup(val);
 633         } else
 634                 add_bent(list, NULL, "setprop", name, val);
 635 
 636         if (verbose) {
 637                 (void) printf("new:");
 638                 print_var(name, list);
 639         }
 640         return (1);
 641 }
 642 
 643 /*
 644  * Returns 1 if bootenv.rc is modified or 0 if no modification was
 645  * necessary.  This allows us to implement non super-user look-up of
 646  * variables by name without the user being yelled at for trying to
 647  * modify the bootenv.rc file.
 648  */
 649 static int
 650 proc_var(char *name, eplist_t *list)
 651 {
 652         register char *val;
 653 
 654         if ((val = strchr(name, '=')) == NULL) {
 655                 print_var(name, list);
 656                 return (0);
 657         } else {
 658                 *val++ = '\0';
 659                 return (set_var(name, val, list));
 660         }
 661 }
 662 
 663 static void
 664 init_benv(benv_des_t *bd, char *file)
 665 {
 666         get_kbenv();
 667 
 668         if (test)
 669                 boottree = "/tmp";
 670         else if ((boottree = (char *)get_propval("boottree", "chosen")) == NULL)
 671                 boottree = strcats("/boot", NULL);
 672 
 673         if (file != NULL)
 674                 bd->name = file;
 675         else
 676                 bd->name = strcats(boottree, "/solaris/bootenv.rc", NULL);
 677 }
 678 
 679 static void
 680 map_benv(benv_des_t *bd)
 681 {
 682         if ((bd->fd = open(bd->name, O_RDONLY)) == -1)
 683                 if (errno == ENOENT)
 684                         return;
 685                 else
 686                         exit(_error(PERROR, "cannot open %s", bd->name));
 687 
 688         if ((bd->len = (size_t)lseek(bd->fd, 0, SEEK_END)) == 0) {
 689                 if (close(bd->fd) == -1)
 690                         exit(_error(PERROR, "close error on %s", bd->name));
 691                 return;
 692         }
 693 
 694         (void) lseek(bd->fd, 0, SEEK_SET);
 695 
 696         if ((bd->adr = mmap((caddr_t)0, bd->len, (PROT_READ | PROT_WRITE),
 697             MAP_PRIVATE, bd->fd, 0)) == MAP_FAILED)
 698                 exit(_error(PERROR, "cannot map %s", bd->name));
 699 }
 700 
 701 static void
 702 unmap_benv(benv_des_t *bd)
 703 {
 704         if (munmap(bd->adr, bd->len) == -1)
 705                 exit(_error(PERROR, "unmap error on %s", bd->name));
 706 
 707         if (close(bd->fd) == -1)
 708                 exit(_error(PERROR, "close error on %s", bd->name));
 709 }
 710 
 711 #define NL      '\n'
 712 #define COMM    '#'
 713 
 714 /*
 715  * Add a comment block to the benv list.
 716  */
 717 static void
 718 add_comm(benv_des_t *bd, char *base, char *last, char **next, int *line)
 719 {
 720         int nl, lines;
 721         char *p;
 722 
 723         nl = 0;
 724         for (p = base, lines = 0; p < last; p++) {
 725                 if (*p == NL) {
 726                         nl++;
 727                         lines++;
 728                 } else if (nl) {
 729                         if (*p != COMM)
 730                                 break;
 731                         nl = 0;
 732                 }
 733         }
 734         *(p - 1) = NULL;
 735         add_bent(bd->elist, base, NULL, NULL, NULL);
 736         *next = p;
 737         *line += lines;
 738 }
 739 
 740 /*
 741  * Parse out an operator (setprop) from the boot environment
 742  */
 743 static char *
 744 parse_cmd(benv_des_t *bd, char **next, int *line)
 745 {
 746         char *strbegin;
 747         char *badeof = "unexpected EOF in %s line %d";
 748         char *syntax = "syntax error in %s line %d";
 749         char *c = *next;
 750 
 751         /*
 752          * Skip spaces or tabs. New lines increase the line count.
 753          */
 754         while (isspace(*c)) {
 755                 if (*c++ == '\n')
 756                         (*line)++;
 757         }
 758 
 759         /*
 760          * Check for a the setprop command.  Currently that's all we
 761          * seem to support.
 762          *
 763          * XXX need support for setbinprop?
 764          */
 765 
 766         /*
 767          * Check first for end of file.  Finding one now would be okay.
 768          * We should also bail if we are at the start of a comment.
 769          */
 770         if (*c == '\0' || *c == COMM) {
 771                 *next = c;
 772                 return (NULL);
 773         }
 774 
 775         strbegin = c;
 776         while (*c && !isspace(*c))
 777                 c++;
 778 
 779         /*
 780          * Check again for end of file.  Finding one now would NOT be okay.
 781          */
 782         if (*c == '\0') {
 783                 exit(_error(NO_PERROR, badeof, bd->name, *line));
 784         }
 785 
 786         *c++ = '\0';
 787         *next = c;
 788 
 789         /*
 790          * Last check is to make sure the command is a setprop!
 791          */
 792         if (strcmp(strbegin, "setprop") != 0) {
 793                 exit(_error(NO_PERROR, syntax, bd->name, *line));
 794                 /* NOTREACHED */
 795         }
 796         return (strbegin);
 797 }
 798 
 799 /*
 800  * Parse out the name (LHS) of a setprop from the boot environment
 801  */
 802 static char *
 803 parse_name(benv_des_t *bd, char **next, int *line)
 804 {
 805         char *strbegin;
 806         char *badeof = "unexpected EOF in %s line %d";
 807         char *syntax = "syntax error in %s line %d";
 808         char *c = *next;
 809 
 810         /*
 811          * Skip spaces or tabs. No tolerance for new lines now.
 812          */
 813         while (isspace(*c)) {
 814                 if (*c++ == '\n')
 815                         exit(_error(NO_PERROR, syntax, bd->name, *line));
 816         }
 817 
 818         /*
 819          * Grab a name for the property to set.
 820          */
 821 
 822         /*
 823          * Check first for end of file.  Finding one now would NOT be okay.
 824          */
 825         if (*c == '\0') {
 826                 exit(_error(NO_PERROR, badeof, bd->name, *line));
 827         }
 828 
 829         strbegin = c;
 830         while (*c && !isspace(*c))
 831                 c++;
 832 
 833         /*
 834          * At this point in parsing we have 'setprop name'.  What follows
 835          * is a newline, other whitespace, or EOF.  Most of the time we
 836          * want to replace a white space character with a NULL to terminate
 837          * the name, and then continue on processing.  A newline here provides
 838          * the most grief.  If we just replace it with a null we'll
 839          * potentially get the setprop on the next line as the value of this
 840          * setprop! So, if the last thing we see is a newline we'll have to
 841          * dup the string.
 842          */
 843         if (isspace(*c)) {
 844                 if (*c == '\n') {
 845                         *c = '\0';
 846                         strbegin = strdup(strbegin);
 847                         *c = '\n';
 848                 } else {
 849                         *c++ = '\0';
 850                 }
 851         }
 852 
 853         *next = c;
 854         return (strbegin);
 855 }
 856 
 857 /*
 858  * Parse out the value (RHS) of a setprop line from the boot environment
 859  */
 860 static char *
 861 parse_value(benv_des_t *bd, char **next, int *line)
 862 {
 863         char *strbegin;
 864         char *badeof = "unexpected EOF in %s line %d";
 865         char *result;
 866         char *c = *next;
 867         char quote;
 868 
 869         /*
 870          * Skip spaces or tabs. A newline here would indicate a
 871          * NULL property value.
 872          */
 873         while (isspace(*c)) {
 874                 if (*c++ == '\n') {
 875                         (*line)++;
 876                         *next = c;
 877                         return (NULL);
 878                 }
 879         }
 880 
 881         /*
 882          * Grab the value of the property to set.
 883          */
 884 
 885         /*
 886          * Check first for end of file.  Finding one now would
 887          * also indicate a NULL property.
 888          */
 889         if (*c == '\0') {
 890                 *next = c;
 891                 return (NULL);
 892         }
 893 
 894         /*
 895          * Value may be quoted, in which case we assume the end of the value
 896          * comes with a closing quote.
 897          *
 898          * We also allow escaped quote characters inside the quoted value.
 899          *
 900          * For obvious reasons we do not attempt to parse variable references.
 901          */
 902         if (*c == '"' || *c == '\'') {
 903                 quote = *c;
 904                 c++;
 905                 strbegin = c;
 906                 result = c;
 907                 while (*c != quote) {
 908                         if (*c == '\\') {
 909                                 c++;
 910                         }
 911                         if (*c == '\0') {
 912                                 break;
 913                         }
 914                         *result++ = *c++;
 915                 }
 916 
 917                 /*
 918                  *  Throw fatal exception if no end quote found.
 919                  */
 920                 if (*c != quote) {
 921                         exit(_error(NO_PERROR, badeof, bd->name, *line));
 922                 }
 923 
 924                 *result = '\0';         /* Terminate the result */
 925                 c++;                    /* and step past the close quote */
 926         } else {
 927                 strbegin = c;
 928                 while (*c && !isspace(*c))
 929                         c++;
 930         }
 931 
 932         /*
 933          * Check again for end of file.  Finding one now is okay.
 934          */
 935         if (*c == '\0') {
 936                 *next = c;
 937                 return (strbegin);
 938         }
 939 
 940         *c++ = '\0';
 941         *next = c;
 942         return (strbegin);
 943 }
 944 
 945 /*
 946  * Add a command to the benv list.
 947  */
 948 static void
 949 add_cmd(benv_des_t *bd, char *last, char **next, int *line)
 950 {
 951         char *cmd, *name, *val;
 952 
 953         while (*next <= last && **next != COMM) {
 954                 if ((cmd = parse_cmd(bd, next, line)) == NULL)
 955                         break;
 956                 name = parse_name(bd, next, line);
 957                 val = parse_value(bd, next, line);
 958                 add_bent(bd->elist, NULL, cmd, name, val);
 959                 (*line)++;
 960         };
 961 
 962 }
 963 
 964 /*
 965  * Parse the benv (bootenv.rc) file and break it into a benv
 966  * list.  List entries may be comment blocks or commands.
 967  */
 968 static void
 969 parse_benv(benv_des_t *bd)
 970 {
 971         int line;
 972         char *pbase, *pend;
 973         char *tok, *tnext;
 974 
 975         line = 1;
 976         pbase = (char *)bd->adr;
 977         pend = pbase + bd->len;
 978 
 979         for (tok = tnext = pbase; tnext < pend && '\0' != *tnext; tok = tnext)
 980                 if (*tok == COMM)
 981                         add_comm(bd, tok, pend, &tnext, &line);
 982                 else
 983                         add_cmd(bd, pend, &tnext, &line);
 984 }
 985 
 986 static void
 987 write_benv(benv_des_t *bd)
 988 {
 989         FILE *fp;
 990         eplist_t *list, *e;
 991         benv_ent_t *bent;
 992         char *name;
 993 
 994         list = bd->elist;
 995 
 996         if (list->next == list)
 997                 return;
 998 
 999         if ((fp = fopen(bd->name, "w")) == NULL)
1000                 exit(_error(PERROR, "cannot open %s", bd->name));
1001 
1002         for (e = list->next; e != list; e = e->next) {
1003                 bent = (benv_ent_t *)e->item;
1004                 name = bent->name;
1005                 if (name) {
1006                         if (bent->val) {
1007                                 (void) fprintf(fp, "%s %s ",
1008                                     bent->cmd, bent->name);
1009                                 put_quoted(fp, bent->val);
1010                                 (void) fprintf(fp, "\n");
1011                         } else {
1012                                 (void) fprintf(fp, "%s %s\n",
1013                                     bent->cmd, bent->name);
1014                         }
1015                 } else {
1016                         (void) fprintf(fp, "%s\n", bent->cmd);
1017                 }
1018         }
1019 
1020         (void) fclose(fp);
1021 }
1022 
1023 static char *
1024 get_line(void)
1025 {
1026         int c;
1027         char *nl;
1028         static char line[256];
1029 
1030         if (fgets(line, sizeof (line), stdin) != NULL) {
1031                 /*
1032                  * Remove newline if present,
1033                  * otherwise discard rest of line.
1034                  */
1035                 if (nl = strchr(line, '\n'))
1036                         *nl = 0;
1037                 else
1038                         while ((c = getchar()) != '\n' && c != EOF)
1039                                 ;
1040                 return (line);
1041         } else
1042                 return (NULL);
1043 }
1044 
1045 int
1046 main(int argc, char **argv)
1047 {
1048         int c;
1049         int updates = 0;
1050         char *usage = "Usage: %s [-v] [-f prom-device]"
1051             " [variable[=value] ...]";
1052         eplist_t *elist;
1053         benv_des_t *bd;
1054         char *file = NULL;
1055 
1056         setpname(argv[0]);
1057 
1058         while ((c = getopt(argc, argv, "f:Itv")) != -1)
1059                 switch (c) {
1060                 case 'v':
1061                         verbose++;
1062                         break;
1063                 case 'f':
1064                         file = optarg;
1065                         break;
1066                 case 't':
1067                         test++;
1068                         break;
1069                 default:
1070                         exit(_error(NO_PERROR, usage, argv[0]));
1071                 }
1072 
1073         (void) uname(&uts_buf);
1074         bd = new_bd();
1075         init_benv(bd, file);
1076 
1077         map_benv(bd);
1078         if (bd->len) {
1079                 parse_benv(bd);
1080                 unmap_benv(bd);
1081         }
1082 
1083         elist = bd->elist;
1084 
1085         if (optind >= argc) {
1086                 print_vars(elist);
1087                 return (0);
1088         } else
1089                 while (optind < argc) {
1090                         /*
1091                          * If "-" specified, read variables from stdin;
1092                          * otherwise, process each argument as a variable
1093                          * print or set request.
1094                          */
1095                         if (strcmp(argv[optind], "-") == 0) {
1096                                 char *line;
1097 
1098                                 while ((line = get_line()) != NULL)
1099                                         updates += proc_var(line, elist);
1100                                 clearerr(stdin);
1101                         } else
1102                                 updates += proc_var(argv[optind], elist);
1103 
1104                         optind++;
1105                 }
1106 
1107         /*
1108          * don't write benv if we are processing delayed writes since
1109          * it is likely that the delayed writes changes bootenv.rc anyway...
1110          */
1111         if (updates)
1112                 write_benv(bd);
1113         close_kbenv();
1114 
1115         return (0);
1116 }