1 files changed, 16 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);
+}