mirror of
https://github.com/moparisthebest/spdylay
synced 2024-11-11 12:05:07 -05:00
436 lines
12 KiB
C++
436 lines
12 KiB
C++
/*
|
|
* Spdylay - SPDY Library
|
|
*
|
|
* Copyright (c) 2012 Tatsuhiro Tsujikawa
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining
|
|
* a copy of this software and associated documentation files (the
|
|
* "Software"), to deal in the Software without restriction, including
|
|
* without limitation the rights to use, copy, modify, merge, publish,
|
|
* distribute, sublicense, and/or sell copies of the Software, and to
|
|
* permit persons to whom the Software is furnished to do so, subject to
|
|
* the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be
|
|
* included in all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
|
* LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
|
* OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
|
* WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
*/
|
|
#include "shrpx_downstream.h"
|
|
|
|
#include <cassert>
|
|
|
|
#include "shrpx_upstream.h"
|
|
#include "shrpx_client_handler.h"
|
|
#include "shrpx_config.h"
|
|
#include "shrpx_error.h"
|
|
#include "util.h"
|
|
|
|
using namespace spdylay;
|
|
|
|
namespace shrpx {
|
|
|
|
Downstream::Downstream(Upstream *upstream, int stream_id, int priority)
|
|
: upstream_(upstream),
|
|
bev_(0),
|
|
stream_id_(stream_id),
|
|
priority_(priority),
|
|
ioctrl_(0),
|
|
request_state_(INITIAL),
|
|
chunked_request_(false),
|
|
request_connection_close_(false),
|
|
response_state_(INITIAL),
|
|
response_http_status_(0),
|
|
chunked_response_(false),
|
|
response_htp_(htparser_new()),
|
|
response_body_buf_(0)
|
|
{
|
|
htparser_init(response_htp_, htp_type_response);
|
|
htparser_set_userdata(response_htp_, this);
|
|
event_base *evbase = upstream_->get_client_handler()->get_evbase();
|
|
bev_ = bufferevent_socket_new
|
|
(evbase, -1,
|
|
BEV_OPT_CLOSE_ON_FREE | BEV_OPT_DEFER_CALLBACKS);
|
|
ioctrl_.set_bev(bev_);
|
|
}
|
|
|
|
Downstream::~Downstream()
|
|
{
|
|
if(ENABLE_LOG) {
|
|
LOG(INFO) << "Deleting downstream" << this;
|
|
}
|
|
if(response_body_buf_) {
|
|
// Passing NULL to evbuffer_free() causes segmentation fault.
|
|
evbuffer_free(response_body_buf_);
|
|
}
|
|
bufferevent_disable(bev_, EV_READ | EV_WRITE);
|
|
bufferevent_free(bev_);
|
|
free(response_htp_);
|
|
if(ENABLE_LOG) {
|
|
LOG(INFO) << "Deleted";
|
|
}
|
|
}
|
|
|
|
void Downstream::pause_read(IOCtrlReason reason)
|
|
{
|
|
ioctrl_.pause_read(reason);
|
|
}
|
|
|
|
bool Downstream::resume_read(IOCtrlReason reason)
|
|
{
|
|
return ioctrl_.resume_read(reason);
|
|
}
|
|
|
|
namespace {
|
|
void check_transfer_encoding_chunked(bool *chunked,
|
|
const Headers::value_type &item)
|
|
{
|
|
if(util::strieq(item.first.c_str(), "transfer-encoding")) {
|
|
if(util::strifind(item.second.c_str(), "chunked")) {
|
|
*chunked = true;
|
|
}
|
|
}
|
|
}
|
|
} // namespace
|
|
|
|
namespace {
|
|
void check_request_connection(bool *connection_close,
|
|
const Headers::value_type &item)
|
|
{
|
|
if(util::strieq(item.first.c_str(), "connection")) {
|
|
if(util::strifind(item.second.c_str(), "close")) {
|
|
*connection_close = true;
|
|
}
|
|
}
|
|
}
|
|
} // namespace
|
|
|
|
void Downstream::add_request_header(const std::string& name,
|
|
const std::string& value)
|
|
{
|
|
request_headers_.push_back(std::make_pair(name, value));
|
|
}
|
|
|
|
void Downstream::set_last_request_header_value(const std::string& value)
|
|
{
|
|
Headers::value_type &item = request_headers_.back();
|
|
item.second = value;
|
|
check_transfer_encoding_chunked(&chunked_request_, item);
|
|
check_request_connection(&request_connection_close_, item);
|
|
}
|
|
|
|
void Downstream::set_request_method(const std::string& method)
|
|
{
|
|
request_method_ = method;
|
|
}
|
|
|
|
void Downstream::set_request_path(const std::string& path)
|
|
{
|
|
request_path_ = path;
|
|
}
|
|
|
|
Upstream* Downstream::get_upstream() const
|
|
{
|
|
return upstream_;
|
|
}
|
|
|
|
int32_t Downstream::get_stream_id() const
|
|
{
|
|
return stream_id_;
|
|
}
|
|
|
|
int Downstream::start_connection()
|
|
{
|
|
bufferevent_setcb(bev_,
|
|
upstream_->get_downstream_readcb(),
|
|
upstream_->get_downstream_writecb(),
|
|
upstream_->get_downstream_eventcb(), this);
|
|
bufferevent_enable(bev_, EV_READ | EV_WRITE);
|
|
bufferevent_set_timeouts(bev_,
|
|
&get_config()->downstream_read_timeout,
|
|
&get_config()->downstream_write_timeout);
|
|
bufferevent_socket_connect
|
|
(bev_,
|
|
// TODO maybe not thread-safe?
|
|
const_cast<sockaddr*>(&get_config()->downstream_addr.sa),
|
|
get_config()->downstream_addrlen);
|
|
return 0;
|
|
}
|
|
|
|
void Downstream::set_request_state(int state)
|
|
{
|
|
request_state_ = state;
|
|
}
|
|
|
|
int Downstream::get_request_state() const
|
|
{
|
|
return request_state_;
|
|
}
|
|
|
|
bool Downstream::get_chunked_request() const
|
|
{
|
|
return chunked_request_;
|
|
}
|
|
|
|
bool Downstream::get_request_connection_close() const
|
|
{
|
|
return request_connection_close_;
|
|
}
|
|
|
|
void Downstream::set_request_connection_close(bool f)
|
|
{
|
|
request_connection_close_ = f;
|
|
}
|
|
|
|
int Downstream::push_request_headers()
|
|
{
|
|
bool xff_found = false;
|
|
std::string hdrs = request_method_;
|
|
hdrs += " ";
|
|
hdrs += request_path_;
|
|
hdrs += " ";
|
|
hdrs += "HTTP/1.1\r\n";
|
|
hdrs += "Host: ";
|
|
hdrs += get_config()->downstream_hostport;
|
|
hdrs += "\r\n";
|
|
// TODO Rewrite user-agent?
|
|
for(Headers::const_iterator i = request_headers_.begin();
|
|
i != request_headers_.end(); ++i) {
|
|
if(util::strieq((*i).first.c_str(), "X-Forwarded-Proto")) {
|
|
continue;
|
|
}
|
|
if(util::strieq((*i).first.c_str(), "user-agent")) {
|
|
hdrs += "User-Agent: ";
|
|
hdrs += get_config()->server_name;
|
|
} else {
|
|
hdrs += (*i).first;
|
|
hdrs += ": ";
|
|
hdrs += (*i).second;
|
|
if(!xff_found && util::strieq((*i).first.c_str(), "X-Forwarded-For")) {
|
|
xff_found = true;
|
|
hdrs += ", ";
|
|
hdrs += upstream_->get_client_handler()->get_ipaddr();
|
|
}
|
|
}
|
|
hdrs += "\r\n";
|
|
}
|
|
hdrs += "Connection: close\r\n";
|
|
if(!xff_found) {
|
|
hdrs += "X-Forwarded-For: ";
|
|
hdrs += upstream_->get_client_handler()->get_ipaddr();
|
|
hdrs += "\r\n";
|
|
}
|
|
hdrs += "X-Forwarded-Proto: https\r\n";
|
|
|
|
hdrs += "\r\n";
|
|
if(ENABLE_LOG) {
|
|
LOG(INFO) << "<downstream> request headers\n" << hdrs;
|
|
}
|
|
evbuffer *output = bufferevent_get_output(bev_);
|
|
evbuffer_add(output, hdrs.c_str(), hdrs.size());
|
|
return 0;
|
|
}
|
|
|
|
int Downstream::push_upload_data_chunk(const uint8_t *data, size_t datalen)
|
|
{
|
|
// Assumes that request headers have already been pushed to output
|
|
// buffer using push_request_headers().
|
|
ssize_t res = 0;
|
|
int rv;
|
|
evbuffer *output = bufferevent_get_output(bev_);
|
|
if(chunked_request_) {
|
|
char chunk_size_hex[16];
|
|
rv = snprintf(chunk_size_hex, sizeof(chunk_size_hex), "%X\r\n",
|
|
static_cast<unsigned int>(datalen));
|
|
evbuffer_add(output, chunk_size_hex, rv);
|
|
res += rv;
|
|
}
|
|
rv = evbuffer_add(output, data, datalen);
|
|
if(rv == -1) {
|
|
return -1;
|
|
}
|
|
res += rv;
|
|
return res;
|
|
}
|
|
|
|
int Downstream::end_upload_data()
|
|
{
|
|
if(chunked_request_) {
|
|
evbuffer *output = bufferevent_get_output(bev_);
|
|
evbuffer_add(output, "0\r\n\r\n", 5);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
const Headers& Downstream::get_response_headers() const
|
|
{
|
|
return response_headers_;
|
|
}
|
|
|
|
void Downstream::add_response_header(const std::string& name,
|
|
const std::string& value)
|
|
{
|
|
response_headers_.push_back(std::make_pair(name, value));
|
|
}
|
|
|
|
void Downstream::set_last_response_header_value(const std::string& value)
|
|
{
|
|
Headers::value_type &item = response_headers_.back();
|
|
item.second = value;
|
|
check_transfer_encoding_chunked(&chunked_response_, item);
|
|
}
|
|
|
|
unsigned int Downstream::get_response_http_status() const
|
|
{
|
|
return response_http_status_;
|
|
}
|
|
|
|
void Downstream::set_response_http_status(unsigned int status)
|
|
{
|
|
response_http_status_ = status;
|
|
}
|
|
|
|
bool Downstream::get_chunked_response() const
|
|
{
|
|
return chunked_response_;
|
|
}
|
|
|
|
namespace {
|
|
int htp_hdrs_completecb(htparser *htp)
|
|
{
|
|
Downstream *downstream;
|
|
downstream = reinterpret_cast<Downstream*>(htparser_get_userdata(htp));
|
|
downstream->set_response_http_status(htparser_get_status(htp));
|
|
downstream->set_response_state(Downstream::HEADER_COMPLETE);
|
|
downstream->get_upstream()->on_downstream_header_complete(downstream);
|
|
return 0;
|
|
}
|
|
} // namespace
|
|
|
|
namespace {
|
|
int htp_hdr_keycb(htparser *htp, const char *data, size_t len)
|
|
{
|
|
Downstream *downstream;
|
|
downstream = reinterpret_cast<Downstream*>(htparser_get_userdata(htp));
|
|
downstream->add_response_header(std::string(data, len), "");
|
|
return 0;
|
|
}
|
|
} // namespace
|
|
|
|
namespace {
|
|
int htp_hdr_valcb(htparser *htp, const char *data, size_t len)
|
|
{
|
|
Downstream *downstream;
|
|
downstream = reinterpret_cast<Downstream*>(htparser_get_userdata(htp));
|
|
downstream->set_last_response_header_value(std::string(data, len));
|
|
return 0;
|
|
}
|
|
} // namespace
|
|
|
|
namespace {
|
|
int htp_bodycb(htparser *htp, const char *data, size_t len)
|
|
{
|
|
Downstream *downstream;
|
|
downstream = reinterpret_cast<Downstream*>(htparser_get_userdata(htp));
|
|
downstream->get_upstream()->on_downstream_body
|
|
(downstream, reinterpret_cast<const uint8_t*>(data), len);
|
|
return 0;
|
|
}
|
|
} // namespace
|
|
|
|
namespace {
|
|
int htp_body_completecb(htparser *htp)
|
|
{
|
|
Downstream *downstream;
|
|
downstream = reinterpret_cast<Downstream*>(htparser_get_userdata(htp));
|
|
downstream->set_response_state(Downstream::MSG_COMPLETE);
|
|
downstream->get_upstream()->on_downstream_body_complete(downstream);
|
|
return 0;
|
|
}
|
|
} // namespace
|
|
|
|
namespace {
|
|
htparse_hooks htp_hooks = {
|
|
0, /*htparse_hook on_msg_begin;*/
|
|
0, /*htparse_data_hook method;*/
|
|
0, /* htparse_data_hook scheme;*/
|
|
0, /* htparse_data_hook host; */
|
|
0, /* htparse_data_hook port; */
|
|
0, /* htparse_data_hook path; */
|
|
0, /* htparse_data_hook args; */
|
|
0, /* htparse_data_hook uri; */
|
|
0, /* htparse_hook on_hdrs_begin; */
|
|
htp_hdr_keycb, /* htparse_data_hook hdr_key; */
|
|
htp_hdr_valcb, /* htparse_data_hook hdr_val; */
|
|
htp_hdrs_completecb, /* htparse_hook on_hdrs_complete; */
|
|
0, /*htparse_hook on_new_chunk;*/
|
|
0, /*htparse_hook on_chunk_complete;*/
|
|
0, /*htparse_hook on_chunks_complete;*/
|
|
htp_bodycb, /* htparse_data_hook body; */
|
|
htp_body_completecb /* htparse_hook on_msg_complete;*/
|
|
};
|
|
} // namespace
|
|
|
|
int Downstream::parse_http_response()
|
|
{
|
|
evbuffer *input = bufferevent_get_input(bev_);
|
|
unsigned char *mem = evbuffer_pullup(input, -1);
|
|
size_t nread = htparser_run(response_htp_, &htp_hooks,
|
|
reinterpret_cast<const char*>(mem),
|
|
evbuffer_get_length(input));
|
|
evbuffer_drain(input, nread);
|
|
if(htparser_get_error(response_htp_) == htparse_error_none) {
|
|
return 0;
|
|
} else {
|
|
if(ENABLE_LOG) {
|
|
LOG(INFO) << "<downstream> http parser failure: "
|
|
<< htparser_get_strerror(response_htp_);
|
|
}
|
|
return SHRPX_ERR_HTTP_PARSE;
|
|
}
|
|
}
|
|
|
|
void Downstream::set_response_state(int state)
|
|
{
|
|
response_state_ = state;
|
|
}
|
|
|
|
int Downstream::get_response_state() const
|
|
{
|
|
return response_state_;
|
|
}
|
|
|
|
namespace {
|
|
void body_buf_cb(evbuffer *body, size_t oldlen, size_t newlen, void *arg)
|
|
{
|
|
Downstream *downstream = reinterpret_cast<Downstream*>(arg);
|
|
if(newlen == 0) {
|
|
downstream->resume_read(SHRPX_NO_BUFFER);
|
|
}
|
|
}
|
|
} // namespace
|
|
|
|
int Downstream::init_response_body_buf()
|
|
{
|
|
assert(response_body_buf_ == 0);
|
|
response_body_buf_ = evbuffer_new();
|
|
if(response_body_buf_ == 0) {
|
|
DIE();
|
|
}
|
|
evbuffer_setcb(response_body_buf_, body_buf_cb, this);
|
|
return 0;
|
|
}
|
|
|
|
evbuffer* Downstream::get_response_body_buf()
|
|
{
|
|
return response_body_buf_;
|
|
}
|
|
|
|
} // namespace shrpx
|