1 files changed, 105 insertions, 0 deletions
diff --git a/emu-radio/common-library/websocket-server.cc b/emu-radio/common-library/websocket-server.cc
new file mode 100644
index 00000000..26a2cf76
--- /dev/null
+++ b/emu-radio/common-library/websocket-server.cc
@@ -0,0 +1,105 @@
+/*
+ * 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 (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;
+
+    }
+}
+\ No newline at end of file