source: branches/2.2.9/mondo/src/common/libmondo-cli.c @ 2426

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