source: trunk/mondo/mondo/mondoarchive/mondo-cli.c @ 292

Last change on this file since 292 was 292, checked in by bcornec, 13 years ago

svn merge -r288:291 $SVN_M/branches/2.06

  • Property svn:keywords set to Id
File size: 26.7 KB
Line 
1/***************************************************************************
2 * $Id: mondo-cli.c 292 2006-01-10 08:51:24Z bcornec $
3 *
4 * Functions for handling command-line arguments passed to mondoarchive.
5 */
6
7/** @def BOOT_LOADER_CHARS The characters allowed for boot loader on this platform. */
8
9#include "../common/my-stuff.h"
10#include "../common/mondostructures.h"
11#include "mondo-cli-EXT.h"
12#include "../common/libmondo.h"
13#ifndef S_SPLINT_S
14#include <pthread.h>
15#endif
16
17//static char cvsid[] = "$Id: mondo-cli.c 292 2006-01-10 08:51:24Z bcornec $";
18
19extern int g_loglevel;
20extern bool g_text_mode;
21extern bool g_skip_floppies;    ///< Whether to skip the creation of boot disks
22extern char g_startdir[MAX_STR_LEN];    ///< ????? @bug ?????
23extern char g_tmpfs_mountpt[MAX_STR_LEN];
24extern bool g_sigpipe;
25
26/*@ file pointer **************************************************/
27extern FILE *g_tape_stream;
28
29/*@ long long *****************************************************/
30extern long long g_tape_posK;
31
32/*@ long **********************************************************/
33extern long g_noof_sets;
34
35/*@ bool******** **************************************************/
36bool g_debugging = FALSE;       ///< ????? @bug ????? @ingroup globalGroup
37bool g_running_live = FALSE;    ///< ????? @bug ????? @ingroup globalGroup
38extern bool g_cd_recovery;
39
40/**
41 * Whether we're restoring from ISOs. Obviously not, since this is the
42 * backup program.
43 * @note You @b MUST declare this variable somewhere in your program if
44 * you use libmondo. Otherwise the link will fail.
45 * @ingroup globalGroup
46 */
47bool g_ISO_restore_mode = FALSE;
48
49
50extern double g_kernel_version;
51
52extern int g_current_media_number;
53
54
55
56extern pid_t g_main_pid;
57
58
59
60
61extern char *resolve_softlinks_to_get_to_actual_device_file(char *);
62
63
64
65/**
66 * @addtogroup cliGroup
67 * @{
68 */
69/**
70 * Populate @p bkpinfo from the command-line parameters stored in @p argc and @p argv.
71 * @param argc The argument count, including the program name; @p argc passed to main().
72 * @param argv The argument vector; @p argv passed to main().
73 * @param bkpinfo The backup information structure to populate.
74 * @return The number of problems with the command line (0 for success).
75 */
76int
77handle_incoming_parameters(int argc, char *argv[],
78                           struct s_bkpinfo *bkpinfo)
79{
80    /*@ int *** */
81    int res = 0;
82    int retval = 0;
83    int i = 0, j;
84
85    /*@ buffers *************** */
86    char *tmp;
87    char *flag_val[128];
88    bool flag_set[128];
89
90    sensibly_set_tmpdir_and_scratchdir(bkpinfo);
91
92    for (i = 0; i < 128; i++) {
93        flag_val[i] = NULL;
94        flag_set[i] = FALSE;
95    }
96    for (j = 1; j <= MAX_NOOF_MEDIA; j++) {
97        bkpinfo->media_size[j] = 650;
98    }                           /* default */
99    res =
100        retrieve_switches_from_command_line(argc, argv, flag_val,
101                                            flag_set);
102    log_it("value: %s", flag_val['s']);
103    retval += res;
104    if (!retval) {
105        res = process_switches(bkpinfo, flag_val, flag_set);
106        retval += res;
107    }
108/*
109  if (!retval)
110    {
111*/
112    log_msg(3, "Switches:-");
113    for (i = 0; i < 128; i++) {
114        if (flag_set[i]) {
115            asprintf(&tmp, "-%c %s", i, flag_val[i]);
116            log_msg(3, tmp);
117            paranoid_free(tmp);
118        }
119    }
120//    }
121    asprintf(&tmp, "rm -Rf %s/tmp.mondo.*", bkpinfo->tmpdir);
122    paranoid_system(tmp);
123    paranoid_free(tmp);
124
125    asprintf(&tmp, "rm -Rf %s/mondo.scratch.*", bkpinfo->scratchdir);
126    paranoid_system(tmp);
127    paranoid_free(tmp);
128
129    sprintf(bkpinfo->tmpdir + strlen(bkpinfo->tmpdir), "/tmp.mondo.%ld",
130            random() % 32767);
131    sprintf(bkpinfo->scratchdir + strlen(bkpinfo->scratchdir),
132            "/mondo.scratch.%ld", random() % 32767);
133
134    asprintf(&tmp, "mkdir -p %s/tmpfs", bkpinfo->tmpdir);
135    paranoid_system(tmp);
136    paranoid_free(tmp);
137
138    asprintf(&tmp, "mkdir -p %s", bkpinfo->scratchdir);
139    paranoid_system(tmp);
140    paranoid_free(tmp);
141
142    if (bkpinfo->nfs_mount[0] != '\0') {
143        store_nfs_config(bkpinfo);
144    }
145    return (retval);
146}
147
148
149/**
150 * Store the sizespec(s) stored in @p value into @p bkpinfo.
151 * @param bkpinfo The backup information structure; the @c bkpinfo->media_size field will be populated.
152 * @param value The sizespec (e.g. "2g", "40m").
153 * @return 0, always.
154 * @bug Return code not needed.
155 */
156int process_the_s_switch(struct s_bkpinfo *bkpinfo, char *value)
157{
158    int j;
159    char *tmp;
160    char *p;
161    char *q;
162    char *comment;
163
164    assert(bkpinfo != NULL);
165    assert(value != NULL);
166
167    bkpinfo->media_size[0] = -1;    /* dummy value */
168    for (j = 1, p = value; j < MAX_NOOF_MEDIA && strchr(p, ',');
169         j++, p = strchr(p, ',') + 1) {
170        asprintf(&tmp, p);
171        q = strchr(tmp, ',');
172        if (q != NULL) {
173            *q = '\0';
174        }
175        bkpinfo->media_size[j] = friendly_sizestr_to_sizelong(tmp);
176        paranoid_free(tmp);
177
178        asprintf(&comment, "media_size[%d] = %ld", j,
179                bkpinfo->media_size[j]);
180        log_msg(3, comment);
181        paranoid_free(comment);
182    }
183    for (; j <= MAX_NOOF_MEDIA; j++) {
184        bkpinfo->media_size[j] = friendly_sizestr_to_sizelong(p);
185    }
186    for (j = 1; j <= MAX_NOOF_MEDIA; j++) {
187        if (bkpinfo->media_size[j] <= 0) {
188            log_msg(1, "You gave media #%d an invalid size\n", j);
189            return (-1);
190        }
191    }
192    return (0);
193}
194
195
196/**
197 * Process mondoarchive's command-line switches.
198 * @param bkpinfo The backup information structure to populate.
199 * @param flag_val An array of the argument passed to each switch (the letter is the index).
200 * If a switch is not set or has no argument, the field in @p flag_val doesn't matter.
201 * @param flag_set An array of <tt>bool</tt>s indexed by switch letter: TRUE if it's set,
202 * FALSE if it's not.
203 * @return The number of problems with the switches, or 0 for success.
204 * @bug Maybe include a list of all switches (inc. intentionally undocumented ones not in the manual!) here?
205 */
206int
207process_switches(struct s_bkpinfo *bkpinfo,
208                 char *flag_val[128], bool flag_set[128])
209{
210
211    /*@ ints *** */
212    int i = 0;
213    int retval = 0;
214    int percent = 0;
215
216    /*@ buffers ** */
217    char *tmp;
218    char *tmp2;
219    char *psz;
220
221    long itbs;
222
223    struct stat buf;
224
225    assert(bkpinfo != NULL);
226    assert(flag_val != NULL);
227    assert(flag_set != NULL);
228
229    bkpinfo->internal_tape_block_size = DEFAULT_INTERNAL_TAPE_BLOCK_SIZE;
230
231    /* compulsory */
232    i = flag_set['c'] + flag_set['i'] + flag_set['n'] +
233        flag_set['t'] + flag_set['u'] + flag_set['r'] +
234        flag_set['w'] + flag_set['C'];
235    if (i == 0) {
236        retval++;
237        log_to_screen("You must specify the media type\n");
238    }
239    if (i > 1) {
240        retval++;
241        log_to_screen("Please specify only one media type\n");
242    }
243    if (flag_set['K']) {
244        g_loglevel = atoi(flag_val['K']);
245        if (g_loglevel < 3) {
246            g_loglevel = 3;
247        }
248    }
249    if (flag_set['L'] && flag_set['0']) {
250        retval++;
251        log_to_screen("You cannot have 'no compression' _and_ LZOP.\n");
252    }
253    bkpinfo->backup_data = flag_set['O'];
254    bkpinfo->verify_data = flag_set['V'];
255    if (flag_set['I'] && !bkpinfo->backup_data) {
256        log_to_screen("-I switch is ignored if just verifying");
257    }
258    if (flag_set['E'] && !bkpinfo->backup_data) {
259        log_to_screen("-E switch is ignored if just verifying");
260    }
261
262    if (!find_home_of_exe("afio")) {
263        if (find_home_of_exe("star")) {
264            flag_set['R'] = TRUE;
265            log_msg(1, "Using star instead of afio");
266        } else {
267            fatal_error
268                ("Neither afio nor star is installed. Please install at least one.");
269        }
270    }
271
272    if (flag_set['R']) {
273        bkpinfo->use_star = TRUE;
274        if (flag_set['L']) {
275            fatal_error("You may not use star and lzop at the same time.");
276        }
277        if (!find_home_of_exe("star")) {
278            fatal_error
279                ("Please install 'star' RPM or tarball if you are going to use -R. Thanks.");
280        }
281    }
282    if (flag_set['W']) {
283        bkpinfo->nonbootable_backup = TRUE;
284        log_to_screen("Warning - you have opted for non-bootable backup");
285        if (flag_set['f'] || flag_set['l']) {
286            log_to_screen
287                ("You don't need to specify bootloader or bootdevice");
288        }
289    }
290    if (flag_set['t'] && flag_set['H']) {
291        fatal_error
292            ("Sorry, you may not nuke w/o warning from tape. Drop -H, please.");
293    }
294    if (flag_set['I']) {
295        if (!strcmp(bkpinfo->include_paths, "/")) {
296            log_msg(2, "'/' is pleonastic.");
297            bkpinfo->include_paths[0] = '\0';
298        }
299        if (bkpinfo->include_paths[0]) {
300            strcat(bkpinfo->include_paths, " ");
301        }
302        if (stat(flag_val['I'], &buf) != 0) {
303            log_msg("ERROR ! %s doesn't exist", flag_val['I']);
304            fatal_error("ERROR ! You specified a directory to include which doesn't exist");
305        }
306        strncpy(bkpinfo->include_paths + strlen(bkpinfo->include_paths),
307                flag_val['I'],
308                MAX_STR_LEN - strlen(bkpinfo->include_paths));
309        log_msg(1, "include_paths is now '%s'", bkpinfo->include_paths);
310        if (bkpinfo->include_paths[0] == '-') {
311            retval++;
312            log_to_screen("Please supply a sensible value with '-I'\n");
313        }
314    }
315
316    if (g_kernel_version >= 2.6 && !flag_set['d']
317        && (flag_set['c'] || flag_set['w'])) {
318        fatal_error
319            ("If you are using the 2.6.x kernel, please specify the CD-R(W) device.");
320    }
321
322
323    if (flag_set['J']) {
324        if (flag_set['I']) {
325            retval++;
326            log_to_screen
327                ("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. :-)");
328        }
329        bkpinfo->make_filelist = FALSE;
330        strcpy(bkpinfo->include_paths, flag_val['J']);
331    }
332    if (flag_set['c'] || flag_set['w'] || flag_set['C'] || flag_set['r']) {
333        if (!flag_set['r'] && g_kernel_version <= 2.5
334            && strstr(flag_val['d'], "/dev/")) {
335            fatal_error
336                ("Please don't give a /dev entry. Give a SCSI node for the parameter of the -d flag.");
337        }
338        if (flag_set['r'] && g_kernel_version <= 2.5
339            && !strstr(flag_val['d'], "/dev/")) {
340            fatal_error
341                ("Please give a /dev entry, not a SCSI node, as the parameter of the -d flag.");
342        }
343        if (g_kernel_version >= 2.6 && !strstr(flag_val['d'], "/dev/")) {
344            log_to_screen
345                ("Linus says 2.6 has a broken ide-scsi module. Proceed at your own risk...");
346        }
347
348        if (system("which cdrecord > /dev/null 2> /dev/null")
349            && system("which dvdrecord > /dev/null 2> /dev/null")) {
350            fatal_error
351                ("Please install dvdrecord/cdrecord and try again.");
352        }
353        if (flag_set['C']) {
354            bkpinfo->cdrw_speed = atoi(flag_val['C']);
355            if (bkpinfo->cdrw_speed < 1) {
356                fatal_error
357                    ("You specified a silly speed for a CD-R[W] drive");
358            }
359            if (!flag_set['L']) {
360                log_to_screen
361                    ("You must use -L with -C. Therefore I am setting it for you.");
362                flag_set['L'] = 1;
363                flag_val['L'] = NULL;
364            }
365        } else {
366            log_msg(3, "flag_val['c'] = %s", flag_val['c']);
367            log_msg(3, "flag_val['w'] = %s", flag_val['w']);
368            if (flag_set['c']) {
369                bkpinfo->cdrw_speed = atoi(flag_val['c']);
370            } else if (flag_set['w']) {
371                bkpinfo->cdrw_speed = atoi(flag_val['w']);
372            } else if (flag_set['r']) {
373                bkpinfo->cdrw_speed = 1;    /*atoi(flag_val['r']); */
374            }
375
376            if (bkpinfo->cdrw_speed < 1) {
377                fatal_error
378                    ("You specified a silly speed for a CD-R[W] drive");
379            }
380        }
381    }
382    if (flag_set['t'] && !flag_set['d']) {
383        log_it("Hmm! No tape drive specified. Let's see what we can do.");
384        if (find_tape_device_and_size(flag_val['d'], tmp)) {
385            fatal_error
386                ("Tape device not specified. I couldn't find it either.");
387        }
388        flag_set['d'] = TRUE;
389        paranoid_free(tmp); // allocation from find_tape_device_and_size
390
391        asprintf(&tmp,
392                "You didn't specify a tape streamer device. I'm assuming %s",
393                flag_val['d']);
394        log_to_screen(tmp);
395        paranoid_free(tmp);
396        percent = 0;
397    }
398
399    if (flag_set['r'])          // DVD
400    {
401        if (flag_set['m']) {
402            fatal_error
403                ("Manual CD tray (-m) not yet supported in conjunction w/ DVD drives. Drop -m.");
404        }
405        if (!flag_set['d']) {
406            if ((flag_val['d'] = find_dvd_device()) != NULL) {
407                flag_set['d'] = TRUE;
408                log_to_screen("I guess DVD drive is at %s", flag_val['d']);
409            }
410        }
411        if (!find_home_of_exe("growisofs")) {
412            fatal_error
413                ("Please install growisofs (probably part of dvd+rw-tools). If you want DVD support, you need it.");
414        }
415        if (!find_home_of_exe("dvd+rw-format")) {
416            fatal_error
417                ("Please install dvd+rw-format (probably part of dvd+rw-tools). If you want DVD support, you need it.");
418        }
419        if (strchr(flag_val['d'], ',')) {
420            fatal_error
421                ("Please don't give a SCSI node. Give a _device_, preferably a /dev entry, for the parameter of the -d flag.");
422        }
423        if (!flag_set['s']) {
424            asprintf(&flag_val['s'], "%dm", DEFAULT_DVD_DISK_SIZE); // 4.7 salesman's GB = 4.482 real GB = 4582 MB
425            log_to_screen
426                ("You did not specify a size (-s) for DVD. I'm guessing %s.",
427                 flag_val['s']);
428            flag_set['s'] = 1;
429        }
430    }
431
432    if (flag_set['t'] || flag_set['u']) {   /* tape size */
433        if (strchr(flag_val['d'], ',')) {
434            fatal_error
435                ("Please don't give a SCSI node. Give a _device_, preferably a /dev entry, for the parameter of the -d flag.");
436        }
437        if (flag_set['O']) {
438            if (flag_set['s']) {
439                if (flag_set['t']) {
440                    fatal_error
441                        ("For the moment, please don't specify a tape size. Mondo should handle end-of-tape gracefully anyway.");
442                }
443                if (process_the_s_switch(bkpinfo, flag_val['s'])) {
444                    fatal_error("Bad -s switch");
445                }
446            } else if (flag_set['u'] || flag_set['t']) {
447                for (i = 0; i <= MAX_NOOF_MEDIA; i++) {
448                    bkpinfo->media_size[i] = 0;
449                }
450            } else {
451                retval++;
452                log_to_screen("Tape size not specified.\n");
453            }
454        }
455    } else {                    /* CD size */
456        if (flag_set['s']) {
457            if (process_the_s_switch(bkpinfo, flag_val['s'])) {
458                fatal_error("Bad -s switch");
459            }
460        }
461        if (flag_set['w']) {
462            bkpinfo->wipe_media_first = TRUE;
463        }                       /* CD-RW */
464    }
465    if (flag_set['n']) {
466        asprintf(&tmp, flag_val['n']);
467        bkpinfo->nfs_mount = tmp;
468        if (!flag_set['d']) {
469            asprintf(&tmp, "/");
470            bkpinfo->nfs_remote_dir = tmp;
471        }
472        asprintf(&tmp, "mount | grep -x \"%s .*\" | cut -d' ' -f3",
473                bkpinfo->nfs_mount);
474        asprintf(&tmp2, call_program_and_get_last_line_of_output(tmp));
475        bkpinfo->isodir = tmp2;
476        paranoid_free(tmp);
477
478        if (strlen(bkpinfo->isodir) < 3) {
479            retval++;
480            log_to_screen("NFS share is not mounted. Please mount it.\n");
481        }
482        log_msg(3, "mount = %s", bkpinfo->nfs_mount);
483        log_msg(3, "isodir= %s", bkpinfo->isodir);
484    }
485    if (flag_set['c']) {
486        bkpinfo->backup_media_type = cdr;
487    }
488    if (flag_set['C']) {
489        bkpinfo->backup_media_type = cdstream;
490    }
491    if (flag_set['i']) {
492        bkpinfo->backup_media_type = iso;
493    }
494    if (flag_set['n']) {
495        bkpinfo->backup_media_type = nfs;
496    }
497    if (flag_set['r']) {
498        bkpinfo->backup_media_type = dvd;
499    }
500    if (flag_set['t']) {
501        bkpinfo->backup_media_type = tape;
502    }
503    if (flag_set['u']) {
504        bkpinfo->backup_media_type = udev;
505    }
506    if (flag_set['w']) {
507        bkpinfo->backup_media_type = cdrw;
508    }
509
510/* optional, popular */
511    if (flag_set['g']) {
512        g_text_mode = FALSE;
513    }
514    if (flag_set['E']) {
515        if (bkpinfo->exclude_paths[0]) {
516            strcat(bkpinfo->exclude_paths, " ");
517        }
518        if (stat(flag_val['E'], &buf) != 0) {
519            log_msg(1, "WARNING ! %s doesn't exist", flag_val['E']);
520        }
521        strncpy(bkpinfo->exclude_paths + strlen(bkpinfo->exclude_paths),
522                flag_val['E'],
523                MAX_STR_LEN - strlen(bkpinfo->exclude_paths));
524    }
525    if (flag_set['e']) {
526        bkpinfo->please_dont_eject = TRUE;
527    }
528    if (flag_set['N'])          // exclude NFS mounts & devices
529    {
530        psz = list_of_NFS_mounts_only();
531        if (bkpinfo->exclude_paths[0]) {
532            strncat(bkpinfo->exclude_paths, " ", MAX_STR_LEN);
533        }
534        strncat(bkpinfo->exclude_paths, psz, MAX_STR_LEN);
535        paranoid_free(psz);
536
537        log_msg(3, "-N means we're now excluding %s",
538                bkpinfo->exclude_paths);
539    }
540    if (strlen(bkpinfo->exclude_paths) >= MAX_STR_LEN) {
541        fatal_error
542            ("Your '-E' parameter is too long. Please use '-J'. (See manual.)");
543    }
544    if (flag_set['b']) {
545        asprintf(&psz, flag_val['b']);
546        log_msg(1, "psz = '%s'", psz);
547        if (psz[strlen(psz) - 1] == 'k') {
548            psz[strlen(psz) - 1] = '\0';
549            itbs = atol(psz) * 1024L;
550        } else {
551            itbs = atol(psz);
552        }
553        paranoid_free(psz);
554
555        log_msg(1, "'%s' --> %ld", flag_val['b'], itbs);
556        log_msg(1, "Internal tape block size is now %ld bytes", itbs);
557        if (itbs % 512 != 0 || itbs < 256 || itbs > 1024L * 1024) {
558            fatal_error
559                ("Are you nuts? Silly, your internal tape block size is. Abort, I shall.");
560        }
561        bkpinfo->internal_tape_block_size = itbs;
562    }
563    if (flag_set['D']) {
564        bkpinfo->differential = 1;
565//      bkpinfo->differential = atoi (flag_val['D']);
566        if ((bkpinfo->differential < 1) || (bkpinfo->differential > 9)) {
567            fatal_error
568                ("The D option should be between 1 and 9 inclusive");
569        }
570    }
571    if (flag_set['x']) {
572        asprintf(&tmp, flag_val['x']);
573        bkpinfo->image_devs = tmp;
574        if (run_program_and_log_output("which partimagehack", 2)) {
575            fatal_error("Please install partimagehack RPM/tarball.");
576        }
577    }
578    if (flag_set['m']) {
579        bkpinfo->manual_cd_tray = TRUE;
580    }
581    if (flag_set['k']) {
582        if (strcasecmp(flag_val['k'], "FAILSAFE")) {
583            asprintf(&tmp, "FAILSAFE");
584            bkpinfo->kernel_path = tmp;
585
586            if (!does_file_exist(bkpinfo->kernel_path)) {
587                retval++;
588                asprintf(&tmp,
589                    "You specified kernel '%s', which does not exist\n",
590                    bkpinfo->kernel_path);
591                log_to_screen(tmp);
592                paranoid_free(tmp);
593            }
594        } else {
595            asprintf(&tmp, flag_val['k']);
596            bkpinfo->kernel_path = tmp;
597        }
598    }
599    if (flag_set['p']) {
600        asprintf(&tmp, bkpinfo->prefix);
601        bkpinfo->prefix = tmp;
602    }
603
604
605    if (flag_set['d']) {        /* backup directory (if ISO/NFS) */
606        if (flag_set['i']) {
607            asprintf(&tmp, flag_val['d']);
608            bkpinfo->isodir = tmp;
609            asprintf(&tmp, "ls -l %s", bkpinfo->isodir);
610            if (run_program_and_log_output(tmp, FALSE)) {
611                fatal_error
612                    ("output folder does not exist - please create it");
613            }
614            paranoid_free(tmp);
615        } else if (flag_set['n']) {
616            asprintf(&tmp, flag_val['d']);
617            bkpinfo->nfs_remote_dir = tmp;
618        } else {                /* backup device (if tape/CD-R/CD-RW) */
619
620            paranoid_alloc(bkpinfo->media_device, flag_val['d']);
621        }
622    }
623
624    if (flag_set['n']) {
625        asprintf(&tmp, "echo hi > %s/%s/.dummy.txt", bkpinfo->isodir,
626                bkpinfo->nfs_remote_dir);
627        if (run_program_and_log_output(tmp, FALSE)) {
628            retval++;
629            paranoid_free(tmp);
630            asprintf(&tmp,
631                    "Are you sure directory '%s' exists in remote dir '%s'?\nIf so, do you have rights to write to it?\n",
632                    bkpinfo->nfs_remote_dir, bkpinfo->nfs_mount);
633            log_to_screen(tmp);
634        }
635        paranoid_free(tmp);
636    }
637
638    if (!flag_set['d']
639        && (flag_set['c'] || flag_set['w'] || flag_set['C'])) {
640        if (g_kernel_version >= 2.6) {
641            if (popup_and_get_string
642                ("Device", "Please specify the device",
643                 bkpinfo->media_device, MAX_STR_LEN / 4)) {
644                retval++;
645                log_to_screen("User opted to cancel.");
646            }
647        } else if ((tmp = find_cdrw_device()) ==  NULL) {
648            paranoid_free(bkpinfo->media_device);
649            bkpinfo->media_device = tmp;
650            retval++;
651            log_to_screen
652                ("Tried and failed to find CD-R[W] drive automatically.\n");
653        } else {
654            flag_set['d'] = TRUE;
655            asprintf(&flag_val['d'], bkpinfo->media_device);
656        }
657    }
658
659    if (!flag_set['d'] && !flag_set['n'] && !flag_set['C']) {
660        retval++;
661        log_to_screen("Please specify the backup device/directory.\n");
662        fatal_error
663            ("You didn't use -d to specify the backup device/directory.");
664    }
665/* optional, obscure */
666    for (i = '0'; i <= '9'; i++) {
667        if (flag_set[i]) {
668            bkpinfo->compression_level = i - '0';
669        }                       /* not '\0' but '0' */
670    }
671    if (flag_set['S']) {
672        asprintf(&tmp, "%s/mondo.scratch.%ld", flag_val['S'],
673                random() % 32768);
674        bkpinfo->scratchdir = tmp;
675    }
676    if (flag_set['T']) {
677        asprintf(&tmp, "%s/tmp.mondo.%ld", flag_val['T'],
678                random() % 32768);
679        bkpinfo->tmpdir = tmp;
680        asprintf(&tmp, "touch %s/.foo.dat", flag_val['T']);
681        if (run_program_and_log_output(tmp, 1)) {
682            retval++;
683            log_to_screen
684                ("Please specify a tempdir which I can write to. :)");
685            fatal_error("I cannot write to the tempdir you specified.");
686        }
687        paranoid_free(tmp);
688
689        asprintf(&tmp, "ln -sf %s/.foo.dat %s/.bar.dat", flag_val['T'],
690                flag_val['T']);
691        if (run_program_and_log_output(tmp, 1)) {
692            retval++;
693            log_to_screen
694                ("Please don't specify a SAMBA or VFAT or NFS tmpdir.");
695            fatal_error("I cannot write to the tempdir you specified.");
696        }
697        paranoid_free(tmp);
698    }
699    if (flag_set['A']) {
700        asprintf(&tmp, flag_val['A']);
701        bkpinfo->call_after_iso = tmp;
702    }
703    if (flag_set['B']) {
704        asprintf(&tmp, flag_val['B']);
705        bkpinfo->call_before_iso = tmp;
706    }
707    if (flag_set['F']) {
708        g_skip_floppies = TRUE;
709    }
710    if (flag_set['H']) {
711        g_cd_recovery = TRUE;
712    }
713    if (flag_set['l']) {
714#ifdef __FreeBSD__
715#  define BOOT_LOADER_CHARS "GLBMR"
716#else
717#  ifdef __IA64__
718#    define BOOT_LOADER_CHARS "GER"
719#  else
720#    define BOOT_LOADER_CHARS "GLR"
721#  endif
722#endif
723        if (!strchr
724            (BOOT_LOADER_CHARS,
725             (bkpinfo->boot_loader = flag_val['l'][0]))) {
726            log_msg(1, "%c? WTF is %c? I need G, L, E or R.",
727                    bkpinfo->boot_loader, bkpinfo->boot_loader);
728            fatal_error
729                ("Please specify GRUB, LILO, ELILO  or RAW with the -l switch");
730        }
731#undef BOOT_LOADER_CHARS
732    }
733    if (flag_set['f']) {
734        tmp = resolve_softlinks_to_get_to_actual_device_file(flag_val['f']);
735        bkpinfo->boot_device = tmp;
736    }
737    if (flag_set['Q']) {
738        if (tmp == NULL) {
739            printf("-f option required when using -Q\n");
740            finish(-1);
741        }
742        i = which_boot_loader(tmp);
743        log_msg(3, "boot loader is %c, residing at %s", i, tmp);
744        printf("boot loader is %c, residing at %s\n", i, tmp);
745        finish(0);
746    }
747    paranoid_free(tmp);
748
749    if (flag_set['P']) {
750        asprintf(&tmp, flag_val['P']);
751        bkpinfo->postnuke_tarball = tmp;
752    }
753    if (flag_set['L']) {
754        bkpinfo->use_lzo = TRUE;
755        if (run_program_and_log_output("which lzop", FALSE)) {
756            retval++;
757            log_to_screen
758                ("Please install LZOP. You can't use '-L' until you do.\n");
759        }
760    }
761
762    if (!flag_set['o']
763        &&
764        !run_program_and_log_output
765        ("grep -i suse /etc/issue.net | grep 64", TRUE)) {
766        bkpinfo->make_cd_use_lilo = TRUE;
767        log_to_screen
768            ("Forcing you to use LILO. SuSE 9.0 (64-bit) has a broken mkfs.vfat binary.");
769    }
770    if (flag_set['o']) {
771        bkpinfo->make_cd_use_lilo = TRUE;
772    }
773#ifndef __FreeBSD__
774    else {
775        if (!is_this_a_valid_disk_format("vfat")) {
776            bkpinfo->make_cd_use_lilo = TRUE;
777            log_to_screen
778                ("Your kernel appears not to support vfat filesystems. I am therefore");
779            log_to_screen
780                ("using LILO instead of SYSLINUX as the CD/floppy's boot loader.");
781        }
782        if (run_program_and_log_output("which mkfs.vfat", FALSE)) {
783            bkpinfo->make_cd_use_lilo = TRUE;
784#ifdef __IA32__
785            log_to_screen
786                ("Your filesystem is missing 'mkfs.vfat', so I cannot use SYSLINUX as");
787            log_to_screen
788                ("your boot loader. I shall therefore use LILO instead.");
789#endif
790#ifdef __IA64__
791            log_to_screen
792                ("Your filesystem is missing 'mkfs.vfat', so I cannot prepare the EFI");
793            log_to_screen("environment correctly. Please install it.");
794            fatal_error("Aborting");
795#endif
796        }
797#ifdef __IA64__
798        /* We force ELILO usage on IA64 */
799        bkpinfo->make_cd_use_lilo = TRUE;
800#endif
801    }
802#endif
803
804    if (bkpinfo->make_cd_use_lilo && !does_file_exist("/boot/boot.b")) {
805        paranoid_system("touch /boot/boot.b");
806    }
807
808    i = flag_set['O'] + flag_set['V'];
809    if (i == 0) {
810        retval++;
811        log_to_screen("Specify backup (-O), verify (-V) or both (-OV).\n");
812    }
813
814/* and finally... */
815
816    return (retval);
817}
818
819
820
821/**
822 * Get the switches from @p argc and @p argv using getopt() and place them in
823 * @p flag_set and @p flag_val.
824 * @param argc The argument count (@p argc passed to main()).
825 * @param argv The argument vector (@p argv passed to main()).
826 * @param flag_val An array indexed by switch letter - if a switch is set and
827 * has an argument then set flag_val[switch] to that argument.
828 * @param flag_set An array indexed by switch letter - if a switch is set then
829 * set flag_set[switch] to TRUE, else set it to FALSE.
830 * @return The number of problems with the command line (0 for success).
831 */
832int
833retrieve_switches_from_command_line(int argc, char *argv[],
834                                    char *flag_val[128],
835                                    bool flag_set[128])
836{
837    /*@ ints ** */
838    int opt = 0;
839    char *tmp;
840    int i = 0;
841    int len;
842
843    /*@ bools *** */
844    bool bad_switches = FALSE;
845
846    assert(flag_val != NULL);
847    assert(flag_set != NULL);
848
849    for (i = 0; i < 128; i++) {
850        flag_val[i] = NULL;
851        flag_set[i] = FALSE;
852    }
853    while ((opt =
854            getopt(argc, argv,
855                   "0123456789A:B:C:DE:FHI:J:K:LNOP:QRS:T:VWb:c:d:ef:gik:l:mn:op:rs:tuw:x:"))
856           != -1) {
857        if (opt == '?') {
858            bad_switches = TRUE;
859            /*log_it("Invalid option: %c\n",optopt); */
860        } else {
861            if (flag_set[optopt]) {
862                bad_switches = TRUE;
863                asprintf(&tmp, "Switch -%c previously defined as %s\n", opt,
864                        flag_val[i]);
865                log_to_screen(tmp);
866                paranoid_free(tmp);
867            } else {
868                flag_set[opt] = TRUE;
869                if (optarg) {
870                    len = strlen(optarg);
871                    if (optarg[0] != '/' && optarg[len - 1] == '/') {
872                        optarg[--len] = '\0';
873                        log_to_screen
874                            ("Warning - param '%s' should not have trailing slash!",
875                             optarg);
876                    }
877                    if (opt == 'd') {
878                        if (strchr(flag_val[opt], '/')
879                            && flag_val[opt][0] != '/') {
880                            asprintf(&tmp,
881                                    "-%c flag --- must be absolute path --- '%s' isn't absolute",
882                                    opt, flag_val[opt]);
883                            log_to_screen(tmp);
884                            paranoid_free(tmp);
885                            bad_switches = TRUE;
886                        }
887                    }
888                    asprintf(&flag_val[opt], optarg);
889                }
890            }
891        }
892    }
893    for (i = optind; i < argc; i++) {
894        bad_switches = TRUE;
895        asprintf(&tmp, "Invalid arg -- %s\n", argv[i]);
896        log_to_screen(tmp);
897        paranoid_free(tmp);
898    }
899    return (bad_switches);
900}
901
902
903
904
905/**
906 * Print a not-so-helpful help message and exit.
907 */
908void help_screen()
909{
910    log_msg(1, "Type 'man mondo-archive' for more information\n");
911    exit(1);
912}
913
914
915/**
916 * Terminate Mondo in response to a signal.
917 * @param sig The signal number received.
918 */
919void terminate_daemon(int sig)
920{
921    char *tmp;
922    char *tmp2;
923
924    switch (sig) {
925    case SIGINT:
926        asprintf(&tmp, "SIGINT signal received from OS");
927        asprintf(&tmp2, "You interrupted me :-)");
928        break;
929    case SIGKILL:
930        asprintf(&tmp, "SIGKILL signal received from OS");
931        asprintf(&tmp2,
932               "I seriously have no clue how this signal even got to me. Something's wrong with your system.");
933        break;
934    case SIGTERM:
935        asprintf(&tmp, "SIGTERM signal received from OS");
936        asprintf(&tmp2, "Got terminate signal");
937        break;
938    case SIGHUP:
939        asprintf(&tmp, "SIGHUP signal received from OS");
940        asprintf(&tmp2, "Hangup on line");
941        break;
942    case SIGSEGV:
943        asprintf(&tmp, "SIGSEGV signal received from OS");
944        asprintf(&tmp2,
945               "Internal programming error. Please send a backtrace as well as your log.");
946        break;
947    case SIGPIPE:
948        asprintf(&tmp, "SIGPIPE signal received from OS");
949        asprintf(&tmp2, "Pipe was broken");
950        break;
951    case SIGABRT:
952        asprintf(&tmp, "SIGABRT signal received from OS");
953        asprintf(&tmp2,
954                "Abort - probably failed assertion. I'm sleeping for a few seconds so you can read the message.");
955        break;
956    default:
957        asprintf(&tmp, "(Unknown)");
958        asprintf(&tmp2, "(Unknown)");
959    }
960
961    log_to_screen(tmp);
962    log_to_screen(tmp2);
963    paranoid_free(tmp);
964    paranoid_free(tmp2);
965    if (sig == SIGABRT) {
966        sleep(10);
967    }
968    kill_buffer();
969    fatal_error
970        ("Mondoarchive is terminating in response to a signal from the OS");
971    finish(254);                // just in case
972}
973
974
975/**
976 * Turn signal-trapping on or off.
977 * @param on If TRUE, turn it on; if FALSE, turn it off (we still trap it, just don't do as much).
978 */
979void set_signals(int on)
980{
981    int signals[] =
982        { SIGTERM, SIGHUP, SIGTRAP, SIGABRT, SIGINT, SIGKILL, SIGSTOP, 0 };
983    int i;
984
985    signal(SIGPIPE, sigpipe_occurred);
986    for (i = 0; signals[i]; i++) {
987        if (on) {
988            signal(signals[i], terminate_daemon);
989        } else {
990            signal(signals[i], termination_in_progress);
991        }
992    }
993}
994
995
996/**
997 * Exit immediately without cleaning up.
998 * @param sig The signal we are exiting due to.
999 */
1000void termination_in_progress(int sig)
1001{
1002    log_msg(1, "Termination in progress");
1003    usleep(1000);
1004    pthread_exit(0);
1005}
1006
1007/* @} - end of cliGroup */
Note: See TracBrowser for help on using the repository browser.