blob: 3d64d15b49919686935986920695ea0cc3d3564d [file] [log] [blame]
// Copyright 2015 The Weave Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "examples/provider/event_http_server.h"
#include <vector>
#include <base/bind.h>
#include <base/time/time.h>
#include <event2/bufferevent_ssl.h>
#include <evhtp.h>
#include <openssl/err.h>
#include "examples/provider/event_task_runner.h"
namespace weave {
namespace examples {
namespace {
std::string GetSslError() {
char error[1000] = {};
ERR_error_string_n(ERR_get_error(), error, sizeof(error));
return error;
}
} // namespace
class HttpServerImpl::RequestImpl : public Request {
public:
RequestImpl(EventPtr<evhtp_request_t> req) : req_(std::move(req)) {
evbuf_t* input_buffer = req_.get()->buffer_in;
data_.resize(evbuffer_get_length(input_buffer));
evbuffer_remove(input_buffer, &data_[0], data_.size());
}
~RequestImpl() {}
std::string GetPath() const override { return req_->uri->path->full; }
std::string GetFirstHeader(const std::string& name) const override {
const char* header = evhtp_header_find(req_->headers_in, name.c_str());
if (!header)
return {};
return header;
}
std::string GetData() { return data_; }
void SendReply(int status_code,
const std::string& data,
const std::string& mime_type) override {
EventPtr<evbuffer> buf{evbuffer_new()};
evbuffer_add(buf.get(), data.data(), data.size());
evhtp_header_key_add(req_->headers_out, "Content-Type", 0);
evhtp_header_val_add(req_->headers_out, mime_type.c_str(), 1);
evhtp_header_key_add(req_->headers_out, "Content-Length", 0);
std::string content_length = std::to_string(evbuffer_get_length(buf.get()));
evhtp_header_val_add(req_->headers_out, content_length.c_str(), 1);
evhtp_send_reply_start(req_.get(), status_code);
evhtp_send_reply_body(req_.get(), buf.get());
evhtp_send_reply_end(req_.get());
}
private:
EventPtr<evhtp_request_t> req_;
std::string data_;
};
HttpServerImpl::HttpServerImpl(EventTaskRunner* task_runner)
: task_runner_{task_runner} {
SSL_load_error_strings();
SSL_library_init();
std::unique_ptr<SSL_CTX, decltype(&SSL_CTX_free)> ctx{
SSL_CTX_new(TLSv1_2_server_method()), &SSL_CTX_free};
CHECK(ctx);
SSL_CTX_set_options(ctx.get(), SSL_OP_SINGLE_DH_USE | SSL_OP_SINGLE_ECDH_USE |
SSL_OP_NO_SSLv2);
std::unique_ptr<EC_KEY, decltype(&EC_KEY_free)> ec_key{
EC_KEY_new_by_curve_name(NID_X9_62_prime256v1), &EC_KEY_free};
CHECK(ec_key) << GetSslError();
CHECK_EQ(1, SSL_CTX_set_tmp_ecdh(ctx.get(), ec_key.get())) << GetSslError();
std::unique_ptr<X509, decltype(&X509_free)> x509{X509_new(), &X509_free};
CHECK(x509);
std::unique_ptr<EVP_PKEY, decltype(&EVP_PKEY_free)> pkey{EVP_PKEY_new(),
&EVP_PKEY_free};
CHECK(pkey);
GenerateX509(x509.get(), pkey.get());
CHECK_EQ(1, SSL_CTX_use_PrivateKey(ctx.get(), pkey.get())) << GetSslError();
CHECK_EQ(1, SSL_CTX_use_certificate(ctx.get(), x509.get())) << GetSslError();
CHECK_EQ(1, SSL_CTX_check_private_key(ctx.get())) << GetSslError();
httpd_.reset(evhtp_new(task_runner_->GetEventBase(), nullptr));
CHECK(httpd_);
httpsd_.reset(evhtp_new(task_runner_->GetEventBase(), nullptr));
CHECK(httpsd_);
httpsd_.get()->ssl_ctx = ctx.release();
CHECK_EQ(0, evhtp_bind_socket(httpd_.get(), "0.0.0.0", GetHttpPort(), -1));
CHECK_EQ(0, evhtp_bind_socket(httpsd_.get(), "0.0.0.0", GetHttpsPort(), -1));
}
void HttpServerImpl::GenerateX509(X509* x509, EVP_PKEY* pkey) {
CHECK(x509) << GetSslError();
X509_set_version(x509, 2);
X509_gmtime_adj(X509_get_notBefore(x509), 0);
X509_gmtime_adj(X509_get_notAfter(x509),
base::TimeDelta::FromDays(365).InSeconds());
CHECK(pkey) << GetSslError();
std::unique_ptr<BIGNUM, decltype(&BN_free)> big_num(BN_new(), &BN_free);
CHECK(BN_set_word(big_num.get(), 65537)) << GetSslError();
auto rsa = RSA_new();
RSA_generate_key_ex(rsa, 2048, big_num.get(), nullptr);
CHECK(EVP_PKEY_assign_RSA(pkey, rsa)) << GetSslError();
X509_set_pubkey(x509, pkey);
CHECK(X509_sign(x509, pkey, EVP_sha256())) << GetSslError();
cert_fingerprint_.resize(EVP_MD_size(EVP_sha256()));
uint32_t len = 0;
CHECK(X509_digest(x509, EVP_sha256(), cert_fingerprint_.data(), &len));
CHECK_EQ(len, cert_fingerprint_.size());
}
void HttpServerImpl::NotFound(evhtp_request_t* req) {
EventPtr<evbuffer> buf{evbuffer_new()};
evbuffer_add_printf(buf.get(), "404 Not Found: %s\n", req->uri->path->full);
evhtp_send_reply_start(req, 404);
evhtp_send_reply_body(req, buf.get());
evhtp_send_reply_end(req);
}
void HttpServerImpl::ProcessRequest(evhtp_request_t* req) {
std::unique_ptr<RequestImpl> request{new RequestImpl{EventPtr<evhtp_request_t>{req}}};
std::string path = request->GetPath();
auto it = handlers_.find(std::make_pair(path, req->htp));
if (it != handlers_.end()) {
return it->second.Run(std::move(request));
}
NotFound(req);
}
void HttpServerImpl::ProcessRequestCallback(evhtp_request_t* req, void* arg) {
static_cast<HttpServerImpl*>(arg)->ProcessRequest(req);
}
void HttpServerImpl::AddHttpRequestHandler(
const std::string& path,
const RequestHandlerCallback& callback) {
handlers_[std::make_pair(path, httpd_.get())] = callback;
evhtp_set_cb(httpd_.get(), path.c_str(), &ProcessRequestCallback, this);
}
void HttpServerImpl::AddHttpsRequestHandler(
const std::string& path,
const RequestHandlerCallback& callback) {
handlers_[std::make_pair(path, httpsd_.get())] = callback;
evhtp_set_cb(httpsd_.get(), path.c_str(), &ProcessRequestCallback, this);
}
void HttpServerImpl::RemoveHttpRequestHandler(const std::string& path) {
handlers_.erase(std::make_pair(path, httpd_.get()));
evhtp_set_cb(httpd_.get(), path.c_str(), nullptr, nullptr);
}
void HttpServerImpl::RemoveHttpsRequestHandler(const std::string& path) {
handlers_.erase(std::make_pair(path, httpsd_.get()));
evhtp_set_cb(httpsd_.get(), path.c_str(), nullptr, nullptr);
}
void HttpServerImpl::ProcessReply(std::shared_ptr<RequestImpl> request,
int status_code,
const std::string& data,
const std::string& mime_type) {}
uint16_t HttpServerImpl::GetHttpPort() const {
return 7780;
}
uint16_t HttpServerImpl::GetHttpsPort() const {
return 7781;
}
base::TimeDelta HttpServerImpl::GetRequestTimeout() const {
return base::TimeDelta::Max();
}
std::vector<uint8_t> HttpServerImpl::GetHttpsCertificateFingerprint() const {
return cert_fingerprint_;
}
} // namespace examples
} // namespace weave