重構阻塞式客戶端程序代碼

     在這幾個月中閱讀和編寫了大量的套接字程序，所以對套接字 程序的編寫也有了新的認識 。現在這裏分享出來與大家交流一下，不足之處還望大家指正。

    這裏的示例程序是修改自 Winsocket入門教程一：多線程阻塞式服務器和阻塞式客戶端程序(TCP) 中的客戶端程序。相對於原來的程序，主要進行了以下幾個方面的處理：

    1.添加了更好的錯誤處理代碼。使用GOTO語句將錯誤處理放到一起，可以使錯誤處理的代碼更利於維護，並且減少代碼的重複(Code duplicate)

    2.使用微軟推薦的函數getaddrinfo函數獲取連接地址，不過冒似使用這種方法要寫的代碼更多。

    3.發送和接收時使用循環發送和接收，以處理一次髮送或者接收不完的情況。

    示例代碼如下

    // 此示例程序參考MSDN Winsocket Client Demo // 示例和自己處理Winsocket程序的經驗而成 // 不足和錯誤之處望大家指正 // 原始DEMO地址：http://msdn.microsoft.com/zh-cn/library/ms737591%28v=VS.85%29.aspx // 如果在引用WinSock2.h的同時需要引用Windows.h // 必須定義下列宏。因爲Windows.h中包含了WinSock.h // 如果不定義下列宏，會引起數據結構和函數重定義的問題 #ifndef WIN32_LEAN_AND_MEAN #define WIN32_LEAN_AND_MEAN #endif #include <cstdio> #include <cstdlib> #include <Windows.h> #include <WinSock2.h> #include <WS2tcpip.h> #include <IPHlpApi.h> // Winsocket程序需要鏈接ws2_32.lib #pragma comment(lib, "ws2_32.lib") int main() { int iRet = 0; WSADATA wd; // Winsocket程序必須要調用WSAStartup對 // ws2_32.dll的初始化，否則下面的函數調用 // 均會失敗 iRet = WSAStartup(MAKEWORD(2, 2), &wd); if (SOCKET_ERROR == iRet) { printf("Call WSAStartup failed, errno = %d./r/n", WSAGetLastError()); goto error_handle1; } // 使用getaddrinfo創建進行連接(connect)所需要的地址 addrinfo aiInst; addrinfo *paiResult = NULL; addrinfo *paiThis = NULL; ZeroMemory(&aiInst, sizeof(aiInst)); aiInst.ai_flags = AI_NUMERICHOST; // 使用IP地址，不使用名稱解析 aiInst.ai_family = AF_INET; // 協議族：IPV4 aiInst.ai_socktype = SOCK_STREAM; // 流式套接字 aiInst.ai_protocol = IPPROTO_TCP; // TCP傳輸 const char *pszHost = "127.0.0.1"; const char *pszService = "10001"; // getaddrinfo根據填入的參數 // 可能會生成一個帶有多個地址的鏈表 // 注意必須釋放paiResult iRet = getaddrinfo(pszHost, pszService, &aiInst, &paiResult); if (SOCKET_ERROR == iRet) { printf("call getaddrinfo failed, errno = %d./r/n", WSAGetLastError()); goto error_handle2; } // 由協議族、套接字類型以及協議類型創建所需要的套接字 // 以上三個參數決定了創建套接字的特性。該特性決定了連 // 接的屬性。例如AF_INET、SOCK_STREAM、IPPROTO_TCP // 創建的套接字擁有TCP連接屬性。即由該套接字創建的連接 // 爲TCP連接 SOCKET sktConn = INVALID_SOCKET; for (paiThis = paiResult; NULL != paiThis; paiThis = paiThis->ai_next) { sktConn = socket(paiResult->ai_family, paiResult->ai_socktype, paiResult->ai_protocol); if (INVALID_SOCKET == sktConn) { printf("call socket failed, errno = %d./r/n", WSAGetLastError()); goto error_handle3; } // 由所創建的套接字和地址連接服務器 iRet = connect(sktConn, paiResult->ai_addr, (int)paiResult->ai_addrlen); if (SOCKET_ERROR == iRet) { printf("call connect failed, errno = %d./r/n", WSAGetLastError()); closesocket(sktConn); sktConn = INVALID_SOCKET; continue; } break; } if (INVALID_SOCKET == sktConn) { goto error_handle3; } // 也可以使用下面的方法創建連接地址 // sockaddr_in adrServ; // ZeroMemory(&adrServ, sizeof(sockaddr_in)); // adrServ.sin_family = AF_INET; // adrServ.sin_port = htons(10001); // adrServ.sin_addr.s_addr = inet_addr("127.0.0.1"); // connect(sktConn, (sockaddr*)&adrServ, sizeof(sockaddr_in)); printf("Connect to %s:%s succeed./r/n", pszHost, pszService); // 如果發送的消息過長 // 可能需要發送多次才能發送完成 // 這裏展示一種簡單的發送方法 const char *pszSend = "Hello world!!!"; int iSndLen = (int)strlen(pszSend); int iHadSnd = 0; do { iRet = send(sktConn, pszSend + iHadSnd, iSndLen, 0); if (0 < iRet) { printf("%d bytes send./r/n", iRet); } else if (0 == iRet) { printf("All data send./r/n"); } else { printf("call send failed, errno = %d./r/n", WSAGetLastError()); goto error_handle4; } iHadSnd += iRet; iSndLen -= iRet; } while (0 < iRet); // 如果不再需要發送數據，則應該使用shutdown配合 // SD_SEND關閉該套接字的功能，以減輕服務器端的 // 資源佔用 iRet = shutdown(sktConn, SD_SEND); if (SOCKET_ERROR == iRet) { printf("call shutdown failed, errno = %d./r/n", WSAGetLastError()); goto error_handle4; } // 接收從服務器發來的數據 // 如果消息過長，則需要多次接收 // 這裏展示一種簡單的接收方法 const int iRcvLen = 511; char szRcvBuf[iRcvLen + 1]; // 0結束符 do { iRet = recv(sktConn, szRcvBuf, iRcvLen, 0); szRcvBuf[iRet] = 0; if (0 < iRet) { printf("%d bytes received: %s./r/n", iRet, szRcvBuf); } else if (0 == iRet) { printf("All data received./r/n"); } else { printf("call recv failed, errno = %d./r/n", WSAGetLastError()); } } while (0 < iRet); error_handle4: // 確定已經收完數據後在關閉套接字(優雅關閉(Grace closed)) // while(0 < recv(sktConn, szRcvBuf, iRcvLen, 0)); closesocket(sktConn); sktConn = INVALID_SOCKET; error_handle3: freeaddrinfo(paiResult); // 釋放連接地址 paiResult = NULL; error_handle2: WSACleanup(); // 清理ws2_32.dll相關資源 error_handle1: system("pause"); return iRet; }