#include <wx/defs.h>
#ifdef __WXMSW__
  // MinGW needs this for getaddrinfo
  #if defined(_WIN32_WINNT)
    #if _WIN32_WINNT < 0x0501
      #undef _WIN32_WINNT
      #define _WIN32_WINNT 0x0501
    #endif
  #else
    #define _WIN32_WINNT 0x0501
  #endif
  #include <winsock2.h>
  #include <ws2tcpip.h>
#endif
#include "FileZilla.h"
#include "socket.h"
#include "threadex.h"
#include <errno.h>
#ifndef __WXMSW__
  #include <sys/types.h>
  #include <sys/socket.h>
  #include <netdb.h>
  #include <fcntl.h>
  #include <arpa/inet.h>
  #include <netinet/in.h>
  #include <netinet/tcp.h>
  #if !defined(MSG_NOSIGNAL) && !defined(SO_NOSIGPIPE)
    #include <signal.h>
  #endif
#endif

// Fixups needed on FreeBSD
#if !defined(EAI_ADDRFAMILY) && defined(EAI_FAMILY)
  #define EAI_ADDRFAMILY EAI_FAMILY
#endif

// --------------------------
// Windows 2000 compatibility
// --------------------------
#ifdef __WXMSW__

// Stupid Win2K has no getaddrinfo
// Apparently it is too hard for the richest company in the
// world to add this simple function with a service pack...

extern "C" 
{
	typedef int (WINAPI *t_getaddrinfo)(const char *nodename, const char *servname,
						const struct addrinfo *hints, struct addrinfo **res);
	typedef void (WINAPI *t_freeaddrinfo)(struct addrinfo* ai);
	typedef int (WINAPI *t_getnameinfo)(const struct sockaddr* sa, socklen_t salen,
						char* host, DWORD hostlen, char* serv, DWORD servlen, int flags);
}

static t_getaddrinfo p_getaddrinfo = 0;
static t_freeaddrinfo p_freeaddrinfo = 0;
static t_getnameinfo p_getnameinfo = 0;

class CGetAddrinfoLoader
{
public:
	CGetAddrinfoLoader()
	{
		HMODULE m_hDll = LoadLibrary(_T("ws2_32.dll"));
		if (m_hDll)
		{
			p_getaddrinfo = (t_getaddrinfo)GetProcAddress(m_hDll, "getaddrinfo");
			if (p_getaddrinfo)
				p_freeaddrinfo = (t_freeaddrinfo)GetProcAddress(m_hDll, "freeaddrinfo");
			p_getnameinfo = (t_getnameinfo)GetProcAddress(m_hDll, "getnameinfo");
		}
	}

	~CGetAddrinfoLoader()
	{
		if (m_hDll)
			FreeLibrary(m_hDll);
	}
protected:
	HMODULE m_hDll;
};
static CGetAddrinfoLoader addrinfo_loader;

#define getaddrinfo p_getaddrinfo
#define freeaddrinfo p_freeaddrinfo
#define getnameinfo p_getnameinfo

#endif //__WXMSW__
// ------------------------------
// End Windows 2000 compatibility
// ------------------------------

#define WAIT_CONNECT 0x01
#define WAIT_READ	 0x02
#define WAIT_WRITE	 0x04
#define WAIT_ACCEPT  0x08
#define WAIT_CLOSE	 0x10
#define WAIT_EVENTCOUNT 5

class CSocketThread;
static std::list<CSocketThread*> waiting_socket_threads;

#ifdef __WXMSW__
static int ConvertMSWErrorCode(int error)
{
	switch (error)
	{
	case WSAEINVAL:
		return EAI_BADFLAGS;
	case WSANO_RECOVERY:
		return EAI_FAIL;
	case WSAEAFNOSUPPORT:
		return EAI_FAMILY;
	case WSA_NOT_ENOUGH_MEMORY:
		return EAI_MEMORY;
	case WSANO_DATA:
		return EAI_NODATA;
	case WSAHOST_NOT_FOUND:
		return EAI_NONAME;
	case WSATYPE_NOT_FOUND:
		return EAI_SERVICE;
	case WSAESOCKTNOSUPPORT:
		return EAI_SOCKTYPE;
	case WSAEWOULDBLOCK:
		return EAGAIN;
	case WSAEMFILE:
		return EMFILE;
	case WSAEINTR:
		return EINTR;
	case WSAEFAULT:
		return EFAULT;
	case WSAEACCES:
		return EACCES;
	default:
		return error;
	}
}
#endif

#ifdef ERRORCODETEST
class CErrorCodeTest
{
public:
	CErrorCodeTest()
	{
		int errors[] = {
			0
		};
		for (int i = 0; errors[i]; i++)
		{
#ifdef __WXMSW__
			int code = ConvertMSWErrorCode(errors[i]);
#else
			int code = errors[i];
#endif
			if (CSocket::GetErrorDescription(code).Len() < 15)
				wxMessageBox(CSocket::GetErrorDescription(code));
		}

	}
};
#endif

CSocketEventDispatcher CSocketEventDispatcher::m_dispatcher;

CSocketEventDispatcher::CSocketEventDispatcher()
{
}

CSocketEventDispatcher::~CSocketEventDispatcher()
{
	for (std::list<CSocketEvent*>::iterator iter = m_pending_events.begin(); iter != m_pending_events.end(); iter++)
		delete *iter;
}

CSocketEventDispatcher& CSocketEventDispatcher::Get()
{
	return m_dispatcher;
}

void CSocketEventDispatcher::SendEvent(CSocketEvent* evt)
{
	wxCriticalSectionLocker lock(m_sync);

	const bool send = m_pending_events.empty() && !m_inside_loop;

	m_pending_events.push_back(evt);

	if (!send)
		return;

	wxCommandEvent tmp(0, 0);
	AddPendingEvent(tmp);
}

void CSocketEventDispatcher::RemovePending(const CSocketEventHandler* pHandler)
{
	wxCriticalSectionLocker lock(m_sync);

	std::list<CSocketEvent*> keep;
	for (std::list<CSocketEvent*>::iterator iter = m_pending_events.begin(); iter != m_pending_events.end(); iter++)
	{
		if ((*iter)->GetSocketEventHandler() != pHandler)
		{
			keep.push_back(*iter);
			continue;
		}

		delete *iter;
	}
	m_pending_events.swap(keep);
}

void CSocketEventDispatcher::RemovePending(const CSocketEventSource* pSource)
{
	wxCriticalSectionLocker lock(m_sync);

	std::list<CSocketEvent*> keep;
	for (std::list<CSocketEvent*>::iterator iter = m_pending_events.begin(); iter != m_pending_events.end(); iter++)
	{
		if ((*iter)->GetSocketEventSource() != pSource)
		{
			keep.push_back(*iter);
			continue;
		}

		delete *iter;
	}
	m_pending_events.swap(keep);
}

void CSocketEventDispatcher::UpdatePending(const CSocketEventHandler* pOldHandler, const CSocketEventSource* pOldSource, CSocketEventHandler* pNewHandler, CSocketEventSource* pNewSource)
{
	wxCriticalSectionLocker lock(m_sync);

	for (std::list<CSocketEvent*>::iterator iter = m_pending_events.begin(); iter != m_pending_events.end(); iter++)
	{
		CSocketEvent* evt = *iter;
		if (evt->GetSocketEventSource() != pOldSource || evt->GetSocketEventHandler() != pOldHandler)
			continue;

		evt->m_pSource = pNewSource;
		evt->m_pSocketEventHandler = pNewHandler;
	}
}

bool CSocketEventDispatcher::ProcessEvent(wxEvent& event)
{
	m_sync.Enter();
	m_inside_loop = true;
	while (!m_pending_events.empty())
	{
		CSocketEvent *evt = m_pending_events.front();
		m_pending_events.pop_front();

		m_sync.Leave();

		CSocketEventHandler* pHandler = evt->GetSocketEventHandler();
		wxASSERT(pHandler);
		pHandler->OnSocketEvent(*evt);

		delete evt;

		m_sync.Enter();
	}
	m_inside_loop = false;
	m_sync.Leave();

	return true;
}

CSocketEvent::CSocketEvent(CSocketEventHandler* pSocketEventHandler, CSocketEventSource* pSource, enum EventType type, const wxChar* data)
	: m_pSource(pSource), m_type(type), m_error(0), m_pSocketEventHandler(pSocketEventHandler)
{
	if (data)
	{
		m_data = new wxChar[wxStrlen(data) + 1];
		wxStrcpy(m_data, data);
	}
	else
		m_data = 0;
}

CSocketEvent::CSocketEvent(CSocketEventHandler* pSocketEventHandler, CSocketEventSource* pSource, enum EventType type, int error /*=0*/)
	: m_pSource(pSource), m_type(type), m_error(error), m_pSocketEventHandler(pSocketEventHandler)
{
	m_data = 0;
}

CSocketEvent::~CSocketEvent()
{
	delete [] m_data;
}

wxString CSocketEvent::GetData() const
{
	if (!m_data)
		return wxEmptyString;

	return m_data;
}

CSocketEventHandler::~CSocketEventHandler()
{
	CSocketEventDispatcher::Get().RemovePending(this);
}

CSocketEventSource::~CSocketEventSource()
{
	CSocketEventDispatcher::Get().RemovePending(this);
}

class CSocketThread : protected wxThreadEx
{
	friend class CSocket;
public:
	CSocketThread()
		: wxThreadEx(wxTHREAD_JOINABLE), m_condition(m_sync)
	{
		m_pSocket = 0;
		m_pHost = 0;
		m_pPort = 0;
#ifdef __WXMSW__
		m_sync_event = WSA_INVALID_EVENT;
#else
		m_pipe[0] = -1;
		m_pipe[1] = -1;
#endif
		m_started = false;
		m_quit = false;
		m_finished = false;

		m_waiting = 0;
		m_triggered = 0;
		m_threadwait = false;

		for (int i = 0; i < WAIT_EVENTCOUNT; i++)
		{
			m_triggered_errors[i] = 0;
		}
	}

	virtual ~CSocketThread()
	{
#ifdef __WXMSW__
		if (m_sync_event != WSA_INVALID_EVENT)
			WSACloseEvent(m_sync_event);
#else
		if (m_pipe[0] != -1)
			close(m_pipe[0]);
		if (m_pipe[1] != -1)
			close(m_pipe[1]);
#endif
	}

	void SetSocket(CSocket* pSocket, bool already_locked = false)
	{
		if (!already_locked)
			m_sync.Lock();
		m_pSocket = pSocket;
		if (m_pHost)
		{
			delete [] m_pHost;
			m_pHost = 0;
		}
		if (m_pPort)
		{
			delete [] m_pPort;
			m_pPort = 0;
		}
		m_waiting = 0;
		if (!already_locked)
			m_sync.Unlock();
	}

	int Connect()
	{
		wxASSERT(m_pSocket);
		if (m_pHost)
			delete [] m_pHost;
		if (m_pPort)
			delete [] m_pPort;

		const wxWX2MBbuf buf = m_pSocket->m_host.mb_str();
		if (!buf)
		{
			m_pHost = 0;
			m_pPort = 0;
			return EINVAL;
		}
		m_pHost = new char[strlen(buf) + 1];
		strcpy(m_pHost, buf);

		// Connect method of CSocket ensures port is in range
		m_pPort = new char[6];
		sprintf(m_pPort, "%d", m_pSocket->m_port);
		m_pPort[5] = 0;

		Start();

		return 0;
	}

	int Start()
	{
		if (m_started)
		{
			m_sync.Lock();
			wxASSERT(m_threadwait);
			m_waiting = 0;
			WakeupThread(true);
			m_sync.Unlock();
			return 0;
		}
		m_started = true;
#ifdef __WXMSW__
		if (m_sync_event == WSA_INVALID_EVENT)
			m_sync_event = WSACreateEvent();
		if (m_sync_event == WSA_INVALID_EVENT)
			return 1;
#else
		if (m_pipe[0] == -1)
		{
			if (pipe(m_pipe))
				return errno;
		}
#endif

		int res = Create();
		if (res != wxTHREAD_NO_ERROR)
			return 1;

		Run();

		return 0;
	}

	// Cancels select or idle wait
	void WakeupThread(bool already_locked)
	{
		if (!already_locked)
			m_sync.Lock();
			
		if (!m_started || m_finished)
		{
			if (!already_locked)
				m_sync.Unlock();
			return;
		}

		if (m_threadwait)
		{
			m_threadwait = false;
			m_condition.Signal();
			if (!already_locked)
				m_sync.Unlock();
			return;
		}

#ifdef __WXMSW__
		WSASetEvent(m_sync_event);
#else
		char tmp = 0;
		write(m_pipe[1], &tmp, 1);
#endif
		if (!already_locked)
			m_sync.Unlock();
	}

protected:

	int TryConnectHost(struct addrinfo *addr)
	{
		if (m_pSocket->m_pEvtHandler)
		{
			CSocketEvent *evt = new CSocketEvent(m_pSocket->m_pEvtHandler, m_pSocket, CSocketEvent::hostaddress, CSocket::AddressToString(addr->ai_addr, addr->ai_addrlen));
			CSocketEventDispatcher::Get().SendEvent(evt);
		}

		int fd = socket(addr->ai_family, addr->ai_socktype, addr->ai_protocol);
		if (fd == -1)
		{
#ifdef __WXMSW__
			int res = ConvertMSWErrorCode(WSAGetLastError());
#else
			int res = errno;
#endif
			if (m_pSocket->m_pEvtHandler)
			{
				CSocketEvent *evt = new CSocketEvent(m_pSocket->GetEventHandler(), m_pSocket, addr->ai_next ? CSocketEvent::connection_next : CSocketEvent::connection, res);
				CSocketEventDispatcher::Get().SendEvent(evt);
			}

			return 0;
		}
		
#if defined(SO_NOSIGPIPE) && !defined(MSG_NOSIGNAL)
		// We do not want SIGPIPE if writing to socket.
		const int value = 1;
		setsockopt(fd, SOL_SOCKET, SO_NOSIGPIPE, &value, sizeof(int));
#endif

		CSocket::DoSetFlags(fd, m_pSocket->m_flags, m_pSocket->m_flags);
		CSocket::DoSetBufferSizes(fd, m_pSocket->m_buffer_sizes[0], m_pSocket->m_buffer_sizes[1]);
		CSocket::SetNonblocking(fd);

		int res = connect(fd, addr->ai_addr, addr->ai_addrlen);
		if (res == -1)
		{
#ifdef __WXMSW__
			// Map to POSIX error codes
			int error = WSAGetLastError();
			if (error == WSAEWOULDBLOCK)
				res = EINPROGRESS;
			else
				res = ConvertMSWErrorCode(WSAGetLastError());
#else
			res = errno;
#endif
		}

		if (res == EINPROGRESS)
		{

			m_pSocket->m_fd = fd;

			bool wait_successful;
			do
			{
				wait_successful = DoWait(WAIT_CONNECT);
				if ((m_triggered & WAIT_CONNECT))
					break;
			} while (wait_successful);

			if (!wait_successful)
			{
#ifdef __WXMSW__
				closesocket(fd);
#else
				close(fd);
#endif
				if (m_pSocket)
					m_pSocket->m_fd = -1;
				return -1;
			}
			m_triggered &= ~WAIT_CONNECT;

			res = m_triggered_errors[0];
		}

		if (res)
		{
			if (m_pSocket->m_pEvtHandler)
			{
				CSocketEvent *evt = new CSocketEvent(m_pSocket->GetEventHandler(), m_pSocket, addr->ai_next ? CSocketEvent::connection_next : CSocketEvent::connection, res);
				CSocketEventDispatcher::Get().SendEvent(evt);
			}

			m_pSocket->m_fd = -1;
#ifdef __WXMSW__
			closesocket(fd);
#else
			close(fd);
#endif
		}
		else
		{
			m_pSocket->m_fd = fd;
			m_pSocket->m_state = CSocket::connected;

			if (m_pSocket->m_pEvtHandler)
			{
				CSocketEvent *evt = new CSocketEvent(m_pSocket->GetEventHandler(), m_pSocket, CSocketEvent::connection, 0);
				CSocketEventDispatcher::Get().SendEvent(evt);
			}

			// We're now interested in all the other nice events
			m_waiting |= WAIT_READ | WAIT_WRITE;

			return 1;
		}

		return 0;
	}

	// Only call while locked
	bool DoConnect()
	{
		char* pHost;
		char* pPort;

		if (!m_pHost || !m_pPort)
		{
			m_pSocket->m_state = CSocket::closed;
			return false;
		}

		pHost = m_pHost;
		m_pHost = 0;
		pPort = m_pPort;
		m_pPort = 0;

		m_sync.Unlock();

		struct addrinfo *addressList = 0;
		struct addrinfo hints = {0};
		hints.ai_family = m_pSocket->m_family;
		hints.ai_socktype = SOCK_STREAM;

#ifdef __WXMSW__
		if (!getaddrinfo)
		{
			// Win2K fallback
			int port = atoi(pPort);

			struct hostent* h = gethostbyname(pHost);

			delete [] pHost;
			delete [] pPort;

			if (!Lock())
			{
				if (m_pSocket)
					m_pSocket->m_state = CSocket::closed;
				return false;
			}

			if (!h)
			{
				if (m_pSocket->m_pEvtHandler)
				{
					int res = ConvertMSWErrorCode(WSAGetLastError());
					CSocketEvent *evt = new CSocketEvent(m_pSocket->GetEventHandler(), m_pSocket, CSocketEvent::connection, res);
					CSocketEventDispatcher::Get().SendEvent(evt);
				}
				m_pSocket->m_state = CSocket::closed;
				return false;
			}

			struct sockaddr_in addr_in = {0};
			addr_in.sin_family = AF_INET;
			addr_in.sin_addr = *((struct in_addr*)h->h_addr_list[0]);
			addr_in.sin_port = htons(port);

			struct addrinfo addr = {0};
			addr.ai_family = AF_INET;
			addr.ai_socktype = SOCK_STREAM;
			addr.ai_addr = (struct sockaddr*)&addr_in;
			addr.ai_addrlen = sizeof(struct sockaddr_in);

			int res = TryConnectHost(&addr);
			if (res == 1)
				return true;

			if (m_pSocket)
			{
				if (!res && m_pSocket->m_pEvtHandler)
				{
					CSocketEvent *evt = new CSocketEvent(m_pSocket->GetEventHandler(), m_pSocket, CSocketEvent::connection, ECONNABORTED);
					CSocketEventDispatcher::Get().SendEvent(evt);
				}
				m_pSocket->m_state = CSocket::closed;
			}
			return false;
		}
#endif
		int res = getaddrinfo(pHost, pPort, &hints, &addressList);

		delete [] pHost;
		delete [] pPort;

		if (!Lock())
		{
			if (!res && addressList)
				freeaddrinfo(addressList);
			if (m_pSocket)
				m_pSocket->m_state = CSocket::closed;
			return false;
		}

		if (res)
		{
#ifdef __WXMSW__
			res = ConvertMSWErrorCode(res);
#endif

			if (m_pSocket->m_pEvtHandler)
			{
				CSocketEvent *evt = new CSocketEvent(m_pSocket->GetEventHandler(), m_pSocket, CSocketEvent::connection, res);
				CSocketEventDispatcher::Get().SendEvent(evt);
			}
			m_pSocket->m_state = CSocket::closed;

			return false;
		}

		for (struct addrinfo *addr = addressList; addr; addr = addr->ai_next)
		{
			res = TryConnectHost(addr);
			if (res == -1)
			{
				freeaddrinfo(addressList);
				if (m_pSocket)
					m_pSocket->m_state = CSocket::closed;
				return false;
			}
			else if (res)
			{
				freeaddrinfo(addressList);
				return true;
			}
		}
		freeaddrinfo(addressList);

		if (m_pSocket->m_pEvtHandler)
		{
			CSocketEvent *evt = new CSocketEvent(m_pSocket->GetEventHandler(), m_pSocket, CSocketEvent::connection, ECONNABORTED);
			CSocketEventDispatcher::Get().SendEvent(evt);
		}
		m_pSocket->m_state = CSocket::closed;

		return false;
	}

	// Obtains lock in all cases.
	// Returns false if thread should quit
	bool Lock()
	{
		m_sync.Lock();
		if (m_quit || !m_pSocket)
			return false;

		return true;
	}

	// Call only while locked
	bool DoWait(int wait)
	{
		m_waiting |= wait;

		while (true)
		{
#ifdef __WXMSW__
			int wait_events = FD_CLOSE;
			if (m_waiting & WAIT_CONNECT)
				wait_events |= FD_CONNECT;
			if (m_waiting & WAIT_READ)
				wait_events |= FD_READ;
			if (m_waiting & WAIT_WRITE)
				wait_events |= FD_WRITE;
			if (m_waiting & WAIT_ACCEPT)
				wait_events |= FD_ACCEPT;
			if (m_waiting & WAIT_CLOSE)
				wait_events |= FD_CLOSE;
			WSAEventSelect(m_pSocket->m_fd, m_sync_event, wait_events);
			m_sync.Unlock();
			WSAWaitForMultipleEvents(1, &m_sync_event, false, WSA_INFINITE, false);

			m_sync.Lock();
			if (m_quit || !m_pSocket)
			{
				return false;
			}

			WSANETWORKEVENTS events;
			int res = WSAEnumNetworkEvents(m_pSocket->m_fd, m_sync_event, &events);
			if (res)
			{
				res = ConvertMSWErrorCode(WSAGetLastError());
				return false;
			}

			if (m_waiting & WAIT_CONNECT)
			{
				if (events.lNetworkEvents & FD_CONNECT)
				{
					m_triggered |= WAIT_CONNECT;
					m_triggered_errors[0] = ConvertMSWErrorCode(events.iErrorCode[FD_CONNECT_BIT]);
					m_waiting &= ~WAIT_CONNECT;
				}
			}
			if (m_waiting & WAIT_READ)
			{
				if (events.lNetworkEvents & FD_READ)
				{
					m_triggered |= WAIT_READ;
					m_triggered_errors[1] = ConvertMSWErrorCode(events.iErrorCode[FD_READ_BIT]);
					m_waiting &= ~WAIT_READ;
				}
			}
			if (m_waiting & WAIT_WRITE)
			{
				if (events.lNetworkEvents & FD_WRITE)
				{
					m_triggered |= WAIT_WRITE;
					m_triggered_errors[2] = ConvertMSWErrorCode(events.iErrorCode[FD_WRITE_BIT]);
					m_waiting &= ~WAIT_WRITE;
				}
			}
			if (m_waiting & WAIT_ACCEPT)
			{
				if (events.lNetworkEvents & FD_ACCEPT)
				{
					m_triggered |= WAIT_ACCEPT;
					m_triggered_errors[3] = ConvertMSWErrorCode(events.iErrorCode[FD_ACCEPT_BIT]);
					m_waiting &= ~WAIT_ACCEPT;
				}
			}
			if (m_waiting & WAIT_CLOSE)
			{
				if (events.lNetworkEvents & FD_CLOSE)
				{
					m_triggered |= WAIT_CLOSE;
					m_triggered_errors[4] = ConvertMSWErrorCode(events.iErrorCode[FD_CLOSE_BIT]);
					m_waiting &= ~WAIT_CLOSE;
				}
			}

			if (m_triggered || !m_waiting)
				return true;
#else
			fd_set readfds;
			fd_set writefds;
			FD_ZERO(&readfds);
			FD_ZERO(&writefds);

			FD_SET(m_pipe[0], &readfds);
			if (!(m_waiting & WAIT_CONNECT))
				FD_SET(m_pSocket->m_fd, &readfds);

			if (m_waiting & (WAIT_WRITE | WAIT_CONNECT))
				FD_SET(m_pSocket->m_fd, &writefds);

			int max = wxMax(m_pipe[0], m_pSocket->m_fd) + 1;

			m_sync.Unlock();

			int res = select(max, &readfds, &writefds, 0, 0);

			m_sync.Lock();
			
			if (res > 0 && FD_ISSET(m_pipe[0], &readfds))
			{
				char buffer[100];
				read(m_pipe[0], buffer, 100);
			}

			if (m_quit || !m_pSocket || m_pSocket->m_fd == -1)
			{
				return false;
			}

			if (!res)
				continue;
			if (res == -1)
			{
				res = errno;
				//printf("select failed: %s\n", (const char *)CSocket::GetErrorDescription(res).mb_str());
				//fflush(stdout);

				if (res == EINTR)
					continue;

				return false;
			}

			if (m_waiting & WAIT_CONNECT)
			{
				if (FD_ISSET(m_pSocket->m_fd, &writefds))
				{
					int error;
					socklen_t len = sizeof(error);
					int res = getsockopt(m_pSocket->m_fd, SOL_SOCKET, SO_ERROR, &error, &len);
					if (res)
						error = errno;
					m_triggered |= WAIT_CONNECT;
					m_triggered_errors[0] = error;
					m_waiting &= ~WAIT_CONNECT;
				}
			}
			else if (m_waiting & WAIT_ACCEPT)
			{
				if (FD_ISSET(m_pSocket->m_fd, &readfds))
				{
					m_triggered |= WAIT_ACCEPT;
					m_waiting &= ~WAIT_ACCEPT;
				}
			}
			else if (m_waiting & WAIT_READ)
			{
				if (FD_ISSET(m_pSocket->m_fd, &readfds))
				{
					m_triggered |= WAIT_READ;
					m_waiting &= ~WAIT_READ;
				}
			}
			if (m_waiting & WAIT_WRITE)
			{
				if (FD_ISSET(m_pSocket->m_fd, &writefds))
				{
					m_triggered |= WAIT_WRITE;
					m_waiting &= ~WAIT_WRITE;
				}
			}

			if (m_triggered || !m_waiting)
				return true;
#endif
		}
	}

	void SendEvents()
	{
		if (!m_pSocket || !m_pSocket->m_pEvtHandler)
			return;
		if (m_triggered & WAIT_READ)
		{
			CSocketEvent *evt = new CSocketEvent(m_pSocket->m_pEvtHandler, m_pSocket, CSocketEvent::read, m_triggered_errors[1]);
			CSocketEventDispatcher::Get().SendEvent(evt);
			m_triggered &= ~WAIT_READ;
		}
		if (m_triggered & WAIT_WRITE)
		{
			CSocketEvent *evt = new CSocketEvent(m_pSocket->m_pEvtHandler, m_pSocket, CSocketEvent::write, m_triggered_errors[2]);
			CSocketEventDispatcher::Get().SendEvent(evt);
			m_triggered &= ~WAIT_WRITE;
		}
		if (m_triggered & WAIT_ACCEPT)
		{
			CSocketEvent *evt = new CSocketEvent(m_pSocket->m_pEvtHandler, m_pSocket, CSocketEvent::connection, m_triggered_errors[3]);
			CSocketEventDispatcher::Get().SendEvent(evt);
			m_triggered &= ~WAIT_ACCEPT;
		}
		if (m_triggered & WAIT_CLOSE)
		{
			CSocketEvent *evt = new CSocketEvent(m_pSocket->m_pEvtHandler, m_pSocket, CSocketEvent::close, m_triggered_errors[4]);
			CSocketEventDispatcher::Get().SendEvent(evt);
			m_triggered &= ~WAIT_CLOSE;
		}
	}

	// Call only while locked
	bool IdleLoop()
	{
		if (m_quit)
			return false;
		while (!m_pSocket || (!m_waiting && !m_pHost))
		{
			m_threadwait = true;
			m_condition.Wait();

			if (m_quit)
				return false;
		}

		return true;
	}

	virtual ExitCode Entry()
	{
		m_sync.Lock();
		while (true)
		{
			if (!IdleLoop())
			{
				m_finished = true;
				m_sync.Unlock();
				return 0;
			}

			if (m_pSocket->m_state == CSocket::listening)
			{
				while (IdleLoop())
				{
					if (m_pSocket->m_fd == -1)
					{
						m_waiting = 0;
						break;
					}
					if (!DoWait(0))
						break;
					SendEvents();
				}
			}
			else
			{
				if (m_pSocket->m_state == CSocket::connecting)
				{
					if (!DoConnect())
						continue;
				}

#ifdef __WXMSW__
				m_waiting |= WAIT_CLOSE;
				int wait_close = WAIT_CLOSE;
#endif
				while (IdleLoop())
				{
					if (m_pSocket->m_fd == -1)
					{
						m_waiting = 0;
						break;
					}
					bool res = DoWait(0);

					if (m_triggered & WAIT_CLOSE && m_pSocket)
					{
						m_pSocket->m_state = CSocket::closing;
#ifdef __WXMSW__
						wait_close = 0;
#endif
					}

					if (!res)
						break;

					SendEvents();
#ifdef __WXMSW__
					m_waiting |= wait_close;
#endif
				}
			}
		}

		m_finished = true;
		return 0;
	}

	CSocket* m_pSocket;

	char* m_pHost;
	char* m_pPort;

#ifdef __WXMSW__
	// We wait on this using WSAWaitForMultipleEvents
	WSAEVENT m_sync_event;
#else
	// A pipe is used to unblock select
	int m_pipe[2];
#endif

	wxMutex m_sync;
	wxCondition m_condition;

	bool m_started;
	bool m_quit;
	bool m_finished;

	// The socket events we are waiting for
	int m_waiting;

	// The triggered socket events
	int m_triggered;
	int m_triggered_errors[WAIT_EVENTCOUNT];

	// Thread waits for instructions
	bool m_threadwait;
};

CSocket::CSocket(CSocketEventHandler* pEvtHandler)
	: m_pEvtHandler(pEvtHandler)
{
#ifdef ERRORCODETEST
	CErrorCodeTest test;
#endif
	m_fd = -1;
	m_state = none;
	m_pSocketThread = 0;
	m_flags = 0;

	m_port = 0;
	m_family = AF_UNSPEC;

	m_buffer_sizes[0] = -1;
	m_buffer_sizes[1] = -1;
}

CSocket::~CSocket()
{
	if (m_state != none)
		Close();

	DetachThread();
}

void CSocket::DetachThread()
{
	if (!m_pSocketThread)
		return;

	m_pSocketThread->m_sync.Lock();
	m_pSocketThread->SetSocket(0, true);
	if (m_pSocketThread->m_finished)
	{
		m_pSocketThread->WakeupThread(true);
		m_pSocketThread->m_sync.Unlock();
		m_pSocketThread->Wait();
		delete m_pSocketThread;
	}
	else
	{
		if (!m_pSocketThread->m_started)
		{
			m_pSocketThread->m_sync.Unlock();
			delete m_pSocketThread;
		}
		else
		{
			m_pSocketThread->m_quit = true;
			m_pSocketThread->WakeupThread(true);
			m_pSocketThread->m_sync.Unlock();
			waiting_socket_threads.push_back(m_pSocketThread);
		}
	}
	m_pSocketThread = 0;

	Cleanup(false);
}

int CSocket::Connect(wxString host, unsigned int port, enum address_family family /*=unsped*/)
{
	if (m_state != none)
		return EISCONN;

	if (port < 1 || port > 65535)
		return EINVAL;

	switch (family)
	{
	case unspec:
		m_family = AF_UNSPEC;
		break;
	case ipv4:
		m_family = AF_INET;
		break;
	case ipv6:
		m_family = AF_INET6;
		break;
	default:
		return EINVAL;
	}

	if (m_pSocketThread && m_pSocketThread->m_started)
	{
		m_pSocketThread->m_sync.Lock();
		if (!m_pSocketThread->m_threadwait)
		{
			m_pSocketThread->WakeupThread(true);
			m_pSocketThread->m_sync.Unlock();
			// Wait a small amount of time
			wxMilliSleep(100);

			m_pSocketThread->m_sync.Lock();
			if (!m_pSocketThread->m_threadwait)
			{
				// Inside a blocking call, e.g. getaddrinfo
				m_pSocketThread->m_sync.Unlock();
				DetachThread();
			}
			else
				m_pSocketThread->m_sync.Unlock();
		}
		else
			m_pSocketThread->m_sync.Unlock();
	}
	if (!m_pSocketThread)
	{
		m_pSocketThread = new CSocketThread();
		m_pSocketThread->SetSocket(this);
	}

	m_state = connecting;

	m_host = host;
	m_port = port;
	int res = m_pSocketThread->Connect();
	if (res)
	{
		m_state = none;
		delete m_pSocketThread;
		m_pSocketThread = 0;
		return res;
	}

	return EINPROGRESS;
}

void CSocket::SetEventHandler(CSocketEventHandler* pEvtHandler)
{
	if (m_pSocketThread)
		m_pSocketThread->m_sync.Lock();

	if (!pEvtHandler)
	{
		CSocketEventDispatcher::Get().RemovePending(m_pEvtHandler);
	}
	else
	{
		if (m_pEvtHandler)
			CSocketEventDispatcher::Get().UpdatePending(m_pEvtHandler, this, pEvtHandler, this);
	}
#ifdef __WXMSW__
	if (pEvtHandler && !m_pEvtHandler && m_state == closing && m_pSocketThread)
	{
		// After getting FD_CLOSE, no further events are recorded, so send
		// it out to new handler manually
		CSocketEvent *evt = new CSocketEvent(pEvtHandler, this, CSocketEvent::close, m_pSocketThread->m_triggered_errors[4]);
		CSocketEventDispatcher::Get().SendEvent(evt);
	}
#else
	wxASSERT(!pEvtHandler || m_state != closing);
#endif

	m_pEvtHandler = pEvtHandler;

	if (m_pSocketThread)
	{
		if (pEvtHandler && m_state == connected)
		{
#ifdef __WXMSW__
			// If a graceful shutdown is going on in background already,
			// no further events are recorded. Send out events we're not
			// waiting for (i.e. they got triggered already) manually.

			if (!(m_pSocketThread->m_waiting & WAIT_WRITE))
			{
				CSocketEvent *evt = new CSocketEvent(pEvtHandler, this, CSocketEvent::write, 0);
				CSocketEventDispatcher::Get().SendEvent(evt);
			}

			CSocketEvent *evt = new CSocketEvent(pEvtHandler, this, CSocketEvent::read, 0);
			CSocketEventDispatcher::Get().SendEvent(evt);
			if (m_pSocketThread->m_waiting & WAIT_WRITE)
			{
				m_pSocketThread->m_waiting &= ~WAIT_READ;
				m_pSocketThread->WakeupThread(true);
			}
#else
			m_pSocketThread->m_waiting |= WAIT_READ | WAIT_WRITE;
			m_pSocketThread->WakeupThread(true);
#endif
		}
		m_pSocketThread->m_sync.Unlock();
	}
}

#define ERRORDECL(c, desc) { c, _T(#c), wxTRANSLATE(desc) },

struct Error_table
{
	int code;
	const wxChar* const name;
	const wxChar* const description;
};

static struct Error_table error_table[] =
{
	ERRORDECL(EACCES, "Permission denied")
	ERRORDECL(EADDRINUSE, "Local address in use")
	ERRORDECL(EAFNOSUPPORT, "The specified address family is not supported")
	ERRORDECL(EINPROGRESS, "Operation in progress")
	ERRORDECL(EINVAL, "Invalid argument passed")
	ERRORDECL(EMFILE, "Process file table overflow")
	ERRORDECL(ENFILE, "System limit of open files exceeded")
	ERRORDECL(ENOBUFS, "Out of memory")
	ERRORDECL(ENOMEM, "Out of memory")
	ERRORDECL(EPERM, "Permission denied")
	ERRORDECL(EPROTONOSUPPORT, "Protocol not supported")
	ERRORDECL(EAGAIN, "Resource temporarily unavailable")
	ERRORDECL(EALREADY, "Operation already in progress")
	ERRORDECL(EBADF, "Bad file descriptor")
	ERRORDECL(ECONNREFUSED, "Connection refused by server")
	ERRORDECL(EFAULT, "Socket address outside address space")
	ERRORDECL(EINTR, "Interrupted by signal")
	ERRORDECL(EISCONN, "Socket already connected")
	ERRORDECL(ENETUNREACH, "Network unreachable")
	ERRORDECL(ENOTSOCK, "File descriptior not a socket")
	ERRORDECL(ETIMEDOUT, "Connection attempt timed out")
	ERRORDECL(EHOSTUNREACH, "No route to host")
	ERRORDECL(ENOTCONN, "Socket not connected")
	ERRORDECL(ENETRESET, "Connection reset by network")
	ERRORDECL(EOPNOTSUPP, "Operation not supported")
	ERRORDECL(ESHUTDOWN, "Socket has been shut down")
	ERRORDECL(EMSGSIZE, "Message too large")
	ERRORDECL(ECONNABORTED, "Connection aborted")
	ERRORDECL(ECONNRESET, "Connection reset by peer")
	ERRORDECL(EPIPE, "Local endpoint has been closed")

	// Getaddrinfo related
#ifndef __WXMSW__
	ERRORDECL(EAI_ADDRFAMILY, "Network host does not have any network addresses in the requested address family")
#endif
	ERRORDECL(EAI_AGAIN, "Temporary failure in name resolution")
	ERRORDECL(EAI_BADFLAGS, "Invalid value for ai_flags")
#ifdef EAI_BADHINTS
	ERRORDECL(EAI_BADHINTS, "Invalid value for hints")
#endif
	ERRORDECL(EAI_FAIL, "Nonrecoverable failure in name resolution")
	ERRORDECL(EAI_FAMILY, "The ai_family member is not supported")
	ERRORDECL(EAI_MEMORY, "Memory allocation failure")
#ifdef EAI_NODATA
	ERRORDECL(EAI_NODATA, "No address associated with nodename")
#endif
	ERRORDECL(EAI_NONAME, "Neither nodename nor servname provided, or not known")
#ifdef EAI_OVERFLOW
	ERRORDECL(EAI_OVERFLOW, "Argument buffer overflow")
#endif
#ifdef EAI_PROTOCOL
	ERRORDECL(EAI_PROTOCOL, "Resolved protocol is unknown")
#endif
	ERRORDECL(EAI_SERVICE, "The servname parameter is not supported for ai_socktype")
	ERRORDECL(EAI_SOCKTYPE, "The ai_socktype member is not supported")
#ifndef __WXMSW__
	ERRORDECL(EAI_SYSTEM, "Other system error")
#endif

	// Codes that have no POSIX equivalence
#ifdef __WXMSW__
	ERRORDECL(WSANOTINITIALISED, "Not initialized, need to call WSAStartup")
	ERRORDECL(WSAENETDOWN, "System's network subsystem has failed")
	ERRORDECL(WSAEPROTOTYPE, "Protocol not supported on given socket type")
	ERRORDECL(WSAESOCKTNOSUPPORT, "Socket type not supported for address family")
	ERRORDECL(WSAEADDRNOTAVAIL, "Cannot assign requested address")
#endif
	{ 0, 0, 0 }
};

wxString CSocket::GetErrorString(int error)
{
	for (int i = 0; error_table[i].code; i++)
	{
		if (error != error_table[i].code)
			continue;

		return error_table[i].name;
	}

	return wxString::Format(_T("%d"), error);
}

wxString CSocket::GetErrorDescription(int error)
{
	for (int i = 0; error_table[i].code; i++)
	{
		if (error != error_table[i].code)
			continue;

		return wxString(error_table[i].name) + _T(" - ") + wxGetTranslation(error_table[i].description);
	}

	return wxString::Format(_T("%d"), error);
}

int CSocket::Close()
{
	int fd;
	if (m_pSocketThread)
	{
		m_pSocketThread->m_sync.Lock();
		fd = m_fd;
		m_fd = -1;
		delete [] m_pSocketThread->m_pHost;
		m_pSocketThread->m_pHost = 0;
		delete [] m_pSocketThread->m_pPort;
		m_pSocketThread->m_pPort = 0;
		if (!m_pSocketThread->m_threadwait)
			m_pSocketThread->WakeupThread(true);
	}
	else
	{
		fd = m_fd;
		m_fd = -1;
	}

	if (fd != -1)
	{
#ifdef __WXMSW__
		closesocket(fd);
#else
		close(fd);
#endif
	}

	m_state = none;

	if (m_pSocketThread)
		m_pSocketThread->m_sync.Unlock();

	if (m_pEvtHandler)
		CSocketEventDispatcher::Get().RemovePending(m_pEvtHandler);

	return 0;
}

enum CSocket::SocketState CSocket::GetState()
{
	enum SocketState state;
	if (m_pSocketThread)
		m_pSocketThread->m_sync.Lock();
	state = m_state;
	if (m_pSocketThread)
		m_pSocketThread->m_sync.Unlock();

	return state;
}

bool CSocket::Cleanup(bool force)
{
	std::list<CSocketThread*>::iterator iter = waiting_socket_threads.begin();
	while (iter != waiting_socket_threads.end())
	{
		std::list<CSocketThread*>::iterator current = iter++;
		CSocketThread* pThread = *current;
		pThread->m_sync.Lock();
		if (!force && !pThread->m_finished)
		{
			pThread->m_sync.Unlock();
			continue;
		}
		pThread->m_sync.Unlock();

		pThread->Wait();
		delete pThread;
		waiting_socket_threads.erase(current);
	}

	return false;
}

int CSocket::Read(void* buffer, unsigned int size, int& error)
{
	int res = recv(m_fd, (char*)buffer, size, 0);

	if (res == -1)
	{
#ifdef __WXMSW__
		error = ConvertMSWErrorCode(WSAGetLastError());
#else
		error = errno;
#endif
		if (error == EAGAIN)
		{
			if (m_pSocketThread)
			{
				m_pSocketThread->m_sync.Lock();
				if (!(m_pSocketThread->m_waiting & WAIT_READ))
				{
					m_pSocketThread->m_waiting |= WAIT_READ;
					m_pSocketThread->WakeupThread(true);
				}
				m_pSocketThread->m_sync.Unlock();
			}
		}
	}
	else
		error = 0;

	return res;
}

int CSocket::Peek(void* buffer, unsigned int size, int& error)
{
	int res = recv(m_fd, (char*)buffer, size, MSG_PEEK);

	if (res == -1)
	{
#ifdef __WXMSW__
		error = ConvertMSWErrorCode(WSAGetLastError());
#else
		error = errno;
#endif
	}
	else
		error = 0;

	return res;
}

int CSocket::Write(const void* buffer, unsigned int size, int& error)
{
#ifdef MSG_NOSIGNAL
	const int flags = MSG_NOSIGNAL;
#else
	const int flags = 0;

#if !defined(SO_NOSIGPIPE) && !defined(__WXMSW__)
	// Some systems have neither. Need to block signal
	struct sigaction old_action;
	struct sigaction action = {0};
	action.sa_handler = SIG_IGN;
	int signal_set = sigaction(SIGPIPE, &action, &old_action);
#endif

#endif

	int res = send(m_fd, (const char*)buffer, size, flags);

#if !defined(MSG_NOSIGNAL) && !defined(SO_NOSIGPIPE) && !defined(__WXMSW__)
	// Restore previous signal handler
	if (!signal_set)
		sigaction(SIGPIPE, &old_action, 0);
#endif

	if (res == -1)
	{
#ifdef __WXMSW__
		error = ConvertMSWErrorCode(WSAGetLastError());
#else
		error = errno;
#endif
		if (error == EAGAIN)
		{
			if (m_pSocketThread)
			{
				m_pSocketThread->m_sync.Lock();
				if (!(m_pSocketThread->m_waiting & WAIT_WRITE))
				{
					m_pSocketThread->m_waiting |= WAIT_WRITE;
					m_pSocketThread->WakeupThread(true);
				}
				m_pSocketThread->m_sync.Unlock();
			}
		}
	}
	else
		error = 0;

	return res;
}

wxString CSocket::AddressToString(const struct sockaddr* addr, int addr_len, bool with_port /*=true*/)
{
	char hostbuf[NI_MAXHOST];
	char portbuf[NI_MAXSERV];

#ifdef __WXMSW__
	if (!getnameinfo)
	{
		// Win2K fallback
		if (addr->sa_family != AF_INET)
			return _T("");
		char* s = inet_ntoa(((struct sockaddr_in*)addr)->sin_addr);
		if (!s)
			return _T("");

		wxString host = wxString(s, wxConvLibc);
		if (!with_port)
			return host;

		return host + wxString::Format(_T(":%d"), (int)ntohs(((struct sockaddr_in*)addr)->sin_port));
	}
#endif

	int res = getnameinfo(addr, addr_len, hostbuf, NI_MAXHOST, portbuf, NI_MAXSERV, NI_NUMERICHOST | NI_NUMERICSERV);
	if (res) // Should never fail
		return _T("");

	wxString host = wxString(hostbuf, wxConvLibc);
	wxString port = wxString(portbuf, wxConvLibc);

	// IPv6 uses colons as separator, need to enclose address
	// to avoid ambiguity if also showing port
	if (with_port && addr->sa_family == AF_INET6)
		host = _T("[") + host + _T("]");

	if (with_port)
		return host + _T(":") + port;
	else
		return host;
}

wxString CSocket::GetLocalIP() const
{
	struct sockaddr_storage addr;
	socklen_t addr_len = sizeof(addr);
	int res = getsockname(m_fd, (sockaddr*)&addr, &addr_len);
	if (res)
		return _T("");

	return AddressToString((sockaddr *)&addr, addr_len, false);
}

wxString CSocket::GetPeerIP() const
{
	struct sockaddr_storage addr;
	socklen_t addr_len = sizeof(addr);
	int res = getpeername(m_fd, (sockaddr*)&addr, &addr_len);
	if (res)
		return _T("");

	return AddressToString((sockaddr *)&addr, addr_len, false);
}

enum CSocket::address_family CSocket::GetAddressFamily() const
{
	struct sockaddr_storage addr;
	socklen_t addr_len = sizeof(addr);
	int res = getsockname(m_fd, (sockaddr*)&addr, &addr_len);
	if (res)
		return unspec;

	switch (((sockaddr*)&addr)->sa_family)
	{
	case AF_INET:
		return ipv4;
	case AF_INET6:
		return ipv6;
	default:
		return unspec;
	}
}

int CSocket::Listen(enum address_family family, int port /*=0*/)
{
	if (m_state != none)
		return EALREADY;

	if (port < 0 || port > 65535)
		return EINVAL;

	switch (family)
	{
	case unspec:
		m_family = AF_UNSPEC;
		break;
	case ipv4:
		m_family = AF_INET;
		break;
	case ipv6:
		m_family = AF_INET6;
		break;
	default:
		return EINVAL;
	}

#ifdef __WXMSW__
	if (!getaddrinfo)
	{
		m_fd = socket(AF_INET, SOCK_STREAM, 0);
		if (m_fd == -1)
			return ConvertMSWErrorCode(WSAGetLastError());

		SetNonblocking(m_fd);

		struct sockaddr_in addr = {0};
		addr.sin_family = AF_INET;
		addr.sin_port = htons(port);

		int res = bind(m_fd, (sockaddr *)&addr, sizeof(addr));
		if (res)
		{
			res = WSAGetLastError();
			closesocket(m_fd);
			m_fd = -1;

			return ConvertMSWErrorCode(res);
		}
	}
	else
#endif
	{
		struct addrinfo hints = {0};
		hints.ai_family = m_family;
		hints.ai_socktype = SOCK_STREAM;
		hints.ai_flags = AI_PASSIVE;
#ifdef AI_NUMERICSERV
		// Some systems like Windows or OS X don't know AI_NUMERICSERV.
		hints.ai_flags |= AI_NUMERICSERV;
#endif

		char portstring[6];
		sprintf(portstring, "%d", port);

		struct addrinfo* addressList = 0;
		int res = getaddrinfo(0, portstring, &hints, &addressList);

		if (res)
		{
#ifdef __WXMSW__
			return ConvertMSWErrorCode(res);
#else
			return res;
#endif
		}

		for (struct addrinfo* addr = addressList; addr; addr = addr->ai_next)
		{
			m_fd = socket(addr->ai_family, addr->ai_socktype, addr->ai_protocol);
#ifdef __WXMSW__
			res = ConvertMSWErrorCode(WSAGetLastError());
#else
			res = errno;
#endif
			if (m_fd == -1)
				continue;

			SetNonblocking(m_fd);

			res = bind(m_fd, addr->ai_addr, addr->ai_addrlen);
			if (!res)
				break;
#ifdef __WXMSW__
			res = ConvertMSWErrorCode(WSAGetLastError());
			closesocket(m_fd);
#else
			res = errno;
			close(m_fd);
#endif
			m_fd = -1;
		}
		freeaddrinfo(addressList);
		if (m_fd == -1)
			return res;
	}

	int res = listen(m_fd, 1);
	if (res)
	{
#ifdef __WXMSW__
		res = ConvertMSWErrorCode(res);
		closesocket(m_fd);
#else
		close(m_fd);
#endif
		m_fd = -1;
		return res;
	}

	m_state = listening;

	m_pSocketThread = new CSocketThread();
	m_pSocketThread->SetSocket(this);

	m_pSocketThread->m_waiting = WAIT_ACCEPT;

	m_pSocketThread->Start();

	return 0;
}

int CSocket::GetLocalPort(int& error)
{
	struct sockaddr_storage addr;
	socklen_t addr_len = sizeof(addr);
	error = getsockname(m_fd, (sockaddr*)&addr, &addr_len);
	if (error)
	{
#ifdef __WXMSW__
		error = ConvertMSWErrorCode(error);
#endif
		return -1;
	}

	if (addr.ss_family == AF_INET)
	{
		struct sockaddr_in* addr_v4 = (sockaddr_in*)&addr;
		return ntohs(addr_v4->sin_port);
	}
	else if (addr.ss_family == AF_INET6)
	{
		struct sockaddr_in6* addr_v6 = (sockaddr_in6*)&addr;
		return ntohs(addr_v6->sin6_port);
	}

	error = EINVAL;
	return -1;
}

int CSocket::GetRemotePort(int& error)
{
	struct sockaddr_storage addr;
	socklen_t addr_len = sizeof(addr);
	error = getpeername(m_fd, (sockaddr*)&addr, &addr_len);
	if (error)
	{
#ifdef __WXMSW__
		error = ConvertMSWErrorCode(error);
#endif
		return -1;
	}

	if (addr.ss_family == AF_INET)
	{
		struct sockaddr_in* addr_v4 = (sockaddr_in*)&addr;
		return ntohs(addr_v4->sin_port);
	}
	else if (addr.ss_family == AF_INET6)
	{
		struct sockaddr_in6* addr_v6 = (sockaddr_in6*)&addr;
		return ntohs(addr_v6->sin6_port);
	}

	error = EINVAL;
	return -1;
}

CSocket* CSocket::Accept(int &error)
{
	if (m_pSocketThread)
	{
		m_pSocketThread->m_sync.Lock();
		m_pSocketThread->m_waiting |= WAIT_ACCEPT;
		m_pSocketThread->WakeupThread(true);
		m_pSocketThread->m_sync.Unlock();
	}
	int fd = accept(m_fd, 0, 0);
	if (fd == -1)
	{
#ifdef __WXMSW__
		error = ConvertMSWErrorCode(WSAGetLastError());
#else
		error = errno;
#endif
		return 0;
	}
	
#if defined(SO_NOSIGPIPE) && !defined(MSG_NOSIGNAL)
	// We do not want SIGPIPE if writing to socket.
	const int value = 1;
	setsockopt(fd, SOL_SOCKET, SO_NOSIGPIPE, &value, sizeof(int));
#endif

	SetNonblocking(fd);

	DoSetBufferSizes(fd, m_buffer_sizes[0], m_buffer_sizes[1]);

	CSocket* pSocket = new CSocket(0);
	pSocket->m_state = connected;
	pSocket->m_fd = fd;
	pSocket->m_pSocketThread = new CSocketThread();
	pSocket->m_pSocketThread->SetSocket(pSocket);
	pSocket->m_pSocketThread->m_waiting = WAIT_READ | WAIT_WRITE;
	pSocket->m_pSocketThread->Start();

	return pSocket;
}

int CSocket::SetNonblocking(int fd)
{
	// Set socket to non-blocking.
#ifdef __WXMSW__
	unsigned long nonblock = 1;
	int res = ioctlsocket(fd, FIONBIO, &nonblock);
	if (!res)
		return 0;
	else
		return ConvertMSWErrorCode(WSAGetLastError());
#else
	int flags = fcntl(fd, F_GETFL);
	if (flags == -1)
		return errno;
	int res = fcntl(fd, F_SETFL, flags | O_NONBLOCK);
	if (res == -1)
		return errno;
	return 0;
#endif
}

void CSocket::SetFlags(int flags)
{
	if (m_pSocketThread)
		m_pSocketThread->m_sync.Lock();

	if (m_fd != -1)
		DoSetFlags(m_fd, flags, flags ^ m_flags);
	m_flags = flags;

	if (m_pSocketThread)
		m_pSocketThread->m_sync.Unlock();
}

int CSocket::DoSetFlags(int fd, int flags, int flags_mask)
{
	if (flags_mask & flag_nodelay)
	{
		const int value = (flags & flag_nodelay) ? 1 : 0;
		int res = setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, (const char*)&value, sizeof(value));
		if (res != 0)
#ifdef __WXMSW__
			return ConvertMSWErrorCode(WSAGetLastError());
#else
			return errno;
#endif
	}
	if (flags_mask & flag_keepalive)
	{
		const int value = (flags & flag_keepalive) ? 1 : 0;
		int res = setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, (const char*)&value, sizeof(value));
		if (res != 0)
#ifdef __WXMSW__
			return ConvertMSWErrorCode(WSAGetLastError());
#else
			return errno;
#endif
	}

	return 0;
}

void CSocket::SetBufferSizes(int size_read, int size_write)
{
	if (m_pSocketThread)
		m_pSocketThread->m_sync.Lock();

	m_buffer_sizes[0] = size_read;
	m_buffer_sizes[1] = size_write;

	if (m_fd != -1)
		DoSetBufferSizes(m_fd, size_read, size_write);

	if (m_pSocketThread)
		m_pSocketThread->m_sync.Unlock();
}

int CSocket::DoSetBufferSizes(int fd, int size_read, int size_write)
{
	if (size_read != -1)
	{
		int res = setsockopt(fd, SOL_SOCKET, SO_RCVBUF, (const char*)&size_read, sizeof(size_read));
		if (res != 0)
#ifdef __WXMSW__
			return ConvertMSWErrorCode(WSAGetLastError());
#else
			return errno;
#endif
	}

	if (size_write != -1)
	{
		int res = setsockopt(fd, SOL_SOCKET, SO_SNDBUF, (const char*)&size_write, sizeof(size_write));
		if (res != 0)
#ifdef __WXMSW__
			return ConvertMSWErrorCode(WSAGetLastError());
#else
			return errno;
#endif
	}

	return 0;
}