socket 请求接收完整的一个http响应（设置recv 接收超时选项SO_RCVTIMEO）
在前面的系列网络编程文章中，我们都是使用socket 自己实现客户端和服务器端来互相发数据测试，现在尝试使用socket 客户端发
送http 请求给某个网站，然后接收网站的响应数据。http 协议参考 这里。
代码如下：
C++ Code
|
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 99 100 101 |
#include<stdio.h>
#include<stdlib.h> #include<string.h> //strlen #include<sys/socket.h> #include<arpa/inet.h> //inet_addr #include<netdb.h> #include<errno.h> int main(int argc , char *argv[]) { int socket_desc; struct sockaddr_in server; char *message; //Create socket socket_desc = socket(AF_INET , SOCK_STREAM , 0); if (socket_desc == -1) { printf("Could not create socket"); } char ip[20] = {0}; char *hostname = "www.google.com.hk"; struct hostent *hp; if ((hp = gethostbyname(hostname)) == NULL) return 1; // #define h_addr h_addr_list[0] strcpy(ip, inet_ntoa(*(struct in_addr *)hp->h_addr_list[0])); server.sin_addr.s_addr = inet_addr(ip); server.sin_family = AF_INET; server.sin_port = htons( 80 ); //Connect to remote server if (connect(socket_desc , (struct sockaddr *)&server , sizeof(server)) < 0) { puts("connect error"); return 1; } puts("Connected\n"); //Send some data message = "GET /?st=1 HTTP/1.1\r\nHost: www.google.com.hk\r\n\r\n"; if( send(socket_desc , message , strlen(message) , 0) < 0) { puts("Send failed"); return 1; } puts("Data Send\n"); struct timeval timeout = {3, 0}; setsockopt(socket_desc, SOL_SOCKET, SO_RCVTIMEO, (char *)&timeout, sizeof(struct timeval)); //Receive a reply from the server //loop int size_recv , total_size = 0; char chunk[512]; while(1) { memset(chunk , 0 , 512); //clear the variable if((size_recv = recv(socket_desc , chunk , 512 , 0) ) == -1) { if (errno == EWOULDBLOCK || errno == EAGAIN) { printf("recv timeout ...\n"); break; } else if (errno == EINTR) { printf("interrupt by signal...\n"); continue; } else if (errno == ENOENT) { printf("recv RST segement...\n"); break; } else { printf("unknown error!\n"); exit(1); } } else if (size_recv == 0) { printf("peer closed ...\n"); break; } else { total_size += size_recv; printf("%s" , chunk); } } printf("Reply received, total_size = %d bytes\n", total_size); return 0; } |
输出如下：
.............................省略................................
从上面的输出可以看到有完整的<html> </html> ，即已经完整接收，但有一点不解的是为什么最后会接收到一个0？
Chunked transfer encoding uses a chunk size of 0 to mark the end of the content.
程序中 struct timeval timeout = {3,0};
setsockopt(socket_desc, SOL_SOCKET, SO_RCVTIMEO, (char *)&timeout, sizeof(struct timeval));
设置超时时间为3s，现在recv 为阻塞接收，如果超时时间内接收缓冲区没有一点数据，则返回-1 且errno = EWOULDBLOCK 。
退出循环，程序结束。
在这里顺便提一下，recv的第四个参数如果设置为MSG_WAITALL，在阻塞模式下不等到指定数目的数据是不会返回的，除非超时时间到或者被信号打断。但在这里我们并不知道对方会发来具体多少数据，所以不能使用这种方法来读取数据，否则可能出现一直阻塞的情况。
注：在阻塞发送时，也有人喜欢设置发送超时，超时判断返回值，如果没有发送完整则继续发送。但实际上本身阻塞发送会一直阻
塞到发送完整才返回，好像二者并无大的区别。
