67 lines
1.5 KiB
C++
67 lines
1.5 KiB
C++
|
#include "mqtt_server.h"
|
||
|
|
|
||
|
|
void MQTTServer::initialize() {
|
||
|
|
evloop = event_base_new();
|
||
|
|
|
||
|
|
if (!evloop) {
|
||
|
|
printf("Could not initialize libevent\n");
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
std::memset(&sin, 0, sizeof(sin));
|
||
|
|
sin.sin_family = AF_INET;
|
||
|
|
evutil_inet_pton(sin.sin_family, bind_address.c_str(), &sin.sin_addr);
|
||
|
|
sin.sin_port = htons(port);
|
||
|
|
|
||
|
|
listener = evconnlistener_new_bind(evloop, MQTTServer::listener_cb,
|
||
|
|
(void *)this,
|
||
|
|
LEV_OPT_REUSEABLE | LEV_OPT_CLOSE_ON_FREE, -1,
|
||
|
|
(struct sockaddr *)&sin, sizeof(sin));
|
||
|
|
|
||
|
|
if (!listener) {
|
||
|
|
std::cerr << "Could not create listener!\n";
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void MQTTServer::listener_cb(struct evconnlistener *listener, evutil_socket_t fd, struct sockaddr *sa, int socklen, void *user_data) {
|
||
|
|
|
||
|
|
MQTTServer *server = static_cast<MQTTServer *>(user_data);
|
||
|
|
|
||
|
|
struct bufferevent *bev;
|
||
|
|
|
||
|
|
bev = bufferevent_socket_new(server->evloop, fd, BEV_OPT_CLOSE_ON_FREE);
|
||
|
|
if (!bev) {
|
||
|
|
std::cerr << "Error constructing bufferevent!\n";
|
||
|
|
event_base_loopbreak(server->evloop);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
server->session_manager->accept_connection(bev);
|
||
|
|
}
|
||
|
|
|
||
|
|
void MQTTServer::loop_once() {
|
||
|
|
event_base_dispatch(evloop);
|
||
|
|
}
|
||
|
|
|
||
|
|
MQTTServer::MQTTServer() {
|
||
|
|
bind_address = "0";
|
||
|
|
port = 1883;
|
||
|
|
|
||
|
|
session_manager = new SessionManager();
|
||
|
|
|
||
|
|
evloop = nullptr;
|
||
|
|
listener = nullptr;
|
||
|
|
}
|
||
|
|
|
||
|
|
MQTTServer::~MQTTServer() {
|
||
|
|
if (event_base_loopexit(evloop, NULL)) {
|
||
|
|
std::cerr << "failed to exit event loop\n";
|
||
|
|
}
|
||
|
|
|
||
|
|
evconnlistener_free(listener);
|
||
|
|
event_base_free(evloop);
|
||
|
|
|
||
|
|
delete session_manager;
|
||
|
|
}
|