多线程并发服务器与客户端 －－－20100514

//服务器

1 #include <stdio.h>

2 #include <sys/types.h>

3 #include <sys/socket.h>

4 #include <arpa/inet.h>

5 #include <unistd.h>

6 #include <stdlib.h>

7 #include <string.h>

8

9 #define SERV_IP "127.0.0.1" //本机IP

10 #define SERV_PORT 8000

11

12 struct ARG{

13 int connfd;

14 struct sockaddr_in client;

15 int other;

16 };

17

18 void *function(void *arg);

19 void sys_err(char* err)

20 {

21 perror(err);

22 exit(1);

23 }

24 int main(void)

25 {

26 int sock_fd;

27 int connfd;

28 int ret;

29 socklen_t addrlen;

30 struct ARG * arg;

31 pthread_t tid;

32

33 struct sockaddr_in cli_addr;

34 struct sockaddr_in serv_addr;

35

36 memset(&serv_addr, 0, sizeof(serv_addr));

37 serv_addr.sin_family = AF_INET;

38 serv_addr.sin_port = htons(SERV_PORT);

39 // serv_addr.sin_addr.s_addr = inet_addr(SERV_IP);

40 inet_pton(AF_INET, SERV_IP, &serv_addr.sin_addr.s_addr);

41

42 sock_fd = socket(AF_INET, SOCK_STREAM, 0);

43 if(sock_fd == -1)

44 sys_err("fail to socket");

45 int opt = SO_REUSEADDR;

46 setsockopt(sock_fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));

47 ret = bind(sock_fd, (struct sockaddr *)&serv_addr, sizeof(serv_addr));

48 if(ret == -1)

49 sys_err("fail to bind");

50

51 ret = listen(sock_fd, 20);

52 if(ret == -1)

53 sys_err("fail to listen");

54 //net_fd = accept(sock_fd, (struct sockaddr* )&cli_addr, &addrlen);

55 //if(net_fd == -1)

56 // sys_err("fail to accept");

57

58 char buf[1024];

59 int readsize;

60 while(1)

61 {

62 if((connfd = accept(sock_fd, (struct sockaddr*)&cli_addr, &addrlen)) == -1)

63 sys_err("fail to accept");

64 arg = (struct ARG *)malloc(sizeof(struct ARG));

65 arg->client = cli_addr;

66 arg->connfd = connfd;

67

68 if(pthread_create(&tid, NULL, function, (void *)arg))

69 sys_err("pthread create error");

70

71

72

73

74

75 }

76

77 close(sock_fd);

78

79 return 0;

80 }

81

82 void *function(void *arg)

83 {

84 int net_fd;

85 int num;

86 char buf[1024];

87 struct sockaddr_in client;

88

89 net_fd = ((struct ARG *)arg)->connfd;

90 client = ((struct ARG *)arg)->client;

91

92 while(1)

93 {

94 num = recv(net_fd, buf, 1024, 0);

95 if(num == 0)

96 {

97 printf("the client<%s> exit/n", inet_ntoa(client.sin_addr));

98 break;

99 }

100 printf("You got message from client:%s/n", inet_ntoa(client.sin_addr));

101

102 write(STDOUT_FILENO, buf, num);

103

104 }

105 free(arg);

106 close(net_fd);

107 pthread_exit(NULL);

108 }

//客户端

1 #include <stdio.h>

2 #include <unistd.h>

3 #include <arpa/inet.h>

4 #include <sys/socket.h>

5 #include <sys/types.h>

6 #include <stdlib.h>

7 #include <string.h>

8

9 #define SERV_IP "127.0.0.1"

10 #define SERV_PORT 8000

11

12 void sys_err(char* err)

13 {

14 perror(err);

15 exit(0);

16 }

17 int main(void)

18 {

19 int sockfd;

20 int ret;

21

22 struct sockaddr_in serv_addr;

23 memset(&serv_addr, 0, sizeof(serv_addr));

24 serv_addr.sin_family = AF_INET;

25 serv_addr.sin_port = htons(SERV_PORT);

26 serv_addr.sin_addr.s_addr = inet_addr(SERV_IP);

27

28 sockfd = socket(AF_INET, SOCK_STREAM, 0);

29 if(sockfd == -1)

30 sys_err("fail to socket");

31 ret = connect(sockfd, (struct sockaddr*)&serv_addr, sizeof(serv_addr));

32 if(ret == -1)

33 sys_err("fail to connect");

34 char buf[1024];

35 int readsize;

36

37 while(1)

38 {

39 readsize = read(STDIN_FILENO, buf, 1024);

40 if(strncmp(buf, "exit", 4) == 0)

41 {

42 printf("client has exited");

43 break;

44 }

45

46 write(sockfd, buf, readsize);

47 }

48 close(sockfd);

49

50 }