void
soup_message_io_finished (SoupMessage *msg)
{
SoupMessagePrivate *priv = SOUP_MESSAGE_GET_PRIVATE (msg);
SoupMessageIOData *io = priv->io_data;
SoupMessageCompletionFn completion_cb;
gpointer completion_data;
SoupMessageIOCompletion completion;
if (!io)
return;
completion_cb = io->completion_cb;
completion_data = io->completion_data;
if ((io->read_state >= SOUP_MESSAGE_IO_STATE_FINISHING &&
io->write_state >= SOUP_MESSAGE_IO_STATE_FINISHING))
completion = SOUP_MESSAGE_IO_COMPLETE;
else
completion = SOUP_MESSAGE_IO_INTERRUPTED;
g_object_ref (msg);
soup_message_io_cleanup (msg);
if (completion_cb)
completion_cb (msg, completion, completion_data);
g_object_unref (msg);
}
static SoupMessageIOData *
new_iostate (SoupMessage *msg, SoupSocket *sock, SoupMessageIOMode mode,
SoupMessageGetHeadersFn get_headers_cb,
SoupMessageParseHeadersFn parse_headers_cb,
gpointer user_data)
{
SoupMessagePrivate *priv = SOUP_MESSAGE_GET_PRIVATE (msg);
SoupMessageIOData *io;
io = g_slice_new0 (SoupMessageIOData);
io->sock = g_object_ref (sock);
io->mode = mode;
io->get_headers_cb = get_headers_cb;
io->parse_headers_cb = parse_headers_cb;
io->user_data = user_data;
io->read_meta_buf = g_byte_array_new ();
io->write_buf = g_string_new (NULL);
io->read_tag = g_signal_connect (io->sock, "readable",
G_CALLBACK (io_read), msg);
io->write_tag = g_signal_connect (io->sock, "writable",
G_CALLBACK (io_write), msg);
io->err_tag = g_signal_connect (io->sock, "disconnected",
G_CALLBACK (io_disconnected), msg);
io->read_state = SOUP_MESSAGE_IO_STATE_NOT_STARTED;
io->write_state = SOUP_MESSAGE_IO_STATE_NOT_STARTED;
if (priv->io_data)
soup_message_io_cleanup (msg);
priv->io_data = io;
return io;
}
static gboolean
soup_input_stream_seek (GSeekable *seekable,
goffset offset,
GSeekType type,
GCancellable *cancellable,
GError **error)
{
GInputStream *stream = G_INPUT_STREAM (seekable);
SoupInputStreamPrivate *priv = SOUP_INPUT_STREAM_GET_PRIVATE (seekable);
char *range;
if (type == G_SEEK_END)
{
/* FIXME: we could send "bytes=-offset", but unless we know the
* Content-Length, we wouldn't be able to answer a tell() properly.
* We could find the Content-Length by doing a HEAD...
*/
g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_NOT_SUPPORTED,
"G_SEEK_END not currently supported");
return FALSE;
}
if (!g_input_stream_set_pending (stream, error))
return FALSE;
soup_session_cancel_message (priv->session, priv->msg, SOUP_STATUS_CANCELLED);
soup_message_io_cleanup (priv->msg);
switch (type)
{
case G_SEEK_CUR:
offset += priv->offset;
/* fall through */
case G_SEEK_SET:
range = g_strdup_printf ("bytes=%"G_GUINT64_FORMAT"-", (guint64)offset);
priv->offset = offset;
break;
case G_SEEK_END:
range = NULL; /* keep compilers happy */
g_return_val_if_reached (FALSE);
break;
default:
g_return_val_if_reached (FALSE);
}
soup_message_headers_remove (priv->msg->request_headers, "Range");
soup_message_headers_append (priv->msg->request_headers, "Range", range);
g_free (range);
soup_input_stream_queue_message (SOUP_INPUT_STREAM (stream));
g_input_stream_clear_pending (stream);
return TRUE;
}
static void
soup_message_io_finished (SoupMessage *msg)
{
g_object_ref (msg);
soup_message_io_cleanup (msg);
if (SOUP_MESSAGE_IS_STARTING (msg))
soup_message_restarted (msg);
else
soup_message_finished (msg);
g_object_unref (msg);
}
static SoupMessageIOData *
new_iostate (SoupMessage *msg, GIOStream *iostream,
GMainContext *async_context, SoupMessageIOMode mode,
SoupMessageGetHeadersFn get_headers_cb,
SoupMessageParseHeadersFn parse_headers_cb,
gpointer header_data,
SoupMessageCompletionFn completion_cb,
gpointer completion_data)
{
SoupMessagePrivate *priv = SOUP_MESSAGE_GET_PRIVATE (msg);
SoupMessageIOData *io;
io = g_slice_new0 (SoupMessageIOData);
io->mode = mode;
io->get_headers_cb = get_headers_cb;
io->parse_headers_cb = parse_headers_cb;
io->header_data = header_data;
io->completion_cb = completion_cb;
io->completion_data = completion_data;
io->iostream = g_object_ref (iostream);
io->istream = SOUP_FILTER_INPUT_STREAM (g_io_stream_get_input_stream (iostream));
io->ostream = g_io_stream_get_output_stream (iostream);
if (async_context)
io->async_context = g_main_context_ref (async_context);
io->read_header_buf = g_byte_array_new ();
io->write_buf = g_string_new (NULL);
io->read_state = SOUP_MESSAGE_IO_STATE_NOT_STARTED;
io->write_state = SOUP_MESSAGE_IO_STATE_NOT_STARTED;
if (priv->io_data)
soup_message_io_cleanup (msg);
priv->io_data = io;
return io;
}
GIOStream *
soup_message_io_steal (SoupMessage *msg)
{
SoupMessagePrivate *priv = SOUP_MESSAGE_GET_PRIVATE (msg);
SoupMessageIOData *io = priv->io_data;
SoupMessageCompletionFn completion_cb;
gpointer completion_data;
GIOStream *iostream;
if (!io || !io->iostream)
return NULL;
iostream = g_object_ref (io->iostream);
completion_cb = io->completion_cb;
completion_data = io->completion_data;
g_object_ref (msg);
soup_message_io_cleanup (msg);
if (completion_cb)
completion_cb (msg, SOUP_MESSAGE_IO_STOLEN, completion_data);
g_object_unref (msg);
return iostream;
}
