1/* 2 * Copyright (c) 1983, 1993 3 * The Regents of the University of California. All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 4. Neither the name of the University nor the names of its contributors 14 * may be used to endorse or promote products derived from this software 15 * without specific prior written permission. 16 * 17 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 20 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 27 * SUCH DAMAGE. 28 */ 29 30#ifndef lint 31#if 0 32static char sccsid[] = "@(#)utilities.c 8.5 (Berkeley) 4/28/95"; 33#endif 34static const char rcsid[] = 35 "$FreeBSD: releng/10.2/sbin/restore/utilities.c 236213 2012-05-29 01:48:06Z kevlo $"; 36#endif /* not lint */ 37 38#include <sys/param.h> 39#include <sys/stat.h> 40 41#include <ufs/ufs/dinode.h> 42#include <ufs/ufs/dir.h> 43 44#include <errno.h> 45#include <limits.h> 46#include <stdio.h> 47#include <stdlib.h> 48#include <string.h> 49#include <unistd.h> 50 51#include "restore.h" 52#include "extern.h" 53 54/* 55 * Insure that all the components of a pathname exist. 56 */ 57void 58pathcheck(char *name) 59{ 60 char *cp; 61 struct entry *ep; 62 char *start; 63 64 start = strchr(name, '/'); 65 if (start == 0) 66 return; 67 for (cp = start; *cp != '\0'; cp++) { 68 if (*cp != '/') 69 continue; 70 *cp = '\0'; 71 ep = lookupname(name); 72 if (ep == NULL) { 73 /* Safe; we know the pathname exists in the dump. */ 74 ep = addentry(name, pathsearch(name)->d_ino, NODE); 75 newnode(ep); 76 } 77 ep->e_flags |= NEW|KEEP; 78 *cp = '/'; 79 } 80} 81 82/* 83 * Change a name to a unique temporary name. 84 */ 85void 86mktempname(struct entry *ep) 87{ 88 char oldname[MAXPATHLEN]; 89 90 if (ep->e_flags & TMPNAME) 91 badentry(ep, "mktempname: called with TMPNAME"); 92 ep->e_flags |= TMPNAME; 93 (void) strcpy(oldname, myname(ep)); 94 freename(ep->e_name); 95 ep->e_name = savename(gentempname(ep)); 96 ep->e_namlen = strlen(ep->e_name); 97 renameit(oldname, myname(ep)); 98} 99 100/* 101 * Generate a temporary name for an entry. 102 */ 103char * 104gentempname(struct entry *ep) 105{ 106 static char name[MAXPATHLEN]; 107 struct entry *np; 108 long i = 0; 109 110 for (np = lookupino(ep->e_ino); 111 np != NULL && np != ep; np = np->e_links) 112 i++; 113 if (np == NULL) 114 badentry(ep, "not on ino list"); 115 (void) sprintf(name, "%s%ld%lu", TMPHDR, i, (u_long)ep->e_ino); 116 return (name); 117} 118 119/* 120 * Rename a file or directory. 121 */ 122void 123renameit(char *from, char *to) 124{ 125 if (!Nflag && rename(from, to) < 0) { 126 fprintf(stderr, "warning: cannot rename %s to %s: %s\n", 127 from, to, strerror(errno)); 128 return; 129 } 130 vprintf(stdout, "rename %s to %s\n", from, to); 131} 132 133/* 134 * Create a new node (directory). 135 */ 136void 137newnode(struct entry *np) 138{ 139 char *cp; 140 141 if (np->e_type != NODE) 142 badentry(np, "newnode: not a node"); 143 cp = myname(np); 144 if (!Nflag && mkdir(cp, 0777) < 0 && !uflag) { 145 np->e_flags |= EXISTED; 146 fprintf(stderr, "warning: %s: %s\n", cp, strerror(errno)); 147 return; 148 } 149 vprintf(stdout, "Make node %s\n", cp); 150} 151 152/* 153 * Remove an old node (directory). 154 */ 155void 156removenode(struct entry *ep) 157{ 158 char *cp; 159 160 if (ep->e_type != NODE) 161 badentry(ep, "removenode: not a node"); 162 if (ep->e_entries != NULL) 163 badentry(ep, "removenode: non-empty directory"); 164 ep->e_flags |= REMOVED; 165 ep->e_flags &= ~TMPNAME; 166 cp = myname(ep); 167 if (!Nflag && rmdir(cp) < 0) { 168 fprintf(stderr, "warning: %s: %s\n", cp, strerror(errno)); 169 return; 170 } 171 vprintf(stdout, "Remove node %s\n", cp); 172} 173 174/* 175 * Remove a leaf. 176 */ 177void 178removeleaf(struct entry *ep) 179{ 180 char *cp; 181 182 if (ep->e_type != LEAF) 183 badentry(ep, "removeleaf: not a leaf"); 184 ep->e_flags |= REMOVED; 185 ep->e_flags &= ~TMPNAME; 186 cp = myname(ep); 187 if (!Nflag && unlink(cp) < 0) { 188 fprintf(stderr, "warning: %s: %s\n", cp, strerror(errno)); 189 return; 190 } 191 vprintf(stdout, "Remove leaf %s\n", cp); 192} 193 194/* 195 * Create a link. 196 */ 197int 198linkit(char *existing, char *new, int type) 199{ 200 201 /* if we want to unlink first, do it now so *link() won't fail */ 202 if (uflag && !Nflag) 203 (void)unlink(new); 204 205 if (type == SYMLINK) { 206 if (!Nflag && symlink(existing, new) < 0) { 207 fprintf(stderr, 208 "warning: cannot create symbolic link %s->%s: %s\n", 209 new, existing, strerror(errno)); 210 return (FAIL); 211 } 212 } else if (type == HARDLINK) { 213 int ret; 214 215 if (!Nflag && (ret = link(existing, new)) < 0) { 216 struct stat s; 217 218 /* 219 * Most likely, the schg flag is set. Clear the 220 * flags and try again. 221 */ 222 if (stat(existing, &s) == 0 && s.st_flags != 0 && 223 chflags(existing, 0) == 0) { 224 ret = link(existing, new); 225 chflags(existing, s.st_flags); 226 } 227 if (ret < 0) { 228 fprintf(stderr, "warning: cannot create " 229 "hard link %s->%s: %s\n", 230 new, existing, strerror(errno)); 231 return (FAIL); 232 } 233 } 234 } else { 235 panic("linkit: unknown type %d\n", type); 236 return (FAIL); 237 } 238 vprintf(stdout, "Create %s link %s->%s\n", 239 type == SYMLINK ? "symbolic" : "hard", new, existing); 240 return (GOOD); 241} 242 243/* 244 * Create a whiteout. 245 */ 246int 247addwhiteout(char *name) 248{ 249 250 if (!Nflag && mknod(name, S_IFWHT, 0) < 0) { 251 fprintf(stderr, "warning: cannot create whiteout %s: %s\n", 252 name, strerror(errno)); 253 return (FAIL); 254 } 255 vprintf(stdout, "Create whiteout %s\n", name); 256 return (GOOD); 257} 258 259/* 260 * Delete a whiteout. 261 */ 262void 263delwhiteout(struct entry *ep) 264{ 265 char *name; 266 267 if (ep->e_type != LEAF) 268 badentry(ep, "delwhiteout: not a leaf"); 269 ep->e_flags |= REMOVED; 270 ep->e_flags &= ~TMPNAME; 271 name = myname(ep); 272 if (!Nflag && undelete(name) < 0) { 273 fprintf(stderr, "warning: cannot delete whiteout %s: %s\n", 274 name, strerror(errno)); 275 return; 276 } 277 vprintf(stdout, "Delete whiteout %s\n", name); 278} 279 280/* 281 * find lowest number file (above "start") that needs to be extracted 282 */ 283ino_t 284lowerbnd(ino_t start) 285{ 286 struct entry *ep; 287 288 for ( ; start < maxino; start++) { 289 ep = lookupino(start); 290 if (ep == NULL || ep->e_type == NODE) 291 continue; 292 if (ep->e_flags & (NEW|EXTRACT)) 293 return (start); 294 } 295 return (start); 296} 297 298/* 299 * find highest number file (below "start") that needs to be extracted 300 */ 301ino_t 302upperbnd(ino_t start) 303{ 304 struct entry *ep; 305 306 for ( ; start > ROOTINO; start--) { 307 ep = lookupino(start); 308 if (ep == NULL || ep->e_type == NODE) 309 continue; 310 if (ep->e_flags & (NEW|EXTRACT)) 311 return (start); 312 } 313 return (start); 314} 315 316/* 317 * report on a badly formed entry 318 */ 319void 320badentry(struct entry *ep, char *msg) 321{ 322 323 fprintf(stderr, "bad entry: %s\n", msg); 324 fprintf(stderr, "name: %s\n", myname(ep)); 325 fprintf(stderr, "parent name %s\n", myname(ep->e_parent)); 326 if (ep->e_sibling != NULL) 327 fprintf(stderr, "sibling name: %s\n", myname(ep->e_sibling)); 328 if (ep->e_entries != NULL) 329 fprintf(stderr, "next entry name: %s\n", myname(ep->e_entries)); 330 if (ep->e_links != NULL) 331 fprintf(stderr, "next link name: %s\n", myname(ep->e_links)); 332 if (ep->e_next != NULL) 333 fprintf(stderr, 334 "next hashchain name: %s\n", myname(ep->e_next)); 335 fprintf(stderr, "entry type: %s\n", 336 ep->e_type == NODE ? "NODE" : "LEAF"); 337 fprintf(stderr, "inode number: %lu\n", (u_long)ep->e_ino); 338 panic("flags: %s\n", flagvalues(ep)); 339} 340 341/* 342 * Construct a string indicating the active flag bits of an entry. 343 */ 344char * 345flagvalues(struct entry *ep) 346{ 347 static char flagbuf[BUFSIZ]; 348 349 (void) strcpy(flagbuf, "|NIL"); 350 flagbuf[0] = '\0'; 351 if (ep->e_flags & REMOVED) 352 (void) strcat(flagbuf, "|REMOVED"); 353 if (ep->e_flags & TMPNAME) 354 (void) strcat(flagbuf, "|TMPNAME"); 355 if (ep->e_flags & EXTRACT) 356 (void) strcat(flagbuf, "|EXTRACT"); 357 if (ep->e_flags & NEW) 358 (void) strcat(flagbuf, "|NEW"); 359 if (ep->e_flags & KEEP) 360 (void) strcat(flagbuf, "|KEEP"); 361 if (ep->e_flags & EXISTED) 362 (void) strcat(flagbuf, "|EXISTED"); 363 return (&flagbuf[1]); 364} 365 366/* 367 * Check to see if a name is on a dump tape. 368 */ 369ino_t 370dirlookup(const char *name) 371{ 372 struct direct *dp; 373 ino_t ino; 374 375 ino = ((dp = pathsearch(name)) == NULL) ? 0 : dp->d_ino; 376 377 if (ino == 0 || TSTINO(ino, dumpmap) == 0) 378 fprintf(stderr, "%s is not on the tape\n", name); 379 return (ino); 380} 381 382/* 383 * Elicit a reply. 384 */ 385int 386reply(char *question) 387{ 388 int c; 389 390 do { 391 fprintf(stderr, "%s? [yn] ", question); 392 (void) fflush(stderr); 393 c = getc(terminal); 394 while (c != '\n' && getc(terminal) != '\n') 395 if (c == EOF) 396 return (FAIL); 397 } while (c != 'y' && c != 'n'); 398 if (c == 'y') 399 return (GOOD); 400 return (FAIL); 401} 402 403/* 404 * handle unexpected inconsistencies 405 */ 406#include <stdarg.h> 407 408void 409panic(const char *fmt, ...) 410{ 411 va_list ap; 412 va_start(ap, fmt); 413 vfprintf(stderr, fmt, ap); 414 va_end(ap); 415 if (yflag) 416 return; 417 if (reply("abort") == GOOD) { 418 if (reply("dump core") == GOOD) 419 abort(); 420 done(1); 421 } 422} 423