《C++编程艺术》第五章 下载工具源码

// Header file for downloader.  Call this file dl.h. 
#include <iostream> 
#include <string> 
#include <windows.h> 
#include <wininet.h> 
#include <fstream> 

using namespace std; 

const int MAX_ERRMSG_SIZE = 80; 

// 用于下载工具抛出异常的类.  
class DLExc {  
    char err[MAX_ERRMSG_SIZE];  
public:  
    DLExc(char *exc) {  
        if(strlen(exc) < MAX_ERRMSG_SIZE)  
            strcpy_s(err, exc);  
    }  

    // Return a pointer to the error message.  
    const char * geterr() {  
        return err;  
    }  
};  

// Download 类处理文件的下载，这个只包含静态函数
// 这个类只支持http1.1以及更高版本的URL下载，因为
// 它需要ranger报头，提供断点下载功能

class Download {  
    static bool ishttp(char *url);  
    static bool httpverOK(HINTERNET hIurl);  
    static bool getfname(char *url, char *fname);  
    static unsigned long openfile(char *url, bool reload,  
        ofstream &fout);  
public:  
    static bool dowld(char *url, bool restart=false,  
        void (*update)(unsigned long, unsigned long)=NULL);  
};  

// A file dowld subsystem.  
#include "afx.h"
#include <windows.h>
#include <wininet.h> 
#include <iostream>
#include <fstream>
#include "dl.h"

#pragma comment(lib, "Wininet.lib")

using namespace std;

#ifndef MAX_ERRMSG_SIZE
#define MAX_ERRMSG_SIZE 80
#endif // !1

#ifndef MAX_FILENAME_SIZE
#define MAX_FILENAME_SIZE 512
#endif

//#ifndef BUFFERSIZE
//#define BUFFERSIZE 1024;  
//#endif
    //const int MAX_FILENAME_SIZE = 512;
    const int BUFFERSIZE = 1024; 
    //const int MAX_ERRMSG_SIZE = 80;
// 用于下载工具抛出异常的类.  
class DowdLoadExc {  
    char err[MAX_ERRMSG_SIZE];  
public:  
    DowdLoadExc(char *exc) {  
        if(strlen(exc) < MAX_ERRMSG_SIZE)  
            strcpy_s(err, exc);  
    }  

    // Return a pointer to the error message.  
    const char * geterr() {  
        return err;  
    }  
};  

// dowld 类处理文件的下载，这个只包含静态函数
// 这个类只支持http1.1以及更高版本的URL下载，因为
// 它需要ranger报头，提供断点下载功能
//class Download {  
//    bool ishttp(char *url);  
//    bool httpverOK(HINTERNET hIurl);  
//    bool getfname(char *url, char *fname);  
//    unsigned long openfile(char *url, bool reload,  
//        ofstream &fout);  
//public:  
//    bool dowld(char *url, bool restart=false,  
//        void (*update)(unsigned long, unsigned long)=NULL);  
//};  

// 
// dowld函数传进一个URL，，如果文件已经在磁盘则继续下载
// 否则新建下载
// reload参数，指示是否再次下载某个已下载过的文件
// update函数指针作为了解下载过程的一种方法，被dowld函数
// 反复调用
bool Download::dowld(char *url, bool reload,   
                        void (*update)(unsigned long, unsigned long)) 
{  

    ofstream fout;                       // 输出流
    char buf[BUFFERSIZE];                 // 输入缓冲区  
    unsigned long numrcved;              // 每次循环已收到字节数 
    unsigned long filelen;               // 文件占用空间大小 
    HINTERNET hIurl = NULL , hInet = NULL;              // hUrl为URL句柄，hInet为连接Internet的句柄
    unsigned long contentlen;            // 内容的数量
    unsigned long len;                   // length of contentlen  
    unsigned long total = 0;             // 总已下载的字节数 
    char header[80];                     // 保存range报头 

    try {  
        //判断是否是http连接
        if(!ishttp(url))  
            throw DowdLoadExc("Must be HTTP url.");  

        // 通过指定的URL打开文件.
        // 如果文件不存在，或要重新下载，则opfile返回
        // 文件大小为0，否则返回已存在文件的大小 
        filelen = openfile(url, reload, fout);  

        // 文件打开后，建立起以Internet的连，这个函数只有一个参数
        // 且这个参数必须为0.  
        if(InternetAttemptConnect(0) != ERROR_SUCCESS)   
            throw DowdLoadExc("Can‘t connect.");  

        // 如果链接可用，就会调用InternetOpen打开这个链接
        // 第一个参数指定程序名，第二个指定使用的访问类型
        // 指定当前这个类型是，第三第四必须为NULL，最后一个指定任意选项  
        hInet = InternetOpen("downloader",  
            INTERNET_OPEN_TYPE_DIRECT,  
            NULL, NULL,  0);  

        if(hInet == NULL)   
            throw DowdLoadExc("Can‘t open connection.");  

        //一旦打开链接就会生成下面的range报头 
        sprintf_s(header, "Range:bytes=%d-", filelen);   

        // 将range报头传给下面的函数，打开指定的URL
        hIurl = InternetOpenUrl(hInet, url,  
            header, -1,  
            INTERNET_FLAG_NO_CACHE_WRITE, 0);  

        if(hIurl == NULL) throw DowdLoadExc("Can‘t open url.");  

        //  检查HTTP的版本是否是1.1或者是更高的版本.  
        if(!httpverOK(hIurl))  
            throw DowdLoadExc("HTTP/1.1 not supported.");  

        // 获取下载内容的数量
        // 内容的长度存储在contentlen，内容长度自动考虑range报头的
        // 请求范围，如果是整个下载，则从0到-（整个文件长度），断点续传
        // 的话，将返回中的一个节点，内容长度的为剩余长度
        // HTTP_QUERY_CONTENT_LENGTH用来获取被请求内容长度, HTTP_QUERY_FLAG_NUMBER
        // 要求这个值以整形值表示
        len = sizeof contentlen;  
        if(!HttpQueryInfo(hIurl,  
            HTTP_QUERY_CONTENT_LENGTH |  
            HTTP_QUERY_FLAG_NUMBER,  
            &contentlen, &len, NULL))  
            throw DowdLoadExc("File or content length not found.");  

        // 如果已存在文件未完成，则继续下载
        if(filelen != contentlen && contentlen)   
            do {  
                // 调用InternetReadFile来读取文件每次读取一个缓冲区就会循环一次
                // 每循环一次都会调用update指向的函数，除非update为NULL
                // 最后一个参数为实际读取成功的字节数
                // 如果成功这个函数返回true
                if(!InternetReadFile(hIurl, &buf,  
                    BUFFERSIZE, &numrcved))  
                    throw DowdLoadExc("Error occurred during dowld.");  

                // 将缓存写入到文件中  
                fout.write((const char *) buf, numrcved);  
                if(!fout.good())   
                    throw DowdLoadExc("Error writing file.");  

                total += numrcved;                                                             // 更新已下载的字节总数  

                // 如果已定义update函数，则调用
                if(update && numrcved > 0)  
                    update(contentlen+filelen, total+filelen);  

            } while(numrcved > 0);                                                          //当每次接收的字节大于0则继续，为0一般就是接收完成
        else  
            if(update)  
                update(filelen, filelen);  

    } catch(DowdLoadExc) {  
        fout.close();  
        InternetCloseHandle(hIurl);  
        InternetCloseHandle(hInet);  

        throw; // rethrow the exception for use by caller  
    }  

    fout.close();  
    InternetCloseHandle(hIurl);  
    InternetCloseHandle(hInet);  

    return true;  
}  

// 检查是否是http1.1版本 
bool Download::httpverOK(HINTERNET hIurl) {  
    char str[80];  
    unsigned long len = 79;  

    // 获取http版本，如果是1.1，str内容为HTTP/1.1.  
    if(!HttpQueryInfo(hIurl, HTTP_QUERY_VERSION, &str, &len, NULL))  
        return false;  

    // First, check major version number.  
    char *p = strchr(str, ‘/‘); 
    p++; 
    if(*p == ‘0‘) return false; // 不支持 HTTP 0.x 

    // Now, find start of minor HTTP version number.  
    p = strchr(str, ‘.‘);  
    p++;  

    // Convert to int.  
    int minorVerNum = atoi(p);  

    if(minorVerNum > 0) return true;  
    return false;  
}  

// 从URL中获取文件名. 
bool Download::getfname(char *url, char *fname) {  
    // Find last /.  
    char *p = strrchr(url, ‘/‘);  

    // Copy filename after the last /.   
    size_t len = strlen(p);
    if(p && ( len < MAX_FILENAME_SIZE)) {  
        //p++;  
        strcpy_s(fname,len,p + 1);  
        return true;  
    }  
    else  
        return false;  
}  

// 打开文件，初始化输出流，返回文件的大小
unsigned long Download::openfile(char *url,  
                                 bool reload,  
                                 ofstream &fout) 
{  
    char fname[MAX_FILENAME_SIZE];  

    if(!getfname(url, fname))   
        throw DowdLoadExc("File name error.");  

    if(!reload)   
        fout.open(fname, ios::binary | ios::out |  
        ios::app | ios::ate);    
    else  
        fout.open(fname, ios::binary | ios::out |  
        ios::trunc);    

    if(!fout)  
        throw DowdLoadExc("Can‘t open output file.");    

    // 返回当前文件的长度.  
    return fout.tellp();  
}  

// 检查这个URL是否是HTTP链接，通过前四个字母检查.  
bool Download::ishttp(char *url) {  
    char str[5] = "";  

    // Get first four characters from URL.  
    strncpy_s(str, url, 4);  

    // Convert to lowercase  
    for(char *p=str; *p; p++) *p = tolower(*p);  

    return !strcmp("http", str);  
}

#include <iostream> 
#include "dl.h" 

// This function displays the download progress as a percentage. 
void showprogress(unsigned long total, unsigned long part) { 
    int val = (int) ((double) part/total*100); 
    cout << val << "%" << endl; 
} 

int main(int argc, char *argv[]) 
{ 

    // This URL is for demonstration purposes only.  Substitute 
    // the URL of the file that you want to download. 
    char url[] = 
        "http://zip.77nt.com/18/%E5%9B%BD%E5%AE%B6%E6%84%8F%E5%BF%9777nt.com_txt18685.txt"; 

    bool reload = false; 

    if(argc==2 && !strcmp(argv[1], "reload")) 
        reload = true; 

    cout << "Beginning download.\n"; 

    try { 
        if(Download::dowld(url, reload, showprogress)) 
            cout << "Download Complete\n"; 
    } catch(DLExc exc) { 
        cout << exc.geterr() << endl; 
        cout << "Download Interrupted\n"; 
    }    

    system("pause");
    return 0; 
}