source: MondoRescue/trunk/mondo/src/common/libmondo-fifo.c @ 932

Last change on this file since 932 was 932, checked in by Bruno Cornec, 14 years ago

merge -r913:931 $SVN_M/branches/stable

  • Property svn:keywords set to Id
File size: 4.5 KB
RevLine 
[42]1/* $Id: libmondo-fifo.c 932 2006-11-15 11:40:15Z bruno $ */
[1]2
3/**
4 * @file
5 * Functions to handle buffering of tape archives as they are read/written.
6 * This used the external program @c buffer mostly.
7 */
8
9#include <unistd.h>
10#include <stdio.h>
[142]11#ifndef S_SPLINT_S
[1]12#include <signal.h>
[142]13#endif
[42]14#include <fcntl.h>
15
[1]16#include <errno.h>
17#include <sys/types.h>
18#include <sys/stat.h>
19#include <sys/ipc.h>
20#include <sys/shm.h>
21#include <sys/sem.h>
22#include <sys/wait.h>
[142]23#ifndef S_SPLINT_S
[1]24#include <pthread.h>
[142]25#endif
[1]26
27#include "my-stuff.h"
28#include "mondostructures.h"
29#include "libmondo.h"
[900]30#include "mr_mem.h"
[1]31
32/**
33 * @addtogroup globalGroup
34 * @{
35 */
36/**
37 * The SIGPIPE handler sets this to TRUE.
38 */
[42]39bool g_sigpipe = FALSE;
[1]40
41/**
42 * PID of the "main" process.
43 */
[42]44pid_t g_mastermind_pid = 0;
[1]45
46/**
47 * Command line with which @c buffer was invoked.
48 */
[42]49char *g_sz_call_to_buffer;
[1]50
51/**
52 * Size of the buffer used with @c buffer.
53 */
[42]54int g_tape_buffer_size_MB = 0;
[1]55
56/* @} - end of globalGroup */
57
[794]58extern char *ps_options;
[932]59extern char *ps_proc_id;
[1]60
61/**
62 * @addtogroup fifoGroup
63 * @{
64 */
65/**
66 * Open a pipe to/from @c buffer.
67 * If buffer does not work at all, we use `dd'.
68 * @param device The device to read from/write to.
69 * @param direction @c 'r' (reading) or @c 'w' (writing).
70 * @return A file pointer to/from the @c buffer process.
71 */
[42]72FILE *open_device_via_buffer(char *device, char direction,
73                             long internal_tape_block_size)
[1]74{
[42]75    char *sz_dir;
76    char keych;
77    char *tmp;
78    FILE *fres;
79    int bufsize;                // in megabytes
80    int res;
81    int wise_upper_limit;
82    int wise_lower_limit;
[1]83
[42]84    assert_string_is_neither_NULL_nor_zerolength(device);
85    assert(direction == 'w' || direction == 'r');
[900]86    mr_asprintf(&sz_dir, "%c", direction);
[42]87    wise_upper_limit = (am_I_in_disaster_recovery_mode()? 8 : 32);
88    wise_lower_limit = 1;       // wise_upper_limit/2 + 1;
[688]89    sync();
[42]90    for (bufsize = wise_upper_limit, res = -1;
91         res != 0 && bufsize >= wise_lower_limit; bufsize--) {
[900]92        mr_asprintf(&tmp,
[42]93                 "dd if=/dev/zero bs=1024 count=16k 2> /dev/null | buffer -o /dev/null -s %ld -m %d%c",
94                 internal_tape_block_size, bufsize, 'm');
95        res = run_program_and_log_output(tmp, 2);
[900]96        mr_free(tmp);
[42]97    }
98    if (!res) {
99        bufsize++;
[900]100        mr_asprintf(&tmp, _("Negotiated max buffer of %d MB "), bufsize);
[42]101        log_to_screen(tmp);
[900]102        mr_free(tmp);
[42]103    } else {
104        bufsize = 0;
105        res = 0;
106        log_to_screen
[507]107            (_("Cannot negotiate a buffer of ANY size. Using dd instead."));
[42]108    }
109    if (direction == 'r') {
110        keych = 'i';
111    } else {
112        keych = 'o';
113    }
114    if (bufsize) {
[900]115        mr_asprintf(&g_sz_call_to_buffer,
[42]116                 "buffer -m %d%c -p%d -B -s%ld -%c %s 2>> %s", bufsize,
117                 'm', (direction == 'r') ? 20 : 75,
118                 internal_tape_block_size, keych, device, MONDO_LOGFILE);
119    } else {
[900]120        mr_asprintf(&g_sz_call_to_buffer, "dd bs=%ld %cf=%s",
[42]121                 internal_tape_block_size, keych, device);
122    }
123    log_msg(2, "Calling buffer --- command = '%s'", g_sz_call_to_buffer);
124    fres = popen(g_sz_call_to_buffer, sz_dir);
[900]125    mr_free(sz_dir);
[42]126    if (fres) {
127        log_msg(2, "Successfully opened ('%c') tape device %s", direction,
128                device);
129    } else {
130        log_msg(2, "Failed to open ('%c') tape device %s", direction,
131                device);
132    }
133    sleep(2);
[900]134    mr_asprintf(&tmp, "ps %s | grep \"%s\"", ps_options, g_sz_call_to_buffer);
[42]135    if (run_program_and_log_output(tmp, 2)) {
136        log_msg(2, "Warning - I think I failed to open tape, actually.");
137    }
[900]138    mr_free(tmp);
[42]139    g_tape_buffer_size_MB = bufsize;
[900]140    mr_asprintf(&tmp, "ps %s | grep buffer | grep -v grep", ps_options);
[42]141    if (run_program_and_log_output(tmp, 1)) {
142        fres = NULL;
[507]143        log_to_screen(_("Failed to open tape streamer. Buffer error."));
[42]144    } else {
[507]145        log_to_screen(_("Buffer successfully started."));
[42]146    }
[900]147    mr_free(tmp);
[42]148    return (fres);
[1]149}
150
151
152/**
153 * Kill @c buffer processes.
[932]154 * Only called in mondoarchive
[1]155 */
156void kill_buffer()
157{
[42]158    char *tmp;
159    char *command;
[1]160
[688]161    sync();
[900]162    mr_asprintf(&command,
[932]163            "ps %s | grep -F \"%s\" | grep -Fv grep | awk '{print $2;}' | grep -v PID | head -1", ps_options,
[687]164            g_sz_call_to_buffer);
[900]165    mr_free(g_sz_call_to_buffer);
[42]166    log_msg(2, "kill_buffer() --- command = %s", command);
[1]167
[688]168    tmp = call_program_and_get_last_line_of_output(command);
[900]169    mr_free(command);
[1]170
[900]171    mr_asprintf(&command, "kill %s", tmp);
[42]172    log_msg(2, "kill_buffer() --- command = %s", command);
[1]173
[42]174    if (strlen(tmp) > 0) {
175        run_program_and_log_output(command, TRUE);
176    }
[900]177    mr_free(tmp);
178    mr_free(command);
[42]179}
[1]180
181
182/**
183 * Handler for SIGPIPE.
184 * @param sig The signal that occurred (hopefully SIGPIPE).
185 */
186void sigpipe_occurred(int sig)
187{
[42]188    g_sigpipe = TRUE;
[1]189}
190
191/* @} - end of fifoGroup */
192
[42]193/* BERLIOS: useless ?
[1]194int
195extract_config_file_from_ramdisk( struct s_bkpinfo *bkpinfo,
196                  char *ramdisk_fname,
197                  char *output_cfg_file,
198                  char *output_mountlist_file);
[42]199*/
Note: See TracBrowser for help on using the repository browser.