source: branches/stable/mondo/src/mondoarchive/main.c @ 1152

Last change on this file since 1152 was 1152, checked in by bruno, 13 years ago

More memory updates around mondoarchive (addition of mr_setenv) (based on ideas sent as patches by Michel)

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