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

Last change on this file since 1152 was 1152, checked in by Bruno Cornec, 17 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.