static void
child_watch_callback(
pid_t pid,
gint status,
gpointer data)
{
XferFilterProcess *self = XFER_FILTER_PROCESS(data);
XferElement *elt = (XferElement *)self;
XMsg *msg;
char *errmsg = NULL;
g_assert(pid == self->child_pid);
self->child_pid = -1; /* it's gone now.. */
if (WIFEXITED(status)) {
int exitcode = WEXITSTATUS(status);
g_debug("%s: process exited with status %d", xfer_element_repr(elt), exitcode);
if (exitcode != 0) {
errmsg = g_strdup_printf("%s exited with status %d",
self->argv[0], exitcode);
}
} else if (WIFSIGNALED(status)) {
int signal = WTERMSIG(status);
if (signal != SIGKILL || !self->child_killed) {
errmsg = g_strdup_printf("%s died on signal %d", self->argv[0], signal);
g_debug("%s: %s", xfer_element_repr(elt), errmsg);
}
}
if (errmsg) {
msg = xmsg_new(XFER_ELEMENT(self), XMSG_INFO, 0);
msg->message = g_strdup("ERROR");
xfer_queue_message(XFER_ELEMENT(self)->xfer, msg);
} else {
msg = xmsg_new(XFER_ELEMENT(self), XMSG_INFO, 0);
msg->message = g_strdup("SUCCESS");
xfer_queue_message(XFER_ELEMENT(self)->xfer, msg);
}
/* if this is an error exit, send an XMSG_ERROR and cancel */
if (!elt->cancelled) {
if (errmsg) {
msg = xmsg_new(XFER_ELEMENT(self), XMSG_ERROR, 0);
msg->message = errmsg;
xfer_queue_message(XFER_ELEMENT(self)->xfer, msg);
xfer_cancel(elt->xfer);
} else if (elt->cancel_on_success) {
xfer_cancel(elt->xfer);
}
}
/* this element is as good as cancelled already, so fall through to XMSG_DONE */
xfer_queue_message(XFER_ELEMENT(self)->xfer, xmsg_new(XFER_ELEMENT(self), XMSG_DONE, 0));
}
static void
send_xfer_done(
XferElementGlue *self)
{
xfer_queue_message(XFER_ELEMENT(self)->xfer,
xmsg_new((XferElement *)self, XMSG_DONE, 0));
}
static gpointer
pull_buffer_impl(
XferElement *elt,
size_t *size)
{
XferFilterCrc *self = (XferFilterCrc *)elt;
char *buf;
XMsg *msg;
if (elt->cancelled) {
/* drain our upstream only if we're expecting an EOF */
if (elt->expect_eof) {
xfer_element_drain_buffers(XFER_ELEMENT(self)->upstream);
}
/* return an EOF */
*size = 0;
return NULL;
}
/* get a buffer from upstream, crc it, and hand it back */
buf = xfer_element_pull_buffer(XFER_ELEMENT(self)->upstream, size);
if (buf) {
crc32_add((uint8_t *)buf, *size, &elt->crc);
} else {
g_debug("sending XMSG_CRC message");
g_debug("crc pull_buffer CRC: %08x",
crc32_finish(&elt->crc));
msg = xmsg_new(elt, XMSG_CRC, 0);
msg->crc = crc32_finish(&elt->crc);
msg->size = elt->crc.size;
xfer_queue_message(elt->xfer, msg);
}
return buf;
}
static void
push_buffer_impl(
XferElement *elt,
gpointer buf,
size_t len)
{
XferFilterCrc *self = (XferFilterCrc *)elt;
XMsg *msg;
/* drop the buffer if we've been cancelled */
if (elt->cancelled) {
amfree(buf);
return;
}
/* crc the given buffer and pass it downstream */
if (buf) {
crc32_add((uint8_t *)buf, len, &elt->crc);
} else {
g_debug("sending XMSG_CRC message to %p", elt);
g_debug("crc push_buffer CRC: %08x",
crc32_finish(&elt->crc));
msg = xmsg_new(elt, XMSG_CRC, 0);
msg->crc = crc32_finish(&elt->crc);
msg->size = elt->crc.size;
xfer_queue_message(elt->xfer, msg);
}
xfer_element_push_buffer(XFER_ELEMENT(self)->downstream, buf, len);
}
static void
pull_and_write(XferElementGlue *self)
{
XferElement *elt = XFER_ELEMENT(self);
int fd = get_write_fd(self);
XMsg *msg;
size_t written;
self->write_fdp = NULL;
while (!elt->cancelled) {
size_t len;
char *buf;
/* get a buffer from upstream */
buf = xfer_element_pull_buffer(elt->upstream, &len);
if (!buf)
break;
/* write it */
if (!elt->downstream->drain_mode) {
written = full_write(fd, buf, len);
if (written < len) {
if (elt->downstream->must_drain) {
g_debug("Error writing to fd %d: %s", fd, strerror(errno));
} else if (elt->downstream->ignore_broken_pipe && errno == EPIPE) {
} else {
if (!elt->cancelled) {
xfer_cancel_with_error(elt,
_("Error writing to fd %d: %s"), fd, strerror(errno));
xfer_cancel(elt->xfer);
wait_until_xfer_cancelled(elt->xfer);
}
amfree(buf);
break;
}
elt->downstream->drain_mode = TRUE;
}
}
crc32_add((uint8_t *)buf, len, &elt->crc);
amfree(buf);
}
if (elt->cancelled && elt->expect_eof)
xfer_element_drain_buffers(elt->upstream);
g_debug("sending XMSG_CRC message %p", elt->downstream);
g_debug("pull_and_write CRC: %08x size %lld",
crc32_finish(&elt->crc), (long long)elt->crc.size);
msg = xmsg_new(elt->downstream, XMSG_CRC, 0);
msg->crc = crc32_finish(&elt->crc);
msg->size = elt->crc.size;
xfer_queue_message(elt->xfer, msg);
/* close the fd we've been writing, as an EOF signal to downstream, and
* set it to -1 to avoid accidental re-use */
close_write_fd(self);
}
static void
read_and_push(
XferElementGlue *self)
{
XferElement *elt = XFER_ELEMENT(self);
int fd = get_read_fd(self);
XMsg *msg;
crc32_init(&elt->crc);
while (!elt->cancelled) {
char *buf = g_malloc(GLUE_BUFFER_SIZE);
gsize len;
int read_error;
/* read a buffer from upstream */
len = read_fully(fd, buf, GLUE_BUFFER_SIZE, &read_error);
if (len < GLUE_BUFFER_SIZE) {
if (read_error) {
if (!elt->cancelled) {
xfer_cancel_with_error(elt,
_("Error reading from fd %d: %s"), fd, strerror(read_error));
g_debug("element-glue: error reading from fd %d: %s",
fd, strerror(read_error));
wait_until_xfer_cancelled(elt->xfer);
}
amfree(buf);
break;
} else if (len == 0) { /* we only count a zero-length read as EOF */
amfree(buf);
break;
}
}
crc32_add((uint8_t *)buf, len, &elt->crc);
xfer_element_push_buffer(elt->downstream, buf, len);
}
if (elt->cancelled && elt->expect_eof)
xfer_element_drain_fd(fd);
/* send an EOF indication downstream */
xfer_element_push_buffer(elt->downstream, NULL, 0);
/* close the read fd, since it's at EOF */
close_read_fd(self);
g_debug("sending XMSG_CRC message");
g_debug("read_and_push CRC: %08x size %lld",
crc32_finish(&elt->crc), (long long)elt->crc.size);
msg = xmsg_new(elt->upstream, XMSG_CRC, 0);
msg->crc = crc32_finish(&elt->crc);
msg->size = elt->crc.size;
xfer_queue_message(elt->xfer, msg);
}
static void
push_buffer_impl(
XferElement *elt,
gpointer buf,
size_t len)
{
XferDestNull *self = (XferDestNull *)elt;
if (buf) {
crc32_add(buf, len, &elt->crc);
} else {
XMsg *msg = xmsg_new((XferElement *)self, XMSG_CRC, 0);
msg->crc = crc32_finish(&elt->crc);
msg->size = elt->crc.size;
xfer_queue_message(XFER_ELEMENT(self)->xfer, msg);
return;
}
if (self->do_verify && !elt->cancelled) {
if (!simpleprng_verify_buffer(&self->prng, buf, len)) {
xfer_cancel_with_error(elt,
"verification of incoming bytestream failed; see stderr for details"),
wait_until_xfer_cancelled(elt->xfer);
amfree(buf);
return;
}
}
self->byte_position += len;
if (!self->sent_info) {
/* send a superfluous message (this is a testing XferElement,
* after all) */
XMsg *msg = xmsg_new((XferElement *)self, XMSG_INFO, 0);
msg->message = g_strdup("Is this thing on?");
xfer_queue_message(XFER_ELEMENT(self)->xfer, msg);
self->sent_info = TRUE;
}
amfree(buf);
}
static gpointer
directtcp_listen_thread(
gpointer data)
{
XferSourceRecovery *self = XFER_SOURCE_RECOVERY(data);
XferElement *elt = XFER_ELEMENT(self);
int result;
DBG(1, "(this is directtcp_listen_thread)");
/* we need to make an outgoing connection to downstream; we do this while
* holding the start_part_mutex, so that a part doesn't get started until
* we're finished with the device */
g_mutex_lock(self->start_part_mutex);
if (elt->cancelled) {
g_mutex_unlock(self->start_part_mutex);
goto send_done;
}
g_assert(self->device != NULL); /* have a device */
g_assert(elt->downstream->input_listen_addrs != NULL); /* downstream listening */
DBG(2, "making DirectTCP connection on device %s", self->device->device_name);
result = device_connect(self->device, FALSE,
elt->downstream->input_listen_addrs,
&self->conn, &elt->cancelled,
self->start_part_mutex, self->abort_cond);
if (result == 1 && !elt->cancelled) {
xfer_cancel_with_error(elt,
_("error making DirectTCP connection: %s"),
device_error_or_status(self->device));
g_mutex_unlock(self->start_part_mutex);
wait_until_xfer_cancelled(elt->xfer);
goto send_done;
} else if (result == 2 || elt->cancelled) {
g_mutex_unlock(self->start_part_mutex);
wait_until_xfer_cancelled(elt->xfer);
goto send_done;
}
DBG(2, "DirectTCP connect succeeded");
return directtcp_common_thread(self);
send_done:
xfer_queue_message(elt->xfer, xmsg_new(elt, XMSG_DONE, 0));
return NULL;
}
static gboolean
start_impl(
XferElement *elt)
{
XferSourceRecovery *self = XFER_SOURCE_RECOVERY(elt);
if (elt->output_mech == XFER_MECH_DIRECTTCP_CONNECT) {
g_assert(elt->output_listen_addrs != NULL);
self->thread = g_thread_create(directtcp_connect_thread, (gpointer)self, FALSE, NULL);
return TRUE; /* we'll send XMSG_DONE */
} else if (elt->output_mech == XFER_MECH_DIRECTTCP_LISTEN) {
g_assert(elt->output_listen_addrs == NULL);
self->thread = g_thread_create(directtcp_listen_thread, (gpointer)self, FALSE, NULL);
return TRUE; /* we'll send XMSG_DONE */
} else {
/* nothing to prepare for - we're ready already! */
DBG(2, "not using DirectTCP: sending XMSG_READY immediately");
xfer_queue_message(elt->xfer, xmsg_new(elt, XMSG_READY, 0));
return FALSE; /* we won't send XMSG_DONE */
}
}
static void
read_and_write(XferElementGlue *self)
{
XferElement *elt = XFER_ELEMENT(self);
/* dynamically allocate a buffer, in case this thread has
* a limited amount of stack allocated */
char *buf = g_malloc(GLUE_BUFFER_SIZE);
int rfd = get_read_fd(self);
int wfd = get_write_fd(self);
XMsg *msg;
crc32_init(&elt->crc);
g_debug("read_and_write: read from %d, write to %d", rfd, wfd);
while (!elt->cancelled) {
size_t len;
/* read from upstream */
len = read_fully(rfd, buf, GLUE_BUFFER_SIZE, NULL);
if (len < GLUE_BUFFER_SIZE) {
if (errno) {
if (!elt->cancelled) {
xfer_cancel_with_error(elt,
_("Error reading from fd %d: %s"), rfd, strerror(errno));
wait_until_xfer_cancelled(elt->xfer);
}
break;
} else if (len == 0) { /* we only count a zero-length read as EOF */
break;
}
}
/* write the buffer fully */
if (!elt->downstream->drain_mode && full_write(wfd, buf, len) < len) {
if (elt->downstream->must_drain) {
g_debug("Could not write to fd %d: %s", wfd, strerror(errno));
} else if (elt->downstream->ignore_broken_pipe && errno == EPIPE) {
} else {
if (!elt->cancelled) {
xfer_cancel_with_error(elt,
_("Could not write to fd %d: %s"),
wfd, strerror(errno));
wait_until_xfer_cancelled(elt->xfer);
}
break;
}
}
crc32_add((uint8_t *)buf, len, &elt->crc);
}
if (elt->cancelled && elt->expect_eof)
xfer_element_drain_fd(rfd);
/* close the read fd. If it's not at EOF, then upstream will get EPIPE, which will hopefully
* kill it and complete the cancellation */
close_read_fd(self);
/* close the fd we've been writing, as an EOF signal to downstream */
close_write_fd(self);
g_debug("read_and_write upstream CRC: %08x size %lld",
crc32_finish(&elt->crc), (long long)elt->crc.size);
g_debug("sending XMSG_CRC message");
msg = xmsg_new(elt->upstream, XMSG_CRC, 0);
msg->crc = crc32_finish(&elt->crc);
msg->size = elt->crc.size;
xfer_queue_message(elt->xfer, msg);
g_debug("read_and_write downstream CRC: %08x size %lld",
crc32_finish(&elt->crc), (long long)elt->crc.size);
g_debug("sending XMSG_CRC message");
msg = xmsg_new(elt->downstream, XMSG_CRC, 0);
msg->crc = crc32_finish(&elt->crc);
msg->size = elt->crc.size;
xfer_queue_message(elt->xfer, msg);
amfree(buf);
}
static void
push_buffer_impl(
XferElement *elt,
gpointer buf,
size_t len)
{
XferElementGlue *self = (XferElementGlue *)elt;
XMsg *msg;
/* accept first, if required */
if (self->on_push & PUSH_ACCEPT_FIRST) {
/* don't accept the next time around */
self->on_push &= ~PUSH_ACCEPT_FIRST;
if (elt->cancelled) {
return;
}
if ((self->output_data_socket = do_directtcp_accept(self,
&self->output_listen_socket)) == -1) {
/* do_directtcp_accept already signalled an error; xfer
* is cancelled */
return;
}
/* write to this new socket */
self->write_fdp = &self->output_data_socket;
}
/* or connect first, if required */
if (self->on_push & PUSH_CONNECT_FIRST) {
/* don't accept the next time around */
self->on_push &= ~PUSH_CONNECT_FIRST;
if (elt->cancelled) {
return;
}
if ((self->output_data_socket = do_directtcp_connect(self,
elt->downstream->input_listen_addrs)) == -1) {
/* do_directtcp_connect already signalled an error; xfer
* is cancelled */
return;
}
/* read from this new socket */
self->write_fdp = &self->output_data_socket;
}
switch (self->on_push) {
case PUSH_TO_RING_BUFFER:
/* just drop packets if the transfer has been cancelled */
if (elt->cancelled) {
amfree(buf);
return;
}
/* make sure there's at least one element free */
amsemaphore_down(self->ring_free_sem);
/* set it */
self->ring[self->ring_head].buf = buf;
self->ring[self->ring_head].size = len;
self->ring_head = (self->ring_head + 1) % GLUE_RING_BUFFER_SIZE;
/* and mark this element as available for reading */
amsemaphore_up(self->ring_used_sem);
return;
case PUSH_TO_FD: {
int fd = get_write_fd(self);
/* if the fd is already closed, it's possible upstream bailed out
* so quickly that we didn't even get a look at the fd. In this
* case we can assume the xfer has been cancelled and just discard
* the data. */
if (fd == -1)
return;
if (elt->cancelled) {
if (!elt->expect_eof || !buf) {
close_write_fd(self);
/* hack to ensure we won't close the fd again, if we get another push */
elt->expect_eof = TRUE;
}
amfree(buf);
return;
}
/* write the full buffer to the fd, or close on EOF */
if (buf) {
if (!elt->downstream->drain_mode &&
full_write(fd, buf, len) < len) {
if (elt->downstream->must_drain) {
g_debug("Error writing to fd %d: %s",
fd, strerror(errno));
} else if (elt->downstream->ignore_broken_pipe &&
errno == EPIPE) {
} else {
if (!elt->cancelled) {
xfer_cancel_with_error(elt,
_("Error writing to fd %d: %s"),
fd, strerror(errno));
wait_until_xfer_cancelled(elt->xfer);
}
/* nothing special to do to handle a cancellation */
}
elt->downstream->drain_mode = TRUE;
}
crc32_add((uint8_t *)buf, len, &elt->crc);
amfree(buf);
} else {
g_debug("sending XMSG_CRC message");
g_debug("push_to_fd CRC: %08x", crc32_finish(&elt->crc));
msg = xmsg_new(elt->downstream, XMSG_CRC, 0);
msg->crc = crc32_finish(&elt->crc);
msg->size = elt->crc.size;
xfer_queue_message(elt->xfer, msg);
close_write_fd(self);
}
return;
}
default:
case PUSH_INVALID:
g_assert_not_reached();
break;
}
}
static gpointer
holding_thread(
gpointer data)
{
XferDestHolding *self = XFER_DEST_HOLDING(data);
XferElement *elt = XFER_ELEMENT(self);
XMsg *msg;
gchar *mesg = NULL;
GTimer *timer = g_timer_new();
DBG(1, "(this is the holding thread)");
/* This is the outer loop, that loops once for each holding file or
* CONTINUE command */
g_mutex_lock(self->state_mutex);
while (1) {
gboolean done;
/* wait until the main thread un-pauses us, and check that we have
* the relevant holding info available */
while (self->paused && !elt->cancelled) {
DBG(9, "waiting to be unpaused");
g_cond_wait(self->state_cond, self->state_mutex);
}
DBG(9, "holding_thread done waiting");
if (elt->cancelled)
break;
self->data_bytes_written = 0;
self->header_bytes_written = 0;
/* new holding file */
if (self->filename == NULL ||
strcmp(self->filename, self->new_filename) != 0) {
char *tmp_filename;
char *pc;
int fd;
ssize_t write_header_size;
if (self->use_bytes < HEADER_BLOCK_BYTES) {
self->chunk_status = CHUNK_NO_ROOM;
goto no_room;
}
tmp_filename = g_strjoin(NULL, self->new_filename, ".tmp", NULL);
pc = strrchr(tmp_filename, '/');
g_assert(pc != NULL);
*pc = '\0';
mkholdingdir(tmp_filename);
*pc = '/';
fd = open(tmp_filename, O_RDWR|O_CREAT|O_TRUNC, 0600);
if (fd < 0) {
self->chunk_status = CHUNK_NO_ROOM;
g_free(mesg);
mesg = g_strdup_printf("Failed to open '%s': %s",
tmp_filename, strerror(errno));
g_free(tmp_filename);
goto no_room;
}
if (self->filename == NULL) {
self->chunk_header->type = F_DUMPFILE;
} else {
self->chunk_header->type = F_CONT_DUMPFILE;
}
self->chunk_header->cont_filename[0] = '\0';
write_header_size = write_header(self, fd);
if (write_header_size != HEADER_BLOCK_BYTES) {
self->chunk_status = CHUNK_NO_ROOM;
mesg = g_strdup_printf("Failed to write header to '%s': %s",
tmp_filename, strerror(errno));
close(fd);
unlink(tmp_filename);
g_free(tmp_filename);
goto no_room;
}
g_free(tmp_filename);
self->use_bytes -= HEADER_BLOCK_BYTES;
/* rewrite old_header */
if (self->filename &&
strcmp(self->filename, self->new_filename) != 0) {
close_chunk(self, self->new_filename);
}
self->filename = self->new_filename;
self->new_filename = NULL;
self->fd = fd;
self->header_bytes_written = HEADER_BLOCK_BYTES;
self->chunk_offset = HEADER_BLOCK_BYTES;
}
DBG(2, "beginning to write chunk");
done = holding_thread_write_chunk(self);
DBG(2, "done writing chunk");
if (!done) /* cancelled */
break;
no_room:
msg = xmsg_new(XFER_ELEMENT(self), XMSG_CHUNK_DONE, 0);
msg->header_size = self->header_bytes_written;
msg->data_size = self->data_bytes_written;
msg->no_room = (self->chunk_status == CHUNK_NO_ROOM);
if (mesg) {
msg->message = mesg;
mesg = NULL;
}
xfer_queue_message(elt->xfer, msg);
/* pause ourselves and await instructions from the main thread */
self->paused = TRUE;
/* if this is the last part, we're done with the chunk loop */
if (self->chunk_status == CHUNK_EOF) {
break;
}
}
g_mutex_unlock(self->state_mutex);
g_debug("sending XMSG_CRC message");
g_debug("xfer-dest-holding CRC: %08x size: %lld",
crc32_finish(&elt->crc), (long long)elt->crc.size);
msg = xmsg_new(XFER_ELEMENT(self), XMSG_CRC, 0);
msg->crc = crc32_finish(&elt->crc);
msg->size = elt->crc.size;
xfer_queue_message(elt->xfer, msg);
msg = xmsg_new(XFER_ELEMENT(self), XMSG_DONE, 0);
msg->duration = g_timer_elapsed(timer, NULL);
g_timer_destroy(timer);
/* tell the main thread we're done */
xfer_queue_message(elt->xfer, msg);
return NULL;
}
static gpointer
pull_buffer_impl(
XferElement *elt,
size_t *size)
{
XferSourceRecovery *self = XFER_SOURCE_RECOVERY(elt);
gpointer buf = NULL;
int result;
int devsize;
XMsg *msg;
g_assert(elt->output_mech == XFER_MECH_PULL_BUFFER);
g_mutex_lock(self->start_part_mutex);
if (elt->size == 0) {
if (elt->offset == 0 && elt->orig_size == 0) {
self->paused = TRUE;
} else {
DBG(2, "xfer-source-recovery sending XMSG_CRC message");
DBG(2, "xfer-source-recovery CRC: %08x size %lld",
crc32_finish(&elt->crc), (long long)elt->crc.size);
msg = xmsg_new(XFER_ELEMENT(self), XMSG_CRC, 0);
msg->crc = crc32_finish(&elt->crc);
msg->size = elt->crc.size;
xfer_queue_message(elt->xfer, msg);
/* the device has signalled EOF (really end-of-part), so clean up instance
* variables and report the EOP to the caller in the form of an xmsg */
DBG(2, "pull_buffer hit EOF; sending XMSG_SEGMENT_DONE");
msg = xmsg_new(XFER_ELEMENT(self), XMSG_SEGMENT_DONE, 0);
msg->size = self->part_size;
if (self->part_timer) {
msg->duration = g_timer_elapsed(self->part_timer, NULL);
g_timer_destroy(self->part_timer);
self->part_timer = NULL;
}
msg->partnum = 0;
msg->fileno = self->device->file;
msg->successful = TRUE;
msg->eof = FALSE;
self->paused = TRUE;
device_clear_bytes_read(self->device);
self->bytes_read += self->part_size;
self->part_size = 0;
self->block_size = 0;
/* don't queue the XMSG_PART_DONE until we've adjusted all of our
* instance variables appropriately */
xfer_queue_message(elt->xfer, msg);
if (self->device->is_eof) {
DBG(2, "pull_buffer hit EOF; sending XMSG_PART_DONE");
msg = xmsg_new(XFER_ELEMENT(self), XMSG_PART_DONE, 0);
msg->size = self->part_size;
if (self->part_timer) {
msg->duration = g_timer_elapsed(self->part_timer, NULL);
g_timer_destroy(self->part_timer);
self->part_timer = NULL;
}
msg->partnum = 0;
msg->fileno = self->device->file;
msg->successful = TRUE;
msg->eof = FALSE;
xfer_queue_message(elt->xfer, msg);
}
}
}
while (1) {
/* make sure we have a device */
while (self->paused && !elt->cancelled)
g_cond_wait(self->start_part_cond, self->start_part_mutex);
/* indicate EOF on an cancel or when there are no more parts */
if (elt->cancelled) {
goto error;
}
if (self->done)
goto error;
/* start the timer if this is the first pull_buffer of this part */
if (!self->part_timer) {
DBG(2, "first pull_buffer of new part");
self->part_timer = g_timer_new();
}
if (elt->size == 0) {
result = -1;
} else {
/* loop until we read a full block, in case the blocks are larger
* than expected */
if (self->block_size == 0)
self->block_size = (size_t)self->device->block_size;
do {
int max_block;
buf = g_malloc(self->block_size);
if (buf == NULL) {
xfer_cancel_with_error(elt,
_("%s: cannot allocate memory"),
self->device->device_name);
g_mutex_unlock(self->start_part_mutex);
wait_until_xfer_cancelled(elt->xfer);
goto error_unlocked;
}
devsize = (int)self->block_size;
if (elt->size < 0)
max_block = -1;
else
max_block = (elt->size+self->block_size-1)/self->block_size;
result = device_read_block(self->device, buf, &devsize, max_block);
*size = devsize;
if (result == 0) {
g_assert(*size > self->block_size);
self->block_size = devsize;
amfree(buf);
}
} while (result == 0);
if (result > 0 &&
(elt->offset ||
(elt->size > 0 && (long long unsigned)elt->size < *size))) {
gpointer buf1 = g_malloc(self->block_size);
if ((long long unsigned)elt->offset > *size) {
g_debug("offset > *size");
} else if ((long long unsigned)elt->offset == *size) {
g_debug("offset == *size");
}
*size -= elt->offset;
if (elt->size > 0 && (size_t)elt->size < *size)
*size = elt->size;
memmove(buf1, buf + elt->offset, *size);
elt->offset = 0;
g_free(buf);
buf = buf1;
}
if (result > 0)
elt->size -= *size;
}
/* if this block was successful, return it */
if (result > 0) {
self->part_size += *size;
break;
}
if (result < 0) {
amfree(buf);
/* if we're not at EOF, it's an error */
if (!self->device->is_eof && elt->size != 0) {
xfer_cancel_with_error(elt,
_("error reading from %s: %s"),
self->device->device_name,
device_error_or_status(self->device));
g_mutex_unlock(self->start_part_mutex);
wait_until_xfer_cancelled(elt->xfer);
goto error_unlocked;
}
DBG(2, "xfer-source-recovery sending XMSG_CRC message");
DBG(2, "xfer-source-recovery CRC: %08x size %lld",
crc32_finish(&elt->crc), (long long)elt->crc.size);
msg = xmsg_new(XFER_ELEMENT(self), XMSG_CRC, 0);
msg->crc = crc32_finish(&elt->crc);
msg->size = elt->crc.size;
xfer_queue_message(elt->xfer, msg);
/* the device has signalled EOF (really end-of-part), so clean up instance
* variables and report the EOP to the caller in the form of an xmsg */
DBG(2, "pull_buffer hit EOF; sending XMSG_PART_DONE");
msg = xmsg_new(XFER_ELEMENT(self), XMSG_PART_DONE, 0);
msg->size = self->part_size;
msg->duration = g_timer_elapsed(self->part_timer, NULL);
msg->partnum = 0;
msg->fileno = self->device->file;
msg->successful = TRUE;
msg->eof = FALSE;
self->paused = TRUE;
self->bytes_read += self->part_size;
device_clear_bytes_read(self->device);
self->part_size = 0;
self->block_size = 0;
if (self->part_timer) {
g_timer_destroy(self->part_timer);
self->part_timer = NULL;
}
/* don't queue the XMSG_PART_DONE until we've adjusted all of our
* instance variables appropriately */
xfer_queue_message(elt->xfer, msg);
if (elt->size == 0) {
g_mutex_unlock(self->start_part_mutex);
return NULL;
}
}
}
g_mutex_unlock(self->start_part_mutex);
if (buf) {
crc32_add(buf, *size, &elt->crc);
}
return buf;
error:
g_mutex_unlock(self->start_part_mutex);
error_unlocked:
*size = 0;
return NULL;
}
/* common code for both directtcp_listen_thread and directtcp_connect_thread;
* this is called after self->conn is filled in and carries out the data
* transfer over that connection. NOTE: start_part_mutex is HELD when this
* function begins */
static gpointer
directtcp_common_thread(
XferSourceRecovery *self)
{
XferElement *elt = XFER_ELEMENT(self);
char *errmsg = NULL;
int result;
/* send XMSG_READY to indicate it's OK to call start_part now */
DBG(2, "directtcp_common_thread sending XMSG_READY");
xfer_queue_message(elt->xfer, xmsg_new(elt, XMSG_READY, 0));
/* now we sit around waiting for signals to write a part */
while (1) {
guint64 actual_size;
XMsg *msg;
while (self->paused && !elt->cancelled) {
DBG(9, "directtcp_common_thread waiting to be un-paused");
g_cond_wait(self->start_part_cond, self->start_part_mutex);
}
DBG(9, "directtcp_common_thread done waiting");
if (elt->cancelled) {
g_mutex_unlock(self->start_part_mutex);
goto close_conn_and_send_done;
}
/* if the device is NULL, we're done */
if (!self->device)
break;
/* read the part */
self->part_timer = g_timer_new();
while (1) {
DBG(2, "directtcp_common_thread reading part from %s", self->device->device_name);
result = device_read_to_connection(self->device, G_MAXUINT64,
&actual_size, &elt->cancelled,
self->start_part_mutex, self->abort_cond);
if (result == 1 && !elt->cancelled) {
xfer_cancel_with_error(elt, _("error reading from device: %s"),
device_error_or_status(self->device));
g_mutex_unlock(self->start_part_mutex);
goto close_conn_and_send_done;
} else if (result == 2 || elt->cancelled) {
g_mutex_unlock(self->start_part_mutex);
goto close_conn_and_send_done;
}
/* break on EOF; otherwise do another read_to_connection */
if (self->device->is_eof) {
break;
}
}
DBG(2, "done reading part; sending XMSG_PART_DONE");
/* the device has signalled EOF (really end-of-part), so clean up instance
* variables and report the EOP to the caller in the form of an xmsg */
msg = xmsg_new(XFER_ELEMENT(self), XMSG_PART_DONE, 0);
msg->size = actual_size;
msg->duration = g_timer_elapsed(self->part_timer, NULL);
msg->partnum = 0;
msg->fileno = self->device->file;
msg->successful = TRUE;
msg->eof = FALSE;
self->paused = TRUE;
g_object_unref(self->device);
self->device = NULL;
self->part_size = 0;
self->block_size = 0;
g_timer_destroy(self->part_timer);
self->part_timer = NULL;
xfer_queue_message(elt->xfer, msg);
}
g_mutex_unlock(self->start_part_mutex);
close_conn_and_send_done:
if (self->conn) {
errmsg = directtcp_connection_close(self->conn);
g_object_unref(self->conn);
self->conn = NULL;
if (errmsg) {
xfer_cancel_with_error(elt, _("error closing DirectTCP connection: %s"), errmsg);
wait_until_xfer_cancelled(elt->xfer);
}
}
xfer_queue_message(elt->xfer, xmsg_new(elt, XMSG_DONE, 0));
return NULL;
}
