1484b5c25SMarcel Moolenaar /*- 21de7b4b8SPedro F. Giffuni * SPDX-License-Identifier: BSD-2-Clause-FreeBSD 31de7b4b8SPedro F. Giffuni * 4484b5c25SMarcel Moolenaar * Copyright (c) 2011 Marcel Moolenaar 5484b5c25SMarcel Moolenaar * All rights reserved. 6484b5c25SMarcel Moolenaar * 7484b5c25SMarcel Moolenaar * Redistribution and use in source and binary forms, with or without 8484b5c25SMarcel Moolenaar * modification, are permitted provided that the following conditions 9484b5c25SMarcel Moolenaar * are met: 10484b5c25SMarcel Moolenaar * 1. Redistributions of source code must retain the above copyright 11484b5c25SMarcel Moolenaar * notice, this list of conditions and the following disclaimer. 12484b5c25SMarcel Moolenaar * 2. Redistributions in binary form must reproduce the above copyright 13484b5c25SMarcel Moolenaar * notice, this list of conditions and the following disclaimer in the 14484b5c25SMarcel Moolenaar * documentation and/or other materials provided with the distribution. 15484b5c25SMarcel Moolenaar * 16484b5c25SMarcel Moolenaar * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR 17484b5c25SMarcel Moolenaar * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 18484b5c25SMarcel Moolenaar * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 19484b5c25SMarcel Moolenaar * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT, 20484b5c25SMarcel Moolenaar * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 21484b5c25SMarcel Moolenaar * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 22484b5c25SMarcel Moolenaar * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 23484b5c25SMarcel Moolenaar * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 24484b5c25SMarcel Moolenaar * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 25484b5c25SMarcel Moolenaar * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 26484b5c25SMarcel Moolenaar */ 27484b5c25SMarcel Moolenaar 285f401182SAlex Richardson #if HAVE_NBTOOL_CONFIG_H 295f401182SAlex Richardson #include "nbtool_config.h" 305f401182SAlex Richardson #endif 315f401182SAlex Richardson 32484b5c25SMarcel Moolenaar #include <sys/cdefs.h> 33484b5c25SMarcel Moolenaar __FBSDID("$FreeBSD$"); 34484b5c25SMarcel Moolenaar 35484b5c25SMarcel Moolenaar #include <sys/param.h> 36484b5c25SMarcel Moolenaar #include <sys/queue.h> 37484b5c25SMarcel Moolenaar #include <sys/sbuf.h> 38484b5c25SMarcel Moolenaar #include <sys/stat.h> 39484b5c25SMarcel Moolenaar #include <sys/types.h> 40484b5c25SMarcel Moolenaar #include <assert.h> 41484b5c25SMarcel Moolenaar #include <errno.h> 42484b5c25SMarcel Moolenaar #include <fcntl.h> 43484b5c25SMarcel Moolenaar #include <grp.h> 44484b5c25SMarcel Moolenaar #include <inttypes.h> 45484b5c25SMarcel Moolenaar #include <pwd.h> 46484b5c25SMarcel Moolenaar #include <stdarg.h> 47484b5c25SMarcel Moolenaar #include <stdbool.h> 48484b5c25SMarcel Moolenaar #include <stddef.h> 49484b5c25SMarcel Moolenaar #include <stdio.h> 50484b5c25SMarcel Moolenaar #include <stdlib.h> 51484b5c25SMarcel Moolenaar #include <string.h> 52484b5c25SMarcel Moolenaar #include <strings.h> 535d71efbeSMarcel Moolenaar #include <time.h> 54484b5c25SMarcel Moolenaar #include <unistd.h> 555f5598b1SEd Maste #include <util.h> 5688e2cc9eSBenno Rice #include <vis.h> 57484b5c25SMarcel Moolenaar 58484b5c25SMarcel Moolenaar #include "makefs.h" 59484b5c25SMarcel Moolenaar 605d71efbeSMarcel Moolenaar #ifndef ENOATTR 61e2ec2f75SPedro F. Giffuni #define ENOATTR ENODATA 625d71efbeSMarcel Moolenaar #endif 635d71efbeSMarcel Moolenaar 64484b5c25SMarcel Moolenaar #define IS_DOT(nm) ((nm)[0] == '.' && (nm)[1] == '\0') 65484b5c25SMarcel Moolenaar #define IS_DOTDOT(nm) ((nm)[0] == '.' && (nm)[1] == '.' && (nm)[2] == '\0') 66484b5c25SMarcel Moolenaar 67484b5c25SMarcel Moolenaar struct mtree_fileinfo { 68484b5c25SMarcel Moolenaar SLIST_ENTRY(mtree_fileinfo) next; 69484b5c25SMarcel Moolenaar FILE *fp; 70484b5c25SMarcel Moolenaar const char *name; 71484b5c25SMarcel Moolenaar u_int line; 72484b5c25SMarcel Moolenaar }; 73484b5c25SMarcel Moolenaar 74484b5c25SMarcel Moolenaar /* Global state used while parsing. */ 75484b5c25SMarcel Moolenaar static SLIST_HEAD(, mtree_fileinfo) mtree_fileinfo = 76484b5c25SMarcel Moolenaar SLIST_HEAD_INITIALIZER(mtree_fileinfo); 77484b5c25SMarcel Moolenaar static fsnode *mtree_root; 78484b5c25SMarcel Moolenaar static fsnode *mtree_current; 79484b5c25SMarcel Moolenaar static fsnode mtree_global; 80484b5c25SMarcel Moolenaar static fsinode mtree_global_inode; 81484b5c25SMarcel Moolenaar static u_int errors, warnings; 82484b5c25SMarcel Moolenaar 83484b5c25SMarcel Moolenaar static void mtree_error(const char *, ...) __printflike(1, 2); 84484b5c25SMarcel Moolenaar static void mtree_warning(const char *, ...) __printflike(1, 2); 85484b5c25SMarcel Moolenaar 86484b5c25SMarcel Moolenaar static int 87484b5c25SMarcel Moolenaar mtree_file_push(const char *name, FILE *fp) 88484b5c25SMarcel Moolenaar { 89484b5c25SMarcel Moolenaar struct mtree_fileinfo *fi; 90484b5c25SMarcel Moolenaar 915f5598b1SEd Maste fi = emalloc(sizeof(*fi)); 92484b5c25SMarcel Moolenaar if (strcmp(name, "-") == 0) 935f5598b1SEd Maste fi->name = estrdup("(stdin)"); 94484b5c25SMarcel Moolenaar else 955f5598b1SEd Maste fi->name = estrdup(name); 96484b5c25SMarcel Moolenaar if (fi->name == NULL) { 97484b5c25SMarcel Moolenaar free(fi); 98484b5c25SMarcel Moolenaar return (ENOMEM); 99484b5c25SMarcel Moolenaar } 100484b5c25SMarcel Moolenaar 101484b5c25SMarcel Moolenaar fi->fp = fp; 102484b5c25SMarcel Moolenaar fi->line = 0; 103484b5c25SMarcel Moolenaar 104484b5c25SMarcel Moolenaar SLIST_INSERT_HEAD(&mtree_fileinfo, fi, next); 105484b5c25SMarcel Moolenaar return (0); 106484b5c25SMarcel Moolenaar } 107484b5c25SMarcel Moolenaar 108484b5c25SMarcel Moolenaar static void 109484b5c25SMarcel Moolenaar mtree_print(const char *msgtype, const char *fmt, va_list ap) 110484b5c25SMarcel Moolenaar { 111484b5c25SMarcel Moolenaar struct mtree_fileinfo *fi; 112484b5c25SMarcel Moolenaar 113484b5c25SMarcel Moolenaar if (msgtype != NULL) { 114484b5c25SMarcel Moolenaar fi = SLIST_FIRST(&mtree_fileinfo); 115484b5c25SMarcel Moolenaar if (fi != NULL) 116484b5c25SMarcel Moolenaar fprintf(stderr, "%s:%u: ", fi->name, fi->line); 117484b5c25SMarcel Moolenaar fprintf(stderr, "%s: ", msgtype); 118484b5c25SMarcel Moolenaar } 119484b5c25SMarcel Moolenaar vfprintf(stderr, fmt, ap); 120484b5c25SMarcel Moolenaar } 121484b5c25SMarcel Moolenaar 122484b5c25SMarcel Moolenaar static void 123484b5c25SMarcel Moolenaar mtree_error(const char *fmt, ...) 124484b5c25SMarcel Moolenaar { 125484b5c25SMarcel Moolenaar va_list ap; 126484b5c25SMarcel Moolenaar 127484b5c25SMarcel Moolenaar va_start(ap, fmt); 128484b5c25SMarcel Moolenaar mtree_print("error", fmt, ap); 129484b5c25SMarcel Moolenaar va_end(ap); 130484b5c25SMarcel Moolenaar 131484b5c25SMarcel Moolenaar errors++; 132484b5c25SMarcel Moolenaar fputc('\n', stderr); 133484b5c25SMarcel Moolenaar } 134484b5c25SMarcel Moolenaar 135484b5c25SMarcel Moolenaar static void 136484b5c25SMarcel Moolenaar mtree_warning(const char *fmt, ...) 137484b5c25SMarcel Moolenaar { 138484b5c25SMarcel Moolenaar va_list ap; 139484b5c25SMarcel Moolenaar 140484b5c25SMarcel Moolenaar va_start(ap, fmt); 141484b5c25SMarcel Moolenaar mtree_print("warning", fmt, ap); 142484b5c25SMarcel Moolenaar va_end(ap); 143484b5c25SMarcel Moolenaar 144484b5c25SMarcel Moolenaar warnings++; 145484b5c25SMarcel Moolenaar fputc('\n', stderr); 146484b5c25SMarcel Moolenaar } 147484b5c25SMarcel Moolenaar 14831c0b21fSSimon J. Gerraty #ifndef MAKEFS_MAX_TREE_DEPTH 14931c0b21fSSimon J. Gerraty # define MAKEFS_MAX_TREE_DEPTH (MAXPATHLEN/2) 15031c0b21fSSimon J. Gerraty #endif 15131c0b21fSSimon J. Gerraty 15231c0b21fSSimon J. Gerraty /* construct path to node->name */ 15331c0b21fSSimon J. Gerraty static char * 15431c0b21fSSimon J. Gerraty mtree_file_path(fsnode *node) 15531c0b21fSSimon J. Gerraty { 15631c0b21fSSimon J. Gerraty fsnode *pnode; 15731c0b21fSSimon J. Gerraty struct sbuf *sb; 15831c0b21fSSimon J. Gerraty char *res, *rp[MAKEFS_MAX_TREE_DEPTH]; 15931c0b21fSSimon J. Gerraty int depth; 16031c0b21fSSimon J. Gerraty 16131c0b21fSSimon J. Gerraty depth = 0; 16231c0b21fSSimon J. Gerraty rp[depth] = node->name; 163e29182f6SDon Lewis for (pnode = node->parent; pnode && depth < MAKEFS_MAX_TREE_DEPTH - 1; 16431c0b21fSSimon J. Gerraty pnode = pnode->parent) { 16531c0b21fSSimon J. Gerraty if (strcmp(pnode->name, ".") == 0) 16631c0b21fSSimon J. Gerraty break; 16731c0b21fSSimon J. Gerraty rp[++depth] = pnode->name; 16831c0b21fSSimon J. Gerraty } 16931c0b21fSSimon J. Gerraty 1704ceda9d3SMarcel Moolenaar sb = sbuf_new_auto(); 17131c0b21fSSimon J. Gerraty if (sb == NULL) { 17231c0b21fSSimon J. Gerraty errno = ENOMEM; 17331c0b21fSSimon J. Gerraty return (NULL); 17431c0b21fSSimon J. Gerraty } 17531c0b21fSSimon J. Gerraty while (depth > 0) { 17631c0b21fSSimon J. Gerraty sbuf_cat(sb, rp[depth--]); 17731c0b21fSSimon J. Gerraty sbuf_putc(sb, '/'); 17831c0b21fSSimon J. Gerraty } 17931c0b21fSSimon J. Gerraty sbuf_cat(sb, rp[depth]); 18031c0b21fSSimon J. Gerraty sbuf_finish(sb); 1815f5598b1SEd Maste res = estrdup(sbuf_data(sb)); 18231c0b21fSSimon J. Gerraty sbuf_delete(sb); 18331c0b21fSSimon J. Gerraty if (res == NULL) 18431c0b21fSSimon J. Gerraty errno = ENOMEM; 18531c0b21fSSimon J. Gerraty return res; 18631c0b21fSSimon J. Gerraty 18731c0b21fSSimon J. Gerraty } 18831c0b21fSSimon J. Gerraty 189484b5c25SMarcel Moolenaar /* mtree_resolve() sets errno to indicate why NULL was returned. */ 190484b5c25SMarcel Moolenaar static char * 191484b5c25SMarcel Moolenaar mtree_resolve(const char *spec, int *istemp) 192484b5c25SMarcel Moolenaar { 193484b5c25SMarcel Moolenaar struct sbuf *sb; 19465b4f270SXin LI char *res, *var = NULL; 195484b5c25SMarcel Moolenaar const char *base, *p, *v; 196484b5c25SMarcel Moolenaar size_t len; 197484b5c25SMarcel Moolenaar int c, error, quoted, subst; 198484b5c25SMarcel Moolenaar 199484b5c25SMarcel Moolenaar len = strlen(spec); 200484b5c25SMarcel Moolenaar if (len == 0) { 201484b5c25SMarcel Moolenaar errno = EINVAL; 202484b5c25SMarcel Moolenaar return (NULL); 203484b5c25SMarcel Moolenaar } 204484b5c25SMarcel Moolenaar 205484b5c25SMarcel Moolenaar c = (len > 1) ? (spec[0] == spec[len - 1]) ? spec[0] : 0 : 0; 206484b5c25SMarcel Moolenaar *istemp = (c == '`') ? 1 : 0; 207484b5c25SMarcel Moolenaar subst = (c == '`' || c == '"') ? 1 : 0; 208484b5c25SMarcel Moolenaar quoted = (subst || c == '\'') ? 1 : 0; 209484b5c25SMarcel Moolenaar 210484b5c25SMarcel Moolenaar if (!subst) { 2115f5598b1SEd Maste res = estrdup(spec + quoted); 2125f5598b1SEd Maste if (quoted) 213484b5c25SMarcel Moolenaar res[len - 2] = '\0'; 214484b5c25SMarcel Moolenaar return (res); 215484b5c25SMarcel Moolenaar } 216484b5c25SMarcel Moolenaar 217484b5c25SMarcel Moolenaar sb = sbuf_new_auto(); 218484b5c25SMarcel Moolenaar if (sb == NULL) { 219484b5c25SMarcel Moolenaar errno = ENOMEM; 220484b5c25SMarcel Moolenaar return (NULL); 221484b5c25SMarcel Moolenaar } 222484b5c25SMarcel Moolenaar 223484b5c25SMarcel Moolenaar base = spec + 1; 224484b5c25SMarcel Moolenaar len -= 2; 225484b5c25SMarcel Moolenaar error = 0; 226484b5c25SMarcel Moolenaar while (len > 0) { 227484b5c25SMarcel Moolenaar p = strchr(base, '$'); 228484b5c25SMarcel Moolenaar if (p == NULL) { 229484b5c25SMarcel Moolenaar sbuf_bcat(sb, base, len); 230484b5c25SMarcel Moolenaar base += len; 231484b5c25SMarcel Moolenaar len = 0; 232484b5c25SMarcel Moolenaar continue; 233484b5c25SMarcel Moolenaar } 234484b5c25SMarcel Moolenaar /* The following is safe. spec always starts with a quote. */ 235484b5c25SMarcel Moolenaar if (p[-1] == '\\') 236484b5c25SMarcel Moolenaar p--; 237484b5c25SMarcel Moolenaar if (base != p) { 238484b5c25SMarcel Moolenaar sbuf_bcat(sb, base, p - base); 239484b5c25SMarcel Moolenaar len -= p - base; 240484b5c25SMarcel Moolenaar base = p; 241484b5c25SMarcel Moolenaar } 242484b5c25SMarcel Moolenaar if (*p == '\\') { 243484b5c25SMarcel Moolenaar sbuf_putc(sb, '$'); 244484b5c25SMarcel Moolenaar base += 2; 245484b5c25SMarcel Moolenaar len -= 2; 246484b5c25SMarcel Moolenaar continue; 247484b5c25SMarcel Moolenaar } 248484b5c25SMarcel Moolenaar /* Skip the '$'. */ 249484b5c25SMarcel Moolenaar base++; 250484b5c25SMarcel Moolenaar len--; 251484b5c25SMarcel Moolenaar /* Handle ${X} vs $X. */ 252484b5c25SMarcel Moolenaar v = base; 253484b5c25SMarcel Moolenaar if (*base == '{') { 254484b5c25SMarcel Moolenaar p = strchr(v, '}'); 255484b5c25SMarcel Moolenaar if (p == NULL) 256484b5c25SMarcel Moolenaar p = v; 257484b5c25SMarcel Moolenaar } else 258484b5c25SMarcel Moolenaar p = v; 259484b5c25SMarcel Moolenaar len -= (p + 1) - base; 260484b5c25SMarcel Moolenaar base = p + 1; 261484b5c25SMarcel Moolenaar 262484b5c25SMarcel Moolenaar if (v == p) { 263484b5c25SMarcel Moolenaar sbuf_putc(sb, *v); 264484b5c25SMarcel Moolenaar continue; 265484b5c25SMarcel Moolenaar } 266484b5c25SMarcel Moolenaar 267484b5c25SMarcel Moolenaar error = ENOMEM; 2685f5598b1SEd Maste var = ecalloc(p - v, 1); 269484b5c25SMarcel Moolenaar memcpy(var, v + 1, p - v - 1); 270484b5c25SMarcel Moolenaar if (strcmp(var, ".CURDIR") == 0) { 271484b5c25SMarcel Moolenaar res = getcwd(NULL, 0); 272484b5c25SMarcel Moolenaar if (res == NULL) 273484b5c25SMarcel Moolenaar break; 274484b5c25SMarcel Moolenaar } else if (strcmp(var, ".PROG") == 0) { 2755f5598b1SEd Maste res = estrdup(getprogname()); 276484b5c25SMarcel Moolenaar } else { 277484b5c25SMarcel Moolenaar v = getenv(var); 278484b5c25SMarcel Moolenaar if (v != NULL) { 2795f5598b1SEd Maste res = estrdup(v); 280484b5c25SMarcel Moolenaar } else 281484b5c25SMarcel Moolenaar res = NULL; 282484b5c25SMarcel Moolenaar } 283484b5c25SMarcel Moolenaar error = 0; 284484b5c25SMarcel Moolenaar 285484b5c25SMarcel Moolenaar if (res != NULL) { 286484b5c25SMarcel Moolenaar sbuf_cat(sb, res); 287484b5c25SMarcel Moolenaar free(res); 288484b5c25SMarcel Moolenaar } 289484b5c25SMarcel Moolenaar free(var); 29065b4f270SXin LI var = NULL; 291484b5c25SMarcel Moolenaar } 292484b5c25SMarcel Moolenaar 29365b4f270SXin LI free(var); 294484b5c25SMarcel Moolenaar sbuf_finish(sb); 295484b5c25SMarcel Moolenaar res = (error == 0) ? strdup(sbuf_data(sb)) : NULL; 296484b5c25SMarcel Moolenaar sbuf_delete(sb); 297484b5c25SMarcel Moolenaar if (res == NULL) 298484b5c25SMarcel Moolenaar errno = ENOMEM; 299484b5c25SMarcel Moolenaar return (res); 300484b5c25SMarcel Moolenaar } 301484b5c25SMarcel Moolenaar 302484b5c25SMarcel Moolenaar static int 303484b5c25SMarcel Moolenaar skip_over(FILE *fp, const char *cs) 304484b5c25SMarcel Moolenaar { 305484b5c25SMarcel Moolenaar int c; 306484b5c25SMarcel Moolenaar 307484b5c25SMarcel Moolenaar c = getc(fp); 308484b5c25SMarcel Moolenaar while (c != EOF && strchr(cs, c) != NULL) 309484b5c25SMarcel Moolenaar c = getc(fp); 310484b5c25SMarcel Moolenaar if (c != EOF) { 311484b5c25SMarcel Moolenaar ungetc(c, fp); 312484b5c25SMarcel Moolenaar return (0); 313484b5c25SMarcel Moolenaar } 314484b5c25SMarcel Moolenaar return (ferror(fp) ? errno : -1); 315484b5c25SMarcel Moolenaar } 316484b5c25SMarcel Moolenaar 317484b5c25SMarcel Moolenaar static int 318484b5c25SMarcel Moolenaar skip_to(FILE *fp, const char *cs) 319484b5c25SMarcel Moolenaar { 320484b5c25SMarcel Moolenaar int c; 321484b5c25SMarcel Moolenaar 322484b5c25SMarcel Moolenaar c = getc(fp); 323484b5c25SMarcel Moolenaar while (c != EOF && strchr(cs, c) == NULL) 324484b5c25SMarcel Moolenaar c = getc(fp); 325484b5c25SMarcel Moolenaar if (c != EOF) { 326484b5c25SMarcel Moolenaar ungetc(c, fp); 327484b5c25SMarcel Moolenaar return (0); 328484b5c25SMarcel Moolenaar } 329484b5c25SMarcel Moolenaar return (ferror(fp) ? errno : -1); 330484b5c25SMarcel Moolenaar } 331484b5c25SMarcel Moolenaar 332484b5c25SMarcel Moolenaar static int 333484b5c25SMarcel Moolenaar read_word(FILE *fp, char *buf, size_t bufsz) 334484b5c25SMarcel Moolenaar { 335484b5c25SMarcel Moolenaar struct mtree_fileinfo *fi; 336484b5c25SMarcel Moolenaar size_t idx, qidx; 337484b5c25SMarcel Moolenaar int c, done, error, esc, qlvl; 338484b5c25SMarcel Moolenaar 339484b5c25SMarcel Moolenaar if (bufsz == 0) 340484b5c25SMarcel Moolenaar return (EINVAL); 341484b5c25SMarcel Moolenaar 342484b5c25SMarcel Moolenaar done = 0; 343484b5c25SMarcel Moolenaar esc = 0; 344484b5c25SMarcel Moolenaar idx = 0; 345484b5c25SMarcel Moolenaar qidx = -1; 346484b5c25SMarcel Moolenaar qlvl = 0; 347484b5c25SMarcel Moolenaar do { 348484b5c25SMarcel Moolenaar c = getc(fp); 349484b5c25SMarcel Moolenaar switch (c) { 350484b5c25SMarcel Moolenaar case EOF: 351484b5c25SMarcel Moolenaar buf[idx] = '\0'; 352484b5c25SMarcel Moolenaar error = ferror(fp) ? errno : -1; 353484b5c25SMarcel Moolenaar if (error == -1) 354484b5c25SMarcel Moolenaar mtree_error("unexpected end of file"); 355484b5c25SMarcel Moolenaar return (error); 35612a5ad47SSimon J. Gerraty case '#': /* comment -- skip to end of line. */ 35712a5ad47SSimon J. Gerraty if (!esc) { 35812a5ad47SSimon J. Gerraty error = skip_to(fp, "\n"); 35912a5ad47SSimon J. Gerraty if (!error) 36012a5ad47SSimon J. Gerraty continue; 36112a5ad47SSimon J. Gerraty } 36212a5ad47SSimon J. Gerraty break; 363484b5c25SMarcel Moolenaar case '\\': 364484b5c25SMarcel Moolenaar esc++; 365484b5c25SMarcel Moolenaar break; 366484b5c25SMarcel Moolenaar case '`': 367484b5c25SMarcel Moolenaar case '\'': 368484b5c25SMarcel Moolenaar case '"': 369484b5c25SMarcel Moolenaar if (esc) 370484b5c25SMarcel Moolenaar break; 371484b5c25SMarcel Moolenaar if (qlvl == 0) { 372484b5c25SMarcel Moolenaar qlvl++; 373484b5c25SMarcel Moolenaar qidx = idx; 374484b5c25SMarcel Moolenaar } else if (c == buf[qidx]) { 375484b5c25SMarcel Moolenaar qlvl--; 376484b5c25SMarcel Moolenaar if (qlvl > 0) { 377484b5c25SMarcel Moolenaar do { 378484b5c25SMarcel Moolenaar qidx--; 379484b5c25SMarcel Moolenaar } while (buf[qidx] != '`' && 380484b5c25SMarcel Moolenaar buf[qidx] != '\'' && 381484b5c25SMarcel Moolenaar buf[qidx] != '"'); 382484b5c25SMarcel Moolenaar } else 383484b5c25SMarcel Moolenaar qidx = -1; 384484b5c25SMarcel Moolenaar } else { 385484b5c25SMarcel Moolenaar qlvl++; 386484b5c25SMarcel Moolenaar qidx = idx; 387484b5c25SMarcel Moolenaar } 388484b5c25SMarcel Moolenaar break; 389484b5c25SMarcel Moolenaar case ' ': 390484b5c25SMarcel Moolenaar case '\t': 391484b5c25SMarcel Moolenaar case '\n': 392484b5c25SMarcel Moolenaar if (!esc && qlvl == 0) { 393484b5c25SMarcel Moolenaar ungetc(c, fp); 394484b5c25SMarcel Moolenaar c = '\0'; 395484b5c25SMarcel Moolenaar done = 1; 396484b5c25SMarcel Moolenaar break; 397484b5c25SMarcel Moolenaar } 398484b5c25SMarcel Moolenaar if (c == '\n') { 399484b5c25SMarcel Moolenaar /* 400484b5c25SMarcel Moolenaar * We going to eat the newline ourselves. 401484b5c25SMarcel Moolenaar */ 402484b5c25SMarcel Moolenaar if (qlvl > 0) 403484b5c25SMarcel Moolenaar mtree_warning("quoted word straddles " 404484b5c25SMarcel Moolenaar "onto next line."); 405484b5c25SMarcel Moolenaar fi = SLIST_FIRST(&mtree_fileinfo); 406484b5c25SMarcel Moolenaar fi->line++; 407484b5c25SMarcel Moolenaar } 408484b5c25SMarcel Moolenaar break; 40988e2cc9eSBenno Rice default: 410484b5c25SMarcel Moolenaar if (esc) 41188e2cc9eSBenno Rice buf[idx++] = '\\'; 412484b5c25SMarcel Moolenaar break; 413484b5c25SMarcel Moolenaar } 414484b5c25SMarcel Moolenaar buf[idx++] = c; 415484b5c25SMarcel Moolenaar esc = 0; 416484b5c25SMarcel Moolenaar } while (idx < bufsz && !done); 417484b5c25SMarcel Moolenaar 418484b5c25SMarcel Moolenaar if (idx >= bufsz) { 419484b5c25SMarcel Moolenaar mtree_error("word too long to fit buffer (max %zu characters)", 420484b5c25SMarcel Moolenaar bufsz); 421484b5c25SMarcel Moolenaar skip_to(fp, " \t\n"); 422484b5c25SMarcel Moolenaar } 423484b5c25SMarcel Moolenaar return (0); 424484b5c25SMarcel Moolenaar } 425484b5c25SMarcel Moolenaar 426484b5c25SMarcel Moolenaar static fsnode * 427484b5c25SMarcel Moolenaar create_node(const char *name, u_int type, fsnode *parent, fsnode *global) 428484b5c25SMarcel Moolenaar { 429484b5c25SMarcel Moolenaar fsnode *n; 430484b5c25SMarcel Moolenaar 4315f5598b1SEd Maste n = ecalloc(1, sizeof(*n)); 4325f5598b1SEd Maste n->name = estrdup(name); 433484b5c25SMarcel Moolenaar n->type = (type == 0) ? global->type : type; 434484b5c25SMarcel Moolenaar n->parent = parent; 435484b5c25SMarcel Moolenaar 4365f5598b1SEd Maste n->inode = ecalloc(1, sizeof(*n->inode)); 437484b5c25SMarcel Moolenaar 438484b5c25SMarcel Moolenaar /* Assign global options/defaults. */ 43983b6064dSEd Maste memcpy(n->inode, global->inode, sizeof(*n->inode)); 440484b5c25SMarcel Moolenaar n->inode->st.st_mode = (n->inode->st.st_mode & ~S_IFMT) | n->type; 441484b5c25SMarcel Moolenaar 442484b5c25SMarcel Moolenaar if (n->type == S_IFLNK) 443484b5c25SMarcel Moolenaar n->symlink = global->symlink; 444484b5c25SMarcel Moolenaar else if (n->type == S_IFREG) 445484b5c25SMarcel Moolenaar n->contents = global->contents; 446484b5c25SMarcel Moolenaar 447484b5c25SMarcel Moolenaar return (n); 448484b5c25SMarcel Moolenaar } 449484b5c25SMarcel Moolenaar 450484b5c25SMarcel Moolenaar static void 451484b5c25SMarcel Moolenaar destroy_node(fsnode *n) 452484b5c25SMarcel Moolenaar { 453484b5c25SMarcel Moolenaar 454484b5c25SMarcel Moolenaar assert(n != NULL); 455484b5c25SMarcel Moolenaar assert(n->name != NULL); 456484b5c25SMarcel Moolenaar assert(n->inode != NULL); 457484b5c25SMarcel Moolenaar 458484b5c25SMarcel Moolenaar free(n->inode); 459484b5c25SMarcel Moolenaar free(n->name); 460484b5c25SMarcel Moolenaar free(n); 461484b5c25SMarcel Moolenaar } 462484b5c25SMarcel Moolenaar 463484b5c25SMarcel Moolenaar static int 464484b5c25SMarcel Moolenaar read_number(const char *tok, u_int base, intmax_t *res, intmax_t min, 465484b5c25SMarcel Moolenaar intmax_t max) 466484b5c25SMarcel Moolenaar { 467484b5c25SMarcel Moolenaar char *end; 468484b5c25SMarcel Moolenaar intmax_t val; 469484b5c25SMarcel Moolenaar 470484b5c25SMarcel Moolenaar val = strtoimax(tok, &end, base); 471484b5c25SMarcel Moolenaar if (end == tok || end[0] != '\0') 472484b5c25SMarcel Moolenaar return (EINVAL); 473484b5c25SMarcel Moolenaar if (val < min || val > max) 474484b5c25SMarcel Moolenaar return (EDOM); 475484b5c25SMarcel Moolenaar *res = val; 476484b5c25SMarcel Moolenaar return (0); 477484b5c25SMarcel Moolenaar } 478484b5c25SMarcel Moolenaar 479484b5c25SMarcel Moolenaar static int 480484b5c25SMarcel Moolenaar read_mtree_keywords(FILE *fp, fsnode *node) 481484b5c25SMarcel Moolenaar { 482484b5c25SMarcel Moolenaar char keyword[PATH_MAX]; 483484b5c25SMarcel Moolenaar char *name, *p, *value; 484e9a03238SBrooks Davis gid_t gid; 485e9a03238SBrooks Davis uid_t uid; 486484b5c25SMarcel Moolenaar struct stat *st, sb; 487484b5c25SMarcel Moolenaar intmax_t num; 488484b5c25SMarcel Moolenaar u_long flset, flclr; 4893afe6a68SEd Maste int error, istemp; 4903afe6a68SEd Maste uint32_t type; 491484b5c25SMarcel Moolenaar 492484b5c25SMarcel Moolenaar st = &node->inode->st; 493484b5c25SMarcel Moolenaar do { 494484b5c25SMarcel Moolenaar error = skip_over(fp, " \t"); 495484b5c25SMarcel Moolenaar if (error) 496484b5c25SMarcel Moolenaar break; 497484b5c25SMarcel Moolenaar 498484b5c25SMarcel Moolenaar error = read_word(fp, keyword, sizeof(keyword)); 499484b5c25SMarcel Moolenaar if (error) 500484b5c25SMarcel Moolenaar break; 501484b5c25SMarcel Moolenaar 502484b5c25SMarcel Moolenaar if (keyword[0] == '\0') 503484b5c25SMarcel Moolenaar break; 504484b5c25SMarcel Moolenaar 505484b5c25SMarcel Moolenaar value = strchr(keyword, '='); 506484b5c25SMarcel Moolenaar if (value != NULL) 507484b5c25SMarcel Moolenaar *value++ = '\0'; 508484b5c25SMarcel Moolenaar 509484b5c25SMarcel Moolenaar /* 510484b5c25SMarcel Moolenaar * We use EINVAL, ENOATTR, ENOSYS and ENXIO to signal 511484b5c25SMarcel Moolenaar * certain conditions: 512484b5c25SMarcel Moolenaar * EINVAL - Value provided for a keyword that does 513484b5c25SMarcel Moolenaar * not take a value. The value is ignored. 514484b5c25SMarcel Moolenaar * ENOATTR - Value missing for a keyword that needs 515484b5c25SMarcel Moolenaar * a value. The keyword is ignored. 516484b5c25SMarcel Moolenaar * ENOSYS - Unsupported keyword encountered. The 517484b5c25SMarcel Moolenaar * keyword is ignored. 518484b5c25SMarcel Moolenaar * ENXIO - Value provided for a keyword that does 519484b5c25SMarcel Moolenaar * not take a value. The value is ignored. 520484b5c25SMarcel Moolenaar */ 521484b5c25SMarcel Moolenaar switch (keyword[0]) { 522484b5c25SMarcel Moolenaar case 'c': 523484b5c25SMarcel Moolenaar if (strcmp(keyword, "contents") == 0) { 524484b5c25SMarcel Moolenaar if (value == NULL) { 525484b5c25SMarcel Moolenaar error = ENOATTR; 526484b5c25SMarcel Moolenaar break; 527484b5c25SMarcel Moolenaar } 5285f5598b1SEd Maste node->contents = estrdup(value); 529484b5c25SMarcel Moolenaar } else 530484b5c25SMarcel Moolenaar error = ENOSYS; 531484b5c25SMarcel Moolenaar break; 532484b5c25SMarcel Moolenaar case 'f': 533484b5c25SMarcel Moolenaar if (strcmp(keyword, "flags") == 0) { 534484b5c25SMarcel Moolenaar if (value == NULL) { 535484b5c25SMarcel Moolenaar error = ENOATTR; 536484b5c25SMarcel Moolenaar break; 537484b5c25SMarcel Moolenaar } 538484b5c25SMarcel Moolenaar flset = flclr = 0; 5395f401182SAlex Richardson #if HAVE_STRUCT_STAT_ST_FLAGS 540484b5c25SMarcel Moolenaar if (!strtofflags(&value, &flset, &flclr)) { 541484b5c25SMarcel Moolenaar st->st_flags &= ~flclr; 542484b5c25SMarcel Moolenaar st->st_flags |= flset; 543484b5c25SMarcel Moolenaar } else 544484b5c25SMarcel Moolenaar error = errno; 5455f401182SAlex Richardson #endif 546484b5c25SMarcel Moolenaar } else 547484b5c25SMarcel Moolenaar error = ENOSYS; 548484b5c25SMarcel Moolenaar break; 549484b5c25SMarcel Moolenaar case 'g': 550484b5c25SMarcel Moolenaar if (strcmp(keyword, "gid") == 0) { 551484b5c25SMarcel Moolenaar if (value == NULL) { 552484b5c25SMarcel Moolenaar error = ENOATTR; 553484b5c25SMarcel Moolenaar break; 554484b5c25SMarcel Moolenaar } 555484b5c25SMarcel Moolenaar error = read_number(value, 10, &num, 556484b5c25SMarcel Moolenaar 0, UINT_MAX); 557484b5c25SMarcel Moolenaar if (!error) 558484b5c25SMarcel Moolenaar st->st_gid = num; 559484b5c25SMarcel Moolenaar } else if (strcmp(keyword, "gname") == 0) { 560484b5c25SMarcel Moolenaar if (value == NULL) { 561484b5c25SMarcel Moolenaar error = ENOATTR; 562484b5c25SMarcel Moolenaar break; 563484b5c25SMarcel Moolenaar } 564e9a03238SBrooks Davis if (gid_from_group(value, &gid) == 0) 565e9a03238SBrooks Davis st->st_gid = gid; 566484b5c25SMarcel Moolenaar else 567e9a03238SBrooks Davis error = EINVAL; 568484b5c25SMarcel Moolenaar } else 569484b5c25SMarcel Moolenaar error = ENOSYS; 570484b5c25SMarcel Moolenaar break; 571484b5c25SMarcel Moolenaar case 'l': 572484b5c25SMarcel Moolenaar if (strcmp(keyword, "link") == 0) { 573484b5c25SMarcel Moolenaar if (value == NULL) { 574484b5c25SMarcel Moolenaar error = ENOATTR; 575484b5c25SMarcel Moolenaar break; 576484b5c25SMarcel Moolenaar } 57788e2cc9eSBenno Rice node->symlink = emalloc(strlen(value) + 1); 57888e2cc9eSBenno Rice if (node->symlink == NULL) { 57988e2cc9eSBenno Rice error = errno; 58088e2cc9eSBenno Rice break; 58188e2cc9eSBenno Rice } 58288e2cc9eSBenno Rice if (strunvis(node->symlink, value) < 0) { 58388e2cc9eSBenno Rice error = errno; 58488e2cc9eSBenno Rice break; 58588e2cc9eSBenno Rice } 586484b5c25SMarcel Moolenaar } else 587484b5c25SMarcel Moolenaar error = ENOSYS; 588484b5c25SMarcel Moolenaar break; 589484b5c25SMarcel Moolenaar case 'm': 590484b5c25SMarcel Moolenaar if (strcmp(keyword, "mode") == 0) { 591484b5c25SMarcel Moolenaar if (value == NULL) { 592484b5c25SMarcel Moolenaar error = ENOATTR; 593484b5c25SMarcel Moolenaar break; 594484b5c25SMarcel Moolenaar } 595484b5c25SMarcel Moolenaar if (value[0] >= '0' && value[0] <= '9') { 596484b5c25SMarcel Moolenaar error = read_number(value, 8, &num, 597484b5c25SMarcel Moolenaar 0, 07777); 598484b5c25SMarcel Moolenaar if (!error) { 599484b5c25SMarcel Moolenaar st->st_mode &= S_IFMT; 600484b5c25SMarcel Moolenaar st->st_mode |= num; 601484b5c25SMarcel Moolenaar } 602484b5c25SMarcel Moolenaar } else { 603484b5c25SMarcel Moolenaar /* Symbolic mode not supported. */ 604484b5c25SMarcel Moolenaar error = EINVAL; 605484b5c25SMarcel Moolenaar break; 606484b5c25SMarcel Moolenaar } 607484b5c25SMarcel Moolenaar } else 608484b5c25SMarcel Moolenaar error = ENOSYS; 609484b5c25SMarcel Moolenaar break; 610484b5c25SMarcel Moolenaar case 'o': 611484b5c25SMarcel Moolenaar if (strcmp(keyword, "optional") == 0) { 612484b5c25SMarcel Moolenaar if (value != NULL) 613484b5c25SMarcel Moolenaar error = ENXIO; 614484b5c25SMarcel Moolenaar node->flags |= FSNODE_F_OPTIONAL; 615484b5c25SMarcel Moolenaar } else 616484b5c25SMarcel Moolenaar error = ENOSYS; 617484b5c25SMarcel Moolenaar break; 618484b5c25SMarcel Moolenaar case 's': 619484b5c25SMarcel Moolenaar if (strcmp(keyword, "size") == 0) { 620484b5c25SMarcel Moolenaar if (value == NULL) { 621484b5c25SMarcel Moolenaar error = ENOATTR; 622484b5c25SMarcel Moolenaar break; 623484b5c25SMarcel Moolenaar } 624484b5c25SMarcel Moolenaar error = read_number(value, 10, &num, 625484b5c25SMarcel Moolenaar 0, INTMAX_MAX); 626484b5c25SMarcel Moolenaar if (!error) 627484b5c25SMarcel Moolenaar st->st_size = num; 628484b5c25SMarcel Moolenaar } else 629484b5c25SMarcel Moolenaar error = ENOSYS; 630484b5c25SMarcel Moolenaar break; 631484b5c25SMarcel Moolenaar case 't': 632484b5c25SMarcel Moolenaar if (strcmp(keyword, "time") == 0) { 633484b5c25SMarcel Moolenaar if (value == NULL) { 634484b5c25SMarcel Moolenaar error = ENOATTR; 635484b5c25SMarcel Moolenaar break; 636484b5c25SMarcel Moolenaar } 637484b5c25SMarcel Moolenaar p = strchr(value, '.'); 638484b5c25SMarcel Moolenaar if (p != NULL) 639484b5c25SMarcel Moolenaar *p++ = '\0'; 640484b5c25SMarcel Moolenaar error = read_number(value, 10, &num, 0, 641484b5c25SMarcel Moolenaar INTMAX_MAX); 642484b5c25SMarcel Moolenaar if (error) 643484b5c25SMarcel Moolenaar break; 644484b5c25SMarcel Moolenaar st->st_atime = num; 645484b5c25SMarcel Moolenaar st->st_ctime = num; 646484b5c25SMarcel Moolenaar st->st_mtime = num; 647286258a9SEd Maste #if HAVE_STRUCT_STAT_ST_MTIMENSEC 6484ceda9d3SMarcel Moolenaar if (p == NULL) 6494ceda9d3SMarcel Moolenaar break; 650484b5c25SMarcel Moolenaar error = read_number(p, 10, &num, 0, 651484b5c25SMarcel Moolenaar INTMAX_MAX); 652484b5c25SMarcel Moolenaar if (error) 653484b5c25SMarcel Moolenaar break; 654286258a9SEd Maste st->st_atimensec = num; 655286258a9SEd Maste st->st_ctimensec = num; 656286258a9SEd Maste st->st_mtimensec = num; 657286258a9SEd Maste #endif 658484b5c25SMarcel Moolenaar } else if (strcmp(keyword, "type") == 0) { 659484b5c25SMarcel Moolenaar if (value == NULL) { 660484b5c25SMarcel Moolenaar error = ENOATTR; 661484b5c25SMarcel Moolenaar break; 662484b5c25SMarcel Moolenaar } 663484b5c25SMarcel Moolenaar if (strcmp(value, "dir") == 0) 664484b5c25SMarcel Moolenaar node->type = S_IFDIR; 665484b5c25SMarcel Moolenaar else if (strcmp(value, "file") == 0) 666484b5c25SMarcel Moolenaar node->type = S_IFREG; 667484b5c25SMarcel Moolenaar else if (strcmp(value, "link") == 0) 668484b5c25SMarcel Moolenaar node->type = S_IFLNK; 669484b5c25SMarcel Moolenaar else 670484b5c25SMarcel Moolenaar error = EINVAL; 671484b5c25SMarcel Moolenaar } else 672484b5c25SMarcel Moolenaar error = ENOSYS; 673484b5c25SMarcel Moolenaar break; 674484b5c25SMarcel Moolenaar case 'u': 675484b5c25SMarcel Moolenaar if (strcmp(keyword, "uid") == 0) { 676484b5c25SMarcel Moolenaar if (value == NULL) { 677484b5c25SMarcel Moolenaar error = ENOATTR; 678484b5c25SMarcel Moolenaar break; 679484b5c25SMarcel Moolenaar } 680484b5c25SMarcel Moolenaar error = read_number(value, 10, &num, 681484b5c25SMarcel Moolenaar 0, UINT_MAX); 682484b5c25SMarcel Moolenaar if (!error) 683484b5c25SMarcel Moolenaar st->st_uid = num; 684484b5c25SMarcel Moolenaar } else if (strcmp(keyword, "uname") == 0) { 685484b5c25SMarcel Moolenaar if (value == NULL) { 686484b5c25SMarcel Moolenaar error = ENOATTR; 687484b5c25SMarcel Moolenaar break; 688484b5c25SMarcel Moolenaar } 689e9a03238SBrooks Davis if (uid_from_user(value, &uid) == 0) 690e9a03238SBrooks Davis st->st_uid = uid; 691484b5c25SMarcel Moolenaar else 692e9a03238SBrooks Davis error = EINVAL; 693484b5c25SMarcel Moolenaar } else 694484b5c25SMarcel Moolenaar error = ENOSYS; 695484b5c25SMarcel Moolenaar break; 696484b5c25SMarcel Moolenaar default: 697484b5c25SMarcel Moolenaar error = ENOSYS; 698484b5c25SMarcel Moolenaar break; 699484b5c25SMarcel Moolenaar } 700484b5c25SMarcel Moolenaar 701484b5c25SMarcel Moolenaar switch (error) { 702484b5c25SMarcel Moolenaar case EINVAL: 703484b5c25SMarcel Moolenaar mtree_error("%s: invalid value '%s'", keyword, value); 704484b5c25SMarcel Moolenaar break; 705484b5c25SMarcel Moolenaar case ENOATTR: 706484b5c25SMarcel Moolenaar mtree_error("%s: keyword needs a value", keyword); 707484b5c25SMarcel Moolenaar break; 708484b5c25SMarcel Moolenaar case ENOSYS: 709484b5c25SMarcel Moolenaar mtree_warning("%s: unsupported keyword", keyword); 710484b5c25SMarcel Moolenaar break; 711484b5c25SMarcel Moolenaar case ENXIO: 712484b5c25SMarcel Moolenaar mtree_error("%s: keyword does not take a value", 713484b5c25SMarcel Moolenaar keyword); 714484b5c25SMarcel Moolenaar break; 715484b5c25SMarcel Moolenaar } 716484b5c25SMarcel Moolenaar } while (1); 717484b5c25SMarcel Moolenaar 718484b5c25SMarcel Moolenaar if (error) 719484b5c25SMarcel Moolenaar return (error); 720484b5c25SMarcel Moolenaar 721484b5c25SMarcel Moolenaar st->st_mode = (st->st_mode & ~S_IFMT) | node->type; 722484b5c25SMarcel Moolenaar 723484b5c25SMarcel Moolenaar /* Nothing more to do for the global defaults. */ 724484b5c25SMarcel Moolenaar if (node->name == NULL) 725484b5c25SMarcel Moolenaar return (0); 726484b5c25SMarcel Moolenaar 727484b5c25SMarcel Moolenaar /* 728484b5c25SMarcel Moolenaar * Be intelligent about the file type. 729484b5c25SMarcel Moolenaar */ 730484b5c25SMarcel Moolenaar if (node->contents != NULL) { 731484b5c25SMarcel Moolenaar if (node->symlink != NULL) { 732484b5c25SMarcel Moolenaar mtree_error("%s: both link and contents keywords " 733484b5c25SMarcel Moolenaar "defined", node->name); 734484b5c25SMarcel Moolenaar return (0); 735484b5c25SMarcel Moolenaar } 736484b5c25SMarcel Moolenaar type = S_IFREG; 73731c0b21fSSimon J. Gerraty } else if (node->type != 0) { 73831c0b21fSSimon J. Gerraty type = node->type; 73931c0b21fSSimon J. Gerraty if (type == S_IFREG) { 74031c0b21fSSimon J. Gerraty /* the named path is the default contents */ 74131c0b21fSSimon J. Gerraty node->contents = mtree_file_path(node); 74231c0b21fSSimon J. Gerraty } 743484b5c25SMarcel Moolenaar } else 744484b5c25SMarcel Moolenaar type = (node->symlink != NULL) ? S_IFLNK : S_IFDIR; 745484b5c25SMarcel Moolenaar 746484b5c25SMarcel Moolenaar if (node->type == 0) 747484b5c25SMarcel Moolenaar node->type = type; 748484b5c25SMarcel Moolenaar 749484b5c25SMarcel Moolenaar if (node->type != type) { 750484b5c25SMarcel Moolenaar mtree_error("%s: file type and defined keywords to not match", 751484b5c25SMarcel Moolenaar node->name); 752484b5c25SMarcel Moolenaar return (0); 753484b5c25SMarcel Moolenaar } 754484b5c25SMarcel Moolenaar 755484b5c25SMarcel Moolenaar st->st_mode = (st->st_mode & ~S_IFMT) | node->type; 756484b5c25SMarcel Moolenaar 757484b5c25SMarcel Moolenaar if (node->contents == NULL) 758484b5c25SMarcel Moolenaar return (0); 759484b5c25SMarcel Moolenaar 760484b5c25SMarcel Moolenaar name = mtree_resolve(node->contents, &istemp); 761484b5c25SMarcel Moolenaar if (name == NULL) 762484b5c25SMarcel Moolenaar return (errno); 763484b5c25SMarcel Moolenaar 764484b5c25SMarcel Moolenaar if (stat(name, &sb) != 0) { 765484b5c25SMarcel Moolenaar mtree_error("%s: contents file '%s' not found", node->name, 766484b5c25SMarcel Moolenaar name); 767484b5c25SMarcel Moolenaar free(name); 768484b5c25SMarcel Moolenaar return (0); 769484b5c25SMarcel Moolenaar } 770484b5c25SMarcel Moolenaar 7711f0ed3c0SBrooks Davis /* 7721f0ed3c0SBrooks Davis * Check for hardlinks. If the contents key is used, then the check 7731f0ed3c0SBrooks Davis * will only trigger if the contents file is a link even if it is used 7741f0ed3c0SBrooks Davis * by more than one file 7751f0ed3c0SBrooks Davis */ 7761f0ed3c0SBrooks Davis if (sb.st_nlink > 1) { 7771f0ed3c0SBrooks Davis fsinode *curino; 7781f0ed3c0SBrooks Davis 7791f0ed3c0SBrooks Davis st->st_ino = sb.st_ino; 7801f0ed3c0SBrooks Davis st->st_dev = sb.st_dev; 7811f0ed3c0SBrooks Davis curino = link_check(node->inode); 7821f0ed3c0SBrooks Davis if (curino != NULL) { 7831f0ed3c0SBrooks Davis free(node->inode); 7841f0ed3c0SBrooks Davis node->inode = curino; 7851f0ed3c0SBrooks Davis node->inode->nlink++; 786*12ad8bdbSAlex Richardson /* Reset st since node->inode has been updated. */ 787*12ad8bdbSAlex Richardson st = &node->inode->st; 7881f0ed3c0SBrooks Davis } 7891f0ed3c0SBrooks Davis } 7901f0ed3c0SBrooks Davis 791484b5c25SMarcel Moolenaar free(node->contents); 792484b5c25SMarcel Moolenaar node->contents = name; 793484b5c25SMarcel Moolenaar st->st_size = sb.st_size; 794484b5c25SMarcel Moolenaar return (0); 795484b5c25SMarcel Moolenaar } 796484b5c25SMarcel Moolenaar 797484b5c25SMarcel Moolenaar static int 798484b5c25SMarcel Moolenaar read_mtree_command(FILE *fp) 799484b5c25SMarcel Moolenaar { 800484b5c25SMarcel Moolenaar char cmd[10]; 801484b5c25SMarcel Moolenaar int error; 802484b5c25SMarcel Moolenaar 803484b5c25SMarcel Moolenaar error = read_word(fp, cmd, sizeof(cmd)); 804484b5c25SMarcel Moolenaar if (error) 805484b5c25SMarcel Moolenaar goto out; 806484b5c25SMarcel Moolenaar 807484b5c25SMarcel Moolenaar error = read_mtree_keywords(fp, &mtree_global); 808484b5c25SMarcel Moolenaar 809484b5c25SMarcel Moolenaar out: 810484b5c25SMarcel Moolenaar skip_to(fp, "\n"); 811484b5c25SMarcel Moolenaar (void)getc(fp); 812484b5c25SMarcel Moolenaar return (error); 813484b5c25SMarcel Moolenaar } 814484b5c25SMarcel Moolenaar 815484b5c25SMarcel Moolenaar static int 816484b5c25SMarcel Moolenaar read_mtree_spec1(FILE *fp, bool def, const char *name) 817484b5c25SMarcel Moolenaar { 818484b5c25SMarcel Moolenaar fsnode *last, *node, *parent; 819484b5c25SMarcel Moolenaar u_int type; 820484b5c25SMarcel Moolenaar int error; 821484b5c25SMarcel Moolenaar 822484b5c25SMarcel Moolenaar assert(name[0] != '\0'); 823484b5c25SMarcel Moolenaar 824484b5c25SMarcel Moolenaar /* 825484b5c25SMarcel Moolenaar * Treat '..' specially, because it only changes our current 826484b5c25SMarcel Moolenaar * directory. We don't create a node for it. We simply ignore 827484b5c25SMarcel Moolenaar * any keywords that may appear on the line as well. 828484b5c25SMarcel Moolenaar * Going up a directory is a little non-obvious. A directory 829484b5c25SMarcel Moolenaar * node has a corresponding '.' child. The parent of '.' is 830484b5c25SMarcel Moolenaar * not the '.' node of the parent directory, but the directory 831484b5c25SMarcel Moolenaar * node within the parent to which the child relates. However, 832484b5c25SMarcel Moolenaar * going up a directory means we need to find the '.' node to 833484b5c25SMarcel Moolenaar * which the directoy node is linked. This we can do via the 834484b5c25SMarcel Moolenaar * first * pointer, because '.' is always the first entry in a 835484b5c25SMarcel Moolenaar * directory. 836484b5c25SMarcel Moolenaar */ 837484b5c25SMarcel Moolenaar if (IS_DOTDOT(name)) { 838484b5c25SMarcel Moolenaar /* This deals with NULL pointers as well. */ 839484b5c25SMarcel Moolenaar if (mtree_current == mtree_root) { 840484b5c25SMarcel Moolenaar mtree_warning("ignoring .. in root directory"); 841484b5c25SMarcel Moolenaar return (0); 842484b5c25SMarcel Moolenaar } 843484b5c25SMarcel Moolenaar 844484b5c25SMarcel Moolenaar node = mtree_current; 845484b5c25SMarcel Moolenaar 846484b5c25SMarcel Moolenaar assert(node != NULL); 847484b5c25SMarcel Moolenaar assert(IS_DOT(node->name)); 848484b5c25SMarcel Moolenaar assert(node->first == node); 849484b5c25SMarcel Moolenaar 850484b5c25SMarcel Moolenaar /* Get the corresponding directory node in the parent. */ 851484b5c25SMarcel Moolenaar node = mtree_current->parent; 852484b5c25SMarcel Moolenaar 853484b5c25SMarcel Moolenaar assert(node != NULL); 854484b5c25SMarcel Moolenaar assert(!IS_DOT(node->name)); 855484b5c25SMarcel Moolenaar 856484b5c25SMarcel Moolenaar node = node->first; 857484b5c25SMarcel Moolenaar 858484b5c25SMarcel Moolenaar assert(node != NULL); 859484b5c25SMarcel Moolenaar assert(IS_DOT(node->name)); 860484b5c25SMarcel Moolenaar assert(node->first == node); 861484b5c25SMarcel Moolenaar 862484b5c25SMarcel Moolenaar mtree_current = node; 863484b5c25SMarcel Moolenaar return (0); 864484b5c25SMarcel Moolenaar } 865484b5c25SMarcel Moolenaar 866484b5c25SMarcel Moolenaar /* 867484b5c25SMarcel Moolenaar * If we don't have a current directory and the first specification 868484b5c25SMarcel Moolenaar * (either implicit or defined) is not '.', then we need to create 869484b5c25SMarcel Moolenaar * a '.' node first (using a recursive call). 870484b5c25SMarcel Moolenaar */ 871484b5c25SMarcel Moolenaar if (!IS_DOT(name) && mtree_current == NULL) { 872484b5c25SMarcel Moolenaar error = read_mtree_spec1(fp, false, "."); 873484b5c25SMarcel Moolenaar if (error) 874484b5c25SMarcel Moolenaar return (error); 875484b5c25SMarcel Moolenaar } 876484b5c25SMarcel Moolenaar 877484b5c25SMarcel Moolenaar /* 878484b5c25SMarcel Moolenaar * Lookup the name in the current directory (if we have a current 879484b5c25SMarcel Moolenaar * directory) to make sure we do not create multiple nodes for the 880484b5c25SMarcel Moolenaar * same component. For non-definitions, if we find a node with the 881484b5c25SMarcel Moolenaar * same name, simply change the current directory. For definitions 882484b5c25SMarcel Moolenaar * more happens. 883484b5c25SMarcel Moolenaar */ 884484b5c25SMarcel Moolenaar last = NULL; 885484b5c25SMarcel Moolenaar node = mtree_current; 886484b5c25SMarcel Moolenaar while (node != NULL) { 887484b5c25SMarcel Moolenaar assert(node->first == mtree_current); 888484b5c25SMarcel Moolenaar 889484b5c25SMarcel Moolenaar if (strcmp(name, node->name) == 0) { 890484b5c25SMarcel Moolenaar if (def == true) { 891b0d9addeSBrooks Davis if (!dupsok) 892b0d9addeSBrooks Davis mtree_error( 893b0d9addeSBrooks Davis "duplicate definition of %s", 894b0d9addeSBrooks Davis name); 895b0d9addeSBrooks Davis else 896b0d9addeSBrooks Davis mtree_warning( 897b0d9addeSBrooks Davis "duplicate definition of %s", 898484b5c25SMarcel Moolenaar name); 899484b5c25SMarcel Moolenaar return (0); 900484b5c25SMarcel Moolenaar } 901484b5c25SMarcel Moolenaar 902484b5c25SMarcel Moolenaar if (node->type != S_IFDIR) { 903484b5c25SMarcel Moolenaar mtree_error("%s is not a directory", name); 904484b5c25SMarcel Moolenaar return (0); 905484b5c25SMarcel Moolenaar } 906484b5c25SMarcel Moolenaar 907484b5c25SMarcel Moolenaar assert(!IS_DOT(name)); 908484b5c25SMarcel Moolenaar 909484b5c25SMarcel Moolenaar node = node->child; 910484b5c25SMarcel Moolenaar 911484b5c25SMarcel Moolenaar assert(node != NULL); 912484b5c25SMarcel Moolenaar assert(IS_DOT(node->name)); 913484b5c25SMarcel Moolenaar 914484b5c25SMarcel Moolenaar mtree_current = node; 915484b5c25SMarcel Moolenaar return (0); 916484b5c25SMarcel Moolenaar } 917484b5c25SMarcel Moolenaar 918484b5c25SMarcel Moolenaar last = node; 919484b5c25SMarcel Moolenaar node = last->next; 920484b5c25SMarcel Moolenaar } 921484b5c25SMarcel Moolenaar 922484b5c25SMarcel Moolenaar parent = (mtree_current != NULL) ? mtree_current->parent : NULL; 923484b5c25SMarcel Moolenaar type = (def == false || IS_DOT(name)) ? S_IFDIR : 0; 924484b5c25SMarcel Moolenaar node = create_node(name, type, parent, &mtree_global); 925484b5c25SMarcel Moolenaar if (node == NULL) 926484b5c25SMarcel Moolenaar return (ENOMEM); 927484b5c25SMarcel Moolenaar 928484b5c25SMarcel Moolenaar if (def == true) { 929484b5c25SMarcel Moolenaar error = read_mtree_keywords(fp, node); 930484b5c25SMarcel Moolenaar if (error) { 931484b5c25SMarcel Moolenaar destroy_node(node); 932484b5c25SMarcel Moolenaar return (error); 933484b5c25SMarcel Moolenaar } 934484b5c25SMarcel Moolenaar } 935484b5c25SMarcel Moolenaar 936484b5c25SMarcel Moolenaar node->first = (mtree_current != NULL) ? mtree_current : node; 937484b5c25SMarcel Moolenaar 938484b5c25SMarcel Moolenaar if (last != NULL) 939484b5c25SMarcel Moolenaar last->next = node; 940484b5c25SMarcel Moolenaar 941484b5c25SMarcel Moolenaar if (node->type != S_IFDIR) 942484b5c25SMarcel Moolenaar return (0); 943484b5c25SMarcel Moolenaar 944484b5c25SMarcel Moolenaar if (!IS_DOT(node->name)) { 945484b5c25SMarcel Moolenaar parent = node; 946484b5c25SMarcel Moolenaar node = create_node(".", S_IFDIR, parent, parent); 947484b5c25SMarcel Moolenaar if (node == NULL) { 948484b5c25SMarcel Moolenaar last->next = NULL; 949484b5c25SMarcel Moolenaar destroy_node(parent); 950484b5c25SMarcel Moolenaar return (ENOMEM); 951484b5c25SMarcel Moolenaar } 952484b5c25SMarcel Moolenaar parent->child = node; 953484b5c25SMarcel Moolenaar node->first = node; 954484b5c25SMarcel Moolenaar } 955484b5c25SMarcel Moolenaar 956484b5c25SMarcel Moolenaar assert(node != NULL); 957484b5c25SMarcel Moolenaar assert(IS_DOT(node->name)); 958484b5c25SMarcel Moolenaar assert(node->first == node); 959484b5c25SMarcel Moolenaar 960484b5c25SMarcel Moolenaar mtree_current = node; 961484b5c25SMarcel Moolenaar if (mtree_root == NULL) 962484b5c25SMarcel Moolenaar mtree_root = node; 963484b5c25SMarcel Moolenaar 964484b5c25SMarcel Moolenaar return (0); 965484b5c25SMarcel Moolenaar } 966484b5c25SMarcel Moolenaar 967484b5c25SMarcel Moolenaar static int 968484b5c25SMarcel Moolenaar read_mtree_spec(FILE *fp) 969484b5c25SMarcel Moolenaar { 97088e2cc9eSBenno Rice char pathspec[PATH_MAX], pathtmp[4*PATH_MAX + 1]; 971484b5c25SMarcel Moolenaar char *cp; 972484b5c25SMarcel Moolenaar int error; 973484b5c25SMarcel Moolenaar 97488e2cc9eSBenno Rice error = read_word(fp, pathtmp, sizeof(pathtmp)); 975484b5c25SMarcel Moolenaar if (error) 976484b5c25SMarcel Moolenaar goto out; 97788e2cc9eSBenno Rice if (strnunvis(pathspec, PATH_MAX, pathtmp) == -1) { 97888e2cc9eSBenno Rice error = errno; 97988e2cc9eSBenno Rice goto out; 98088e2cc9eSBenno Rice } 98188e2cc9eSBenno Rice error = 0; 982484b5c25SMarcel Moolenaar 983484b5c25SMarcel Moolenaar cp = strchr(pathspec, '/'); 984484b5c25SMarcel Moolenaar if (cp != NULL) { 985484b5c25SMarcel Moolenaar /* Absolute pathname */ 986484b5c25SMarcel Moolenaar mtree_current = mtree_root; 987484b5c25SMarcel Moolenaar 988484b5c25SMarcel Moolenaar do { 989484b5c25SMarcel Moolenaar *cp++ = '\0'; 990484b5c25SMarcel Moolenaar 99117b3e6f7SBrooks Davis /* Disallow '..' as a component. */ 99217b3e6f7SBrooks Davis if (IS_DOTDOT(pathspec)) { 99317b3e6f7SBrooks Davis mtree_error("absolute path cannot contain " 99417b3e6f7SBrooks Davis ".. component"); 995484b5c25SMarcel Moolenaar goto out; 996484b5c25SMarcel Moolenaar } 997484b5c25SMarcel Moolenaar 99817b3e6f7SBrooks Davis /* Ignore multiple adjacent slashes and '.'. */ 99917b3e6f7SBrooks Davis if (pathspec[0] != '\0' && !IS_DOT(pathspec)) 1000484b5c25SMarcel Moolenaar error = read_mtree_spec1(fp, false, pathspec); 1001484b5c25SMarcel Moolenaar memmove(pathspec, cp, strlen(cp) + 1); 1002484b5c25SMarcel Moolenaar cp = strchr(pathspec, '/'); 1003484b5c25SMarcel Moolenaar } while (!error && cp != NULL); 1004484b5c25SMarcel Moolenaar 1005484b5c25SMarcel Moolenaar /* Disallow '.' and '..' as the last component. */ 1006484b5c25SMarcel Moolenaar if (!error && (IS_DOT(pathspec) || IS_DOTDOT(pathspec))) { 1007484b5c25SMarcel Moolenaar mtree_error("absolute path cannot contain . or .. " 1008484b5c25SMarcel Moolenaar "components"); 1009484b5c25SMarcel Moolenaar goto out; 1010484b5c25SMarcel Moolenaar } 1011484b5c25SMarcel Moolenaar } 1012484b5c25SMarcel Moolenaar 1013484b5c25SMarcel Moolenaar /* Ignore absolute specfications that end with a slash. */ 1014484b5c25SMarcel Moolenaar if (!error && pathspec[0] != '\0') 1015484b5c25SMarcel Moolenaar error = read_mtree_spec1(fp, true, pathspec); 1016484b5c25SMarcel Moolenaar 1017484b5c25SMarcel Moolenaar out: 1018484b5c25SMarcel Moolenaar skip_to(fp, "\n"); 1019484b5c25SMarcel Moolenaar (void)getc(fp); 1020484b5c25SMarcel Moolenaar return (error); 1021484b5c25SMarcel Moolenaar } 1022484b5c25SMarcel Moolenaar 1023484b5c25SMarcel Moolenaar fsnode * 1024484b5c25SMarcel Moolenaar read_mtree(const char *fname, fsnode *node) 1025484b5c25SMarcel Moolenaar { 1026484b5c25SMarcel Moolenaar struct mtree_fileinfo *fi; 1027484b5c25SMarcel Moolenaar FILE *fp; 1028484b5c25SMarcel Moolenaar int c, error; 1029484b5c25SMarcel Moolenaar 1030484b5c25SMarcel Moolenaar /* We do not yet support nesting... */ 1031484b5c25SMarcel Moolenaar assert(node == NULL); 1032484b5c25SMarcel Moolenaar 1033484b5c25SMarcel Moolenaar if (strcmp(fname, "-") == 0) 1034484b5c25SMarcel Moolenaar fp = stdin; 1035484b5c25SMarcel Moolenaar else { 1036484b5c25SMarcel Moolenaar fp = fopen(fname, "r"); 1037484b5c25SMarcel Moolenaar if (fp == NULL) 1038484b5c25SMarcel Moolenaar err(1, "Can't open `%s'", fname); 1039484b5c25SMarcel Moolenaar } 1040484b5c25SMarcel Moolenaar 1041484b5c25SMarcel Moolenaar error = mtree_file_push(fname, fp); 1042484b5c25SMarcel Moolenaar if (error) 1043484b5c25SMarcel Moolenaar goto out; 1044484b5c25SMarcel Moolenaar 104583b6064dSEd Maste memset(&mtree_global, 0, sizeof(mtree_global)); 104683b6064dSEd Maste memset(&mtree_global_inode, 0, sizeof(mtree_global_inode)); 1047484b5c25SMarcel Moolenaar mtree_global.inode = &mtree_global_inode; 1048484b5c25SMarcel Moolenaar mtree_global_inode.nlink = 1; 1049a92b1688SMarcel Moolenaar mtree_global_inode.st.st_nlink = 1; 1050484b5c25SMarcel Moolenaar mtree_global_inode.st.st_atime = mtree_global_inode.st.st_ctime = 1051484b5c25SMarcel Moolenaar mtree_global_inode.st.st_mtime = time(NULL); 1052484b5c25SMarcel Moolenaar errors = warnings = 0; 1053484b5c25SMarcel Moolenaar 1054484b5c25SMarcel Moolenaar setgroupent(1); 1055484b5c25SMarcel Moolenaar setpassent(1); 1056484b5c25SMarcel Moolenaar 1057484b5c25SMarcel Moolenaar mtree_root = node; 1058484b5c25SMarcel Moolenaar mtree_current = node; 1059484b5c25SMarcel Moolenaar do { 1060484b5c25SMarcel Moolenaar /* Start of a new line... */ 1061484b5c25SMarcel Moolenaar fi = SLIST_FIRST(&mtree_fileinfo); 1062484b5c25SMarcel Moolenaar fi->line++; 1063484b5c25SMarcel Moolenaar 1064484b5c25SMarcel Moolenaar error = skip_over(fp, " \t"); 1065484b5c25SMarcel Moolenaar if (error) 1066484b5c25SMarcel Moolenaar break; 1067484b5c25SMarcel Moolenaar 1068484b5c25SMarcel Moolenaar c = getc(fp); 1069484b5c25SMarcel Moolenaar if (c == EOF) { 1070484b5c25SMarcel Moolenaar error = ferror(fp) ? errno : -1; 1071484b5c25SMarcel Moolenaar break; 1072484b5c25SMarcel Moolenaar } 1073484b5c25SMarcel Moolenaar 1074484b5c25SMarcel Moolenaar switch (c) { 1075484b5c25SMarcel Moolenaar case '\n': /* empty line */ 1076484b5c25SMarcel Moolenaar error = 0; 1077484b5c25SMarcel Moolenaar break; 1078484b5c25SMarcel Moolenaar case '#': /* comment -- skip to end of line. */ 1079484b5c25SMarcel Moolenaar error = skip_to(fp, "\n"); 1080484b5c25SMarcel Moolenaar if (!error) 1081484b5c25SMarcel Moolenaar (void)getc(fp); 1082484b5c25SMarcel Moolenaar break; 1083484b5c25SMarcel Moolenaar case '/': /* special commands */ 1084484b5c25SMarcel Moolenaar error = read_mtree_command(fp); 1085484b5c25SMarcel Moolenaar break; 1086484b5c25SMarcel Moolenaar default: /* specification */ 1087484b5c25SMarcel Moolenaar ungetc(c, fp); 1088484b5c25SMarcel Moolenaar error = read_mtree_spec(fp); 1089484b5c25SMarcel Moolenaar break; 1090484b5c25SMarcel Moolenaar } 1091484b5c25SMarcel Moolenaar } while (!error); 1092484b5c25SMarcel Moolenaar 1093484b5c25SMarcel Moolenaar endpwent(); 1094484b5c25SMarcel Moolenaar endgrent(); 1095484b5c25SMarcel Moolenaar 1096484b5c25SMarcel Moolenaar if (error <= 0 && (errors || warnings)) { 1097484b5c25SMarcel Moolenaar warnx("%u error(s) and %u warning(s) in mtree manifest", 1098484b5c25SMarcel Moolenaar errors, warnings); 1099484b5c25SMarcel Moolenaar if (errors) 1100484b5c25SMarcel Moolenaar exit(1); 1101484b5c25SMarcel Moolenaar } 1102484b5c25SMarcel Moolenaar 1103484b5c25SMarcel Moolenaar out: 1104484b5c25SMarcel Moolenaar if (error > 0) 1105484b5c25SMarcel Moolenaar errc(1, error, "Error reading mtree file"); 1106484b5c25SMarcel Moolenaar 1107484b5c25SMarcel Moolenaar if (fp != stdin) 1108484b5c25SMarcel Moolenaar fclose(fp); 1109484b5c25SMarcel Moolenaar 1110484b5c25SMarcel Moolenaar if (mtree_root != NULL) 1111484b5c25SMarcel Moolenaar return (mtree_root); 1112484b5c25SMarcel Moolenaar 1113484b5c25SMarcel Moolenaar /* Handle empty specifications. */ 1114484b5c25SMarcel Moolenaar node = create_node(".", S_IFDIR, NULL, &mtree_global); 1115484b5c25SMarcel Moolenaar node->first = node; 1116484b5c25SMarcel Moolenaar return (node); 1117484b5c25SMarcel Moolenaar } 1118