ret_t
cherokee_buffer_add_fsize (cherokee_buffer_t *buf, CST_OFFSET size)
{
ret_t ret;
int remain;
const char ord[] = "KMGTPE";
const char *o = ord;
ret = cherokee_buffer_ensure_size (buf, buf->len + 8);
if (unlikely (ret != ret_ok))
return ret;
if (size < 973)
return cherokee_buffer_add_ulong10 (buf, (culong_t)size);
do {
remain = (int)(size & 1023);
size >>= 10;
if (size >= 973) {
++o;
continue;
}
if (size < 9 || (size == 9 && remain < 973)) {
if ((remain = ((remain * 5) + 256) / 512) >= 10)
++size, remain = 0;
return cherokee_buffer_add_va_fixed (buf, "%d.%d%c", (int) size, remain, *o);
}
if (remain >= 512)
++size;
return cherokee_buffer_add_va_fixed (buf, "%3d%c", (int) size, *o);
} while (1);
return ret_ok;
}
ret_t
cherokee_url_build_string (cherokee_url_t *url, cherokee_buffer_t *buf)
{
cherokee_buffer_add_buffer (buf, &url->host);
if (((url->protocol == http) && (url->port != 80)) ||
((url->protocol == https) && (url->port != 443))) {
cherokee_buffer_add_char (buf, ':');
cherokee_buffer_add_ulong10 (buf, (culong_t) url->port);
}
cherokee_buffer_add_buffer (buf, &url->request);
return ret_ok;
}
ret_t
cherokee_url_build_string (cherokee_url_t *url, cherokee_buffer_t *buf)
{
cherokee_buffer_add_buffer (buf, &url->host);
if (! http_port_is_standard (url->port, (url->protocol == https)))
{
cherokee_buffer_add_char (buf, ':');
cherokee_buffer_add_ulong10 (buf, (culong_t) url->port);
}
cherokee_buffer_add_buffer (buf, &url->request);
return ret_ok;
}
ret_t
cherokee_handler_proxy_hosts_get (cherokee_handler_proxy_hosts_t *hosts,
cherokee_source_t *src,
cherokee_handler_proxy_poll_t **poll,
cuint_t reuse_max)
{
ret_t ret;
CHEROKEE_MUTEX_LOCK (&hosts->hosts_mutex);
/* Build the index name */
cherokee_buffer_clean (&hosts->tmp);
cherokee_buffer_add_buffer (&hosts->tmp, &src->host);
cherokee_buffer_add_char (&hosts->tmp, ':');
cherokee_buffer_add_ulong10 (&hosts->tmp, src->port);
/* Check the hosts tree */
ret = cherokee_avl_get (&hosts->hosts, &hosts->tmp, (void **)poll);
switch (ret) {
case ret_ok:
break;
case ret_not_found: {
cherokee_handler_proxy_poll_t *n;
ret = cherokee_handler_proxy_poll_new (&n, reuse_max);
if (ret != ret_ok)
return ret;
cherokee_avl_add (&hosts->hosts, &hosts->tmp, n);
*poll = n;
break;
}
default:
goto error;
}
/* Got it */
CHEROKEE_MUTEX_UNLOCK (&hosts->hosts_mutex);
return ret_ok;
error:
CHEROKEE_MUTEX_LOCK (&hosts->hosts_mutex);
return ret_error;
}
ret_t
cherokee_handler_error_add_headers (cherokee_handler_error_t *hdl, cherokee_buffer_t *buffer)
{
cherokee_connection_t *conn = HANDLER_CONN(hdl);
/* It has special headers on protocol upgrading
*/
switch (conn->upgrade) {
case http_upgrade_nothing:
break;
case http_upgrade_tls10:
cherokee_buffer_add_str (buffer, "Upgrade: TLS/1.0, HTTP/1.1"CRLF);
break;
case http_upgrade_http11:
cherokee_buffer_add_str (buffer, "Upgrade: HTTP/1.1"CRLF);
break;
default:
SHOULDNT_HAPPEN;
}
/* 1xx, 204 and 304 (Not Modified) responses have to be managed
* by "content" handlers, anyway this test ensures that
* it'll never send wrong and unrelated headers in case that
* a 1xx, 204 or 304 response is managed by this handler.
* 304 responses should only include the
* Last-Modified, ETag, Expires and Cache-Control headers.
*/
if (!http_code_with_body (conn->error_code)) {
return ret_ok;
}
if (cherokee_connection_should_include_length(conn)) {
HANDLER(hdl)->support |= hsupport_length;
if (conn->error_code == http_range_not_satisfiable) {
/* The handler that attended the request has put the content
* length in conn->range_end in order to allow it to send the
* right length to the client.
*
* "Content-Range: bytes *" "/" FMT_OFFSET CRLF
*/
cherokee_buffer_add_str (buffer, "Content-Range: bytes */");
cherokee_buffer_add_ullong10(buffer, (cullong_t)conn->range_end);
cherokee_buffer_add_str (buffer, CRLF);
}
cherokee_buffer_add_str (buffer, "Content-Length: ");
cherokee_buffer_add_ulong10 (buffer, (culong_t) hdl->content.len);
cherokee_buffer_add_str (buffer, CRLF);
}
/* Usual headers
*/
cherokee_buffer_add_str (buffer, "Content-Type: text/html"CRLF);
if (http_type_400(conn->error_code) ||
http_type_500(conn->error_code))
{
cherokee_buffer_add_str (buffer, "Cache-Control: no-cache"CRLF);
cherokee_buffer_add_str (buffer, "Pragma: no-cache"CRLF);
}
return ret_ok;
}
