Skip to content
UDP_Protocol.h 1.14 KiB
Newer Older
Valerio Pastore's avatar
Valerio Pastore committed
/*
 *
 *  Created on: Mar 1, 2021
 *      Author: astrisw
 */
astri's avatar
astri committed
#pragma once
Valerio Pastore's avatar
Valerio Pastore committed

#include <Base_Protocol.h>

/*
Valerio Pastore's avatar
Valerio Pastore committed
 * class UDPProtocol
Valerio Pastore's avatar
Valerio Pastore committed
 * Implementation of a UDP socket
 * @member field fd_sock: socket file descriptor
 * @member field sockaddr_in: struct of the client address.
 */
Valerio Pastore's avatar
Valerio Pastore committed
namespace inaf::oasbo::ConnectionProtocols{
Valerio Pastore's avatar
Valerio Pastore committed

class UDPProtocol : public BaseProtocol{
protected:
	int srv_sock;
	int cli_sock;
astri's avatar
astri committed

Valerio Pastore's avatar
Valerio Pastore committed
	struct sockaddr_in cliaddr;
	struct sockaddr_in srvaddr;
	void split_ip_port(const std::string& ip_port, std::string& ip, std::string& port);
	int receiveAtLeastHeaderSizeBytes(uint8_t *buff, int headerSize, int packetSize);
	void resetPacket(PacketLib::BasePacket &, int bytes);

	UDPProtocol(std::string ip, int prt);
	UDPProtocol();

astri's avatar
astri committed

public:
	std::string ip{};
	int port;

	~UDPProtocol();

	int connectToServer() override;
	int connectToClient() override;
	int closeConnectionToServer() override;
	int closeConnectionToClient() override;
	bool isConnectedToClient() const override;
	bool isConnectedToServer() const override;
	int receiveFromClient(PacketLib::BasePacket &) override;
	int sendToServer(PacketLib::BasePacket &) override;
astri's avatar
astri committed

	class Builder;