cuberite-2a/source/cSocket.h

#pragma once

#ifdef _WIN32
#define WIN32_LEAN_AND_MEAN
#include <WinSock.h>
#ifdef SendMessage
#undef SendMessage
#endif
#endif	// _WIN32


class cSocket
{
#ifdef _WIN32
	typedef SOCKET xSocket;
#else
	typedef int xSocket;
#endif

public:
	cSocket() : m_Socket( 0 ) {}

	cSocket( xSocket a_Socket );
	~cSocket();

	bool IsValid();
	void CloseSocket();

	operator const xSocket() const;
	xSocket GetSocket() const;
	void SetSocket( xSocket a_Socket );

	int SetReuseAddress();
	static int WSAStartup();

	static const char* GetLastErrorString();
	static cSocket CreateSocket();

	inline static bool IsSocketError( int a_ReturnedValue )
	{
#ifdef _WIN32
		return (a_ReturnedValue == SOCKET_ERROR || a_ReturnedValue == 0);
#else
		return (a_ReturnedValue <= 0);
#endif
	}

	struct SockAddr_In
	{
		short Family;
		unsigned short Port;
		unsigned long Address;
	};

	static const short ADDRESS_FAMILY_INTERNET = 2;
	static const unsigned long INTERNET_ADDRESS_ANY = 0;

	int Bind( SockAddr_In& a_Address );
	int Listen( int a_Backlog );
	cSocket Accept();
	int Receive( char* a_Buffer, unsigned int a_Length, unsigned int a_Flags );

	char* GetIPString() { return m_IPString; }

private:
	xSocket m_Socket;
	char* m_IPString;
};