hmtsocket.h

VC++高级编程技巧与示例

// HMTSocket.h: interface for the CMTSocket class.
//
//////////////////////////////////////////////////////////////////////

#if !defined(AFX_HMTSOCKET_H__13A17B64_2ACF_4585_911D_F03483039B65__INCLUDED_)
#define AFX_HMTSOCKET_H__13A17B64_2ACF_4585_911D_F03483039B65__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

#include <afxsock.h>		// MFC socket extensions
#include <afxtempl.h>
#include "Afx.h"

class CMTSocketException
{
public:
	DWORD m_ErrorCode;
	CString m_EInfo;
	int m_ErrorType;
	DWORD GetErrorNumber();
	LPCTSTR GetErrorMsg();
	CMTSocketException(LPCTSTR einfo);
	virtual ~CMTSocketException();
};

class CMTSocket  
{

	LPCTSTR m_SrvrAddr;
	int m_Port;
	SOCKET m_Socket;
	sockaddr_in m_Sa;
	CList <SOCKET,SOCKET> m_Clients;
public:
	SOCKET GetSocket();
	SOCKET Accept();
	BOOL Bind();
	BOOL Close();
	BOOL Close(SOCKET sock);
	CMTSocket();
	virtual ~CMTSocket();
	BOOL Connect();
	BOOL Create(unsigned port=0, LPCTSTR Srvr=0, unsigned socktype=SOCK_STREAM);
	BOOL Listen();
	int Receive(SOCKET sock,char FAR* buf,int len,int RecvAll=TRUE,int flags=0);
	int Receive(char FAR* buf,int len,int flags=0);
	int ReceiveFrom(SOCKET sock,
		             char * lpBuf, int nBufLen,
		             CString& rSocketAddress, 
					 UINT& rSocketPort,
					 int nFlags=0
					 );
	int ReceiveFrom(char * lpBuf, int nBufLen,
		             CString& rSocketAddress, 
					 UINT& rSocketPort,
					 int RecvAll=TRUE,
					 int nFlags=0
					 );
	BOOL Select(fd_set FAR * readfds,               
				fd_set FAR * writefds,              
				fd_set FAR * exceptfds,             
				const struct timeval FAR * timeout);
	int Send(SOCKET sock,const char * buf, int len, int flags=0);
	int Send(const char * buf, int len, int flags=0);
	int SendTo(SOCKET sock,
		        const char * lpBuf,
		        int nBufLen,
		        UINT nHostPort,
				LPCTSTR lpszHostAddress=NULL,
				int nFlags = 0);
	int SendTo(const char * lpBuf,
		        int nBufLen,
		        UINT nHostPort,
				LPCTSTR lpszHostAddress=NULL,
				int nFlags = 0);
	BOOL SetSockOpt(SOCKET sock,
		            int nOptionName,
					const char * lpOptionValue, 
					int nOptionLen, 
					int nLevel = SOL_SOCKET
					);
	BOOL SetSockOpt(int nOptionName,
					const char * lpOptionValue, 
					int nOptionLen, 
					int nLevel = SOL_SOCKET
					);
	BOOL ShutDown(SOCKET sock);
	BOOL ShutDown();

};
#endif // !defined(AFX_HMTSOCKET_H__13A17B64_2ACF_4585_911D_F03483039B65__INCLUDED_)