socket 请求接收完整的一个http响应(设置recv 接收超时选项SO_RCVTIMEO)

#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.baidu.com";
	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.baidu.com\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;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值