Unigine::Socket Class

#ifndef __APP_WORLD_LOGIC_H__
#define __APP_WORLD_LOGIC_H__

#include <UnigineLogic.h>
#include <UnigineStreams.h>
 
 // UDP port to be used
const int UDP_PORT = 8889;

// UDP receive buffer size 
const int RECV_SIZE = 7;

// UDP send buffer size
const int SEND_SIZE = 7;

/* ... */

/// Class representing the Server Socket
class ServerSocket
{
public:
	/// Server constructor
	ServerSocket()
	{
		// creating a UDP socket
		socket = Unigine::Socket::create(Unigine::Socket::SOCKET_TYPE_DGRAM);

		// opening a socket on the specified port with a specified broadcast address
		socket->open("127.255.255.255", UDP_PORT);

		// setting the size of the sending buffer
		socket->send(SEND_SIZE);

		// setting the socket as a broadcasting one
		socket->broadcast();

		// setting the socket as a non-blocking one
		socket->nonblock();
	}

	/// Server destructor
	~ServerSocket()
	{
		// closing the socket
		socket->close();

		// destroying the socket
		socket.deleteLater();
	}

	/// method sending a message to a certain client
	void send_message(int client_num, const char* message)
	{
		// preparing a message to be received by a client with a given client_id
		Unigine::BlobPtr blob = Unigine::Blob::create();
		blob->clear();
		blob->writeShort(client_num);
		blob->write(message, strlen(message));
		int size = blob->getSize();

		// sending the message
		socket->write(blob->getData(), size);
		blob.clear();
	}

private:
	// socket pointer
	Unigine::SocketPtr socket;
};

/// Class representing the Client Socket
class ClientSocket {

public:

	/// Client constructor
	ClientSocket()
	{
		// creating a UDP socket
		socket = Unigine::Socket::create(Unigine::Socket::SOCKET_TYPE_DGRAM);

		// opening a socket on the specified port 

		socket->open(UDP_PORT);

		// setting the size of the receiving buffer
		socket->recv(RECV_SIZE);

		// binding the socket to an address figured out from the host used for socket initialization
		socket->bind();

		// setting the socket as a non-blocking one
		socket->nonblock();
	}

	/// Client destructor
	~ClientSocket()
	{
		// closing the socket
		socket->close();

		// destroying the socket
		socket.deleteLater();
	}

	void setID(int num)
	{
		// setting client's ID
		id = num;
	}

	/// method checking for received packes from the server
	int update()
	{
		// preparing a blob to read the message into
		Unigine::BlobPtr temp_blob = Unigine::Blob::create();
		temp_blob->clear();

		// reading data from the socket
		socket->readStream(temp_blob, RECV_SIZE);

		if (temp_blob->getSize() > 0) {

			// setting current position to start
			temp_blob->seekSet(0);

			// getting client's ID
			int num_client = temp_blob->readShort();

			// checking if the received message is addressed to this particular client and processing it
			if (num_client == id) {
				Unigine::Log::message("\nClient[%d] - OPERATION_CODE: %s", id, temp_blob->readLine().get());
			}
		}
		return 1;
	}
private:
	// socket pointer
	Unigine::SocketPtr socket;

	// client ID
	int id = 0;
};

class AppWorldLogic : public Unigine::WorldLogic
{

public:
	AppWorldLogic();
	virtual ~AppWorldLogic();

	int init() override;

	int update() override;
	int postUpdate() override;
	int updatePhysics() override;

	int shutdown() override;

	int save(const Unigine::StreamPtr &stream) override;
	int restore(const Unigine::StreamPtr &stream) override;
private:

	// declaring client and server sockets
	ServerSocket server_socket;
	ClientSocket clients[5];
};

#endif // __APP_WORLD_LOGIC_H__

// AppWorldLogic.cpp
#include <UnigineInput.h>
#include <UnigineGame.h>

using namespace Unigine;

/* ... */

int AppWorldLogic::init() 
{
	//initializing clients IDs
	for (int i = 0; i < 5; i++)
		clients[i].setID(i+1);

	return 1;
}

int AppWorldLogic::update() 
{
	// sending messages on keys pressed to clients 2 and 5
	if (Input::isKeyDown(Input::KEY_E))
		server_socket.send_message(2, "E");
	else if (Input::isKeyDown(Input::KEY_W))
		server_socket.send_message(5, "W");
	else if (Input::isKeyDown(Input::KEY_A))
		server_socket.send_message(5, "A");
	else if (Input::isKeyDown(Input::KEY_S))
		server_socket.send_message(5, "S");
	else if (Input::isKeyDown(Input::KEY_D))
		server_socket.send_message(5, "D");

	// updating clients
	for (int i = 0; i < 5; i++)
		clients[i].update();

	return 1;
}
/* ... */