source: branches/3.2/mondo/src/mondoarchive/mondoarchive.c @ 3468

Last change on this file since 3468 was 3468, checked in by bruno, 4 years ago
  • Avoids mondoarchive to crash on an allocation error when there is a syntax error
  • Property svn:keywords set to Id
File size: 13.1 KB
Line 
1/***************************************************************************
2$Id: mondoarchive.c 3468 2015-09-11 15:06:04Z bruno $
3* The main file for mondoarchive.
4*/
5
6/************************* #include statements *************************/
7#include <pthread.h>
8#include <stdio.h>
9#include <stdlib.h>
10#include "my-stuff.h"
11#include "mr_mem.h"
12#include "mr_str.h"
13#include "../common/mondostructures.h"
14#include "../common/libmondo.h"
15#include "../common/libmondo-cli-EXT.h"
16#include "../common/libmondo-tools-EXT.h"
17#include "mondoarchive.h"
18
19// for CVS
20//static char cvsid[] = "$Id: mondoarchive.c 3468 2015-09-11 15:06:04Z bruno $";
21
22/************************* external variables *************************/
23extern void set_signals(int);
24extern int g_current_media_number;
25extern int g_currentY;
26extern bool g_text_mode;
27extern char *g_boot_mountpt;
28extern bool g_remount_floppy_at_end;
29extern char *g_cdrw_drive_is_here;
30static char *g_cdrom_drive_is_here = NULL;
31static char *g_dvd_drive_is_here = NULL;
32extern double g_kernel_version;
33
34/***************** global vars, used only by main.c ******************/
35long diffs = 0L;
36
37extern t_bkptype g_backup_media_type;
38extern int g_loglevel;
39
40extern char *g_magicdev_command;
41
42/**
43 * Whether we're restoring from ISOs. Obviously not, since this is the
44 * backup program.
45 * @note You @b MUST declare this variable somewhere in your program if
46 * you use libmondo. Otherwise the link will fail.
47 * @ingroup globalGroup
48 */
49bool g_ISO_restore_mode = FALSE;
50
51/* Whether we should fail immediately at first error */
52bool g_fail_immediately = FALSE;
53
54/* Do we use extended attributes and acl ?
55 *  * By default no, use --acl & --attr options to force their usage */
56char *g_getfacl = NULL;
57char *g_getfattr = NULL;
58
59/* Reference to global bkpinfo */
60struct s_bkpinfo *bkpinfo;
61
62/* No cleanup for the moment */
63void (*mr_cleanup)(void) = NULL;
64
65/* To be coded */
66void free_MR_global_filenames(void) {
67}
68
69/****************** subroutines used only by main.c ******************/
70
71
72/**
73 * Print a "don't panic" message to the log and a message about the logfile to the screen.
74 */
75void welcome_to_mondoarchive(void)
76{
77    char *tmp = NULL;
78
79    log_msg(0, "Mondo Archive v%s --- http://www.mondorescue.org", PACKAGE_VERSION);
80    log_msg(0, "running %s binaries", get_architecture());
81    tmp = get_uname_m();
82    log_msg(0, "running on %s architecture", tmp);
83    mr_free(tmp);
84    log_msg(0, "-----------------------------------------------------------");
85    log_msg(0, "NB: Mondo logs almost everything, so don't panic if you see");
86    log_msg(0, "some error messages.  Please read them carefully before you");
87    log_msg(0, "decide to break out in a cold sweat.    Despite (or perhaps");
88    log_msg(0, "because of) the wealth of messages. some users are inclined");
89    log_msg(0, "to stop reading this log. If Mondo stopped for some reason,");
90    log_msg(0, "chances are it's detailed here.  More than likely there's a");
91    log_msg(0, "message at the very end of this log that will tell you what");
92    log_msg(0, "is wrong. Please read it!                          -Devteam");
93    log_msg(0, "-----------------------------------------------------------");
94
95    log_msg(0, "Zero...");
96    log_msg(1, "One...");
97    log_msg(2, "Two...");
98    log_msg(3, "Three...");
99    log_msg(4, "Four...");
100    log_msg(5, "Five...");
101    log_msg(6, "Six...");
102    log_msg(7, "Seven...");
103    log_msg(8, "Eight...");
104    printf("See %s for details of backup run.\n", MONDO_LOGFILE);
105}
106
107
108/**
109 * Do whatever is necessary to insure a successful backup on the Linux distribution
110 * of the day.
111 */
112void distro_specific_kludges_at_start_of_mondoarchive(void)
113{
114    log_msg(2, "Unmounting old ramdisks if necessary");
115    stop_magicdev_if_necessary();   // for RH+Gnome users
116    /*
117    run_program_and_log_output
118        ("umount `mount | grep shm | grep mondo | cut -d' ' -f3`", 2);
119        */
120    unmount_supermounts_if_necessary(); // for Mandrake users whose CD-ROMs are supermounted
121    //  stop_autofs_if_necessary(); // for Xandros users
122    mount_boot_if_necessary();  // for Gentoo users with non-mounted /boot partitions
123    clean_up_KDE_desktop_if_necessary();    // delete various misc ~/.* files that get in the way
124}
125
126
127
128/**
129 * Undo whatever was done by distro_specific_kludges_at_start_of_mondoarchive().
130 */
131void distro_specific_kludges_at_end_of_mondoarchive(void)
132{
133    log_msg(2, "Restarting magicdev if necessary");
134    sync();
135    restart_magicdev_if_necessary();    // for RH+Gnome users
136
137    log_msg(2, "Restarting autofs if necessary");
138    sync();
139    //  restart_autofs_if_necessary(); // for Xandros users
140
141    log_msg(2, "Restarting supermounts if necessary");
142    sync();
143    remount_supermounts_if_necessary(); // for Mandrake users
144
145    log_msg(2, "Unmounting /boot if necessary");
146    sync();
147    unmount_boot_if_necessary();    // for Gentoo users
148
149//  log_msg( 2, "Cleaning up KDE desktop");
150//  clean_up_KDE_desktop_if_necessary();
151}
152
153
154/**
155 * Backup/verify the user's data.
156 * What did you think it did, anyway? :-)
157 */
158int main(int argc, char *argv[])
159{
160    char *tmp = NULL;
161    char *tmp1 = NULL;
162    int res = 0;
163    int retval = 0;
164    char *say_at_end = NULL;
165    FILE *fin = NULL;
166
167    /* If -V, -v or --version then echo version no. and quit */
168    if (argc == 2 && (!strcmp(argv[argc - 1], "-v") || !strcmp(argv[argc - 1], "-V") || !strcmp(argv[argc - 1], "--version"))) {
169        printf("mondoarchive v%s\nSee man page for help\n", PACKAGE_VERSION);
170        if (getuid() != 0) fprintf(stderr, "Please run the mondoarchive command as root.\n");
171        exit(0);
172    }
173
174    /* Make sure I'm root; abort if not */
175    if (getuid() != 0) {
176        fprintf(stderr, "Please run as root.\n");
177        exit(127);
178    }
179
180    /* initialize log file with time stamp */
181    unlink(MONDO_LOGFILE);
182    log_msg(0, "Time started: %s", mr_date());
183
184    /* Initialize variables */
185    if (!(bkpinfo = (struct s_bkpinfo *)malloc(sizeof(struct s_bkpinfo)))) {
186        fprintf(stderr, "Cannot malloc bkpinfo\n");
187        exit(-1);
188    }
189    /*  Now on we can use finish to exit as bkpingo has been initialized */
190    log_msg(9, "reset_bkpinfo");
191    reset_bkpinfo();
192
193    /* Memory allocation is done in those functions */
194    malloc_libmondo_global_strings();
195
196    /* make sure PATH environmental variable allows access to mkfs, fdisk, etc. */
197    mr_asprintf(tmp1,"%s:/sbin:/usr/sbin:/usr/local/sbin",getenv("PATH"));
198    setenv("PATH", tmp1, 1);
199    mr_free(tmp1);
200
201    /* Add the ARCH environment variable for ia64 purposes */
202    mr_asprintf(tmp1,"%s",get_architecture());
203    setenv("ARCH", tmp1, 1);
204    mr_free(tmp1);
205
206    /* Add MONDO_SHARE environment variable for mindi */
207    setenv_mondo_share();
208
209    /* Configure the bkpinfo structure, global file paths, etc. */
210    g_main_pid = getpid();
211    log_msg(9, "This");
212
213    set_signals(TRUE);          // catch SIGTERM, etc.
214    run_program_and_log_output("dmesg -n1", TRUE);
215
216    log_msg(9, "Next");
217    make_hole_for_dir(MONDO_CACHE);
218
219    welcome_to_mondoarchive();
220    distro_specific_kludges_at_start_of_mondoarchive();
221    g_kernel_version = get_kernel_version();
222
223    if (argc == 4 && !strcmp(argv[1], "getfattr")) {
224        g_loglevel = 10;
225        g_text_mode = TRUE;
226        setup_newt_stuff();
227        if (!strstr(argv[2], "filelist")) {
228            printf("Sorry - filelist goes first\n");
229            finish(1);
230        } else {
231            finish(get_fattr_list(argv[2], argv[3]));
232        }
233        finish(0);
234    }
235    if (argc == 4 && !strcmp(argv[1], "setfattr")) {
236        g_loglevel = 10;
237        g_text_mode = TRUE;
238        setup_newt_stuff();
239        finish(set_fattr_list(argv[2], argv[3]));
240    }
241
242    if (argc == 3 && !strcmp(argv[1], "wildcards")) {
243        g_loglevel = 10;
244        g_text_mode = TRUE;
245        setup_newt_stuff();
246        tmp1 = mr_stresc(argv[2], "[]*?", '\\', '\'');
247        printf("in=%s; out=%s\n", argv[2], tmp1);
248        mr_free(tmp1);
249        finish(1);
250    }
251
252    if (argc == 4 && !strcmp(argv[1], "getfacl")) {
253        g_loglevel = 10;
254        g_text_mode = TRUE;
255        setup_newt_stuff();
256        if (!strstr(argv[2], "filelist")) {
257            printf("Sorry - filelist goes first\n");
258            finish(1);
259        } else {
260            finish(get_acl_list(argv[2], argv[3]));
261        }
262        finish(0);
263    }
264    if (argc == 4 && !strcmp(argv[1], "setfacl")) {
265        g_loglevel = 10;
266        g_text_mode = TRUE;
267        setup_newt_stuff();
268        finish(set_acl_list(argv[2], argv[3]));
269    }
270    if (argc >= 2 && !strcmp(argv[1], "mkraidtab")) {
271        g_loglevel = 10;
272        g_text_mode = TRUE;
273        setup_newt_stuff();
274#undef MDSTAT_FILE
275#define MDSTAT_FILE "/tmp/mdstat"
276        if (!(fin = fopen(MDSTAT_FILE, "r"))) {
277            log_msg(1, "Could not open %s.", MDSTAT_FILE);
278                finish(1);
279        }
280
281        create_raidtab_from_mdstat(MDSTAT_FILE,"/tmp/raidtab");
282        finish(0);
283    }
284
285    if (argc > 2 && !strcmp(argv[1], "find-cd")) {
286        g_loglevel = 10;
287        g_text_mode = TRUE;
288        setup_newt_stuff();
289        malloc_string(tmp);
290        if (find_cdrw_device(tmp)) {
291            printf("Failed to find CDR-RW drive\n");
292        } else {
293            printf("CD-RW is at %s\n", tmp);
294        }
295        tmp[0] = '\0';
296        if (find_cdrom_device(tmp, atoi(argv[2]))) {
297            printf("Failed to find CD-ROM drive\n");
298        } else {
299            printf("CD-ROM is at %s\n", tmp);
300        }
301        mr_free(tmp);
302        finish(0);
303    }
304
305    if (argc > 2 && !strcmp(argv[1], "find-dvd")) {
306        g_loglevel = 10;
307        g_text_mode = TRUE;
308        setup_newt_stuff();
309        malloc_string(tmp);
310        if (find_dvd_device(tmp, atoi(argv[2]))) {
311            printf("Failed to find DVD drive\n");
312        } else {
313            printf("DVD is at %s\n", tmp);
314        }
315        mr_free(tmp);
316        finish(0);
317    }
318
319    if (argc > 2 && !strcmp(argv[1], "disksize")) {
320        printf("%s --> %ld\n", argv[2], get_phys_size_of_drive(argv[2]));
321        finish(0);
322    }
323    if (argc > 2 && !strcmp(argv[1], "test-dev")) {
324        if (is_dev_an_NTFS_dev(argv[2])) {
325            printf("%s is indeed an NTFS dev\n", argv[2]);
326        } else {
327            printf("%s is _not_ an NTFS dev\n", argv[2]);
328        }
329        finish(0);
330    }
331
332    if (pre_param_configuration()) {
333        fatal_error("Pre-param initialization phase failed. Please review the error messages above, make the specified changes, then try again. Exiting...");
334    }
335
336    /* Process command line, if there is one. If not, ask user for info. */
337    setup_newt_stuff();
338    if (argc == 1) {
339        g_text_mode = FALSE;
340        res = interactively_obtain_media_parameters_from_user(TRUE);    /* yes, archiving */
341        if (res) {
342            fatal_error("Syntax error. Please review the parameters you have supplied and try again.");
343        }
344    } else {
345        res = handle_incoming_parameters(argc, argv);
346        if (res) {
347            printf("Errors were detected in the command line you supplied.\n");
348            printf("Please review the log file - %s\n", MONDO_LOGFILE );
349            log_msg(1, "Mondoarchive will now exit.");
350            finish(1);
351        }
352    }
353
354    /* Finish configuring global structures */
355    if (post_param_configuration()) {
356        fatal_error("Post-param initialization phase failed. Perhaps bad parameters were supplied to mondoarchive? Please review the documentation, error messages and logs. Exiting...");
357    }
358
359    /* If we're meant to backup then backup */
360    if (bkpinfo->backup_data) {
361        res = backup_data();
362        retval += res;
363        if (res) {
364            mr_asprintf(say_at_end, "Data archived. Please check the logs, just as a precaution. ");
365        } else {
366            mr_asprintf(say_at_end, "Data archived OK. ");
367        }
368    }
369
370/* If we're meant to verify then verify */
371    if (bkpinfo->verify_data) {
372        res = verify_data();
373        if (res < 0) {
374            mr_asprintf(tmp, "%d difference%c found.", -res, (-res != 1) ? 's' : ' ');
375            mr_asprintf(say_at_end, "%s", tmp);
376            log_to_screen(tmp);
377            mr_free(tmp);
378            res = 0;
379        }
380        retval += res;
381    }
382
383/* Report result of entire operation (success? errors?) */
384    if (!retval) {
385        mvaddstr_and_log_it(g_currentY++, 0, "Backup and/or verify ran to completion. Everything appears to be fine.");
386    } else {
387        mvaddstr_and_log_it(g_currentY++, 0, "Backup and/or verify ran to completion. However, errors did occur.");
388    }
389
390    if (does_file_exist(MINDI_CACHE"/mondorescue.iso")) {
391        log_to_screen(MINDI_CACHE"/mondorescue.iso, a boot/utility CD, is available if you want it.");
392    }
393
394    if (length_of_file(MONDO_CACHE"/changed.files") > 2) {
395        if (g_text_mode) {
396            log_to_screen("Type 'less "MONDO_CACHE"/changed.files' to see which files don't match the archives");
397        } else {
398            log_msg(1, "Type 'less "MONDO_CACHE"/changed.files' to see which files don't match the archives");
399            log_msg(2, "Calling popup_changelist_from_file()");
400            popup_changelist_from_file(MONDO_CACHE"/changed.files");
401            log_msg(2, "Returned from popup_changelist_from_file()");
402        }
403    } else {
404        unlink(MONDO_CACHE"/changed.files");
405    }
406    if (say_at_end != NULL) {
407        log_to_screen(say_at_end);
408        mr_free(say_at_end);
409    }
410    mr_asprintf(tmp, "umount %s/tmpfs", bkpinfo->tmpdir);
411    run_program_and_log_output(tmp, TRUE);
412    mr_free(tmp);
413    if (bkpinfo->backup_media_type == usb) {
414        log_msg(1, "Unmounting USB device.");
415        if (bkpinfo->media_device == NULL) {
416            fatal_error("USB device set to NULL");
417        }
418        mr_asprintf(tmp, "umount %s1", bkpinfo->media_device);
419        run_program_and_log_output(tmp, TRUE);
420        mr_free(tmp);
421    }
422
423    run_program_and_log_output("mount", 2);
424
425    paranoid_system("rm -f "MONDO_CACHE"/last-backup.aborted");
426    if (!retval) {
427        printf("Mondoarchive ran OK.\n");
428    } else {
429        printf("Errors occurred during backup. Please check logfile.\n");
430    }
431    distro_specific_kludges_at_end_of_mondoarchive();
432    set_signals(FALSE);
433
434    free_libmondo_global_strings();
435   
436
437    if (!g_cdrom_drive_is_here) {
438        log_msg(10, "FYI, g_cdrom_drive_is_here was never used");
439    }
440    if (!g_dvd_drive_is_here) {
441        log_msg(10, "FYI, g_dvd_drive_is_here was never used");
442    }
443
444    /* finalize log file with time stamp */
445    log_msg(0, "Time finished: %s", mr_date());
446
447    if (chdir("/tmp")) {
448        // FIXME
449    }
450
451    if (!g_text_mode) {
452        popup_and_OK("Mondo Archive has finished its run. Please press ENTER to return to the shell prompt.");
453        log_to_screen("See %s for details of backup run.", MONDO_LOGFILE);
454    } else {
455        printf("See %s for details of backup run.\n", MONDO_LOGFILE);
456    }
457    finish(retval);
458
459    return EXIT_SUCCESS;
460}
Note: See TracBrowser for help on using the repository browser.