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

Last change on this file since 219 was 219, checked in by bcornec, 14 years ago

merge -r176:218 $SVN_M/branches/2.05

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