1 files changed, 95 insertions, 0 deletions
diff --git a/Websocket/websocket-server.cpp b/Websocket/websocket-server.cpp
new file mode 100644
index 00000000..ee79ccdb
--- /dev/null
+++ b/Websocket/websocket-server.cpp
@@ -0,0 +1,95 @@
+/*
+ * Copyright (c) 2017 Cisco and/or its affiliates.
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at:
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "websocket-server.h"
+
+WebSocketServer::WebSocketServer(unsigned short port)
+      : m_isRunning(false)
+{
+  try {
+    // Initialize Asio
+    server.init_asio();
+
+    // Set reuse address option
+    server.set_reuse_addr(true);
+
+    // Register our message handler
+    server.set_message_handler(bind(&WebSocketServer::onMessage, this, &server, ::_1, ::_2));
+
+    // Listen on port
+    server.listen(boost::asio::ip::tcp::v4(),port);
+
+  } catch (websocketpp::exception const & e) {
+    std::cout << e.what() << std::endl;
+  } catch (...) {
+    std::cout << "Exception occurred.." << std::endl;
+  }
+}
+
+void
+WebSocketServer::setHandler(const HandlerFunction &handler)
+{
+  this->handler = handler;
+}
+
+void
+WebSocketServer::start()
+{
+  m_isRunning = true;
+  int retries = 5;
+
+  while (m_isRunning && retries > 0) {
+
+    try {
+      // Start the server accept loop
+      server.start_accept();
+
+      //Set interrupt callbacks
+
+      boost::asio::io_service io_service;
+      boost::asio::signal_set signals(server.get_io_service(), SIGINT, SIGQUIT);
+
+      signals.async_wait([this](const boost::system::error_code &errorCode, int) {
+        std::cout << "Gracefully terminating websocket server" << std::endl;
+        this->m_isRunning = false;
+        this->server.stop();
+      });
+
+      // Start the ASIO io_service run loop
+      server.run();
+
+    } catch (websocketpp::exception const &e) {
+      std::cout << e.what() << std::endl;
+    }
+    catch (...) {
+      std::cout << "Exception occurred.." << std::endl;
+    }
+
+    retries--;
+  }
+}
+
+void
+WebSocketServer::onMessage(Server *s, websocketpp::connection_hdl hdl, message_ptr msg)
+{
+  try {
+    handler(s, hdl, msg, (uint8_t *) msg->get_payload().c_str(), msg->get_payload().size());
+  } catch (const websocketpp::lib::error_code &e) {
+
+    std::cout << "Reply sent failed because: " << e
+              << "(" << e.message() << ")" << std::endl;
+
+  }
+}