Handle extension packets using binary frames

The subprotocol shall use binary frames, and AO protocol stays separated
within the text frames.

2 files changed, 19 insertions, 0 deletions

diff --git a/src/network/websocketconnection.cpp b/src/network/websocketconnection.cpp
index d51cdca..fedbe11 100644
--- a/src/network/websocketconnection.cpp
+++ b/src/network/websocketconnection.cpp
@@ -14,6 +14,7 @@ WebSocketConnection::WebSocketConnection(AOApplication *ao_app, QObject *parent)
   connect(m_socket, &QWebSocket::errorOccurred, this, &WebSocketConnection::onError);
   connect(m_socket, &QWebSocket::stateChanged, this, &WebSocketConnection::onStateChanged);
   connect(m_socket, &QWebSocket::textMessageReceived, this, &WebSocketConnection::onTextMessageReceived);
+  connect(m_socket, &QWebSocket::binaryMessageReceived, this, &WebSocketConnection::onBinaryMessageReceived);
 }
 
 WebSocketConnection::~WebSocketConnection()
@@ -55,6 +56,11 @@ void WebSocketConnection::sendPacket(AOPacket packet)
   m_socket->sendTextMessage(packet.toString(true));
 }
 
+void WebSocketConnection::sendExMessage(const QByteArray &msg)
+{
+  m_socket->sendBinaryMessage(msg);
+}
+
 void WebSocketConnection::onError()
 {
   Q_EMIT errorOccurred(m_socket->errorString());
@@ -95,3 +101,13 @@ void WebSocketConnection::onTextMessageReceived(QString message)
 
   Q_EMIT receivedPacket(AOPacket(header, raw_content));
 }
+
+void WebSocketConnection::onBinaryMessageReceived(const QByteArray message)
+{
+  if (message.isEmpty())
+  {
+    return;
+  }
+
+  Q_EMIT receivedExMessage(message);
+}
diff --git a/src/network/websocketconnection.h b/src/network/websocketconnection.h
index 9d9f7d0..1f5a842 100644
--- a/src/network/websocketconnection.h
+++ b/src/network/websocketconnection.h
@@ -23,6 +23,7 @@ public:
   void disconnectFromServer();
 
   void sendPacket(AOPacket packet);
+  void sendExMessage(const QByteArray &msg);
 
 Q_SIGNALS:
   void connectedToServer();
@@ -30,6 +31,7 @@ Q_SIGNALS:
   void errorOccurred(QString error);
 
   void receivedPacket(AOPacket packet);
+  void receivedExMessage(QByteArray msg);
 
 private:
   AOApplication *ao_app;
@@ -41,4 +43,5 @@ private Q_SLOTS:
   void onError();
   void onStateChanged(QAbstractSocket::SocketState state);
   void onTextMessageReceived(QString message);
+  void onBinaryMessageReceived(QByteArray message);
 };