]> git.wh0rd.org - dump.git/blob - restore/main.c
Better comparision (including directories).
[dump.git] / restore / main.c
1 /*
2 * Ported to Linux's Second Extended File System as part of the
3 * dump and restore backup suit
4 * Remy Card <card@Linux.EU.Org>, 1994-1997
5 * Stelian Pop <stelian@popies.net>, 1999-2000
6 * Stelian Pop <stelian@popies.net> - AlcĂ´ve <www.alcove.com>, 2000-2002
7 */
8
9 /*
10 * Copyright (c) 1983, 1993
11 * The Regents of the University of California. All rights reserved.
12 *
13 * Redistribution and use in source and binary forms, with or without
14 * modification, are permitted provided that the following conditions
15 * are met:
16 * 1. Redistributions of source code must retain the above copyright
17 * notice, this list of conditions and the following disclaimer.
18 * 2. Redistributions in binary form must reproduce the above copyright
19 * notice, this list of conditions and the following disclaimer in the
20 * documentation and/or other materials provided with the distribution.
21 * 3. Neither the name of the University nor the names of its contributors
22 * may be used to endorse or promote products derived from this software
23 * without specific prior written permission.
24 *
25 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
26 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
27 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
28 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
29 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
30 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
31 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
32 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
33 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
34 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
35 * SUCH DAMAGE.
36 */
37
38 #ifndef lint
39 static const char rcsid[] =
40 "$Id: main.c,v 1.47 2004/12/15 11:00:01 stelian Exp $";
41 #endif /* not lint */
42
43 #include <config.h>
44 #include <compatlfs.h>
45 #include <sys/types.h>
46 #include <fcntl.h>
47 #include <sys/param.h>
48 #include <sys/stat.h>
49 #include <errno.h>
50
51 #ifdef __linux__
52 #include <sys/time.h>
53 #include <time.h>
54 #ifdef HAVE_EXT2FS_EXT2_FS_H
55 #include <ext2fs/ext2_fs.h>
56 #else
57 #include <linux/ext2_fs.h>
58 #endif
59 #include <bsdcompat.h>
60 #include <signal.h>
61 #include <string.h>
62 #else /* __linux__ */
63 #ifdef sunos
64 #include <signal.h>
65 #include <string.h>
66 #include <sys/fcntl.h>
67 #include <bsdcompat.h>
68 #include <sys/mtio.h>
69 #else
70 #include <ufs/ufs/dinode.h>
71 #endif
72 #endif /* __linux__ */
73 #include <protocols/dumprestore.h>
74
75 #include <compaterr.h>
76 #include <stdio.h>
77 #include <stdlib.h>
78 #include <unistd.h>
79
80 #ifdef __linux__
81 #include <ext2fs/ext2fs.h>
82 #include <getopt.h>
83 #endif
84
85 #include "pathnames.h"
86 #include "restore.h"
87 #include "extern.h"
88
89 int abortifconnerr = 1; /* set to 1 if lib dumprmt.o should exit on connection errors
90 otherwise just print a message using msg */
91
92 int aflag = 0, bflag = 0, cvtflag = 0, dflag = 0, vflag = 0, yflag = 0;
93 int hflag = 1, mflag = 1, Mflag = 0, Nflag = 0, Vflag = 0, zflag = 0;
94 int uflag = 0, lflag = 0, Lflag = 0, oflag = 0;
95 int ufs2flag = 0;
96 char *Afile = NULL;
97 int dokerberos = 0;
98 char command = '\0';
99 long dumpnum = 1;
100 long volno = 0;
101 long ntrec;
102 char *dumpmap = NULL;
103 char *usedinomap = NULL;
104 dump_ino_t maxino;
105 time_t dumptime;
106 time_t dumpdate;
107 FILE *terminal;
108 char *tmpdir;
109 int compare_ignore_not_found;
110 int compare_errors;
111 char filesys[NAMELEN];
112 static const char *stdin_opt = NULL;
113 char *bot_script = NULL;
114 dump_ino_t volinfo[TP_NINOS];
115
116 #ifdef USE_QFA
117 FILE *gTapeposfp;
118 char *gTapeposfile;
119 char gTps[255];
120 long gSeekstart;
121 int tapeposflag;
122 int gTapeposfd;
123 int createtapeposflag;
124 unsigned long qfadumpdate;
125 long long curtapepos;
126 #endif /* USE_QFA */
127
128 long smtc_errno;
129
130 #if defined(__linux__) || defined(sunos)
131 char *__progname;
132 #endif
133
134 static void obsolete __P((int *, char **[]));
135 static void usage __P((void));
136 static void use_stdin __P((const char *));
137
138 #define FORCED_UMASK (077)
139
140 int
141 main(int argc, char *argv[])
142 {
143 int ch;
144 dump_ino_t ino;
145 char *inputdev = _PATH_DEFTAPE;
146 char *symtbl = "./restoresymtable";
147 char *p, name[MAXPATHLEN];
148 FILE *filelist = NULL;
149 char fname[MAXPATHLEN];
150 mode_t orig_umask;
151 #ifdef DEBUG_QFA
152 time_t tistart, tiend, titaken;
153 #endif
154 #ifdef USE_QFA
155 tapeposflag = 0;
156 createtapeposflag = 0;
157 #endif /* USE_QFA */
158
159 /* Temp files should *not* be readable. We set permissions later. */
160 orig_umask = umask(FORCED_UMASK);
161 filesys[0] = '\0';
162 #if defined(__linux__) || defined(sunos)
163 __progname = argv[0];
164 #endif
165
166 if (argc < 2)
167 usage();
168
169 if ((inputdev = getenv("TAPE")) == NULL)
170 inputdev = _PATH_DEFTAPE;
171 if ((tmpdir = getenv("TMPDIR")) == NULL)
172 tmpdir = _PATH_TMP;
173 if ((tmpdir = strdup(tmpdir)) == NULL)
174 err(1, "malloc tmpdir");
175 for (p = tmpdir + strlen(tmpdir) - 1; p >= tmpdir && *p == '/'; p--)
176 ;
177 obsolete(&argc, &argv);
178 while ((ch = getopt(argc, argv,
179 "aA:b:CcdD:f:F:hi"
180 #ifdef KERBEROS
181 "k"
182 #endif
183 "lL:mMNo"
184 #ifdef USE_QFA
185 "P:Q:"
186 #endif
187 "Rrs:tT:uvVxX:y")) != -1)
188 switch(ch) {
189 case 'a':
190 aflag = 1;
191 break;
192 case 'A':
193 Afile = optarg;
194 aflag = 1;
195 break;
196 case 'b':
197 /* Change default tape blocksize. */
198 bflag = 1;
199 ntrec = strtol(optarg, &p, 10);
200 if (*p)
201 errx(1, "illegal blocksize -- %s", optarg);
202 if (ntrec <= 0)
203 errx(1, "block size must be greater than 0");
204 break;
205 case 'c':
206 cvtflag = 1;
207 break;
208 case 'D':
209 strncpy(filesys, optarg, NAMELEN);
210 filesys[NAMELEN - 1] = '\0';
211 break;
212 case 'T':
213 tmpdir = optarg;
214 break;
215 case 'd':
216 dflag = 1;
217 break;
218 case 'f':
219 if( !strcmp(optarg,"-") )
220 use_stdin("-f");
221 inputdev = optarg;
222 break;
223 case 'F':
224 bot_script = optarg;
225 break;
226 case 'h':
227 hflag = 0;
228 break;
229 #ifdef KERBEROS
230 case 'k':
231 dokerberos = 1;
232 break;
233 #endif
234 case 'C':
235 case 'i':
236 #ifdef USE_QFA
237 case 'P':
238 #endif
239 case 'R':
240 case 'r':
241 case 't':
242 case 'x':
243 if (command != '\0')
244 errx(1,
245 "%c and %c options are mutually exclusive",
246 ch, command);
247 command = ch;
248 #ifdef USE_QFA
249 if (ch == 'P') {
250 gTapeposfile = optarg;
251 createtapeposflag = 1;
252 }
253 #endif
254
255 break;
256 case 'l':
257 lflag = 1;
258 break;
259 case 'L':
260 Lflag = strtol(optarg, &p, 10);
261 if (*p)
262 errx(1, "illegal limit -- %s", optarg);
263 if (Lflag < 0)
264 errx(1, "limit must be greater than 0");
265 break;
266 case 'm':
267 mflag = 0;
268 break;
269 case 'M':
270 Mflag = 1;
271 break;
272 case 'N':
273 Nflag = 1;
274 break;
275 case 'o':
276 oflag = 1;
277 break;
278 #ifdef USE_QFA
279 case 'Q':
280 gTapeposfile = optarg;
281 tapeposflag = 1;
282 aflag = 1;
283 break;
284 #endif
285 case 's':
286 /* Dumpnum (skip to) for multifile dump tapes. */
287 dumpnum = strtol(optarg, &p, 10);
288 if (*p)
289 errx(1, "illegal dump number -- %s", optarg);
290 if (dumpnum <= 0)
291 errx(1, "dump number must be greater than 0");
292 break;
293 case 'u':
294 uflag = 1;
295 break;
296 case 'v':
297 vflag = 1;
298 break;
299 case 'V':
300 Vflag = 1;
301 break;
302 case 'X':
303 if( !strcmp(optarg,"-") ) {
304 use_stdin("-X");
305 filelist = stdin;
306 }
307 else
308 if ( !(filelist = fopen(optarg,"r")) )
309 errx(1, "can't open file for reading -- %s", optarg);
310 break;
311 case 'y':
312 yflag = 1;
313 break;
314 default:
315 usage();
316 }
317 argc -= optind;
318 argv += optind;
319
320 if (command == '\0')
321 errx(1, "none of C, i, R, r, t or x options specified");
322
323 #ifdef USE_QFA
324 if (!mflag && tapeposflag)
325 errx(1, "m and Q options are mutually exclusive");
326
327 if (tapeposflag && command != 'i' && command != 'x' && command != 't')
328 errx(1, "Q option is not valid for %c command", command);
329 #endif
330
331 if (Afile && command != 'i' && command != 'x' && command != 't')
332 errx(1, "A option is not valid for %c command", command);
333
334 if (signal(SIGINT, onintr) == SIG_IGN)
335 (void) signal(SIGINT, SIG_IGN);
336 if (signal(SIGTERM, onintr) == SIG_IGN)
337 (void) signal(SIGTERM, SIG_IGN);
338 setlinebuf(stderr);
339
340 atexit(cleanup);
341
342 if (command == 'C' && inputdev[0] != '/' && strcmp(inputdev, "-")
343 #ifdef RRESTORE
344 && !strchr(inputdev, ':')
345 #endif
346 ) {
347 /* since we chdir into the directory we are comparing
348 * to, we must retain the full tape path */
349 char wd[MAXPATHLEN], fullpathinput[MAXPATHLEN];
350 if (!getcwd(wd, MAXPATHLEN))
351 err(1, "can't get current directory");
352 snprintf(fullpathinput, MAXPATHLEN, "%s/%s", wd, inputdev);
353 fullpathinput[MAXPATHLEN - 1] = '\0';
354 setinput(fullpathinput);
355 }
356 else
357 setinput(inputdev);
358
359 if (argc == 0 && !filelist) {
360 argc = 1;
361 *--argv = ".";
362 }
363
364 #ifdef USE_QFA
365 if (tapeposflag) {
366 msg("reading QFA positions from %s\n", gTapeposfile);
367 if ((gTapeposfp = fopen(gTapeposfile, "r")) == NULL)
368 errx(1, "can't open file for reading -- %s",
369 gTapeposfile);
370 /* start reading header info */
371 if (fgets(gTps, sizeof(gTps), gTapeposfp) == NULL)
372 errx(1, "not requested format of -- %s", gTapeposfile);
373 gTps[strlen(gTps) - 1] = 0; /* delete end of line */
374 if (strcmp(gTps, QFA_MAGIC) != 0)
375 errx(1, "not requested format of -- %s", gTapeposfile);
376 if (fgets(gTps, sizeof(gTps), gTapeposfp) == NULL)
377 errx(1, "not requested format of -- %s", gTapeposfile);
378 gTps[strlen(gTps) - 1] = 0;
379 if (strcmp(gTps, QFA_VERSION) != 0)
380 errx(1, "not requested format of -- %s", gTapeposfile);
381 /* read dumpdate */
382 if (fgets(gTps, sizeof(gTps), gTapeposfp) == NULL)
383 errx(1, "not requested format of -- %s", gTapeposfile);
384 gTps[strlen(gTps) - 1] = 0;
385 qfadumpdate = atol(gTps);
386 /* read empty line */
387 if (fgets(gTps, sizeof(gTps), gTapeposfp) == NULL)
388 errx(1, "not requested format of -- %s", gTapeposfile);
389 gTps[strlen(gTps) - 1] = 0;
390 /* read table header line */
391 if (fgets(gTps, sizeof(gTps), gTapeposfp) == NULL)
392 errx(1, "not requested format of -- %s", gTapeposfile);
393 gTps[strlen(gTps) - 1] = 0;
394 /* end reading header info */
395 /* tape position table starts here */
396 gSeekstart = ftell(gTapeposfp); /* remember for later use */
397 #ifdef sunos
398 if (GetSCSIIDFromPath(inputdev, &scsiid)) {
399 errx(1, "can't get SCSI-ID for %s\n", inputdev);
400 }
401 if (scsiid < 0) {
402 errx(1, "can't get SCSI-ID for %s\n", inputdev);
403 }
404 sprintf(smtcpath, "/dev/rsmtc%ld,0", scsiid);
405 if ((fdsmtc = open(smtcpath, O_RDWR)) == -1) {
406 errx(1, "can't open smtc device: %s, %d\n", smtcpath, errno);
407 }
408 #endif
409 }
410 #endif /* USE_QFA */
411
412 switch (command) {
413 /*
414 * Compare contents of tape.
415 */
416 case 'C': {
417 struct STAT stbuf;
418
419 Vprintf(stdout, "Begin compare restore\n");
420 compare_ignore_not_found = 0;
421 compare_errors = 0;
422 setup();
423 printf("filesys = %s\n", filesys);
424 if (STAT(filesys, &stbuf) < 0)
425 err(1, "cannot stat directory %s", filesys);
426 if (chdir(filesys) < 0)
427 err(1, "cannot cd to %s", filesys);
428 compare_ignore_not_found = dumptime > 0;
429 initsymtable((char *)0);
430 extractdirs(1);
431 treescan(".", ROOTINO, nodeupdates);
432 compareleaves();
433 comparedirmodes();
434 checkrestore();
435 if (compare_errors) {
436 printf("Some files were modified!\n");
437 exit(2);
438 }
439 break;
440 }
441
442 /*
443 * Interactive mode.
444 */
445 case 'i':
446 setup();
447 extractdirs(1);
448 initsymtable(NULL);
449 runcmdshell();
450 break;
451 /*
452 * Incremental restoration of a file system.
453 */
454 case 'r':
455 aflag = 1; /* in -r or -R mode, -a is default */
456 setup();
457 if (dumptime > 0) {
458 /*
459 * This is an incremental dump tape.
460 */
461 Vprintf(stdout, "Begin incremental restore\n");
462 initsymtable(symtbl);
463 extractdirs(1);
464 removeoldleaves();
465 Vprintf(stdout, "Calculate node updates.\n");
466 treescan(".", ROOTINO, nodeupdates);
467 findunreflinks();
468 removeoldnodes();
469 } else {
470 /*
471 * This is a level zero dump tape.
472 */
473 Vprintf(stdout, "Begin level 0 restore\n");
474 initsymtable((char *)0);
475 extractdirs(1);
476 Vprintf(stdout, "Calculate extraction list.\n");
477 treescan(".", ROOTINO, nodeupdates);
478 }
479 createleaves(symtbl);
480 createlinks();
481 setdirmodes(FORCE);
482 checkrestore();
483 if (dflag) {
484 Vprintf(stdout, "Verify the directory structure\n");
485 treescan(".", ROOTINO, verifyfile);
486 }
487 dumpsymtable(symtbl, (long)1);
488 break;
489 /*
490 * Resume an incremental file system restoration.
491 */
492 case 'R':
493 aflag = 1; /* in -r or -R mode, -a is default */
494 initsymtable(symtbl);
495 skipmaps();
496 skipdirs();
497 createleaves(symtbl);
498 createlinks();
499 setdirmodes(FORCE);
500 checkrestore();
501 dumpsymtable(symtbl, (long)1);
502 break;
503
504 /* handle file names from either text file (-X) or the command line */
505 #define NEXTFILE(p) \
506 p = NULL; \
507 if (argc) { \
508 --argc; \
509 p = *argv++; \
510 } \
511 else if (filelist) { \
512 if ((p = fgets(fname, MAXPATHLEN, filelist))) { \
513 if ( *p && *(p + strlen(p) - 1) == '\n' ) /* possible null string */ \
514 *(p + strlen(p) - 1) = '\0'; \
515 if ( !*p ) /* skip empty lines */ \
516 continue; \
517 } \
518 }
519
520 /*
521 * List contents of tape.
522 */
523 case 't':
524 setup();
525 extractdirs(0);
526 initsymtable((char *)0);
527 printvolinfo();
528 for (;;) {
529 NEXTFILE(p);
530 if (!p)
531 break;
532 canon(p, name, sizeof(name));
533 ino = dirlookup(name);
534 if (ino == 0)
535 continue;
536 treescan(name, ino, listfile);
537 }
538 break;
539 /*
540 * Batch extraction of tape contents.
541 */
542 case 'x':
543 #ifdef DEBUG_QFA
544 tistart = time(NULL);
545 #endif
546 setup();
547 extractdirs(1);
548 initsymtable((char *)0);
549 for (;;) {
550 NEXTFILE(p);
551 if (!p)
552 break;
553 canon(p, name, sizeof(name));
554 ino = dirlookup(name);
555 if (ino == 0)
556 continue;
557 if (mflag)
558 pathcheck(name);
559 treescan(name, ino, addfile);
560 }
561 createfiles();
562 createlinks();
563 setdirmodes(oflag ? FORCE : 0);
564 if (dflag)
565 checkrestore();
566 #ifdef sunos
567 if (fdsmtc != -1) {
568 close(fdsmtc);
569 }
570 #endif /* sunos */
571 #ifdef DEBUG_QFA
572 tiend = time(NULL);
573 titaken = tiend - tistart;
574 msg("restore took %d:%02d:%02d\n", titaken / 3600,
575 (titaken % 3600) / 60, titaken % 60);
576 #endif /* DEBUG_QFA */
577 break;
578 #ifdef USE_QFA
579 case 'P':
580 #ifdef DEBUG_QFA
581 tistart = time(NULL);
582 #endif
583 #ifdef sunos
584 if (GetSCSIIDFromPath(inputdev, &scsiid)) {
585 errx(1, "can't get SCSI-ID for %s\n", inputdev);
586 }
587 if (scsiid < 0) {
588 errx(1, "can't get SCSI-ID for %s\n", inputdev);
589 }
590 sprintf(smtcpath, "/dev/rsmtc%ld,0", scsiid);
591 if ((fdsmtc = open(smtcpath, O_RDWR)) == -1) {
592 errx(1, "can't open smtc device: %s, %d\n", smtcpath, errno);
593 }
594 #endif /* sunos */
595 setup();
596 msg("writing QFA positions to %s\n", gTapeposfile);
597 (void) umask(orig_umask);
598 if ((gTapeposfd = open(gTapeposfile, O_WRONLY|O_CREAT|O_TRUNC,
599 S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP
600 |S_IROTH|S_IWOTH)) < 0)
601 errx(1, "can't create tapeposfile\n");
602 (void) umask(FORCED_UMASK);
603 /* print QFA-file header */
604 sprintf(gTps, "%s\n%s\n%ld\n\n", QFA_MAGIC, QFA_VERSION, (unsigned long)spcl.c_date);
605 if (write(gTapeposfd, gTps, strlen(gTps)) != (ssize_t)strlen(gTps))
606 errx(1, "can't write tapeposfile\n");
607 sprintf(gTps, "ino\ttapeno\ttapepos\n");
608 if (write(gTapeposfd, gTps, strlen(gTps)) != (ssize_t)strlen(gTps))
609 errx(1, "can't write tapeposfile\n");
610
611 extractdirs(1);
612 initsymtable((char *)0);
613 for (;;) {
614 NEXTFILE(p);
615 if (!p)
616 break;
617 canon(p, name, sizeof(name));
618 ino = dirlookup(name);
619 if (ino == 0)
620 continue;
621 if (mflag)
622 pathcheck(name);
623 treescan(name, ino, addfile);
624 }
625 createfiles();
626 #ifdef sunos
627 if (fdsmtc != -1) {
628 close(fdsmtc);
629 }
630 #endif /* sunos */
631 #ifdef DEBUG_QFA
632 tiend = time(NULL);
633 titaken = tiend - tistart;
634 msg("writing QFA positions took %d:%02d:%02d\n", titaken / 3600,
635 (titaken % 3600) / 60, titaken % 60);
636 #endif /* DEBUG_QFA */
637 break;
638 #endif /* USE_QFA */
639 }
640 exit(0);
641 /* NOTREACHED */
642 return 0; /* gcc shut up */
643 }
644
645 static void
646 usage(void)
647 {
648 char white[MAXPATHLEN];
649 const char *ext2ver, *ext2date;
650
651 memset(white, ' ', MAXPATHLEN);
652 white[MIN(strlen(__progname), MAXPATHLEN - 1)] = '\0';
653
654 #ifdef __linux__
655 ext2fs_get_library_version(&ext2ver, &ext2date);
656 (void)fprintf(stderr, "%s %s (using libext2fs %s of %s)\n",
657 __progname, _DUMP_VERSION, ext2ver, ext2date);
658 #else
659 (void)fprintf(stderr, "%s %s\n", __progname, _DUMP_VERSION);
660 #endif
661
662 #ifdef KERBEROS
663 #define kerbflag "k"
664 #else
665 #define kerbflag
666 #endif
667
668 #ifdef USE_QFA
669 #define qfaflag "[-Q file] "
670 #else
671 #define qfaflag
672 #endif
673
674 fprintf(stderr,
675 "usage:"
676 "\t%s -C [-cd" kerbflag "lMvVy] [-b blocksize] [-D filesystem] [-f file]\n"
677 "\t%s [-F script] [-L limit] [-s fileno]\n"
678 "\t%s -i [-acdh" kerbflag "lmMouvVy] [-A file] [-b blocksize] [-f file]\n"
679 "\t%s [-F script] " qfaflag "[-s fileno]\n"
680 #ifdef USE_QFA
681 "\t%s -P file [-acdh" kerbflag "lmMuvVy] [-A file] [-b blocksize]\n"
682 "\t%s [-f file] [-F script] [-s fileno] [-X filelist] [file ...]\n"
683 #endif
684 "\t%s -r [-cd" kerbflag "lMuvVy] [-b blocksize] [-f file] [-F script]\n"
685 "\t%s [-s fileno] [-T directory]\n"
686 "\t%s -R [-cd" kerbflag "lMuvVy] [-b blocksize] [-f file] [-F script]\n"
687 "\t%s [-s fileno] [-T directory]\n"
688 "\t%s -t [-cdh" kerbflag "lMuvVy] [-A file] [-b blocksize] [-f file]\n"
689 "\t%s [-F script] " qfaflag "[-s fileno] [-X filelist] [file ...]\n"
690 "\t%s -x [-acdh" kerbflag "lmMouvVy] [-A file] [-b blocksize] [-f file]\n"
691 "\t%s [-F script] " qfaflag "[-s fileno] [-X filelist] [file ...]\n",
692 __progname, white,
693 __progname, white,
694 #ifdef USE_QFA
695 __progname, white,
696 #endif
697 __progname, white,
698 __progname, white,
699 __progname, white,
700 __progname, white);
701 exit(1);
702 }
703
704 /*
705 * obsolete --
706 * Change set of key letters and ordered arguments into something
707 * getopt(3) will like.
708 */
709 static void
710 obsolete(int *argcp, char **argvp[])
711 {
712 int argc, flags;
713 char *ap, **argv, *flagsp = NULL, **nargv, *p = NULL;
714
715 /* Setup. */
716 argv = *argvp;
717 argc = *argcp;
718
719 /* Return if no arguments or first argument has leading dash. */
720 ap = argv[1];
721 if (argc == 1 || *ap == '-')
722 return;
723
724 /* Allocate space for new arguments. */
725 if ((*argvp = nargv = malloc((argc + 1) * sizeof(char *))) == NULL ||
726 (p = flagsp = malloc(strlen(ap) + 2)) == NULL)
727 err(1, "malloc args");
728
729 *nargv++ = *argv;
730 argv += 2, argc -= 2;
731
732 for (flags = 0; *ap; ++ap) {
733 switch (*ap) {
734 case 'A':
735 case 'b':
736 case 'D':
737 case 'f':
738 case 'F':
739 case 'L':
740 case 'Q':
741 case 's':
742 case 'T':
743 case 'X':
744 if (*argv == NULL) {
745 warnx("option requires an argument -- %c", *ap);
746 usage();
747 }
748 if ((nargv[0] = malloc(strlen(*argv) + 2 + 1)) == NULL)
749 err(1, "malloc arg");
750 nargv[0][0] = '-';
751 nargv[0][1] = *ap;
752 (void)strcpy(&nargv[0][2], *argv);
753 ++argv;
754 ++nargv;
755 break;
756 default:
757 if (!flags) {
758 *p++ = '-';
759 flags = 1;
760 }
761 *p++ = *ap;
762 break;
763 }
764 }
765
766 /* Terminate flags. */
767 if (flags) {
768 *p = '\0';
769 *nargv++ = flagsp;
770 }
771
772 /* Copy remaining arguments. */
773 while ((*nargv++ = *argv++));
774
775 /* Update argument count. */
776 *argcp = nargv - *argvp - 1;
777 }
778
779 /*
780 * use_stdin --
781 * reserve stdin for opt (avoid conflicts)
782 */
783 void
784 use_stdin(const char *opt)
785 {
786 if (stdin_opt)
787 errx(1, "can't handle standard input for both %s and %s",
788 stdin_opt, opt);
789 stdin_opt = opt;
790 }