source: trunk/mondo/src/common/libmondo-fork.c @ 1161

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

Update trunk with the latest feedback from stable (obtained with meld) - First pass

  • Property svn:keywords set to Id
File size: 18.2 KB
Line 
1/* libmondo-fork.c - subroutines for handling forking/pthreads/etc.
2 * $Id: libmondo-fork.c 1161 2007-02-14 12:13:49Z bruno $
3 */
4#include "my-stuff.h"
5#include "mr_mem.h"
6#include "mondostructures.h"
7#include "libmondo-fork.h"
8#include "libmondo-string-EXT.h"
9#include "newt-specific-EXT.h"
10#include "libmondo-files-EXT.h"
11#include "libmondo-tools-EXT.h"
12#include "mr_mem.h"
13
14extern bool g_text_mode;
15pid_t g_buffer_pid = 0;
16
17
18/**
19 * Call a program and retrieve its last line of output.
20 * @param call The program to run.
21 * @return The last line of its output.
22 * @note The returned value should be freed by caller
23 */
24char *call_program_and_get_last_line_of_output(char *call)
25{
26    char *result = NULL;
27    FILE *fin = NULL;
28    size_t n = 0;
29
30    assert_string_is_neither_NULL_nor_zerolength(call);
31
32    if ((fin = popen(call, "r"))) {
33        for (mr_getline(&result, &n, fin); !feof(fin); mr_getline(&result, &n, fin));
34        paranoid_pclose(fin);
35    } else {
36        log_OS_error("Unable to popen call");
37    }
38    return(result);
39}
40
41#define MONDO_POPMSG  _("Your PC will not retract the CD tray automatically. Please call mondoarchive with the -m (manual CD tray) flag.")
42
43/**
44 * Call mkisofs to create an ISO image.
45 * @param bkpinfo The backup information structure. Fields used:
46 * - @c bkpinfo->manual_cd_tray
47 * - @c bkpinfo->backup_media_type
48 * - @c bkpinfo->please_dont_eject_when_restoring
49 * @param basic_call The call to mkisofs. May contain tokens that will be resolved to actual data. The tokens are:
50 * - @c _ISO_ will become the ISO file (@p isofile)
51 * - @c _CD#_ becomes the CD number (@p cd_no)
52 * - @c _ERR_ becomes the logfile (@p g_logfile)
53 * @param isofile Replaces @c _ISO_ in @p basic_call. Should probably be the ISO image to create (-o parameter to mkisofs).
54 * @param cd_no Replaces @c _CD#_ in @p basic_call. Should probably be the CD number.
55 * @param logstub Unused.
56 * @param what_i_am_doing The action taking place (e.g. "Making ISO #1"). Used as the title of the progress dialog.
57 * @return Exit code of @c mkisofs (0 is success, anything else indicates failure).
58 * @bug @p logstub is unused.
59 */
60int
61eval_call_to_make_ISO(struct s_bkpinfo *bkpinfo,
62                      char *basic_call, char *isofile,
63                      int cd_no, char *logstub, char *what_i_am_doing)
64{
65
66    /*@ int's  *** */
67    int retval = 0;
68
69
70    /*@ buffers      *** */
71    char *midway_call, *ultimate_call, *tmp;
72    char *p = NULL; 
73    char *cd_number_str = NULL;
74    char *command = NULL;
75   
76
77/*@***********   End Variables ***************************************/
78
79    mr_msg(3, "Starting");
80    assert(bkpinfo != NULL);
81    assert_string_is_neither_NULL_nor_zerolength(basic_call);
82    assert_string_is_neither_NULL_nor_zerolength(isofile);
83    assert_string_is_neither_NULL_nor_zerolength(logstub);
84
85    midway_call = mr_malloc(1200);
86    ultimate_call = mr_malloc(1200);
87    tmp = mr_malloc(1200);
88
89    mr_asprintf(&cd_number_str, "%d", cd_no);
90    resolve_naff_tokens(midway_call, basic_call, isofile, "_ISO_");
91    resolve_naff_tokens(tmp, midway_call, cd_number_str, "_CD#_");
92    mr_free(cd_number_str);
93
94    resolve_naff_tokens(ultimate_call, tmp, MONDO_LOGFILE, "_ERR_");
95    mr_msg(4, "basic call = '%s'", basic_call);
96    mr_msg(4, "midway_call = '%s'", midway_call);
97    mr_msg(4, "tmp = '%s'", tmp);
98    mr_msg(4, "ultimate call = '%s'", ultimate_call);
99    mr_asprintf(&command, "%s >> %s", ultimate_call, MONDO_LOGFILE);
100
101    log_to_screen
102        (_("Please be patient. Do not be alarmed by on-screen inactivity."));
103    mr_msg(4, "Calling open_evalcall_form() with what_i_am_doing='%s'",
104            what_i_am_doing);
105    if (bkpinfo->manual_cd_tray) {
106        /* Find everything after a 2>> and remove it */
107        p = strstr(command, "2>>");
108        if (p) {
109            for (; *p != ' ' && *p != '\0'; p++) {
110                *p = ' ';
111            }
112        }
113#ifndef _XWIN
114        if (!g_text_mode) {
115            newtSuspend();
116        }
117#endif
118        mr_msg(1, "command = '%s'", command);
119        retval += system(command);
120        if (!g_text_mode) {
121            newtResume();
122        }
123        if (retval) {
124            mr_msg(2, "Basic call '%s' returned an error.", basic_call);
125            popup_and_OK(_("Press ENTER to continue."));
126            popup_and_OK
127                (_("mkisofs and/or cdrecord returned an error. CD was not created"));
128        }
129    }
130    /* if text mode then do the above & RETURN; if not text mode, do this... */
131    else {
132        mr_msg(3, "command = '%s'", command);
133//      yes_this_is_a_goto:
134        retval =
135            run_external_binary_with_percentage_indicator_NEW
136            (what_i_am_doing, command);
137    }
138    mr_free(command);
139
140    mr_free(midway_call);
141    mr_free(ultimate_call);
142    mr_free(tmp);
143    return (retval);
144}
145
146
147/**
148 * Run a program and log its output (stdout and stderr) to the logfile.
149 * @param program The program to run. Passed to the shell, so you can use pipes etc.
150 * @param debug_level If @p g_loglevel is higher than this, do not log the output.
151 * @return The exit code of @p program (depends on the command, but 0 almost always indicates success).
152 */
153int run_program_and_log_output(char *program, int debug_level)
154{
155    /*@ buffer ****************************************************** */
156    char *callstr = NULL;
157    char *incoming = NULL;
158
159    /*@ int ********************************************************* */
160    int res = 0;
161    size_t n = 0;
162    bool log_if_failure = FALSE;
163    bool log_if_success = FALSE;
164
165    /*@ pointers *************************************************** */
166    FILE *fin;
167    char *p;
168
169    /*@ end vars *************************************************** */
170
171    assert(program != NULL);
172    if (!program[0]) {
173        mr_msg(2, "Warning - asked to run zerolength program");
174        return (1);
175    }
176
177    if (debug_level <= g_loglevel) {
178        log_if_success = TRUE;
179        log_if_failure = TRUE;
180    }
181    mr_asprintf(&callstr,
182            "%s > /tmp/mondo-run-prog-thing.tmp 2> /tmp/mondo-run-prog-thing.err",
183            program);
184    while ((p = strchr(callstr, '\r'))) {
185        *p = ' ';
186    }
187    while ((p = strchr(callstr, '\n'))) {
188        *p = ' ';
189    }                           /* single '=' is intentional */
190
191
192    res = system(callstr);
193    if (((res == 0) && log_if_success) || ((res != 0) && log_if_failure)) {
194        mr_msg(0, "running: %s", callstr);
195        mr_msg(0,
196                "--------------------------------start of output-----------------------------");
197    }
198    mr_free(callstr);
199
200    if (log_if_failure
201        &&
202        system
203        ("cat /tmp/mondo-run-prog-thing.err >> /tmp/mondo-run-prog-thing.tmp 2> /dev/null"))
204    {
205        log_OS_error("Command failed");
206    }
207    unlink("/tmp/mondo-run-prog-thing.err");
208    fin = fopen("/tmp/mondo-run-prog-thing.tmp", "r");
209    if (fin) {
210        for (mr_getline(&incoming, &n, fin); !feof(fin);
211             mr_getline(&incoming, &n, fin)) {
212            /* patch by Heiko Schlittermann */
213            p = incoming;
214            while (p && *p) {
215                if ((p = strchr(p, '%'))) {
216                    memmove(p, p + 1, strlen(p) + 1);
217                    p += 2;
218                }
219            }
220            /* end of patch */
221            strip_spaces(incoming);
222            if ((res == 0 && log_if_success)
223                || (res != 0 && log_if_failure)) {
224                mr_msg(0, incoming);
225            }
226        }
227        mr_free(incoming);
228        paranoid_fclose(fin);
229    }
230    unlink("/tmp/mondo-run-prog-thing.tmp");
231    if ((res == 0 && log_if_success) || (res != 0 && log_if_failure)) {
232        mr_msg(0,
233                "--------------------------------end of output------------------------------");
234        if (res) {
235            mr_msg(0, "...ran with res=%d", res);
236        } else {
237            mr_msg(0, "...ran just fine. :-)");
238        }
239    }
240//  else
241//    { mr_msg (0, "-------------------------------ran w/ res=%d------------------------------", res); }
242    return (res);
243}
244
245
246/**
247 * Run a program and log its output to the screen.
248 * @param basic_call The program to run.
249 * @param what_i_am_doing The title of the evalcall form.
250 * @return The return value of the command (varies, but 0 almost always means success).
251 * @see run_program_and_log_output
252 * @see log_to_screen
253 */
254int run_program_and_log_to_screen(char *basic_call, char *what_i_am_doing)
255{
256    /*@ int ******************************************************** */
257    int retval = 0;
258    int res = 0;
259    int i;
260
261    /*@ pointers **************************************************** */
262    FILE *fin;
263
264    /*@ buffers **************************************************** */
265    char *tmp = NULL;
266    char *command = NULL;
267    char *lockfile = NULL;
268
269    /*@ end vars *************************************************** */
270
271    assert_string_is_neither_NULL_nor_zerolength(basic_call);
272
273    mr_asprintf(&lockfile, "/tmp/mojo-jojo.blah.XXXXXX");
274    mkstemp(lockfile);
275    mr_asprintf(&command,
276            "echo hi > %s ; %s >> %s 2>> %s; res=$?; sleep 1; rm -f %s; exit $res",
277            lockfile, basic_call, MONDO_LOGFILE, MONDO_LOGFILE, lockfile);
278    open_evalcall_form(what_i_am_doing);
279    mr_asprintf(&tmp, "Executing %s", basic_call);
280    mr_msg(2, tmp);
281    mr_free(tmp);
282
283    if (!(fin = popen(command, "r"))) {
284        log_OS_error("Unable to popen-in command");
285        mr_asprintf(&tmp, _("Failed utterly to call '%s'"), command);
286        log_to_screen(tmp);
287        mr_free(tmp);
288        mr_free(lockfile);
289        mr_free(command);
290        return (1);
291    }
292    mr_free(command);
293
294    if (!does_file_exist(lockfile)) {
295        log_to_screen(_("Waiting for external binary to start"));
296        for (i = 0; i < 60 && !does_file_exist(lockfile); sleep(1), i++) {
297            mr_msg(3, "Waiting for lockfile %s to exist", lockfile);
298        }
299    }
300    /* This works on Newt, and it gives quicker updates. */
301    for (; does_file_exist(lockfile); sleep(1)) {
302        log_file_end_to_screen(MONDO_LOGFILE, "");
303        update_evalcall_form(1);
304    }
305    /* Evaluate the status returned by pclose to get the exit code of the called program. */
306    errno = 0;
307    res = pclose(fin);
308    /* Log actual pclose errors. */
309    if (errno) {
310        mr_msg(5, "pclose err: %d", errno);
311    }
312    /* Check if we have a valid status. If we do, extract the called program's exit code. */
313    /* If we don't, highlight this fact by returning -1. */
314    if (WIFEXITED(res)) {
315        retval = WEXITSTATUS(res);
316    } else {
317        retval = -1;
318    }
319    close_evalcall_form();
320    unlink(lockfile);
321    mr_free(lockfile);
322
323    return (retval);
324}
325
326
327/**
328 * Apparently used. @bug This has a purpose, but what?
329 */
330#define PIMP_START_SZ "STARTSTARTSTART9ff3kff9a82gv34r7fghbkaBBC2T231hc81h42vws8"
331#define PIMP_END_SZ "ENDENDEND0xBBC10xBBC2T231hc81h42vws89ff3kff9a82gv34r7fghbka"
332
333
334
335
336int copy_from_src_to_dest(FILE * f_orig, FILE * f_archived, char direction)
337{
338// if dir=='w' then copy from orig to archived
339// if dir=='r' then copy from archived to orig
340    char *tmp = NULL;
341    char *buf = NULL;
342    long int bytes_to_be_read, bytes_read_in, bytes_written_out =
343        0, bufcap, subsliceno = 0;
344    int retval = 0;
345    FILE *fin = NULL;
346    FILE *fout = NULL;
347    FILE *ftmp = NULL;
348
349    mr_msg(5, "Opening.");
350    bufcap = 256L * 1024L;
351    buf = mr_malloc(bufcap);
352
353    if (direction == 'w') {
354        fin = f_orig;
355        fout = f_archived;
356        mr_asprintf(&tmp, "%-64s", PIMP_START_SZ);
357        if (fwrite(tmp, 1, 64, fout) != 64) {
358            fatal_error("Can't write the introductory block");
359        }
360        mr_free(tmp);
361
362        while (1) {
363            bytes_to_be_read = bytes_read_in = fread(buf, 1, bufcap, fin);
364            if (bytes_read_in == 0) {
365                break;
366            }
367            mr_asprintf(&tmp, "%-64ld", bytes_read_in);
368            if (fwrite(tmp, 1, 64, fout) != 64) {
369                fatal_error("Cannot write introductory block");
370            }
371            mr_free(tmp);
372
373            mr_msg(7,
374                    "subslice #%ld --- I have read %ld of %ld bytes in from f_orig",
375                    subsliceno, bytes_read_in, bytes_to_be_read);
376            bytes_written_out += fwrite(buf, 1, bytes_read_in, fout);
377            mr_asprintf(&tmp, "%-64ld", subsliceno);
378            if (fwrite(tmp, 1, 64, fout) != 64) {
379                fatal_error("Cannot write post-thingy block");
380            }
381            mr_free(tmp);
382
383            mr_msg(7, "Subslice #%d written OK", subsliceno);
384            subsliceno++;
385        }
386        mr_asprintf(&tmp, "%-64ld", 0L);
387        if (fwrite(tmp, 1, 64L, fout) != 64L) {
388            fatal_error("Cannot write final introductory block");
389        }
390    } else {
391        fin = f_archived;
392        fout = f_orig;
393        if (!(tmp = malloc(64L))) {
394            fatal_error("Failed to malloc() tmp");
395        }
396        if (fread(tmp, 1, 64L, fin) != 64L) {
397            fatal_error("Cannot read the introductory block");
398        }
399        mr_msg(5, "tmp is %s", tmp);
400        if (!strstr(tmp, PIMP_START_SZ)) {
401            fatal_error("Can't find intro blk");
402        }
403        if (fread(tmp, 1, 64L, fin) != 64L) {
404            fatal_error("Cannot read introductory blk");
405        }
406        bytes_to_be_read = atol(tmp);
407        while (bytes_to_be_read > 0) {
408            mr_msg(7, "subslice#%ld, bytes=%ld", subsliceno,
409                    bytes_to_be_read);
410            bytes_read_in = fread(buf, 1, bytes_to_be_read, fin);
411            if (bytes_read_in != bytes_to_be_read) {
412                fatal_error
413                    ("Danger, WIll Robinson. Failed to read whole subvol from archives.");
414            }
415            bytes_written_out += fwrite(buf, 1, bytes_read_in, fout);
416            if (fread(tmp, 1, 64L, fin) != 64L) {
417                fatal_error("Cannot read post-thingy block");
418            }
419            if (atol(tmp) != subsliceno) {
420                mr_msg(1, "Wanted subslice %ld but got %ld ('%s')",
421                        subsliceno, atol(tmp), tmp);
422            }
423            mr_msg(7, "Subslice #%ld read OK", subsliceno);
424            subsliceno++;
425            if (fread(tmp, 1, 64L, fin) != 64L) {
426                fatal_error("Cannot read introductory block");
427            }
428            bytes_to_be_read = atol(tmp);
429        }
430    }
431
432//  mr_msg(4, "Written %ld of %ld bytes", bytes_written_out, bytes_read_in);
433
434    if (direction == 'w') {
435        mr_free(tmp);
436        mr_asprintf(&tmp, "%-64s", PIMP_END_SZ);
437        if (fwrite(tmp, 1, 64L, fout) != 64L) {
438            fatal_error("Can't write the final block");
439        }
440        mr_free(tmp);
441    } else {
442        mr_msg(1, "tmpA is %s", tmp);
443        if (!strstr(tmp, PIMP_END_SZ)) {
444            if (fread(tmp, 1, 64L, fin) != 64L) {
445                fatal_error("Can't read the final block");
446            }
447            mr_msg(5, "tmpB is %s", tmp);
448            if (!strstr(tmp, PIMP_END_SZ)) {
449                ftmp = fopen("/tmp/out.leftover", "w");
450                bytes_read_in = fread(tmp, 1, 64L, fin);
451                mr_msg(1, "bytes_read_in = %ld", bytes_read_in);
452//      if (bytes_read_in!=128+64) { fatal_error("Can't read the terminating block"); }
453                fwrite(tmp, 1, bytes_read_in, ftmp);
454               
455                mr_free(tmp);
456                if (!(tmp = malloc(512))) {
457                    fatal_error("Failed to malloc() tmp");
458                }
459                /* BERLIOS : strange ???
460                s-printf(tmp, "I am here - %llu", ftello(fin));
461                mr_msg(0, tmp);
462                */
463                fread(tmp, 1, 512, fin);
464                mr_msg(0, "tmp = '%s'", tmp);
465                fwrite(tmp, 1, 512, ftmp);
466                fclose(ftmp);
467                fatal_error("Missing terminating block");
468            }
469        }
470        mr_free(tmp);
471    }
472
473    mr_free(buf);
474    mr_msg(3, "Successfully copied %ld bytes", bytes_written_out);
475    return (retval);
476}
477
478/**
479 * Feed @p input_device through ntfsclone to @p output_fname.
480 * @param input_device The device to image.
481 * @param output_fname The file to write.
482 * @return 0 for success, nonzero for failure.
483 */
484int feed_into_ntfsprog(char *input_device, char *output_fname)
485{
486// BACKUP
487    int res = -1;
488    char *command = NULL;
489    char *tmp = NULL;
490
491    if (!does_file_exist(input_device)) {
492        fatal_error("input device does not exist");
493    }
494    tmp = find_home_of_exe("ntfsclone");
495    if (!tmp) {
496        fatal_error("ntfsclone not found");
497    }
498    mr_free(tmp);
499
500    mr_asprintf(&command, "ntfsclone --force --save-image --overwrite %s %s", output_fname, input_device);
501    res = run_program_and_log_output(command, 5);
502    mr_free(command);
503
504    unlink(output_fname);
505    return (res);
506}
507
508
509int run_external_binary_with_percentage_indicator_OLD(char *tt, char *cmd)
510{
511
512    int res = 0;
513    int percentage = 0;
514    int maxpc = 0;
515    int pcno = 0;
516    int last_pcno = 0;
517
518    char *command = NULL;
519    char *tempfile = NULL;
520    FILE *pin = NULL;
521
522    assert_string_is_neither_NULL_nor_zerolength(cmd);
523
524    tempfile = call_program_and_get_last_line_of_output("mktemp -q /tmp/mondo.XXXXXXXX");
525    mr_asprintf(&command, "%s >> %s 2>> %s; rm -f %s", cmd, tempfile, tempfile,
526            tempfile);
527    mr_msg(3, command);
528    open_evalcall_form(tt);
529
530    if (!(pin = popen(command, "r"))) {
531        log_OS_error("fmt err");
532        mr_free(command);
533        mr_free(tempfile);
534        return (1);
535    }
536    mr_free(command);
537
538    maxpc = 100;
539// OLD OLD OLD OLD OLD OLD OLD OLD OLD OLD OLD OLD
540    for (sleep(1); does_file_exist(tempfile); sleep(1)) {
541        pcno = grab_percentage_from_last_line_of_file(MONDO_LOGFILE);
542        if (pcno < 0 || pcno > 100) {
543            mr_msg(5, "Weird pc#");
544            continue;
545        }
546        percentage = pcno * 100 / maxpc;
547        if (pcno <= 5 && last_pcno > 40) {
548            close_evalcall_form();
549            open_evalcall_form("_(Verifying...");
550        }
551        last_pcno = pcno;
552        update_evalcall_form(percentage);
553    }
554// OLD OLD OLD OLD OLD OLD OLD OLD OLD OLD OLD OLD
555    close_evalcall_form();
556    if (pclose(pin)) {
557        res++;
558        log_OS_error("Unable to pclose");
559    }
560    unlink(tempfile);
561    mr_free(tempfile);
562    return (res);
563}
564
565
566void *run_prog_in_bkgd_then_exit(void *info)
567{
568    char *sz_command;
569    static int res = 4444;
570
571    res = 999;
572    sz_command = (char *) info;
573    mr_msg(4, "sz_command = '%s'", sz_command);
574    res = system(sz_command);
575    if (res > 256 && res != 4444) {
576        res = res / 256;
577    }
578    mr_msg(4, "child res = %d", res);
579    sz_command[0] = '\0';
580    pthread_exit((void *) (&res));
581}
582
583
584int run_external_binary_with_percentage_indicator_NEW(char *tt, char *cmd)
585{
586
587    /*@ int *************************************************************** */
588    int res = 0;
589    int percentage = 0;
590    int maxpc = 100;
591    int pcno = 0;
592    int last_pcno = 0;
593    int counter = 0;
594
595    /*@ buffers *********************************************************** */
596    char *command = NULL;
597    /*@ pointers ********************************************************** */
598    static int chldres = 0;
599    int *pchild_result;
600    pthread_t childthread;
601
602    pchild_result = &chldres;
603    assert_string_is_neither_NULL_nor_zerolength(cmd);
604    assert_string_is_neither_NULL_nor_zerolength(tt);
605    *pchild_result = 999;
606
607    mr_asprintf(&command, "%s 2>> %s", cmd, MONDO_LOGFILE);
608    mr_msg(3, "command = '%s'", command);
609    if ((res =
610         pthread_create(&childthread, NULL, run_prog_in_bkgd_then_exit,
611                        (void *) command))) {
612        fatal_error("Unable to create an archival thread");
613    }
614
615    mr_msg(8, "Parent running");
616    open_evalcall_form(tt);
617    for (sleep(1); command[0] != '\0'; sleep(1)) {
618        pcno = grab_percentage_from_last_line_of_file(MONDO_LOGFILE);
619        if (pcno <= 0 || pcno > 100) {
620            mr_msg(8, "Weird pc#");
621            continue;
622        }
623        percentage = pcno * 100 / maxpc;
624        if (pcno <= 5 && last_pcno >= 40) {
625            close_evalcall_form();
626            open_evalcall_form(_("Verifying..."));
627        }
628        if (counter++ >= 5) {
629            counter = 0;
630            log_file_end_to_screen(MONDO_LOGFILE, "");
631        }
632        last_pcno = pcno;
633        update_evalcall_form(percentage);
634    }
635    mr_free(command);
636
637    log_file_end_to_screen(MONDO_LOGFILE, "");
638    close_evalcall_form();
639    pthread_join(childthread, (void *) (&pchild_result));
640    if (pchild_result) {
641        res = *pchild_result;
642    } else {
643        res = 666;
644    }
645    mr_msg(3, "Parent res = %d", res);
646    return (res);
647}
648
649
650/**
651 * Feed @p input_fifo through ntfsclone (restore) to @p output_device.
652 * @param input_fifo The ntfsclone file to read.
653 * @param output_device Where to put the output.
654 * @return The return value of ntfsclone (0 for success).
655 */
656int feed_outfrom_ntfsprog(char *output_device, char *input_fifo)
657{
658// RESTORE
659    int res = -1;
660    char *command = NULL;
661    char *tmp = NULL;
662
663    tmp = find_home_of_exe("ntfsclone");
664    if (!tmp) {
665        fatal_error("ntfsclone not found");
666    }
667    mr_free(tmp);
668
669    mr_asprintf(&command, "ntfsclone --force --restore-image --overwrite %s %s", output_device, input_fifo);
670    res = run_program_and_log_output(command, 5);
671    mr_free(command);
672    return (res);
673}
Note: See TracBrowser for help on using the repository browser.