基于UDP的select函数用法

心不留意外尘 2017-07-17

展开全文

http://blog.csdn.net/FeeLang/article/details/4983317

2009

[cpp] view plain copy

print?

/************SERVER**************/
#include <winsock2.h>
#include <stdio.h>
#define PORT_A 11111
#define PORT_B 22222
void main(int argc, char **argv)
{
WSADATA wsaData; // 套接口信息数据
SOCKET socka; // 套接口a
SOCKET sockb; // 套接口b
int nPortA = PORT_A;
int nPortB = PORT_B;
fd_set rfd; // 读描述符集
timeval timeout; // 定时变量
sockaddr_in addr; // 告诉sock 应该在什么地方licence
char recv_buf[1024]; // 接收缓冲区
int nRecLen; // 客户端地址长度!!!!!!
sockaddr_in cli; // 客户端地址
int nRet; // select返回值
if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
{
printf("failed to laod winsock!/n");
return;
}
socka = socket(AF_INET, SOCK_DGRAM, 0); // 创建数据报socka
if (socka == INVALID_SOCKET)
{
printf("socket()/n");
return;
}
sockb = socket(AF_INET, SOCK_DGRAM, 0); // 创建数据报sockb
if (sockb == INVALID_SOCKET)
{
printf("socket()/n");
return;
}
memset(&addr, 0, sizeof(addr));
addr.sin_family = AF_INET; // IP协议
addr.sin_port = htons(nPortA); // 端口
addr.sin_addr.s_addr = htonl(INADDR_ANY); // 在本机的所有ip上开始监听
if (bind(socka, (sockaddr*)&addr, sizeof(addr)) == SOCKET_ERROR)// bind socka
{
printf("bind()/n");
return;
}
memset(&addr, 0, sizeof(addr));
addr.sin_family = AF_INET; // IP协议
addr.sin_port = htons(nPortB); // 端口
addr.sin_addr.s_addr = htonl(INADDR_ANY); // 在本机的所有ip上开始监听
if (bind(sockb, (sockaddr*)&addr, sizeof(addr)) == SOCKET_ERROR) // bind sockb
{
printf("bind()/n");
return;
}
// 设置超时时间为6s
timeout.tv_sec = 6;
timeout.tv_usec = 0;
memset(recv_buf, 0, sizeof(recv_buf)); // 清空接收缓冲区
while (true)
{
FD_ZERO(&rfd); // 在使用之前总是要清空
// 开始使用select
FD_SET(socka, &rfd); // 把socka放入要测试的描述符集中
FD_SET(sockb, &rfd); // 把sockb放入要测试的描述符集中
nRet = select(0, &rfd, NULL, NULL, &timeout);// 检测是否有套接口是否可读
if (nRet == SOCKET_ERROR)
{
printf("select()/n");
return;
}
else if (nRet == 0) // 超时
{
printf("timeout/n");
closesocket(socka);
closesocket(sockb);
break;
}
else // 检测到有套接口可读
{
if (FD_ISSET(socka, &rfd)) // socka可读
{
nRecLen = sizeof(cli);
int nRecEcho = recvfrom(socka, recv_buf, sizeof(recv_buf), 0, (sockaddr*)&cli, &nRecLen);
if (nRecEcho == INVALID_SOCKET)
{
printf("recvfrom()/n");
break;
}
printf("data to port 11111: %s/n", recv_buf);
}
if (FD_ISSET(sockb, &rfd)) // sockb 可读
{
nRecLen = sizeof(cli);
int nRecEcho = recvfrom(sockb, recv_buf, sizeof(recv_buf), 0, (sockaddr*)&cli, &nRecLen);
if (nRecEcho == INVALID_SOCKET)
{
printf("recvfrom()/n");
break;
}
printf("data to port 22222: %s/n", recv_buf);
}
}
}
WSACleanup();
}

[cpp] view plain copy

print?

/************CLIENT*************/
#include <winsock2.h>
#include <stdio.h>
#include <stdlib.h>
#define SERVER_PORT_A 11111 // 服务器端口A
#define SERVER_PORT_B 22222 // 服务器端口B
typedef struct tagSERVER // 服务器
{
char* ip; // ip地址
int nPort; // 端口号
} SERVER, *PSERVER;
int SendData(SOCKET s, char *ip, int nPort, char *pData); // 发送数据到IP:nPort
int main(int argc, char **argv)
{
int i;
WSADATA wsaData; // socket数据
SOCKET sClient; // 客户端套接口
char send_buf[] = "hello! I am LiangFei whoes SNO=06060734"; // 发送的数据内容
int nSend; // 发送数据后的返回值
// 服务器
SERVER sers[] = { {"127.0.0.1", SERVER_PORT_A},
{"127.0.0.1", SERVER_PORT_B} };
int nSerCount = sizeof(sers) / sizeof(sers[0]); // 服务器个数
if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) // 启动socket
{
printf("failed to start up socket!/n");
return 0;
}
// 建立客户端数据包套接口
sClient = socket(AF_INET, SOCK_DGRAM, 0);
if (sClient == INVALID_SOCKET)
{
printf("socket()/n");
return 0;
}
for (i = 0; i < nSerCount; i++)
{
nSend = SendData(sClient, sers[i].ip, sers[i].nPort, send_buf); // 发送数据
if (nSend == 0) // 发送失败
{
return 0;
}
else if (nSend == SOCKET_ERROR) // 套接口出错
{
printf("sendto()/n");
return 0;
}
}
closesocket(sClient); // 关闭套接口
WSACleanup(); // 卸载winsock
return 0;
}
int SendData(SOCKET s, char *ip, int nPort, char *pData)
{
sockaddr_in ser; // 服务器端地址
ser.sin_family = AF_INET; // IP协议
ser.sin_port = htons(nPort); // 端口号
ser.sin_addr.s_addr = inet_addr(ip); // IP地址
int nLen = sizeof(ser); // 服务器地址长度
return sendto(s, pData, strlen(pData) + 1, 0, (sockaddr*)&ser, nLen); // 向服务器发送数据