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

Last change on this file since 2698 was 2698, checked in by bruno, 9 years ago

r4174@localhost: bruno | 2011-01-26 07:04:13 +0100

  • Fix a bug in process_switches around exclude_paths check now that it's setup with NULL
  • Property svn:keywords set to Id
File size: 31.4 KB
Line 
1/***************************************************************************
2$Id: libmondo-cli.c 2698 2011-01-27 18:30:49Z 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@]*%s[/]* .*\" | cut -d' ' -f3", bkpinfo->netfs_user,
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 if (strstr(bkpinfo->netfs_proto, "nfs")) {
536                sprintf(tmp, "mount %s", bkpinfo->netfs_mount);
537            } else {
538                log_to_screen("Protocol %s not supported yet for network backups.\n", bkpinfo->netfs_proto);
539                fatal_error("Bad Protocol\n");
540            }
541            if (system(tmp)) {
542                log_to_screen("Unable to mount Network share %s. Please mount manually.\n", bkpinfo->netfs_mount);
543                retval++;
544            } else {
545                sprintf(tmp, "mount | grep -E \"^[%s@]*%s[/]* .*\" | cut -d' ' -f3", bkpinfo->netfs_user,
546                    bkpinfo->netfs_mount);
547                strncpy(bkpinfo->isodir,
548                        call_program_and_get_last_line_of_output(tmp),
549                        MAX_STR_LEN / 4);
550                if (strlen(bkpinfo->isodir) < 3) {
551                    retval++;
552                    log_to_screen("Network share %s is strangely not mounted. Please mount manually...\n", bkpinfo->netfs_mount);
553                }
554            }
555        }
556        log_msg(3, "proto = %s", bkpinfo->netfs_proto);
557        log_msg(3, "mount = %s", bkpinfo->netfs_mount);
558        if (bkpinfo->netfs_user) {
559            log_msg(3, "user = %s", bkpinfo->netfs_user);
560        }
561        log_msg(3, "isodir= %s", bkpinfo->isodir);
562    }
563
564    if (flag_set['c']) {
565        bkpinfo->backup_media_type = cdr;
566    }
567    if (flag_set['C']) {
568        bkpinfo->backup_media_type = cdstream;
569    }
570    if (flag_set['i']) {
571        bkpinfo->backup_media_type = iso;
572    }
573    if (flag_set['n']) {
574        bkpinfo->backup_media_type = netfs;
575        /* Never try to eject a Network device */
576        bkpinfo->please_dont_eject = TRUE;
577    }
578    if (flag_set['r']) {
579        bkpinfo->backup_media_type = dvd;
580    }
581    if (flag_set['t']) {
582        bkpinfo->backup_media_type = tape;
583    }
584    if (flag_set['u']) {
585        bkpinfo->backup_media_type = udev;
586    }
587    if (flag_set['w']) {
588        bkpinfo->backup_media_type = cdrw;
589    }
590    if (flag_set['U']) {
591        bkpinfo->backup_media_type = usb;
592        /* Never try to eject a USB device */
593        bkpinfo->please_dont_eject = TRUE;
594    }
595    if (flag_set['z']) {
596        if (find_home_of_exe("getfattr")) {
597            mr_asprintf(&g_getfattr,"getfattr");
598        }
599        if (find_home_of_exe("getfacl")) {
600            mr_asprintf(&g_getfacl,"getfacl");
601        }
602    }
603
604    /* optional, popular */
605    if (flag_set['g']) {
606        g_text_mode = FALSE;
607    }
608
609    if (flag_set['E']) {
610        if (bkpinfo->exclude_paths && (bkpinfo->exclude_paths[0] == '-')) {
611            retval++;
612            log_to_screen("Please supply a sensible value with '-E'\n");
613        }
614        mr_strcat(bkpinfo->exclude_paths, " ");
615        mr_asprintf(&tmp1, "%s", flag_val['E']);
616
617        p = tmp1;
618        q = tmp1;
619
620        /* Cut the flag_val['E'] in parts containing all paths to test them */
621        while (p != NULL) {
622            q = strchr(p, ' ');
623            if (q != NULL) {
624                *q = '\0';
625                /* Fix bug 14 where ending / cause a problem later
626                * so handled here for the moment */
627                q--;
628                if (*q == '/') {
629                    *q = '\0';
630                }
631                q++;
632                /* End of bug fix */
633                if ((stat(p, &buf) != 0) && (! bkpinfo->restore_data)) {
634                    log_msg(1, "WARNING ! %s doesn't exist", p);
635                }
636                p = q+1 ;
637            } else {
638                if ((stat(p, &buf) != 0) && (! bkpinfo->restore_data)) {
639                    log_msg(1, "WARNING ! %s doesn't exist", p);
640                }
641                p = NULL;
642            }
643        }
644        paranoid_free(tmp1);
645        lastpos = 0;
646
647        mr_make_devlist_from_pathlist(flag_val['E'], 'E');
648        log_msg(4, "Finished with the -E option");
649    }
650
651    if (flag_set['e']) {
652        bkpinfo->please_dont_eject = TRUE;
653    }
654
655    if ((flag_set['N']) && (! bkpinfo->restore_data))       // exclude Network mounts & devices
656    {
657        mr_asprintf(&psz, "%s", list_of_NETFS_mounts_only());
658        mr_strcat(bkpinfo->exclude_paths, "%s ", psz);
659        mr_free(psz);
660
661        if (bkpinfo->exclude_paths != NULL) {
662            log_msg(3, "-N means we're now excluding %s", bkpinfo->exclude_paths);
663        }
664    }
665
666    if (flag_set['b']) {
667        mr_asprintf(&psz, "%s", flag_val['b']);
668        log_msg(1, "psz = '%s'", psz);
669        if (psz[strlen(psz) - 1] == 'k') {
670            psz[strlen(psz) - 1] = '\0';
671            itbs = atol(psz) * 1024L;
672        } else {
673            itbs = atol(psz);
674        }
675        mr_free(psz);
676        log_msg(1, "'%s' --> %ld", flag_val['b'], itbs);
677        log_msg(1, "Internal tape block size is now %ld bytes", itbs);
678        if (itbs % 512 != 0 || itbs < 256 || itbs > 1024L * 1024) {
679            fatal_error
680                ("Are you nuts? Silly, your internal tape block size is. Abort, I shall.");
681        }
682        bkpinfo->internal_tape_block_size = itbs;
683    }
684
685    if ((flag_set['D']) && (! bkpinfo->restore_data)) {
686        bkpinfo->differential = 1;
687//      bkpinfo->differential = atoi (flag_val['D']);
688        if ((bkpinfo->differential < 1) || (bkpinfo->differential > 9)) {
689            fatal_error
690                ("The D option should be between 1 and 9 inclusive");
691        }
692    }
693
694    if (flag_set['x']) {
695        strncpy(bkpinfo->image_devs, flag_val['x'], MAX_STR_LEN / 4);
696        if ((run_program_and_log_output("which ntfsclone", 2)) && (! bkpinfo->restore_data)) {
697            fatal_error("Please install ntfsprogs package/tarball.");
698        }
699    }
700
701    if (flag_set['m']) {
702        bkpinfo->manual_cd_tray = TRUE;
703    }
704
705    if ((flag_set['k']) && (! bkpinfo->restore_data)) {
706        strncpy(bkpinfo->kernel_path, flag_val['k'], MAX_STR_LEN);
707        if (!strcmp(bkpinfo->kernel_path, "failsafe")) {
708            strcpy(bkpinfo->kernel_path, "FAILSAFE");
709        }
710        if (strcmp(bkpinfo->kernel_path, "FAILSAFE")
711            && !does_file_exist(bkpinfo->kernel_path)) {
712            retval++;
713            sprintf(tmp,
714                    "You specified kernel '%s', which does not exist\n",
715                    bkpinfo->kernel_path);
716            log_to_screen(tmp);
717        }
718    }
719
720    if (flag_set['p']) {
721        strncpy(bkpinfo->prefix, flag_val['p'], MAX_STR_LEN / 4);
722        log_msg(1,"Prefix forced to %s",bkpinfo->prefix);
723    }
724
725    if (flag_set['d']) {        /* backup directory (if ISO/NETFS) */
726        if (flag_set['i']) {
727            strncpy(bkpinfo->isodir, flag_val['d'], MAX_STR_LEN / 4);
728            sprintf(tmp, "ls -l %s", bkpinfo->isodir);
729            if (run_program_and_log_output(tmp, 2)) {
730                fatal_error
731                    ("output folder does not exist - please create it");
732            }
733        } else if (flag_set['n']) {
734            strncpy(bkpinfo->netfs_remote_dir, flag_val['d'], MAX_STR_LEN);
735        } else {                /* backup device (if tape/CD-R/CD-RW) */
736            strncpy(bkpinfo->media_device, flag_val['d'], MAX_STR_LEN / 4);
737        }
738    }
739
740    if ((flag_set['n']) && (! bkpinfo->restore_data)) {
741        mr_asprintf(&tmp1,"%s/%s/.dummy.txt", bkpinfo->isodir,bkpinfo->netfs_remote_dir);
742        if ((bkpinfo->netfs_user) && (strstr(bkpinfo->netfs_proto,"nfs"))) {
743            sprintf(tmp, "su - %s -c \"echo hi > %s\"", bkpinfo->netfs_user, tmp1);
744        } else {
745            sprintf(tmp, "echo hi > %s", tmp1);
746        }
747        if (run_program_and_log_output(tmp, 2)) {
748            retval++;
749            sprintf(tmp,
750                    "Are you sure directory '%s' exists in remote dir '%s'?\nIf so, do you have rights to write to it?\n",
751                    bkpinfo->netfs_remote_dir, bkpinfo->netfs_mount);
752            log_to_screen(tmp);
753        }
754        unlink(tmp1);
755        paranoid_free(tmp1);
756    }
757
758    if (!flag_set['d']
759        && (flag_set['c'] || flag_set['w'] || flag_set['C'])) {
760        if (g_kernel_version >= 2.6) {
761            if (popup_and_get_string
762                ("Device", "Please specify the device",
763                 bkpinfo->media_device, MAX_STR_LEN / 4)) {
764                retval++;
765                log_to_screen("User opted to cancel.");
766            }
767        } else if (find_cdrw_device(bkpinfo->media_device)) {
768            retval++;
769            log_to_screen
770                ("Tried and failed to find CD-R[W] drive automatically.\n");
771        } else {
772            flag_set['d'] = TRUE;
773            strncpy(flag_val['d'], bkpinfo->media_device, MAX_STR_LEN / 4);
774        }
775    }
776
777    if ((!flag_set['d'] && !flag_set['n'] && !flag_set['C']) && (! bkpinfo->restore_data)) {
778        retval++;
779        log_to_screen("Please specify the backup device/directory.\n");
780        fatal_error
781            ("You didn't use -d to specify the backup device/directory.");
782    }
783
784    for (i = '0'; i <= '9'; i++) {
785        if (flag_set[i]) {
786            bkpinfo->compression_level = i - '0';
787        }                       /* not '\0' but '0' */
788    }
789
790    if (flag_set['S']) {
791        sprintf(bkpinfo->scratchdir, "%s/mondo.scratch.%ld", flag_val['S'],
792                random() % 32768);
793    }
794
795    if (flag_set['T']) {
796        setup_tmpdir(flag_val['T']);
797        sprintf(tmp, "touch %s/.foo.dat", bkpinfo->tmpdir);
798        if (run_program_and_log_output(tmp, 1)) {
799            retval++;
800            log_to_screen
801                ("Please specify a tempdir which I can write to. :)");
802            fatal_error("I cannot write to the tempdir you specified.");
803        }
804        sprintf(tmp, "ln -sf %s/.foo.dat %s/.bar.dat", bkpinfo->tmpdir, bkpinfo->tmpdir);
805        if (run_program_and_log_output(tmp, 1)) {
806            retval++;
807            log_to_screen
808                ("Please don't specify a SAMBA or VFAT or NFS tmpdir.");
809            fatal_error("I cannot write to the tempdir you specified.");
810        }
811    }
812
813    if ((flag_set['A']) && (! bkpinfo->restore_data)) {
814        strncpy(bkpinfo->call_after_iso, flag_val['A'], MAX_STR_LEN);
815    }
816
817    if ((flag_set['B']) && (! bkpinfo->restore_data)) {
818        strncpy(bkpinfo->call_before_iso, flag_val['B'], MAX_STR_LEN);
819    }
820
821    if ((flag_set['H']) && (! bkpinfo->restore_data)) {
822        g_cd_recovery = TRUE;
823    }
824
825    if ((flag_set['l']) && (! bkpinfo->restore_data)) {
826#ifdef __FreeBSD__
827#  define BOOT_LOADER_CHARS "GLBMR"
828#else
829#  ifdef __IA64__
830#    define BOOT_LOADER_CHARS "GER"
831#  else
832#    define BOOT_LOADER_CHARS "GLR"
833#  endif
834#endif
835        if (!strchr
836            (BOOT_LOADER_CHARS,
837             (bkpinfo->boot_loader = flag_val['l'][0]))) {
838            log_msg(1, "%c? WTF is %c? I need G, L, E or R.",
839                    bkpinfo->boot_loader, bkpinfo->boot_loader);
840            fatal_error
841                ("Please specify GRUB, LILO, ELILO  or RAW with the -l switch");
842        }
843#undef BOOT_LOADER_CHARS
844    }
845
846    if (flag_set['f']) {
847        strncpy(bkpinfo->boot_device,
848                resolve_softlinks_to_get_to_actual_device_file(flag_val
849                                                               ['f']),
850                MAX_STR_LEN / 4);
851    }
852
853    if ((flag_set['P']) && (! bkpinfo->restore_data)) {
854        strncpy(bkpinfo->postnuke_tarball, flag_val['P'], MAX_STR_LEN);
855    }
856
857    if (flag_set['Q']) {
858        i = which_boot_loader(tmp);
859        log_msg(3, "boot loader is %c, residing at %s", i, tmp);
860        printf("boot loader is %c, residing at %s\n", i, tmp);
861        finish(0);
862    }
863
864    if ((flag_set['L']) && (! bkpinfo->restore_data)) {
865        bkpinfo->use_lzo = TRUE;
866        if (run_program_and_log_output("which lzop", 2)) {
867            retval++;
868            log_to_screen
869                ("Please install LZOP. You can't use '-L' until you do.\n");
870        }
871    }
872
873    if ((flag_set['G']) && (! bkpinfo->restore_data)) {
874        bkpinfo->use_gzip = TRUE;
875        if (run_program_and_log_output("which gzip", 2)) {
876            retval++;
877            log_to_screen
878                ("Please install gzip. You can't use '-G' until you do.\n");
879        }
880    }
881
882    bkpinfo->use_obdr = FALSE;
883    if (flag_set['o']) {
884        if ((!flag_set['t']) && (! bkpinfo->restore_data)) {
885            log_to_screen("OBDR support is only available for tapes. Use the -t option");
886            fatal_error("Aborting");
887        }
888        bkpinfo->use_obdr = TRUE;
889    }
890
891#ifndef __FreeBSD__
892    if ((!is_this_a_valid_disk_format("vfat")) && (! bkpinfo->restore_data)) {
893        bkpinfo->make_cd_use_lilo = TRUE;
894        log_to_screen
895            ("Your kernel appears not to support vfat filesystems. I am therefore");
896        log_to_screen
897            ("using LILO instead of SYSLINUX as the media boot loader.");
898    }
899    if ((run_program_and_log_output("which mkfs.vfat", 2)) && (! bkpinfo->restore_data)) {
900        bkpinfo->make_cd_use_lilo = TRUE;
901#ifdef __IA32__
902        log_to_screen
903            ("Your filesystem is missing 'mkfs.vfat', so I cannot use SYSLINUX as");
904        log_to_screen
905            ("your boot loader. I shall therefore use LILO instead.");
906#endif
907#ifdef __IA64__
908        log_to_screen
909            ("Your filesystem is missing 'mkfs.vfat', so I cannot prepare the EFI");
910        log_to_screen("environment correctly. Please install it.");
911        fatal_error("Aborting");
912#endif
913    }
914#ifdef __IA64__
915    /* We force ELILO usage on IA64 */
916    bkpinfo->make_cd_use_lilo = TRUE;
917#endif
918#endif
919
920    if (! bkpinfo->restore_data) {
921        i = flag_set['O'] + flag_set['V'];
922        if (i == 0) {
923            retval++;
924            log_to_screen("Specify backup (-O), verify (-V) or both (-OV).\n");
925        }
926    }
927
928    if ((! bkpinfo->restore_data) && (flag_set['Z'])) {
929            fatal_error
930                ("The -Z switch is only valid in restore mode");
931    }
932
933    if (flag_set['Z']) {
934        if (! strcmp(flag_val['Z'], "nuke")) {
935            bkpinfo->restore_mode = nuke;
936        } else if (! strcmp(flag_val['Z'], "interactive")) {
937            bkpinfo->restore_mode = interactive;
938        } else if (! strcmp(flag_val['Z'], "compare")) {
939            bkpinfo->restore_mode = compare;
940        } else if (! strcmp(flag_val['Z'], "mbr")) {
941            bkpinfo->restore_mode = mbr;
942        } else if (! strcmp(flag_val['Z'], "iso")) {
943            bkpinfo->restore_mode = isoonly;
944        } else if (! strcmp(flag_val['Z'], "isonuke")) {
945            bkpinfo->restore_mode = isonuke;
946        } else {
947            bkpinfo->restore_mode = interactive;
948        }
949    }
950
951/* and finally... */
952
953    paranoid_free(tmp);
954    return (retval);
955}
956
957
958
959/**
960 * Get the switches from @p argc and @p argv using getopt() and place them in
961 * @p flag_set and @p flag_val.
962 * @param argc The argument count (@p argc passed to main()).
963 * @param argv The argument vector (@p argv passed to main()).
964 * @param flag_val An array indexed by switch letter - if a switch is set and
965 * has an argument then set flag_val[switch] to that argument.
966 * @param flag_set An array indexed by switch letter - if a switch is set then
967 * set flag_set[switch] to TRUE, else set it to FALSE.
968 * @return The number of problems with the command line (0 for success).
969 */
970int
971retrieve_switches_from_command_line(int argc, char *argv[],
972                                    char flag_val[128][MAX_STR_LEN],
973                                    bool flag_set[128])
974{
975    /*@ ints ** */
976    int opt = 0;
977    char *tmp = NULL;
978    int i = 0;
979    int len;
980
981    /*@ bools *** */
982    bool bad_switches = FALSE;
983
984    assert(flag_val != NULL);
985    assert(flag_set != NULL);
986
987    for (i = 0; i < 128; i++) {
988        flag_val[i][0] = '\0';
989        flag_set[i] = FALSE;
990    }
991    while ((opt =
992            getopt(argc, argv, MONDO_OPTIONS))
993           != -1) {
994        if (opt == '?') {
995            bad_switches = TRUE;
996            /*log_it("Invalid option: %c\n",optopt); */
997        } else {
998            if (flag_set[opt]) {
999                bad_switches = TRUE;
1000                mr_asprintf(&tmp, "Switch -%c previously defined as %s\n", opt, flag_val[opt]);
1001                log_to_screen(tmp);
1002                paranoid_free(tmp);
1003            } else {
1004                flag_set[opt] = TRUE;
1005                if (optarg) {
1006                    len = strlen(optarg);
1007                    if (optarg[0] != '/' && optarg[len - 1] == '/') {
1008                        optarg[--len] = '\0';
1009                        log_to_screen
1010                            ("Warning - param '%s' should not have trailing slash!",
1011                             optarg);
1012                    }
1013                    if (opt == 'd') {
1014                        if (strchr(flag_val[opt], '/')
1015                            && flag_val[opt][0] != '/') {
1016                            mr_asprintf(&tmp, "-%c flag --- must be absolute path --- '%s' isn't absolute", opt, flag_val[opt]);
1017                            log_to_screen(tmp);
1018                            paranoid_free(tmp);
1019                            bad_switches = TRUE;
1020                        }
1021                    }
1022                    strcpy(flag_val[opt], optarg);
1023                }
1024            }
1025        }
1026    }
1027    for (i = optind; i < argc; i++) {
1028        bad_switches = TRUE;
1029        mr_asprintf(&tmp, "Invalid arg -- %s\n", argv[i]);
1030        log_to_screen(tmp);
1031        paranoid_free(tmp);
1032    }
1033    return (bad_switches);
1034}
1035
1036
1037
1038
1039/**
1040 * Print a not-so-helpful help message and exit.
1041 */
1042void help_screen()
1043{
1044    log_msg(1, "Type 'man mondoarchive' for more information\n");
1045    exit(1);
1046}
1047
1048
1049/**
1050 * Terminate Mondo in response to a signal.
1051 * @param sig The signal number received.
1052 */
1053void terminate_daemon(int sig)
1054{
1055    char *tmp = NULL;
1056    char *tmp2 = NULL;
1057
1058    switch (sig) {
1059    case SIGINT:
1060        mr_asprintf(&tmp, "SIGINT");
1061        mr_asprintf(&tmp2, "You interrupted me :-)");
1062        break;
1063    case SIGKILL:
1064        mr_asprintf(&tmp, "SIGKILL");
1065        mr_asprintf(&tmp2, "I seriously have no clue how this signal even got to me. Something's wrong with your system.");
1066        break;
1067    case SIGTERM:
1068        mr_asprintf(&tmp, "SIGTERM");
1069        mr_asprintf(&tmp2, "Got terminate signal");
1070        break;
1071    case SIGHUP:
1072        mr_asprintf(&tmp, "SIGHUP");
1073        mr_asprintf(&tmp2, "Hangup on line");
1074        break;
1075    case SIGSEGV:
1076        mr_asprintf(&tmp, "SIGSEGV");
1077        mr_asprintf(&tmp2, "Internal programming error. Please send a backtrace as well as your log.");
1078        break;
1079    case SIGPIPE:
1080        mr_asprintf(&tmp, "SIGPIPE");
1081        mr_asprintf(&tmp2, "Pipe was broken");
1082        break;
1083    case SIGABRT:
1084        mr_asprintf(&tmp, "SIGABRT");
1085        mr_asprintf(&tmp2, "Abort - probably failed assertion. I'm sleeping for a few seconds so you can read the message.");
1086        break;
1087    default:
1088        mr_asprintf(&tmp, "(Unknown)");
1089        mr_asprintf(&tmp2, "(Unknown)");
1090    }
1091
1092    mr_strcat(tmp, " signal received from OS");
1093    log_to_screen(tmp);
1094    paranoid_free(tmp);
1095
1096    log_to_screen(tmp2);
1097    paranoid_free(tmp2);
1098    if (sig == SIGABRT) {
1099        sleep(10);
1100    }
1101    kill_buffer();
1102
1103    free_MR_global_filenames();
1104
1105    fatal_error
1106        ("MondoRescue is terminating in response to a signal from the OS");
1107    finish(254);                // just in case
1108}
1109
1110
1111
1112
1113/**
1114 * Turn signal-trapping on or off.
1115 * @param on If TRUE, turn it on; if FALSE, turn it off (we still trap it, just don't do as much).
1116 */
1117void set_signals(int on)
1118{
1119    int signals[] =
1120        { SIGTERM, SIGHUP, SIGTRAP, SIGABRT, SIGINT, SIGKILL, SIGSTOP, 0 };
1121    int i;
1122
1123    signal(SIGPIPE, sigpipe_occurred);
1124    for (i = 0; signals[i]; i++) {
1125        if (on) {
1126            signal(signals[i], terminate_daemon);
1127        } else {
1128            signal(signals[i], termination_in_progress);
1129        }
1130    }
1131}
1132
1133
1134
1135
1136/**
1137 * Exit immediately without cleaning up.
1138 * @param sig The signal we are exiting due to.
1139 */
1140void termination_in_progress(int sig)
1141{
1142    log_msg(1, "Termination in progress");
1143    usleep(1000);
1144    pthread_exit(0);
1145}
1146
1147/* @} - end of cliGroup */
Note: See TracBrowser for help on using the repository browser.