socket.h

ReactOS是一些高手根据Windows XP的内核编写出的类XP。内核实现机理和API函数调用几乎相同。甚至可以兼容XP的程序。喜欢研究系统内核的人可以看一看。

/*
 * COPYRIGHT:   See COPYING in the top level directory
 * PROJECT:     ReactOS HTTP Daemon
 * FILE:        include/socket.h
 */
#ifndef __SOCKET_H
#define __SOCKET_H
#include <stdio.h>
#include <winsock2.h>
#include <thread.h>
#include <list.h>
#include <exception>
#include <assert.h>

#define MAX_PENDING_CONNECTS 4      // The backlog allowed for listen()

VOID InitWinsock();
VOID DeinitWinsock();

class CSocket;
class CClientSocket;
class CServerClientSocket;
class CServerClientThread;
class CServerSocket;

typedef CSocket* LPCSocket;
typedef CClientSocket* LPCClientSocket;
typedef CServerClientSocket* LPCServerClientSocket;
typedef CServerClientThread* LPCServerClientThread;
typedef CServerSocket* LPCServerSocket;

class ESocket {
public:
	ESocket() { Description = NULL; }
	ESocket(LPTSTR description)  { Description = description; }
	LPTSTR what()  { return Description; }
protected:
	LPTSTR Description;
};

class ESocketWinsock : public ESocket {
public:
	ESocketWinsock(LPTSTR description) { Description = description; }
};

class ESocketDll : public ESocket {
public:
	ESocketDll(LPTSTR description) { Description = description; }
};

class ESocketOpen : public ESocket {
public:
	ESocketOpen(LPTSTR description) { Description = description; }
};

class ESocketClose : public ESocket {
public:
	ESocketClose(LPTSTR description) { Description = description; }
};

class ESocketSend : public ESocket {
public:
	ESocketSend(LPTSTR description) { Description = description; }
};

class ESocketReceive : public ESocket {
public:
	ESocketReceive(LPTSTR description) { Description = description; }
};


class CSocket {
public:
	CSocket();
	virtual ~CSocket();
	virtual SOCKET GetSocket();
	virtual VOID SetSocket(SOCKET socket);
	virtual SOCKADDR_IN GetSockAddrIn();
	virtual VOID SetSockAddrIn(SOCKADDR_IN sockaddrin);
	virtual VOID SetEvents(LONG lEvents);
	virtual LONG GetEvents();
    virtual VOID SetPort( UINT nPort) {};
	virtual VOID Open();
	virtual VOID Close();
	virtual INT Transmit( LPCSTR lpsBuffer,  UINT nLength) { return 0; };
	virtual INT Receive(LPSTR lpsBuffer,  UINT nLength) { return 0; };
	virtual INT SendText( LPCSTR lpsStr) { return 0; };
protected:
	SOCKET Socket;
	SOCKADDR_IN SockAddrIn;
	WSAEVENT Event;
	UINT Port;
	BOOL Active;
private:
	LONG Events;
};

class CServerClientSocket : public CSocket {
public:
	CServerClientSocket() {};
	CServerClientSocket(LPCServerSocket lpServerSocket);
	CServerSocket *GetServerSocket();
	virtual INT Transmit( LPCSTR lpsBuffer,  UINT nLength);
	virtual INT Receive(LPSTR lpsBuffer,  UINT nLength);
	virtual INT SendText( LPCSTR lpsText);
	virtual VOID MessageLoop();
	virtual VOID OnRead() {};
	//virtual VOID OnWrite() {};
	virtual VOID OnClose() {};
protected:
	LPCServerSocket ServerSocket;
};

class CServerClientThread : public CThread {
public:
	CServerClientThread() {};
	CServerClientThread(CServerClientSocket *socket);
	virtual ~CServerClientThread();
protected:
	CServerClientSocket *ClientSocket;
};

class CServerSocket : public CSocket {
public:
	CServerSocket();
	virtual ~CServerSocket();
	virtual VOID SetPort( UINT nPort);
	virtual VOID Open();
	virtual VOID Close();
	virtual LPCServerClientSocket OnGetSocket(LPCServerSocket lpServerSocket);
	virtual LPCServerClientThread OnGetThread(LPCServerClientSocket lpSocket);
	virtual VOID OnAccept( LPCServerClientThread lpThread) {};
	virtual VOID MessageLoop();
	VOID InsertClient(LPCServerClientThread lpClient);
	VOID RemoveClient(LPCServerClientThread lpClient);
protected:
	CList<LPCServerClientThread> Connections;
};

#endif /* __SOCKET_H */