winsock select 学习代码(1)

// SelectCli.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include <winsock2.h>
#include <conio.h>

#pragma comment(lib,"ws2_32")

#define SOCKET_MAXCNT	64
#define THREAD_NUM		2		//	实验代码  与服务器配合 不可改变连接线程数
#define CONNECT_PORT	8773	
#define CONNECT_ADDR	"127.0.0.1"


DWORD WINAPI SocketFunc(LPVOID pM)  
{  
	SOCKET	sock;
	char buf[100] = {0};
	int result;
	struct sockaddr_in cli_addr;
	int num = *(int*)pM;

	sock = socket(AF_INET,SOCK_STREAM,0);

	cli_addr.sin_family = AF_INET;
	cli_addr.sin_port = htons(CONNECT_PORT);
	cli_addr.sin_addr.s_addr = inet_addr(CONNECT_ADDR);


	result = connect(sock,(struct sockaddr*)&cli_addr,sizeof(cli_addr));
	if(result == SOCKET_ERROR)
	{
		printf("%d thread: connect error\n",num);
		exit(1);
	}

	while(1)
	{
	
		sprintf(buf,"%d thread: %s",num,"hello world!\n");
		printf(buf);
		result = send(sock,buf,100,0);
		if(result == SOCKET_ERROR)
		{
			printf("2 ,%s\n","send error");
		}
		Sleep(3000);
		
	}

	exit(1);
	return 0;
}  



int _tmain(int argc, _TCHAR* argv[])
{
	int num[SOCKET_MAXCNT] = {0};
	WSADATA	wsa_data;

	// 初始化数组 传递给线程  区别各个线程号
	for(int i = 0 ;i < SOCKET_MAXCNT;i++)
	{
		num[i] = i;
	}

	WSAStartup(WINSOCK_VERSION,&wsa_data);
	
	for(int i = 0; i < THREAD_NUM;i++)
	{
		HANDLE hThread1 = CreateThread(NULL,0,SocketFunc,&num[i],0,NULL);
	}


	while(1)
	{
		Sleep(100000);
	}
	return 0;
}

// SelectSrv.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include <winsock2.h>
#include <conio.h>

#pragma comment(lib,"ws2_32")

#define SOCKET_MAXCNT	64

#define CONNECT_PORT	8773	
#define CONNECT_ADDR	"127.0.0.1"


int _tmain(int argc, _TCHAR* argv[])
{
	WSADATA wsa_data;
	fd_set read_all;
	SOCKET	srv_listen = INVALID_SOCKET;
	struct sockaddr_in srv_addr;
	int result,ready_cnt;
	int addr_len = sizeof(srv_addr);
	SOCKET	recv_sock[2] = {INVALID_SOCKET,INVALID_SOCKET};

	WSAStartup(MAKEWORD(2,2),&wsa_data);

	srv_listen = socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);

	srv_addr.sin_family = AF_INET;
	srv_addr.sin_port = htons(CONNECT_PORT);
	srv_addr.sin_addr.s_addr = INADDR_ANY;

	result = bind(srv_listen,(struct sockaddr*)&srv_addr,addr_len);
	if(result == SOCKET_ERROR)
	{
		perror("err ");
		printf("line %d \n",__LINE__);
		return -1;
	}

	listen(srv_listen,SOMAXCONN);

	FD_ZERO(&read_all);
	FD_SET(srv_listen,&read_all);
	fd_set read_set;
	while(1)
	{
		read_set = read_all;
		ready_cnt = select(0,&read_set,NULL,NULL,NULL);
		if(ready_cnt == SOCKET_ERROR)
		{
			perror("err ");
			printf("line %d \n",__LINE__);
			return -1;
		}

		if(FD_ISSET(srv_listen,&read_set)) 
		{
			struct sockaddr_in faddr;
			int addr_len = sizeof(faddr);
			static int sock_num = 0;
			if(sock_num > 1)
			{
				perror("connect socket is too much\n ");
				printf("line %d \n",__LINE__);
				return -1;
			}
			recv_sock[sock_num] = accept(srv_listen,(struct sockaddr*)&faddr,&addr_len);
			if(recv_sock[sock_num] == INVALID_SOCKET)
			{
				perror("err ");
				printf("line %d \n",__LINE__);
				return -1;
			}
			FD_SET(recv_sock[sock_num],&read_all);
			sock_num++;
			continue;
		}

		for(int i = 0 ; i < 2;i++)
		{
			char buf[100] = {0};
			if(!FD_ISSET(recv_sock[i],&read_set))
			{
				continue;
			}
			result = recv(recv_sock[i],buf,100,NULL);
			if(result != SOCKET_ERROR)
			{
				printf("recv thread .%s\n",buf);
			}
		}
	}

	getch();
	return 0;
}