Refactoring project layout

2025-11-08 14:43:21 +00:00 · 2021-10-19 19:45:50 -07:00
parent 816f0a6e7c
commit 2a0b45fde1
110 changed files with 1151 additions and 230 deletions
--- a/src/RESTAPI/RESTAPI_webSocketServer.cpp
+++ b/src/RESTAPI/RESTAPI_webSocketServer.cpp
@@ -0,0 +1,140 @@
+//
+// Created by stephane bourque on 2021-08-12.
+//
+
+#include "RESTAPI_webSocketServer.h"
+#include "Poco/Net/WebSocket.h"
+#include "Poco/Net/HTTPResponse.h"
+#include "Poco/JSON/Object.h"
+#include "Poco/JSON/Parser.h"
+#include "Poco/JSON/Stringifier.h"
+
+#include "SerialNumberCache.h"
+
+#include "framework/AuthClient.h"
+#include "framework/Utils.h"
+
+namespace OpenWifi {
+
+	void RESTAPI_webSocketServer::DoGet() {
+
+		//	try and upgrade this session to websocket...
+		if(Request->find("Upgrade") != Request->end() && Poco::icompare((*Request)["Upgrade"], "websocket") == 0) {
+			try
+			{
+				Poco::Net::WebSocket WS(*Request, *Response);
+				Logger_.information("WebSocket connection established.");
+				int flags;
+				int n;
+				bool Authenticated=false;
+				bool Done=false;
+				do
+				{
+					Poco::Buffer<char>			IncomingFrame(0);
+					n = WS.receiveFrame(IncomingFrame, flags);
+					auto Op = flags & Poco::Net::WebSocket::FRAME_OP_BITMASK;
+					switch(Op) {
+						case Poco::Net::WebSocket::FRAME_OP_PING: {
+							WS.sendFrame("", 0,
+										   (int)Poco::Net::WebSocket::FRAME_OP_PONG |
+										   (int)Poco::Net::WebSocket::FRAME_FLAG_FIN);
+							}
+							break;
+						case Poco::Net::WebSocket::FRAME_OP_PONG: {
+							}
+							break;
+						case Poco::Net::WebSocket::FRAME_OP_TEXT: {
+								IncomingFrame.append(0);
+								if(!Authenticated) {
+									std::string Frame{IncomingFrame.begin()};
+									auto Tokens = Utils::Split(Frame,':');
+									if(Tokens.size()==2 && AuthClient()->IsTokenAuthorized(Tokens[1], UserInfo_)) {
+										Authenticated=true;
+										std::string S{"Welcome! Bienvenue! Bienvenidos!"};
+										WS.sendFrame(S.c_str(),S.size());
+									} else {
+										std::string S{"Invalid token. Closing connection."};
+										WS.sendFrame(S.c_str(),S.size());
+										Done=true;
+									}
+
+								} else {
+									try {
+										Poco::JSON::Parser P;
+										auto Obj = P.parse(IncomingFrame.begin())
+													   .extract<Poco::JSON::Object::Ptr>();
+										std::string Answer;
+										Process(Obj, Answer);
+										if (!Answer.empty())
+											WS.sendFrame(Answer.c_str(), Answer.size());
+										else {
+											WS.sendFrame("{}", 2);
+										}
+									} catch (const Poco::JSON::JSONException & E) {
+										Logger_.log(E);
+									}
+								}
+							}
+							break;
+						default:
+							{
+
+							}
+					}
+				}
+				while (!Done && (n > 0 && (flags & Poco::Net::WebSocket::FRAME_OP_BITMASK) != Poco::Net::WebSocket::FRAME_OP_CLOSE));
+				Logger_.information("WebSocket connection closed.");
+			}
+			catch (const Poco::Net::WebSocketException & E)
+			{
+				Logger_.log(E);
+				switch (E.code())
+				{
+				case Poco::Net::WebSocket::WS_ERR_HANDSHAKE_UNSUPPORTED_VERSION:
+					Response->set("Sec-WebSocket-Version", Poco::Net::WebSocket::WEBSOCKET_VERSION);
+					// fallthrough
+					case Poco::Net::WebSocket::WS_ERR_NO_HANDSHAKE:
+						case Poco::Net::WebSocket::WS_ERR_HANDSHAKE_NO_VERSION:
+							case Poco::Net::WebSocket::WS_ERR_HANDSHAKE_NO_KEY:
+								Response->setStatusAndReason(Poco::Net::HTTPResponse::HTTP_BAD_REQUEST);
+								Response->setContentLength(0);
+								Response->send();
+								break;
+				}
+			}
+			catch (const Poco::Exception &E) {
+				Logger_.log(E);
+			}
+		}
+	}
+
+	void RESTAPI_webSocketServer::Process(const Poco::JSON::Object::Ptr &O, std::string &Answer ) {
+		try {
+			if (O->has("command")) {
+				auto Command = O->get("command").toString();
+				if (Command == "serial_number_search" && O->has("serial_prefix")) {
+					auto Prefix = O->get("serial_prefix").toString();
+					uint64_t HowMany = 32;
+					if (O->has("howMany"))
+						HowMany = O->get("howMany");
+					Logger_.information(Poco::format("serial_number_search: %s", Prefix));
+					if (!Prefix.empty() && Prefix.length() < 13) {
+						std::vector<uint64_t> Numbers;
+						SerialNumberCache()->FindNumbers(Prefix, 50, Numbers);
+						Poco::JSON::Array A;
+						for (const auto &i : Numbers)
+							A.add(Utils::int_to_hex(i));
+						Poco::JSON::Object AO;
+						AO.set("serialNumbers", A);
+						AO.set("command","serial_number_search");
+						std::ostringstream SS;
+						Poco::JSON::Stringifier::stringify(AO, SS);
+						Answer = SS.str();
+					}
+				}
+			}
+		} catch (const Poco::Exception &E) {
+			Logger_.log(E);
+		}
+	}
+}