diff --git a/build/CMakeLists.txt b/build/CMakeLists.txt index 0b96a1e6..779740da 100644 --- a/build/CMakeLists.txt +++ b/build/CMakeLists.txt @@ -110,7 +110,7 @@ set (CLIENT_SRC ) if(WITH_WEBSOCKETS) - list (APPEND CLIENT_SRC "${LIBI2PD_SRC_DIR}/Websocket.cpp") + list (APPEND CLIENT_SRC "${LIBI2PD_CLIENT_SRC_DIR}/Websocket.cpp") endif () add_library(i2pdclient ${CLIENT_SRC}) diff --git a/libi2pd_client/Websocket.cpp b/libi2pd_client/Websocket.cpp index 585677f2..3d456655 100644 --- a/libi2pd_client/Websocket.cpp +++ b/libi2pd_client/Websocket.cpp @@ -1,4 +1,4 @@ -#ifdef USE_EVENTS +#ifdef WITH_EVENTS #include "Websocket.h" #include "Log.h" @@ -22,7 +22,7 @@ namespace i2p typedef websocketpp::server ServerImpl; typedef websocketpp::connection_hdl ServerConn; - + class WebsocketServerImpl : public EventListener { private: @@ -39,20 +39,20 @@ namespace i2p m_server.set_open_handler(std::bind(&WebsocketServerImpl::ConnOpened, this, std::placeholders::_1)); m_server.set_close_handler(std::bind(&WebsocketServerImpl::ConnClosed, this, std::placeholders::_1)); m_server.set_message_handler(std::bind(&WebsocketServerImpl::OnConnMessage, this, std::placeholders::_1, std::placeholders::_2)); - + m_server.listen(boost::asio::ip::address::from_string(addr), port); } ~WebsocketServerImpl() { } - + void Start() { m_run = true; m_server.start_accept(); m_ws_thread = new std::thread([&] () { while(m_run) { - try { + try { m_server.run(); } catch (std::exception & e ) { LogPrint(eLogError, "Websocket server: ", e.what()); @@ -61,7 +61,7 @@ namespace i2p }); m_ev_thread = new std::thread([&] () { while(m_run) { - try { + try { m_Service.run(); break; } catch (std::exception & e ) { @@ -82,7 +82,7 @@ namespace i2p delete m_ev_thread; } m_ev_thread = nullptr; - + if(m_ws_thread) { m_ws_thread->join(); delete m_ws_thread; @@ -95,7 +95,7 @@ namespace i2p std::lock_guard lock(m_connsMutex); m_conns.insert(c); } - + void ConnClosed(ServerConn c) { std::lock_guard lock(m_connsMutex); @@ -123,20 +123,20 @@ namespace i2p LogPrint(eLogDebug, "Websocket schedule tick"); boost::posix_time::seconds dlt(1); m_WebsocketTicker.expires_from_now(dlt); - m_WebsocketTicker.async_wait(std::bind(&WebsocketServerImpl::HandleTick, this, std::placeholders::_1)); + m_WebsocketTicker.async_wait(std::bind(&WebsocketServerImpl::HandleTick, this, std::placeholders::_1)); } - + /** @brief called from m_ev_thread */ void HandlePumpEvent(const EventType & ev, const uint64_t & val) { EventType e; for (const auto & i : ev) e[i.first] = i.second; - + e["number"] = std::to_string(val); HandleEvent(e); } - + /** @brief called from m_ws_thread */ void HandleEvent(const EventType & ev) { @@ -155,7 +155,7 @@ namespace i2p con->send(s); } } - + private: typedef std::set > ConnList; bool m_run; @@ -175,7 +175,7 @@ namespace i2p delete m_impl; } - + void WebsocketServer::Start() { m_impl->Start(); @@ -185,7 +185,7 @@ namespace i2p { m_impl->Stop(); } - + EventListener * WebsocketServer::ToListener() { return m_impl;