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

Last change on this file since 262 was 262, checked in by bcornec, 13 years ago

merge -r253:261 $SVN_M/branches/2.05

  • Property svn:keywords set to Id
File size: 12.2 KB
Line 
1/*
2 * $Id: main.c 262 2005-12-23 16:25:52Z 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 262 2005-12-23 16:25:52Z 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    /* Add MONDO_LIB environment variable for mindi */
175    setenv_mondo_lib();
176
177    unlink(MONDO_LOGFILE);
178
179/* Configure the bkpinfo structure, global file paths, etc. */
180    g_main_pid = getpid();
181    log_msg(9, "This");
182
183    register_pid(g_main_pid, "mondo");
184    set_signals(TRUE);          // catch SIGTERM, etc.
185    run_program_and_log_output("date", 1);
186    run_program_and_log_output("dmesg -n1", TRUE);
187
188    log_msg(9, "Next");
189    welcome_to_mondoarchive();
190    distro_specific_kludges_at_start_of_mondoarchive();
191    // BERLIOS : too early, bkpinfo is not initialized ??
192    //sprintf(g_erase_tmpdir_and_scratchdir, "rm -Rf %s %s", bkpinfo->tmpdir, bkpinfo->scratchdir);
193    g_kernel_version = get_kernel_version();
194
195    if (argc == 4 && !strcmp(argv[1], "getfattr")) {
196        g_loglevel = 10;
197        g_text_mode = TRUE;
198        setup_newt_stuff();
199        if (!strstr(argv[2], "filelist")) {
200            printf("Sorry - filelist goes first\n");
201            finish(1);
202        } else {
203            finish(get_fattr_list(argv[2], argv[3]));
204        }
205        finish(0);
206    }
207    if (argc == 4 && !strcmp(argv[1], "setfattr")) {
208        g_loglevel = 10;
209        g_text_mode = TRUE;
210        setup_newt_stuff();
211        finish(set_fattr_list(argv[2], argv[3]));
212    }
213
214    if (argc == 3 && !strcmp(argv[1], "wildcards")) {
215        g_loglevel = 10;
216        g_text_mode = TRUE;
217        setup_newt_stuff();
218        turn_wildcard_chars_into_literal_chars(tmp, argv[2]);
219        printf("in=%s; out=%s\n", argv[2], tmp);
220        paranoid_free(tmp);
221        finish(1);
222    }
223
224    if (argc == 4 && !strcmp(argv[1], "getfacl")) {
225        g_loglevel = 10;
226        g_text_mode = TRUE;
227        setup_newt_stuff();
228        if (!strstr(argv[2], "filelist")) {
229            printf("Sorry - filelist goes first\n");
230            finish(1);
231        } else {
232            finish(get_acl_list(argv[2], argv[3]));
233        }
234        finish(0);
235    }
236    if (argc == 4 && !strcmp(argv[1], "setfacl")) {
237        g_loglevel = 10;
238        g_text_mode = TRUE;
239        setup_newt_stuff();
240        finish(set_acl_list(argv[2], argv[3]));
241    }
242
243    if (argc > 2 && !strcmp(argv[1], "find-cd")) {
244        g_loglevel = 10;
245        g_text_mode = TRUE;
246        setup_newt_stuff();
247        if ((tmp = find_cdrw_device()) == NULL) {
248            printf("Failed to find CDR-RW drive\n");
249        } else {
250            printf("CD-RW is at %s\n", tmp);
251        }
252        paranoid_free(tmp);
253
254        if ((tmp = find_cdrom_device(FALSE)) == NULL) {
255            printf("Failed to find CD-ROM drive\n");
256        } else {
257            printf("CD-ROM is at %s\n", tmp);
258        }
259        paranoid_free(tmp);
260        finish(0);
261    }
262
263    if (argc > 2 && !strcmp(argv[1], "find-dvd")) {
264        g_loglevel = 10;
265        g_text_mode = TRUE;
266        setup_newt_stuff();
267        if ((tmp = find_dvd_device()) == NULL) {
268            printf("Failed to find DVD drive\n");
269        } else {
270            printf("DVD is at %s\n", tmp);
271        }
272        paranoid_free(tmp);
273        finish(0);
274    }
275
276    if (argc > 2 && !strcmp(argv[1], "disksize")) {
277        printf("%s --> %ld\n", argv[2], get_phys_size_of_drive(argv[2]));
278        finish(0);
279    }
280    if (argc > 2 && !strcmp(argv[1], "test-dev")) {
281        if (is_dev_an_NTFS_dev(argv[2])) {
282            printf("%s is indeed an NTFS dev\n", argv[2]);
283        } else {
284            printf("%s is _not_ an NTFS dev\n", argv[2]);
285        }
286        finish(0);
287    }
288
289    if (pre_param_configuration(bkpinfo)) {
290        fatal_error
291            ("Pre-param initialization phase failed. Please review the error messages above, make the specified changes, then try again. Exiting...");
292    }
293
294/* Process command line, if there is one. If not, ask user for info. */
295    if (argc == 1) {
296        g_text_mode = FALSE;
297        setup_newt_stuff();
298        res = interactively_obtain_media_parameters_from_user(bkpinfo, TRUE);   /* yes, archiving */
299        if (res) {
300            fatal_error
301                ("Syntax error. Please review the parameters you have supplied and try again.");
302        }
303    } else {
304        res = handle_incoming_parameters(argc, argv, bkpinfo);
305        if (res) {
306            printf
307                ("Errors were detected in the command line you supplied.\n");
308            printf("Please review the log file - " MONDO_LOGFILE "\n");
309            log_msg(1, "Mondoarchive will now exit.");
310            finish(1);
311        }
312        setup_newt_stuff();
313    }
314
315/* Finish configuring global structures */
316    if (post_param_configuration(bkpinfo)) {
317        fatal_error
318            ("Post-param initialization phase failed. Perhaps bad parameters were supplied to mondoarchive? Please review the documentation, error messages and logs. Exiting...");
319    }
320
321    log_to_screen
322        ("BusyBox's sources are available from http://www.busybox.net");
323
324    /* If we're meant to backup then backup */
325    if (bkpinfo->backup_data) {
326        res = backup_data(bkpinfo);
327        retval += res;
328        if (res) {
329            asprintf(&say_at_end,
330                   "Data archived. Please check the logs, just as a precaution. ");
331        } else {
332            asprintf(&say_at_end, "Data archived OK. ");
333        }
334    }
335
336/* If we're meant to verify then verify */
337    if (bkpinfo->verify_data) {
338        res = verify_data(bkpinfo);
339        if (res < 0) {
340            asprintf(&say_at_end, "%d difference%c found.", -res,
341                    (-res != 1) ? 's' : ' ');
342            res = 0;
343        }
344        retval += res;
345    }
346
347    /* Offer to write floppy disk images to physical disks */
348    if (bkpinfo->backup_data && !g_skip_floppies) {
349        res = offer_to_write_boot_floppies_to_physical_disks(bkpinfo);
350        retval += res;
351    }
352
353    /* Report result of entire operation (success? errors?) */
354    if (retval == 0) {
355        mvaddstr_and_log_it(g_currentY++, 0,
356                            "Backup and/or verify ran to completion. Everything appears to be fine.");
357    } else {
358        mvaddstr_and_log_it(g_currentY++, 0,
359                            "Backup and/or verify ran to completion. However, errors did occur.");
360    }
361
362    if (does_file_exist("/root/images/mindi/mondorescue.iso")) {
363        log_to_screen
364            ("/root/images/mindi/mondorescue.iso, a boot/utility CD, is available if you want it.");
365    }
366
367
368    if (length_of_file("/tmp/changed.files") > 2) {
369        if (g_text_mode) {
370            log_to_screen
371                ("Type 'less /tmp/changed.files' to see which files don't match the archives");
372        } else {
373            log_msg(1,
374                    "Type 'less /tmp/changed.files' to see which files don't match the archives");
375            log_msg(2, "Calling popup_changelist_from_file()");
376            popup_changelist_from_file("/tmp/changed.files");
377            log_msg(2, "Returned from popup_changelist_from_file()");
378        }
379    } else {
380        unlink("/tmp/changed.files");
381    }
382    log_to_screen(say_at_end);
383    paranoid_free(say_at_end);
384
385    asprintf(&tmp, "umount %s/tmpfs", bkpinfo->tmpdir);
386    run_program_and_log_output(tmp, TRUE);
387    paranoid_free(tmp);
388
389    sprintf(g_erase_tmpdir_and_scratchdir, "rm -Rf %s %s", bkpinfo->tmpdir,
390            bkpinfo->scratchdir);
391    run_program_and_log_output(g_erase_tmpdir_and_scratchdir, TRUE);
392
393    run_program_and_log_output("mount", 2);
394
395    system("rm -f /var/cache/mondo-archive/last-backup.aborted");
396    system("rm -Rf /tmp.mondo.* /mondo.scratch.*");
397    if (retval == 0) {
398        printf("Mondoarchive ran OK.\n");
399    } else {
400        printf("Errors occurred during backup. Please check logfile.\n");
401    }
402    distro_specific_kludges_at_end_of_mondoarchive();
403    register_pid(0, "mondo");
404    set_signals(FALSE);
405    chdir("/tmp");              // just in case there's something wrong with g_erase_tmpdir_and_scratchdir
406    system(g_erase_tmpdir_and_scratchdir);
407    free_libmondo_global_strings();
408    paranoid_free(bkpinfo);
409
410    unlink("/tmp/filelist.full");
411    unlink("/tmp/filelist.full.gz");
412
413    run_program_and_log_output("date", 1);
414
415    if (!g_text_mode) {
416        popup_and_OK
417            ("Mondo Archive has finished its run. Please press ENTER to return to the shell prompt.");
418        log_to_screen("See %s for details of backup run.", MONDO_LOGFILE);
419        finish(retval);
420    } else {
421        printf("See %s for details of backup run.\n", MONDO_LOGFILE);
422        exit(retval);
423    }
424
425    return EXIT_SUCCESS;
426}
Note: See TracBrowser for help on using the repository browser.