TCP_IP—协议分析与应用编程_套接字编程.rar

/******************************************************************************* 程序名称：UDPExample.cpp 功 能：控制台命令状态运行的UDP通信程序 调试环境：VC++.NET 2003 作 者：Betta.LI 说 明：充当服务器进程时的运行方式：UDPExample 端口号 充当客户进程时的运行方式：UDPExample 端口号 服务器IP地址 服务器端口号 *******************************************************************************/ #include <winsock2.h> #include <iostream> using namespace std; #include <string.h> #pragma comment(lib, "ws2_32.lib") void main(int argc, char *argv[]) { const int DEFAULT_PORT = 5050; const int DATA_BUFFER_LENGHT = 1024; SOCKET locatesocket; SOCKADDR_IN locateAddr,remoteAddr;//本地和远程主机地址 int locatePort,remotePort; //本地和远程套接字端口号 int err; int iLen,iSend,iRecv;//远程端地址长度，发送数据长度，接收数据长度 char apply_Message[] = "I am a Client!\n"; //请求连接信息 char send_buf[DATA_BUFFER_LENGHT]="\0",recv_buf[DATA_BUFFER_LENGHT]="\0";//发送数据缓冲区，接收数据缓冲区 bool clientFlag = false ; //客户端标志，即请求方 WORD wVersionRequested; WSADATA wsaData; //根据参数设置相应的套接字参数 if (argc < 4) { cout<<"Server waiting"<<endl; if (argc < 2) locatePort = DEFAULT_PORT; else locatePort = atoi(argv[1]); } else { locatePort = atoi(argv[1]); remotePort = atoi(argv[3]); remoteAddr.sin_family = AF_INET; remoteAddr.sin_port = htons(remotePort); remoteAddr.sin_addr.S_un.S_addr = inet_addr(argv[2]); clientFlag = true; } //加载Winsocket动态库 wVersionRequested = MAKEWORD( 2, 2 ); err = WSAStartup( wVersionRequested, &wsaData ); if ( err != 0 ) { cout<<"Failed to load Winsock.\n"; return; } //创建本地套接字 locatesocket = socket(AF_INET,SOCK_DGRAM,0); if (locatesocket == INVALID_SOCKET) { cout<<"socket() Failed:"<<GetLastError()<<endl; return ; } //设置本地套接字相关参数 locateAddr.sin_addr.S_un.S_addr = htonl(INADDR_ANY); locateAddr.sin_family = AF_INET; locateAddr.sin_port = htons(locatePort); err = bind(locatesocket,(SOCKADDR*)&locateAddr,sizeof(SOCKADDR)); if (err != 0) { cout<<"bind() Failed:"<<GetLastError()<<endl; return ; } iLen = sizeof(remoteAddr); //客户进程首先发送数据 if (clientFlag) { iSend = sendto(locatesocket,apply_Message,strlen(apply_Message)+1,0,(SOCKADDR *)&remoteAddr,sizeof(remoteAddr)); if (iSend == SOCKET_ERROR) { cout<<"sendto() Failed:"<<GetLastError()<<endl; } else if (iSend > 0){ cout<<"Message:"<<send_buf<<" have sent to "<<inet_ntoa(remoteAddr.sin_addr)<<":"<<ntohs(remoteAddr.sin_port)<<endl; } } //循环处理双方数据传输 while (1) { iRecv = recvfrom(locatesocket,recv_buf,DATA_BUFFER_LENGHT,0,(SOCKADDR*)&remoteAddr,&iLen); if (iRecv == SOCKET_ERROR) { cout<<"recvfrom() Failed:"<<GetLastError()<<endl; break ; } else if (iRecv > 0) { cout<<"Receive:"<<recv_buf<<" From:"<<inet_ntoa(remoteAddr.sin_addr)<<":"<<ntohs(remoteAddr.sin_port)<<endl; } cin>>send_buf; iSend = sendto(locatesocket,send_buf,strlen(send_buf)+1,0,(SOCKADDR *)&remoteAddr,sizeof(remoteAddr)); if (iSend == SOCKET_ERROR) { cout<<"sendto() Failed:"<<GetLastError()<<endl; break ; } else if (iSend > 0){ cout<<"Message:"<<send_buf<<" have sent to "<<inet_ntoa(remoteAddr.sin_addr)<<":"<<ntohs(remoteAddr.sin_port)<<endl; } } //关闭套接字和清理系统资源 closesocket(locatesocket); WSACleanup(); }