linux c http下载带确认进度条

今天药忘吃喽~ 2021-07-29 00:07 300阅读 0赞

代码如下：

#include <sys/ioctl.h>
    #include <poll.h>
    #include <sys/time.h>
    #include <sys/socket.h>
    #include <ctype.h>
    #include <signal.h>
    #include <fcntl.h>
    #include <sys/stat.h>
    #include <netdb.h>
    #include <stdio.h>
    #include <stdlib.h>
    #include <arpa/inet.h>
    #include <unistd.h>
    #include <netinet/in.h>
    #include <sys/types.h>
    #include <strings.h>
    #include <string.h>
    #include <errno.h>
     
    typedef struct HTTP_INFO{
    	unsigned long file_size;
    	char content_type[256];
    	char file_name[256];
    	char ip_addr[INET_ADDRSTRLEN];
    	char aliases[256];
    	int status_code;
    }HTTP_INFO;
     
    typedef struct HOST_INFO{
    	char host_name[256];
    	char file_path[1024];
    	unsigned short port;
    	char new_name[256];
    }HOST_INFO;
     
    HOST_INFO host_info;
    HTTP_INFO http_info;
     
    void print_error(const char *msg)
    {
    	perror(msg);
    	exit(1);
    }
     
    //获取服务端地址和别名
    int get_http_info(const char *host_name,HTTP_INFO *http_info)
    {
    	//判断是否为IP地址
    	int a,b,c,d;
    	if (sscanf(host_name,"%d.%d.%d.%d",&a,&b,&c,&d) == 4){
    		if (a >= 0x0 && a <= 0xff && b >= 0x0 && b <= 0xff && c >=0 && c <= 0xff && d >=0 && d <= 0xff){
    			strcpy(http_info->ip_addr,host_name);
    			strcpy(http_info->aliases,"no aliases");
    			return 0;
    		}
    	}
    	struct hostent *phost = gethostbyname(host_name);
    	if (!phost)
    		return -1;
    	/*char **ptr = NULL,temp[INET_ADDRSTRLEN];
    	for (ptr = phost->h_addr_list;*ptr != NULL;ptr++)
    	{
    		bzero(temp,sizeof(temp));
    		printf("对方服务器地址:%s\n",inet_ntop(AF_INET,*ptr,temp,sizeof(temp)));
    	}
    	for (ptr = phost->h_aliases;*ptr != NULL;ptr++)
    	{
    		printf("域名别名:%s\n",*ptr);
    	}*/
    	inet_ntop(AF_INET,phost->h_addr_list[0],http_info->ip_addr,sizeof(http_info->ip_addr));
    	if (*phost->h_aliases != NULL)
    		strcpy(http_info->aliases,phost->h_aliases[0]);
    	else
    		strcpy(http_info->aliases,"no aliases");
    	return 0;
    }
     
    //发送http报文请求
    int send_http_header(int sfd,HOST_INFO host_info)
    {
    	char http_header[BUFSIZ];
    	bzero(http_header,sizeof(http_header));
    	sprintf(http_header,\
    			"GET %s HTTP/1.1\r\n"\
    			"Host: %s\r\n"\
    			"Connection: Keep-Alive\r\n"\
    			"Content-Type: application/octet-stream\r\n"\
    			"\r\n",host_info.file_path,host_info.host_name);
    	return write(sfd,http_header,strlen(http_header));
    }
     
    //解析http报文头
    int parse_http_header(int sfd,HTTP_INFO *http_info)
    {
    	char buffer[BUFSIZ],temp[BUFSIZ],*ptr;
    	bzero(buffer,sizeof(buffer));
    	bzero(temp,sizeof(temp));
    	int len,n = 0;
    	while((len = read(sfd,buffer,1)) != 0){
    		temp[n] = *buffer;
    		if (*buffer == '\n'){
    			ptr = strstr(temp,"HTTP/1.1");
    			if (ptr != NULL){
    				ptr = strchr(ptr,' ');
    				ptr++;
    				http_info->status_code = atoi(ptr);
    			}
    			ptr = strstr(temp,"Content-Length:");
    			if (ptr != NULL){
    				ptr = strchr(ptr,':');
    				ptr++;
    				http_info->file_size = strtoul(ptr,NULL,10);
    			}
    			ptr = strstr(temp,"Content-Type:");
    			if (ptr != NULL){
    				ptr = strchr(ptr,':');	
    				ptr++;
    				strcpy(http_info->content_type,ptr);
    				http_info->content_type[strlen(ptr) - 1] = '\0';
    			}
    			//printf("%s",temp);
    			if (temp[0] == '\r' && temp[1] == '\n')
    				break;
    			bzero(temp,sizeof(temp));
    			n = -1;
    		}
    		n++;
    	}
    	return sfd;
    }
     
    int connect_server(const char *ip_addr,unsigned short port)
    {
    	int sfd = socket(AF_INET,SOCK_STREAM,0);
    	if (sfd == -1)
    		print_error("socket");
     
    	struct sockaddr_in serv_addr;
    	serv_addr.sin_family = AF_INET;
     
    	//默认80端口
    	serv_addr.sin_port = htons(port);
     
    	//根据域名获取到ip地址
    	int ret = inet_pton(AF_INET,ip_addr,&serv_addr.sin_addr.s_addr);
    	if (ret == -1)
    		print_error("inet_pton");
    	
    	int flags = fcntl(sfd,F_GETFL,0);
     
    	flags |= O_NONBLOCK;
    	fcntl(sfd,F_SETFL,flags);
     
    	//连接服务器
    	ret = connect(sfd,(struct sockaddr *)&serv_addr,sizeof(serv_addr));
    	if (ret != 0 && errno == EINPROGRESS){
    		printf("connecting server,please wait for 10 seconds\n");
    		struct pollfd pfd;
    		socklen_t optlen;
    		int optval,timeout = 10 *1000;
    		pfd.fd = sfd;
    		pfd.events = POLLOUT;
    		
    		if ((ret = poll(&pfd,1,timeout)) == 1){
    			//建立连接成功
    			optlen = sizeof(optval);
    			ret = getsockopt(sfd,SOL_SOCKET,SO_ERROR,&optval,&optlen);
    			errno = optval;
    			ret = optval == 0 ? 0 : -1;
    			if (!ret)
    				printf("connection %s\n",strerror(errno));
    		}
    		else if (!ret){
    			//连接超时
    			errno = ETIMEDOUT;
    			ret = -1;
    		}
    		else
    			print_error("poll failed");
    	}
    	flags &= ~O_NONBLOCK;
    	fcntl(sfd,F_SETFL,flags);
    	return ret == -1 ? -1 : sfd;
    }
     
    //打印进度条
    void print_progress_bar(const char *file_name,float sum,float file_size)
    {
    	float percent = (sum / file_size) * 100;
    	char *sign = "#";
    	if ((int)percent != 0){
    		sign = (char *)malloc((int)percent + 1);
    		strncpy(sign,"####################################################",(int) percent);
    	}
    	printf("%s %7.2f%% [%-*.*s] %.2f/%.2f mb\r",file_name,percent,50,(int)percent / 2,sign,sum / 1024.0 / 1024.0,file_size / 1024.0 / 1024.0);
    	if ((int)percent != 0)
    		free(sign);
    	fflush(stdout);
    }
     
    unsigned long download(int sfd,HOST_INFO host_info,HTTP_INFO http_info)
    {
    	umask(0111);
    	int len;
    	unsigned long sum = 0;
    	char buffer[BUFSIZ] = { 0 };
    	int fd = open(host_info.new_name,O_CREAT | O_WRONLY | O_TRUNC,0777);
    	if (fd == -1)
    		print_error("open");
    	while((len = read(sfd,buffer,sizeof(buffer))) > 0){
    		write(fd,buffer,len);
    		sum+=len;
    		print_progress_bar(host_info.new_name,(float)sum,(float)http_info.file_size);
    		if (http_info.file_size == sum){
    			printf("\n");
    			break;
    		}
    	}
    	close(fd);
    	close(sfd);
    	return sum;	
    }
     
    //解析url
    void parse_http_url(char *url,HOST_INFO *host_info)
    {
    	char *protocol[] = {"http://","https://",NULL};
    	int i,len = 0;
     
    	//删除http/https协议头
    	for (i = 0;protocol[i] != NULL;i++)
    		if (!strncmp(url,protocol[i],strlen(protocol[i]))){
    			len = strlen(protocol[i]);
    			break;
    		}
     
    	
    	char *ptr = strchr(url + len,':');
    	if (ptr != NULL){
    		//提取域名
    		strncpy(host_info->host_name,url + len,strlen(url + len) - strlen(ptr));
    		//提取端口
    		sscanf(++ptr,"%5hu",&host_info->port);
    		//判断端口是否为有效端口
    		if (host_info->port > 65535){
    			printf("invalid port\n");
    			exit(1);
    		}
    		//提取文件路径
    		while(*ptr != '/')
    			ptr++;
    		strcpy(host_info->file_path,ptr);
    		//提取文件名
    		ptr = strrchr(host_info->file_path,'/');
    		ptr++;
    		strcpy(host_info->new_name,ptr);
    	}
    	else{
    		ptr = strchr(url + len,'/');
    		//提取域名
    		strncpy(host_info->host_name,url + len,strlen(url + len) - strlen(ptr));
    		//设置为默认端口
    		host_info->port = 80;
    		//提取文件路径
    		strcpy(host_info->file_path,ptr);
    		//提取文件名
    		ptr = strrchr(host_info->file_path,'/');
    		ptr++;
    		strcpy(host_info->new_name,ptr);
    	}
    }
     
    //获取下载的文件大小
    unsigned long get_file_size(const char *file_name)
    {
    	struct stat buf;
    	if (stat(file_name,&buf) == -1)
    		return 0;
    	else
    		return buf.st_size;
    }
     
    //get请求
    int http_get(char *url,char *new_name)
    {
    	bzero(&host_info,sizeof(host_info));
    	bzero(&http_info,sizeof(http_info));
     
    	//解析url
    	parse_http_url(url,&host_info);
    	
    	if (new_name != NULL)
    		strcpy(host_info.new_name,new_name);
     
    	//根据域名获取IP地址
    	if (get_http_info(host_info.host_name,&http_info) == -1)
    		print_error("gethostbyname");
    	
    	//连接服务器
    	int sfd = connect_server(http_info.ip_addr,host_info.port);
    	if (sfd == -1)
    		print_error("connect failed");
    	printf("waiting for http response\n");
    	//发送http报文
    	int	ret = send_http_header(sfd,host_info);	
    	if (ret == -1)
    		print_error("write");
     
    	//解析http报头
    	sfd = parse_http_header(sfd,&http_info);
    	
    	printf("http response:\n\tstatus code: %d\n",http_info.status_code);
    	printf("\thost: %s:%hu\n",http_info.ip_addr,host_info.port);
    	printf("\taliases: %s\n",http_info.aliases);
    	printf("\tcontent-type:%s\n",http_info.content_type);
    	if (http_info.file_size > 1024*1024)	
    		printf("\tcontent-length: %.2f mb\n",(float)http_info.file_size / 1024.0 / 1024.0);
    	else
    		printf("\tcontent-length: %lu bytes\n",http_info.file_size);
     
    	if (http_info.status_code != 200)
    		printf("warning:not found file %s\n",host_info.new_name);
    	
    	printf("are you sure to download this file?\n");
    	printf("please enter any key to continue\n");
    	getchar();
    	
    	//下载文件
    //	printf("%s\n",host_info.new_name);
    	unsigned long download_size = download(sfd,host_info,http_info);
    	unsigned long file_size = get_file_size(host_info.new_name);
    //	printf("dowload_size %d\n",download_size);
    //	printf("file_size %lu\n",file_size);
    	if (download_size != file_size){
    		printf("download %s failure\n",host_info.new_name);
    		remove(host_info.new_name);
    	}
    	return 0;
    }
     
    //如果按下了ctrl+c
    void sig_handler(int num)
    {
    	if (num == SIGINT){
    		printf("\ndownload %s abort\n",host_info.new_name);
    		if (!access(host_info.new_name,F_OK))
    			remove(host_info.new_name);
    		putc('\n',stdout);
    		exit(0);
    	}
    }
     
     
    int main(int argc,char *argv[])
    {
    	if (argc < 2){
    		printf("Usage:\n");
    		printf("./download url\n");
    		printf("./download url \"new file name\"\n");
    		return 0;
    	}
    	//捕捉ctrl+c
    	signal(SIGINT,sig_handler);
    	http_get(argv[1],argv[2]);
    	return 0;
    }