From: http://blog.csdn.net/jasonliuvip/article/details/22591531

最近在看《linux高性能服务器编程》,在此做个日记,以激励自己,同时分享于有需要的朋友。


1. 读取和设置socket文件描述符属性:

[cpp]  view plain copy 在CODE上查看代码片 派生到我的代码片
  1. #include <sys/socket.h>  
  2.   
  3. int getsockopt(int sockfd, int level, int option_name, void *option_value, socklen_t *restrict option_len);  
  4.   
  5. int setsockopt(int sockfd, int level, int option_name, const void *option_value, socklen_t option_len);  
sockfd参数指定被操作的目标socket。 

level参数指定要操作哪个协议的选项,即属性,比如:IPv4, IPv6, TCP 和通用socket选项。

option_name参数指定选项的名字。

option_value参数指定选项的值。

option_len参参数指定选项的长度。

调用成功时返回0, 失败时返回-1, 并设置errno。


2. 对服务器而言,有部分socket选项只能在调用listen前设置才会有效。因为连接socket只能由accept调用返回,而accept从listen监听队列中接受的连接至少已经完成了TCP三次握手的前两个步聚,listen监听队列的连接至少已进入SYN_RCVD状态,这时服务器已经往被连接上发送TCP同步报文。


3. SO_REUSeADDR选项:重用本地地址

未设置此项前,若服务端开启后,又关闭,此时sock处于TIME_WAIT状态,与之绑定的socket地址不可重用,而导致再次开启服务端失败。

经过setsockopt设置之后, 即使处于TIME_WAIT些状态也可以立即被重用。

[cpp]  view plain copy 在CODE上查看代码片 派生到我的代码片
  1. int reuse = 1;  
  2. setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &reuse, sizof(reuse));  


4. SO_RCVBUF 和 SO_SNDBUF :TCP接收缓冲区和发送缓冲区的大小

当然,即使我们设置了这两项的大小时, 系统都会自动将其加倍, 并且不得小于某个最小值。

TCP接收缓冲区的最小值是 256 字节, 而发送缓冲区的最小值是 2048 字节。(不同系统可能会有差异)

这么做的目的是确保一个TCP连接拥有足够多的空闲缓冲区来处理拥塞。

[java]  view plain copy 在CODE上查看代码片 派生到我的代码片
  1. setsockopt(sock, SOL_SOCKET, SO_SNDBUF, &sendbuf, sizeof(sendbuf));  
  2. getsockopt(sock, SOL_SOCKET, SO_SNDBUF, &sendbuf, (socklen_t*)&len);  
  3.   
  4. setsockopt(sock, SOL_SOCKET, SO_RCVBUF, &recvbuf, sizeof(recvbuf));  
  5. getsockopt(sock, SOL_SOCKET, SO_RCVBUF, &recvbuf, (socklen_t*)&len);  



5. 代码

[cpp]  view plain copy 在CODE上查看代码片 派生到我的代码片
  1. //服务端  
  2. #include <stdio.h>  
  3. #include <stdlib.h>  
  4. #include <unistd.h>  
  5. #include <string.h>  
  6. #include <errno.h>  
  7. #include <sys/socket.h>  
  8. #include <arpa/inet.h>  
  9. #include <assert.h>  
  10.   
  11. #define BUFFER_SIZE 1024  
  12.   
  13. int main(int argc, char **argv)  
  14. {  
  15.     if (argc <= 3) {  
  16.         printf("Usage: %s ip port revc_size\n", basename(argv[0]));  
  17.         return 1;  
  18.     }  
  19.       
  20.     const char *ip = argv[1];  
  21.     int port = atoi(argv[2]);  
  22.       
  23.     struct sockaddr_in address;  
  24.     bzero(&address, sizeof(address));  
  25.     address.sin_family = AF_INET;  
  26.     address.sin_port = htons(port);  
  27.     inet_pton(AF_INET, ip, &address.sin_addr);  
  28.       
  29.     int sock = socket(PF_INET, SOCK_STREAM, 0);  
  30.     assert(sock >= 0);  
  31.   
  32.         //设置地址可重用  
  33.     int reuse = 1;  
  34.     setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &reuse, sizof(reuse));  
  35.       
  36.     int recvbuf = atoi(argv[3]);  
  37.     int len = sizeof(recvbuf);  
  38.       
  39.         //设置接受缓冲区大小  
  40.     setsockopt(sock, SOL_SOCKET, SO_RCVBUF, &recvbuf, sizeof(recvbuf));  
  41.         //获取系统修改后的大小  
  42.         getsockopt(sock, SOL_SOCKET, SO_RCVBUF, &recvbuf, (socklen_t*)&len);  
  43.     printf("the receive buffer size after setting is %d\n", recvbuf);  
  44.       
  45.     int ret = bind(sock, (struct sockaddr*)&address, sizeof(address));  
  46.     assert(ret != -1);  
  47.       
  48.     ret = listen(sock, 5);  
  49.     assert(ret != -1);  
  50.       
  51.     struct sockaddr_in client;  
  52.     socklen_t client_addrlength = sizeof(client);  
  53.       
  54.     int connfd = accept(sock, (struct sockaddr*)&client, &client_addrlength);  
  55.     if (connfd < 0) {  
  56.         printf("errno is: %d\n", errno);  
  57.     }  
  58.     else {  
  59.         char buffer[BUFFER_SIZE];  
  60.         memset(buffer, '\0', BUFFER_SIZE);  
  61.           
  62.         while (recv(connfd, buffer, BUFFER_SIZE-1, 0) > 0);  
  63.           
  64.         printf("recv: %s\n", buffer);  
  65.         close(connfd);  
  66.     }  
  67.       
  68.     close(sock);  
  69.       
  70.     return 0;  
  71. }  

运行后:

jason@lightning:~/myproject/test_recv$ ./test_recv localhost 8000 50
the receive buffer size after setting is 2280

很明显被修改过了, 我们给的50, 被改为2280。



[cpp]  view plain copy 在CODE上查看代码片 派生到我的代码片
  1. //客户端  
  2. #include <stdio.h>  
  3. #include <sys/socket.h>  
  4. #include <arpa/inet.h>  
  5. #include <assert.h>  
  6. #include <unistd.h>  
  7. #include <stdlib.h>  
  8. #include <string.h>  
  9.   
  10. #define BUFFER_SIZE 1024  
  11.   
  12. int main(int argc, char **argv)  
  13. {  
  14.     if(argc <= 3) {  
  15.         fprintf(stderr, "Usage: %s ip port send_buffer_size\n",   
  16.                 basename(argv[0]));  
  17.         return 1;  
  18.     }  
  19.       
  20.     const char *ip = argv[1];  
  21.     int port = atoi(argv[2]);  
  22.       
  23.     struct sockaddr_in server_address;  
  24.     bzero(&server_address, sizeof(server_address));  
  25.     server_address.sin_family = AF_INET;  
  26.     server_address.sin_port = htons(port);  
  27.     inet_pton(AF_INET, ip, &server_address.sin_addr);  
  28.   
  29.     int sock = socket(PF_INET, SOCK_STREAM, 0);  
  30.     assert(sock >= 0);  
  31.       
  32.     int sendbuf = atoi(argv[3]);  
  33.     int len = sizeof(sendbuf);  
  34.       
  35.         //设置发送缓冲区大小  
  36.     setsockopt(sock, SOL_SOCKET, SO_SNDBUF, &sendbuf, sizeof(sendbuf));  
  37.         //获取系统修改后的大小  
  38.         getsockopt(sock, SOL_SOCKET, SO_SNDBUF, &sendbuf, (socklen_t*)&len);  
  39.       
  40.     printf("the tcp send buffer size after setting is %d\n",  
  41.             sendbuf);  
  42.       
  43.     if (connect(sock, (struct sockaddr*)&server_address, sizeof(server_address)) != -1) {  
  44.         char buffer[BUFFER_SIZE];  
  45.         memset(buffer, 'a', BUFFER_SIZE);  
  46.         send(sock, buffer, BUFFER_SIZE, 0);  
  47.     }  
  48.     else {  
  49.         printf("connect %s failed\n", ip);  
  50.     }  
  51.       
  52.     close(sock);  
  53.       
  54.     return 0;  
  55. }  

运行后:

jason@lightning:~/myproject/test_send$ ./test_send localhost 8000 2000
the tcp send buffer size after setting is 4000

给的是2000, 被改成4000了。


Logo

更多推荐