线程TCP服务器

1. #include <stdlib.h>  
2. #include <stdio.h>  
3. #include <errno.h>  
4. #include <string.h>  
5. #include <netdb.h>  
6. #include <sys/types.h>  
7. #include <netinet/in.h>  
8. #include <sys/socket.h>  
9.   
10. #define portnumber 3333  
11.   
12. void * read_msg(void *arg)  
13. {  
14.     int fd = *((int *)arg);  
15.     int nread = 0;  
16.     char buffer[1024];  
17.   
18.     while((nread = read(fd,buffer,sizeof(buffer))) > 0)  
19.     {  
20.         buffer[nread] = '\0';  
21.         printf("get client message: %s\n",buffer);  
22.         memset(buffer,0,sizeof(buffer));  
23.     }  
24. }  
25.   
26. void * write_msg(void * arg)  
27. {  
28.      
29.     int fd = *((int *)arg);  
30.       
31.     while(1)  
32.     {  
33.         write(fd,"hello",5);  
34.         sleep(2);  
35.     }  
36. }  
37.   
38. int main(int argc, char *argv[])  
39. {  
40.     int sockfd,new_fd;  
41.     struct sockaddr_in server_addr;  
42.     struct sockaddr_in client_addr;  
43.     int sin_size;  
44.     int nbytes;  
45.     char buffer[1024];  
46.   
47.     pthread_t id;  
48.   
49.   
50.     /* 服务器端开始建立sockfd描述符 */  
51.     if((sockfd=socket(AF_INET,SOCK_STREAM,0))==-1) // AF_INET:IPV4;SOCK_STREAM:TCP  
52.     {  
53.         fprintf(stderr,"Socket error:%s\n\a",strerror(errno));  
54.         exit(1);  
55.     }  
56.   
57.     /* 服务器端填充 sockaddr结构 */  
58.     bzero(&server_addr,sizeof(struct sockaddr_in)); // 初始化,置0  
59.     server_addr.sin_family=AF_INET;                 // Internet  
60.     server_addr.sin_addr.s_addr=htonl(INADDR_ANY);  // (将本机器上的long数据转化为网络上的long数据)和任何主机通信  //INADDR_ANY 表示可以接收任意IP地址的数据，即绑定到所有的IP  
61.     //server_addr.sin_addr.s_addr=inet_addr("192.168.1.1");  //用于绑定到一个固定IP,inet_addr用于把数字加格式的ip转化为整形ip  
62.     server_addr.sin_port=htons(portnumber);         // (将本机器上的short数据转化为网络上的short数据)端口号  
63.   
64.     /* 捆绑sockfd描述符到IP地址 */  
65.     if(bind(sockfd,(struct sockaddr *)(&server_addr),sizeof(struct sockaddr))==-1)  
66.     {  
67.         fprintf(stderr,"Bind error:%s\n\a",strerror(errno));  
68.         exit(1);  
69.     }  
70.   
71.     /* 设置允许连接的最大客户端数 */  
72.     if(listen(sockfd,5)==-1)  
73.     {  
74.         fprintf(stderr,"Listen error:%s\n\a",strerror(errno));  
75.         exit(1);  
76.     }  
77.   
78.     while(1)  
79.     {  
80.         /* 服务器阻塞,直到客户程序建立连接 */  
81.         sin_size=sizeof(struct sockaddr_in);  
82.   
83.         printf("accepting!\n");  
84.         if((new_fd=accept(sockfd,(struct sockaddr *)(&client_addr),&sin_size))==-1)  
85.         {  
86.             fprintf(stderr,"Accept error:%s\n\a",strerror(errno));  
87.             exit(1);  
88.         }  
89.         fprintf(stderr,"Server get connection from %s\n",inet_ntoa(client_addr.sin_addr)); // 将网络地址转换成.字符串  
90.           
91.         pthread_create(&id,NULL,(void *)read_msg,(void *)&new_fd);  
92.         pthread_create(&id,NULL,(void *)write_msg,(void *)&new_fd);  
93.   
94.     }  
95.   
96.     /* 结束通讯 */  
97.     close(sockfd);  
98.     exit(0);  
99. }