static void
_client_react(struct gridd_client_s *client)
{
EXTRA_ASSERT(client != NULL);
EXTRA_ASSERT(client->abstract.vtable == &VTABLE_CLIENT);
if (!client)
return;
GError *err = NULL;
client->tv_step = oio_ext_monotonic_time ();
retry:
if (!(err = _client_manage_event(client))) {
if (client->step == REP_READING_SIZE && client->reply
&& client->reply->len >= 4)
goto retry;
}
else {
_client_reset_request(client);
_client_reset_reply(client);
_client_reset_cnx(client);
client->error = err;
client->step = STATUS_FAILED;
}
}
static gboolean
_client_expire(struct gridd_client_s *client, gint64 now)
{
EXTRA_ASSERT(client != NULL);
EXTRA_ASSERT(client->abstract.vtable == &VTABLE_CLIENT);
if (_client_finished(client))
return FALSE;
if (!_client_expired(client, now))
return FALSE;
_client_reset_cnx(client);
client->error = NEWERROR(ERRCODE_READ_TIMEOUT, "Timeout");
client->step = STATUS_FAILED;
return FALSE;
}
static void
_client_free(struct gridd_client_s *client)
{
EXTRA_ASSERT(client != NULL);
EXTRA_ASSERT(client->abstract.vtable == &VTABLE_CLIENT);
_client_reset_reply(client);
_client_reset_request(client);
_client_reset_cnx(client);
_client_reset_target(client);
_client_reset_error(client);
if (client->reply)
g_byte_array_free(client->reply, TRUE);
client->fd = -1;
SLICE_FREE (struct gridd_client_s, client);
}
static void
_client_free(struct gridd_client_s *client)
{
EXTRA_ASSERT(client != NULL);
EXTRA_ASSERT(client->abstract.vtable == &VTABLE_CLIENT);
_client_reset_reply(client);
_client_reset_request(client);
_client_reset_cnx(client);
_client_reset_target(client);
_client_reset_error(client);
if (client->reply)
g_byte_array_free(client->reply, TRUE);
if (client->past_url)
g_string_free(client->past_url, TRUE);
memset(client, 0, sizeof(*client));
client->fd = -1;
g_free(client);
}
static GError*
_client_connect_addr(struct gridd_client_s *client,
const struct addr_info_s *ai)
{
EXTRA_ASSERT(client != NULL);
EXTRA_ASSERT(client->abstract.vtable == &VTABLE_CLIENT);
if (NULL == ai || !ai->port)
return NEWERROR(CODE_INTERNAL_ERROR, "Invalid parameter");
g_assert(client != NULL);
_client_reset_cnx(client);
_client_reset_target(client);
_client_reset_reply(client);
grid_addrinfo_to_string(ai, client->orig_url, URL_MAXLEN);
memcpy(client->url, client->orig_url, URL_MAXLEN);
client->step = NONE;
return NULL;
}
static GError*
_client_connect_url(struct gridd_client_s *client, const gchar *url)
{
EXTRA_ASSERT(client != NULL);
EXTRA_ASSERT(client->abstract.vtable == &VTABLE_CLIENT);
if (NULL == url || !url[0])
return NEWERROR(CODE_INTERNAL_ERROR, "Bad address");
if (!metautils_url_valid_for_connect(url))
return NEWERROR(400, "Bad address [%s]", url);
g_assert(client != NULL);
_client_reset_cnx(client);
_client_reset_target(client);
_client_reset_reply(client);
g_strlcpy(client->orig_url, url, URL_MAXLEN);
memcpy(client->url, client->orig_url, URL_MAXLEN);
client->step = NONE;
return NULL;
}
static GError *
_client_manage_reply(struct gridd_client_s *client, MESSAGE reply)
{
GError *err;
gint status = 0;
gchar *message = NULL;
if (!metaXClient_reply_simple(reply, &status, &message, NULL))
return NEWERROR(500, "Invalid reply");
switch (status / 100) {
case 0:
err = NEWERROR(status, "net error: %s", message);
g_free(message);
metautils_pclose(&(client->fd));
client->step = STATUS_FAILED;
return err;
case 1: /* Informational reply : */
g_get_current_time(&(client->tv_step));
client->step = REP_READING_SIZE;
g_free(message);
return NULL;
case 2:
g_get_current_time(&(client->tv_step));
client->step = (status==200) ? STATUS_OK : REP_READING_SIZE;
if (client->step == STATUS_OK) {
if (!client->keepalive)
metautils_pclose(&(client->fd));
}
g_free(message);
if (client->on_reply) {
if (!client->on_reply(client->ctx, reply))
return NEWERROR(500, "Handler error");
}
return NULL;
case 3: /* redirection */
if (status == CODE_REDIRECT) {
/* Reset the context */
_client_reset_reply(client);
_client_reset_cnx(client);
client->sent_bytes = 0;
if ((++ client->nb_redirects) > 7) {
g_free(message);
return NEWERROR(CODE_TOOMANY_REDIRECT,
"Too many redirections");
}
/* Save the current URL to avoid looping, and check
* for a potential loop */
g_string_append_c(client->past_url, '|');
g_string_append(client->past_url, client->url);
if (_client_looped(client, message)) {
g_free(message);
return NEWERROR(CODE_LOOP_REDIRECT,
"Looping on redirections");
}
/* Replace the URL */
memset(client->url, 0, sizeof(client->url));
g_strlcpy(client->url, message, sizeof(client->url)-1);
if (NULL != (err = _client_connect(client))) {
g_free(message);
g_prefix_error(&err, "Redirection error: Connect error: ");
return err;
}
g_free(message);
return NULL;
}
/* FALLTHROUGH */
default: /* all other are considered errors */
err = NEWERROR(status, "Request error: %s", message);
g_free(message);
if (!client->keepalive)
_client_reset_cnx(client);
_client_reset_reply(client);
return err;
}
}
static GError *
_client_manage_reply(struct gridd_client_s *client, MESSAGE reply)
{
GError *err;
guint status = 0;
gchar *message = NULL;
if (NULL != (err = metaXClient_reply_simple(reply, &status, &message))) {
g_prefix_error (&err, "reply: ");
return err;
}
if (CODE_IS_NETWORK_ERROR(status)) {
err = NEWERROR(status, "net error: %s", message);
g_free(message);
metautils_pclose(&(client->fd));
client->step = STATUS_FAILED;
return err;
}
if (status == CODE_TEMPORARY) {
g_get_current_time(&(client->tv_step));
client->step = REP_READING_SIZE;
g_free(message);
return NULL;
}
if (CODE_IS_OK(status)) {
g_get_current_time(&(client->tv_step));
client->step = (status==CODE_FINAL_OK) ? STATUS_OK : REP_READING_SIZE;
if (client->step == STATUS_OK) {
if (!client->keepalive)
metautils_pclose(&(client->fd));
}
g_free(message);
if (client->on_reply) {
if (!client->on_reply(client->ctx, reply))
return NEWERROR(CODE_INTERNAL_ERROR, "Handler error");
}
return NULL;
}
if (status == CODE_REDIRECT) {
/* Reset the context */
_client_reset_reply(client);
_client_reset_cnx(client);
client->sent_bytes = 0;
if ((++ client->nb_redirects) > 7) {
g_free(message);
return NEWERROR(CODE_TOOMANY_REDIRECT, "Too many redirections");
}
/* Save the current URL to avoid looping, and check
* for a potential loop */
g_string_append_c(client->past_url, '|');
g_string_append(client->past_url, client->url);
if (_client_looped(client, message)) {
g_free(message);
return NEWERROR(CODE_LOOP_REDIRECT, "Looping on redirections");
}
/* Replace the URL */
memset(client->url, 0, sizeof(client->url));
g_strlcpy(client->url, message, sizeof(client->url)-1);
if (NULL != (err = _client_connect(client))) {
g_free(message);
g_prefix_error(&err, "Redirection error: Connect error: ");
return err;
}
g_free(message);
return NULL;
}
/* all other are considered errors */
err = NEWERROR(status, "Request error: %s", message);
g_free(message);
if (!client->keepalive)
_client_reset_cnx(client);
_client_reset_reply(client);
return err;
}
static GError *
_client_manage_reply(struct gridd_client_s *client, MESSAGE reply)
{
GError *err;
guint status = 0;
gchar *message = NULL;
if (NULL != (err = metaXClient_reply_simple(reply, &status, &message))) {
g_prefix_error (&err, "reply: ");
return err;
}
STRING_STACKIFY(message);
if (CODE_IS_NETWORK_ERROR(status)) {
err = NEWERROR(status, "net error: %s", message);
metautils_pclose(&(client->fd));
client->step = STATUS_FAILED;
return err;
}
if (status == CODE_TEMPORARY) {
client->step = REP_READING_SIZE;
return NULL;
}
if (CODE_IS_OK(status)) {
client->step = (status==CODE_FINAL_OK) ? STATUS_OK : REP_READING_SIZE;
if (client->step == STATUS_OK) {
if (!client->keepalive)
metautils_pclose(&(client->fd));
}
if (client->on_reply) {
if (!client->on_reply(client->ctx, reply))
return NEWERROR(CODE_INTERNAL_ERROR, "Handler error");
}
return NULL;
}
if (status == CODE_REDIRECT && !client->forbid_redirect) {
/* Reset the context */
_client_reset_reply(client);
_client_reset_cnx(client);
client->sent_bytes = 0;
if ((++ client->nb_redirects) > 3)
return NEWERROR(CODE_TOOMANY_REDIRECT, "Too many redirections");
/* Replace the URL */
g_strlcpy(client->url, message, URL_MAXLEN);
if (NULL != (err = _client_connect(client)))
g_prefix_error(&err, "Redirection error: Connect error: ");
return err;
}
/* all other are considered errors */
if (status != CODE_REDIRECT)
err = NEWERROR(status, "Request error: %s", message);
else
err = NEWERROR(status, "%s", message);
if (!client->keepalive)
_client_reset_cnx(client);
_client_reset_reply(client);
return err;
}
