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

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