windows 下多socket连接（二）

在window下使用select实现多socket的连接修订版本

// WeChat.cpp : Defines the entry point for the console application.

//


#include "stdafx.h"
#include <stdio.h>
#include <string>
#include <vector>
#include <winsock2.h>
#pragma comment(lib, "ws2_32.lib")


#include "Util.h"


#define MAX_NUM 10
#define DATA_NUM 1024


using namespace std;


int sock_fd;
int client_fd[MAX_NUM];
struct fd_set read_set;
struct fd_set write_set;
struct fd_set select_read_set;
int max_fd;
int new_conn_fd;
struct timeval timeout = {0, 5000};
char recv_buf[DATA_NUM] = {0};




static DWORD WINAPI ThreadAccept(void *p);


void main(void )
{
sock_fd = -1;
max_fd = -1;
new_conn_fd = -1;
FD_ZERO(&read_set);
FD_ZERO(&write_set);
FD_ZERO(&select_read_set);


for (int i = 0; i < MAX_NUM; i++)
{
client_fd[i] = -1;
}



WORD wVersionRequested;
WSAData lpWSAData;
wVersionRequested = MAKEWORD(1, 1);


int err = WSAStartup(wVersionRequested, &lpWSAData);
if (!err)
{
printf("WSAStartup Succeed!\n");
}
else
{
printf("WSAStartup Fail!\n");
return ;
}


sock_fd = socket(AF_INET, SOCK_STREAM, 0);


SOCKADDR_IN serAdd;
serAdd.sin_family = AF_INET;
serAdd.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
serAdd.sin_port = htons(8000);
bind(sock_fd, (SOCKADDR *)&serAdd, sizeof(SOCKADDR));


listen(sock_fd, SOMAXCONN);


printf("server is running...\n");


HANDLE threadMsg = CreateThread(NULL, 0, ThreadAccept, NULL, 0, NULL);
if (threadMsg != NULL && threadMsg != INVALID_HANDLE_VALUE)
{
CloseHandle(threadMsg);
threadMsg = NULL;
}


while(1)
{
Sleep(1000);
}


WSACleanup();
}


DWORD WINAPI ThreadAccept(void *p)
{
printf("ThreadAccept start\n");
FD_SET(sock_fd, &read_set);
while(true)
{
max_fd = sock_fd;


for (int i = 0; i < MAX_NUM; i++)
{
if (max_fd < client_fd[i])
{
max_fd = client_fd[i];
}
}


select_read_set = read_set;


int ret = select(max_fd + 1, &select_read_set, NULL, NULL, &timeout);
if (ret == 0)
{
//printf("timeout\n");
}
else if (ret < 0)
{
printf("select error\n");
}
else
{
if (FD_ISSET(sock_fd, &select_read_set))
{
printf("new client comes\n");

SOCKADDR_IN clientAdd;
int len = sizeof(SOCKADDR);


new_conn_fd = accept(sock_fd, (SOCKADDR *)&clientAdd, &len);
if (new_conn_fd < 0)
{
printf("Fail to accept\n");
exit(1);
}
else
{
for (int i = 0; i < MAX_NUM; i++)
{
if (client_fd[i] == -1)
{
client_fd[i] = new_conn_fd;
FD_SET(new_conn_fd, &read_set);
break;
}
if (max_fd < new_conn_fd)
{
max_fd = new_conn_fd;
}
}
}
}
else
{
for (int i = 0; i < MAX_NUM; i++)
{
if (client_fd[i] == -1)
{
continue;
}


memset(recv_buf, 0, sizeof(recv_buf));


if (FD_ISSET(client_fd[i], &select_read_set))
{
int num = ::recv(client_fd[i], recv_buf, DATA_NUM, 0);
if (num < 0)
{
printf("Client(%d) left\n", client_fd[i]);  
FD_CLR(client_fd[i], &read_set);  
closesocket(client_fd[i]);
client_fd[i] = -1;  
}
else if (num > 0)
{
recv_buf[num] = '\0';
printf("Recieve client(%d) data\n", client_fd[i]);  
printf("Data: %s\n\n", recv_buf); 
}
else if (num == 0)
{
printf("Client(%d) exit\n", client_fd[i]);  
FD_CLR(client_fd[i], &read_set);  
closesocket(client_fd[i]);  
client_fd[i] = -1;  
}
}
}
}
}
}
printf("ThreadReceSendMsg end\n");
return 1;

}

在window平台select的第一个参数是可以忽略的，代码可以更简洁点