source: MondoRescue/trunk/mondo/mondo/mondoarchive/mondo-cli.c@ 300

Last change on this file since 300 was 300, checked in by bcornec, 18 years ago

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