?? hmtsocket.cpp
字號:
// HMTSocket.cpp: implementation of the CMTSocket class.
//
//////////////////////////////////////////////////////////////////////
#include "stdafx.h"
#include "HMTSocket.h"
#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif
//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////
CMTSocket::CMTSocket()
{
m_Socket = NULL;
m_SrvrAddr=NULL;
m_Port=0;
memset(&m_Sa,0,sizeof(m_Sa));
}
CMTSocket::~CMTSocket()
{
}
//返回m_Socket的值用于操作：
SOCKET CMTSocket::GetSocket()
{
return m_Socket;
}
/************************/
/* 與服務器相關的代碼：*/
/************************/
//創建一個套接字：
BOOL CMTSocket::Create(unsigned int port, LPCTSTR Srvr, unsigned int socktype)
{
if(!AfxSocketInit())
{
throw new CMTSocketException("初始化Winsock動態連接庫出錯！");
return FALSE;
}
if((m_Socket=socket(AF_INET,socktype,0))==INVALID_SOCKET)
{
throw new CMTSocketException("創建套接口失??！");
return FALSE;
}
m_SrvrAddr=Srvr;
m_Port=port;
return TRUE;
}
BOOL CMTSocket::Bind()
{
memset(&m_Sa,0,sizeof(m_Sa));
//準備sockaddr_in 結構：
m_Sa.sin_family=AF_INET;
m_Sa.sin_port=htons(m_Port);
m_Sa.sin_addr.s_addr=INADDR_ANY;
//進行捆綁：
if(bind(m_Socket,(sockaddr * )&m_Sa,sizeof(m_Sa))== SOCKET_ERROR)
{
throw new CMTSocketException("捆綁失敗！");
return FALSE;
}
return TRUE;
}
BOOL CMTSocket::Listen()
{
if(listen(m_Socket,5)==SOCKET_ERROR)
{
throw new CMTSocketException("Listen（）調用失敗！");
return FALSE;
}
return TRUE;
}
SOCKET CMTSocket::Accept()
{
SOCKET sock=0;
int nl=sizeof(m_Sa);
if((sock=accept(m_Socket,(sockaddr *)&m_Sa,&nl))==INVALID_SOCKET)
{
throw new CMTSocketException("Accept調用失?。?quot;);
return FALSE;
}
m_Clients.AddTail(sock);
return sock;
}
BOOL CMTSocket::Close(SOCKET sock)
{
if(sock==NULL)
{
throw new CMTSocketException("套接字為0，無法關閉！");
return FALSE;
}
if(closesocket(sock)==SOCKET_ERROR)
{
throw new CMTSocketException("關閉套接口失??！");
return FALSE;
}
return TRUE;
}
BOOL CMTSocket::Close()
{
return Close(m_Socket);
}
/************************/
/* 與客戶機相關的代碼：*/
/************************/
//連接到服務器：
BOOL CMTSocket::Connect()
{
if(m_SrvrAddr==NULL)
{
WSASetLastError(WSAEINVAL);
throw new CMTSocketException("連接失敗！");
return FALSE;
}
memset(&m_Sa,0,sizeof(m_Sa));
m_Sa.sin_family = AF_INET;
m_Sa.sin_addr.s_addr = inet_addr(( LPSTR )m_SrvrAddr);
if (m_Sa.sin_addr.s_addr == INADDR_NONE)
{
LPHOSTENT lphost;
lphost = gethostbyname(( LPSTR )m_SrvrAddr);
if (lphost != NULL)
{
m_Sa.sin_addr.s_addr = ((LPIN_ADDR)lphost->h_addr)->s_addr;
}
else
{
WSASetLastError(WSAEINVAL);
throw new CMTSocketException("連接失??！");
return FALSE;
}
}
m_Sa.sin_port = htons((u_short)m_Port);
if(connect(m_Socket,(sockaddr * )&m_Sa,sizeof(m_Sa))==SOCKET_ERROR)
{
if(GetLastError()!=WSAECONNREFUSED)
{
throw new CMTSocketException("連接失??！");
return FALSE;
}
else
{
return FALSE;
}
}
return TRUE;
}
//設置套接字屬性：
BOOL CMTSocket::SetSockOpt(SOCKET sock,
int nOptionName,
const char * lpOptionValue,
int nOptionLen,
int nLevel
)
{
if(setsockopt(sock,nLevel,nOptionName,lpOptionValue,nOptionLen)==SOCKET_ERROR)
{
throw new CMTSocketException("套接口設置失?。?quot;);
return FALSE;
}
return TRUE;
}
BOOL CMTSocket::SetSockOpt(int nOptionName,
const char * lpOptionValue,
int nOptionLen,
int nLevel
)
{
return SetSockOpt(m_Socket,nOptionName,lpOptionValue,nOptionLen,nLevel);
}
/************************/
/* 與讀、寫相關的代碼：*/
/************************/
//從套接口讀取字節流：
int CMTSocket::Receive(SOCKET sock,char FAR* buf,int len,int RecvAll,int flags)
{
int re;
re=1;
if(RecvAll)
{
while(re!=SOCKET_ERROR&&len>0)
{
if((re=recv(sock,buf,len,flags))==SOCKET_ERROR)
{
throw new CMTSocketException("讀取數據失敗！");
return SOCKET_ERROR;
}
buf+=re;
len-=re;
}
}
else
{
if((re=recv(sock,buf,len,flags))==SOCKET_ERROR)
{
throw new CMTSocketException("讀取數據失?。?quot;);
return SOCKET_ERROR;
}
}
return re;
}
int CMTSocket::Receive(char FAR* buf,int len,int flags)
{
return Receive(m_Socket,buf,len,flags);
}
//從套接口讀取字節流，并且返回數據的源地址：
int CMTSocket::ReceiveFrom(SOCKET sock,char * lpBuf, int nBufLen,
CString& rSocketAddress,
UINT& rSocketPort,
int nFlags)
{
int re;
memset(&m_Sa, 0, sizeof(m_Sa));
int nl= sizeof(m_Sa);
re=1;
re=recvfrom(sock,lpBuf,nBufLen,nFlags,(sockaddr * )&m_Sa, &nl);
if(re!= SOCKET_ERROR)
{
rSocketPort = ntohs(m_Sa.sin_port);
rSocketAddress = inet_ntoa(m_Sa.sin_addr);
}
else
{
throw new CMTSocketException("以Recvfrom（）方式讀取數據失敗！");
return FALSE;
}
return re;
}
int CMTSocket::ReceiveFrom(char * lpBuf, int nBufLen,
CString& rSocketAddress,
UINT& rSocketPort,
int RecvAll,int nFlags)
{
return ReceiveFrom(m_Socket,lpBuf,nBufLen,rSocketAddress,rSocketPort,nFlags);
}
//發送數據：
int CMTSocket::Send(SOCKET sock,const char * buf, int len, int flags)
{
int re;
if((re=send(sock,buf,len,flags))==SOCKET_ERROR)
{
throw new CMTSocketException("發送數據失??！");
return SOCKET_ERROR;
}
return re;
}
int CMTSocket::Send(const char * buf, int len, int flags)
{
return Send(m_Socket,buf,len,flags);
}
//發送數據，并且返回地址：
int CMTSocket::SendTo(SOCKET sock,
const char * lpBuf,
int nBufLen,
UINT nHostPort,
LPCTSTR lpszHostAddress,
int nFlags)
{
int re;
memset(&m_Sa,0,sizeof(m_Sa));
LPSTR addr=(LPSTR)m_SrvrAddr;
m_Sa.sin_family = AF_INET;
if (addr == NULL)
{
m_Sa.sin_addr.s_addr = htonl(INADDR_BROADCAST);
}
else
{
m_Sa.sin_addr.s_addr = inet_addr(addr);
if (m_Sa.sin_addr.s_addr == INADDR_NONE)
{
LPHOSTENT lphost;
lphost = gethostbyname(addr);
if (lphost != NULL)
{
m_Sa.sin_addr.s_addr = ((LPIN_ADDR)lphost->h_addr)->s_addr;
}
else
{
WSASetLastError(WSAEINVAL);
throw new CMTSocketException("以snedto（）方式發送數據失??！");
return FALSE;
}
}
}
m_Sa.sin_port = htons((u_short)nHostPort);
int nl=sizeof(m_Sa);
if((re=sendto(sock,lpBuf,nBufLen,nFlags,(sockaddr *)&m_Sa,nl))==SOCKET_ERROR)
{
throw new CMTSocketException("以snedto（）方式發送數據失敗！");
return SOCKET_ERROR;
}
return re;
}
int CMTSocket::SendTo(const char * lpBuf,
int nBufLen,
UINT nHostPort,
LPCTSTR lpszHostAddress,
int nFlags)
{
return SendTo(m_Socket,lpBuf,nBufLen,nHostPort,lpszHostAddress,nFlags);
}
//利用該函數“文明地”關閉套接口：
BOOL CMTSocket::ShutDown(SOCKET sock)
{
if(shutdown(sock,0)==SOCKET_ERROR)
{
throw new CMTSocketException("文明關閉套接口失?。?quot;);
return FALSE;
}
return TRUE;
}
BOOL CMTSocket::ShutDown()
{
return ShutDown(m_Socket);
}
//確定套接口的狀態：
BOOL CMTSocket::Select(fd_set FAR * readfds,
fd_set FAR * writefds,
fd_set FAR * exceptfds,
const struct timeval FAR * timeout
)
{
if(select(0,readfds,writefds,exceptfds,timeout)==SOCKET_ERROR)
{
throw new CMTSocketException("確定套接口調用失敗！");
return FALSE;
}
return TRUE;
}
//////////////////////////////////////////////////////////////////////
// CMTSocketException Class
//////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////
CMTSocketException::CMTSocketException(LPCTSTR einfo)
{
m_EInfo=einfo;
m_EInfo=m_EInfo+"\r\n";
}
CMTSocketException::~CMTSocketException()
{
}
LPCTSTR CMTSocketException::GetErrorMsg()
{
m_ErrorCode=WSAGetLastError();
switch(m_ErrorCode)
{
case WSAEACCES:
{
m_EInfo=m_EInfo+"錯誤WSAEACCES：對套接口的訪問方式非法！";
break;
}
case WSAEADDRINUSE:
{
m_EInfo=m_EInfo+"錯誤WSAEADDRINUSE：試圖將套接口捆綁到正在使用的地址或端口！";
break;
}
case WSAEADDRNOTAVAIL:
{
m_EInfo=m_EInfo+"錯誤WSAEADDRNOTAVAIL：指定的地址或端口非法！";
break;
}
case WSAEAFNOSUPPORT:
{
m_EInfo=m_EInfo+"錯誤WSAEAFNOSUPPORT：地址同目前協議不兼容！";
break;
}
case WSAEALREADY:
{
m_EInfo=m_EInfo+"錯誤WSAEALREADY：當前操作正在執行！";
break;
}
case WSAECONNABORTED:
{
m_EInfo=m_EInfo+"錯誤WSAECONNABORTED：同服務器的連接中斷！";
break;
}
case WSAECONNREFUSED:
{
m_EInfo=m_EInfo+"錯誤WSAECONNREFUSED：同服務器的連接被拒絕！";
break;
}
case WSAECONNRESET:
{
m_EInfo=m_EInfo+"錯誤WSAECONNRESET：同服務器的連接被服務器強行中斷！";
break;
}
case WSAEDESTADDRREQ:
{
m_EInfo=m_EInfo+"錯誤WSAEDESTADDRREQ：沒有指明目標地址！";
break;
}
case WSAEFAULT:
{
m_EInfo=m_EInfo+"錯誤WSAEFAULT：錯誤的地址！";
break;
}
case WSAEHOSTDOWN:
{
m_EInfo=m_EInfo+"錯誤WSAEHOSTDOWN：服務器死鎖！";
break;
}
case WSAEHOSTUNREACH:
{
m_EInfo=m_EInfo+"錯誤WSAEHOSTUNREACH：試圖同無法到達的服務器相連接！";
break;
}
case WSAEINPROGRESS:
{
m_EInfo=m_EInfo+"錯誤WSAEINPROGRESS：只允許有一個阻塞的函數調用！";
break;
}
case WSAEINTR:
{
m_EInfo=m_EInfo+"錯誤WSAEINTR：阻塞函數調用被終止！";
break;
}
case WSAEINVAL:
{
m_EInfo=m_EInfo+"錯誤WSAEINVAL：參數無效！";
break;
}
case WSAEISCONN:
{
m_EInfo=m_EInfo+"錯誤WSAEISCONN：套接口處于連接狀態中！";
break;
}
case WSAEMFILE:
{
m_EInfo=m_EInfo+"錯誤WSAEMFILE：被打開的套接口太多！";
break;
}
case WSAEMSGSIZE:
{
m_EInfo=m_EInfo+"錯誤WSAEMSGSIZE：數據報套接口中傳送的信息太長！";
break;
}
case WSAENETDOWN :
{
m_EInfo=m_EInfo+"錯誤WSAENETDOWN：網絡系統死鎖！";
break;
}
case WSAENETRESET :
{
m_EInfo=m_EInfo+"錯誤WSAENETRESET：操作過程出錯，連接中斷！";
break;
}
case WSAENETUNREACH :
{
m_EInfo=m_EInfo+"錯誤WSAENETUNREACH：無法連接到網絡！";
break;
}
case WSAENOBUFS :
{
m_EInfo=m_EInfo+"錯誤WSAENOBUFS：緩沖區已滿，無法進行操作！";
break;
}
case WSAENOPROTOOPT :
{
m_EInfo=m_EInfo+"錯誤WSAENOPROTOOPT：無效的套接口選項！";
break;
}
case WSAENOTCONN :
{
m_EInfo=m_EInfo+"錯誤WSAENOTCONN：無法進行讀寫操作！";
break;
}
case WSAENOTSOCK :
{
m_EInfo=m_EInfo+"錯誤WSAENOTSOCK：試圖對非套接口類型的變量進行操作！";
break;
}
case WSAEOPNOTSUPP :
{
m_EInfo=m_EInfo+"錯誤WSAEOPNOTSUPP：不支持這種操作！";
break;
}
case WSAEPFNOSUPPORT :
{
m_EInfo=m_EInfo+"錯誤WSAEPFNOSUPPORT：不支持當前協議族！";
break;
}
case WSAEPROCLIM :
{
m_EInfo=m_EInfo+"錯誤WSAEPROCLIM：使用Windows Sock的應用程序太多！";
break;
}
case WSAEPROTONOSUPPORT :
{
m_EInfo=m_EInfo+"錯誤WSAEPROTONOSUPPORT：當前協議不被支持！";
break;
}
case WSAEPROTOTYPE :
{
m_EInfo=m_EInfo+"錯誤WSAEPROTOTYPE：當前協議不支持指定的套接口類型！";
break;
}
case WSAESHUTDOWN :
{
m_EInfo=m_EInfo+"錯誤WSAESHUTDOWN：套接口已經關閉，無法發送數據！";
break;
}
case WSAESOCKTNOSUPPORT :
{
m_EInfo=m_EInfo+"錯誤WSAESOCKTNOSUPPORT：指定的套接口類型不被支持！";
break;
}
case WSAETIMEDOUT :
{
m_EInfo=m_EInfo+"錯誤WSAETIMEDOUT：連接超時！";
break;
}
case 10109:
{
m_EInfo=m_EInfo+"錯誤WSATYPE_NOT_FOUND：無法找到指定的類！";
break;
}
case WSAEWOULDBLOCK :
{
m_EInfo=m_EInfo+"錯誤WSAEWOULDBLOCK：資源暫時無法使用！";
break;
}
case WSAHOST_NOT_FOUND :
{
m_EInfo=m_EInfo+"錯誤WSAHOST_NOT_FOUND：找不到服務器！";
break;
}
case WSANOTINITIALISED:
{
m_EInfo=m_EInfo+"錯誤WSANOTINITIALISED：沒有調用WSAStartup（）初始化！";
break;
}
case WSANO_DATA:
{
m_EInfo=m_EInfo+"錯誤WSANO_DATA：指定的機器名稱存在，但相應的數據不存在！";
break;
}
case WSANO_RECOVERY:
{
m_EInfo=m_EInfo+"錯誤WSANO_RECOVERY：無法恢復的錯誤（同機器名稱的查找相關）！";
break;
}
case WSASYSNOTREADY :
{
m_EInfo=m_EInfo+"錯誤WSASYSNOTREADY：Windows Socket 系統不能工作！";
break;
}
case WSATRY_AGAIN :
{
m_EInfo=m_EInfo+"錯誤WSATRY_AGAIN：主機名解析時沒有發現授權服務器！";
break;
}
case WSAVERNOTSUPPORTED:
{
m_EInfo=m_EInfo+"錯誤WSAVERNOTSUPPORTED：無法初始化服務提供者！";
break;
}
case WSAEDISCON:
{
m_EInfo=m_EInfo+"錯誤WSAEDISCON：服務器已經\"文明地\"關閉了！";
break;
}
default:
{
m_EInfo=m_EInfo+"未知網絡錯誤！";
break;
}
};
return m_EInfo;
}
DWORD CMTSocketException::GetErrorNumber()
{
return m_ErrorCode=WSAGetLastError();
}
?? 快捷鍵說明
復制代碼
Ctrl + C
搜索代碼
Ctrl + F
全屏模式
F11
切換主題
Ctrl + Shift + D
顯示快捷鍵
?
增大字號
Ctrl + =
減小字號
Ctrl + -