nghttp2/src/shrpx_thread_event_receiver.cc

92 lines
3.2 KiB
C++
Raw Normal View History

2012-06-05 18:26:04 +02:00
/*
2014-03-30 12:09:21 +02:00
* nghttp2 - HTTP/2 C Library
2012-06-05 18:26:04 +02:00
*
* 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_thread_event_receiver.h"
2012-06-08 14:40:03 +02:00
#include <unistd.h>
2012-06-05 18:26:04 +02:00
#include "shrpx_ssl.h"
#include "shrpx_log.h"
#include "shrpx_client_handler.h"
#include "shrpx_http2_session.h"
2012-06-05 18:26:04 +02:00
namespace shrpx {
ThreadEventReceiver::ThreadEventReceiver(event_base *evbase,
SSL_CTX *ssl_ctx,
Http2Session *http2session)
: evbase_(evbase),
ssl_ctx_(ssl_ctx),
http2session_(http2session),
rate_limit_group_(bufferevent_rate_limit_group_new
(evbase_, get_config()->worker_rate_limit_cfg))
2012-06-05 18:26:04 +02:00
{}
ThreadEventReceiver::~ThreadEventReceiver()
{
bufferevent_rate_limit_group_free(rate_limit_group_);
}
2012-06-05 18:26:04 +02:00
void ThreadEventReceiver::on_read(bufferevent *bev)
{
2013-12-20 15:36:24 +01:00
auto input = bufferevent_get_input(bev);
2012-06-05 18:26:04 +02:00
while(evbuffer_get_length(input) >= sizeof(WorkerEvent)) {
WorkerEvent wev;
int nread = evbuffer_remove(input, &wev, sizeof(wev));
if(nread == -1) {
TLOG(FATAL, this) << "evbuffer_remove() failed";
continue;
}
if(nread != sizeof(wev)) {
TLOG(FATAL, this) << "evbuffer_remove() removed fewer bytes. Expected:"
<< sizeof(wev) << " Actual:" << nread;
continue;
}
if(LOG_ENABLED(INFO)) {
TLOG(INFO, this) << "WorkerEvent: client_fd=" << wev.client_fd
<< ", addrlen=" << wev.client_addrlen;
2012-06-05 18:26:04 +02:00
}
2013-12-20 15:36:24 +01:00
auto evbase = bufferevent_get_base(bev);
auto client_handler = ssl::accept_connection(evbase, rate_limit_group_,
ssl_ctx_,
2013-12-20 15:36:24 +01:00
wev.client_fd,
&wev.client_addr.sa,
wev.client_addrlen);
2012-06-05 18:26:04 +02:00
if(client_handler) {
client_handler->set_http2_session(http2session_);
if(LOG_ENABLED(INFO)) {
TLOG(INFO, this) << "CLIENT_HANDLER:" << client_handler << " created";
2012-06-05 18:26:04 +02:00
}
} else {
if(LOG_ENABLED(INFO)) {
TLOG(ERROR, this) << "ClientHandler creation failed";
2012-06-05 18:26:04 +02:00
}
close(wev.client_fd);
}
}
}
} // namespace shrpx