winsock select 学习代码(1)

<br>// 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;
}