source: trunk/mondo/mondo/mondoarchive/main.c @ 252

Last change on this file since 252 was 252, checked in by bcornec, 14 years ago

merge -r249:251 $SVN_M/branches/2.05

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