WinSock溫習一

sock編程，之前接觸過一段時間，但是沒空寫寫。

現在先從最基本的select模型開始吧，簡單的echoserver示例。

 

服務端，sock創建的步驟，順序不能亂

1.WSAStartup  // 加載winsock.dll

2.sock              // 創建sock

3.bind              // 綁定sock

4.listen            // 偵聽sock

5.select           // select模型的關鍵

6.accept          // 響應客戶端的連接

7.recv              // 接受數據

8.WSACleanup // winsock.dll卸載

 

其中，需要注意MAKEWORD宏接受的兩個參數分別是主版本號和次版本號，WSADATA結構的wVersion參數低位上存放主版本號，高位上存放次版本號。

 

#Server

#define FD_SETSIZE 256 #include <WINSOCK2.H> #include <iostream> #pragma comment(lib, "WSOCK32") class CWinSockServer { public: CWinSockServer(); virtual ~CWinSockServer(); public: bool InitSocket(); void StartServer(); void ClearSock() ; private: int m_nPort; SOCKET m_sock; char m_szBuf[1024]; std::vector<SOCKET> m_vecSock; };

 

CWinSockServer::CWinSockServer() :m_nPort(9714) { ::memset(m_szBuf, 0, sizeof(m_szBuf)); } CWinSockServer::~CWinSockServer() { } void CWinSockServer::ClearSock() { ::WSACleanup(); } bool CWinSockServer::InitSocket() { WORD wVersion = MAKEWORD(2, 0); WSADATA wsaData; int nRet = WSAStartup(wVersion, &wsaData); if (0 != nRet) { std::cout << "/nWSAStartup(wVersion, &wsaData) Error:" << ::WSAGetLastError(); return false; } if (LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 0) { ClearSock(); std::cout << "/nthe request version could not found./n"; return false; } return true; } void CWinSockServer::StartServer() { if (!InitSocket()) { return; } m_sock = ::socket(AF_INET, SOCK_STREAM, 0); if (INVALID_SOCKET == m_sock) { std::cout << "/nCreate sock error." << ::WSAGetLastError(); ClearSock(); return; } SOCKADDR_IN sockAddr; sockAddr.sin_family = AF_INET; sockAddr.sin_port = ::htons(m_nPort); sockAddr.sin_addr.S_un.S_addr = htonl(INADDR_ANY); if (SOCKET_ERROR == ::bind(m_sock, (SOCKADDR*)&sockAddr, sizeof(sockAddr))) { std::cout << "bind error" << WSAGetLastError(); ClearSock(); return; } listen(m_sock, FD_SETSIZE); // 非阻塞 unsigned long unBlock = 1; if(SOCKET_ERROR == ::ioctlsocket(m_sock, FIONBIO, (unsigned long*)&unBlock)) { std::cout << "/nSet unblock error."; ClearSock(); return; } std::cout << "Sock Server run.../n"; fd_set fdRead; timeval time = {0, 1}; while(true) { FD_ZERO(&fdRead); FD_SET(m_sock, &fdRead); for(size_t i=0; i < m_vecSock.size(); ++i) { FD_SET(m_vecSock[i], &fdRead); } //fd_set fdRead = fdSocket; if(::select(FD_SETSIZE, &fdRead, NULL, NULL, &time) > 0) { if (FD_ISSET(m_sock, &fdRead)) { SOCKADDR_IN addr; int nLen = sizeof(addr); SOCKET client = ::accept(m_sock, (SOCKADDR*)&addr, &nLen); std::cout << "RSCV : " << ::inet_ntoa(addr.sin_addr) << "/n"; m_vecSock.push_back(client); } else { for (std::size_t i=0; i < m_vecSock.size(); ++i) { if (FD_ISSET(m_vecSock[i], &fdRead)) { ::memset(m_szBuf, 0, sizeof(m_szBuf)); int nRecv = ::recv(m_vecSock[i], m_szBuf, sizeof(m_szBuf), 0); if (nRecv > 0) { std::cout << "/nRecv Data(" << sizeof(m_szBuf) << "):[[" << m_szBuf << "]]/n"; } else { ::closesocket(m_vecSock[i]); FD_CLR(m_vecSock[i], &fdRead); std::cout << "/n------------Client closed." << "--------/n"; } } } } } } FD_CLR(m_sock, &fdRead); closesocket(m_sock); for (std::size_t i = 0; i < m_vecSock.size(); ++i) { FD_CLR(m_vecSock[i], &fdRead); closesocket(m_vecSock[i]); } }

 

 

 

#client

客戶端比較簡單。創建步驟：

1.WSAStartup  // 加載winsock.dll

2.sock              // 創建sock

3.connect         // 連接服務端

4.send             // 發送數據

5.WSACleanup // winsock.dll卸載

 #include <WINSOCK2.H> #include <iostream> #include <string> #pragma comment(lib, "WSOCK32") class CWinsockClient { public: CWinsockClient(); virtual ~CWinsockClient(); public: bool InitClient(); bool SendData(const char* pBuf, unsigned int unLen); void ClearSock(); void DestorySock(); private: SOCKET m_sock; int m_nPort; std::string m_strIP; };

 

CWinsockClient::CWinsockClient() :m_nPort(9714), m_strIP("127.0.0.1") { } CWinsockClient::~CWinsockClient() { } void CWinsockClient::ClearSock() { ::WSACleanup(); } bool CWinsockClient::InitClient() { WSADATA wsaData; int nRet = ::WSAStartup(MAKEWORD(2, 0), &wsaData); if (nRet) { std::cout << "/nWSAStartup failed with error " << WSAGetLastError(); return false; } m_sock = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); if (INVALID_SOCKET == m_sock) { std::cout <<"/nsocket failed with error " << WSAGetLastError(); ClearSock(); return false; } SOCKADDR_IN serverAddr; serverAddr.sin_family = AF_INET; serverAddr.sin_port = htons(m_nPort); serverAddr.sin_addr.S_un.S_addr = inet_addr(m_strIP.c_str()); std::cout << "/nTrying to connect to " << ::inet_ntoa(serverAddr.sin_addr) << ": " << ::htons(serverAddr.sin_port); if (SOCKET_ERROR == ::connect(m_sock, (SOCKADDR*)&serverAddr, sizeof(serverAddr))) { std::cout << "/nConnect failed with error " << WSAGetLastError(); closesocket(m_sock); ClearSock(); return false; } std::cout << "/nConnection success."; return true; } bool CWinsockClient::SendData(const char* pBuf, unsigned int unLen) { if (NULL == pBuf) { return false; } int nRet = ::send(m_sock, pBuf, unLen, 0); if (SOCKET_ERROR == nRet) { std::cout << "/nSend failed with error " << WSAGetLastError(); closesocket(m_sock); WSACleanup(); return false; } std::cout << "Send data success. " << nRet << "byte./n"; return true; } void CWinsockClient::DestorySock() { closesocket(m_sock); ClearSock(); }