UNIX网络编程(十七)-使用select 函数的TCP&UDP 回射服务器程

无双

原文出处http://www.xinxi888.com/Library/www/UNIX/UNIX17.HTM
感谢原作者的贡献

使用select 函数的TCP&UDP 回射服务器程序是并发TCP回射服务器程序与迭代UDP回射

服务器程序使用select复用组合为TCP和UDP套接口的单个服务器程序.

TCP并发服务器

1. 
   
2. // TCP并发服务器
   
3. int main(int argc, char **argv)
   
4. { int listenfd, connfd;
   
5. pid_t childpid;
   
6. socklen_t clilen;
   
7. struct sockaddr_in cliaddr, servaddr;
   
8. void sig-chld(int);
   
9. 
   
10. listenfd = Socket(AF_INET, SOCK_STREAM, 0);
    
11. 
    
12. bzero(&servaddr, sizeof(servaddr));
    
13. servaddr.sin_family = AF_INET;
    
14. servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    
15. servaddr.sin_port = htons(SERV_PORT);
    
16. 
    
17. Bind(listenfd, (SA *) &servaddr, sizeof(servaddr));
    
18. 
    
19. Listen(listenfd, LISTENQ);
    
20. Signal(SIGCHLD, sig_chld); /* must call waitpid() */
    
21. 
    
22. for ( ; ; ) {
    
23. clilen = sizeof(cliaddr);
    
24. if ( (connfd = accept(listenfd, (SA *) &cliaddr, &clilen)) < 0) {
    
25. if (errno == EINTR)
    
26. continue; /* back to for() */
    
27. else
    
28. err_sys("accept error");
    
29. }
    
30. if ( (childpid = Fork()) == 0) { /* child process */
    
31. Close(listenfd); /* close listening socket */
    
32. str_echo(connfd); /* process the request */
    
33. exit(0);
    
34. }
    
35. Close(connfd); /* parent closes connected socket */
    
36. }
    
37. }
    

复制代码

用select函数的TCP&UDP回射服务器程序

1. 
   
2. int main(int argc, char **argv)
   
3. { int listenfd, connfd, udpfd, nready, maxfdp1;
   
4. char mesg[MAXLINE]; pid_t childpid;
   
5. fd_set rset;
   
6. ssize_t n;
   
7. socklen_t len;
   
8. const int on = 1;
   
9. struct sockaddr_in cliaddr, servaddr;
   
10. void sig_chld(int);
    
11. /*create listening TCP socket */
    
12. listenfd = Socket(AF_INET, SOCK_STREAM, 0);
    
13. bzero(&servaddr, sizeof(servaddr));
    
14. servaddr.sin_family = AF_INET;
    
15. servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    
16. servaddr.sin_port = htons(SERV_PORT);
    
17. Setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));
    
18. Bind(listenfd, (SA *) &servaddr, sizeof(servaddr));
    
19. Listen(listenfd, LISTENQ);
    
20. /*create UDP socket */
    
21. udpfd = Socket(AF_INET, SOCK_DGRAM, 0);
    
22. bzero(&servaddr, sizeof(servaddr));
    
23. servaddr.sin_family = AF_INET;
    
24. servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    
25. servaddr.sin_port = htons(SERV_PORT);
    
26. Bind(udpfd, (SA *) &servaddr, sizeof(servaddr));
    
27. Signal(SIGCHLD, sig_chld); /* must call waitpid() */
    
28. FD_ZERO(&rset);
    
29. maxfdp1 = max(listenfd, udpfd) + 1
    
30. for ( ; ; ) {
    
31. FD_SET(listenfd, &rset);
    
32. FD_SET(udpfd, &rset);
    
33. if ( (nready = select(maxfdp1, &rset, NULL, NULL, NULL)) < 0) {
    
34. if (errno == EINTR) continue; /* back to for() */
    
35. else err_sys("select error");
    
36. }
    
37. if (FD_ISSET(listenfd, &rset)) {
    
38. len = sizeof(cliaddr);
    
39. connfd = Accept(listenfd, (SA *) &cliaddr, &len);
    
40. if ( (childpid = Fork()) == 0) { /* child process */
    
41. Close(listenfd); /* close listening socket */
    
42. str_echo(connfd); /* process the request */
    
43. exit(0);
    
44. }
    
45. Close(connfd); /* parent closes connected socket */
    
46. }
    
47. if (FD_ISSET(udpfd, &rset)) {
    
48. len = sizeof(cliaddr);
    
49. n = Recvfrom(udpfd, mesg, MAXLINE, 0, (SA *) &cliaddr, &len);
    
50. Sendto(udpfd, mesg, n, 0, (SA *) &cliaddr, len);
    
51. }
    
52. }
    
53. }
    
54. 
    

复制代码

程序流程

定义变量

int listenfd ： TCP监听套接口
int connfd: TCP已连接套接口,
int udpfd：UDP 套接口,
int nready：select函数中的准备好的描述字个数
int maxfdp1：select函数中的最大的描述字个数
char mesg[MAXLINE]：回射的字符串
pid_t childpid: TCP已连接套接口的进程编号
fd_set rset：select函数中的监听读描述字集
ssize_t n：回射的字符串的长度
socklen_t len：套接口地址长度
const int on = 1; 基本套接口选项SO_REUSEADDR的参数
SA cliaddr, servaddr ：服务器客户套接口地址结构
void sig_chld(int);信号处理函数

创建TCP监听套接口

listenfd = Socket(AF_INET, SOCK_STREAM, 0); //建立TCP套接口描述字
bzero(&servaddr, sizeof(servaddr)); //地址结构清零
servaddr.sin_family = AF_INET; //IPv4协议
servaddr.sin_addr.s_addr = htonl(INADDR_ANY); //内核指定地址
servaddr.sin_port = htons(SERV_PORT); //总所诸知端口#9877
// 允许启动一个监听服务器并捆绑其众所周知端口，即使以前建立的将此端口用作它们的本地端口的连接仍存在。
Setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));
Bind(listenfd, (SA *) &servaddr, sizeof(servaddr)); //绑定本机地址和端口
Listen(listenfd, LISTENQ); //监听

创建UPD套接口

udpfd = Socket(AF_INET, SOCK_DGRAM, 0); //建立UDP套接口描述字
bzero(&servaddr, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
servaddr.sin_port = htons(SERV_PORT);
Bind(udpfd, (SA *) &servaddr, sizeof(servaddr)); //绑定本机地址和端口