source: branches/2.2.5/mondo/src/mondoarchive/mondo-cli.c @ 1745

Last change on this file since 1745 was 1745, checked in by Bruno Cornec, 12 years ago
  • Fix a command build format error
  • USB should never be ejected
  • Property svn:keywords set to Id
File size: 32.1 KB
Line 
1/***************************************************************************
2mondo-cli.c
3-------------------
4begin                : Fri Apr 19 16:40:35 EDT 2002
5copyright        : (C) 2002 Mondo  Hugo Rabson
6email                : Hugo Rabson <hugorabson@msn.com>
7edited by            : by Stan Benoit 4/2002
8email                : troff@nakedsoul.org
9cvsid                : $Id: mondo-cli.c 1745 2007-10-30 11:18:10Z bruno $
10 ***************************************************************************/
11
12/***************************************************************************
13 *                                                                         *
14 *   This program is free software; you can redistribute it and/or modify  *
15 *   it under the terms of the GNU General Public License as published by  *
16 *   the Free Software Foundation; either version 2 of the License, or     *
17 *   (at your option) any later version.                                   *
18 *                                                                         *
19 ***************************************************************************/
20
21/***************************************************************************
22UPDATE LOG
23
24
2508/04
26- if user specifies a dumb -T value, abort
27
2807/22
29- handle -Q sensibly (don't abort if only -Q supplied)
30
3107/17
32- better checking of NFS dir's validity
33
3406/19
35- added AUX_VER
36
3704/17
38- added '-b' support
3904/03
40- added star support
41
4202/06
43- fixed "Please give /dev entry" error msg
44
4501/20/2004
46- added 2.6.x-kernel "device = /dev/...." support to param reader
47- better handling of SCSI and /dev entries
48- touch /boot/boot.b if necessary
49
5009/26/2003
51- typo in command-line handling of 'r'
52
5309/25
54- added DVD write support
55
5609/24
57- if tape backup (-t) but user doesn't specify tape drive then
58  make an educated guess using find_tape_device_and_size()
59
6009/18
61- insist on partimagehack being present if -x found
62
6309/16
64- added support for multiple -I's and -E's in call to mondoarchive
65- the '-D' flag doesn't take a value now
66
6707/14
68- fatal error if -E too short
69
7005/14
71- if 'n' and user doesn't have rights to write to output dir
72  then error out
73
7405/05
75- added Joshua Oreman's FreeBSD patches
76
7705/04
78- added Herman Kuster's multi-level bkp patch
79
8005/02
81- write errors to screen, not just to log
82
8304/25
84- added lots of assert()'s and log_OS_error()'s
85
8604/21
87- line 570 --- resolve boot device if softlink
88
8904/08
90- changed a bunch of fprintf(stderr,...)'s to log_it()'s
91- fixed final / removal in retrieve_switches_from_command_line()
92
9304/04
94- added -e
95
9603/15
97- reject relative paths if -d flag (Alessandro Polverini)
98
9901/02/2003
100- added -J flag (to let user specify incoming filelist)
101
10212/13/2002
103- various strcpy() calls replaced with strncpy() calls
104
10512/10
106- added g_loglevel
107
10811/25
109- line 614 --- if commmand-line param ends in '/' then drop that '/'
110
11111/19
112- rewrote finish() to kill processes more softly
113- if user calls with '-l RAW' then
114
11509/01 - 09/30
116- added -N to let user exclude all NFS-related mounts and devices
117- run_program_and_log_output() now takes boolean operator to specify
118  whether it will log its activities in the event of _success_
119
12008/01 - 08/31
121- if no device specified when backing up to tape streamer then assume /dev/st0
122- abort if tape user specifies tape size (just for testing porpoises)
123- tape users no longer need to specify tape size
124- added some comments
125- added bkpinfo->backup_media_type
126- renamed from mondo-archive.c to mondo-cli.c
127
12807/01 - 07/31
129- better check for mkfs.vfat
130- more sanity-checking for -d flag
131- if isodir does not exist then abort
132- removed "the rest are..." comment from log
133- do not permit '-H' and '-t' in the same command line
134- if not '-o' then insist on vfat-friendly kernel
135
13606/01 - 06/30
137- added signal-trapping
138- added '-W' (won't make bootable CD or floppies) flag
139- added missing 'j++' to -s switch's code
140- expanded -s switch
141- added -u switch
142- if dvdrecord exists and cdrecord doesn't then use
143  dvdrecord instead of cdrecord
144
14504/01 - 04/30
146- if CD-ROM is supermounted then unmount, run, and remount
147- replace MONDO_VERSION #define with VERSION from ../config.h
148- if CD-ROM is mounted at start then abort
149- improved homedir-locating code
150- improved "backup will occupy N CD's" calculation a bit
151- added -m (manual CD tray) flag
152
153
154*******************************************************************/
155
156/**
157 * @file
158 * Functions for handling command-line arguments passed to mondoarchive.
159 */
160
161/** @def BOOT_LOADER_CHARS The characters allowed for boot loader on this platform. */
162
163#include <pthread.h>
164#include "../common/my-stuff.h"
165#include "../common/mondostructures.h"
166#include "mondo-cli-EXT.h"
167#include "../common/libmondo.h"
168
169//static char cvsid[] = "$Id: mondo-cli.c 1745 2007-10-30 11:18:10Z bruno $";
170
171extern int g_loglevel;
172extern bool g_text_mode;
173extern bool g_skip_floppies;    ///< Whether to skip the creation of boot disks
174extern char g_startdir[MAX_STR_LEN];    ///< ????? @bug ?????
175extern char g_tmpfs_mountpt[MAX_STR_LEN];
176extern bool g_sigpipe;
177
178/*@ file pointer **************************************************/
179extern FILE *g_tape_stream;
180
181/*@ long long *****************************************************/
182extern long long g_tape_posK;
183
184/*@ long **********************************************************/
185extern long g_noof_sets;
186
187/*@ bool******** **************************************************/
188bool g_debugging = FALSE;       ///< ????? @bug ????? @ingroup globalGroup
189bool g_running_live = FALSE;    ///< ????? @bug ????? @ingroup globalGroup
190extern bool g_cd_recovery;
191
192extern void setup_tmpdir(char *path);
193
194/**
195 * Whether we're restoring from ISOs. Obviously not, since this is the
196 * backup program.
197 * @note You @b MUST declare this variable somewhere in your program if
198 * you use libmondo. Otherwise the link will fail.
199 * @ingroup globalGroup
200 */
201bool g_ISO_restore_mode = FALSE;
202
203
204extern double g_kernel_version;
205extern int g_current_media_number;
206extern pid_t g_main_pid;
207extern char *resolve_softlinks_to_get_to_actual_device_file(char *);
208
209/* Do we use extended attributes and acl ?
210 * By default no, use --acl & --attr options to force their usage */
211char *g_getfacl = NULL;
212char *g_getfattr = NULL;
213
214/* Reference to global bkpinfo */
215extern struct s_bkpinfo *bkpinfo;
216
217/**
218 * @addtogroup cliGroup
219 * @{
220 */
221/**
222 * Populate @p bkpinfo from the command-line parameters stored in @p argc and @p argv.
223 * @param argc The argument count, including the program name; @p argc passed to main().
224 * @param argv The argument vector; @p argv passed to main().
225 * @param bkpinfo The backup information structure to populate.
226 * @return The number of problems with the command line (0 for success).
227 */
228int
229handle_incoming_parameters(int argc, char *argv[])
230{
231    /*@ int *** */
232    int res = 0;
233    int retval = 0;
234    int i = 0, j;
235
236    /*@ buffers *************** */
237    char *tmp;
238    char flag_val[128][MAX_STR_LEN];
239    bool flag_set[128];
240
241    tmp = malloc(9*MAX_STR_LEN);
242    sensibly_set_tmpdir_and_scratchdir();
243    for (i = 0; i < 128; i++) {
244        flag_val[i][0] = '\0';
245        flag_set[i] = FALSE;
246    }
247    //  strcpy (bkpinfo->scratchdir, "/home");
248    for (j = 1; j <= MAX_NOOF_MEDIA; j++) {
249        bkpinfo->media_size[j] = 650;
250    }                           /* default */
251    res =
252        retrieve_switches_from_command_line(argc, argv, flag_val,
253                                            flag_set);
254    retval += res;
255    if (!retval) {
256        res = process_switches(flag_val, flag_set);
257        retval += res;
258    }
259/*
260  if (!retval)
261    {
262*/
263    log_msg(3, "Switches:-");
264    for (i = 0; i < 128; i++) {
265        if (flag_set[i]) {
266            sprintf(tmp, "-%c %s", i, flag_val[i]);
267            log_msg(3, tmp);
268        }
269    }
270//    }
271    sprintf(tmp, "rm -Rf %s/tmp.mondo.*", bkpinfo->tmpdir);
272    paranoid_system(tmp);
273    sprintf(tmp, "rm -Rf %s/mondo.scratch.*", bkpinfo->scratchdir);
274    paranoid_system(tmp);
275    sprintf(bkpinfo->scratchdir + strlen(bkpinfo->scratchdir),
276            "/mondo.scratch.%ld", random() % 32767);
277    sprintf(tmp, "mkdir -p %s/tmpfs", bkpinfo->tmpdir);
278    paranoid_system(tmp);
279    sprintf(tmp, "mkdir -p %s", bkpinfo->scratchdir);
280    paranoid_system(tmp);
281    paranoid_free(tmp);
282    return (retval);
283}
284
285
286
287
288/**
289 * Store the sizespec(s) stored in @p value into @p bkpinfo.
290 * @param bkpinfo The backup information structure; the @c bkpinfo->media_size field will be populated.
291 * @param value The sizespec (e.g. "2g", "40m").
292 * @return 0, always.
293 * @bug Return code not needed.
294 */
295int process_the_s_switch(char *value)
296{
297    int j;
298    char tmp[MAX_STR_LEN], *p, comment[MAX_STR_LEN];
299
300    assert(bkpinfo != NULL);
301    assert(value != NULL);
302
303    bkpinfo->media_size[0] = -1;    /* dummy value */
304    for (j = 1, p = value; j < MAX_NOOF_MEDIA && strchr(p, ',');
305         j++, p = strchr(p, ',') + 1) {
306        strncpy(tmp, p, MAX_STR_LEN);
307        *(strchr(tmp, ',')) = '\0';
308        bkpinfo->media_size[j] = friendly_sizestr_to_sizelong(tmp);
309        sprintf(comment, "media_size[%d] = %ld", j,
310                bkpinfo->media_size[j]);
311        log_msg(3, comment);
312    }
313    for (; j <= MAX_NOOF_MEDIA; j++) {
314        bkpinfo->media_size[j] = friendly_sizestr_to_sizelong(p);
315    }
316//      bkpinfo->media_size[0] = bkpinfo->media_size[MAX_NOOF_MEDIA];
317    for (j = 1; j <= MAX_NOOF_MEDIA; j++) {
318        if (bkpinfo->media_size[j] <= 0) {
319            log_msg(1, "You gave media #%d an invalid size\n", j);
320            return (-1);
321        }
322    }
323    return (0);
324}
325
326
327
328/**
329 * Process mondoarchive's command-line switches.
330 * @param bkpinfo The backup information structure to populate.
331 * @param flag_val An array of the argument passed to each switch (the letter is the index).
332 * If a switch is not set or has no argument, the field in @p flag_val doesn't matter.
333 * @param flag_set An array of <tt>bool</tt>s indexed by switch letter: TRUE if it's set,
334 * FALSE if it's not.
335 * @return The number of problems with the switches, or 0 for success.
336 * @bug Maybe include a list of all switches (inc. intentionally undocumented ones not in the manual!) here?
337 */
338int
339process_switches(char flag_val[128][MAX_STR_LEN], bool flag_set[128])
340{
341
342    /*@ ints *** */
343    int i = 0;
344    int retval = 0;
345    int percent = 0;
346
347    /*@ buffers ** */
348    char *tmp;
349    char *tmp1;
350    char *psz;
351    char *p;
352    char *q;
353
354    long itbs = 0L;
355
356    struct stat buf;
357
358    malloc_string(tmp);
359    malloc_string(psz);
360
361    assert(bkpinfo != NULL);
362    assert(flag_val != NULL);
363    assert(flag_set != NULL);
364
365    bkpinfo->internal_tape_block_size = DEFAULT_INTERNAL_TAPE_BLOCK_SIZE;
366
367/* compulsory */
368    i = flag_set['c'] + flag_set['i'] + flag_set['n'] +
369        flag_set['t'] + flag_set['u'] + flag_set['r'] +
370        flag_set['w'] + flag_set['C'] + flag_set['U'];
371    if (i == 0) {
372        retval++;
373        log_to_screen("You must specify the media type\n");
374    }
375    if (i > 1) {
376        retval++;
377        log_to_screen("Please specify only one media type\n");
378    }
379    if (flag_set['K']) {
380        g_loglevel = atoi(flag_val['K']);
381        if (g_loglevel < 3) {
382            g_loglevel = 3;
383        }
384    }
385    if (flag_set['L'] && flag_set['0']) {
386        retval++;
387        log_to_screen("You cannot have 'no compression' _and_ LZOP.\n");
388    }
389    bkpinfo->backup_data = flag_set['O'];
390    bkpinfo->verify_data = flag_set['V'];
391    if (flag_set['I'] && !bkpinfo->backup_data) {
392        log_to_screen("-I switch is ignored if just verifying");
393    }
394    if (flag_set['E'] && !bkpinfo->backup_data) {
395        log_to_screen("-E switch is ignored if just verifying");
396    }
397
398    if (!find_home_of_exe("afio")) {
399        if (find_home_of_exe("star")) {
400            flag_set['R'] = TRUE;
401            log_msg(1, "Using star instead of afio");
402        } else {
403            fatal_error
404                ("Neither afio nor star is installed. Please install at least one.");
405        }
406    }
407
408    if (flag_set['R']) {
409        bkpinfo->use_star = TRUE;
410        if (flag_set['L']) {
411            fatal_error("You may not use star and lzop at the same time.");
412        }
413        if (!find_home_of_exe("star")) {
414            fatal_error
415                ("Please install 'star' RPM or tarball if you are going to use -R. Thanks.");
416        }
417    }
418    if (flag_set['W']) {
419        bkpinfo->nonbootable_backup = TRUE;
420        log_to_screen("Warning - you have opted for non-bootable backup");
421        if (flag_set['f'] || flag_set['l']) {
422            log_to_screen
423                ("You don't need to specify bootloader or bootdevice");
424        }
425    }
426    if (flag_set['t'] && flag_set['H']) {
427        fatal_error
428            ("Sorry, you may not nuke w/o warning from tape. Drop -H, please.");
429    }
430    if (flag_set['I']) {
431        if (!strcmp(bkpinfo->include_paths, "/")) {
432            log_msg(2, "'/' is pleonastic.");
433            bkpinfo->include_paths[0] = '\0';
434        }
435        if (bkpinfo->include_paths[0]) {
436            strcat(bkpinfo->include_paths, " ");
437        }
438        asprintf(&tmp1, flag_val['I']);
439        p = tmp1;
440        q = tmp1;
441
442        /* Cut the flag_val['I'] in parts containing all paths to test them */
443        while (p != NULL) {
444            q = strchr(p, ' ');
445            if (q != NULL) {
446                *q = '\0';
447                if (stat(p, &buf) != 0) {
448                    log_msg(1, "ERROR ! %s doesn't exist", p);
449                    fatal_error("ERROR ! You specified a directory to include which doesn't exist");
450                }
451                p = q+1 ;
452            } else {
453                if (stat(p, &buf) != 0) {
454                    log_msg(1, "ERROR ! %s doesn't exist", p);
455                    fatal_error("ERROR ! You specified a directory to include which doesn't exist");
456                }
457                p = NULL;
458            }
459        }
460        paranoid_free(tmp1);
461
462        strncpy(bkpinfo->include_paths + strlen(bkpinfo->include_paths),
463                flag_val['I'],
464                4*MAX_STR_LEN - strlen(bkpinfo->include_paths));
465        log_msg(1, "include_paths is now '%s'", bkpinfo->include_paths);
466        if (bkpinfo->include_paths[0] == '-') {
467            retval++;
468            log_to_screen("Please supply a sensible value with '-I'\n");
469        }
470    }
471
472    if (g_kernel_version >= 2.6 && !flag_set['d']
473        && (flag_set['c'] || flag_set['w'])) {
474        fatal_error
475            ("If you are using the 2.6.x kernel, please specify the CD-R(W) device.");
476    }
477
478
479    if (flag_set['J']) {
480        if (flag_set['I']) {
481            retval++;
482            log_to_screen
483                ("Please do not use -J in combination with -I. If you want to make a list of files to backup, that's fine, use -J <filename> but please don't muddy the waters by combining -J with -I. Thanks. :-)");
484        }
485        bkpinfo->make_filelist = FALSE;
486        strcpy(bkpinfo->include_paths, flag_val['J']);
487    }
488    if (flag_set['c'] || flag_set['w'] || flag_set['C'] || flag_set['r']) {
489        if (!flag_set['r'] && g_kernel_version <= 2.5
490            && strstr(flag_val['d'], "/dev/")) {
491            fatal_error
492                ("Please don't give a /dev entry. Give a SCSI node for the parameter of the -d flag.");
493        }
494        if (flag_set['r'] && g_kernel_version <= 2.5
495            && !strstr(flag_val['d'], "/dev/")) {
496            fatal_error
497                ("Please give a /dev entry, not a SCSI node, as the parameter of the -d flag.");
498        }
499        if (g_kernel_version >= 2.6 && !strstr(flag_val['d'], "/dev/")) {
500            log_to_screen
501                ("Linus says 2.6 has a broken ide-scsi module. Proceed at your own risk...");
502        }
503
504        if (system("which cdrecord > /dev/null 2> /dev/null")
505            && system("which dvdrecord > /dev/null 2> /dev/null")) {
506            fatal_error
507                ("Please install dvdrecord/cdrecord and try again.");
508        }
509        if (flag_set['C']) {
510            bkpinfo->cdrw_speed = atoi(flag_val['C']);
511            if (bkpinfo->cdrw_speed < 1) {
512                fatal_error
513                    ("You specified a silly speed for a CD-R[W] drive");
514            }
515            if (!flag_set['L']) {
516                log_to_screen
517                    ("You must use -L with -C. Therefore I am setting it for you.");
518                flag_set['L'] = 1;
519                flag_val['L'][0] = '\0';
520            }
521        } else {
522            log_msg(3, "flag_val['c'] = %s", flag_val['c']);
523            log_msg(3, "flag_val['w'] = %s", flag_val['w']);
524//    log_msg(3, "flag_set['r'] = %i", flag_set['r'] );
525            if (flag_set['c']) {
526                bkpinfo->cdrw_speed = atoi(flag_val['c']);
527            } else if (flag_set['w']) {
528                bkpinfo->cdrw_speed = atoi(flag_val['w']);
529            } else if (flag_set['r']) {
530                bkpinfo->cdrw_speed = 1;    /*atoi(flag_val['r']); */
531            }
532
533            if (bkpinfo->cdrw_speed < 1) {
534                fatal_error
535                    ("You specified a silly speed for a CD-R[W] drive");
536            }
537        }
538    }
539    if (flag_set['t'] && !flag_set['d']) {
540        log_it("Hmm! No tape drive specified. Let's see what we can do.");
541        if (find_tape_device_and_size(flag_val['d'], tmp)) {
542            fatal_error
543                ("Tape device not specified. I couldn't find it either.");
544        }
545        flag_set['d'] = TRUE;
546        sprintf(tmp,
547                "You didn't specify a tape streamer device. I'm assuming %s",
548                flag_val['d']);
549        log_to_screen(tmp);
550        percent = 0;
551    }
552
553    if (flag_set['U'])          // USB
554    {
555        if (! flag_set['d']) {
556            fatal_error
557                ("You need to specify a device file with -d for bootable USB device creation");
558        }
559        if (!flag_set['s']) {
560            fatal_error("You did not specify a size (-s) for your USB device. Aborting");
561        }
562    }
563
564    if (flag_set['r'])          // DVD
565    {
566        if (flag_set['m']) {
567            fatal_error
568                ("Manual CD tray (-m) not yet supported in conjunction w/ DVD drives. Drop -m.");
569        }
570        if (!flag_set['d']) {
571            if (!find_dvd_device(flag_val['d'], FALSE)) {
572                flag_set['d'] = TRUE;
573                log_to_screen("I guess DVD drive is at %s", flag_val['d']);
574            }
575        }
576        if (!find_home_of_exe("growisofs")) {
577            fatal_error
578                ("Please install growisofs (probably part of dvd+rw-tools). If you want DVD support, you need it.");
579        }
580        if (!find_home_of_exe("dvd+rw-format")) {
581            fatal_error
582                ("Please install dvd+rw-format (probably part of dvd+rw-tools). If you want DVD support, you need it.");
583        }
584        if (strchr(flag_val['d'], ',')) {
585            fatal_error
586                ("Please don't give a SCSI node. Give a _device_, preferably a /dev entry, for the parameter of the -d flag.");
587        }
588        if (!flag_set['s']) {
589            sprintf(flag_val['s'], "%d", DEFAULT_DVD_DISK_SIZE);    // 4.7 salesman's GB = 4.482 real GB = 4582 MB
590            strcat(flag_val['s'], "m");
591            log_to_screen
592                ("You did not specify a size (-s) for DVD. I'm guessing %s.",
593                 flag_val['s']);
594            flag_set['s'] = 1;
595        }
596/*
597      if (flag_set['Z']) {
598      bkpinfo->blank_dvd_first = TRUE;
599      }
600*/
601    }
602
603    if (flag_set['t'] || flag_set['u']) {   /* tape size */
604        if (strchr(flag_val['d'], ',')) {
605            fatal_error
606                ("Please don't give a SCSI node. Give a _device_, preferably a /dev entry, for the parameter of the -d flag.");
607        }
608        if (flag_set['O']) {
609            if (flag_set['s']) {
610                if (flag_set['t']) {
611                    fatal_error
612                        ("For the moment, please don't specify a tape size. Mondo should handle end-of-tape gracefully anyway.");
613                }
614                if (process_the_s_switch(flag_val['s'])) {
615                    fatal_error("Bad -s switch");
616                }
617            } else if (flag_set['u'] || flag_set['t']) {
618                for (i = 0; i <= MAX_NOOF_MEDIA; i++) {
619                    bkpinfo->media_size[i] = 0;
620                }
621            } else {
622                retval++;
623                log_to_screen("Tape size not specified.\n");
624            }
625        }
626    } else {                    /* CD|USB size */
627        if (flag_set['s']) {
628            if (process_the_s_switch(flag_val['s'])) {
629                fatal_error("Bad -s switch");
630            }
631        }
632        if (flag_set['w']) {
633            bkpinfo->wipe_media_first = TRUE;
634        }                       /* CD-RW */
635    }
636    if (flag_set['n']) {
637        strncpy(bkpinfo->nfs_mount, flag_val['n'], MAX_STR_LEN);
638        if (!flag_set['d']) {
639            strncpy(bkpinfo->nfs_remote_dir, "/", MAX_STR_LEN);
640        }
641        sprintf(tmp, "mount | grep -E \"^%s .*\" | cut -d' ' -f3",
642                bkpinfo->nfs_mount);
643        strncpy(bkpinfo->isodir,
644                call_program_and_get_last_line_of_output(tmp),
645                MAX_STR_LEN / 4);
646        if (strlen(bkpinfo->isodir) < 3) {
647            retval++;
648            log_to_screen("NFS share is not mounted. Please mount it.\n");
649        }
650        log_msg(3, "mount = %s", bkpinfo->nfs_mount);
651        log_msg(3, "isodir= %s", bkpinfo->isodir);
652    }
653    if (flag_set['c']) {
654        bkpinfo->backup_media_type = cdr;
655    }
656    if (flag_set['C']) {
657        bkpinfo->backup_media_type = cdstream;
658    }
659    if (flag_set['i']) {
660        bkpinfo->backup_media_type = iso;
661    }
662    if (flag_set['n']) {
663        bkpinfo->backup_media_type = nfs;
664    }
665    if (flag_set['r']) {
666        bkpinfo->backup_media_type = dvd;
667    }
668    if (flag_set['t']) {
669        bkpinfo->backup_media_type = tape;
670    }
671    if (flag_set['u']) {
672        bkpinfo->backup_media_type = udev;
673    }
674    if (flag_set['w']) {
675        bkpinfo->backup_media_type = cdrw;
676    }
677    if (flag_set['U']) {
678        bkpinfo->backup_media_type = usb;
679        /* Never try to eject a USB device */
680        bkpinfo->please_dont_eject = FALSE;
681    }
682    if (flag_set['z']) {
683        if (find_home_of_exe("getfattr")) {
684            asprintf(&g_getfattr,"getfattr");
685        }
686        if (find_home_of_exe("getfacl")) {
687            asprintf(&g_getfacl,"getfacl");
688        }
689    }
690
691    /* optional, popular */
692    if (flag_set['g']) {
693        g_text_mode = FALSE;
694    }
695
696    if (flag_set['E']) {
697        if (bkpinfo->exclude_paths[0]) {
698            strcat(bkpinfo->exclude_paths, " ");
699        }
700        asprintf(&tmp1, flag_val['E']);
701        p = tmp1;
702        q = tmp1;
703
704        /* Cut the flag_val['E'] in parts containing all paths to test them */
705        while (p != NULL) {
706            q = strchr(p, ' ');
707            if (q != NULL) {
708                *q = '\0';
709                /* Fix bug 14 where ending / cause a problem later
710                 * so handled here for the moment */
711                q--;
712                if (*q == '/') {
713                    *q = '\0';
714                }
715                q++;
716                /* End of bug fix */
717                if (stat(p, &buf) != 0) {
718                    log_msg(1, "WARNING ! %s doesn't exist", p);
719                }
720                p = q+1 ;
721            } else {
722                if (stat(p, &buf) != 0) {
723                    log_msg(1, "WARNING ! %s doesn't exist", p);
724                }
725                p = NULL;
726            }
727        }
728        paranoid_free(tmp1);
729
730        strncpy(bkpinfo->exclude_paths + strlen(bkpinfo->exclude_paths),
731                flag_val['E'],
732                4*MAX_STR_LEN - strlen(bkpinfo->exclude_paths));
733    }
734    if (flag_set['e']) {
735        bkpinfo->please_dont_eject = TRUE;
736    }
737    if (flag_set['N'])          // exclude NFS mounts & devices
738    {
739//      strncpy(psz, list_of_NFS_devices_and_mounts(), MAX_STR_LEN);
740        strncpy(psz, list_of_NFS_mounts_only(), MAX_STR_LEN);
741        if (bkpinfo->exclude_paths[0]) {
742            strncat(bkpinfo->exclude_paths, " ", 4*MAX_STR_LEN);
743        }
744        strncat(bkpinfo->exclude_paths, psz, 4*MAX_STR_LEN);
745        log_msg(3, "-N means we're now excluding %s",
746                bkpinfo->exclude_paths);
747    }
748    if (strlen(bkpinfo->exclude_paths) >= 4*MAX_STR_LEN) {
749        fatal_error
750            ("Your '-E' parameter is too long. Increase MAX_STR_LEN");
751    }
752    if (flag_set['b']) {
753        strcpy(psz, flag_val['b']);
754        log_msg(1, "psz = '%s'", psz);
755        if (psz[strlen(psz) - 1] == 'k') {
756            psz[strlen(psz) - 1] = '\0';
757            itbs = atol(psz) * 1024L;
758        } else {
759            itbs = atol(psz);
760        }
761        log_msg(1, "'%s' --> %ld", flag_val['b'], itbs);
762        log_msg(1, "Internal tape block size is now %ld bytes", itbs);
763        if (itbs % 512 != 0 || itbs < 256 || itbs > 1024L * 1024) {
764            fatal_error
765                ("Are you nuts? Silly, your internal tape block size is. Abort, I shall.");
766        }
767        bkpinfo->internal_tape_block_size = itbs;
768    }
769    if (flag_set['D']) {
770        bkpinfo->differential = 1;
771//      bkpinfo->differential = atoi (flag_val['D']);
772        if ((bkpinfo->differential < 1) || (bkpinfo->differential > 9)) {
773            fatal_error
774                ("The D option should be between 1 and 9 inclusive");
775        }
776    }
777    if (flag_set['x']) {
778        strncpy(bkpinfo->image_devs, flag_val['x'], MAX_STR_LEN / 4);
779        if (run_program_and_log_output("which ntfsclone", 2)) {
780            fatal_error("Please install ntfsprogs package/tarball.");
781        }
782    }
783    if (flag_set['m']) {
784        bkpinfo->manual_cd_tray = TRUE;
785    }
786    if (flag_set['k']) {
787        strncpy(bkpinfo->kernel_path, flag_val['k'], MAX_STR_LEN);
788        if (!strcmp(bkpinfo->kernel_path, "failsafe")) {
789            strcpy(bkpinfo->kernel_path, "FAILSAFE");
790        }
791        if (strcmp(bkpinfo->kernel_path, "FAILSAFE")
792            && !does_file_exist(bkpinfo->kernel_path)) {
793            retval++;
794            sprintf(tmp,
795                    "You specified kernel '%s', which does not exist\n",
796                    bkpinfo->kernel_path);
797            log_to_screen(tmp);
798        }
799    }
800    if (flag_set['p']) {
801        strncpy(bkpinfo->prefix, flag_val['p'], MAX_STR_LEN / 4);
802    }
803
804
805    if (flag_set['d']) {        /* backup directory (if ISO/NFS) */
806        if (flag_set['i']) {
807            strncpy(bkpinfo->isodir, flag_val['d'], MAX_STR_LEN / 4);
808            sprintf(tmp, "ls -l %s", bkpinfo->isodir);
809            if (run_program_and_log_output(tmp, 2)) {
810                fatal_error
811                    ("output folder does not exist - please create it");
812            }
813        } else if (flag_set['n']) {
814            strncpy(bkpinfo->nfs_remote_dir, flag_val['d'], MAX_STR_LEN);
815        } else {                /* backup device (if tape/CD-R/CD-RW) */
816
817            strncpy(bkpinfo->media_device, flag_val['d'], MAX_STR_LEN / 4);
818        }
819    }
820
821    if (flag_set['n']) {
822        sprintf(tmp, "echo hi > %s/%s/.dummy.txt", bkpinfo->isodir,
823                bkpinfo->nfs_remote_dir);
824        if (run_program_and_log_output(tmp, 2)) {
825            retval++;
826            sprintf(tmp,
827                    "Are you sure directory '%s' exists in remote dir '%s'?\nIf so, do you have rights to write to it?\n",
828                    bkpinfo->nfs_remote_dir, bkpinfo->nfs_mount);
829            log_to_screen(tmp);
830        }
831    }
832
833    if (!flag_set['d']
834        && (flag_set['c'] || flag_set['w'] || flag_set['C'])) {
835        if (g_kernel_version >= 2.6) {
836            if (popup_and_get_string
837                ("Device", "Please specify the device",
838                 bkpinfo->media_device, MAX_STR_LEN / 4)) {
839                retval++;
840                log_to_screen("User opted to cancel.");
841            }
842        } else if (find_cdrw_device(bkpinfo->media_device)) {
843            retval++;
844            log_to_screen
845                ("Tried and failed to find CD-R[W] drive automatically.\n");
846        } else {
847            flag_set['d'] = TRUE;
848            strncpy(flag_val['d'], bkpinfo->media_device, MAX_STR_LEN / 4);
849        }
850    }
851
852    if (!flag_set['d'] && !flag_set['n'] && !flag_set['C']) {
853        retval++;
854        log_to_screen("Please specify the backup device/directory.\n");
855        fatal_error
856            ("You didn't use -d to specify the backup device/directory.");
857    }
858/* optional, obscure */
859    for (i = '0'; i <= '9'; i++) {
860        if (flag_set[i]) {
861            bkpinfo->compression_level = i - '0';
862        }                       /* not '\0' but '0' */
863    }
864    if (flag_set['S']) {
865        sprintf(bkpinfo->scratchdir, "%s/mondo.scratch.%ld", flag_val['S'],
866                random() % 32768);
867    }
868    if (flag_set['T']) {
869        setup_tmpdir(flag_val['T']);
870        sprintf(tmp, "touch %s/.foo.dat", bkpinfo->tmpdir);
871        if (run_program_and_log_output(tmp, 1)) {
872            retval++;
873            log_to_screen
874                ("Please specify a tempdir which I can write to. :)");
875            fatal_error("I cannot write to the tempdir you specified.");
876        }
877        sprintf(tmp, "ln -sf %s/.foo.dat %s/.bar.dat", bkpinfo->tmpdir, bkpinfo->tmpdir);
878        if (run_program_and_log_output(tmp, 1)) {
879            retval++;
880            log_to_screen
881                ("Please don't specify a SAMBA or VFAT or NFS tmpdir.");
882            fatal_error("I cannot write to the tempdir you specified.");
883        }
884    }
885    if (flag_set['A']) {
886        strncpy(bkpinfo->call_after_iso, flag_val['A'], MAX_STR_LEN);
887    }
888    if (flag_set['B']) {
889        strncpy(bkpinfo->call_before_iso, flag_val['B'], MAX_STR_LEN);
890    }
891    if (flag_set['F']) {
892        g_skip_floppies = TRUE;
893    }
894    if (flag_set['H']) {
895        g_cd_recovery = TRUE;
896    }
897    if (flag_set['l']) {
898#ifdef __FreeBSD__
899#  define BOOT_LOADER_CHARS "GLBMR"
900#else
901#  ifdef __IA64__
902#    define BOOT_LOADER_CHARS "GER"
903#  else
904#    define BOOT_LOADER_CHARS "GLR"
905#  endif
906#endif
907        if (!strchr
908            (BOOT_LOADER_CHARS,
909             (bkpinfo->boot_loader = flag_val['l'][0]))) {
910            log_msg(1, "%c? WTF is %c? I need G, L, E or R.",
911                    bkpinfo->boot_loader, bkpinfo->boot_loader);
912            fatal_error
913                ("Please specify GRUB, LILO, ELILO  or RAW with the -l switch");
914        }
915#undef BOOT_LOADER_CHARS
916    }
917    if (flag_set['f']) {
918        strncpy(bkpinfo->boot_device,
919                resolve_softlinks_to_get_to_actual_device_file(flag_val
920                                                               ['f']),
921                MAX_STR_LEN / 4);
922    }
923    if (flag_set['P']) {
924        strncpy(bkpinfo->postnuke_tarball, flag_val['P'], MAX_STR_LEN);
925    }
926    if (flag_set['Q']) {
927        i = which_boot_loader(tmp);
928        log_msg(3, "boot loader is %c, residing at %s", i, tmp);
929        printf("boot loader is %c, residing at %s\n", i, tmp);
930        finish(0);
931    }
932    if (flag_set['L']) {
933        bkpinfo->use_lzo = TRUE;
934        if (run_program_and_log_output("which lzop", 2)) {
935            retval++;
936            log_to_screen
937                ("Please install LZOP. You can't use '-L' until you do.\n");
938        }
939    }
940
941    if (flag_set['G']) {
942        bkpinfo->use_gzip = TRUE;
943        if (run_program_and_log_output("which gzip", 2)) {
944            retval++;
945            log_to_screen
946                ("Please install gzip. You can't use '-G' until you do.\n");
947        }
948    }
949
950    if (!flag_set['o']
951        &&
952        !run_program_and_log_output
953        ("grep -Ei suse /etc/issue.net | grep -E '9.0' | grep 64", 2)) {
954        bkpinfo->make_cd_use_lilo = TRUE;
955        log_to_screen
956            ("Forcing you to use LILO. SuSE 9.0 (64-bit) has a broken mkfs.vfat binary.");
957    }
958    if (flag_set['o']) {
959        bkpinfo->make_cd_use_lilo = TRUE;
960    }
961#ifndef __FreeBSD__
962    else {
963        if (!is_this_a_valid_disk_format("vfat")) {
964            bkpinfo->make_cd_use_lilo = TRUE;
965            log_to_screen
966                ("Your kernel appears not to support vfat filesystems. I am therefore");
967            log_to_screen
968                ("using LILO instead of SYSLINUX as the CD/floppy's boot loader.");
969        }
970        if (run_program_and_log_output("which mkfs.vfat", 2)) {
971            bkpinfo->make_cd_use_lilo = TRUE;
972#ifdef __IA32__
973            log_to_screen
974                ("Your filesystem is missing 'mkfs.vfat', so I cannot use SYSLINUX as");
975            log_to_screen
976                ("your boot loader. I shall therefore use LILO instead.");
977#endif
978#ifdef __IA64__
979            log_to_screen
980                ("Your filesystem is missing 'mkfs.vfat', so I cannot prepare the EFI");
981            log_to_screen("environment correctly. Please install it.");
982            fatal_error("Aborting");
983#endif
984        }
985#ifdef __IA64__
986        /* We force ELILO usage on IA64 */
987        bkpinfo->make_cd_use_lilo = TRUE;
988#endif
989    }
990#endif
991
992    if (bkpinfo->make_cd_use_lilo && !does_file_exist("/boot/boot.b")) {
993        paranoid_system("touch /boot/boot.b");
994    }
995
996    i = flag_set['O'] + flag_set['V'];
997    if (i == 0) {
998        retval++;
999        log_to_screen("Specify backup (-O), verify (-V) or both (-OV).\n");
1000    }
1001
1002/* and finally... */
1003
1004    paranoid_free(tmp);
1005    paranoid_free(psz);
1006    return (retval);
1007}
1008
1009
1010
1011/**
1012 * Get the switches from @p argc and @p argv using getopt() and place them in
1013 * @p flag_set and @p flag_val.
1014 * @param argc The argument count (@p argc passed to main()).
1015 * @param argv The argument vector (@p argv passed to main()).
1016 * @param flag_val An array indexed by switch letter - if a switch is set and
1017 * has an argument then set flag_val[switch] to that argument.
1018 * @param flag_set An array indexed by switch letter - if a switch is set then
1019 * set flag_set[switch] to TRUE, else set it to FALSE.
1020 * @return The number of problems with the command line (0 for success).
1021 */
1022int
1023retrieve_switches_from_command_line(int argc, char *argv[],
1024                                    char flag_val[128][MAX_STR_LEN],
1025                                    bool flag_set[128])
1026{
1027    /*@ ints ** */
1028    int opt = 0;
1029    char tmp[MAX_STR_LEN];
1030    int i = 0;
1031    int len;
1032
1033    /*@ bools *** */
1034    bool bad_switches = FALSE;
1035
1036    assert(flag_val != NULL);
1037    assert(flag_set != NULL);
1038
1039    for (i = 0; i < 128; i++) {
1040        flag_val[i][0] = '\0';
1041        flag_set[i] = FALSE;
1042    }
1043    while ((opt =
1044            getopt(argc, argv,
1045                   "0123456789A:B:C:DE:FGHI:J:K:LNOP:QRS:T:UVWb:c:d:ef:gik:l:mn:op:rs:tuw:x:z"))
1046           != -1) {
1047        if (opt == '?') {
1048            bad_switches = TRUE;
1049            /*log_it("Invalid option: %c\n",optopt); */
1050        } else {
1051            if (flag_set[optopt]) {
1052                bad_switches = TRUE;
1053                sprintf(tmp, "Switch -%c previously defined as %s\n", opt,
1054                        flag_val[i]);
1055                log_to_screen(tmp);
1056            } else {
1057                flag_set[opt] = TRUE;
1058                if (optarg) {
1059                    len = strlen(optarg);
1060                    if (optarg[0] != '/' && optarg[len - 1] == '/') {
1061                        optarg[--len] = '\0';
1062                        log_to_screen
1063                            ("Warning - param '%s' should not have trailing slash!",
1064                             optarg);
1065                    }
1066                    if (opt == 'd') {
1067                        if (strchr(flag_val[opt], '/')
1068                            && flag_val[opt][0] != '/') {
1069                            sprintf(tmp,
1070                                    "-%c flag --- must be absolute path --- '%s' isn't absolute",
1071                                    opt, flag_val[opt]);
1072                            log_to_screen(tmp);
1073                            bad_switches = TRUE;
1074                        }
1075                    }
1076                    strcpy(flag_val[opt], optarg);
1077                }
1078            }
1079        }
1080    }
1081    for (i = optind; i < argc; i++) {
1082        bad_switches = TRUE;
1083        sprintf(tmp, "Invalid arg -- %s\n", argv[i]);
1084        log_to_screen(tmp);
1085    }
1086    return (bad_switches);
1087}
1088
1089
1090
1091
1092/**
1093 * Print a not-so-helpful help message and exit.
1094 */
1095void help_screen()
1096{
1097    log_msg(1, "Type 'man mondo-archive' for more information\n");
1098    exit(1);
1099}
1100
1101
1102/**
1103 * Terminate Mondo in response to a signal.
1104 * @param sig The signal number received.
1105 */
1106void terminate_daemon(int sig)
1107{
1108    char tmp[64];
1109    char tmp2[MAX_STR_LEN];
1110    //  char command[512];
1111    //  pid_t pid;
1112
1113    switch (sig) {
1114    case SIGINT:
1115        sprintf(tmp, "SIGINT");
1116        strcpy(tmp2, "You interrupted me :-)");
1117        break;
1118    case SIGKILL:
1119        sprintf(tmp, "SIGKILL");
1120        strcpy(tmp2,
1121               "I seriously have no clue how this signal even got to me. Something's wrong with your system.");
1122        break;
1123    case SIGTERM:
1124        sprintf(tmp, "SIGTERM");
1125        strcpy(tmp2, "Got terminate signal");
1126        break;
1127    case SIGHUP:
1128        sprintf(tmp, "SIGHUP");
1129        strcpy(tmp2, "Hangup on line");
1130        break;
1131    case SIGSEGV:
1132        sprintf(tmp, "SIGSEGV");
1133        strcpy(tmp2,
1134               "Internal programming error. Please send a backtrace as well as your log.");
1135        break;
1136    case SIGPIPE:
1137        sprintf(tmp, "SIGPIPE");
1138        strcpy(tmp2, "Pipe was broken");
1139        break;
1140    case SIGABRT:
1141        sprintf(tmp, "SIGABRT");
1142        sprintf(tmp2,
1143                "Abort - probably failed assertion. I'm sleeping for a few seconds so you can read the message.");
1144        break;
1145    default:
1146        sprintf(tmp, "(Unknown)");
1147    }
1148
1149    strcat(tmp, " signal received from OS");
1150    log_to_screen(tmp);
1151    log_to_screen(tmp2);
1152    if (sig == SIGABRT) {
1153        sleep(10);
1154    }
1155    kill_buffer();
1156    fatal_error
1157        ("Mondoarchive is terminating in response to a signal from the OS");
1158    finish(254);                // just in case
1159}
1160
1161
1162
1163
1164/**
1165 * Turn signal-trapping on or off.
1166 * @param on If TRUE, turn it on; if FALSE, turn it off (we still trap it, just don't do as much).
1167 */
1168void set_signals(int on)
1169{
1170    int signals[] =
1171        { SIGTERM, SIGHUP, SIGTRAP, SIGABRT, SIGINT, SIGKILL, SIGSTOP, 0 };
1172    int i;
1173
1174    signal(SIGPIPE, sigpipe_occurred);
1175    for (i = 0; signals[i]; i++) {
1176        if (on) {
1177            signal(signals[i], terminate_daemon);
1178        } else {
1179            signal(signals[i], termination_in_progress);
1180        }
1181    }
1182}
1183
1184
1185
1186
1187/**
1188 * Exit immediately without cleaning up.
1189 * @param sig The signal we are exiting due to.
1190 */
1191void termination_in_progress(int sig)
1192{
1193    log_msg(1, "Termination in progress");
1194    usleep(1000);
1195    pthread_exit(0);
1196}
1197
1198/* @} - end of cliGroup */
Note: See TracBrowser for help on using the repository browser.