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

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

memory management continues:

  • mondoarchive handled completely
  • bkpinfo, begining of dyn. alloc.
  • lot of changes around memory everywhere

=> even if it compiles, i'm pretty sure it doesn't work yet (even not tried)

  • Property svn:keywords set to Id
File size: 28.9 KB
Line 
1/* $Id: libmondo-fork.c 171 2005-12-08 16:20:29Z 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 171 2005-12-08 16:20:29Z 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    size_t 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    size_t 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 i;
305
306    /*@ pointers **************************************************** */
307    FILE *fin;
308
309    /*@ buffers **************************************************** */
310    char *tmp;
311    char *command;
312    char *lockfile;
313
314    /*@ end vars *************************************************** */
315
316    assert_string_is_neither_NULL_nor_zerolength(basic_call);
317
318    asprintf(&lockfile, "/tmp/mojo-jojo.blah.XXXXXX");
319    mkstemp(lockfile);
320    asprintf(&command,
321            "echo hi > %s ; %s >> %s 2>> %s; res=$?; sleep 1; rm -f %s; exit $res",
322            lockfile, basic_call, MONDO_LOGFILE, MONDO_LOGFILE, lockfile);
323    open_evalcall_form(what_i_am_doing);
324    asprintf(&tmp, "Executing %s", basic_call);
325    log_msg(2, tmp);
326    paranoid_free(tmp);
327
328    if (!(fin = popen(command, "r"))) {
329        log_OS_error("Unable to popen-in command");
330        asprintf(&tmp, "Failed utterly to call '%s'", command);
331        log_to_screen(tmp);
332        paranoid_free(tmp);
333        paranoid_free(lockfile);
334        paranoid_free(command);
335        return (1);
336    }
337    paranoid_free(command);
338
339    if (!does_file_exist(lockfile)) {
340        log_to_screen("Waiting for external binary to start");
341        for (i = 0; i < 60 && !does_file_exist(lockfile); sleep(1), i++) {
342            log_msg(3, "Waiting for lockfile %s to exist", lockfile);
343        }
344    }
345    /* This works on Newt, and it gives quicker updates. */
346    for (; does_file_exist(lockfile); sleep(1)) {
347        log_file_end_to_screen(MONDO_LOGFILE, "");
348        update_evalcall_form(1);
349    }
350    /* Evaluate the status returned by pclose to get the exit code of the called program. */
351    errno = 0;
352    res = pclose(fin);
353    /* Log actual pclose errors. */
354    if (errno)
355        log_msg(5, "pclose err: %d", errno);
356    /* Check if we have a valid status. If we do, extract the called program's exit code. */
357    /* If we don't, highlight this fact by returning -1. */
358    if (WIFEXITED(res)) {
359        retval = WEXITSTATUS(res);
360    } else {
361        retval = -1;
362    }
363    close_evalcall_form();
364    unlink(lockfile);
365    paranoid_free(lockfile);
366
367    return (retval);
368}
369
370
371/**
372 * Thread callback to run a command (partimage) in the background.
373 * @param xfb A transfer block of @c char, containing:
374 * - xfb:[0] A marker, should be set to 2. Decremented to 1 while the command is running and 0 when it's finished.
375 * - xfb:[1] The command's return value, if xfb:[0] is 0.
376 * - xfb+2:  A <tt>NULL</tt>-terminated string containing the command to be run.
377 * @return NULL to pthread_join.
378 */
379void *call_partimage_in_bkgd(void *xfb)
380{
381    char *transfer_block;
382    int retval = 0;
383
384    g_buffer_pid = getpid();
385    unlink("/tmp/null");
386    log_msg(1, "starting");
387    transfer_block = (char *) xfb;
388    transfer_block[0]--;        // should now be 1
389    retval = system(transfer_block + 2);
390    if (retval) {
391        log_OS_error("partimage returned an error");
392    }
393    transfer_block[1] = retval;
394    transfer_block[0]--;        // should now be 0
395    g_buffer_pid = 0;
396    log_msg(1, "returning");
397    pthread_exit(NULL);
398}
399
400
401/**
402 * File to touch if we want partimage to wait for us.
403 */
404#define PAUSE_PARTIMAGE_FNAME "/tmp/PAUSE-PARTIMAGE-FOR-MONDO"
405
406/**
407 * Apparently used. @bug This has a purpose, but what?
408 */
409#define PIMP_START_SZ "STARTSTARTSTART9ff3kff9a82gv34r7fghbkaBBC2T231hc81h42vws8"
410#define PIMP_END_SZ "ENDENDEND0xBBC10xBBC2T231hc81h42vws89ff3kff9a82gv34r7fghbka"
411
412/**
413 * Marker to start the next subvolume for Partimage.
414 */
415#define NEXT_SUBVOL_PLEASE "I-grew-up-on-the-crime-side,-the-New-York-Times-side,-where-staying-alive-was-no-jive"
416
417/**
418 * Marker to end the partimage file.
419 */
420#define NO_MORE_SUBVOLS "On-second-hand,momma-bounced-on-old-man,-and-so-we-moved-to-Shaolin-Land."
421
422int copy_from_src_to_dest(FILE * f_orig, FILE * f_archived, char direction)
423{
424// if dir=='w' then copy from orig to archived
425// if dir=='r' then copy from archived to orig
426    char *tmp;
427    char *buf;
428    long int bytes_to_be_read, bytes_read_in, bytes_written_out =
429        0, bufcap, subsliceno = 0;
430    int retval = 0;
431    FILE *fin;
432    FILE *fout;
433    FILE *ftmp;
434
435    log_msg(5, "Opening.");
436    bufcap = 256L * 1024L;
437    if (!(buf = malloc(bufcap))) {
438        fatal_error("Failed to malloc() buf");
439    }
440
441    if (direction == 'w') {
442        fin = f_orig;
443        fout = f_archived;
444        asprintf(&tmp, "%-64s", PIMP_START_SZ);
445        if (fwrite(tmp, 1, 64, fout) != 64) {
446            fatal_error("Can't write the introductory block");
447        }
448        paranoid_free(tmp);
449
450        while (1) {
451            bytes_to_be_read = bytes_read_in = fread(buf, 1, bufcap, fin);
452            if (bytes_read_in == 0) {
453                break;
454            }
455            asprintf(&tmp, "%-64ld", bytes_read_in);
456            if (fwrite(tmp, 1, 64, fout) != 64) {
457                fatal_error("Cannot write introductory block");
458            }
459            paranoid_free(tmp);
460
461            log_msg(7,
462                    "subslice #%ld --- I have read %ld of %ld bytes in from f_orig",
463                    subsliceno, bytes_read_in, bytes_to_be_read);
464            bytes_written_out += fwrite(buf, 1, bytes_read_in, fout);
465            asprintf(&tmp, "%-64ld", subsliceno);
466            if (fwrite(tmp, 1, 64, fout) != 64) {
467                fatal_error("Cannot write post-thingy block");
468            }
469            paranoid_free(tmp);
470
471            log_msg(7, "Subslice #%d written OK", subsliceno);
472            subsliceno++;
473        }
474        asprintf(&tmp, "%-64ld", 0L);
475        if (fwrite(tmp, 1, 64L, fout) != 64L) {
476            fatal_error("Cannot write final introductory block");
477        }
478    } else {
479        fin = f_archived;
480        fout = f_orig;
481        if (!(tmp = malloc(64L))) {
482            fatal_error("Failed to malloc() tmp");
483        }
484        if (fread(tmp, 1, 64L, fin) != 64L) {
485            fatal_error("Cannot read the introductory block");
486        }
487        log_msg(5, "tmp is %s", tmp);
488        if (!strstr(tmp, PIMP_START_SZ)) {
489            fatal_error("Can't find intro blk");
490        }
491        if (fread(tmp, 1, 64L, fin) != 64L) {
492            fatal_error("Cannot read introductory blk");
493        }
494        bytes_to_be_read = atol(tmp);
495        while (bytes_to_be_read > 0) {
496            log_msg(7, "subslice#%ld, bytes=%ld", subsliceno,
497                    bytes_to_be_read);
498            bytes_read_in = fread(buf, 1, bytes_to_be_read, fin);
499            if (bytes_read_in != bytes_to_be_read) {
500                fatal_error
501                    ("Danger, WIll Robinson. Failed to read whole subvol from archives.");
502            }
503            bytes_written_out += fwrite(buf, 1, bytes_read_in, fout);
504            if (fread(tmp, 1, 64, fin) != 64) {
505                fatal_error("Cannot read post-thingy block");
506            }
507            if (atol(tmp) != subsliceno) {
508                log_msg(1, "Wanted subslice %ld but got %ld ('%s')",
509                        subsliceno, atol(tmp), tmp);
510            }
511            log_msg(7, "Subslice #%ld read OK", subsliceno);
512            subsliceno++;
513            if (fread(tmp, 1, 64, fin) != 64) {
514                fatal_error("Cannot read introductory block");
515            }
516            bytes_to_be_read = atol(tmp);
517        }
518    }
519
520//  log_msg(4, "Written %ld of %ld bytes", bytes_written_out, bytes_read_in);
521
522    if (direction == 'w') {
523        asprintf(&tmp, "%-64s", PIMP_END_SZ);
524        if (fwrite(tmp, 1, 64, fout) != 64) {
525            fatal_error("Can't write the final block");
526        }
527        paranoid_free(tmp);
528    } else {
529        log_msg(1, "tmpA is %s", tmp);
530        if (!strstr(tmp, PIMP_END_SZ)) {
531            if (fread(tmp, 1, 64, fin) != 64) {
532                fatal_error("Can't read the final block");
533            }
534            log_msg(5, "tmpB is %s", tmp);
535            if (!strstr(tmp, PIMP_END_SZ)) {
536                ftmp = fopen("/tmp/out.leftover", "w");
537                bytes_read_in = fread(tmp, 1, 64, fin);
538                log_msg(1, "bytes_read_in = %ld", bytes_read_in);
539//      if (bytes_read_in!=128+64) { fatal_error("Can't read the terminating block"); }
540                fwrite(tmp, 1, bytes_read_in, ftmp);
541                paranoid_free(tmp);
542               
543                if (!(tmp = malloc(512))) {
544                    fatal_error("Failed to malloc() tmp");
545                }
546                fread(tmp, 1, 512, fin);
547                log_msg(0, "tmp = '%s'", tmp);
548                fwrite(tmp, 1, 512, ftmp);
549                fclose(ftmp);
550                fatal_error("Missing terminating block");
551            }
552        }
553        paranoid_free(tmp);
554    }
555
556    paranoid_free(buf);
557    log_msg(3, "Successfully copied %ld bytes", bytes_written_out);
558    return (retval);
559}
560
561
562/**
563 * Call partimage from @p input_device to @p output_fname.
564 * @param input_device The device to read.
565 * @param output_fname The file to write.
566 * @return 0 for success, nonzero for failure.
567 */
568int dynamically_create_pipes_and_copy_from_them_to_output_file(char
569                                                               *input_device, char
570                                                               *output_fname)
571{
572    char *curr_fifo;
573    char *prev_fifo = NULL;
574    char *next_fifo;
575    char *command;
576    char *sz_call_to_partimage;
577    int fifo_number = 0;
578    struct stat buf;
579    pthread_t partimage_thread;
580    int res = 0;
581    char *tmpstub;
582    FILE *fout;
583    FILE *fin;
584    char *tmp;
585
586    log_msg(1, "g_tmpfs_mountpt = %s", g_tmpfs_mountpt);
587    if (g_tmpfs_mountpt && g_tmpfs_mountpt[0]
588        && does_file_exist(g_tmpfs_mountpt)) {
589        asprintf(&tmpstub, g_tmpfs_mountpt);
590    } else {
591        asprintf(&tmpstub, "/tmp");
592    }
593    paranoid_system("rm -f /tmp/*PARTIMAGE*");
594    asprintf(&command, "rm -Rf %s/pih-fifo-*", tmpstub);
595    paranoid_system(command);
596    paranoid_free(command);
597
598    asprintf(&tmp, "%s/pih-fifo-%ld", tmpstub, (long int) random());
599    paranoid_free(tmpstub);
600    tmpstub = tmp;
601    paranoid_free(tmp);
602
603    mkfifo(tmpstub, S_IRWXU | S_IRWXG); // never used, though...
604    asprintf(&curr_fifo, "%s.%03d", tmpstub, fifo_number);
605    asprintf(&next_fifo, "%s.%03d", tmpstub, fifo_number + 1);
606    mkfifo(curr_fifo, S_IRWXU | S_IRWXG);
607    mkfifo(next_fifo, S_IRWXU | S_IRWXG);   // make sure _next_ fifo already exists before we call partimage
608    asprintf(&sz_call_to_partimage,
609            "%c%cpartimagehack " PARTIMAGE_PARAMS
610            " save %s %s > /tmp/stdout 2> /tmp/stderr", 2, 0, input_device,
611            tmpstub);
612    log_msg(5, "curr_fifo   = %s", curr_fifo);
613    log_msg(5, "next_fifo   = %s", next_fifo);
614    log_msg(5, "sz_call_to_partimage call is '%s'",
615            sz_call_to_partimage + 2);
616    if (!lstat(output_fname, &buf) && S_ISREG(buf.st_mode)) {
617        log_msg(5, "Deleting %s", output_fname);
618        unlink(output_fname);
619    }
620    if (!(fout = fopen(output_fname, "w"))) {
621        fatal_error("Unable to openout to output_fname");
622    }
623    res =
624        pthread_create(&partimage_thread, NULL, call_partimage_in_bkgd,
625                       (void *) sz_call_to_partimage);
626    if (res) {
627        fatal_error("Failed to create thread to call partimage");
628    }
629    log_msg(1, "Running fore/back at same time");
630    log_to_screen("Working with partimagehack...");
631    while (sz_call_to_partimage[0] > 0) {
632        asprintf(&tmp, "%s\n", NEXT_SUBVOL_PLEASE);
633        if (fwrite(tmp, 1, 128, fout) != 128) {
634            fatal_error("Cannot write interim block");
635        }
636        paranoid_free(tmp);
637
638        log_msg(5, "fifo_number=%d", fifo_number);
639        log_msg(4, "Cat'ting %s", curr_fifo);
640        if (!(fin = fopen(curr_fifo, "r"))) {
641            fatal_error("Unable to openin from fifo");
642        }
643        if (prev_fifo !=  NULL) {
644            log_msg(5, "Deleting %s", prev_fifo);
645            unlink(prev_fifo);      // just in case
646            paranoid_free(prev_fifo);
647        }
648        copy_from_src_to_dest(fin, fout, 'w');
649        paranoid_fclose(fin);
650        fifo_number++;
651
652        prev_fifo = curr_fifo;
653        curr_fifo = next_fifo;
654        log_msg(5, "Creating %s", next_fifo);
655        asprintf(&next_fifo, "%s.%03d", tmpstub, fifo_number + 1);
656        mkfifo(next_fifo, S_IRWXU | S_IRWXG);   // make sure _next_ fifo exists before we cat this one
657        system("sync");
658        sleep(5);
659    }
660    asprintf(&tmp, "%s\n", NO_MORE_SUBVOLS);
661    if (fwrite(tmp, 1, 128, fout) != 128) {
662        fatal_error("Cannot write interim block");
663    }
664    if (fwrite(tmp, 1, 128, fout) != 128) {
665        fatal_error("Cannot write interim block");
666    }
667    if (fwrite(tmp, 1, 128, fout) != 128) {
668        fatal_error("Cannot write interim block");
669    }
670    if (fwrite(tmp, 1, 128, fout) != 128) {
671        fatal_error("Cannot write interim block");
672    }
673    paranoid_free(tmp);
674    paranoid_fclose(fout);
675    log_to_screen("Cleaning up after partimagehack...");
676    log_msg(3, "Final fifo_number=%d", fifo_number);
677    paranoid_system("sync");
678    unlink(next_fifo);
679    paranoid_free(next_fifo);
680
681    unlink(curr_fifo);
682    paranoid_free(curr_fifo);
683
684    unlink(prev_fifo);
685    paranoid_free(prev_fifo);
686
687    log_to_screen("Finished cleaning up.");
688
689//  if (!lstat(sz_wait_for_this_file, &statbuf))
690//    { log_msg(3, "WARNING! %s was not processed.", sz_wait_for_this_file); }
691    log_msg(2, "Waiting for pthread_join() to join.");
692    pthread_join(partimage_thread, NULL);
693    res = sz_call_to_partimage[1];
694    paranoid_free(sz_call_to_partimage);
695    log_msg(2, "pthread_join() joined OK.");
696    log_msg(1, "Partimagehack(save) returned %d", res);
697    unlink(tmpstub);
698    paranoid_free(tmpstub);
699
700    return (res);
701}
702
703
704/**
705 * Feed @p input_device through partimage to @p output_fname.
706 * @param input_device The device to image.
707 * @param output_fname The file to write.
708 * @return 0 for success, nonzero for failure.
709 */
710int feed_into_partimage(char *input_device, char *output_fname)
711{
712// BACKUP
713    int res;
714
715    if (!does_file_exist(input_device)) {
716        fatal_error("input device does not exist");
717    }
718    if (!find_home_of_exe("partimagehack")) {
719        fatal_error("partimagehack not found");
720    }
721    res =
722        dynamically_create_pipes_and_copy_from_them_to_output_file
723        (input_device, output_fname);
724    return (res);
725}
726
727
728int run_external_binary_with_percentage_indicator_OLD(char *tt, char *cmd)
729{
730
731    /*@ int *************************************************************** */
732    int res = 0;
733    int percentage = 0;
734    int maxpc = 0;
735    int pcno = 0;
736    int last_pcno = 0;
737
738    /*@ buffers *********************************************************** */
739    char *command;
740    char *tempfile;
741    /*@ pointers ********************************************************** */
742    FILE *pin;
743
744    assert_string_is_neither_NULL_nor_zerolength(cmd);
745
746    asprintf(&tempfile,
747           call_program_and_get_last_line_of_output
748           ("mktemp -q /tmp/mondo.XXXXXXXX"));
749    asprintf(&command, "%s >> %s 2>> %s; rm -f %s", cmd, tempfile, tempfile,
750            tempfile);
751    log_msg(3, command);
752    open_evalcall_form(tt);
753
754    if (!(pin = popen(command, "r"))) {
755        log_OS_error("fmt err");
756        paranoid_free(command);
757        paranoid_free(tempfile);
758        return (1);
759    }
760    paranoid_free(command);
761
762    maxpc = 100;
763// OLD OLD OLD OLD OLD OLD OLD OLD OLD OLD OLD OLD
764    for (sleep(1); does_file_exist(tempfile); sleep(1)) {
765        pcno = grab_percentage_from_last_line_of_file(MONDO_LOGFILE);
766        if (pcno < 0 || pcno > 100) {
767            log_msg(5, "Weird pc#");
768            continue;
769        }
770        percentage = pcno * 100 / maxpc;
771        if (pcno <= 5 && last_pcno > 40) {
772            close_evalcall_form();
773            open_evalcall_form("Verifying...");
774        }
775        last_pcno = pcno;
776        update_evalcall_form(percentage);
777    }
778// OLD OLD OLD OLD OLD OLD OLD OLD OLD OLD OLD OLD
779    close_evalcall_form();
780    if (pclose(pin)) {
781        res++;
782        log_OS_error("Unable to pclose");
783    }
784    unlink(tempfile);
785    paranoid_free(tempfile);
786    return (res);
787}
788
789
790void *run_prog_in_bkgd_then_exit(void *info)
791{
792    char *sz_command;
793    static int res = 4444;
794
795    res = 999;
796    sz_command = (char *) info;
797    log_msg(4, "sz_command = '%s'", sz_command);
798    res = system(sz_command);
799    if (res > 256 && res != 4444) {
800        res = res / 256;
801    }
802    log_msg(4, "child res = %d", res);
803    sz_command[0] = '\0';
804    pthread_exit((void *) (&res));
805}
806
807
808int run_external_binary_with_percentage_indicator_NEW(char *tt, char *cmd)
809{
810
811    /*@ int *************************************************************** */
812    int res = 0;
813    int percentage = 0;
814    int maxpc = 100;
815    int pcno = 0;
816    int last_pcno = 0;
817    int counter = 0;
818
819    /*@ buffers *********************************************************** */
820    char *command;
821    /*@ pointers ********************************************************** */
822    static int chldres = 0;
823    int *pchild_result;
824    pthread_t childthread;
825
826    pchild_result = &chldres;
827    assert_string_is_neither_NULL_nor_zerolength(cmd);
828    assert_string_is_neither_NULL_nor_zerolength(tt);
829    *pchild_result = 999;
830
831    asprintf(&command, "%s 2>> %s", cmd, MONDO_LOGFILE);
832    log_msg(3, "command = '%s'", command);
833    if ((res =
834         pthread_create(&childthread, NULL, run_prog_in_bkgd_then_exit,
835                        (void *) command))) {
836        fatal_error("Unable to create an archival thread");
837    }
838
839    log_msg(8, "Parent running");
840    open_evalcall_form(tt);
841    for (sleep(1); command[0] != '\0'; sleep(1)) {
842        pcno = grab_percentage_from_last_line_of_file(MONDO_LOGFILE);
843        if (pcno <= 0 || pcno > 100) {
844            log_msg(8, "Weird pc#");
845            continue;
846        }
847        percentage = pcno * 100 / maxpc;
848        if (pcno <= 5 && last_pcno >= 40) {
849            close_evalcall_form();
850            open_evalcall_form("Verifying...");
851        }
852        if (counter++ >= 5) {
853            counter = 0;
854            log_file_end_to_screen(MONDO_LOGFILE, "");
855        }
856        last_pcno = pcno;
857        update_evalcall_form(percentage);
858    }
859    paranoid_free(command);
860
861    log_file_end_to_screen(MONDO_LOGFILE, "");
862    close_evalcall_form();
863    pthread_join(childthread, (void *) (&pchild_result));
864    if (pchild_result) {
865        res = *pchild_result;
866    } else {
867        res = 666;
868    }
869    log_msg(3, "Parent res = %d", res);
870    return (res);
871}
872
873#define PIH_LOG "/var/log/partimage-debug.log"
874
875/**
876 * Feed @p input_fifo through partimage (restore) to @p output_device.
877 * @param input_fifo The partimage file to read.
878 * @param output_device Where to put the output.
879 * @return The return value of partimagehack (0 for success).
880 * @bug Probably unnecessary, as the partimage is just a sparse file. We could use @c dd to restore it.
881 */
882int feed_outfrom_partimage(char *output_device, char *input_fifo)
883{
884// RESTORE
885    char *tmp;
886    char *stuff;
887    char *sz_call_to_partimage;
888    pthread_t partimage_thread;
889    int res;
890    char *curr_fifo;
891    char *prev_fifo = NULL;
892    char *oldest_fifo = NULL;
893    char *next_fifo;
894    char *afternxt_fifo;
895    int fifo_number = 0;
896    char *tmpstub;
897    FILE *fin;
898    FILE *fout;
899
900    log_msg(1, "output_device=%s", output_device);
901    log_msg(1, "input_fifo=%s", input_fifo);
902    asprintf(&tmpstub, "/tmp");
903
904    log_msg(1, "tmpstub was %s", tmpstub);
905    asprintf(&stuff, tmpstub);
906    paranoid_free(tmpstub);
907
908    asprintf(&tmpstub, "%s/pih-fifo-%ld", stuff, (long int) random());
909    paranoid_free(stuff);
910
911    log_msg(1, "tmpstub is now %s", tmpstub);
912    unlink("/tmp/PARTIMAGEHACK-POSITION");
913    unlink(PAUSE_PARTIMAGE_FNAME);
914    paranoid_system("rm -f /tmp/*PARTIMAGE*");
915    asprintf(&curr_fifo, "%s.%03d", tmpstub, fifo_number);
916    asprintf(&next_fifo, "%s.%03d", tmpstub, fifo_number + 1);
917    asprintf(&afternxt_fifo, "%s.%03d", tmpstub, fifo_number + 2);
918    mkfifo(PIH_LOG, S_IRWXU | S_IRWXG);
919    mkfifo(curr_fifo, S_IRWXU | S_IRWXG);
920    mkfifo(next_fifo, S_IRWXU | S_IRWXG);   // make sure _next_ fifo already exists before we call partimage
921    mkfifo(afternxt_fifo, S_IRWXU | S_IRWXG);
922    system("cat " PIH_LOG " > /dev/null &");
923    log_msg(3, "curr_fifo   = %s", curr_fifo);
924    log_msg(3, "next_fifo   = %s", next_fifo);
925    if (!does_file_exist(input_fifo)) {
926        fatal_error("input fifo does not exist");
927    }
928    if (!(fin = fopen(input_fifo, "r"))) {
929        fatal_error("Unable to openin from input_fifo");
930    }
931    if (!find_home_of_exe("partimagehack")) {
932        fatal_error("partimagehack not found");
933    }
934    asprintf(&sz_call_to_partimage,
935            "%c%cpartimagehack " PARTIMAGE_PARAMS
936            " restore %s %s > /dev/null 2>> %s", 2, 0, output_device, curr_fifo,
937            MONDO_LOGFILE);
938    log_msg(1, "output_device = %s", output_device);
939    log_msg(1, "curr_fifo = %s", curr_fifo);
940    log_msg(1, "sz_call_to_partimage+2 = %s", sz_call_to_partimage + 2);
941    res =
942        pthread_create(&partimage_thread, NULL, call_partimage_in_bkgd,
943                       (void *) sz_call_to_partimage);
944    if (res) {
945        fatal_error("Failed to create thread to call partimage");
946    }
947    log_msg(1, "Running fore/back at same time");
948    log_msg(2, " Trying to openin %s", input_fifo);
949    if (!does_file_exist(input_fifo)) {
950        log_msg(2, "Warning - %s does not exist", input_fifo);
951    }
952    while (!does_file_exist("/tmp/PARTIMAGEHACK-POSITION")) {
953        log_msg(6, "Waiting for partimagehack (restore) to start");
954        sleep(1);
955    }
956
957    if (!(tmp = malloc(128))) {
958        fatal_error("Failed to malloc() tmp");
959    }
960    while (sz_call_to_partimage[0] > 0) {
961        if (fread(tmp, 1, 128, fin) != 128) {
962            fatal_error("Cannot read introductory block");
963        }
964        if (strstr(tmp, NEXT_SUBVOL_PLEASE)) {
965            log_msg(2, "Great. Next subvol coming up.");
966        } else if (strstr(tmp, NO_MORE_SUBVOLS)) {
967            log_msg(2, "Great. That was the last subvol.");
968            break;
969        } else {
970            log_msg(2, "WTF is this? '%s'", tmp);
971            fatal_error("Unknown interim block");
972        }
973        if (feof(fin)) {
974            log_msg(1, "Eof(fin) detected. Breaking.");
975            break;
976        }
977        log_msg(3, "Processing subvol %d", fifo_number);
978        log_msg(5, "fifo_number=%d", fifo_number);
979        if (!(fout = fopen(curr_fifo, "w"))) {
980            fatal_error("Cannot openout to curr_fifo");
981        }
982        copy_from_src_to_dest(fout, fin, 'r');
983        paranoid_fclose(fout);
984        fifo_number++;
985        if (oldest_fifo != NULL) {
986            log_msg(6, "Deleting %s", oldest_fifo);
987            unlink(oldest_fifo);    // just in case
988            paranoid_free(oldest_fifo);
989        }
990        oldest_fifo = prev_fifo;
991        prev_fifo = curr_fifo;
992        curr_fifo = next_fifo;
993        next_fifo = afternxt_fifo;
994        asprintf(&afternxt_fifo, "%s.%03d", tmpstub, fifo_number + 2);
995        log_msg(6, "Creating %s", afternxt_fifo);
996        mkfifo(afternxt_fifo, S_IRWXU | S_IRWXG);   // make sure _next_ fifo already exists before we access current fifo
997        fflush(fin);
998//      system("sync");
999        usleep(1000L * 100L);
1000    }
1001    paranoid_free(tmp);
1002    paranoid_free(tmpstub);
1003
1004    paranoid_fclose(fin);
1005    paranoid_system("sync");
1006    log_msg(1, "Partimagehack has finished. Great. Fin-closing.");
1007    log_msg(1, "Waiting for pthread_join");
1008    pthread_join(partimage_thread, NULL);
1009    res = sz_call_to_partimage[1];
1010    paranoid_free(sz_call_to_partimage);
1011
1012    log_msg(1, "Yay. Partimagehack (restore) returned %d", res);
1013    unlink(prev_fifo);
1014    paranoid_free(prev_fifo);
1015
1016    unlink(curr_fifo);
1017    paranoid_free(curr_fifo);
1018
1019    unlink(next_fifo);
1020    paranoid_free(next_fifo);
1021
1022    unlink(afternxt_fifo);
1023    paranoid_free(afternxt_fifo);
1024
1025    unlink(PIH_LOG);
1026    /* BERLIOS : pas de unlink(oldest_fifo) ??? */
1027    paranoid_free(oldest_fifo);
1028
1029    return (res);
1030}
Note: See TracBrowser for help on using the repository browser.