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

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

2 prototypes added
Free memory which was lost otherwise
(Patches from Michel Loiseleur <mloiseleur_at_linagora.com>)

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