static void
io_error (SoupSocket *sock, SoupMessage *msg, GError *error)
{
SoupMessagePrivate *priv = SOUP_MESSAGE_GET_PRIVATE (msg);
SoupMessageIOData *io = priv->io_data;
if (error && error->domain == SOUP_SSL_ERROR) {
soup_message_set_status_full (msg,
SOUP_STATUS_SSL_FAILED,
error->message);
} else if (io->mode == SOUP_MESSAGE_IO_CLIENT &&
io->read_state <= SOUP_MESSAGE_IO_STATE_HEADERS &&
io->read_meta_buf->len == 0 &&
!g_error_matches (error, G_IO_ERROR, G_IO_ERROR_TIMED_OUT) &&
request_is_idempotent (msg)) {
/* Connection got closed, but we can safely try again */
priv->io_status = SOUP_MESSAGE_IO_STATUS_QUEUED;
} else if (!SOUP_STATUS_IS_TRANSPORT_ERROR (msg->status_code))
soup_message_set_status (msg, SOUP_STATUS_IO_ERROR);
if (error)
g_error_free (error);
soup_message_io_finished (msg);
}
static void
io_run (SoupMessage *msg, gboolean blocking)
{
SoupMessagePrivate *priv = SOUP_MESSAGE_GET_PRIVATE (msg);
SoupMessageIOData *io = priv->io_data;
GError *error = NULL;
GCancellable *cancellable;
if (io->io_source) {
g_source_destroy (io->io_source);
g_source_unref (io->io_source);
io->io_source = NULL;
}
g_object_ref (msg);
cancellable = io->cancellable ? g_object_ref (io->cancellable) : NULL;
if (io_run_until (msg, blocking,
SOUP_MESSAGE_IO_STATE_DONE,
SOUP_MESSAGE_IO_STATE_DONE,
cancellable, &error)) {
soup_message_io_finished (msg);
} else if (g_error_matches (error, G_IO_ERROR, G_IO_ERROR_WOULD_BLOCK)) {
g_clear_error (&error);
io->io_source = soup_message_io_get_source (msg, NULL, io_run_ready, msg);
g_source_attach (io->io_source, io->async_context);
} else if (error && priv->io_data == io) {
if (g_error_matches (error, SOUP_HTTP_ERROR, SOUP_STATUS_TRY_AGAIN))
io->item->state = SOUP_MESSAGE_RESTARTING;
else if (error->domain == G_TLS_ERROR) {
soup_message_set_status_full (msg,
SOUP_STATUS_SSL_FAILED,
error->message);
} else if (!SOUP_STATUS_IS_TRANSPORT_ERROR (msg->status_code))
soup_message_set_status (msg, SOUP_STATUS_IO_ERROR);
g_error_free (error);
soup_message_io_finished (msg);
} else if (error)
g_error_free (error);
g_object_unref (msg);
g_clear_object (&cancellable);
}
static void
io_error (SoupSocket *sock, SoupMessage *msg, GError *error)
{
if (!SOUP_STATUS_IS_TRANSPORT_ERROR (msg->status_code)) {
if (error && error->domain == SOUP_SSL_ERROR) {
soup_message_set_status_full (msg,
SOUP_STATUS_SSL_FAILED,
error->message);
} else
soup_message_set_status (msg, SOUP_STATUS_IO_ERROR);
}
if (error)
g_error_free (error);
soup_message_io_finished (msg);
}
static void
io_read (SoupSocket *sock, SoupMessage *msg)
{
SoupMessagePrivate *priv = SOUP_MESSAGE_GET_PRIVATE (msg);
SoupMessageIOData *io = priv->io_data;
guint status;
read_more:
switch (io->read_state) {
case SOUP_MESSAGE_IO_STATE_NOT_STARTED:
return;
case SOUP_MESSAGE_IO_STATE_HEADERS:
if (!read_metadata (msg, TRUE))
return;
/* We need to "rewind" io->read_meta_buf back one line.
* That SHOULD be two characters (CR LF), but if the
* web server was stupid, it might only be one.
*/
if (io->read_meta_buf->len < 3 ||
io->read_meta_buf->data[io->read_meta_buf->len - 2] == '\n')
io->read_meta_buf->len--;
else
io->read_meta_buf->len -= 2;
io->read_meta_buf->data[io->read_meta_buf->len] = '\0';
status = io->parse_headers_cb (msg, (char *)io->read_meta_buf->data,
io->read_meta_buf->len,
&io->read_encoding,
io->user_data);
g_byte_array_set_size (io->read_meta_buf, 0);
if (status != SOUP_STATUS_OK) {
/* Either we couldn't parse the headers, or they
* indicated something that would mean we wouldn't
* be able to parse the body. (Eg, unknown
* Transfer-Encoding.). Skip the rest of the
* reading, and make sure the connection gets
* closed when we're done.
*/
soup_message_set_status (msg, status);
soup_message_headers_append (msg->request_headers,
"Connection", "close");
io->read_state = SOUP_MESSAGE_IO_STATE_FINISHING;
break;
}
if (io->read_encoding == SOUP_ENCODING_CONTENT_LENGTH) {
SoupMessageHeaders *hdrs =
(io->mode == SOUP_MESSAGE_IO_CLIENT) ?
msg->response_headers : msg->request_headers;
io->read_length = soup_message_headers_get_content_length (hdrs);
}
if (io->mode == SOUP_MESSAGE_IO_CLIENT &&
SOUP_STATUS_IS_INFORMATIONAL (msg->status_code)) {
if (msg->status_code == SOUP_STATUS_CONTINUE &&
io->write_state == SOUP_MESSAGE_IO_STATE_BLOCKING) {
/* Pause the reader, unpause the writer */
io->read_state =
SOUP_MESSAGE_IO_STATE_BLOCKING;
io->write_state =
io_body_state (io->write_encoding);
} else {
/* Just stay in HEADERS */
io->read_state = SOUP_MESSAGE_IO_STATE_HEADERS;
}
} else if (io->mode == SOUP_MESSAGE_IO_SERVER &&
soup_message_headers_get_expectations (msg->request_headers) & SOUP_EXPECTATION_CONTINUE) {
/* The client requested a Continue response. The
* got_headers handler may change this to something
* else though.
*/
soup_message_set_status (msg, SOUP_STATUS_CONTINUE);
io->write_state = SOUP_MESSAGE_IO_STATE_HEADERS;
io->read_state = SOUP_MESSAGE_IO_STATE_BLOCKING;
} else {
io->read_state = io_body_state (io->read_encoding);
/* If the client was waiting for a Continue
* but got something else, then it's done
* writing.
*/
if (io->mode == SOUP_MESSAGE_IO_CLIENT &&
io->write_state == SOUP_MESSAGE_IO_STATE_BLOCKING)
io->write_state = SOUP_MESSAGE_IO_STATE_FINISHING;
}
if (io->mode == SOUP_MESSAGE_IO_CLIENT &&
SOUP_STATUS_IS_INFORMATIONAL (msg->status_code)) {
SOUP_MESSAGE_IO_PREPARE_FOR_CALLBACK;
soup_message_got_informational (msg);
soup_message_cleanup_response (msg);
SOUP_MESSAGE_IO_RETURN_IF_CANCELLED_OR_PAUSED;
} else {
SOUP_MESSAGE_IO_PREPARE_FOR_CALLBACK;
soup_message_got_headers (msg);
SOUP_MESSAGE_IO_RETURN_IF_CANCELLED_OR_PAUSED;
}
break;
case SOUP_MESSAGE_IO_STATE_BLOCKING:
io_write (sock, msg);
/* As in the io_write case, we *must* return here. */
return;
case SOUP_MESSAGE_IO_STATE_BODY:
if (!read_body_chunk (msg))
return;
got_body:
if (!io_handle_sniffing (msg, TRUE)) {
/* If the message was paused (as opposed to
* cancelled), we need to make sure we wind up
* back here when it's unpaused, even if it
* was doing a chunked or EOF-terminated read
* before.
*/
if (io == priv->io_data) {
io->read_state = SOUP_MESSAGE_IO_STATE_BODY;
io->read_encoding = SOUP_ENCODING_CONTENT_LENGTH;
io->read_length = 0;
}
return;
}
io->read_state = SOUP_MESSAGE_IO_STATE_FINISHING;
SOUP_MESSAGE_IO_PREPARE_FOR_CALLBACK;
soup_message_got_body (msg);
SOUP_MESSAGE_IO_RETURN_IF_CANCELLED_OR_PAUSED;
break;
case SOUP_MESSAGE_IO_STATE_CHUNK_SIZE:
if (!read_metadata (msg, FALSE))
return;
io->read_length = strtoul ((char *)io->read_meta_buf->data, NULL, 16);
g_byte_array_set_size (io->read_meta_buf, 0);
if (io->read_length > 0)
io->read_state = SOUP_MESSAGE_IO_STATE_CHUNK;
else
io->read_state = SOUP_MESSAGE_IO_STATE_TRAILERS;
break;
case SOUP_MESSAGE_IO_STATE_CHUNK:
if (!read_body_chunk (msg))
return;
io->read_state = SOUP_MESSAGE_IO_STATE_CHUNK_END;
break;
case SOUP_MESSAGE_IO_STATE_CHUNK_END:
if (!read_metadata (msg, FALSE))
return;
g_byte_array_set_size (io->read_meta_buf, 0);
io->read_state = SOUP_MESSAGE_IO_STATE_CHUNK_SIZE;
break;
case SOUP_MESSAGE_IO_STATE_TRAILERS:
if (!read_metadata (msg, FALSE))
return;
if (io->read_meta_buf->len <= SOUP_MESSAGE_IO_EOL_LEN)
goto got_body;
/* FIXME: process trailers */
g_byte_array_set_size (io->read_meta_buf, 0);
break;
case SOUP_MESSAGE_IO_STATE_FINISHING:
if (io->read_tag) {
g_signal_handler_disconnect (io->sock, io->read_tag);
io->read_tag = 0;
}
io->read_state = SOUP_MESSAGE_IO_STATE_DONE;
if (io->mode == SOUP_MESSAGE_IO_SERVER) {
io->write_state = SOUP_MESSAGE_IO_STATE_HEADERS;
io_write (sock, msg);
} else
soup_message_io_finished (msg);
return;
case SOUP_MESSAGE_IO_STATE_DONE:
default:
g_return_if_reached ();
}
goto read_more;
}
static void
io_write (SoupSocket *sock, SoupMessage *msg)
{
SoupMessagePrivate *priv = SOUP_MESSAGE_GET_PRIVATE (msg);
SoupMessageIOData *io = priv->io_data;
write_more:
switch (io->write_state) {
case SOUP_MESSAGE_IO_STATE_NOT_STARTED:
return;
case SOUP_MESSAGE_IO_STATE_HEADERS:
if (!io->write_buf->len) {
io->get_headers_cb (msg, io->write_buf,
&io->write_encoding,
io->user_data);
if (!io->write_buf->len) {
soup_message_io_pause (msg);
return;
}
}
if (!write_data (msg, io->write_buf->str,
io->write_buf->len, FALSE))
return;
g_string_truncate (io->write_buf, 0);
if (io->write_encoding == SOUP_ENCODING_CONTENT_LENGTH) {
SoupMessageHeaders *hdrs =
(io->mode == SOUP_MESSAGE_IO_CLIENT) ?
msg->request_headers : msg->response_headers;
io->write_length = soup_message_headers_get_content_length (hdrs);
}
if (io->mode == SOUP_MESSAGE_IO_SERVER &&
SOUP_STATUS_IS_INFORMATIONAL (msg->status_code)) {
if (msg->status_code == SOUP_STATUS_CONTINUE) {
/* Stop and wait for the body now */
io->write_state =
SOUP_MESSAGE_IO_STATE_BLOCKING;
io->read_state = io_body_state (io->read_encoding);
} else {
/* We just wrote a 1xx response
* header, so stay in STATE_HEADERS.
* (The caller will pause us from the
* wrote_informational callback if he
* is not ready to send the final
* response.)
*/
}
} else if (io->mode == SOUP_MESSAGE_IO_CLIENT &&
soup_message_headers_get_expectations (msg->request_headers) & SOUP_EXPECTATION_CONTINUE) {
/* Need to wait for the Continue response */
io->write_state = SOUP_MESSAGE_IO_STATE_BLOCKING;
io->read_state = SOUP_MESSAGE_IO_STATE_HEADERS;
} else {
io->write_state = io_body_state (io->write_encoding);
/* If the client was waiting for a Continue
* but we sent something else, then they're
* now done writing.
*/
if (io->mode == SOUP_MESSAGE_IO_SERVER &&
io->read_state == SOUP_MESSAGE_IO_STATE_BLOCKING)
io->read_state = SOUP_MESSAGE_IO_STATE_FINISHING;
}
SOUP_MESSAGE_IO_PREPARE_FOR_CALLBACK;
if (SOUP_STATUS_IS_INFORMATIONAL (msg->status_code)) {
soup_message_wrote_informational (msg);
soup_message_cleanup_response (msg);
} else
soup_message_wrote_headers (msg);
SOUP_MESSAGE_IO_RETURN_IF_CANCELLED_OR_PAUSED;
break;
case SOUP_MESSAGE_IO_STATE_BLOCKING:
io_read (sock, msg);
/* If io_read reached a point where we could write
* again, it would have recursively called io_write.
* So (a) we don't need to try to keep writing, and
* (b) we can't anyway, because msg may have been
* destroyed.
*/
return;
case SOUP_MESSAGE_IO_STATE_BODY:
if (!io->write_length && io->write_encoding != SOUP_ENCODING_EOF) {
wrote_body:
io->write_state = SOUP_MESSAGE_IO_STATE_FINISHING;
SOUP_MESSAGE_IO_PREPARE_FOR_CALLBACK;
soup_message_wrote_body (msg);
SOUP_MESSAGE_IO_RETURN_IF_CANCELLED_OR_PAUSED;
break;
}
if (!io->write_chunk) {
io->write_chunk = soup_message_body_get_chunk (io->write_body, io->write_body_offset);
if (!io->write_chunk) {
soup_message_io_pause (msg);
return;
}
if (io->write_chunk->length > io->write_length &&
io->write_encoding != SOUP_ENCODING_EOF) {
/* App is trying to write more than it
* claimed it would; we have to truncate.
*/
SoupBuffer *truncated =
soup_buffer_new_subbuffer (io->write_chunk,
0, io->write_length);
soup_buffer_free (io->write_chunk);
io->write_chunk = truncated;
} else if (io->write_encoding == SOUP_ENCODING_EOF &&
!io->write_chunk->length)
goto wrote_body;
}
if (!write_data (msg, io->write_chunk->data,
io->write_chunk->length, TRUE))
return;
if (io->mode == SOUP_MESSAGE_IO_SERVER)
soup_message_body_wrote_chunk (io->write_body, io->write_chunk);
io->write_body_offset += io->write_chunk->length;
soup_buffer_free (io->write_chunk);
io->write_chunk = NULL;
SOUP_MESSAGE_IO_PREPARE_FOR_CALLBACK;
soup_message_wrote_chunk (msg);
SOUP_MESSAGE_IO_RETURN_IF_CANCELLED_OR_PAUSED;
break;
case SOUP_MESSAGE_IO_STATE_CHUNK_SIZE:
if (!io->write_chunk) {
io->write_chunk = soup_message_body_get_chunk (io->write_body, io->write_body_offset);
if (!io->write_chunk) {
soup_message_io_pause (msg);
return;
}
g_string_append_printf (io->write_buf, "%lx\r\n",
(unsigned long) io->write_chunk->length);
io->write_body_offset += io->write_chunk->length;
}
if (!write_data (msg, io->write_buf->str,
io->write_buf->len, FALSE))
return;
g_string_truncate (io->write_buf, 0);
if (io->write_chunk->length == 0) {
/* The last chunk has no CHUNK_END... */
io->write_state = SOUP_MESSAGE_IO_STATE_TRAILERS;
break;
}
io->write_state = SOUP_MESSAGE_IO_STATE_CHUNK;
/* fall through */
case SOUP_MESSAGE_IO_STATE_CHUNK:
if (!write_data (msg, io->write_chunk->data,
io->write_chunk->length, TRUE))
return;
if (io->mode == SOUP_MESSAGE_IO_SERVER)
soup_message_body_wrote_chunk (io->write_body, io->write_chunk);
soup_buffer_free (io->write_chunk);
io->write_chunk = NULL;
io->write_state = SOUP_MESSAGE_IO_STATE_CHUNK_END;
SOUP_MESSAGE_IO_PREPARE_FOR_CALLBACK;
soup_message_wrote_chunk (msg);
SOUP_MESSAGE_IO_RETURN_IF_CANCELLED_OR_PAUSED;
/* fall through */
case SOUP_MESSAGE_IO_STATE_CHUNK_END:
if (!write_data (msg, SOUP_MESSAGE_IO_EOL,
SOUP_MESSAGE_IO_EOL_LEN, FALSE))
return;
io->write_state = SOUP_MESSAGE_IO_STATE_CHUNK_SIZE;
break;
case SOUP_MESSAGE_IO_STATE_TRAILERS:
if (!write_data (msg, SOUP_MESSAGE_IO_EOL,
SOUP_MESSAGE_IO_EOL_LEN, FALSE))
return;
io->write_state = SOUP_MESSAGE_IO_STATE_FINISHING;
SOUP_MESSAGE_IO_PREPARE_FOR_CALLBACK;
soup_message_wrote_body (msg);
SOUP_MESSAGE_IO_RETURN_IF_CANCELLED_OR_PAUSED;
/* fall through */
case SOUP_MESSAGE_IO_STATE_FINISHING:
if (io->write_tag) {
g_signal_handler_disconnect (io->sock, io->write_tag);
io->write_tag = 0;
}
io->write_state = SOUP_MESSAGE_IO_STATE_DONE;
if (io->mode == SOUP_MESSAGE_IO_CLIENT) {
io->read_state = SOUP_MESSAGE_IO_STATE_HEADERS;
io_read (sock, msg);
} else
soup_message_io_finished (msg);
return;
case SOUP_MESSAGE_IO_STATE_DONE:
default:
g_return_if_reached ();
}
goto write_more;
}