Leetcode: 06/01

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

<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;

}

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

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

}