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 13831c0b21fSSimon J. Gerraty #ifndef MAKEFS_MAX_TREE_DEPTH 13931c0b21fSSimon J. Gerraty # define MAKEFS_MAX_TREE_DEPTH (MAXPATHLEN/2) 14031c0b21fSSimon J. Gerraty #endif 14131c0b21fSSimon J. Gerraty 14231c0b21fSSimon J. Gerraty /* construct path to node->name */ 14331c0b21fSSimon J. Gerraty static char * 14431c0b21fSSimon J. Gerraty mtree_file_path(fsnode *node) 14531c0b21fSSimon J. Gerraty { 14631c0b21fSSimon J. Gerraty fsnode *pnode; 14731c0b21fSSimon J. Gerraty struct sbuf *sb; 14831c0b21fSSimon J. Gerraty char *res, *rp[MAKEFS_MAX_TREE_DEPTH]; 14931c0b21fSSimon J. Gerraty int depth; 15031c0b21fSSimon J. Gerraty 15131c0b21fSSimon J. Gerraty depth = 0; 15231c0b21fSSimon J. Gerraty rp[depth] = node->name; 15331c0b21fSSimon J. Gerraty for (pnode = node->parent; pnode && depth < MAKEFS_MAX_TREE_DEPTH; 15431c0b21fSSimon J. Gerraty pnode = pnode->parent) { 15531c0b21fSSimon J. Gerraty if (strcmp(pnode->name, ".") == 0) 15631c0b21fSSimon J. Gerraty break; 15731c0b21fSSimon J. Gerraty rp[++depth] = pnode->name; 15831c0b21fSSimon J. Gerraty } 15931c0b21fSSimon J. Gerraty 16031c0b21fSSimon J. Gerraty sb = sbuf_new(NULL, NULL, 0, SBUF_AUTOEXTEND); 16131c0b21fSSimon J. Gerraty if (sb == NULL) { 16231c0b21fSSimon J. Gerraty errno = ENOMEM; 16331c0b21fSSimon J. Gerraty return (NULL); 16431c0b21fSSimon J. Gerraty } 16531c0b21fSSimon J. Gerraty while (depth > 0) { 16631c0b21fSSimon J. Gerraty sbuf_cat(sb, rp[depth--]); 16731c0b21fSSimon J. Gerraty sbuf_putc(sb, '/'); 16831c0b21fSSimon J. Gerraty } 16931c0b21fSSimon J. Gerraty sbuf_cat(sb, rp[depth]); 17031c0b21fSSimon J. Gerraty sbuf_finish(sb); 17131c0b21fSSimon J. Gerraty res = strdup(sbuf_data(sb)); 17231c0b21fSSimon J. Gerraty sbuf_delete(sb); 17331c0b21fSSimon J. Gerraty if (res == NULL) 17431c0b21fSSimon J. Gerraty errno = ENOMEM; 17531c0b21fSSimon J. Gerraty return res; 17631c0b21fSSimon J. Gerraty 17731c0b21fSSimon J. Gerraty } 17831c0b21fSSimon 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; 511e9a03238SBrooks Davis gid_t gid; 512e9a03238SBrooks Davis uid_t uid; 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 } 588e9a03238SBrooks Davis if (gid_from_group(value, &gid) == 0) 589e9a03238SBrooks Davis st->st_gid = gid; 590484b5c25SMarcel Moolenaar else 591e9a03238SBrooks Davis error = EINVAL; 592484b5c25SMarcel Moolenaar } else 593484b5c25SMarcel Moolenaar error = ENOSYS; 594484b5c25SMarcel Moolenaar break; 595484b5c25SMarcel Moolenaar case 'l': 596484b5c25SMarcel Moolenaar if (strcmp(keyword, "link") == 0) { 597484b5c25SMarcel Moolenaar if (value == NULL) { 598484b5c25SMarcel Moolenaar error = ENOATTR; 599484b5c25SMarcel Moolenaar break; 600484b5c25SMarcel Moolenaar } 601484b5c25SMarcel Moolenaar node->symlink = strdup(value); 602484b5c25SMarcel Moolenaar } else 603484b5c25SMarcel Moolenaar error = ENOSYS; 604484b5c25SMarcel Moolenaar break; 605484b5c25SMarcel Moolenaar case 'm': 606484b5c25SMarcel Moolenaar if (strcmp(keyword, "mode") == 0) { 607484b5c25SMarcel Moolenaar if (value == NULL) { 608484b5c25SMarcel Moolenaar error = ENOATTR; 609484b5c25SMarcel Moolenaar break; 610484b5c25SMarcel Moolenaar } 611484b5c25SMarcel Moolenaar if (value[0] >= '0' && value[0] <= '9') { 612484b5c25SMarcel Moolenaar error = read_number(value, 8, &num, 613484b5c25SMarcel Moolenaar 0, 07777); 614484b5c25SMarcel Moolenaar if (!error) { 615484b5c25SMarcel Moolenaar st->st_mode &= S_IFMT; 616484b5c25SMarcel Moolenaar st->st_mode |= num; 617484b5c25SMarcel Moolenaar } 618484b5c25SMarcel Moolenaar } else { 619484b5c25SMarcel Moolenaar /* Symbolic mode not supported. */ 620484b5c25SMarcel Moolenaar error = EINVAL; 621484b5c25SMarcel Moolenaar break; 622484b5c25SMarcel Moolenaar } 623484b5c25SMarcel Moolenaar } else 624484b5c25SMarcel Moolenaar error = ENOSYS; 625484b5c25SMarcel Moolenaar break; 626484b5c25SMarcel Moolenaar case 'o': 627484b5c25SMarcel Moolenaar if (strcmp(keyword, "optional") == 0) { 628484b5c25SMarcel Moolenaar if (value != NULL) 629484b5c25SMarcel Moolenaar error = ENXIO; 630484b5c25SMarcel Moolenaar node->flags |= FSNODE_F_OPTIONAL; 631484b5c25SMarcel Moolenaar } else 632484b5c25SMarcel Moolenaar error = ENOSYS; 633484b5c25SMarcel Moolenaar break; 634484b5c25SMarcel Moolenaar case 's': 635484b5c25SMarcel Moolenaar if (strcmp(keyword, "size") == 0) { 636484b5c25SMarcel Moolenaar if (value == NULL) { 637484b5c25SMarcel Moolenaar error = ENOATTR; 638484b5c25SMarcel Moolenaar break; 639484b5c25SMarcel Moolenaar } 640484b5c25SMarcel Moolenaar error = read_number(value, 10, &num, 641484b5c25SMarcel Moolenaar 0, INTMAX_MAX); 642484b5c25SMarcel Moolenaar if (!error) 643484b5c25SMarcel Moolenaar st->st_size = num; 644484b5c25SMarcel Moolenaar } else 645484b5c25SMarcel Moolenaar error = ENOSYS; 646484b5c25SMarcel Moolenaar break; 647484b5c25SMarcel Moolenaar case 't': 648484b5c25SMarcel Moolenaar if (strcmp(keyword, "time") == 0) { 649484b5c25SMarcel Moolenaar if (value == NULL) { 650484b5c25SMarcel Moolenaar error = ENOATTR; 651484b5c25SMarcel Moolenaar break; 652484b5c25SMarcel Moolenaar } 653484b5c25SMarcel Moolenaar p = strchr(value, '.'); 654484b5c25SMarcel Moolenaar if (p != NULL) 655484b5c25SMarcel Moolenaar *p++ = '\0'; 656484b5c25SMarcel Moolenaar error = read_number(value, 10, &num, 0, 657484b5c25SMarcel Moolenaar INTMAX_MAX); 658484b5c25SMarcel Moolenaar if (error) 659484b5c25SMarcel Moolenaar break; 660484b5c25SMarcel Moolenaar st->st_atime = num; 661484b5c25SMarcel Moolenaar st->st_ctime = num; 662484b5c25SMarcel Moolenaar st->st_mtime = num; 663484b5c25SMarcel Moolenaar error = read_number(p, 10, &num, 0, 664484b5c25SMarcel Moolenaar INTMAX_MAX); 665484b5c25SMarcel Moolenaar if (error) 666484b5c25SMarcel Moolenaar break; 667484b5c25SMarcel Moolenaar if (num != 0) 668484b5c25SMarcel Moolenaar error = EINVAL; 669484b5c25SMarcel Moolenaar } else if (strcmp(keyword, "type") == 0) { 670484b5c25SMarcel Moolenaar if (value == NULL) { 671484b5c25SMarcel Moolenaar error = ENOATTR; 672484b5c25SMarcel Moolenaar break; 673484b5c25SMarcel Moolenaar } 674484b5c25SMarcel Moolenaar if (strcmp(value, "dir") == 0) 675484b5c25SMarcel Moolenaar node->type = S_IFDIR; 676484b5c25SMarcel Moolenaar else if (strcmp(value, "file") == 0) 677484b5c25SMarcel Moolenaar node->type = S_IFREG; 678484b5c25SMarcel Moolenaar else if (strcmp(value, "link") == 0) 679484b5c25SMarcel Moolenaar node->type = S_IFLNK; 680484b5c25SMarcel Moolenaar else 681484b5c25SMarcel Moolenaar error = EINVAL; 682484b5c25SMarcel Moolenaar } else 683484b5c25SMarcel Moolenaar error = ENOSYS; 684484b5c25SMarcel Moolenaar break; 685484b5c25SMarcel Moolenaar case 'u': 686484b5c25SMarcel Moolenaar if (strcmp(keyword, "uid") == 0) { 687484b5c25SMarcel Moolenaar if (value == NULL) { 688484b5c25SMarcel Moolenaar error = ENOATTR; 689484b5c25SMarcel Moolenaar break; 690484b5c25SMarcel Moolenaar } 691484b5c25SMarcel Moolenaar error = read_number(value, 10, &num, 692484b5c25SMarcel Moolenaar 0, UINT_MAX); 693484b5c25SMarcel Moolenaar if (!error) 694484b5c25SMarcel Moolenaar st->st_uid = num; 695484b5c25SMarcel Moolenaar } else if (strcmp(keyword, "uname") == 0) { 696484b5c25SMarcel Moolenaar if (value == NULL) { 697484b5c25SMarcel Moolenaar error = ENOATTR; 698484b5c25SMarcel Moolenaar break; 699484b5c25SMarcel Moolenaar } 700e9a03238SBrooks Davis if (uid_from_user(value, &uid) == 0) 701e9a03238SBrooks Davis st->st_uid = uid; 702484b5c25SMarcel Moolenaar else 703e9a03238SBrooks Davis error = EINVAL; 704484b5c25SMarcel Moolenaar } else 705484b5c25SMarcel Moolenaar error = ENOSYS; 706484b5c25SMarcel Moolenaar break; 707484b5c25SMarcel Moolenaar default: 708484b5c25SMarcel Moolenaar error = ENOSYS; 709484b5c25SMarcel Moolenaar break; 710484b5c25SMarcel Moolenaar } 711484b5c25SMarcel Moolenaar 712484b5c25SMarcel Moolenaar switch (error) { 713484b5c25SMarcel Moolenaar case EINVAL: 714484b5c25SMarcel Moolenaar mtree_error("%s: invalid value '%s'", keyword, value); 715484b5c25SMarcel Moolenaar break; 716484b5c25SMarcel Moolenaar case ENOATTR: 717484b5c25SMarcel Moolenaar mtree_error("%s: keyword needs a value", keyword); 718484b5c25SMarcel Moolenaar break; 719484b5c25SMarcel Moolenaar case ENOSYS: 720484b5c25SMarcel Moolenaar mtree_warning("%s: unsupported keyword", keyword); 721484b5c25SMarcel Moolenaar break; 722484b5c25SMarcel Moolenaar case ENXIO: 723484b5c25SMarcel Moolenaar mtree_error("%s: keyword does not take a value", 724484b5c25SMarcel Moolenaar keyword); 725484b5c25SMarcel Moolenaar break; 726484b5c25SMarcel Moolenaar } 727484b5c25SMarcel Moolenaar } while (1); 728484b5c25SMarcel Moolenaar 729484b5c25SMarcel Moolenaar if (error) 730484b5c25SMarcel Moolenaar return (error); 731484b5c25SMarcel Moolenaar 732484b5c25SMarcel Moolenaar st->st_mode = (st->st_mode & ~S_IFMT) | node->type; 733484b5c25SMarcel Moolenaar 734484b5c25SMarcel Moolenaar /* Nothing more to do for the global defaults. */ 735484b5c25SMarcel Moolenaar if (node->name == NULL) 736484b5c25SMarcel Moolenaar return (0); 737484b5c25SMarcel Moolenaar 738484b5c25SMarcel Moolenaar /* 739484b5c25SMarcel Moolenaar * Be intelligent about the file type. 740484b5c25SMarcel Moolenaar */ 741484b5c25SMarcel Moolenaar if (node->contents != NULL) { 742484b5c25SMarcel Moolenaar if (node->symlink != NULL) { 743484b5c25SMarcel Moolenaar mtree_error("%s: both link and contents keywords " 744484b5c25SMarcel Moolenaar "defined", node->name); 745484b5c25SMarcel Moolenaar return (0); 746484b5c25SMarcel Moolenaar } 747484b5c25SMarcel Moolenaar type = S_IFREG; 74831c0b21fSSimon J. Gerraty } else if (node->type != 0) { 74931c0b21fSSimon J. Gerraty type = node->type; 75031c0b21fSSimon J. Gerraty if (type == S_IFREG) { 75131c0b21fSSimon J. Gerraty /* the named path is the default contents */ 75231c0b21fSSimon J. Gerraty node->contents = mtree_file_path(node); 75331c0b21fSSimon J. Gerraty } 754484b5c25SMarcel Moolenaar } else 755484b5c25SMarcel Moolenaar type = (node->symlink != NULL) ? S_IFLNK : S_IFDIR; 756484b5c25SMarcel Moolenaar 757484b5c25SMarcel Moolenaar if (node->type == 0) 758484b5c25SMarcel Moolenaar node->type = type; 759484b5c25SMarcel Moolenaar 760484b5c25SMarcel Moolenaar if (node->type != type) { 761484b5c25SMarcel Moolenaar mtree_error("%s: file type and defined keywords to not match", 762484b5c25SMarcel Moolenaar node->name); 763484b5c25SMarcel Moolenaar return (0); 764484b5c25SMarcel Moolenaar } 765484b5c25SMarcel Moolenaar 766484b5c25SMarcel Moolenaar st->st_mode = (st->st_mode & ~S_IFMT) | node->type; 767484b5c25SMarcel Moolenaar 768484b5c25SMarcel Moolenaar if (node->contents == NULL) 769484b5c25SMarcel Moolenaar return (0); 770484b5c25SMarcel Moolenaar 771484b5c25SMarcel Moolenaar name = mtree_resolve(node->contents, &istemp); 772484b5c25SMarcel Moolenaar if (name == NULL) 773484b5c25SMarcel Moolenaar return (errno); 774484b5c25SMarcel Moolenaar 775484b5c25SMarcel Moolenaar if (stat(name, &sb) != 0) { 776484b5c25SMarcel Moolenaar mtree_error("%s: contents file '%s' not found", node->name, 777484b5c25SMarcel Moolenaar name); 778484b5c25SMarcel Moolenaar free(name); 779484b5c25SMarcel Moolenaar return (0); 780484b5c25SMarcel Moolenaar } 781484b5c25SMarcel Moolenaar 782484b5c25SMarcel Moolenaar free(node->contents); 783484b5c25SMarcel Moolenaar node->contents = name; 784484b5c25SMarcel Moolenaar st->st_size = sb.st_size; 785484b5c25SMarcel Moolenaar return (0); 786484b5c25SMarcel Moolenaar } 787484b5c25SMarcel Moolenaar 788484b5c25SMarcel Moolenaar static int 789484b5c25SMarcel Moolenaar read_mtree_command(FILE *fp) 790484b5c25SMarcel Moolenaar { 791484b5c25SMarcel Moolenaar char cmd[10]; 792484b5c25SMarcel Moolenaar int error; 793484b5c25SMarcel Moolenaar 794484b5c25SMarcel Moolenaar error = read_word(fp, cmd, sizeof(cmd)); 795484b5c25SMarcel Moolenaar if (error) 796484b5c25SMarcel Moolenaar goto out; 797484b5c25SMarcel Moolenaar 798484b5c25SMarcel Moolenaar error = read_mtree_keywords(fp, &mtree_global); 799484b5c25SMarcel Moolenaar 800484b5c25SMarcel Moolenaar out: 801484b5c25SMarcel Moolenaar skip_to(fp, "\n"); 802484b5c25SMarcel Moolenaar (void)getc(fp); 803484b5c25SMarcel Moolenaar return (error); 804484b5c25SMarcel Moolenaar } 805484b5c25SMarcel Moolenaar 806484b5c25SMarcel Moolenaar static int 807484b5c25SMarcel Moolenaar read_mtree_spec1(FILE *fp, bool def, const char *name) 808484b5c25SMarcel Moolenaar { 809484b5c25SMarcel Moolenaar fsnode *last, *node, *parent; 810484b5c25SMarcel Moolenaar u_int type; 811484b5c25SMarcel Moolenaar int error; 812484b5c25SMarcel Moolenaar 813484b5c25SMarcel Moolenaar assert(name[0] != '\0'); 814484b5c25SMarcel Moolenaar 815484b5c25SMarcel Moolenaar /* 816484b5c25SMarcel Moolenaar * Treat '..' specially, because it only changes our current 817484b5c25SMarcel Moolenaar * directory. We don't create a node for it. We simply ignore 818484b5c25SMarcel Moolenaar * any keywords that may appear on the line as well. 819484b5c25SMarcel Moolenaar * Going up a directory is a little non-obvious. A directory 820484b5c25SMarcel Moolenaar * node has a corresponding '.' child. The parent of '.' is 821484b5c25SMarcel Moolenaar * not the '.' node of the parent directory, but the directory 822484b5c25SMarcel Moolenaar * node within the parent to which the child relates. However, 823484b5c25SMarcel Moolenaar * going up a directory means we need to find the '.' node to 824484b5c25SMarcel Moolenaar * which the directoy node is linked. This we can do via the 825484b5c25SMarcel Moolenaar * first * pointer, because '.' is always the first entry in a 826484b5c25SMarcel Moolenaar * directory. 827484b5c25SMarcel Moolenaar */ 828484b5c25SMarcel Moolenaar if (IS_DOTDOT(name)) { 829484b5c25SMarcel Moolenaar /* This deals with NULL pointers as well. */ 830484b5c25SMarcel Moolenaar if (mtree_current == mtree_root) { 831484b5c25SMarcel Moolenaar mtree_warning("ignoring .. in root directory"); 832484b5c25SMarcel Moolenaar return (0); 833484b5c25SMarcel Moolenaar } 834484b5c25SMarcel Moolenaar 835484b5c25SMarcel Moolenaar node = mtree_current; 836484b5c25SMarcel Moolenaar 837484b5c25SMarcel Moolenaar assert(node != NULL); 838484b5c25SMarcel Moolenaar assert(IS_DOT(node->name)); 839484b5c25SMarcel Moolenaar assert(node->first == node); 840484b5c25SMarcel Moolenaar 841484b5c25SMarcel Moolenaar /* Get the corresponding directory node in the parent. */ 842484b5c25SMarcel Moolenaar node = mtree_current->parent; 843484b5c25SMarcel Moolenaar 844484b5c25SMarcel Moolenaar assert(node != NULL); 845484b5c25SMarcel Moolenaar assert(!IS_DOT(node->name)); 846484b5c25SMarcel Moolenaar 847484b5c25SMarcel Moolenaar node = node->first; 848484b5c25SMarcel Moolenaar 849484b5c25SMarcel Moolenaar assert(node != NULL); 850484b5c25SMarcel Moolenaar assert(IS_DOT(node->name)); 851484b5c25SMarcel Moolenaar assert(node->first == node); 852484b5c25SMarcel Moolenaar 853484b5c25SMarcel Moolenaar mtree_current = node; 854484b5c25SMarcel Moolenaar return (0); 855484b5c25SMarcel Moolenaar } 856484b5c25SMarcel Moolenaar 857484b5c25SMarcel Moolenaar /* 858484b5c25SMarcel Moolenaar * If we don't have a current directory and the first specification 859484b5c25SMarcel Moolenaar * (either implicit or defined) is not '.', then we need to create 860484b5c25SMarcel Moolenaar * a '.' node first (using a recursive call). 861484b5c25SMarcel Moolenaar */ 862484b5c25SMarcel Moolenaar if (!IS_DOT(name) && mtree_current == NULL) { 863484b5c25SMarcel Moolenaar error = read_mtree_spec1(fp, false, "."); 864484b5c25SMarcel Moolenaar if (error) 865484b5c25SMarcel Moolenaar return (error); 866484b5c25SMarcel Moolenaar } 867484b5c25SMarcel Moolenaar 868484b5c25SMarcel Moolenaar /* 869484b5c25SMarcel Moolenaar * Lookup the name in the current directory (if we have a current 870484b5c25SMarcel Moolenaar * directory) to make sure we do not create multiple nodes for the 871484b5c25SMarcel Moolenaar * same component. For non-definitions, if we find a node with the 872484b5c25SMarcel Moolenaar * same name, simply change the current directory. For definitions 873484b5c25SMarcel Moolenaar * more happens. 874484b5c25SMarcel Moolenaar */ 875484b5c25SMarcel Moolenaar last = NULL; 876484b5c25SMarcel Moolenaar node = mtree_current; 877484b5c25SMarcel Moolenaar while (node != NULL) { 878484b5c25SMarcel Moolenaar assert(node->first == mtree_current); 879484b5c25SMarcel Moolenaar 880484b5c25SMarcel Moolenaar if (strcmp(name, node->name) == 0) { 881484b5c25SMarcel Moolenaar if (def == true) { 882b0d9addeSBrooks Davis if (!dupsok) 883b0d9addeSBrooks Davis mtree_error( 884b0d9addeSBrooks Davis "duplicate definition of %s", 885b0d9addeSBrooks Davis name); 886b0d9addeSBrooks Davis else 887b0d9addeSBrooks Davis mtree_warning( 888b0d9addeSBrooks Davis "duplicate definition of %s", 889484b5c25SMarcel Moolenaar name); 890484b5c25SMarcel Moolenaar return (0); 891484b5c25SMarcel Moolenaar } 892484b5c25SMarcel Moolenaar 893484b5c25SMarcel Moolenaar if (node->type != S_IFDIR) { 894484b5c25SMarcel Moolenaar mtree_error("%s is not a directory", name); 895484b5c25SMarcel Moolenaar return (0); 896484b5c25SMarcel Moolenaar } 897484b5c25SMarcel Moolenaar 898484b5c25SMarcel Moolenaar assert(!IS_DOT(name)); 899484b5c25SMarcel Moolenaar 900484b5c25SMarcel Moolenaar node = node->child; 901484b5c25SMarcel Moolenaar 902484b5c25SMarcel Moolenaar assert(node != NULL); 903484b5c25SMarcel Moolenaar assert(IS_DOT(node->name)); 904484b5c25SMarcel Moolenaar 905484b5c25SMarcel Moolenaar mtree_current = node; 906484b5c25SMarcel Moolenaar return (0); 907484b5c25SMarcel Moolenaar } 908484b5c25SMarcel Moolenaar 909484b5c25SMarcel Moolenaar last = node; 910484b5c25SMarcel Moolenaar node = last->next; 911484b5c25SMarcel Moolenaar } 912484b5c25SMarcel Moolenaar 913484b5c25SMarcel Moolenaar parent = (mtree_current != NULL) ? mtree_current->parent : NULL; 914484b5c25SMarcel Moolenaar type = (def == false || IS_DOT(name)) ? S_IFDIR : 0; 915484b5c25SMarcel Moolenaar node = create_node(name, type, parent, &mtree_global); 916484b5c25SMarcel Moolenaar if (node == NULL) 917484b5c25SMarcel Moolenaar return (ENOMEM); 918484b5c25SMarcel Moolenaar 919484b5c25SMarcel Moolenaar if (def == true) { 920484b5c25SMarcel Moolenaar error = read_mtree_keywords(fp, node); 921484b5c25SMarcel Moolenaar if (error) { 922484b5c25SMarcel Moolenaar destroy_node(node); 923484b5c25SMarcel Moolenaar return (error); 924484b5c25SMarcel Moolenaar } 925484b5c25SMarcel Moolenaar } 926484b5c25SMarcel Moolenaar 927484b5c25SMarcel Moolenaar node->first = (mtree_current != NULL) ? mtree_current : node; 928484b5c25SMarcel Moolenaar 929484b5c25SMarcel Moolenaar if (last != NULL) 930484b5c25SMarcel Moolenaar last->next = node; 931484b5c25SMarcel Moolenaar 932484b5c25SMarcel Moolenaar if (node->type != S_IFDIR) 933484b5c25SMarcel Moolenaar return (0); 934484b5c25SMarcel Moolenaar 935484b5c25SMarcel Moolenaar if (!IS_DOT(node->name)) { 936484b5c25SMarcel Moolenaar parent = node; 937484b5c25SMarcel Moolenaar node = create_node(".", S_IFDIR, parent, parent); 938484b5c25SMarcel Moolenaar if (node == NULL) { 939484b5c25SMarcel Moolenaar last->next = NULL; 940484b5c25SMarcel Moolenaar destroy_node(parent); 941484b5c25SMarcel Moolenaar return (ENOMEM); 942484b5c25SMarcel Moolenaar } 943484b5c25SMarcel Moolenaar parent->child = node; 944484b5c25SMarcel Moolenaar node->first = node; 945484b5c25SMarcel Moolenaar } 946484b5c25SMarcel Moolenaar 947484b5c25SMarcel Moolenaar assert(node != NULL); 948484b5c25SMarcel Moolenaar assert(IS_DOT(node->name)); 949484b5c25SMarcel Moolenaar assert(node->first == node); 950484b5c25SMarcel Moolenaar 951484b5c25SMarcel Moolenaar mtree_current = node; 952484b5c25SMarcel Moolenaar if (mtree_root == NULL) 953484b5c25SMarcel Moolenaar mtree_root = node; 954484b5c25SMarcel Moolenaar 955484b5c25SMarcel Moolenaar return (0); 956484b5c25SMarcel Moolenaar } 957484b5c25SMarcel Moolenaar 958484b5c25SMarcel Moolenaar static int 959484b5c25SMarcel Moolenaar read_mtree_spec(FILE *fp) 960484b5c25SMarcel Moolenaar { 961484b5c25SMarcel Moolenaar char pathspec[PATH_MAX]; 962484b5c25SMarcel Moolenaar char *cp; 963484b5c25SMarcel Moolenaar int error; 964484b5c25SMarcel Moolenaar 965484b5c25SMarcel Moolenaar error = read_word(fp, pathspec, sizeof(pathspec)); 966484b5c25SMarcel Moolenaar if (error) 967484b5c25SMarcel Moolenaar goto out; 968484b5c25SMarcel Moolenaar 969484b5c25SMarcel Moolenaar cp = strchr(pathspec, '/'); 970484b5c25SMarcel Moolenaar if (cp != NULL) { 971484b5c25SMarcel Moolenaar /* Absolute pathname */ 972484b5c25SMarcel Moolenaar mtree_current = mtree_root; 973484b5c25SMarcel Moolenaar 974484b5c25SMarcel Moolenaar do { 975484b5c25SMarcel Moolenaar *cp++ = '\0'; 976484b5c25SMarcel Moolenaar 977*17b3e6f7SBrooks Davis /* Disallow '..' as a component. */ 978*17b3e6f7SBrooks Davis if (IS_DOTDOT(pathspec)) { 979*17b3e6f7SBrooks Davis mtree_error("absolute path cannot contain " 980*17b3e6f7SBrooks Davis ".. component"); 981484b5c25SMarcel Moolenaar goto out; 982484b5c25SMarcel Moolenaar } 983484b5c25SMarcel Moolenaar 984*17b3e6f7SBrooks Davis /* Ignore multiple adjacent slashes and '.'. */ 985*17b3e6f7SBrooks Davis if (pathspec[0] != '\0' && !IS_DOT(pathspec)) 986484b5c25SMarcel Moolenaar error = read_mtree_spec1(fp, false, pathspec); 987484b5c25SMarcel Moolenaar memmove(pathspec, cp, strlen(cp) + 1); 988484b5c25SMarcel Moolenaar cp = strchr(pathspec, '/'); 989484b5c25SMarcel Moolenaar } while (!error && cp != NULL); 990484b5c25SMarcel Moolenaar 991484b5c25SMarcel Moolenaar /* Disallow '.' and '..' as the last component. */ 992484b5c25SMarcel Moolenaar if (!error && (IS_DOT(pathspec) || IS_DOTDOT(pathspec))) { 993484b5c25SMarcel Moolenaar mtree_error("absolute path cannot contain . or .. " 994484b5c25SMarcel Moolenaar "components"); 995484b5c25SMarcel Moolenaar goto out; 996484b5c25SMarcel Moolenaar } 997484b5c25SMarcel Moolenaar } 998484b5c25SMarcel Moolenaar 999484b5c25SMarcel Moolenaar /* Ignore absolute specfications that end with a slash. */ 1000484b5c25SMarcel Moolenaar if (!error && pathspec[0] != '\0') 1001484b5c25SMarcel Moolenaar error = read_mtree_spec1(fp, true, pathspec); 1002484b5c25SMarcel Moolenaar 1003484b5c25SMarcel Moolenaar out: 1004484b5c25SMarcel Moolenaar skip_to(fp, "\n"); 1005484b5c25SMarcel Moolenaar (void)getc(fp); 1006484b5c25SMarcel Moolenaar return (error); 1007484b5c25SMarcel Moolenaar } 1008484b5c25SMarcel Moolenaar 1009484b5c25SMarcel Moolenaar fsnode * 1010484b5c25SMarcel Moolenaar read_mtree(const char *fname, fsnode *node) 1011484b5c25SMarcel Moolenaar { 1012484b5c25SMarcel Moolenaar struct mtree_fileinfo *fi; 1013484b5c25SMarcel Moolenaar FILE *fp; 1014484b5c25SMarcel Moolenaar int c, error; 1015484b5c25SMarcel Moolenaar 1016484b5c25SMarcel Moolenaar /* We do not yet support nesting... */ 1017484b5c25SMarcel Moolenaar assert(node == NULL); 1018484b5c25SMarcel Moolenaar 1019484b5c25SMarcel Moolenaar if (strcmp(fname, "-") == 0) 1020484b5c25SMarcel Moolenaar fp = stdin; 1021484b5c25SMarcel Moolenaar else { 1022484b5c25SMarcel Moolenaar fp = fopen(fname, "r"); 1023484b5c25SMarcel Moolenaar if (fp == NULL) 1024484b5c25SMarcel Moolenaar err(1, "Can't open `%s'", fname); 1025484b5c25SMarcel Moolenaar } 1026484b5c25SMarcel Moolenaar 1027484b5c25SMarcel Moolenaar error = mtree_file_push(fname, fp); 1028484b5c25SMarcel Moolenaar if (error) 1029484b5c25SMarcel Moolenaar goto out; 1030484b5c25SMarcel Moolenaar 1031484b5c25SMarcel Moolenaar bzero(&mtree_global, sizeof(mtree_global)); 1032484b5c25SMarcel Moolenaar bzero(&mtree_global_inode, sizeof(mtree_global_inode)); 1033484b5c25SMarcel Moolenaar mtree_global.inode = &mtree_global_inode; 1034484b5c25SMarcel Moolenaar mtree_global_inode.nlink = 1; 1035484b5c25SMarcel Moolenaar mtree_global_inode.st.st_atime = mtree_global_inode.st.st_ctime = 1036484b5c25SMarcel Moolenaar mtree_global_inode.st.st_mtime = time(NULL); 1037484b5c25SMarcel Moolenaar errors = warnings = 0; 1038484b5c25SMarcel Moolenaar 1039484b5c25SMarcel Moolenaar setgroupent(1); 1040484b5c25SMarcel Moolenaar setpassent(1); 1041484b5c25SMarcel Moolenaar 1042484b5c25SMarcel Moolenaar mtree_root = node; 1043484b5c25SMarcel Moolenaar mtree_current = node; 1044484b5c25SMarcel Moolenaar do { 1045484b5c25SMarcel Moolenaar /* Start of a new line... */ 1046484b5c25SMarcel Moolenaar fi = SLIST_FIRST(&mtree_fileinfo); 1047484b5c25SMarcel Moolenaar fi->line++; 1048484b5c25SMarcel Moolenaar 1049484b5c25SMarcel Moolenaar error = skip_over(fp, " \t"); 1050484b5c25SMarcel Moolenaar if (error) 1051484b5c25SMarcel Moolenaar break; 1052484b5c25SMarcel Moolenaar 1053484b5c25SMarcel Moolenaar c = getc(fp); 1054484b5c25SMarcel Moolenaar if (c == EOF) { 1055484b5c25SMarcel Moolenaar error = ferror(fp) ? errno : -1; 1056484b5c25SMarcel Moolenaar break; 1057484b5c25SMarcel Moolenaar } 1058484b5c25SMarcel Moolenaar 1059484b5c25SMarcel Moolenaar switch (c) { 1060484b5c25SMarcel Moolenaar case '\n': /* empty line */ 1061484b5c25SMarcel Moolenaar error = 0; 1062484b5c25SMarcel Moolenaar break; 1063484b5c25SMarcel Moolenaar case '#': /* comment -- skip to end of line. */ 1064484b5c25SMarcel Moolenaar error = skip_to(fp, "\n"); 1065484b5c25SMarcel Moolenaar if (!error) 1066484b5c25SMarcel Moolenaar (void)getc(fp); 1067484b5c25SMarcel Moolenaar break; 1068484b5c25SMarcel Moolenaar case '/': /* special commands */ 1069484b5c25SMarcel Moolenaar error = read_mtree_command(fp); 1070484b5c25SMarcel Moolenaar break; 1071484b5c25SMarcel Moolenaar default: /* specification */ 1072484b5c25SMarcel Moolenaar ungetc(c, fp); 1073484b5c25SMarcel Moolenaar error = read_mtree_spec(fp); 1074484b5c25SMarcel Moolenaar break; 1075484b5c25SMarcel Moolenaar } 1076484b5c25SMarcel Moolenaar } while (!error); 1077484b5c25SMarcel Moolenaar 1078484b5c25SMarcel Moolenaar endpwent(); 1079484b5c25SMarcel Moolenaar endgrent(); 1080484b5c25SMarcel Moolenaar 1081484b5c25SMarcel Moolenaar if (error <= 0 && (errors || warnings)) { 1082484b5c25SMarcel Moolenaar warnx("%u error(s) and %u warning(s) in mtree manifest", 1083484b5c25SMarcel Moolenaar errors, warnings); 1084484b5c25SMarcel Moolenaar if (errors) 1085484b5c25SMarcel Moolenaar exit(1); 1086484b5c25SMarcel Moolenaar } 1087484b5c25SMarcel Moolenaar 1088484b5c25SMarcel Moolenaar out: 1089484b5c25SMarcel Moolenaar if (error > 0) 1090484b5c25SMarcel Moolenaar errc(1, error, "Error reading mtree file"); 1091484b5c25SMarcel Moolenaar 1092484b5c25SMarcel Moolenaar if (fp != stdin) 1093484b5c25SMarcel Moolenaar fclose(fp); 1094484b5c25SMarcel Moolenaar 1095484b5c25SMarcel Moolenaar if (mtree_root != NULL) 1096484b5c25SMarcel Moolenaar return (mtree_root); 1097484b5c25SMarcel Moolenaar 1098484b5c25SMarcel Moolenaar /* Handle empty specifications. */ 1099484b5c25SMarcel Moolenaar node = create_node(".", S_IFDIR, NULL, &mtree_global); 1100484b5c25SMarcel Moolenaar node->first = node; 1101484b5c25SMarcel Moolenaar return (node); 1102484b5c25SMarcel Moolenaar } 1103