source: trunk/mondo/mondo/common/libmondo-files.c @ 110

Last change on this file since 110 was 110, checked in by andree, 14 years ago

busybox ping wants a space between option '-c' and its argument.

  • Property svn:keywords set to Id
File size: 36.7 KB
Line 
1/* libmondo-files.c                                  file manipulation
2   $Id: libmondo-files.c 110 2005-11-06 14:05:44Z andree $
3.
4
5
604/16/04
7- find_home_of_exe() really does return NULL now if file not found
8
903/22/04
10- added mode_of_file()
11
1210/02/03
13- cleaned up grab_percentage_from_last_line_of_file()
14
1509/18
16- added int make_grub_install_scriptlet()
17
1809/16
19- cleaned up mkisofs feedback window
20
2109/12
22- fixed Teuton-hostile bug in size_of_all_biggiefiles_K()
23
2409/05
25- added size_of_partition_in_mountlist_K()
26- size_of_all_biggiefiles_K() now calls get_phys_size_of_drive(fname)
27
2807/02
29- fixed calls to popup_and_get_string()
30
3105/19
32- added CP_BIN
33
3405/05
35- added Joshua Oreman's FreeBSD patches
36
3705/04
38- find_home_of_exe() now returns NULL if file not found
39
4004/26
41- if >4 media est'd, say one meeeellion
42
4304/25
44- fixed minor bug in find_home_of_exe()
45
4604/24
47- added lots of assert()'s and log_OS_error()'s
48
4904/07
50- fix find_home_of_exe()
51- cleaned up code a bit
52
5303/27
54- copy_mondo_and_mindi_stuff --- if _homedir_/payload.tgz exists then untar it to CD
55
5601/14/2003
57- if backup media type == nfs then don't estimate no. of media reqd
58
5911/25/2002
60- don't log/echo estimated # of media required if >=50
61
6210/01 - 11/09
63- chmod uses 0x, not decimal :)
64- added is_this_file_compressed()
65- replace convoluted grep with wc (KP)
66
6709/01 - 09/30
68- only show "number of media" estimate if no -x
69- run_program_and_log_output() now takes boolean operator to specify
70  whether it will log its activities in the event of _success_
71
7208/01 - 08/31
73- handle unknown media sizes
74- cleaned up some log_it() calls
75
7607/24
77- created
78*/
79
80/**
81 * @file
82 * Functions to manipulate files.
83 */
84
85
86#include "my-stuff.h"
87#include "mondostructures.h"
88#include "libmondo-files.h"
89
90#include "lib-common-externs.h"
91
92#include "libmondo-tools-EXT.h"
93#include "libmondo-gui-EXT.h"
94#include "libmondo-devices-EXT.h"
95#include "libmondo-fork-EXT.h"
96#include "libmondo-string-EXT.h"
97
98/*@unused@*/
99//static char cvsid[] = "$Id: libmondo-files.c 110 2005-11-06 14:05:44Z andree $";
100
101extern char err_log_lines[NOOF_ERR_LINES][MAX_STR_LEN];
102
103extern int g_currentY;
104extern char *g_mondo_home;
105
106/**
107 * @addtogroup fileGroup
108 * @{
109 */
110/**
111 * Get an md5 checksum of the specified file.
112 * @param filename The file to checksum.
113 * @return The 32-character ASCII representation of the 128-bit checksum.
114 * @note The returned string points to static storage that will be overwritten with each call.
115 */
116char *calc_checksum_of_file(char *filename)
117{
118    /*@ buffers ***************************************************** */
119    static char output[MAX_STR_LEN];
120    char command[MAX_STR_LEN * 2];
121    char tmp[MAX_STR_LEN];
122
123    /*@ pointers **************************************************** */
124    char *p;
125    FILE *fin;
126
127    /*@ initialize pointers ***************************************** */
128
129    p = output;
130
131    /*@************************************************************** */
132
133    assert_string_is_neither_NULL_nor_zerolength(filename);
134    if (does_file_exist(filename)) {
135        sprintf(command, "md5sum \"%s\"", filename);
136        fin = popen(command, "r");
137        if (fin) {
138            (void) fgets(output, MAX_STR_LEN, fin);
139            p = strchr(output, ' ');
140            paranoid_pclose(fin);
141        }
142    } else {
143        sprintf(tmp, "File '%s' not found; cannot calc checksum",
144                filename);
145        log_it(tmp);
146    }
147    if (p) {
148        *p = '\0';
149    }
150    return (output);
151}
152
153
154/**
155 * Get a not-quite-unique representation of some of the file's @c stat properties.
156 * The returned string has the form <tt>size-mtime-ctime</tt>.
157 * @param curr_fname The file to generate the "checksum" for.
158 * @return The "checksum".
159 * @note The returned string points to static storage that will be overwritten with each call.
160 */
161char *calc_file_ugly_minichecksum(char *curr_fname)
162{
163
164    /*@ buffers ***************************************************** */
165    static char curr_cksum[1000];
166
167    /*@ pointers **************************************************** */
168
169    /*@ structures ************************************************** */
170    struct stat buf;
171
172    /*@ initialize data *************************************************** */
173    curr_cksum[0] = '\0';
174
175    /*@************************************************************** */
176
177    assert_string_is_neither_NULL_nor_zerolength(curr_fname);
178    if (lstat(curr_fname, &buf)) {
179        return (curr_cksum);    // empty
180    }
181
182    sprintf(curr_cksum, "%ld-%ld-%ld", (long) (buf.st_size),
183            (long) (buf.st_mtime), (long) (buf.st_ctime));
184    return (curr_cksum);
185}
186
187
188
189/**
190 * Get the number of lines in @p filename.
191 * @param filename The file to count lines in.
192 * @return The number of lines in @p filename.
193 * @bug This function uses the shell and "wc -l"; it should probably be rewritten in C.
194 */
195long count_lines_in_file(char *filename)
196{
197
198    /*@ buffers ***************************************************** */
199    char command[MAX_STR_LEN * 2];
200    char incoming[MAX_STR_LEN];
201    char tmp[MAX_STR_LEN];
202
203    /*@ long ******************************************************** */
204    long noof_lines = -1L;
205
206    /*@ pointers **************************************************** */
207    FILE *fin;
208
209    /*@ initialize [0] to null ******************************************** */
210    incoming[0] = '\0';
211
212    assert_string_is_neither_NULL_nor_zerolength(filename);
213    if (!does_file_exist(filename)) {
214        sprintf(tmp,
215                "%s does not exist, so I cannot found the number of lines in it",
216                filename);
217        log_it(tmp);
218        return (0);
219    }
220    sprintf(command, "cat %s | wc -l", filename);
221    if (!does_file_exist(filename)) {
222        return (-1);
223    }
224    fin = popen(command, "r");
225    if (fin) {
226        if (feof(fin)) {
227            noof_lines = 0;
228        } else {
229            (void) fgets(incoming, MAX_STR_LEN - 1, fin);
230            while (strlen(incoming) > 0
231                   && incoming[strlen(incoming) - 1] < 32) {
232                incoming[strlen(incoming) - 1] = '\0';
233            }
234            noof_lines = atol(incoming);
235        }
236        paranoid_pclose(fin);
237    }
238    return (noof_lines);
239}
240
241
242/**
243 * Check for existence of given @p filename.
244 * @param filename The file to check for.
245 * @return TRUE if it exists, FALSE otherwise.
246 */
247bool does_file_exist(char *filename)
248{
249
250    /*@ structures ************************************************** */
251    struct stat buf;
252
253    /*@************************************************************** */
254
255    assert(filename != NULL);
256    //  assert_string_is_neither_NULL_nor_zerolength(filename);
257    if (lstat(filename, &buf)) {
258        log_msg(20, "%s does not exist", filename);
259        return (FALSE);
260    } else {
261        log_msg(20, "%s exists", filename);
262        return (TRUE);
263    }
264}
265
266
267
268
269
270
271/**
272 * Modify @p inout (a file containing a list of files) to only contain files
273 * that exist.
274 * @param inout The filelist to operate on.
275 * @note The original file is renamed beforehand, so it will not be accessible
276 * while the modification is in progress.
277 */
278void exclude_nonexistent_files(char *inout)
279{
280    char infname[MAX_STR_LEN];
281    char outfname[MAX_STR_LEN];
282    char tmp[MAX_STR_LEN];
283    char incoming[MAX_STR_LEN];
284
285    /*@ int ********************************************************* */
286    int i;
287
288    /*@ pointers **************************************************** */
289    FILE *fin, *fout;
290
291
292    /*@ end vars *********************************************************** */
293
294    assert_string_is_neither_NULL_nor_zerolength(inout);
295    sprintf(infname, "%s.in", inout);
296    sprintf(outfname, "%s", inout);
297    sprintf(tmp, "cp -f %s %s", inout, infname);
298    run_program_and_log_output(tmp, FALSE);
299    if (!(fin = fopen(infname, "r"))) {
300        log_OS_error("Unable to openin infname");
301        return;
302    }
303    if (!(fout = fopen(outfname, "w"))) {
304        log_OS_error("Unable to openout outfname");
305        return;
306    }
307    for (fgets(incoming, MAX_STR_LEN, fin); !feof(fin);
308         fgets(incoming, MAX_STR_LEN, fin)) {
309        i = strlen(incoming) - 1;
310        if (i >= 0 && incoming[i] < 32) {
311            incoming[i] = '\0';
312        }
313        if (does_file_exist(incoming)) {
314            fprintf(fout, "%s\n", incoming);
315        } else {
316            sprintf(tmp, "Excluding '%s'-nonexistent\n", incoming);
317            log_it(tmp);
318        }
319    }
320    paranoid_fclose(fout);
321    paranoid_fclose(fin);
322    unlink(infname);
323}
324
325
326
327
328
329
330
331
332
333/**
334 * Attempt to find the user's kernel by calling Mindi.
335 * If Mindi can't find the kernel, ask user. If @p kernel is not empty,
336 * don't do anything.
337 * @param kernel Where to put the found kernel.
338 * @return 0 for success, 1 for failure.
339 */
340int figure_out_kernel_path_interactively_if_necessary(char *kernel)
341{
342    char tmp[MAX_STR_LEN];
343
344    if (!kernel[0]) {
345        strcpy(kernel,
346               call_program_and_get_last_line_of_output
347               ("mindi --findkernel 2> /dev/null"));
348    }
349    log_it("Calling Mindi with kernel path of '%s'", kernel);
350    while (!kernel[0]) {
351        if (!ask_me_yes_or_no
352            ("Kernel not found or invalid. Choose another?")) {
353            return (1);
354        }
355        if (!popup_and_get_string
356            ("Kernel path",
357             "What is the full path and filename of your kernel, please?",
358             kernel, MAX_STR_LEN / 4)) {
359            fatal_error
360                ("Kernel not found. Please specify with the '-k' flag.");
361        }
362        sprintf(tmp, "User says kernel is at %s", kernel);
363        log_it(tmp);
364    }
365    return (0);
366}
367
368
369
370
371
372
373/**
374 * Find location of specified executable in user's PATH.
375 * @param fname The basename of the executable to search for (e.g. @c afio).
376 * @return The full path to the executable, or "" if it does not exist, or NULL if @c file could not be found.
377 * @note The returned string points to static storage that will be overwritten with each call.
378 * @bug The checks with @c file and @c dirname seem pointless. If @c incoming is "", then you're calling
379 * <tt>dirname 2\>/dev/null</tt> or <tt>file 2\>/dev/null | cut -d':' -f1 2\>/dev/null</tt>, which basically amounts
380 * to nothing.
381 */
382char *find_home_of_exe(char *fname)
383{
384    /*@ buffers ********************* */
385    static char output[MAX_STR_LEN];
386    char *incoming;
387    char *command;
388
389    malloc_string(incoming);
390    malloc_string(command);
391    incoming[0] = '\0';
392    /*@******************************* */
393
394    assert_string_is_neither_NULL_nor_zerolength(fname);
395    sprintf(command, "which %s 2> /dev/null", fname);
396    strcpy(incoming, call_program_and_get_last_line_of_output(command));
397    if (incoming[0] == '\0') {
398        if (system("which file > /dev/null 2> /dev/null")) {
399            paranoid_free(incoming);
400            paranoid_free(command);
401            output[0] = '\0';
402            return (NULL);      // forget it :)
403        }
404        sprintf(command,
405                "file %s 2> /dev/null | cut -d':' -f1 2> /dev/null",
406                incoming);
407        strcpy(incoming,
408               call_program_and_get_last_line_of_output(command));
409    }
410    if (incoming[0] == '\0')    // yes, it is == '\0' twice, not once :)
411    {
412        sprintf(command, "dirname %s 2> /dev/null", incoming);
413        strcpy(incoming,
414               call_program_and_get_last_line_of_output(command));
415    }
416    strcpy(output, incoming);
417    if (output[0] != '\0' && does_file_exist(output)) {
418        log_msg(4, "find_home_of_exe () --- Found %s at %s", fname,
419                incoming);
420    } else {
421        output[0] = '\0';
422        log_msg(4, "find_home_of_exe() --- Could not find %s", fname);
423    }
424    paranoid_free(incoming);
425    paranoid_free(command);
426    if (!output[0]) {
427        return (NULL);
428    } else {
429        return (output);
430    }
431}
432
433
434
435
436
437
438
439
440/**
441 * Get the last sequence of digits surrounded by non-digits in the first 32k of
442 * a file.
443 * @param logfile The file to look in.
444 * @return The number found, or 0 if none.
445 */
446int get_trackno_from_logfile(char *logfile)
447{
448
449    /*@ pointers ********************************************************* */
450    FILE *fin;
451
452    /*@ int ************************************************************** */
453    int trackno = 0;
454    size_t len = 0;
455
456    /*@ buffer ************************************************************ */
457    char datablock[32701];
458
459    assert_string_is_neither_NULL_nor_zerolength(logfile);
460    if (!(fin = fopen(logfile, "r"))) {
461        log_OS_error("Unable to open logfile");
462        fatal_error("Unable to open logfile to read trackno");
463    }
464    len = fread(datablock, 1, 32700, fin);
465    paranoid_fclose(fin);
466    if (len <= 0) {
467        return (0);
468    }
469    for (; len > 0 && !isdigit(datablock[len - 1]); len--);
470    datablock[len--] = '\0';
471    for (; len > 0 && isdigit(datablock[len - 1]); len--);
472    trackno = atoi(datablock + len);
473    /*
474       sprintf(tmp,"datablock=%s; trackno=%d",datablock+len, trackno);
475       log_it(tmp);
476     */
477    return (trackno);
478}
479
480
481
482
483
484
485
486/**
487 * Get a percentage from the last line of @p filename. We look for the string
488 * "% done" on the last line and, if we find it, grab the number before the last % sign.
489 * @param filename The file to get the percentage from.
490 * @return The percentage found, or 0 for error.
491 */
492int grab_percentage_from_last_line_of_file(char *filename)
493{
494
495    /*@ buffers ***************************************************** */
496    char tmp[MAX_STR_LEN];
497    char lastline[MAX_STR_LEN];
498    char command[MAX_STR_LEN];
499    /*@ pointers **************************************************** */
500    char *p;
501
502    /*@ int's ******************************************************* */
503    int i;
504
505    for (i = NOOF_ERR_LINES - 1;
506         i >= 0 && !strstr(err_log_lines[i], "% Done")
507         && !strstr(err_log_lines[i], "% done"); i--);
508    if (i < 0) {
509        sprintf(command,
510                "tail -n3 %s | fgrep -i \"%c\" | tail -n1 | awk '{print $0;}'",
511                filename, '%');
512        strcpy(lastline,
513               call_program_and_get_last_line_of_output(command));
514        if (!lastline[0]) {
515            return (0);
516        }
517    } else {
518        strcpy(lastline, err_log_lines[i]);
519    }
520
521    p = strrchr(lastline, '%');
522    if (p) {
523        *p = '\0';
524    }
525//  log_msg(2, "lastline='%s', ", p, lastline);
526    if (!p) {
527        return (0);
528    }
529    *p = '\0';
530    for (p--; *p != ' ' && p != lastline; p--);
531    if (p != lastline) {
532        p++;
533    }
534    i = atoi(p);
535
536    sprintf(tmp, "'%s' --> %d", p, i);
537//     log_to_screen(tmp);
538
539    return (i);
540}
541
542
543
544
545
546/**
547 * Return the last line of @p filename.
548 * @param filename The file to get the last line of.
549 * @return The last line of the file.
550 * @note The returned string points to static storage that will be overwritten with each call.
551 */
552char *last_line_of_file(char *filename)
553{
554    /*@ buffers ***************************************************** */
555    static char output[MAX_STR_LEN];
556    static char command[MAX_STR_LEN * 2];
557    static char tmp[MAX_STR_LEN];
558
559    /*@ pointers **************************************************** */
560    FILE *fin;
561
562    /*@ end vars **************************************************** */
563
564    if (!does_file_exist(filename)) {
565        sprintf(tmp, "Tring to get last line of nonexistent file (%s)",
566                filename);
567        log_it(tmp);
568        output[0] = '\0';
569        return (output);
570    }
571    sprintf(command, "cat %s | tail -n1", filename);
572    fin = popen(command, "r");
573    (void) fgets(output, MAX_STR_LEN, fin);
574    paranoid_pclose(fin);
575    while (strlen(output) > 0 && output[strlen(output) - 1] < 32) {
576        output[strlen(output) - 1] = '\0';
577    }
578    return (output);
579}
580
581/**
582 * Get the length of @p filename in bytes.
583 * @param filename The file to get the length of.
584 * @return The length of the file, or -1 for error.
585 */
586long long length_of_file(char *filename)
587{
588    /*@ pointers *************************************************** */
589    FILE *fin;
590
591    /*@ long long ************************************************* */
592    long long length;
593
594    fin = fopen(filename, "r");
595    if (!fin) {
596        log_it("filename=%s", filename);
597        log_OS_error("Unable to openin filename");
598        return (-1);
599    }
600    fseek(fin, 0, SEEK_END);
601    length = ftell(fin);
602    paranoid_fclose(fin);
603    return (length);
604}
605
606
607
608/**
609 * ?????
610 * @bug I don't know what this function does. However, it seems orphaned, so it should probably be removed.
611 */
612int
613make_checksum_list_file(char *filelist, char *cksumlist, char *comppath)
614{
615    /*@ pointers **************************************************** */
616    FILE *fin;
617    FILE *fout;
618
619    /*@ int   ******************************************************* */
620    int percentage;
621    int i;
622    int counter = 0;
623
624    /*@ buffer ****************************************************** */
625    char stub_fname[1000];
626    char curr_fname[1000];
627    char curr_cksum[1000];
628    char tmp[1000];
629
630    /*@ long [long] ************************************************* */
631    long long filelist_length;
632    long curr_pos;
633    long start_time;
634    long current_time;
635    long time_taken;
636    long time_remaining;
637
638    /*@ end vars *************************************************** */
639
640    start_time = get_time();
641    filelist_length = length_of_file(filelist);
642    sprintf(tmp, "filelist = %s; cksumlist = %s", filelist, cksumlist);
643    log_it(tmp);
644    fin = fopen(filelist, "r");
645    if (fin == NULL) {
646        log_OS_error("Unable to fopen-in filelist");
647        log_to_screen("Can't open filelist");
648        return (1);
649    }
650    fout = fopen(cksumlist, "w");
651    if (fout == NULL) {
652        log_OS_error("Unable to openout cksumlist");
653        paranoid_fclose(fin);
654        log_to_screen("Can't open checksum list");
655        return (1);
656    }
657    for (fgets(stub_fname, 999, fin); !feof(fin);
658         fgets(stub_fname, 999, fin)) {
659        if (stub_fname[(i = strlen(stub_fname) - 1)] < 32) {
660            stub_fname[i] = '\0';
661        }
662        sprintf(tmp, "%s%s", comppath, stub_fname);
663        strcpy(curr_fname, tmp + 1);
664        strcpy(curr_cksum, calc_file_ugly_minichecksum(curr_fname));
665        fprintf(fout, "%s\t%s\n", curr_fname, curr_cksum);
666        if (counter++ > 12) {
667            current_time = get_time();
668            counter = 0;
669            curr_fname[37] = '\0';
670            curr_pos = ftell(fin) / 1024;
671            percentage = (int) (curr_pos * 100 / filelist_length);
672            time_taken = current_time - start_time;
673            if (percentage == 0) {
674                /*              printf("%0d%% done      \r",percentage); */
675            } else {
676                time_remaining =
677                    time_taken * 100 / (long) (percentage) - time_taken;
678                sprintf(tmp,
679                        "%02d%% done   %02d:%02d taken   %02d:%02d remaining  %-37s\r",
680                        percentage, (int) (time_taken / 60),
681                        (int) (time_taken % 60),
682                        (int) (time_remaining / 60),
683                        (int) (time_remaining % 60), curr_fname);
684                log_to_screen(tmp);
685            }
686            sync();
687        }
688    }
689    paranoid_fclose(fout);
690    paranoid_fclose(fin);
691    log_it("Done.");
692    return (0);
693}
694
695
696/**
697 * Create the directory @p outdir_fname and all parent directories. Equivalent to <tt>mkdir -p</tt>.
698 * @param outdir_fname The directory to create.
699 * @return The return value of @c mkdir.
700 */
701/* BERLIOS: This function shouldn't call system at all */
702int make_hole_for_dir(char *outdir_fname)
703{
704    char tmp[MAX_STR_LEN * 2];
705    int res = 0;
706
707    assert_string_is_neither_NULL_nor_zerolength(outdir_fname);
708    sprintf(tmp, "mkdir -p %s", outdir_fname);
709    res = system(tmp);
710    return (res);
711}
712
713
714/**
715 * Create the parent directories of @p outfile_fname.
716 * @param outfile_fname The file to make a "hole" for.
717 * @return 0, always.
718 * @bug Return value unnecessary.
719 */
720/* BERLIOS: This function shouldn't call system at all */
721int make_hole_for_file(char *outfile_fname)
722{
723    /*@ buffer ****************************************************** */
724    char command[MAX_STR_LEN * 2];
725
726    /*@ int  ******************************************************** */
727    int res = 0;
728
729    /*@ end vars *************************************************** */
730
731    assert_string_is_neither_NULL_nor_zerolength(outfile_fname);
732    assert(!strstr(outfile_fname, MNT_CDROM));
733    assert(!strstr(outfile_fname, "/dev/cdrom"));
734    sprintf(command, "mkdir -p \"%s\" 2> /dev/null", outfile_fname);
735    res += system(command);
736    sprintf(command, "rmdir \"%s\" 2> /dev/null", outfile_fname);
737    res += system(command);
738    sprintf(command, "rm -f \"%s\" 2> /dev/null", outfile_fname);
739    res += system(command);
740    unlink(outfile_fname);
741    return (0);
742}
743
744
745
746
747/**
748 * Get the number of lines in @p filelist_fname that contain the string @p wildcard.
749 * @param filelist_fname The file to search through.
750 * @param wildcard The string to search for. This is @e not a shell glob or a regular expression.
751 * @return The number of lines matched.
752 */
753long noof_lines_that_match_wildcard(char *filelist_fname, char *wildcard)
754{
755    /*@ long ******************************************************* */
756    long matches = 0;
757
758    /*@ pointers *************************************************** */
759    FILE *fin;
760
761    /*@ buffers **************************************************** */
762    char incoming[MAX_STR_LEN];
763
764    /*@ end vars *************************************************** */
765
766
767    fin = fopen(filelist_fname, "r");
768
769    if (!fin) {
770        log_OS_error("Unable to openin filelist_fname");
771        return (0);
772    }
773    (void) fgets(incoming, MAX_STR_LEN - 1, fin);
774    while (!feof(fin)) {
775        if (strstr(incoming, wildcard)) {
776            matches++;
777        }
778        (void) fgets(incoming, MAX_STR_LEN - 1, fin);
779    }
780    paranoid_fclose(fin);
781    return (matches);
782}
783
784
785
786
787/**
788 * Register our PID in a file in /var/run.
789 * The PID will be put in /var/run/monitas-<tt>name_str</tt>.pid.
790 * @param pid 0 to remove file, anything else to create it.
791 * @param name_str The basename of the PID file (e.g. "mondo" or "server")
792 * @note This function does not provide support against multiple instances, unless you check for that yourself.
793 */
794void register_pid(pid_t pid, char *name_str)
795{
796    char tmp[MAX_STR_LEN + 1], lockfile_fname[MAX_STR_LEN + 1];
797    int res;
798    FILE *fin;
799
800    sprintf(lockfile_fname, "/var/run/monitas-%s.pid", name_str);
801    if (!pid) {
802        log_it("Unregistering PID");
803        if (unlink(lockfile_fname)) {
804            log_it("Error unregistering PID");
805        }
806        return;
807    }
808    if (does_file_exist(lockfile_fname)) {
809        tmp[0] = '\0';
810        if ((fin = fopen(lockfile_fname, "r"))) {
811            (void) fgets(tmp, MAX_STR_LEN, fin);
812            paranoid_fclose(fin);
813        } else {
814            log_OS_error("Unable to openin lockfile_fname");
815        }
816        pid = (pid_t) atol(tmp);
817        sprintf(tmp, "ps %ld > /dev/null 2> /dev/null", (long int) pid);
818        res = system(tmp);
819        if (!res) {
820            log_it
821                ("I believe the daemon is already running. If it isn't, please delete %s and try again.",
822                 lockfile_fname);
823        }
824    }
825    sprintf(tmp, "echo %ld > %s 2> /dev/null", (long int) getpid(),
826            lockfile_fname);
827    if (system(tmp)) {
828        fatal_error("Cannot register PID");
829    }
830}
831
832
833
834/**
835 * Determine the size (in KB) of @p dev in the mountlist in <tt>tmpdir</tt>/mountlist.txt.
836 * @param tmpdir The tempdir where the mountlist is stored.
837 * @param dev The device to search for.
838 * @return The size of the partition in KB.
839 */
840long size_of_partition_in_mountlist_K(char *tmpdir, char *dev)
841{
842    char command[MAX_STR_LEN];
843    char mountlist[MAX_STR_LEN];
844    char sz_res[MAX_STR_LEN];
845    long file_len_K;
846
847    sprintf(mountlist, "%s/mountlist.txt", tmpdir);
848    sprintf(command,
849            "cat %s/mountlist.txt | grep \"%s \" | head -n1 | awk '{print $4;}'",
850            tmpdir, dev);
851    log_it(command);
852    strcpy(sz_res, call_program_and_get_last_line_of_output(command));
853    file_len_K = atol(sz_res);
854    log_msg(4, "%s --> %s --> %ld", command, sz_res, file_len_K);
855    return (file_len_K);
856}
857
858/**
859 * Calculate the total size (in KB) of all the biggiefiles in this backup.
860 * @param bkpinfo The backup information structure. Only the @c bkpinfo->tmpdir field is used.
861 * @return The total size of all biggiefiles in KB.
862 */
863long size_of_all_biggiefiles_K(struct s_bkpinfo *bkpinfo)
864{
865    /*@ buffers ***************************************************** */
866    char *fname;
867    char *biggielist;
868    char *comment;
869
870    /*@ long ******************************************************** */
871    long scratchL = 0;
872    long file_len_K;
873
874    /*@ pointers *************************************************** */
875    FILE *fin = NULL;
876
877    /*@ end vars *************************************************** */
878
879    malloc_string(fname);
880    malloc_string(biggielist);
881    malloc_string(comment);
882    log_it("Calculating size of all biggiefiles (in total)");
883    sprintf(biggielist, "%s/biggielist.txt", bkpinfo->tmpdir);
884    log_it("biggielist = %s", biggielist);
885    if (!(fin = fopen(biggielist, "r"))) {
886        log_OS_error
887            ("Cannot open biggielist. OK, so estimate is based on filesets only.");
888    } else {
889        log_msg(4, "Reading it...");
890        for (fgets(fname, MAX_STR_LEN, fin); !feof(fin);
891             fgets(fname, MAX_STR_LEN, fin)) {
892            if (fname[strlen(fname) - 1] <= 32) {
893                fname[strlen(fname) - 1] = '\0';
894            }
895            if (0 == strncmp(fname, "/dev/", 5)) {
896                file_len_K = get_phys_size_of_drive(fname) * 1024L;
897            } else {
898                file_len_K = (long) (length_of_file(fname) / 1024);
899            }
900            if (file_len_K > 0) {
901                scratchL += file_len_K;
902                log_msg(4, "%s --> %ld K", fname, file_len_K);
903            }
904            sprintf(comment,
905                    "After adding %s, scratchL+%ld now equals %ld", fname,
906                    file_len_K, scratchL);
907            log_msg(4, comment);
908            if (feof(fin)) {
909                break;
910            }
911        }
912    }
913    log_it("Closing...");
914    paranoid_fclose(fin);
915    log_it("Finished calculating total size of all biggiefiles");
916    paranoid_free(fname);
917    paranoid_free(biggielist);
918    paranoid_free(comment);
919    return (scratchL);
920}
921
922/**
923 * Determine the amount of space (in KB) occupied by a mounted CD.
924 * This can also be used to find the space used for other directories.
925 * @param mountpt The mountpoint/directory to check.
926 * @return The amount of space occupied in KB.
927 */
928long long space_occupied_by_cd(char *mountpt)
929{
930    /*@ buffer ****************************************************** */
931    char tmp[MAX_STR_LEN];
932    char command[MAX_STR_LEN * 2];
933    long long llres;
934    /*@ pointers **************************************************** */
935    char *p;
936    FILE *fin;
937
938    /*@ end vars *************************************************** */
939
940    sprintf(command, "du -sk %s", mountpt);
941    fin = popen(command, "r");
942    (void) fgets(tmp, MAX_STR_LEN, fin);
943    paranoid_pclose(fin);
944    p = strchr(tmp, '\t');
945    if (p) {
946        *p = '\0';
947    }
948    for (p = tmp, llres = 0; *p != '\0'; p++) {
949        llres *= 10;
950        llres += (int) (*p - '0');
951    }
952    return (llres);
953}
954
955
956/**
957 * Update a CRC checksum to include another character.
958 * @param crc The original CRC checksum.
959 * @param c The character to add.
960 * @return The new CRC checksum.
961 * @ingroup utilityGroup
962 */
963unsigned int updcrc(unsigned int crc, unsigned int c)
964{
965    unsigned int tmp;
966    tmp = (crc >> 8) ^ c;
967    crc = (crc << 8) ^ crctttab[tmp & 255];
968    return crc;
969}
970
971/**
972 * Update a reverse CRC checksum to include another character.
973 * @param crc The original CRC checksum.
974 * @param c The character to add.
975 * @return The new CRC checksum.
976 * @ingroup utilityGroup
977 */
978unsigned int updcrcr(unsigned int crc, unsigned int c)
979{
980    unsigned int tmp;
981    tmp = crc ^ c;
982    crc = (crc >> 8) ^ crc16tab[tmp & 0xff];
983    return crc;
984}
985
986
987
988
989/**
990 * Check for an executable on the user's system; write a message to the
991 * screen and the log if we can't find it.
992 * @param fname The executable basename to look for.
993 * @return 0 if it's found, nonzero if not.
994 */
995int whine_if_not_found(char *fname)
996{
997    /*@ buffers *** */
998    char command[MAX_STR_LEN * 2];
999    char errorstr[MAX_STR_LEN];
1000
1001
1002    sprintf(command, "which %s > /dev/null 2> /dev/null", fname);
1003    sprintf(errorstr,
1004            "Please install '%s'. I cannot find it on your system.",
1005            fname);
1006    if (system(command)) {
1007        log_to_screen(errorstr);
1008        log_to_screen
1009            ("There may be hyperlink at http://www.mondorescue.com which");
1010        log_to_screen("will take you to the relevant (missing) package.");
1011        return (1);
1012    } else {
1013        return (0);
1014    }
1015}
1016
1017
1018
1019
1020
1021
1022/**
1023 * Create a data file at @p fname containing @p contents.
1024 * The data actually can be multiple lines, despite the name.
1025 * @param fname The file to create.
1026 * @param contents The data to put in it.
1027 * @return 0 for success, 1 for failure.
1028 */
1029int write_one_liner_data_file(char *fname, char *contents)
1030{
1031    /*@ pointers *************************************************** */
1032    FILE *fout;
1033    int res = 0;
1034
1035    /*@ end vars *************************************************** */
1036
1037    assert_string_is_neither_NULL_nor_zerolength(fname);
1038    if (!contents) {
1039        log_it("%d: Warning - writing NULL to %s", __LINE__, fname);
1040    }
1041    if (!(fout = fopen(fname, "w"))) {
1042        log_it("fname=%s");
1043        log_OS_error("Unable to openout fname");
1044        return (1);
1045    }
1046    fprintf(fout, "%s\n", contents);
1047    paranoid_fclose(fout);
1048    return (res);
1049}
1050
1051
1052
1053/**
1054 * Read @p fname into @p contents.
1055 * @param fname The file to read.
1056 * @param contents Where to put its contents.
1057 * @return 0 for success, nonzero for failure.
1058 */
1059int read_one_liner_data_file(char *fname, char *contents)
1060{
1061    /*@ pointers *************************************************** */
1062    FILE *fin;
1063    int res = 0;
1064    int i;
1065
1066    /*@ end vars *************************************************** */
1067
1068    assert_string_is_neither_NULL_nor_zerolength(fname);
1069    if (!contents) {
1070        log_it("%d: Warning - reading NULL from %s", __LINE__, fname);
1071    }
1072    if (!(fin = fopen(fname, "r"))) {
1073        log_it("fname=%s", fname);
1074        log_OS_error("Unable to openin fname");
1075        return (1);
1076    }
1077    fscanf(fin, "%s\n", contents);
1078    i = strlen(contents);
1079    if (i > 0 && contents[i - 1] < 32) {
1080        contents[i - 1] = '\0';
1081    }
1082    paranoid_fclose(fin);
1083    return (res);
1084}
1085
1086
1087
1088
1089
1090
1091
1092
1093
1094/**
1095 * Copy the files that Mondo/Mindi need to run to the scratchdir or tempdir.
1096 * Currently this includes: copy Mondo's home directory to scratchdir, untar "mondo_home/payload.tgz"
1097 * if it exists, copy LAST-FILELIST-NUMBER to scratchdir, copy mondorestore
1098 * and post-nuke.tgz (if it exists) to tmpdir, and run "hostname > scratchdir/HOSTNAME".
1099 * @param bkpinfo The backup information structure. Fields used:
1100 * - @c bkpinfo->postnuke_tarball
1101 * - @c bkpinfo->scratchdir
1102 * - @c bkpinfo->tmpdir
1103 */
1104void copy_mondo_and_mindi_stuff_to_scratchdir(struct s_bkpinfo *bkpinfo)
1105{
1106    /*@ Char buffers ** */
1107    char command[MAX_STR_LEN * 2];
1108    char tmp[MAX_STR_LEN];
1109    char old_pwd[MAX_STR_LEN];
1110
1111    mvaddstr_and_log_it(g_currentY, 0,
1112                        "Copying Mondo's core files to the scratch directory");
1113
1114    /* BERLIOS: Why do we need to do it here as well ? */
1115    log_msg(4, "g_mondo_home='%s'", g_mondo_home);
1116    if ((g_mondo_home == NULL) || strlen(g_mondo_home) < 2) {
1117        paranoid_free(g_mondo_home);
1118        g_mondo_home = find_and_store_mondoarchives_home();
1119    }
1120    sprintf(command, CP_BIN " --parents -pRdf %s %s", g_mondo_home,
1121            bkpinfo->scratchdir);
1122
1123    log_msg(4, "command = %s", command);
1124    if (run_program_and_log_output(command, 1)) {
1125        fatal_error("Failed to copy Mondo's stuff to scratchdir");
1126    }
1127
1128    sprintf(tmp, "%s/payload.tgz", g_mondo_home);
1129    if (does_file_exist(tmp)) {
1130        log_it("Untarring payload %s to scratchdir %s", tmp,
1131               bkpinfo->scratchdir);
1132        (void) getcwd(old_pwd, MAX_STR_LEN - 1);
1133        chdir(bkpinfo->scratchdir);
1134        sprintf(command, "tar -zxvf %s", tmp);
1135        if (run_program_and_log_output(command, FALSE)) {
1136            fatal_error("Failed to untar payload");
1137        }
1138        chdir(old_pwd);
1139    }
1140
1141    sprintf(command, "cp -f %s/LAST-FILELIST-NUMBER %s", bkpinfo->tmpdir,
1142            bkpinfo->scratchdir);
1143
1144    if (run_program_and_log_output(command, FALSE)) {
1145        fatal_error("Failed to copy LAST-FILELIST-NUMBER to scratchdir");
1146    }
1147
1148    strcpy(tmp,
1149           call_program_and_get_last_line_of_output("which mondorestore"));
1150    if (!tmp[0]) {
1151        fatal_error
1152            ("'which mondorestore' returned null. Where's your mondorestore? `which` can't find it. That's odd. Did you install mondorestore?");
1153    }
1154    sprintf(command, "cp -f %s %s", tmp, bkpinfo->tmpdir);
1155    if (run_program_and_log_output(command, FALSE)) {
1156        fatal_error("Failed to copy mondorestore to tmpdir");
1157    }
1158
1159    sprintf(command, "hostname > %s/HOSTNAME", bkpinfo->scratchdir);
1160    paranoid_system(command);
1161
1162    if (bkpinfo->postnuke_tarball[0]) {
1163        sprintf(command, "cp -f %s %s/post-nuke.tgz",
1164                bkpinfo->postnuke_tarball, bkpinfo->tmpdir);
1165        if (run_program_and_log_output(command, FALSE)) {
1166            fatal_error("Unable to copy post-nuke tarball to tmpdir");
1167        }
1168    }
1169
1170
1171    mvaddstr_and_log_it(g_currentY++, 74, "Done.");
1172}
1173
1174
1175
1176
1177
1178/**
1179 * Store the client's NFS configuration in files to be restored at restore-time.
1180 * Assumes that @c bkpinfo->media_type = nfs, but does not check for this.
1181 * @param bkpinfo The backup information structure. Fields used:
1182 * - @c nfs_mount
1183 * - @c nfs_remote_dir
1184 * - @c tmpdir
1185 */
1186void store_nfs_config(struct s_bkpinfo *bkpinfo)
1187{
1188
1189    /*@ buffers ******** */
1190    char outfile[MAX_STR_LEN];
1191    char nfs_dev[MAX_STR_LEN];
1192    char nfs_mount[MAX_STR_LEN];
1193    char nfs_client_ipaddr[MAX_STR_LEN];
1194    char nfs_server_ipaddr[MAX_STR_LEN];
1195    char tmp[MAX_STR_LEN];
1196    char command[MAX_STR_LEN * 2];
1197
1198    /*@ pointers ***** */
1199    char *p;
1200    FILE *fout;
1201
1202
1203
1204    log_it("Storing NFS configuration");
1205    strcpy(tmp, bkpinfo->nfs_mount);
1206    p = strchr(tmp, ':');
1207    if (!p) {
1208        fatal_error
1209            ("NFS mount doesn't have a colon in it, e.g. 192.168.1.4:/home/nfs");
1210    }
1211    *(p++) = '\0';
1212    strcpy(nfs_server_ipaddr, tmp);
1213    strcpy(nfs_mount, p);
1214    sprintf(command,
1215            "ifconfig | tr '\n' '#' | sed s/##// | tr '#' ' ' | tr '' '\n' | head -n1 | cut -d' ' -f1");
1216    strcpy(nfs_dev, call_program_and_get_last_line_of_output(command));
1217    sprintf(command,
1218            "ifconfig | tr '\n' '#' | sed s/##// | tr '#' ' ' | tr '' '\\n' | head -n1 | tr -s '\t' ' ' | cut -d' ' -f7 | cut -d':' -f2");
1219    strcpy(nfs_client_ipaddr,
1220           call_program_and_get_last_line_of_output(command));
1221    sprintf(tmp,
1222            "nfs_client_ipaddr=%s; nfs_server_ipaddr=%s; nfs_mount=%s",
1223            nfs_client_ipaddr, nfs_server_ipaddr, nfs_mount);
1224    if (strlen(nfs_dev) < 2) {
1225        fatal_error
1226            ("Unable to find ethN (eth0, eth1, ...) adapter via NFS mount you specified.");
1227    }
1228    sprintf(outfile, "%s/start-nfs", bkpinfo->tmpdir);
1229    sprintf(tmp, "outfile = %s", outfile);
1230    log_it(tmp);
1231    if (!(fout = fopen(outfile, "w"))) {
1232        fatal_error("Cannot store NFS config");
1233    }
1234    fprintf(fout, "ifconfig lo 127.0.0.1  # config loopback\n");
1235    fprintf(fout, "ifconfig %s %s; # config client\n", nfs_dev,
1236            nfs_client_ipaddr);
1237    fprintf(fout, "ping -c 1 %s; # ping server\n", nfs_server_ipaddr);
1238    fprintf(fout, "mount -t nfs -o nolock %s /tmp/isodir\n",
1239            bkpinfo->nfs_mount);
1240    fprintf(fout, "exit 0\n");
1241    paranoid_fclose(fout);
1242    chmod(outfile, 0777);
1243    make_hole_for_dir("/var/cache/mondo-archive");
1244
1245//  paranoid_system ("mkdir -p /var/cache/mondo-archive 2> /dev/null");
1246
1247    sprintf(tmp, "cp -f %s /var/cache/mondo-archive", outfile);
1248    run_program_and_log_output(tmp, FALSE);
1249
1250    sprintf(tmp, "%s/NFS-DEV", bkpinfo->tmpdir);
1251    write_one_liner_data_file(tmp, nfs_dev);
1252
1253    sprintf(tmp, "%s/NFS-CLIENT-IPADDR", bkpinfo->tmpdir);
1254    write_one_liner_data_file(tmp, nfs_client_ipaddr);
1255    sprintf(tmp, "%s/NFS-SERVER-IPADDR", bkpinfo->tmpdir);
1256    write_one_liner_data_file(tmp, nfs_server_ipaddr);
1257    sprintf(tmp, "%s/NFS-SERVER-MOUNT", bkpinfo->tmpdir);
1258    write_one_liner_data_file(tmp, bkpinfo->nfs_mount);
1259    sprintf(tmp, "%s/NFS-SERVER-PATH", bkpinfo->tmpdir);
1260    write_one_liner_data_file(tmp, bkpinfo->nfs_remote_dir);
1261    log_it("Finished storing NFS configuration");
1262}
1263
1264
1265
1266
1267
1268
1269/**
1270 * Determine the approximate number of media that the backup will take up,
1271 * and tell the user. The uncompressed size is estimated as size_of_all_biggiefiles_K()
1272 * plus (noof_sets x bkpinfo->optimal_set_size). The compression factor is estimated as
1273 * 2/3 for LZO and 1/2 for bzip2. The data is not saved anywhere. If there are any
1274 * "imagedevs", the estimate is not shown as it will be wildly inaccurate.
1275 * If there are more than 50 media estimated, the estimate will not be shown.
1276 * @param bkpinfo The backup information structure. Fields used:
1277 * - @c bkpinfo->backup_media_type
1278 * - @c bkpinfo->image_devs
1279 * - @c bkpinfo->media_size
1280 * - @c bkpinfo->optimal_set_size
1281 * - @c bkpinfo->use_lzo
1282 * @param noof_sets The number of filesets created.
1283 * @ingroup archiveGroup
1284 */
1285void
1286estimate_noof_media_required(struct s_bkpinfo *bkpinfo, long noof_sets)
1287{
1288    /*@ buffers *************** */
1289    char tmp[MAX_STR_LEN];
1290
1291    /*@ long long ************* */
1292    long long scratchLL;
1293
1294    if (bkpinfo->media_size[1] <= 0 || bkpinfo->backup_media_type == nfs) {
1295        log_to_screen("Number of media required: UNKNOWN");
1296        return;
1297    }
1298
1299    log_it("Estimating number of media required...");
1300    scratchLL =
1301        (long long) (noof_sets) * (long long) (bkpinfo->optimal_set_size)
1302        + (long long) (size_of_all_biggiefiles_K(bkpinfo));
1303    scratchLL = (scratchLL / 1024) / bkpinfo->media_size[1];
1304    scratchLL++;
1305    if (bkpinfo->use_lzo) {
1306        scratchLL = (scratchLL * 2) / 3;
1307    } else {
1308        scratchLL = scratchLL / 2;
1309    }
1310    if (!scratchLL) {
1311        scratchLL++;
1312    }
1313    if (scratchLL <= 1) {
1314        sprintf(tmp,
1315                "Your backup will probably occupy a single CD/tape/ISO. Maybe two.");
1316    } else {
1317        sprintf(tmp, "Your backup will occupy approximately %s media.",
1318                number_to_text((int) (scratchLL + 1)));
1319    }
1320    if (!bkpinfo->image_devs[0] && (scratchLL < 50)) {
1321        log_to_screen(tmp);
1322    }
1323}
1324
1325
1326/**
1327 * Get the last suffix of @p instr.
1328 * If @p instr was "httpd.log.gz", we would return "gz".
1329 * @param instr The filename to get the suffix of.
1330 * @return The suffix (without a dot), or "" if none.
1331 * @note The returned string points to static storage that will be overwritten with each call.
1332 */
1333char *sz_last_suffix(char *instr)
1334{
1335    static char outstr[MAX_STR_LEN];
1336    char *p;
1337
1338    p = strrchr(instr, '.');
1339    if (!p) {
1340        outstr[0] = '\0';
1341    } else {
1342        strcpy(outstr, p);
1343    }
1344    return (outstr);
1345}
1346
1347
1348/**
1349 * Determine whether a file is compressed. This is done
1350 * by reading through the "do-not-compress-these" file distributed with Mondo.
1351 * @param filename The file to check.
1352 * @return TRUE if it's compressed, FALSE if not.
1353 */
1354bool is_this_file_compressed(char *filename)
1355{
1356    char do_not_compress_these[MAX_STR_LEN];
1357    char tmp[MAX_STR_LEN];
1358    char *p;
1359
1360    sprintf(tmp, "%s/do-not-compress-these", g_mondo_home);
1361    if (!does_file_exist(tmp)) {
1362        return (FALSE);
1363    }
1364    strcpy(do_not_compress_these, last_line_of_file(tmp));
1365    for (p = do_not_compress_these; p != NULL; p++) {
1366        strcpy(tmp, p);
1367        if (strchr(tmp, ' ')) {
1368            *(strchr(tmp, ' ')) = '\0';
1369        }
1370        if (!strcmp(sz_last_suffix(filename), tmp)) {   /*printf("MATCH\n"); */
1371            return (TRUE);
1372        }
1373        if (!(p = strchr(p, ' '))) {
1374            break;
1375        }
1376    }
1377    return (FALSE);
1378}
1379
1380
1381
1382int mode_of_file(char *fname)
1383{
1384    struct stat buf;
1385
1386    if (lstat(fname, &buf)) {
1387        return (-1);
1388    }                           // error
1389    else {
1390        return (buf.st_mode);
1391    }
1392}
1393
1394
1395
1396
1397/**
1398 * Create a small script that mounts /boot, calls @c grub-install, and syncs the disks.
1399 * @param outfile Where to put the script.
1400 * @return 0 for success, 1 for failure.
1401 */
1402int make_grub_install_scriptlet(char *outfile)
1403{
1404    FILE *fout;
1405    char *tmp;
1406    int retval = 0;
1407
1408    malloc_string(tmp);
1409    if ((fout = fopen(outfile, "w"))) {
1410        fprintf(fout,
1411                "#!/bin/sh\n\nmount /boot > /dev/null 2> /dev/null\ngrub-install $@\nres=$?\nsync;sync;sync\nexit $res\n");
1412        paranoid_fclose(fout);
1413        log_msg(2, "Created %s", outfile);
1414        sprintf(tmp, "chmod +x %s", outfile);
1415        paranoid_system(tmp);
1416        retval = 0;
1417    } else {
1418        retval = 1;
1419    }
1420    paranoid_free(tmp);
1421    return (retval);
1422}
1423
1424/* @} - end fileGroup */
Note: See TracBrowser for help on using the repository browser.