cpp/funstraw
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Switch from epoll to libevent, to add OS X support.

Browse Source
This commit is contained in:
Ian Gulliver
2015-02-07 16:44:42 +01:00
parent 2c9b9f97de
commit 88c3c69e52
4 changed files with 64 additions and 98 deletions
Download Patch File Download Diff File Expand all files Collapse all files

103
crypto.cc
View File

@@ -13,15 +13,6 @@
#include "crypto.h"
CryptoBase::CryptoBase(const int fd)
: fd_(fd) {}
CryptoBase::~CryptoBase() {
if (close(fd_)) {
perror("close");
}
}
void CryptoBase::GenKey(std::string* key) {
unsigned char buf[crypto_secretbox_KEYBYTES];
randombytes_buf(buf, crypto_secretbox_KEYBYTES);
@@ -37,72 +28,64 @@ void CryptoBase::GenKeyPair(std::string* secret_key, std::string* public_key) {
}
CryptoPubServer::CryptoPubServer(const int fd, const std::string secret_key)
: CryptoBase(fd),
secret_key_(secret_key),
epoll_fd_(epoll_create(256)) {
epoll_event event = {
.events = EPOLLIN,
.data = {
.ptr = this,
},
};
epoll_ctl(epoll_fd_, EPOLL_CTL_ADD, fd, &event);
CryptoPubServer::CryptoPubServer(const std::string secret_key)
: secret_key_(secret_key),
event_base_(event_base_new()) {
struct sockaddr_in6 server_addr = {0};
server_addr.sin6_family = AF_INET6;
server_addr.sin6_addr = in6addr_any;
server_addr.sin6_port = htons(4990);
listener_ = evconnlistener_new_bind(event_base_, &CryptoPubServer::OnNewConn, this, LEV_OPT_REUSEABLE|LEV_OPT_CLOSE_ON_FREE, -1, (struct sockaddr*)&server_addr, sizeof(server_addr));
}
CryptoPubServer::~CryptoPubServer() {
if (close(epoll_fd_)) {
perror("close");
}
evconnlistener_free(listener_);
event_base_free(event_base_);
}
void CryptoPubServer::OnReadable() {
struct sockaddr_in6 client;
socklen_t client_len = sizeof(client);
auto client_fd = accept(fd_, (struct sockaddr*) &client, &client_len);
if (client_fd == -1) {
perror("accept");
return;
}
void CryptoPubServer::OnNewConn(struct evconnlistener* listener, int client_fd, struct sockaddr* client_addr_, int client_addrlen, void* this__) {
auto this_ = (CryptoPubServer*)this__;
assert(client_addr_->sa_family == AF_INET6);
auto client_addr = (struct sockaddr_in6*)client_addr_;
char buf[128];
inet_ntop(AF_INET6, &client.sin6_addr, buf, 128);
std::cerr << "New connection from [" << buf << "]:" << ntohs(client.sin6_port) << std::endl;
auto peer = new CryptoPubServerConnection(client_fd, secret_key_);
epoll_event event = {
.events = EPOLLIN,
.data = {
.ptr = peer,
},
};
epoll_ctl(epoll_fd_, EPOLL_CTL_ADD, client_fd, &event);
inet_ntop(AF_INET6, &client_addr->sin6_addr, buf, 128);
std::cerr << "New connection from [" << buf << "]:" << ntohs(client_addr->sin6_port) << std::endl;
auto bev = bufferevent_socket_new(this_->event_base_, client_fd, BEV_OPT_CLOSE_ON_FREE);
bufferevent_enable(bev, EV_READ);
bufferevent_disable(bev, EV_WRITE);
auto peer = new CryptoPubServerConnection(bev, this_->secret_key_);
bufferevent_setcb(bev, &CryptoPubServerConnection::OnReadable, NULL, &CryptoPubServerConnection::OnError, peer);
}
void CryptoPubServer::Loop() {
while (1) {
epoll_event events[10];
auto num_events = epoll_wait(epoll_fd_, events, 10, -1);
for (int i = 0; i < num_events; i++) {
if (events[i].events & EPOLLIN) {
auto obj = (CryptoBase*) events[i].data.ptr;
obj->OnReadable();
}
}
}
event_base_dispatch(event_base_);
}
CryptoPubServerConnection::CryptoPubServerConnection(const int fd, const std::string secret_key)
: CryptoBase(fd),
CryptoPubServerConnection::CryptoPubServerConnection(struct bufferevent* bev, const std::string secret_key)
: bev_(bev),
secret_key_(secret_key),
state_(AWAITING_HANDSHAKE) {
}
void CryptoPubServerConnection::OnReadable() {
char buf[128];
if (read(fd_, buf, 128) == 0) {
delete this;
return;
}
CryptoPubServerConnection::~CryptoPubServerConnection() {
std::cerr << "Connection closed" << std::endl;
bufferevent_free(bev_);
}
void CryptoPubServerConnection::OnReadable(struct bufferevent* bev, void* this__) {
std::cerr << "OnReadable" << std::endl;
auto this_ = (CryptoPubServerConnection*)this__;
char buf[128];
bufferevent_read(bev, buf, 128);
}
void CryptoPubServerConnection::OnError(struct bufferevent* bev, const short what, void* this__) {
std::cerr << "OnError" << std::endl;
auto this_ = (CryptoPubServerConnection*)this__;
delete this_;
}