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

Last change on this file since 130 was 130, checked in by bcornec, 14 years ago

memory management revision on libmondo-fork.c

  • Property svn:keywords set to Id
File size: 29.2 KB
Line 
1/* $Id: libmondo-fork.c 130 2005-11-22 16:47:53Z bcornec $
2  subroutines for handling forking/pthreads/etc.
3*/
4
5
6#include "my-stuff.h"
7#include "mondostructures.h"
8#include "libmondo-fork.h"
9#include "libmondo-string-EXT.h"
10#include "libmondo-gui-EXT.h"
11#include "libmondo-files-EXT.h"
12#include "libmondo-tools-EXT.h"
13#include "lib-common-externs.h"
14
15/*@unused@*/
16//static char cvsid[] = "$Id: libmondo-fork.c 130 2005-11-22 16:47:53Z bcornec $";
17
18extern char *g_tmpfs_mountpt;
19extern t_bkptype g_backup_media_type;
20extern bool g_text_mode;
21pid_t g_buffer_pid = 0;
22
23
24/**
25 * Call a program and retrieve its last line of output.
26 * @param call The program to run.
27 * @return The last line of its output.
28 * @note The returned value points to static storage that will be overwritten with each call.
29 */
30char *call_program_and_get_last_line_of_output(char *call)
31{
32    /*@ buffers ***************************************************** */
33    static char *result = NULL;
34    char *tmp = NULL;
35
36    /*@ pointers **************************************************** */
37    FILE *fin;
38
39    int n = 0;
40
41    /*@******************************************************************** */
42
43    assert_string_is_neither_NULL_nor_zerolength(call);
44    if ((fin = popen(call, "r"))) {
45        for (getline(&tmp, &n, fin); !feof(fin);
46             getline(&tmp, &n, fin)) {
47            if (strlen(tmp) > 1) {
48                if (result != NULL) {
49                    paranoid_free(result);
50                }
51                asprintf(&result, tmp);
52            }
53        }
54        paranoid_pclose(fin);
55    } else {
56        log_OS_error("Unable to popen call");
57    }
58    strip_spaces(result);
59    paranoid_free(tmp);
60    return (result);
61}
62
63#define MONDO_POPMSG  "Your PC will not retract the CD tray automatically. Please call mondoarchive with the -m (manual CD tray) flag."
64
65/**
66 * Call mkisofs to create an ISO image.
67 * @param bkpinfo The backup information structure. Fields used:
68 * - @c bkpinfo->manual_cd_tray
69 * - @c bkpinfo->backup_media_type
70 * - @c bkpinfo->please_dont_eject_when_restoring
71 * @param basic_call The call to mkisofs. May contain tokens that will be resolved to actual data. The tokens are:
72 * - @c _ISO_ will become the ISO file (@p isofile)
73 * - @c _CD#_ becomes the CD number (@p cd_no)
74 * - @c _ERR_ becomes the logfile (@p g_logfile)
75 * @param isofile Replaces @c _ISO_ in @p basic_call. Should probably be the ISO image to create (-o parameter to mkisofs).
76 * @param cd_no Replaces @c _CD#_ in @p basic_call. Should probably be the CD number.
77 * @param logstub Unused.
78 * @param what_i_am_doing The action taking place (e.g. "Making ISO #1"). Used as the title of the progress dialog.
79 * @return Exit code of @c mkisofs (0 is success, anything else indicates failure).
80 * @bug @p logstub is unused.
81 */
82int
83eval_call_to_make_ISO(struct s_bkpinfo *bkpinfo,
84                      char *basic_call, char *isofile,
85                      int cd_no, char *logstub, char *what_i_am_doing)
86{
87
88    /*@ int's  *** */
89    int retval = 0;
90
91
92    /*@ buffers      *** */
93    char *midway_call, *ultimate_call, *tmp, *command,
94        *cd_number_str;
95    char *p;
96
97/*@***********   End Variables ***************************************/
98
99    log_msg(3, "Starting");
100    assert(bkpinfo != NULL);
101    assert_string_is_neither_NULL_nor_zerolength(basic_call);
102    assert_string_is_neither_NULL_nor_zerolength(isofile);
103    assert_string_is_neither_NULL_nor_zerolength(logstub);
104    if (!(midway_call = malloc(1200))) {
105        fatal_error("Cannot malloc midway_call");
106    }
107    if (!(ultimate_call = malloc(1200))) {
108        fatal_error("Cannot malloc ultimate_call");
109    }
110    if (!(tmp = malloc(1200))) {
111        fatal_error("Cannot malloc tmp");
112    }
113
114    asprintf(&cd_number_str, "%d", cd_no);
115    resolve_naff_tokens(midway_call, basic_call, isofile, "_ISO_");
116    resolve_naff_tokens(tmp, midway_call, cd_number_str, "_CD#_");
117    paranoid_free(cd_number_str);
118
119    resolve_naff_tokens(ultimate_call, tmp, MONDO_LOGFILE, "_ERR_");
120    log_msg(4, "basic call = '%s'", basic_call);
121    log_msg(4, "midway_call = '%s'", midway_call);
122    log_msg(4, "tmp = '%s'", tmp);
123    log_msg(4, "ultimate call = '%s'", ultimate_call);
124    asprintf(&command, "%s >> %s", ultimate_call, MONDO_LOGFILE);
125
126    log_to_screen
127        ("Please be patient. Do not be alarmed by on-screen inactivity.");
128    log_msg(4, "Calling open_evalcall_form() with what_i_am_doing='%s'",
129            what_i_am_doing);
130    strcpy(tmp, command);
131    if (bkpinfo->manual_cd_tray) {
132        p = strstr(tmp, "2>>");
133        if (p) {
134            sprintf(p, "   ");
135            while (*p == ' ') {
136                p++;
137            }
138            for (; *p != ' '; p++) {
139                *p = ' ';
140            }
141        }
142        paranoid_free(command);
143        asprintf(&command, tmp);
144#ifndef _XWIN
145        if (!g_text_mode) {
146            newtSuspend();
147        }
148#endif
149        log_msg(1, "command = '%s'", command);
150        retval += system(command);
151        if (!g_text_mode) {
152            newtResume();
153        }
154        if (retval) {
155            log_msg(2, "Basic call '%s' returned an error.", basic_call);
156            popup_and_OK("Press ENTER to continue.");
157            popup_and_OK
158                ("mkisofs and/or cdrecord returned an error. CD was not created");
159        }
160    }
161    /* if text mode then do the above & RETURN; if not text mode, do this... */
162    else {
163        log_msg(3, "command = '%s'", command);
164//      yes_this_is_a_goto:
165        retval =
166            run_external_binary_with_percentage_indicator_NEW
167            (what_i_am_doing, command);
168    }
169    paranoid_free(command);
170
171    paranoid_free(midway_call);
172    paranoid_free(ultimate_call);
173    paranoid_free(tmp);
174    return (retval);
175}
176
177
178/**
179 * Run a program and log its output (stdout and stderr) to the logfile.
180 * @param program The program to run. Passed to the shell, so you can use pipes etc.
181 * @param debug_level If @p g_loglevel is higher than this, do not log the output.
182 * @return The exit code of @p program (depends on the command, but 0 almost always indicates success).
183 */
184int run_program_and_log_output(char *program, int debug_level)
185{
186    /*@ buffer ****************************************************** */
187    char *callstr;
188    char *incoming = NULL;
189    char tmp[MAX_STR_LEN * 2];
190
191    /*@ int ********************************************************* */
192    int res;
193    int i;
194    int n = 0;
195    int len;
196    bool log_if_failure = FALSE;
197    bool log_if_success = FALSE;
198
199    /*@ pointers *************************************************** */
200    FILE *fin;
201    char *p;
202
203    /*@ end vars *************************************************** */
204
205    assert(program != NULL);
206    if (!program[0]) {
207        log_msg(2, "Warning - asked to run zerolength program");
208        return (1);
209    }
210
211    if (debug_level <= g_loglevel) {
212        log_if_success = TRUE;
213        log_if_failure = TRUE;
214    }
215    asprintf(&callstr,
216            "%s > /tmp/mondo-run-prog-thing.tmp 2> /tmp/mondo-run-prog-thing.err",
217            program);
218    while ((p = strchr(callstr, '\r'))) {
219        *p = ' ';
220    }
221    while ((p = strchr(callstr, '\n'))) {
222        *p = ' ';
223    }                           /* single '=' is intentional */
224
225
226    len = (int) strlen(program);
227    for (i = 0; i < 35 - len / 2; i++) {
228        tmp[i] = '-';
229    }
230    tmp[i] = '\0';
231    strcat(tmp, " ");
232    strcat(tmp, program);
233    strcat(tmp, " ");
234    for (i = 0; i < 35 - len / 2; i++) {
235        strcat(tmp, "-");
236    }
237    res = system(callstr);
238    if (((res == 0) && log_if_success) || ((res != 0) && log_if_failure)) {
239        log_msg(0, "running: %s", callstr);
240        log_msg(0,
241                "--------------------------------start of output-----------------------------");
242    }
243    paranoid_free(callstr);
244
245    if (log_if_failure
246        &&
247        system
248        ("cat /tmp/mondo-run-prog-thing.err >> /tmp/mondo-run-prog-thing.tmp 2> /dev/null"))
249    {
250        log_OS_error("Command failed");
251    }
252    unlink("/tmp/mondo-run-prog-thing.err");
253    fin = fopen("/tmp/mondo-run-prog-thing.tmp", "r");
254    if (fin) {
255        for (getline(&incoming, &n, fin); !feof(fin);
256             getline(&incoming, &n, fin)) {
257            /* patch by Heiko Schlittermann */
258            p = incoming;
259            while (p && *p) {
260                if ((p = strchr(p, '%'))) {
261                    memmove(p, p + 1, strlen(p) + 1);
262                    p += 2;
263                }
264            }
265            /* end of patch */
266            strip_spaces(incoming);
267            if ((res == 0 && log_if_success)
268                || (res != 0 && log_if_failure)) {
269                log_msg(0, incoming);
270            }
271        }
272        paranoid_free(incoming);
273        paranoid_fclose(fin);
274    }
275    unlink("/tmp/mondo-run-prog-thing.tmp");
276    if ((res == 0 && log_if_success) || (res != 0 && log_if_failure)) {
277        log_msg(0,
278                "--------------------------------end of output------------------------------");
279        if (res) {
280            log_msg(0, "...ran with res=%d", res);
281        } else {
282            log_msg(0, "...ran just fine. :-)");
283        }
284    }
285//  else
286//    { log_msg (0, "-------------------------------ran w/ res=%d------------------------------", res); }
287    return (res);
288}
289
290
291/**
292 * Run a program and log its output to the screen.
293 * @param basic_call The program to run.
294 * @param what_i_am_doing The title of the evalcall form.
295 * @return The return value of the command (varies, but 0 almost always means success).
296 * @see run_program_and_log_output
297 * @see log_to_screen
298 */
299int run_program_and_log_to_screen(char *basic_call, char *what_i_am_doing)
300{
301    /*@ int ******************************************************** */
302    int retval = 0;
303    int res = 0;
304    int n = 0;
305    int i;
306
307    /*@ pointers **************************************************** */
308    FILE *fin;
309
310    /*@ buffers **************************************************** */
311    char *tmp;
312    char *command;
313    char *lockfile;
314
315    /*@ end vars *************************************************** */
316
317    assert_string_is_neither_NULL_nor_zerolength(basic_call);
318
319    asprintf(&lockfile, "/tmp/mojo-jojo.blah.XXXXXX");
320    mkstemp(lockfile);
321    asprintf(&command,
322            "echo hi > %s ; %s >> %s 2>> %s; res=$?; sleep 1; rm -f %s; exit $res",
323            lockfile, basic_call, MONDO_LOGFILE, MONDO_LOGFILE, lockfile);
324    open_evalcall_form(what_i_am_doing);
325    asprintf(&tmp, "Executing %s", basic_call);
326    log_msg(2, tmp);
327    paranoid_free(tmp);
328
329    if (!(fin = popen(command, "r"))) {
330        log_OS_error("Unable to popen-in command");
331        asprintf(&tmp, "Failed utterly to call '%s'", command);
332        log_to_screen(tmp);
333        paranoid_free(tmp);
334        paranoid_free(lockfile);
335        paranoid_free(command);
336        return (1);
337    }
338    paranoid_free(command);
339
340    if (!does_file_exist(lockfile)) {
341        log_to_screen("Waiting for external binary to start");
342        for (i = 0; i < 60 && !does_file_exist(lockfile); sleep(1), i++) {
343            log_msg(3, "Waiting for lockfile %s to exist", lockfile);
344        }
345    }
346#ifdef _XWIN
347    /* This only can update when newline goes into the file,
348       but it's *much* prettier/faster on Qt. */
349    tmp = NULL;
350    while (does_file_exist(lockfile)) {
351        while (!feof(fin)) {
352            if (!getline(&tmp, &n, fin))
353                break;
354            log_to_screen(tmp);
355        }
356        usleep(500000);
357    }
358    paranoid_free(tmp);
359#else
360    /* This works on Newt, and it gives quicker updates. */
361    for (; does_file_exist(lockfile); sleep(1)) {
362        log_file_end_to_screen(MONDO_LOGFILE, "");
363        update_evalcall_form(1);
364    }
365#endif
366    /* Evaluate the status returned by pclose to get the exit code of the called program. */
367    errno = 0;
368    res = pclose(fin);
369    /* Log actual pclose errors. */
370    if (errno)
371        log_msg(5, "pclose err: %d", errno);
372    /* Check if we have a valid status. If we do, extract the called program's exit code. */
373    /* If we don't, highlight this fact by returning -1. */
374    if (WIFEXITED(res)) {
375        retval = WEXITSTATUS(res);
376    } else {
377        retval = -1;
378    }
379    close_evalcall_form();
380    unlink(lockfile);
381    paranoid_free(lockfile);
382
383    return (retval);
384}
385
386
387/**
388 * Thread callback to run a command (partimage) in the background.
389 * @param xfb A transfer block of @c char, containing:
390 * - xfb:[0] A marker, should be set to 2. Decremented to 1 while the command is running and 0 when it's finished.
391 * - xfb:[1] The command's return value, if xfb:[0] is 0.
392 * - xfb+2:  A <tt>NULL</tt>-terminated string containing the command to be run.
393 * @return NULL to pthread_join.
394 */
395void *call_partimage_in_bkgd(void *xfb)
396{
397    char *transfer_block;
398    int retval = 0;
399
400    g_buffer_pid = getpid();
401    unlink("/tmp/null");
402    log_msg(1, "starting");
403    transfer_block = (char *) xfb;
404    transfer_block[0]--;        // should now be 1
405    retval = system(transfer_block + 2);
406    if (retval) {
407        log_OS_error("partimage returned an error");
408    }
409    transfer_block[1] = retval;
410    transfer_block[0]--;        // should now be 0
411    g_buffer_pid = 0;
412    log_msg(1, "returning");
413    pthread_exit(NULL);
414}
415
416
417/**
418 * File to touch if we want partimage to wait for us.
419 */
420#define PAUSE_PARTIMAGE_FNAME "/tmp/PAUSE-PARTIMAGE-FOR-MONDO"
421
422/**
423 * Apparently used. @bug This has a purpose, but what?
424 */
425#define PIMP_START_SZ "STARTSTARTSTART9ff3kff9a82gv34r7fghbkaBBC2T231hc81h42vws8"
426#define PIMP_END_SZ "ENDENDEND0xBBC10xBBC2T231hc81h42vws89ff3kff9a82gv34r7fghbka"
427
428/**
429 * Marker to start the next subvolume for Partimage.
430 */
431#define NEXT_SUBVOL_PLEASE "I-grew-up-on-the-crime-side,-the-New-York-Times-side,-where-staying-alive-was-no-jive"
432
433/**
434 * Marker to end the partimage file.
435 */
436#define NO_MORE_SUBVOLS "On-second-hand,momma-bounced-on-old-man,-and-so-we-moved-to-Shaolin-Land."
437
438int copy_from_src_to_dest(FILE * f_orig, FILE * f_archived, char direction)
439{
440// if dir=='w' then copy from orig to archived
441// if dir=='r' then copy from archived to orig
442    char *tmp;
443    char *buf;
444    long int bytes_to_be_read, bytes_read_in, bytes_written_out =
445        0, bufcap, subsliceno = 0;
446    int retval = 0;
447    FILE *fin;
448    FILE *fout;
449    FILE *ftmp;
450
451    log_msg(5, "Opening.");
452    bufcap = 256L * 1024L;
453    if (!(buf = malloc(bufcap))) {
454        fatal_error("Failed to malloc() buf");
455    }
456
457    if (direction == 'w') {
458        fin = f_orig;
459        fout = f_archived;
460        asprintf(&tmp, "%-64s", PIMP_START_SZ);
461        if (fwrite(tmp, 1, 64, fout) != 64) {
462            fatal_error("Can't write the introductory block");
463        }
464        paranoid_free(tmp);
465
466        while (1) {
467            bytes_to_be_read = bytes_read_in = fread(buf, 1, bufcap, fin);
468            if (bytes_read_in == 0) {
469                break;
470            }
471            asprintf(&tmp, "%-64ld", bytes_read_in);
472            if (fwrite(tmp, 1, 64, fout) != 64) {
473                fatal_error("Cannot write introductory block");
474            }
475            paranoid_free(tmp);
476
477            log_msg(7,
478                    "subslice #%ld --- I have read %ld of %ld bytes in from f_orig",
479                    subsliceno, bytes_read_in, bytes_to_be_read);
480            bytes_written_out += fwrite(buf, 1, bytes_read_in, fout);
481            asprintf(&tmp, "%-64ld", subsliceno);
482            if (fwrite(tmp, 1, 64, fout) != 64) {
483                fatal_error("Cannot write post-thingy block");
484            }
485            paranoid_free(tmp);
486
487            log_msg(7, "Subslice #%d written OK", subsliceno);
488            subsliceno++;
489        }
490        asprintf(&tmp, "%-64ld", 0L);
491        if (fwrite(tmp, 1, 64L, fout) != 64L) {
492            fatal_error("Cannot write final introductory block");
493        }
494    } else {
495        fin = f_archived;
496        fout = f_orig;
497        if (!(tmp = malloc(64L))) {
498            fatal_error("Failed to malloc() tmp");
499        }
500        if (fread(tmp, 1, 64L, fin) != 64L) {
501            fatal_error("Cannot read the introductory block");
502        }
503        log_msg(5, "tmp is %s", tmp);
504        if (!strstr(tmp, PIMP_START_SZ)) {
505            fatal_error("Can't find intro blk");
506        }
507        if (fread(tmp, 1, 64L, fin) != 64L) {
508            fatal_error("Cannot read introductory blk");
509        }
510        bytes_to_be_read = atol(tmp);
511        while (bytes_to_be_read > 0) {
512            log_msg(7, "subslice#%ld, bytes=%ld", subsliceno,
513                    bytes_to_be_read);
514            bytes_read_in = fread(buf, 1, bytes_to_be_read, fin);
515            if (bytes_read_in != bytes_to_be_read) {
516                fatal_error
517                    ("Danger, WIll Robinson. Failed to read whole subvol from archives.");
518            }
519            bytes_written_out += fwrite(buf, 1, bytes_read_in, fout);
520            if (fread(tmp, 1, 64, fin) != 64) {
521                fatal_error("Cannot read post-thingy block");
522            }
523            if (atol(tmp) != subsliceno) {
524                log_msg(1, "Wanted subslice %ld but got %ld ('%s')",
525                        subsliceno, atol(tmp), tmp);
526            }
527            log_msg(7, "Subslice #%ld read OK", subsliceno);
528            subsliceno++;
529            if (fread(tmp, 1, 64, fin) != 64) {
530                fatal_error("Cannot read introductory block");
531            }
532            bytes_to_be_read = atol(tmp);
533        }
534    }
535
536//  log_msg(4, "Written %ld of %ld bytes", bytes_written_out, bytes_read_in);
537
538    if (direction == 'w') {
539        asprintf(&tmp, "%-64s", PIMP_END_SZ);
540        if (fwrite(tmp, 1, 64, fout) != 64) {
541            fatal_error("Can't write the final block");
542        }
543        paranoid_free(tmp);
544    } else {
545        log_msg(1, "tmpA is %s", tmp);
546        if (!strstr(tmp, PIMP_END_SZ)) {
547            if (fread(tmp, 1, 64, fin) != 64) {
548                fatal_error("Can't read the final block");
549            }
550            log_msg(5, "tmpB is %s", tmp);
551            if (!strstr(tmp, PIMP_END_SZ)) {
552                ftmp = fopen("/tmp/out.leftover", "w");
553                bytes_read_in = fread(tmp, 1, 64, fin);
554                log_msg(1, "bytes_read_in = %ld", bytes_read_in);
555//      if (bytes_read_in!=128+64) { fatal_error("Can't read the terminating block"); }
556                fwrite(tmp, 1, bytes_read_in, ftmp);
557                paranoid_free(tmp);
558               
559                if (!(tmp = malloc(512))) {
560                    fatal_error("Failed to malloc() tmp");
561                }
562                fread(tmp, 1, 512, fin);
563                log_msg(0, "tmp = '%s'", tmp);
564                fwrite(tmp, 1, 512, ftmp);
565                fclose(ftmp);
566                fatal_error("Missing terminating block");
567            }
568        }
569        paranoid_free(tmp);
570    }
571
572    paranoid_free(buf);
573    log_msg(3, "Successfully copied %ld bytes", bytes_written_out);
574    return (retval);
575}
576
577
578/**
579 * Call partimage from @p input_device to @p output_fname.
580 * @param input_device The device to read.
581 * @param output_fname The file to write.
582 * @return 0 for success, nonzero for failure.
583 */
584int dynamically_create_pipes_and_copy_from_them_to_output_file(char
585                                                               *input_device, char
586                                                               *output_fname)
587{
588    char *curr_fifo;
589    char *prev_fifo = NULL;
590    char *next_fifo;
591    char *command;
592    char *sz_call_to_partimage;
593    int fifo_number = 0;
594    struct stat buf;
595    pthread_t partimage_thread;
596    int res = 0;
597    char *tmpstub;
598    FILE *fout;
599    FILE *fin;
600    char *tmp;
601
602    log_msg(1, "g_tmpfs_mountpt = %s", g_tmpfs_mountpt);
603    if (g_tmpfs_mountpt && g_tmpfs_mountpt[0]
604        && does_file_exist(g_tmpfs_mountpt)) {
605        asprintf(&tmpstub, g_tmpfs_mountpt);
606    } else {
607        asprintf(&tmpstub, "/tmp");
608    }
609    paranoid_system("rm -f /tmp/*PARTIMAGE*");
610    asprintf(&command, "rm -Rf %s/pih-fifo-*", tmpstub);
611    paranoid_system(command);
612    paranoid_free(command);
613
614    asprintf(&tmp, "%s/pih-fifo-%ld", tmpstub, (long int) random());
615    paranoid_free(tmpstub);
616    tmpstub = tmp;
617    paranoid_free(tmp);
618
619    mkfifo(tmpstub, S_IRWXU | S_IRWXG); // never used, though...
620    asprintf(&curr_fifo, "%s.%03d", tmpstub, fifo_number);
621    asprintf(&next_fifo, "%s.%03d", tmpstub, fifo_number + 1);
622    mkfifo(curr_fifo, S_IRWXU | S_IRWXG);
623    mkfifo(next_fifo, S_IRWXU | S_IRWXG);   // make sure _next_ fifo already exists before we call partimage
624    asprintf(&sz_call_to_partimage,
625            "%c%cpartimagehack " PARTIMAGE_PARAMS
626            " save %s %s > /tmp/stdout 2> /tmp/stderr", 2, 0, input_device,
627            tmpstub);
628    log_msg(5, "curr_fifo   = %s", curr_fifo);
629    log_msg(5, "next_fifo   = %s", next_fifo);
630    log_msg(5, "sz_call_to_partimage call is '%s'",
631            sz_call_to_partimage + 2);
632    if (!lstat(output_fname, &buf) && S_ISREG(buf.st_mode)) {
633        log_msg(5, "Deleting %s", output_fname);
634        unlink(output_fname);
635    }
636    if (!(fout = fopen(output_fname, "w"))) {
637        fatal_error("Unable to openout to output_fname");
638    }
639    res =
640        pthread_create(&partimage_thread, NULL, call_partimage_in_bkgd,
641                       (void *) sz_call_to_partimage);
642    if (res) {
643        fatal_error("Failed to create thread to call partimage");
644    }
645    log_msg(1, "Running fore/back at same time");
646    log_to_screen("Working with partimagehack...");
647    while (sz_call_to_partimage[0] > 0) {
648        asprintf(&tmp, "%s\n", NEXT_SUBVOL_PLEASE);
649        if (fwrite(tmp, 1, 128, fout) != 128) {
650            fatal_error("Cannot write interim block");
651        }
652        paranoid_free(tmp);
653
654        log_msg(5, "fifo_number=%d", fifo_number);
655        log_msg(4, "Cat'ting %s", curr_fifo);
656        if (!(fin = fopen(curr_fifo, "r"))) {
657            fatal_error("Unable to openin from fifo");
658        }
659        if (prev_fifo !=  NULL) {
660            log_msg(5, "Deleting %s", prev_fifo);
661            unlink(prev_fifo);      // just in case
662            paranoid_free(prev_fifo);
663        }
664        copy_from_src_to_dest(fin, fout, 'w');
665        paranoid_fclose(fin);
666        fifo_number++;
667
668        prev_fifo = curr_fifo;
669        curr_fifo = next_fifo;
670        log_msg(5, "Creating %s", next_fifo);
671        asprintf(&next_fifo, "%s.%03d", tmpstub, fifo_number + 1);
672        mkfifo(next_fifo, S_IRWXU | S_IRWXG);   // make sure _next_ fifo exists before we cat this one
673        system("sync");
674        sleep(5);
675    }
676    asprintf(&tmp, "%s\n", NO_MORE_SUBVOLS);
677    if (fwrite(tmp, 1, 128, fout) != 128) {
678        fatal_error("Cannot write interim block");
679    }
680    if (fwrite(tmp, 1, 128, fout) != 128) {
681        fatal_error("Cannot write interim block");
682    }
683    if (fwrite(tmp, 1, 128, fout) != 128) {
684        fatal_error("Cannot write interim block");
685    }
686    if (fwrite(tmp, 1, 128, fout) != 128) {
687        fatal_error("Cannot write interim block");
688    }
689    paranoid_free(tmp);
690    paranoid_fclose(fout);
691    log_to_screen("Cleaning up after partimagehack...");
692    log_msg(3, "Final fifo_number=%d", fifo_number);
693    paranoid_system("sync");
694    unlink(next_fifo);
695    paranoid_free(next_fifo);
696
697    unlink(curr_fifo);
698    paranoid_free(curr_fifo);
699
700    unlink(prev_fifo);
701    paranoid_free(prev_fifo);
702
703    log_to_screen("Finished cleaning up.");
704
705//  if (!lstat(sz_wait_for_this_file, &statbuf))
706//    { log_msg(3, "WARNING! %s was not processed.", sz_wait_for_this_file); }
707    log_msg(2, "Waiting for pthread_join() to join.");
708    pthread_join(partimage_thread, NULL);
709    res = sz_call_to_partimage[1];
710    paranoid_free(sz_call_to_partimage);
711    log_msg(2, "pthread_join() joined OK.");
712    log_msg(1, "Partimagehack(save) returned %d", res);
713    unlink(tmpstub);
714    paranoid_free(tmpstub);
715
716    return (res);
717}
718
719
720/**
721 * Feed @p input_device through partimage to @p output_fname.
722 * @param input_device The device to image.
723 * @param output_fname The file to write.
724 * @return 0 for success, nonzero for failure.
725 */
726int feed_into_partimage(char *input_device, char *output_fname)
727{
728// BACKUP
729    int res;
730
731    if (!does_file_exist(input_device)) {
732        fatal_error("input device does not exist");
733    }
734    if (!find_home_of_exe("partimagehack")) {
735        fatal_error("partimagehack not found");
736    }
737    res =
738        dynamically_create_pipes_and_copy_from_them_to_output_file
739        (input_device, output_fname);
740    return (res);
741}
742
743
744int run_external_binary_with_percentage_indicator_OLD(char *tt, char *cmd)
745{
746
747    /*@ int *************************************************************** */
748    int res = 0;
749    int percentage = 0;
750    int maxpc = 0;
751    int pcno = 0;
752    int last_pcno = 0;
753
754    /*@ buffers *********************************************************** */
755    char *command;
756    char *tempfile;
757    /*@ pointers ********************************************************** */
758    FILE *pin;
759
760    assert_string_is_neither_NULL_nor_zerolength(cmd);
761
762    asprintf(&tempfile,
763           call_program_and_get_last_line_of_output
764           ("mktemp -q /tmp/mondo.XXXXXXXX"));
765    asprintf(&command, "%s >> %s 2>> %s; rm -f %s", cmd, tempfile, tempfile,
766            tempfile);
767    log_msg(3, command);
768    open_evalcall_form(tt);
769
770    if (!(pin = popen(command, "r"))) {
771        log_OS_error("fmt err");
772        paranoid_free(command);
773        paranoid_free(tempfile);
774        return (1);
775    }
776    paranoid_free(command);
777
778    maxpc = 100;
779// OLD OLD OLD OLD OLD OLD OLD OLD OLD OLD OLD OLD
780    for (sleep(1); does_file_exist(tempfile); sleep(1)) {
781        pcno = grab_percentage_from_last_line_of_file(MONDO_LOGFILE);
782        if (pcno < 0 || pcno > 100) {
783            log_msg(5, "Weird pc#");
784            continue;
785        }
786        percentage = pcno * 100 / maxpc;
787        if (pcno <= 5 && last_pcno > 40) {
788            close_evalcall_form();
789            open_evalcall_form("Verifying...");
790        }
791        last_pcno = pcno;
792        update_evalcall_form(percentage);
793    }
794// OLD OLD OLD OLD OLD OLD OLD OLD OLD OLD OLD OLD
795    close_evalcall_form();
796    if (pclose(pin)) {
797        res++;
798        log_OS_error("Unable to pclose");
799    }
800    unlink(tempfile);
801    paranoid_free(tempfile);
802    return (res);
803}
804
805
806void *run_prog_in_bkgd_then_exit(void *info)
807{
808    char *sz_command;
809    static int res = 4444;
810
811    res = 999;
812    sz_command = (char *) info;
813    log_msg(4, "sz_command = '%s'", sz_command);
814    res = system(sz_command);
815    if (res > 256 && res != 4444) {
816        res = res / 256;
817    }
818    log_msg(4, "child res = %d", res);
819    sz_command[0] = '\0';
820    pthread_exit((void *) (&res));
821}
822
823
824int run_external_binary_with_percentage_indicator_NEW(char *tt, char *cmd)
825{
826
827    /*@ int *************************************************************** */
828    int res = 0;
829    int percentage = 0;
830    int maxpc = 100;
831    int pcno = 0;
832    int last_pcno = 0;
833    int counter = 0;
834
835    /*@ buffers *********************************************************** */
836    char *command;
837    /*@ pointers ********************************************************** */
838    static int chldres = 0;
839    int *pchild_result;
840    pthread_t childthread;
841
842    pchild_result = &chldres;
843    assert_string_is_neither_NULL_nor_zerolength(cmd);
844    assert_string_is_neither_NULL_nor_zerolength(tt);
845    *pchild_result = 999;
846
847    asprintf(&command, "%s 2>> %s", cmd, MONDO_LOGFILE);
848    log_msg(3, "command = '%s'", command);
849    if ((res =
850         pthread_create(&childthread, NULL, run_prog_in_bkgd_then_exit,
851                        (void *) command))) {
852        fatal_error("Unable to create an archival thread");
853    }
854
855    log_msg(8, "Parent running");
856    open_evalcall_form(tt);
857    for (sleep(1); command[0] != '\0'; sleep(1)) {
858        pcno = grab_percentage_from_last_line_of_file(MONDO_LOGFILE);
859        if (pcno <= 0 || pcno > 100) {
860            log_msg(8, "Weird pc#");
861            continue;
862        }
863        percentage = pcno * 100 / maxpc;
864        if (pcno <= 5 && last_pcno >= 40) {
865            close_evalcall_form();
866            open_evalcall_form("Verifying...");
867        }
868        if (counter++ >= 5) {
869            counter = 0;
870            log_file_end_to_screen(MONDO_LOGFILE, "");
871        }
872        last_pcno = pcno;
873        update_evalcall_form(percentage);
874    }
875    paranoid_free(command);
876
877    log_file_end_to_screen(MONDO_LOGFILE, "");
878    close_evalcall_form();
879    pthread_join(childthread, (void *) (&pchild_result));
880    if (pchild_result) {
881        res = *pchild_result;
882    } else {
883        res = 666;
884    }
885    log_msg(3, "Parent res = %d", res);
886    return (res);
887}
888
889#define PIH_LOG "/var/log/partimage-debug.log"
890
891/**
892 * Feed @p input_fifo through partimage (restore) to @p output_device.
893 * @param input_fifo The partimage file to read.
894 * @param output_device Where to put the output.
895 * @return The return value of partimagehack (0 for success).
896 * @bug Probably unnecessary, as the partimage is just a sparse file. We could use @c dd to restore it.
897 */
898int feed_outfrom_partimage(char *output_device, char *input_fifo)
899{
900// RESTORE
901    char *tmp;
902    char *stuff;
903    char *sz_call_to_partimage;
904    pthread_t partimage_thread;
905    int res;
906    char *curr_fifo;
907    char *prev_fifo;
908    char *oldest_fifo = NULL;
909    char *next_fifo;
910    char *afternxt_fifo;
911    int fifo_number = 0;
912    char *tmpstub;
913    FILE *fin;
914    FILE *fout;
915
916    log_msg(1, "output_device=%s", output_device);
917    log_msg(1, "input_fifo=%s", input_fifo);
918    asprintf(&tmpstub, "/tmp");
919
920    log_msg(1, "tmpstub was %s", tmpstub);
921    asprintf(&stuff, tmpstub);
922    paranoid_free(tmpstub);
923
924    asprintf(&tmpstub, "%s/pih-fifo-%ld", stuff, (long int) random());
925    paranoid_free(stuff);
926
927    log_msg(1, "tmpstub is now %s", tmpstub);
928    unlink("/tmp/PARTIMAGEHACK-POSITION");
929    unlink(PAUSE_PARTIMAGE_FNAME);
930    paranoid_system("rm -f /tmp/*PARTIMAGE*");
931    asprintf(&curr_fifo, "%s.%03d", tmpstub, fifo_number);
932    asprintf(&next_fifo, "%s.%03d", tmpstub, fifo_number + 1);
933    asprintf(&afternxt_fifo, "%s.%03d", tmpstub, fifo_number + 2);
934    mkfifo(PIH_LOG, S_IRWXU | S_IRWXG);
935    mkfifo(curr_fifo, S_IRWXU | S_IRWXG);
936    mkfifo(next_fifo, S_IRWXU | S_IRWXG);   // make sure _next_ fifo already exists before we call partimage
937    mkfifo(afternxt_fifo, S_IRWXU | S_IRWXG);
938    system("cat " PIH_LOG " > /dev/null &");
939    log_msg(3, "curr_fifo   = %s", curr_fifo);
940    log_msg(3, "next_fifo   = %s", next_fifo);
941    if (!does_file_exist(input_fifo)) {
942        fatal_error("input fifo does not exist");
943    }
944    if (!(fin = fopen(input_fifo, "r"))) {
945        fatal_error("Unable to openin from input_fifo");
946    }
947    if (!find_home_of_exe("partimagehack")) {
948        fatal_error("partimagehack not found");
949    }
950    asprintf(&sz_call_to_partimage,
951            "%c%cpartimagehack " PARTIMAGE_PARAMS
952            " restore %s %s > /dev/null 2>> %s", 2, 0, output_device, curr_fifo,
953            MONDO_LOGFILE);
954    log_msg(1, "output_device = %s", output_device);
955    log_msg(1, "curr_fifo = %s", curr_fifo);
956    log_msg(1, "sz_call_to_partimage+2 = %s", sz_call_to_partimage + 2);
957    res =
958        pthread_create(&partimage_thread, NULL, call_partimage_in_bkgd,
959                       (void *) sz_call_to_partimage);
960    if (res) {
961        fatal_error("Failed to create thread to call partimage");
962    }
963    log_msg(1, "Running fore/back at same time");
964    log_msg(2, " Trying to openin %s", input_fifo);
965    if (!does_file_exist(input_fifo)) {
966        log_msg(2, "Warning - %s does not exist", input_fifo);
967    }
968    while (!does_file_exist("/tmp/PARTIMAGEHACK-POSITION")) {
969        log_msg(6, "Waiting for partimagehack (restore) to start");
970        sleep(1);
971    }
972
973    if (!(tmp = malloc(128))) {
974        fatal_error("Failed to malloc() tmp");
975    }
976    while (sz_call_to_partimage[0] > 0) {
977        if (fread(tmp, 1, 128, fin) != 128) {
978            fatal_error("Cannot read introductory block");
979        }
980        if (strstr(tmp, NEXT_SUBVOL_PLEASE)) {
981            log_msg(2, "Great. Next subvol coming up.");
982        } else if (strstr(tmp, NO_MORE_SUBVOLS)) {
983            log_msg(2, "Great. That was the last subvol.");
984            break;
985        } else {
986            log_msg(2, "WTF is this? '%s'", tmp);
987            fatal_error("Unknown interim block");
988        }
989        if (feof(fin)) {
990            log_msg(1, "Eof(fin) detected. Breaking.");
991            break;
992        }
993        log_msg(3, "Processing subvol %d", fifo_number);
994        log_msg(5, "fifo_number=%d", fifo_number);
995        if (!(fout = fopen(curr_fifo, "w"))) {
996            fatal_error("Cannot openout to curr_fifo");
997        }
998        copy_from_src_to_dest(fout, fin, 'r');
999        paranoid_fclose(fout);
1000        fifo_number++;
1001        if (oldest_fifo != NULL) {
1002            log_msg(6, "Deleting %s", oldest_fifo);
1003            unlink(oldest_fifo);    // just in case
1004            paranoid_free(oldest_fifo);
1005        }
1006        oldest_fifo = prev_fifo;
1007        prev_fifo = curr_fifo;
1008        curr_fifo = next_fifo;
1009        next_fifo = afternxt_fifo;
1010        asprintf(&afternxt_fifo, "%s.%03d", tmpstub, fifo_number + 2);
1011        log_msg(6, "Creating %s", afternxt_fifo);
1012        mkfifo(afternxt_fifo, S_IRWXU | S_IRWXG);   // make sure _next_ fifo already exists before we access current fifo
1013        fflush(fin);
1014//      system("sync");
1015        usleep(1000L * 100L);
1016    }
1017    paranoid_free(tmp);
1018    paranoid_free(tmpstub);
1019
1020    paranoid_fclose(fin);
1021    paranoid_system("sync");
1022    log_msg(1, "Partimagehack has finished. Great. Fin-closing.");
1023    log_msg(1, "Waiting for pthread_join");
1024    pthread_join(partimage_thread, NULL);
1025    res = sz_call_to_partimage[1];
1026    paranoid_free(sz_call_to_partimage);
1027
1028    log_msg(1, "Yay. Partimagehack (restore) returned %d", res);
1029    unlink(prev_fifo);
1030    paranoid_free(prev_fifo);
1031
1032    unlink(curr_fifo);
1033    paranoid_free(curr_fifo);
1034
1035    unlink(next_fifo);
1036    paranoid_free(next_fifo);
1037
1038    unlink(afternxt_fifo);
1039    paranoid_free(afternxt_fifo);
1040
1041    unlink(PIH_LOG);
1042    /* BERLIOS : pas de unlink(oldest_fifo) ??? */
1043    paranoid_free(oldest_fifo);
1044
1045    return (res);
1046}
Note: See TracBrowser for help on using the repository browser.