1484b5c25SMarcel Moolenaar /*- 2484b5c25SMarcel Moolenaar * Copyright (c) 2011 Marcel Moolenaar 3484b5c25SMarcel Moolenaar * All rights reserved. 4484b5c25SMarcel Moolenaar * 5484b5c25SMarcel Moolenaar * Redistribution and use in source and binary forms, with or without 6484b5c25SMarcel Moolenaar * modification, are permitted provided that the following conditions 7484b5c25SMarcel Moolenaar * are met: 8484b5c25SMarcel Moolenaar * 1. Redistributions of source code must retain the above copyright 9484b5c25SMarcel Moolenaar * notice, this list of conditions and the following disclaimer. 10484b5c25SMarcel Moolenaar * 2. Redistributions in binary form must reproduce the above copyright 11484b5c25SMarcel Moolenaar * notice, this list of conditions and the following disclaimer in the 12484b5c25SMarcel Moolenaar * documentation and/or other materials provided with the distribution. 13484b5c25SMarcel Moolenaar * 14484b5c25SMarcel Moolenaar * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR 15484b5c25SMarcel Moolenaar * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 16484b5c25SMarcel Moolenaar * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 17484b5c25SMarcel Moolenaar * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT, 18484b5c25SMarcel Moolenaar * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 19484b5c25SMarcel Moolenaar * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 20484b5c25SMarcel Moolenaar * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 21484b5c25SMarcel Moolenaar * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 22484b5c25SMarcel Moolenaar * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 23484b5c25SMarcel Moolenaar * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 24484b5c25SMarcel Moolenaar */ 25484b5c25SMarcel Moolenaar 26484b5c25SMarcel Moolenaar #include <sys/cdefs.h> 27484b5c25SMarcel Moolenaar __FBSDID("$FreeBSD$"); 28484b5c25SMarcel Moolenaar 29484b5c25SMarcel Moolenaar #include <sys/param.h> 30484b5c25SMarcel Moolenaar #include <sys/queue.h> 31484b5c25SMarcel Moolenaar #include <sys/sbuf.h> 32484b5c25SMarcel Moolenaar #include <sys/stat.h> 33484b5c25SMarcel Moolenaar #include <sys/types.h> 34484b5c25SMarcel Moolenaar #include <assert.h> 35484b5c25SMarcel Moolenaar #include <errno.h> 36484b5c25SMarcel Moolenaar #include <fcntl.h> 37484b5c25SMarcel Moolenaar #include <grp.h> 38484b5c25SMarcel Moolenaar #include <inttypes.h> 39484b5c25SMarcel Moolenaar #include <pwd.h> 40484b5c25SMarcel Moolenaar #include <stdarg.h> 41484b5c25SMarcel Moolenaar #include <stdbool.h> 42484b5c25SMarcel Moolenaar #include <stddef.h> 43484b5c25SMarcel Moolenaar #include <stdio.h> 44484b5c25SMarcel Moolenaar #include <stdlib.h> 45484b5c25SMarcel Moolenaar #include <string.h> 46484b5c25SMarcel Moolenaar #include <strings.h> 47484b5c25SMarcel Moolenaar #include <unistd.h> 48484b5c25SMarcel Moolenaar 49484b5c25SMarcel Moolenaar #include "makefs.h" 50484b5c25SMarcel Moolenaar 51484b5c25SMarcel Moolenaar #define IS_DOT(nm) ((nm)[0] == '.' && (nm)[1] == '\0') 52484b5c25SMarcel Moolenaar #define IS_DOTDOT(nm) ((nm)[0] == '.' && (nm)[1] == '.' && (nm)[2] == '\0') 53484b5c25SMarcel Moolenaar 54484b5c25SMarcel Moolenaar struct mtree_fileinfo { 55484b5c25SMarcel Moolenaar SLIST_ENTRY(mtree_fileinfo) next; 56484b5c25SMarcel Moolenaar FILE *fp; 57484b5c25SMarcel Moolenaar const char *name; 58484b5c25SMarcel Moolenaar u_int line; 59484b5c25SMarcel Moolenaar }; 60484b5c25SMarcel Moolenaar 61484b5c25SMarcel Moolenaar /* Global state used while parsing. */ 62484b5c25SMarcel Moolenaar static SLIST_HEAD(, mtree_fileinfo) mtree_fileinfo = 63484b5c25SMarcel Moolenaar SLIST_HEAD_INITIALIZER(mtree_fileinfo); 64484b5c25SMarcel Moolenaar static fsnode *mtree_root; 65484b5c25SMarcel Moolenaar static fsnode *mtree_current; 66484b5c25SMarcel Moolenaar static fsnode mtree_global; 67484b5c25SMarcel Moolenaar static fsinode mtree_global_inode; 68484b5c25SMarcel Moolenaar static u_int errors, warnings; 69484b5c25SMarcel Moolenaar 70484b5c25SMarcel Moolenaar static void mtree_error(const char *, ...) __printflike(1, 2); 71484b5c25SMarcel Moolenaar static void mtree_warning(const char *, ...) __printflike(1, 2); 72484b5c25SMarcel Moolenaar 73484b5c25SMarcel Moolenaar static int 74484b5c25SMarcel Moolenaar mtree_file_push(const char *name, FILE *fp) 75484b5c25SMarcel Moolenaar { 76484b5c25SMarcel Moolenaar struct mtree_fileinfo *fi; 77484b5c25SMarcel Moolenaar 78484b5c25SMarcel Moolenaar fi = malloc(sizeof(*fi)); 79484b5c25SMarcel Moolenaar if (fi == NULL) 80484b5c25SMarcel Moolenaar return (ENOMEM); 81484b5c25SMarcel Moolenaar 82484b5c25SMarcel Moolenaar if (strcmp(name, "-") == 0) 83484b5c25SMarcel Moolenaar fi->name = strdup("(stdin)"); 84484b5c25SMarcel Moolenaar else 85484b5c25SMarcel Moolenaar fi->name = strdup(name); 86484b5c25SMarcel Moolenaar if (fi->name == NULL) { 87484b5c25SMarcel Moolenaar free(fi); 88484b5c25SMarcel Moolenaar return (ENOMEM); 89484b5c25SMarcel Moolenaar } 90484b5c25SMarcel Moolenaar 91484b5c25SMarcel Moolenaar fi->fp = fp; 92484b5c25SMarcel Moolenaar fi->line = 0; 93484b5c25SMarcel Moolenaar 94484b5c25SMarcel Moolenaar SLIST_INSERT_HEAD(&mtree_fileinfo, fi, next); 95484b5c25SMarcel Moolenaar return (0); 96484b5c25SMarcel Moolenaar } 97484b5c25SMarcel Moolenaar 98484b5c25SMarcel Moolenaar static void 99484b5c25SMarcel Moolenaar mtree_print(const char *msgtype, const char *fmt, va_list ap) 100484b5c25SMarcel Moolenaar { 101484b5c25SMarcel Moolenaar struct mtree_fileinfo *fi; 102484b5c25SMarcel Moolenaar 103484b5c25SMarcel Moolenaar if (msgtype != NULL) { 104484b5c25SMarcel Moolenaar fi = SLIST_FIRST(&mtree_fileinfo); 105484b5c25SMarcel Moolenaar if (fi != NULL) 106484b5c25SMarcel Moolenaar fprintf(stderr, "%s:%u: ", fi->name, fi->line); 107484b5c25SMarcel Moolenaar fprintf(stderr, "%s: ", msgtype); 108484b5c25SMarcel Moolenaar } 109484b5c25SMarcel Moolenaar vfprintf(stderr, fmt, ap); 110484b5c25SMarcel Moolenaar } 111484b5c25SMarcel Moolenaar 112484b5c25SMarcel Moolenaar static void 113484b5c25SMarcel Moolenaar mtree_error(const char *fmt, ...) 114484b5c25SMarcel Moolenaar { 115484b5c25SMarcel Moolenaar va_list ap; 116484b5c25SMarcel Moolenaar 117484b5c25SMarcel Moolenaar va_start(ap, fmt); 118484b5c25SMarcel Moolenaar mtree_print("error", fmt, ap); 119484b5c25SMarcel Moolenaar va_end(ap); 120484b5c25SMarcel Moolenaar 121484b5c25SMarcel Moolenaar errors++; 122484b5c25SMarcel Moolenaar fputc('\n', stderr); 123484b5c25SMarcel Moolenaar } 124484b5c25SMarcel Moolenaar 125484b5c25SMarcel Moolenaar static void 126484b5c25SMarcel Moolenaar mtree_warning(const char *fmt, ...) 127484b5c25SMarcel Moolenaar { 128484b5c25SMarcel Moolenaar va_list ap; 129484b5c25SMarcel Moolenaar 130484b5c25SMarcel Moolenaar va_start(ap, fmt); 131484b5c25SMarcel Moolenaar mtree_print("warning", fmt, ap); 132484b5c25SMarcel Moolenaar va_end(ap); 133484b5c25SMarcel Moolenaar 134484b5c25SMarcel Moolenaar warnings++; 135484b5c25SMarcel Moolenaar fputc('\n', stderr); 136484b5c25SMarcel Moolenaar } 137484b5c25SMarcel Moolenaar 138*31c0b21fSSimon J. Gerraty #ifndef MAKEFS_MAX_TREE_DEPTH 139*31c0b21fSSimon J. Gerraty # define MAKEFS_MAX_TREE_DEPTH (MAXPATHLEN/2) 140*31c0b21fSSimon J. Gerraty #endif 141*31c0b21fSSimon J. Gerraty 142*31c0b21fSSimon J. Gerraty /* construct path to node->name */ 143*31c0b21fSSimon J. Gerraty static char * 144*31c0b21fSSimon J. Gerraty mtree_file_path(fsnode *node) 145*31c0b21fSSimon J. Gerraty { 146*31c0b21fSSimon J. Gerraty fsnode *pnode; 147*31c0b21fSSimon J. Gerraty struct sbuf *sb; 148*31c0b21fSSimon J. Gerraty char *res, *rp[MAKEFS_MAX_TREE_DEPTH]; 149*31c0b21fSSimon J. Gerraty int depth; 150*31c0b21fSSimon J. Gerraty 151*31c0b21fSSimon J. Gerraty depth = 0; 152*31c0b21fSSimon J. Gerraty rp[depth] = node->name; 153*31c0b21fSSimon J. Gerraty for (pnode = node->parent; pnode && depth < MAKEFS_MAX_TREE_DEPTH; 154*31c0b21fSSimon J. Gerraty pnode = pnode->parent) { 155*31c0b21fSSimon J. Gerraty if (strcmp(pnode->name, ".") == 0) 156*31c0b21fSSimon J. Gerraty break; 157*31c0b21fSSimon J. Gerraty rp[++depth] = pnode->name; 158*31c0b21fSSimon J. Gerraty } 159*31c0b21fSSimon J. Gerraty 160*31c0b21fSSimon J. Gerraty sb = sbuf_new(NULL, NULL, 0, SBUF_AUTOEXTEND); 161*31c0b21fSSimon J. Gerraty if (sb == NULL) { 162*31c0b21fSSimon J. Gerraty errno = ENOMEM; 163*31c0b21fSSimon J. Gerraty return (NULL); 164*31c0b21fSSimon J. Gerraty } 165*31c0b21fSSimon J. Gerraty while (depth > 0) { 166*31c0b21fSSimon J. Gerraty sbuf_cat(sb, rp[depth--]); 167*31c0b21fSSimon J. Gerraty sbuf_putc(sb, '/'); 168*31c0b21fSSimon J. Gerraty } 169*31c0b21fSSimon J. Gerraty sbuf_cat(sb, rp[depth]); 170*31c0b21fSSimon J. Gerraty sbuf_finish(sb); 171*31c0b21fSSimon J. Gerraty res = strdup(sbuf_data(sb)); 172*31c0b21fSSimon J. Gerraty sbuf_delete(sb); 173*31c0b21fSSimon J. Gerraty if (res == NULL) 174*31c0b21fSSimon J. Gerraty errno = ENOMEM; 175*31c0b21fSSimon J. Gerraty return res; 176*31c0b21fSSimon J. Gerraty 177*31c0b21fSSimon J. Gerraty } 178*31c0b21fSSimon J. Gerraty 179484b5c25SMarcel Moolenaar /* mtree_resolve() sets errno to indicate why NULL was returned. */ 180484b5c25SMarcel Moolenaar static char * 181484b5c25SMarcel Moolenaar mtree_resolve(const char *spec, int *istemp) 182484b5c25SMarcel Moolenaar { 183484b5c25SMarcel Moolenaar struct sbuf *sb; 184484b5c25SMarcel Moolenaar char *res, *var; 185484b5c25SMarcel Moolenaar const char *base, *p, *v; 186484b5c25SMarcel Moolenaar size_t len; 187484b5c25SMarcel Moolenaar int c, error, quoted, subst; 188484b5c25SMarcel Moolenaar 189484b5c25SMarcel Moolenaar len = strlen(spec); 190484b5c25SMarcel Moolenaar if (len == 0) { 191484b5c25SMarcel Moolenaar errno = EINVAL; 192484b5c25SMarcel Moolenaar return (NULL); 193484b5c25SMarcel Moolenaar } 194484b5c25SMarcel Moolenaar 195484b5c25SMarcel Moolenaar c = (len > 1) ? (spec[0] == spec[len - 1]) ? spec[0] : 0 : 0; 196484b5c25SMarcel Moolenaar *istemp = (c == '`') ? 1 : 0; 197484b5c25SMarcel Moolenaar subst = (c == '`' || c == '"') ? 1 : 0; 198484b5c25SMarcel Moolenaar quoted = (subst || c == '\'') ? 1 : 0; 199484b5c25SMarcel Moolenaar 200484b5c25SMarcel Moolenaar if (!subst) { 201484b5c25SMarcel Moolenaar res = strdup(spec + quoted); 202484b5c25SMarcel Moolenaar if (res != NULL && quoted) 203484b5c25SMarcel Moolenaar res[len - 2] = '\0'; 204484b5c25SMarcel Moolenaar return (res); 205484b5c25SMarcel Moolenaar } 206484b5c25SMarcel Moolenaar 207484b5c25SMarcel Moolenaar sb = sbuf_new_auto(); 208484b5c25SMarcel Moolenaar if (sb == NULL) { 209484b5c25SMarcel Moolenaar errno = ENOMEM; 210484b5c25SMarcel Moolenaar return (NULL); 211484b5c25SMarcel Moolenaar } 212484b5c25SMarcel Moolenaar 213484b5c25SMarcel Moolenaar base = spec + 1; 214484b5c25SMarcel Moolenaar len -= 2; 215484b5c25SMarcel Moolenaar error = 0; 216484b5c25SMarcel Moolenaar while (len > 0) { 217484b5c25SMarcel Moolenaar p = strchr(base, '$'); 218484b5c25SMarcel Moolenaar if (p == NULL) { 219484b5c25SMarcel Moolenaar sbuf_bcat(sb, base, len); 220484b5c25SMarcel Moolenaar base += len; 221484b5c25SMarcel Moolenaar len = 0; 222484b5c25SMarcel Moolenaar continue; 223484b5c25SMarcel Moolenaar } 224484b5c25SMarcel Moolenaar /* The following is safe. spec always starts with a quote. */ 225484b5c25SMarcel Moolenaar if (p[-1] == '\\') 226484b5c25SMarcel Moolenaar p--; 227484b5c25SMarcel Moolenaar if (base != p) { 228484b5c25SMarcel Moolenaar sbuf_bcat(sb, base, p - base); 229484b5c25SMarcel Moolenaar len -= p - base; 230484b5c25SMarcel Moolenaar base = p; 231484b5c25SMarcel Moolenaar } 232484b5c25SMarcel Moolenaar if (*p == '\\') { 233484b5c25SMarcel Moolenaar sbuf_putc(sb, '$'); 234484b5c25SMarcel Moolenaar base += 2; 235484b5c25SMarcel Moolenaar len -= 2; 236484b5c25SMarcel Moolenaar continue; 237484b5c25SMarcel Moolenaar } 238484b5c25SMarcel Moolenaar /* Skip the '$'. */ 239484b5c25SMarcel Moolenaar base++; 240484b5c25SMarcel Moolenaar len--; 241484b5c25SMarcel Moolenaar /* Handle ${X} vs $X. */ 242484b5c25SMarcel Moolenaar v = base; 243484b5c25SMarcel Moolenaar if (*base == '{') { 244484b5c25SMarcel Moolenaar p = strchr(v, '}'); 245484b5c25SMarcel Moolenaar if (p == NULL) 246484b5c25SMarcel Moolenaar p = v; 247484b5c25SMarcel Moolenaar } else 248484b5c25SMarcel Moolenaar p = v; 249484b5c25SMarcel Moolenaar len -= (p + 1) - base; 250484b5c25SMarcel Moolenaar base = p + 1; 251484b5c25SMarcel Moolenaar 252484b5c25SMarcel Moolenaar if (v == p) { 253484b5c25SMarcel Moolenaar sbuf_putc(sb, *v); 254484b5c25SMarcel Moolenaar continue; 255484b5c25SMarcel Moolenaar } 256484b5c25SMarcel Moolenaar 257484b5c25SMarcel Moolenaar error = ENOMEM; 258484b5c25SMarcel Moolenaar var = calloc(p - v, 1); 259484b5c25SMarcel Moolenaar if (var == NULL) 260484b5c25SMarcel Moolenaar break; 261484b5c25SMarcel Moolenaar 262484b5c25SMarcel Moolenaar memcpy(var, v + 1, p - v - 1); 263484b5c25SMarcel Moolenaar if (strcmp(var, ".CURDIR") == 0) { 264484b5c25SMarcel Moolenaar res = getcwd(NULL, 0); 265484b5c25SMarcel Moolenaar if (res == NULL) 266484b5c25SMarcel Moolenaar break; 267484b5c25SMarcel Moolenaar } else if (strcmp(var, ".PROG") == 0) { 268484b5c25SMarcel Moolenaar res = strdup(getprogname()); 269484b5c25SMarcel Moolenaar if (res == NULL) 270484b5c25SMarcel Moolenaar break; 271484b5c25SMarcel Moolenaar } else { 272484b5c25SMarcel Moolenaar v = getenv(var); 273484b5c25SMarcel Moolenaar if (v != NULL) { 274484b5c25SMarcel Moolenaar res = strdup(v); 275484b5c25SMarcel Moolenaar if (res == NULL) 276484b5c25SMarcel Moolenaar break; 277484b5c25SMarcel Moolenaar } else 278484b5c25SMarcel Moolenaar res = NULL; 279484b5c25SMarcel Moolenaar } 280484b5c25SMarcel Moolenaar error = 0; 281484b5c25SMarcel Moolenaar 282484b5c25SMarcel Moolenaar if (res != NULL) { 283484b5c25SMarcel Moolenaar sbuf_cat(sb, res); 284484b5c25SMarcel Moolenaar free(res); 285484b5c25SMarcel Moolenaar } 286484b5c25SMarcel Moolenaar free(var); 287484b5c25SMarcel Moolenaar } 288484b5c25SMarcel Moolenaar 289484b5c25SMarcel Moolenaar sbuf_finish(sb); 290484b5c25SMarcel Moolenaar res = (error == 0) ? strdup(sbuf_data(sb)) : NULL; 291484b5c25SMarcel Moolenaar sbuf_delete(sb); 292484b5c25SMarcel Moolenaar if (res == NULL) 293484b5c25SMarcel Moolenaar errno = ENOMEM; 294484b5c25SMarcel Moolenaar return (res); 295484b5c25SMarcel Moolenaar } 296484b5c25SMarcel Moolenaar 297484b5c25SMarcel Moolenaar static int 298484b5c25SMarcel Moolenaar skip_over(FILE *fp, const char *cs) 299484b5c25SMarcel Moolenaar { 300484b5c25SMarcel Moolenaar int c; 301484b5c25SMarcel Moolenaar 302484b5c25SMarcel Moolenaar c = getc(fp); 303484b5c25SMarcel Moolenaar while (c != EOF && strchr(cs, c) != NULL) 304484b5c25SMarcel Moolenaar c = getc(fp); 305484b5c25SMarcel Moolenaar if (c != EOF) { 306484b5c25SMarcel Moolenaar ungetc(c, fp); 307484b5c25SMarcel Moolenaar return (0); 308484b5c25SMarcel Moolenaar } 309484b5c25SMarcel Moolenaar return (ferror(fp) ? errno : -1); 310484b5c25SMarcel Moolenaar } 311484b5c25SMarcel Moolenaar 312484b5c25SMarcel Moolenaar static int 313484b5c25SMarcel Moolenaar skip_to(FILE *fp, const char *cs) 314484b5c25SMarcel Moolenaar { 315484b5c25SMarcel Moolenaar int c; 316484b5c25SMarcel Moolenaar 317484b5c25SMarcel Moolenaar c = getc(fp); 318484b5c25SMarcel Moolenaar while (c != EOF && strchr(cs, c) == NULL) 319484b5c25SMarcel Moolenaar c = getc(fp); 320484b5c25SMarcel Moolenaar if (c != EOF) { 321484b5c25SMarcel Moolenaar ungetc(c, fp); 322484b5c25SMarcel Moolenaar return (0); 323484b5c25SMarcel Moolenaar } 324484b5c25SMarcel Moolenaar return (ferror(fp) ? errno : -1); 325484b5c25SMarcel Moolenaar } 326484b5c25SMarcel Moolenaar 327484b5c25SMarcel Moolenaar static int 328484b5c25SMarcel Moolenaar read_word(FILE *fp, char *buf, size_t bufsz) 329484b5c25SMarcel Moolenaar { 330484b5c25SMarcel Moolenaar struct mtree_fileinfo *fi; 331484b5c25SMarcel Moolenaar size_t idx, qidx; 332484b5c25SMarcel Moolenaar int c, done, error, esc, qlvl; 333484b5c25SMarcel Moolenaar 334484b5c25SMarcel Moolenaar if (bufsz == 0) 335484b5c25SMarcel Moolenaar return (EINVAL); 336484b5c25SMarcel Moolenaar 337484b5c25SMarcel Moolenaar done = 0; 338484b5c25SMarcel Moolenaar esc = 0; 339484b5c25SMarcel Moolenaar idx = 0; 340484b5c25SMarcel Moolenaar qidx = -1; 341484b5c25SMarcel Moolenaar qlvl = 0; 342484b5c25SMarcel Moolenaar do { 343484b5c25SMarcel Moolenaar c = getc(fp); 344484b5c25SMarcel Moolenaar switch (c) { 345484b5c25SMarcel Moolenaar case EOF: 346484b5c25SMarcel Moolenaar buf[idx] = '\0'; 347484b5c25SMarcel Moolenaar error = ferror(fp) ? errno : -1; 348484b5c25SMarcel Moolenaar if (error == -1) 349484b5c25SMarcel Moolenaar mtree_error("unexpected end of file"); 350484b5c25SMarcel Moolenaar return (error); 351484b5c25SMarcel Moolenaar case '\\': 352484b5c25SMarcel Moolenaar esc++; 353484b5c25SMarcel Moolenaar if (esc == 1) 354484b5c25SMarcel Moolenaar continue; 355484b5c25SMarcel Moolenaar break; 356484b5c25SMarcel Moolenaar case '`': 357484b5c25SMarcel Moolenaar case '\'': 358484b5c25SMarcel Moolenaar case '"': 359484b5c25SMarcel Moolenaar if (esc) 360484b5c25SMarcel Moolenaar break; 361484b5c25SMarcel Moolenaar if (qlvl == 0) { 362484b5c25SMarcel Moolenaar qlvl++; 363484b5c25SMarcel Moolenaar qidx = idx; 364484b5c25SMarcel Moolenaar } else if (c == buf[qidx]) { 365484b5c25SMarcel Moolenaar qlvl--; 366484b5c25SMarcel Moolenaar if (qlvl > 0) { 367484b5c25SMarcel Moolenaar do { 368484b5c25SMarcel Moolenaar qidx--; 369484b5c25SMarcel Moolenaar } while (buf[qidx] != '`' && 370484b5c25SMarcel Moolenaar buf[qidx] != '\'' && 371484b5c25SMarcel Moolenaar buf[qidx] != '"'); 372484b5c25SMarcel Moolenaar } else 373484b5c25SMarcel Moolenaar qidx = -1; 374484b5c25SMarcel Moolenaar } else { 375484b5c25SMarcel Moolenaar qlvl++; 376484b5c25SMarcel Moolenaar qidx = idx; 377484b5c25SMarcel Moolenaar } 378484b5c25SMarcel Moolenaar break; 379484b5c25SMarcel Moolenaar case ' ': 380484b5c25SMarcel Moolenaar case '\t': 381484b5c25SMarcel Moolenaar case '\n': 382484b5c25SMarcel Moolenaar if (!esc && qlvl == 0) { 383484b5c25SMarcel Moolenaar ungetc(c, fp); 384484b5c25SMarcel Moolenaar c = '\0'; 385484b5c25SMarcel Moolenaar done = 1; 386484b5c25SMarcel Moolenaar break; 387484b5c25SMarcel Moolenaar } 388484b5c25SMarcel Moolenaar if (c == '\n') { 389484b5c25SMarcel Moolenaar /* 390484b5c25SMarcel Moolenaar * We going to eat the newline ourselves. 391484b5c25SMarcel Moolenaar */ 392484b5c25SMarcel Moolenaar if (qlvl > 0) 393484b5c25SMarcel Moolenaar mtree_warning("quoted word straddles " 394484b5c25SMarcel Moolenaar "onto next line."); 395484b5c25SMarcel Moolenaar fi = SLIST_FIRST(&mtree_fileinfo); 396484b5c25SMarcel Moolenaar fi->line++; 397484b5c25SMarcel Moolenaar } 398484b5c25SMarcel Moolenaar break; 399484b5c25SMarcel Moolenaar case 'a': 400484b5c25SMarcel Moolenaar if (esc) 401484b5c25SMarcel Moolenaar c = '\a'; 402484b5c25SMarcel Moolenaar break; 403484b5c25SMarcel Moolenaar case 'b': 404484b5c25SMarcel Moolenaar if (esc) 405484b5c25SMarcel Moolenaar c = '\b'; 406484b5c25SMarcel Moolenaar break; 407484b5c25SMarcel Moolenaar case 'f': 408484b5c25SMarcel Moolenaar if (esc) 409484b5c25SMarcel Moolenaar c = '\f'; 410484b5c25SMarcel Moolenaar break; 411484b5c25SMarcel Moolenaar case 'n': 412484b5c25SMarcel Moolenaar if (esc) 413484b5c25SMarcel Moolenaar c = '\n'; 414484b5c25SMarcel Moolenaar break; 415484b5c25SMarcel Moolenaar case 'r': 416484b5c25SMarcel Moolenaar if (esc) 417484b5c25SMarcel Moolenaar c = '\r'; 418484b5c25SMarcel Moolenaar break; 419484b5c25SMarcel Moolenaar case 't': 420484b5c25SMarcel Moolenaar if (esc) 421484b5c25SMarcel Moolenaar c = '\t'; 422484b5c25SMarcel Moolenaar break; 423484b5c25SMarcel Moolenaar case 'v': 424484b5c25SMarcel Moolenaar if (esc) 425484b5c25SMarcel Moolenaar c = '\v'; 426484b5c25SMarcel Moolenaar break; 427484b5c25SMarcel Moolenaar } 428484b5c25SMarcel Moolenaar buf[idx++] = c; 429484b5c25SMarcel Moolenaar esc = 0; 430484b5c25SMarcel Moolenaar } while (idx < bufsz && !done); 431484b5c25SMarcel Moolenaar 432484b5c25SMarcel Moolenaar if (idx >= bufsz) { 433484b5c25SMarcel Moolenaar mtree_error("word too long to fit buffer (max %zu characters)", 434484b5c25SMarcel Moolenaar bufsz); 435484b5c25SMarcel Moolenaar skip_to(fp, " \t\n"); 436484b5c25SMarcel Moolenaar } 437484b5c25SMarcel Moolenaar return (0); 438484b5c25SMarcel Moolenaar } 439484b5c25SMarcel Moolenaar 440484b5c25SMarcel Moolenaar static fsnode * 441484b5c25SMarcel Moolenaar create_node(const char *name, u_int type, fsnode *parent, fsnode *global) 442484b5c25SMarcel Moolenaar { 443484b5c25SMarcel Moolenaar fsnode *n; 444484b5c25SMarcel Moolenaar 445484b5c25SMarcel Moolenaar n = calloc(1, sizeof(*n)); 446484b5c25SMarcel Moolenaar if (n == NULL) 447484b5c25SMarcel Moolenaar return (NULL); 448484b5c25SMarcel Moolenaar 449484b5c25SMarcel Moolenaar n->name = strdup(name); 450484b5c25SMarcel Moolenaar if (n->name == NULL) { 451484b5c25SMarcel Moolenaar free(n); 452484b5c25SMarcel Moolenaar return (NULL); 453484b5c25SMarcel Moolenaar } 454484b5c25SMarcel Moolenaar 455484b5c25SMarcel Moolenaar n->type = (type == 0) ? global->type : type; 456484b5c25SMarcel Moolenaar n->parent = parent; 457484b5c25SMarcel Moolenaar 458484b5c25SMarcel Moolenaar n->inode = calloc(1, sizeof(*n->inode)); 459484b5c25SMarcel Moolenaar if (n->inode == NULL) { 460484b5c25SMarcel Moolenaar free(n->name); 461484b5c25SMarcel Moolenaar free(n); 462484b5c25SMarcel Moolenaar return (NULL); 463484b5c25SMarcel Moolenaar } 464484b5c25SMarcel Moolenaar 465484b5c25SMarcel Moolenaar /* Assign global options/defaults. */ 466484b5c25SMarcel Moolenaar bcopy(global->inode, n->inode, sizeof(*n->inode)); 467484b5c25SMarcel Moolenaar n->inode->st.st_mode = (n->inode->st.st_mode & ~S_IFMT) | n->type; 468484b5c25SMarcel Moolenaar 469484b5c25SMarcel Moolenaar if (n->type == S_IFLNK) 470484b5c25SMarcel Moolenaar n->symlink = global->symlink; 471484b5c25SMarcel Moolenaar else if (n->type == S_IFREG) 472484b5c25SMarcel Moolenaar n->contents = global->contents; 473484b5c25SMarcel Moolenaar 474484b5c25SMarcel Moolenaar return (n); 475484b5c25SMarcel Moolenaar } 476484b5c25SMarcel Moolenaar 477484b5c25SMarcel Moolenaar static void 478484b5c25SMarcel Moolenaar destroy_node(fsnode *n) 479484b5c25SMarcel Moolenaar { 480484b5c25SMarcel Moolenaar 481484b5c25SMarcel Moolenaar assert(n != NULL); 482484b5c25SMarcel Moolenaar assert(n->name != NULL); 483484b5c25SMarcel Moolenaar assert(n->inode != NULL); 484484b5c25SMarcel Moolenaar 485484b5c25SMarcel Moolenaar free(n->inode); 486484b5c25SMarcel Moolenaar free(n->name); 487484b5c25SMarcel Moolenaar free(n); 488484b5c25SMarcel Moolenaar } 489484b5c25SMarcel Moolenaar 490484b5c25SMarcel Moolenaar static int 491484b5c25SMarcel Moolenaar read_number(const char *tok, u_int base, intmax_t *res, intmax_t min, 492484b5c25SMarcel Moolenaar intmax_t max) 493484b5c25SMarcel Moolenaar { 494484b5c25SMarcel Moolenaar char *end; 495484b5c25SMarcel Moolenaar intmax_t val; 496484b5c25SMarcel Moolenaar 497484b5c25SMarcel Moolenaar val = strtoimax(tok, &end, base); 498484b5c25SMarcel Moolenaar if (end == tok || end[0] != '\0') 499484b5c25SMarcel Moolenaar return (EINVAL); 500484b5c25SMarcel Moolenaar if (val < min || val > max) 501484b5c25SMarcel Moolenaar return (EDOM); 502484b5c25SMarcel Moolenaar *res = val; 503484b5c25SMarcel Moolenaar return (0); 504484b5c25SMarcel Moolenaar } 505484b5c25SMarcel Moolenaar 506484b5c25SMarcel Moolenaar static int 507484b5c25SMarcel Moolenaar read_mtree_keywords(FILE *fp, fsnode *node) 508484b5c25SMarcel Moolenaar { 509484b5c25SMarcel Moolenaar char keyword[PATH_MAX]; 510484b5c25SMarcel Moolenaar char *name, *p, *value; 511484b5c25SMarcel Moolenaar struct group *grent; 512484b5c25SMarcel Moolenaar struct passwd *pwent; 513484b5c25SMarcel Moolenaar struct stat *st, sb; 514484b5c25SMarcel Moolenaar intmax_t num; 515484b5c25SMarcel Moolenaar u_long flset, flclr; 516484b5c25SMarcel Moolenaar int error, istemp, type; 517484b5c25SMarcel Moolenaar 518484b5c25SMarcel Moolenaar st = &node->inode->st; 519484b5c25SMarcel Moolenaar do { 520484b5c25SMarcel Moolenaar error = skip_over(fp, " \t"); 521484b5c25SMarcel Moolenaar if (error) 522484b5c25SMarcel Moolenaar break; 523484b5c25SMarcel Moolenaar 524484b5c25SMarcel Moolenaar error = read_word(fp, keyword, sizeof(keyword)); 525484b5c25SMarcel Moolenaar if (error) 526484b5c25SMarcel Moolenaar break; 527484b5c25SMarcel Moolenaar 528484b5c25SMarcel Moolenaar if (keyword[0] == '\0') 529484b5c25SMarcel Moolenaar break; 530484b5c25SMarcel Moolenaar 531484b5c25SMarcel Moolenaar value = strchr(keyword, '='); 532484b5c25SMarcel Moolenaar if (value != NULL) 533484b5c25SMarcel Moolenaar *value++ = '\0'; 534484b5c25SMarcel Moolenaar 535484b5c25SMarcel Moolenaar /* 536484b5c25SMarcel Moolenaar * We use EINVAL, ENOATTR, ENOSYS and ENXIO to signal 537484b5c25SMarcel Moolenaar * certain conditions: 538484b5c25SMarcel Moolenaar * EINVAL - Value provided for a keyword that does 539484b5c25SMarcel Moolenaar * not take a value. The value is ignored. 540484b5c25SMarcel Moolenaar * ENOATTR - Value missing for a keyword that needs 541484b5c25SMarcel Moolenaar * a value. The keyword is ignored. 542484b5c25SMarcel Moolenaar * ENOSYS - Unsupported keyword encountered. The 543484b5c25SMarcel Moolenaar * keyword is ignored. 544484b5c25SMarcel Moolenaar * ENXIO - Value provided for a keyword that does 545484b5c25SMarcel Moolenaar * not take a value. The value is ignored. 546484b5c25SMarcel Moolenaar */ 547484b5c25SMarcel Moolenaar switch (keyword[0]) { 548484b5c25SMarcel Moolenaar case 'c': 549484b5c25SMarcel Moolenaar if (strcmp(keyword, "contents") == 0) { 550484b5c25SMarcel Moolenaar if (value == NULL) { 551484b5c25SMarcel Moolenaar error = ENOATTR; 552484b5c25SMarcel Moolenaar break; 553484b5c25SMarcel Moolenaar } 554484b5c25SMarcel Moolenaar node->contents = strdup(value); 555484b5c25SMarcel Moolenaar } else 556484b5c25SMarcel Moolenaar error = ENOSYS; 557484b5c25SMarcel Moolenaar break; 558484b5c25SMarcel Moolenaar case 'f': 559484b5c25SMarcel Moolenaar if (strcmp(keyword, "flags") == 0) { 560484b5c25SMarcel Moolenaar if (value == NULL) { 561484b5c25SMarcel Moolenaar error = ENOATTR; 562484b5c25SMarcel Moolenaar break; 563484b5c25SMarcel Moolenaar } 564484b5c25SMarcel Moolenaar flset = flclr = 0; 565484b5c25SMarcel Moolenaar if (!strtofflags(&value, &flset, &flclr)) { 566484b5c25SMarcel Moolenaar st->st_flags &= ~flclr; 567484b5c25SMarcel Moolenaar st->st_flags |= flset; 568484b5c25SMarcel Moolenaar } else 569484b5c25SMarcel Moolenaar error = errno; 570484b5c25SMarcel Moolenaar } else 571484b5c25SMarcel Moolenaar error = ENOSYS; 572484b5c25SMarcel Moolenaar break; 573484b5c25SMarcel Moolenaar case 'g': 574484b5c25SMarcel Moolenaar if (strcmp(keyword, "gid") == 0) { 575484b5c25SMarcel Moolenaar if (value == NULL) { 576484b5c25SMarcel Moolenaar error = ENOATTR; 577484b5c25SMarcel Moolenaar break; 578484b5c25SMarcel Moolenaar } 579484b5c25SMarcel Moolenaar error = read_number(value, 10, &num, 580484b5c25SMarcel Moolenaar 0, UINT_MAX); 581484b5c25SMarcel Moolenaar if (!error) 582484b5c25SMarcel Moolenaar st->st_gid = num; 583484b5c25SMarcel Moolenaar } else if (strcmp(keyword, "gname") == 0) { 584484b5c25SMarcel Moolenaar if (value == NULL) { 585484b5c25SMarcel Moolenaar error = ENOATTR; 586484b5c25SMarcel Moolenaar break; 587484b5c25SMarcel Moolenaar } 588484b5c25SMarcel Moolenaar grent = getgrnam(value); 589484b5c25SMarcel Moolenaar if (grent != NULL) 590484b5c25SMarcel Moolenaar st->st_gid = grent->gr_gid; 591484b5c25SMarcel Moolenaar else 592484b5c25SMarcel Moolenaar error = errno; 593484b5c25SMarcel Moolenaar } else 594484b5c25SMarcel Moolenaar error = ENOSYS; 595484b5c25SMarcel Moolenaar break; 596484b5c25SMarcel Moolenaar case 'l': 597484b5c25SMarcel Moolenaar if (strcmp(keyword, "link") == 0) { 598484b5c25SMarcel Moolenaar if (value == NULL) { 599484b5c25SMarcel Moolenaar error = ENOATTR; 600484b5c25SMarcel Moolenaar break; 601484b5c25SMarcel Moolenaar } 602484b5c25SMarcel Moolenaar node->symlink = strdup(value); 603484b5c25SMarcel Moolenaar } else 604484b5c25SMarcel Moolenaar error = ENOSYS; 605484b5c25SMarcel Moolenaar break; 606484b5c25SMarcel Moolenaar case 'm': 607484b5c25SMarcel Moolenaar if (strcmp(keyword, "mode") == 0) { 608484b5c25SMarcel Moolenaar if (value == NULL) { 609484b5c25SMarcel Moolenaar error = ENOATTR; 610484b5c25SMarcel Moolenaar break; 611484b5c25SMarcel Moolenaar } 612484b5c25SMarcel Moolenaar if (value[0] >= '0' && value[0] <= '9') { 613484b5c25SMarcel Moolenaar error = read_number(value, 8, &num, 614484b5c25SMarcel Moolenaar 0, 07777); 615484b5c25SMarcel Moolenaar if (!error) { 616484b5c25SMarcel Moolenaar st->st_mode &= S_IFMT; 617484b5c25SMarcel Moolenaar st->st_mode |= num; 618484b5c25SMarcel Moolenaar } 619484b5c25SMarcel Moolenaar } else { 620484b5c25SMarcel Moolenaar /* Symbolic mode not supported. */ 621484b5c25SMarcel Moolenaar error = EINVAL; 622484b5c25SMarcel Moolenaar break; 623484b5c25SMarcel Moolenaar } 624484b5c25SMarcel Moolenaar } else 625484b5c25SMarcel Moolenaar error = ENOSYS; 626484b5c25SMarcel Moolenaar break; 627484b5c25SMarcel Moolenaar case 'o': 628484b5c25SMarcel Moolenaar if (strcmp(keyword, "optional") == 0) { 629484b5c25SMarcel Moolenaar if (value != NULL) 630484b5c25SMarcel Moolenaar error = ENXIO; 631484b5c25SMarcel Moolenaar node->flags |= FSNODE_F_OPTIONAL; 632484b5c25SMarcel Moolenaar } else 633484b5c25SMarcel Moolenaar error = ENOSYS; 634484b5c25SMarcel Moolenaar break; 635484b5c25SMarcel Moolenaar case 's': 636484b5c25SMarcel Moolenaar if (strcmp(keyword, "size") == 0) { 637484b5c25SMarcel Moolenaar if (value == NULL) { 638484b5c25SMarcel Moolenaar error = ENOATTR; 639484b5c25SMarcel Moolenaar break; 640484b5c25SMarcel Moolenaar } 641484b5c25SMarcel Moolenaar error = read_number(value, 10, &num, 642484b5c25SMarcel Moolenaar 0, INTMAX_MAX); 643484b5c25SMarcel Moolenaar if (!error) 644484b5c25SMarcel Moolenaar st->st_size = num; 645484b5c25SMarcel Moolenaar } else 646484b5c25SMarcel Moolenaar error = ENOSYS; 647484b5c25SMarcel Moolenaar break; 648484b5c25SMarcel Moolenaar case 't': 649484b5c25SMarcel Moolenaar if (strcmp(keyword, "time") == 0) { 650484b5c25SMarcel Moolenaar if (value == NULL) { 651484b5c25SMarcel Moolenaar error = ENOATTR; 652484b5c25SMarcel Moolenaar break; 653484b5c25SMarcel Moolenaar } 654484b5c25SMarcel Moolenaar p = strchr(value, '.'); 655484b5c25SMarcel Moolenaar if (p != NULL) 656484b5c25SMarcel Moolenaar *p++ = '\0'; 657484b5c25SMarcel Moolenaar error = read_number(value, 10, &num, 0, 658484b5c25SMarcel Moolenaar INTMAX_MAX); 659484b5c25SMarcel Moolenaar if (error) 660484b5c25SMarcel Moolenaar break; 661484b5c25SMarcel Moolenaar st->st_atime = num; 662484b5c25SMarcel Moolenaar st->st_ctime = num; 663484b5c25SMarcel Moolenaar st->st_mtime = num; 664484b5c25SMarcel Moolenaar error = read_number(p, 10, &num, 0, 665484b5c25SMarcel Moolenaar INTMAX_MAX); 666484b5c25SMarcel Moolenaar if (error) 667484b5c25SMarcel Moolenaar break; 668484b5c25SMarcel Moolenaar if (num != 0) 669484b5c25SMarcel Moolenaar error = EINVAL; 670484b5c25SMarcel Moolenaar } else if (strcmp(keyword, "type") == 0) { 671484b5c25SMarcel Moolenaar if (value == NULL) { 672484b5c25SMarcel Moolenaar error = ENOATTR; 673484b5c25SMarcel Moolenaar break; 674484b5c25SMarcel Moolenaar } 675484b5c25SMarcel Moolenaar if (strcmp(value, "dir") == 0) 676484b5c25SMarcel Moolenaar node->type = S_IFDIR; 677484b5c25SMarcel Moolenaar else if (strcmp(value, "file") == 0) 678484b5c25SMarcel Moolenaar node->type = S_IFREG; 679484b5c25SMarcel Moolenaar else if (strcmp(value, "link") == 0) 680484b5c25SMarcel Moolenaar node->type = S_IFLNK; 681484b5c25SMarcel Moolenaar else 682484b5c25SMarcel Moolenaar error = EINVAL; 683484b5c25SMarcel Moolenaar } else 684484b5c25SMarcel Moolenaar error = ENOSYS; 685484b5c25SMarcel Moolenaar break; 686484b5c25SMarcel Moolenaar case 'u': 687484b5c25SMarcel Moolenaar if (strcmp(keyword, "uid") == 0) { 688484b5c25SMarcel Moolenaar if (value == NULL) { 689484b5c25SMarcel Moolenaar error = ENOATTR; 690484b5c25SMarcel Moolenaar break; 691484b5c25SMarcel Moolenaar } 692484b5c25SMarcel Moolenaar error = read_number(value, 10, &num, 693484b5c25SMarcel Moolenaar 0, UINT_MAX); 694484b5c25SMarcel Moolenaar if (!error) 695484b5c25SMarcel Moolenaar st->st_uid = num; 696484b5c25SMarcel Moolenaar } else if (strcmp(keyword, "uname") == 0) { 697484b5c25SMarcel Moolenaar if (value == NULL) { 698484b5c25SMarcel Moolenaar error = ENOATTR; 699484b5c25SMarcel Moolenaar break; 700484b5c25SMarcel Moolenaar } 701484b5c25SMarcel Moolenaar pwent = getpwnam(value); 702484b5c25SMarcel Moolenaar if (pwent != NULL) 703484b5c25SMarcel Moolenaar st->st_uid = pwent->pw_uid; 704484b5c25SMarcel Moolenaar else 705484b5c25SMarcel Moolenaar error = errno; 706484b5c25SMarcel Moolenaar } else 707484b5c25SMarcel Moolenaar error = ENOSYS; 708484b5c25SMarcel Moolenaar break; 709484b5c25SMarcel Moolenaar default: 710484b5c25SMarcel Moolenaar error = ENOSYS; 711484b5c25SMarcel Moolenaar break; 712484b5c25SMarcel Moolenaar } 713484b5c25SMarcel Moolenaar 714484b5c25SMarcel Moolenaar switch (error) { 715484b5c25SMarcel Moolenaar case EINVAL: 716484b5c25SMarcel Moolenaar mtree_error("%s: invalid value '%s'", keyword, value); 717484b5c25SMarcel Moolenaar break; 718484b5c25SMarcel Moolenaar case ENOATTR: 719484b5c25SMarcel Moolenaar mtree_error("%s: keyword needs a value", keyword); 720484b5c25SMarcel Moolenaar break; 721484b5c25SMarcel Moolenaar case ENOSYS: 722484b5c25SMarcel Moolenaar mtree_warning("%s: unsupported keyword", keyword); 723484b5c25SMarcel Moolenaar break; 724484b5c25SMarcel Moolenaar case ENXIO: 725484b5c25SMarcel Moolenaar mtree_error("%s: keyword does not take a value", 726484b5c25SMarcel Moolenaar keyword); 727484b5c25SMarcel Moolenaar break; 728484b5c25SMarcel Moolenaar } 729484b5c25SMarcel Moolenaar } while (1); 730484b5c25SMarcel Moolenaar 731484b5c25SMarcel Moolenaar if (error) 732484b5c25SMarcel Moolenaar return (error); 733484b5c25SMarcel Moolenaar 734484b5c25SMarcel Moolenaar st->st_mode = (st->st_mode & ~S_IFMT) | node->type; 735484b5c25SMarcel Moolenaar 736484b5c25SMarcel Moolenaar /* Nothing more to do for the global defaults. */ 737484b5c25SMarcel Moolenaar if (node->name == NULL) 738484b5c25SMarcel Moolenaar return (0); 739484b5c25SMarcel Moolenaar 740484b5c25SMarcel Moolenaar /* 741484b5c25SMarcel Moolenaar * Be intelligent about the file type. 742484b5c25SMarcel Moolenaar */ 743484b5c25SMarcel Moolenaar if (node->contents != NULL) { 744484b5c25SMarcel Moolenaar if (node->symlink != NULL) { 745484b5c25SMarcel Moolenaar mtree_error("%s: both link and contents keywords " 746484b5c25SMarcel Moolenaar "defined", node->name); 747484b5c25SMarcel Moolenaar return (0); 748484b5c25SMarcel Moolenaar } 749484b5c25SMarcel Moolenaar type = S_IFREG; 750*31c0b21fSSimon J. Gerraty } else if (node->type != 0) { 751*31c0b21fSSimon J. Gerraty type = node->type; 752*31c0b21fSSimon J. Gerraty if (type == S_IFREG) { 753*31c0b21fSSimon J. Gerraty /* the named path is the default contents */ 754*31c0b21fSSimon J. Gerraty node->contents = mtree_file_path(node); 755*31c0b21fSSimon J. Gerraty } 756484b5c25SMarcel Moolenaar } else 757484b5c25SMarcel Moolenaar type = (node->symlink != NULL) ? S_IFLNK : S_IFDIR; 758484b5c25SMarcel Moolenaar 759484b5c25SMarcel Moolenaar if (node->type == 0) 760484b5c25SMarcel Moolenaar node->type = type; 761484b5c25SMarcel Moolenaar 762484b5c25SMarcel Moolenaar if (node->type != type) { 763484b5c25SMarcel Moolenaar mtree_error("%s: file type and defined keywords to not match", 764484b5c25SMarcel Moolenaar node->name); 765484b5c25SMarcel Moolenaar return (0); 766484b5c25SMarcel Moolenaar } 767484b5c25SMarcel Moolenaar 768484b5c25SMarcel Moolenaar st->st_mode = (st->st_mode & ~S_IFMT) | node->type; 769484b5c25SMarcel Moolenaar 770484b5c25SMarcel Moolenaar if (node->contents == NULL) 771484b5c25SMarcel Moolenaar return (0); 772484b5c25SMarcel Moolenaar 773484b5c25SMarcel Moolenaar name = mtree_resolve(node->contents, &istemp); 774484b5c25SMarcel Moolenaar if (name == NULL) 775484b5c25SMarcel Moolenaar return (errno); 776484b5c25SMarcel Moolenaar 777484b5c25SMarcel Moolenaar if (stat(name, &sb) != 0) { 778484b5c25SMarcel Moolenaar mtree_error("%s: contents file '%s' not found", node->name, 779484b5c25SMarcel Moolenaar name); 780484b5c25SMarcel Moolenaar free(name); 781484b5c25SMarcel Moolenaar return (0); 782484b5c25SMarcel Moolenaar } 783484b5c25SMarcel Moolenaar 784484b5c25SMarcel Moolenaar free(node->contents); 785484b5c25SMarcel Moolenaar node->contents = name; 786484b5c25SMarcel Moolenaar st->st_size = sb.st_size; 787484b5c25SMarcel Moolenaar return (0); 788484b5c25SMarcel Moolenaar } 789484b5c25SMarcel Moolenaar 790484b5c25SMarcel Moolenaar static int 791484b5c25SMarcel Moolenaar read_mtree_command(FILE *fp) 792484b5c25SMarcel Moolenaar { 793484b5c25SMarcel Moolenaar char cmd[10]; 794484b5c25SMarcel Moolenaar int error; 795484b5c25SMarcel Moolenaar 796484b5c25SMarcel Moolenaar error = read_word(fp, cmd, sizeof(cmd)); 797484b5c25SMarcel Moolenaar if (error) 798484b5c25SMarcel Moolenaar goto out; 799484b5c25SMarcel Moolenaar 800484b5c25SMarcel Moolenaar error = read_mtree_keywords(fp, &mtree_global); 801484b5c25SMarcel Moolenaar 802484b5c25SMarcel Moolenaar out: 803484b5c25SMarcel Moolenaar skip_to(fp, "\n"); 804484b5c25SMarcel Moolenaar (void)getc(fp); 805484b5c25SMarcel Moolenaar return (error); 806484b5c25SMarcel Moolenaar } 807484b5c25SMarcel Moolenaar 808484b5c25SMarcel Moolenaar static int 809484b5c25SMarcel Moolenaar read_mtree_spec1(FILE *fp, bool def, const char *name) 810484b5c25SMarcel Moolenaar { 811484b5c25SMarcel Moolenaar fsnode *last, *node, *parent; 812484b5c25SMarcel Moolenaar u_int type; 813484b5c25SMarcel Moolenaar int error; 814484b5c25SMarcel Moolenaar 815484b5c25SMarcel Moolenaar assert(name[0] != '\0'); 816484b5c25SMarcel Moolenaar 817484b5c25SMarcel Moolenaar /* 818484b5c25SMarcel Moolenaar * Treat '..' specially, because it only changes our current 819484b5c25SMarcel Moolenaar * directory. We don't create a node for it. We simply ignore 820484b5c25SMarcel Moolenaar * any keywords that may appear on the line as well. 821484b5c25SMarcel Moolenaar * Going up a directory is a little non-obvious. A directory 822484b5c25SMarcel Moolenaar * node has a corresponding '.' child. The parent of '.' is 823484b5c25SMarcel Moolenaar * not the '.' node of the parent directory, but the directory 824484b5c25SMarcel Moolenaar * node within the parent to which the child relates. However, 825484b5c25SMarcel Moolenaar * going up a directory means we need to find the '.' node to 826484b5c25SMarcel Moolenaar * which the directoy node is linked. This we can do via the 827484b5c25SMarcel Moolenaar * first * pointer, because '.' is always the first entry in a 828484b5c25SMarcel Moolenaar * directory. 829484b5c25SMarcel Moolenaar */ 830484b5c25SMarcel Moolenaar if (IS_DOTDOT(name)) { 831484b5c25SMarcel Moolenaar /* This deals with NULL pointers as well. */ 832484b5c25SMarcel Moolenaar if (mtree_current == mtree_root) { 833484b5c25SMarcel Moolenaar mtree_warning("ignoring .. in root directory"); 834484b5c25SMarcel Moolenaar return (0); 835484b5c25SMarcel Moolenaar } 836484b5c25SMarcel Moolenaar 837484b5c25SMarcel Moolenaar node = mtree_current; 838484b5c25SMarcel Moolenaar 839484b5c25SMarcel Moolenaar assert(node != NULL); 840484b5c25SMarcel Moolenaar assert(IS_DOT(node->name)); 841484b5c25SMarcel Moolenaar assert(node->first == node); 842484b5c25SMarcel Moolenaar 843484b5c25SMarcel Moolenaar /* Get the corresponding directory node in the parent. */ 844484b5c25SMarcel Moolenaar node = mtree_current->parent; 845484b5c25SMarcel Moolenaar 846484b5c25SMarcel Moolenaar assert(node != NULL); 847484b5c25SMarcel Moolenaar assert(!IS_DOT(node->name)); 848484b5c25SMarcel Moolenaar 849484b5c25SMarcel Moolenaar node = node->first; 850484b5c25SMarcel Moolenaar 851484b5c25SMarcel Moolenaar assert(node != NULL); 852484b5c25SMarcel Moolenaar assert(IS_DOT(node->name)); 853484b5c25SMarcel Moolenaar assert(node->first == node); 854484b5c25SMarcel Moolenaar 855484b5c25SMarcel Moolenaar mtree_current = node; 856484b5c25SMarcel Moolenaar return (0); 857484b5c25SMarcel Moolenaar } 858484b5c25SMarcel Moolenaar 859484b5c25SMarcel Moolenaar /* 860484b5c25SMarcel Moolenaar * If we don't have a current directory and the first specification 861484b5c25SMarcel Moolenaar * (either implicit or defined) is not '.', then we need to create 862484b5c25SMarcel Moolenaar * a '.' node first (using a recursive call). 863484b5c25SMarcel Moolenaar */ 864484b5c25SMarcel Moolenaar if (!IS_DOT(name) && mtree_current == NULL) { 865484b5c25SMarcel Moolenaar error = read_mtree_spec1(fp, false, "."); 866484b5c25SMarcel Moolenaar if (error) 867484b5c25SMarcel Moolenaar return (error); 868484b5c25SMarcel Moolenaar } 869484b5c25SMarcel Moolenaar 870484b5c25SMarcel Moolenaar /* 871484b5c25SMarcel Moolenaar * Lookup the name in the current directory (if we have a current 872484b5c25SMarcel Moolenaar * directory) to make sure we do not create multiple nodes for the 873484b5c25SMarcel Moolenaar * same component. For non-definitions, if we find a node with the 874484b5c25SMarcel Moolenaar * same name, simply change the current directory. For definitions 875484b5c25SMarcel Moolenaar * more happens. 876484b5c25SMarcel Moolenaar */ 877484b5c25SMarcel Moolenaar last = NULL; 878484b5c25SMarcel Moolenaar node = mtree_current; 879484b5c25SMarcel Moolenaar while (node != NULL) { 880484b5c25SMarcel Moolenaar assert(node->first == mtree_current); 881484b5c25SMarcel Moolenaar 882484b5c25SMarcel Moolenaar if (strcmp(name, node->name) == 0) { 883484b5c25SMarcel Moolenaar if (def == true) { 884484b5c25SMarcel Moolenaar mtree_error("duplicate definition of %s", 885484b5c25SMarcel Moolenaar name); 886484b5c25SMarcel Moolenaar return (0); 887484b5c25SMarcel Moolenaar } 888484b5c25SMarcel Moolenaar 889484b5c25SMarcel Moolenaar if (node->type != S_IFDIR) { 890484b5c25SMarcel Moolenaar mtree_error("%s is not a directory", name); 891484b5c25SMarcel Moolenaar return (0); 892484b5c25SMarcel Moolenaar } 893484b5c25SMarcel Moolenaar 894484b5c25SMarcel Moolenaar assert(!IS_DOT(name)); 895484b5c25SMarcel Moolenaar 896484b5c25SMarcel Moolenaar node = node->child; 897484b5c25SMarcel Moolenaar 898484b5c25SMarcel Moolenaar assert(node != NULL); 899484b5c25SMarcel Moolenaar assert(IS_DOT(node->name)); 900484b5c25SMarcel Moolenaar 901484b5c25SMarcel Moolenaar mtree_current = node; 902484b5c25SMarcel Moolenaar return (0); 903484b5c25SMarcel Moolenaar } 904484b5c25SMarcel Moolenaar 905484b5c25SMarcel Moolenaar last = node; 906484b5c25SMarcel Moolenaar node = last->next; 907484b5c25SMarcel Moolenaar } 908484b5c25SMarcel Moolenaar 909484b5c25SMarcel Moolenaar parent = (mtree_current != NULL) ? mtree_current->parent : NULL; 910484b5c25SMarcel Moolenaar type = (def == false || IS_DOT(name)) ? S_IFDIR : 0; 911484b5c25SMarcel Moolenaar node = create_node(name, type, parent, &mtree_global); 912484b5c25SMarcel Moolenaar if (node == NULL) 913484b5c25SMarcel Moolenaar return (ENOMEM); 914484b5c25SMarcel Moolenaar 915484b5c25SMarcel Moolenaar if (def == true) { 916484b5c25SMarcel Moolenaar error = read_mtree_keywords(fp, node); 917484b5c25SMarcel Moolenaar if (error) { 918484b5c25SMarcel Moolenaar destroy_node(node); 919484b5c25SMarcel Moolenaar return (error); 920484b5c25SMarcel Moolenaar } 921484b5c25SMarcel Moolenaar } 922484b5c25SMarcel Moolenaar 923484b5c25SMarcel Moolenaar node->first = (mtree_current != NULL) ? mtree_current : node; 924484b5c25SMarcel Moolenaar 925484b5c25SMarcel Moolenaar if (last != NULL) 926484b5c25SMarcel Moolenaar last->next = node; 927484b5c25SMarcel Moolenaar 928484b5c25SMarcel Moolenaar if (node->type != S_IFDIR) 929484b5c25SMarcel Moolenaar return (0); 930484b5c25SMarcel Moolenaar 931484b5c25SMarcel Moolenaar if (!IS_DOT(node->name)) { 932484b5c25SMarcel Moolenaar parent = node; 933484b5c25SMarcel Moolenaar node = create_node(".", S_IFDIR, parent, parent); 934484b5c25SMarcel Moolenaar if (node == NULL) { 935484b5c25SMarcel Moolenaar last->next = NULL; 936484b5c25SMarcel Moolenaar destroy_node(parent); 937484b5c25SMarcel Moolenaar return (ENOMEM); 938484b5c25SMarcel Moolenaar } 939484b5c25SMarcel Moolenaar parent->child = node; 940484b5c25SMarcel Moolenaar node->first = node; 941484b5c25SMarcel Moolenaar } 942484b5c25SMarcel Moolenaar 943484b5c25SMarcel Moolenaar assert(node != NULL); 944484b5c25SMarcel Moolenaar assert(IS_DOT(node->name)); 945484b5c25SMarcel Moolenaar assert(node->first == node); 946484b5c25SMarcel Moolenaar 947484b5c25SMarcel Moolenaar mtree_current = node; 948484b5c25SMarcel Moolenaar if (mtree_root == NULL) 949484b5c25SMarcel Moolenaar mtree_root = node; 950484b5c25SMarcel Moolenaar 951484b5c25SMarcel Moolenaar return (0); 952484b5c25SMarcel Moolenaar } 953484b5c25SMarcel Moolenaar 954484b5c25SMarcel Moolenaar static int 955484b5c25SMarcel Moolenaar read_mtree_spec(FILE *fp) 956484b5c25SMarcel Moolenaar { 957484b5c25SMarcel Moolenaar char pathspec[PATH_MAX]; 958484b5c25SMarcel Moolenaar char *cp; 959484b5c25SMarcel Moolenaar int error; 960484b5c25SMarcel Moolenaar 961484b5c25SMarcel Moolenaar error = read_word(fp, pathspec, sizeof(pathspec)); 962484b5c25SMarcel Moolenaar if (error) 963484b5c25SMarcel Moolenaar goto out; 964484b5c25SMarcel Moolenaar 965484b5c25SMarcel Moolenaar cp = strchr(pathspec, '/'); 966484b5c25SMarcel Moolenaar if (cp != NULL) { 967484b5c25SMarcel Moolenaar /* Absolute pathname */ 968484b5c25SMarcel Moolenaar mtree_current = mtree_root; 969484b5c25SMarcel Moolenaar 970484b5c25SMarcel Moolenaar do { 971484b5c25SMarcel Moolenaar *cp++ = '\0'; 972484b5c25SMarcel Moolenaar 973484b5c25SMarcel Moolenaar /* Disallow '.' and '..' as components. */ 974484b5c25SMarcel Moolenaar if (IS_DOT(pathspec) || IS_DOTDOT(pathspec)) { 975484b5c25SMarcel Moolenaar mtree_error("absolute path cannot contain . " 976484b5c25SMarcel Moolenaar "or .. components"); 977484b5c25SMarcel Moolenaar goto out; 978484b5c25SMarcel Moolenaar } 979484b5c25SMarcel Moolenaar 980484b5c25SMarcel Moolenaar /* Ignore multiple adjacent slashes. */ 981484b5c25SMarcel Moolenaar if (pathspec[0] != '\0') 982484b5c25SMarcel Moolenaar error = read_mtree_spec1(fp, false, pathspec); 983484b5c25SMarcel Moolenaar memmove(pathspec, cp, strlen(cp) + 1); 984484b5c25SMarcel Moolenaar cp = strchr(pathspec, '/'); 985484b5c25SMarcel Moolenaar } while (!error && cp != NULL); 986484b5c25SMarcel Moolenaar 987484b5c25SMarcel Moolenaar /* Disallow '.' and '..' as the last component. */ 988484b5c25SMarcel Moolenaar if (!error && (IS_DOT(pathspec) || IS_DOTDOT(pathspec))) { 989484b5c25SMarcel Moolenaar mtree_error("absolute path cannot contain . or .. " 990484b5c25SMarcel Moolenaar "components"); 991484b5c25SMarcel Moolenaar goto out; 992484b5c25SMarcel Moolenaar } 993484b5c25SMarcel Moolenaar } 994484b5c25SMarcel Moolenaar 995484b5c25SMarcel Moolenaar /* Ignore absolute specfications that end with a slash. */ 996484b5c25SMarcel Moolenaar if (!error && pathspec[0] != '\0') 997484b5c25SMarcel Moolenaar error = read_mtree_spec1(fp, true, pathspec); 998484b5c25SMarcel Moolenaar 999484b5c25SMarcel Moolenaar out: 1000484b5c25SMarcel Moolenaar skip_to(fp, "\n"); 1001484b5c25SMarcel Moolenaar (void)getc(fp); 1002484b5c25SMarcel Moolenaar return (error); 1003484b5c25SMarcel Moolenaar } 1004484b5c25SMarcel Moolenaar 1005484b5c25SMarcel Moolenaar fsnode * 1006484b5c25SMarcel Moolenaar read_mtree(const char *fname, fsnode *node) 1007484b5c25SMarcel Moolenaar { 1008484b5c25SMarcel Moolenaar struct mtree_fileinfo *fi; 1009484b5c25SMarcel Moolenaar FILE *fp; 1010484b5c25SMarcel Moolenaar int c, error; 1011484b5c25SMarcel Moolenaar 1012484b5c25SMarcel Moolenaar /* We do not yet support nesting... */ 1013484b5c25SMarcel Moolenaar assert(node == NULL); 1014484b5c25SMarcel Moolenaar 1015484b5c25SMarcel Moolenaar if (strcmp(fname, "-") == 0) 1016484b5c25SMarcel Moolenaar fp = stdin; 1017484b5c25SMarcel Moolenaar else { 1018484b5c25SMarcel Moolenaar fp = fopen(fname, "r"); 1019484b5c25SMarcel Moolenaar if (fp == NULL) 1020484b5c25SMarcel Moolenaar err(1, "Can't open `%s'", fname); 1021484b5c25SMarcel Moolenaar } 1022484b5c25SMarcel Moolenaar 1023484b5c25SMarcel Moolenaar error = mtree_file_push(fname, fp); 1024484b5c25SMarcel Moolenaar if (error) 1025484b5c25SMarcel Moolenaar goto out; 1026484b5c25SMarcel Moolenaar 1027484b5c25SMarcel Moolenaar bzero(&mtree_global, sizeof(mtree_global)); 1028484b5c25SMarcel Moolenaar bzero(&mtree_global_inode, sizeof(mtree_global_inode)); 1029484b5c25SMarcel Moolenaar mtree_global.inode = &mtree_global_inode; 1030484b5c25SMarcel Moolenaar mtree_global_inode.nlink = 1; 1031484b5c25SMarcel Moolenaar mtree_global_inode.st.st_atime = mtree_global_inode.st.st_ctime = 1032484b5c25SMarcel Moolenaar mtree_global_inode.st.st_mtime = time(NULL); 1033484b5c25SMarcel Moolenaar errors = warnings = 0; 1034484b5c25SMarcel Moolenaar 1035484b5c25SMarcel Moolenaar setgroupent(1); 1036484b5c25SMarcel Moolenaar setpassent(1); 1037484b5c25SMarcel Moolenaar 1038484b5c25SMarcel Moolenaar mtree_root = node; 1039484b5c25SMarcel Moolenaar mtree_current = node; 1040484b5c25SMarcel Moolenaar do { 1041484b5c25SMarcel Moolenaar /* Start of a new line... */ 1042484b5c25SMarcel Moolenaar fi = SLIST_FIRST(&mtree_fileinfo); 1043484b5c25SMarcel Moolenaar fi->line++; 1044484b5c25SMarcel Moolenaar 1045484b5c25SMarcel Moolenaar error = skip_over(fp, " \t"); 1046484b5c25SMarcel Moolenaar if (error) 1047484b5c25SMarcel Moolenaar break; 1048484b5c25SMarcel Moolenaar 1049484b5c25SMarcel Moolenaar c = getc(fp); 1050484b5c25SMarcel Moolenaar if (c == EOF) { 1051484b5c25SMarcel Moolenaar error = ferror(fp) ? errno : -1; 1052484b5c25SMarcel Moolenaar break; 1053484b5c25SMarcel Moolenaar } 1054484b5c25SMarcel Moolenaar 1055484b5c25SMarcel Moolenaar switch (c) { 1056484b5c25SMarcel Moolenaar case '\n': /* empty line */ 1057484b5c25SMarcel Moolenaar error = 0; 1058484b5c25SMarcel Moolenaar break; 1059484b5c25SMarcel Moolenaar case '#': /* comment -- skip to end of line. */ 1060484b5c25SMarcel Moolenaar error = skip_to(fp, "\n"); 1061484b5c25SMarcel Moolenaar if (!error) 1062484b5c25SMarcel Moolenaar (void)getc(fp); 1063484b5c25SMarcel Moolenaar break; 1064484b5c25SMarcel Moolenaar case '/': /* special commands */ 1065484b5c25SMarcel Moolenaar error = read_mtree_command(fp); 1066484b5c25SMarcel Moolenaar break; 1067484b5c25SMarcel Moolenaar default: /* specification */ 1068484b5c25SMarcel Moolenaar ungetc(c, fp); 1069484b5c25SMarcel Moolenaar error = read_mtree_spec(fp); 1070484b5c25SMarcel Moolenaar break; 1071484b5c25SMarcel Moolenaar } 1072484b5c25SMarcel Moolenaar } while (!error); 1073484b5c25SMarcel Moolenaar 1074484b5c25SMarcel Moolenaar endpwent(); 1075484b5c25SMarcel Moolenaar endgrent(); 1076484b5c25SMarcel Moolenaar 1077484b5c25SMarcel Moolenaar if (error <= 0 && (errors || warnings)) { 1078484b5c25SMarcel Moolenaar warnx("%u error(s) and %u warning(s) in mtree manifest", 1079484b5c25SMarcel Moolenaar errors, warnings); 1080484b5c25SMarcel Moolenaar if (errors) 1081484b5c25SMarcel Moolenaar exit(1); 1082484b5c25SMarcel Moolenaar } 1083484b5c25SMarcel Moolenaar 1084484b5c25SMarcel Moolenaar out: 1085484b5c25SMarcel Moolenaar if (error > 0) 1086484b5c25SMarcel Moolenaar errc(1, error, "Error reading mtree file"); 1087484b5c25SMarcel Moolenaar 1088484b5c25SMarcel Moolenaar if (fp != stdin) 1089484b5c25SMarcel Moolenaar fclose(fp); 1090484b5c25SMarcel Moolenaar 1091484b5c25SMarcel Moolenaar if (mtree_root != NULL) 1092484b5c25SMarcel Moolenaar return (mtree_root); 1093484b5c25SMarcel Moolenaar 1094484b5c25SMarcel Moolenaar /* Handle empty specifications. */ 1095484b5c25SMarcel Moolenaar node = create_node(".", S_IFDIR, NULL, &mtree_global); 1096484b5c25SMarcel Moolenaar node->first = node; 1097484b5c25SMarcel Moolenaar return (node); 1098484b5c25SMarcel Moolenaar } 1099