Socket套接字編程（二）——UDP編程

TCP編程完了,下面輪到UDP了，這個是面向無連接的，更加簡單。

先熟悉下流程：

UDP型套接字：

服務器端（接收端）程序 ：
1、創建套接字（socket）。
2、將套接字綁定到一個本地地址和端口上（bind）。
3、等待接收數據（recvfrom）。
4、關閉套接字。

客戶端（發送端）程序 ：
1、創建套接字（socket）。
2、向服務器發送數據（sendto）。
3、關閉套接字。

下面繼續直接上代碼：

服務器端

#include <winsock2.h> #include <iostream> using namespace std; #pragma comment(lib, "WS2_32.lib") int StartUDPSrv() { #pragma region 初始化套接字庫 WSADATA wsaData; int iResult = WSAStartup(MAKEWORD(1, 1), &wsaData); //版本1.1 if (NO_ERROR != iResult) { cout<<"Error at WSAStartup()/n"; return -1; } if ( LOBYTE( wsaData.wVersion ) != 1 || HIBYTE( wsaData.wVersion ) != 1 ) { WSACleanup( ); return -1; }//如果不是1.1的則退出 #pragma endregion 初始化套接字庫 #pragma region 創建套接字 // Create a SOCKET for listening for incoming connection requests SOCKET socketSrv = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP); if (INVALID_SOCKET == socketSrv) { cout<<"Error at socket():/n"<< WSAGetLastError(); WSACleanup(); return -1; } #pragma endregion 創建套接字 #pragma region 綁定套接字 int err; SOCKADDR_IN addrSrv; addrSrv.sin_family = AF_INET; addrSrv.sin_addr.S_un.S_addr = htonl(INADDR_ANY); addrSrv.sin_port = htons(8086); err = bind(socketSrv, (SOCKADDR*) &addrSrv, sizeof(addrSrv)); if (SOCKET_ERROR == err) { cout<<"Error at Bind!/n"<< WSAGetLastError(); closesocket(socketSrv); WSACleanup(); return -1; } #pragma endregion 綁定套接字 #pragma region 等待接收 SOCKADDR_IN addrClient; int len = sizeof(sockaddr); char recvBuf[100]; while (1) { memset(recvBuf, '*', sizeof(char)*100); //置空接受數組 err = recvfrom(socketSrv, recvBuf, strlen(recvBuf), 0, (SOCKADDR*) &addrClient, &len); int len2 = strlen(recvBuf); if (SOCKET_ERROR == err) { cout<<"Error at RecvFrom Srv!/n"<< WSAGetLastError(); /*closesocket(socketSrv); WSACleanup(); return -1;*/ } cout<<recvBuf<<endl; /************************************************************************/ /* 發送數據 */ /************************************************************************/ char sendBuf[100]; sprintf_s(sendBuf, 100, "Welcome %s to This UDP Service!", inet_ntoa(addrClient.sin_addr)); err = sendto(socketSrv, sendBuf, strlen(sendBuf)+1, 0, (SOCKADDR*) &addrClient, sizeof(SOCKADDR)); if (SOCKET_ERROR == err) { cout<<"Error at SrvSendto!/n"<< WSAGetLastError(); closesocket(socketSrv); WSACleanup(); return -1; } } #pragma endregion 等待接收 closesocket(socketSrv); WSACleanup(); return 1; } int main(void) { StartUDPSrv(); //system("pause"); return 0; }

客戶端代碼：

 

#include <winsock2.h> #include <iostream> using namespace std; #pragma comment(lib, "WS2_32.lib") int CallUDPClient() { #pragma region 初始化套接字庫 WSADATA wsaData; int iResult = WSAStartup(MAKEWORD(1, 1), &wsaData); //版本1.1 if (NO_ERROR != iResult) { cout<<"Error at WSAStartup()/n"; return -1; } if ( LOBYTE( wsaData.wVersion ) != 1 || HIBYTE( wsaData.wVersion ) != 1 ) { WSACleanup( ); return -1; }//如果不是1.1的則退出 #pragma endregion 初始化套接字庫 #pragma region 創建套接字 // Create a SOCKET for listening for incoming connection requests SOCKET socketClient = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP); if (INVALID_SOCKET == socketClient) { cout<<"Error at socket():/n"<< WSAGetLastError(); WSACleanup(); return -1; } #pragma endregion 創建套接字 #pragma region 發送數據 SOCKADDR_IN addrSrv; addrSrv.sin_family = AF_INET; addrSrv.sin_addr.S_un.S_addr = inet_addr("127.0.0.1"); addrSrv.sin_port = htons(8086); int len = sizeof(SOCKADDR); //char sendBuf[100] = "Hello! This is UDP Client!"; char sendBuf[100]; gets_s(sendBuf); int err = sendto(socketClient, sendBuf, strlen(sendBuf)+1, 0, (SOCKADDR*) &addrSrv, len); if (SOCKET_ERROR == err) { cout<<"Error at Sendto Client:/n"<< WSAGetLastError(); closesocket(socketClient); WSACleanup(); return -1; } #pragma endregion 發送數據 #pragma region 接受數據 char recvBuf[100]; err = recvfrom(socketClient, recvBuf, strlen(recvBuf), 0, (SOCKADDR*)&addrSrv, &len); if (SOCKET_ERROR == err) { cout<<"Error at RecvFrom Client:/n"<< WSAGetLastError(); closesocket(socketClient); WSACleanup(); return -1; } cout<<recvBuf<<endl; #pragma endregion 接受數據 closesocket(socketClient); WSACleanup(); return 1; } int main() { CallUDPClient(); system("pause"); return 0; }

 

測試下，代碼沒多大問題。

 

但現在我們的代碼存在和TCP的時候同樣的一個問題，我們設置的服務器端接受數據的大小爲1024；但如果大於這個怎麼辦呢？嘿嘿，下回分解。