Ensure downward compatibility

Lord-Grey · Jan 26, 2025 · 197289c · 197289c
1 parent 54e3eb2
commit 197289c
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 15 deletions.
diff --git a/include/flatbufserver/FlatBufferConnection.h b/include/flatbufserver/FlatBufferConnection.h
@@ -1,4 +1,5 @@
-#pragma once
+#ifndef FLATBUFFERCONNECTION_H
+#define FLATBUFFERCONNECTION_H
 
 // Qt includes
 #include <QString>
@@ -150,3 +151,5 @@ private slots:
 	flatbuffers::FlatBufferBuilder _builder;
 	bool _isRegistered;
 };
+
+#endif // FLATBUFFERCONNECTION_H
diff --git a/libsrc/flatbufserver/FlatBufferConnection.cpp b/libsrc/flatbufserver/FlatBufferConnection.cpp
@@ -49,14 +49,14 @@ void FlatBufferConnection::connectToRemoteHost()
 	if (_socket.state() == QAbstractSocket::UnconnectedState)
 	{
 		Info(_log, "Connecting to target host: %s, port [%u]", QSTRING_CSTR(_host), _port);
-	   _socket.connectToHost(_host, _port);
+		_socket.connectToHost(_host, _port);
 	}
 }
 
 void FlatBufferConnection::onDisconnected()
 {
 	_isRegistered = false,
-	Info(_log, "Disconnected from target host: %s, port [%u]", QSTRING_CSTR(_host), _port);
+			Info(_log, "Disconnected from target host: %s, port [%u]", QSTRING_CSTR(_host), _port);
 	emit isDisconnected();	isClientRegistered();
 	_timer.start();
 }
@@ -166,10 +166,10 @@ void FlatBufferConnection::readData()
 	while(_receiveBuffer.size() >= 4)
 	{
 		uint32_t messageSize =
-			((_receiveBuffer[0]<<24) & 0xFF000000) |
-			((_receiveBuffer[1]<<16) & 0x00FF0000) |
-			((_receiveBuffer[2]<< 8) & 0x0000FF00) |
-			((_receiveBuffer[3]    ) & 0x000000FF);
+				((_receiveBuffer[0]<<24) & 0xFF000000) |
+				((_receiveBuffer[1]<<16) & 0x00FF0000) |
+				((_receiveBuffer[2]<< 8) & 0x0000FF00) |
+				((_receiveBuffer[3]    ) & 0x000000FF);
 
 		// check if we can read a complete message
 		if((uint32_t) _receiveBuffer.size() < messageSize + 4) return;
@@ -216,18 +216,19 @@ bool FlatBufferConnection::parseReply(const hyperionnet::Reply *reply)
 
 		const auto registered = reply->registered();
 
-		// We got a client is registered reply.
-		if (registered == -1 || registered != _priority)
+		if (!_isRegistered)
 		{
-			_isRegistered = false;
-		}
-		else
-		{
-			if (!_isRegistered)
+			// We got a client is registered reply.
+			if (registered == -1 || registered != _priority)
+			{
+				registerClient(_origin, _priority);
+			}
+			else
 			{
+
 				_timer.stop();
-				Debug(_log,"Client \"%s\" registered successfully with target host: %s, port [%u]", QSTRING_CSTR(_origin), QSTRING_CSTR(_host), _port);
 				_isRegistered = true;
+				Debug(_log,"Client \"%s\" registered successfully with target host: %s, port [%u]", QSTRING_CSTR(_origin), QSTRING_CSTR(_host), _port);
 				emit isReadyToSend();
 			}
 		}