libtgvoip/os/posix/NetworkSocketPosix.h

//
// libtgvoip is free and unencumbered public domain software.
// For more information, see http://unlicense.org or the UNLICENSE file
// you should have received with this source code distribution.
//

#ifndef LIBTGVOIP_NETWORKSOCKETPOSIX_H
#define LIBTGVOIP_NETWORKSOCKETPOSIX_H

#include "../../NetworkSocket.h"
#include "../../Buffers.h"
#include <vector>
#include <sys/select.h>
#include <pthread.h>

namespace tgvoip {

class SocketSelectCancellerPosix : public SocketSelectCanceller{
friend class NetworkSocketPosix;
public:
	SocketSelectCancellerPosix();
	virtual ~SocketSelectCancellerPosix();
	virtual void CancelSelect();
private:
	int pipeRead;
	int pipeWrite;
};

class NetworkSocketPosix : public NetworkSocket{
public:
	NetworkSocketPosix(NetworkProtocol protocol);
	virtual ~NetworkSocketPosix();
	virtual void Send(NetworkPacket packet) override;
	virtual NetworkPacket Receive(size_t maxLen) override;
	virtual void Open() override;
	virtual void Close() override;
	virtual void Connect(const NetworkAddress address, uint16_t port) override;
	virtual std::string GetLocalInterfaceInfo(NetworkAddress* v4addr, NetworkAddress* v6addr) override;
	virtual void OnActiveInterfaceChanged() override;
	virtual uint16_t GetLocalPort() override;

	static std::string V4AddressToString(uint32_t address);
	static std::string V6AddressToString(const unsigned char address[16]);
	static uint32_t StringToV4Address(std::string address);
	static void StringToV6Address(std::string address, unsigned char* out);
	static NetworkAddress ResolveDomainName(std::string name);
	static bool Select(std::vector<NetworkSocket*>& readFds, std::vector<NetworkSocket*>& writeFds, std::vector<NetworkSocket*>& errorFds, SocketSelectCanceller* canceller);

	virtual NetworkAddress GetConnectedAddress() override;

	virtual uint16_t GetConnectedPort() override;

	virtual void SetTimeouts(int sendTimeout, int recvTimeout) override;
	virtual bool OnReadyToSend() override;

protected:
	virtual void SetMaxPriority() override;

private:
	static int GetDescriptorFromSocket(NetworkSocket* socket);
	std::atomic<int> fd;
	bool needUpdateNat64Prefix;
	bool nat64Present;
	double switchToV6at;
	bool isV4Available;
	std::atomic<bool> closing;
	NetworkAddress tcpConnectedAddress=NetworkAddress::Empty();
	uint16_t tcpConnectedPort;
    NetworkPacket pendingOutgoingPacket=NetworkPacket::Empty();
    Buffer recvBuffer=Buffer(2048);
};

}

#endif //LIBTGVOIP_NETWORKSOCKETPOSIX_H