1 | /*************************************************************************** |
---|
2 | * $Id: main.c 1143 2007-02-12 02:16:04Z 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 1143 2007-02-12 02:16:04Z bruno $"; |
---|
30 | |
---|
31 | /************************* external variables *************************/ |
---|
32 | extern void set_signals(int); |
---|
33 | extern int g_current_media_number; |
---|
34 | extern void register_pid(pid_t, char *); |
---|
35 | extern int g_currentY; |
---|
36 | extern bool g_text_mode; |
---|
37 | extern char *g_boot_mountpt; |
---|
38 | extern bool g_remount_cdrom_at_end, g_remount_floppy_at_end; |
---|
39 | extern char *g_mondo_home; |
---|
40 | extern char *g_tmpfs_mountpt; |
---|
41 | extern char *g_erase_tmpdir_and_scratchdir; |
---|
42 | extern char *g_cdrw_drive_is_here; |
---|
43 | static char *g_cdrom_drive_is_here = NULL; |
---|
44 | static char *g_dvd_drive_is_here = NULL; |
---|
45 | extern double g_kernel_version; |
---|
46 | |
---|
47 | /***************** global vars, used only by main.c ******************/ |
---|
48 | bool g_skip_floppies; |
---|
49 | long diffs; |
---|
50 | char *ps_options = "auxww"; |
---|
51 | char *ps_proc_id = "$2"; |
---|
52 | |
---|
53 | extern t_bkptype g_backup_media_type; |
---|
54 | extern 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 | */ |
---|
62 | void welcome_to_mondoarchive() |
---|
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 | |
---|
101 | extern 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 | */ |
---|
107 | void distro_specific_kludges_at_start_of_mondoarchive() |
---|
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 | */ |
---|
124 | void distro_specific_kludges_at_end_of_mondoarchive() |
---|
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 | */ |
---|
155 | int 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 | } |
---|