libcurl 编程进度条和range请求源码示例

在下面的源码中，我将展示如何使用libcurl提供的进度条功能，以及如何发送range请求, 同时提供了限速功能。

源码如下：

//g++ -g curl_range.cpp -o curl_range -lcurl -lm
//
#include <iostream>
#include <string>
#include <math.h>
#include <curl/curl.h>

using namespace std;

int progress_func(void* ptr, double TotalToDownload, double NowDownloaded, double TotalToUpload, double NowUpload)
{
    //how wide you want the progress bar to be ?
    int totalDot = 80;

    double fractionDownloaded = 0.0;
    if(TotalToDownload != 0)
        fractionDownloaded = NowDownloaded / TotalToDownload;//注意0不能为分母
    else
        fractionDownloaded = 0;
    //the full part of progress bar
    int dot = round(fractionDownloaded * totalDot);

    //create the progress bar, but control to print
    if(dot % 10 == 0){
        printf("total: %0.0f, now: %0.0f\n", TotalToDownload, NowDownloaded);
        int i = 0;
        printf("%3.0f%% [", fractionDownloaded * 100);
        for(; i < dot; i++)
            printf("="); // full part
        for(; i < totalDot; i++)
            printf(" "); // remainder part
        printf("]\n");
        fflush(stdout); //avoid output buffering problems
    }

    return 0;
}

int download (string url, string local_file, int down_speed)
{
CURL *curl;
CURLcode res;
FILE *fp;

curl = curl_easy_init ();
if (curl)
{
//Open File
fp = fopen (local_file.c_str (), "w");
if (fp == NULL)
cout << "File cannot be opened" << endl;

curl_easy_setopt (curl, CURLOPT_URL, url.c_str ());
curl_easy_setopt (curl, CURLOPT_FOLLOWLOCATION, 1);
//这里限速 XX KB/s
curl_easy_setopt (curl, CURLOPT_MAX_RECV_SPEED_LARGE, (curl_off_t) down_speed * 1024);
        curl_easy_setopt (curl, CURLOPT_RANGE, "0-100000000"); //设置range请求, 只下载前100MB
// curl_easy_setopt(curl, CURLOPT_VERBOSE, 1);
curl_easy_setopt (curl, CURLOPT_WRITEFUNCTION, NULL);
curl_easy_setopt (curl, CURLOPT_WRITEDATA, fp);
curl_easy_setopt (curl, CURLOPT_NOPROGRESS, 0);
        //禁用内部CURL进度显示条,假如我们提供了自定义的
curl_easy_setopt (curl, CURLOPT_PROGRESSFUNCTION, progress_func);

//限速下载
res = curl_easy_perform (curl);
if (res)
cout << "Cannot grab the File!\n";
}
//Clean up the resources
curl_easy_cleanup (curl);
//Close the file
fclose (fp);
return 0;
}

int main (int argc, char *argv[])
{
string url ("http://cdimage.ubuntu.com/releases/14.04/release/ubuntu-14.04-desktop-amd64+mac.iso");
string filepath ("./a.iso");
int downspeed = 600;

int ret = download (url, filepath, downspeed);
cout << "download [result]: " << ret << endl;

return 0;
}

下载过程中，和下载完成时的运行截图：

libcurl 编程进度条和range请求源码示例

几个知识点强调：

CURLcode curl_easy_setopt(CURL *handle, CURLoption option, parameter);
1. 设置下载数据的回调函数.
option:
CURLOPT_WRITEFUNCTION //设置回调函数,这个必须得有，否则libcurl的下载数据将直接显示在终端
回调函数原型为: size_t function( void *ptr, size_t size, size_t nmemb, void *userp); 必须返回数据长度，函数将在libcurl接收到数据后被调用。
void *ptr是下载回来的数据.
void *userp是用户指针, 用户通过这个指针传输自己的数据.
CURLOPT_WRITEDATA
　　设置回调函数中的void *userp指针的来源。
2. 下载进度控制.
option:
CURLOPT_NOPROGRESS
为了使CURLOPT_PROGRESSFUNCTION被调用. CURLOPT_NOPROGRESS必须被设置为false.如果没有下面的进度函数，将使用默认的进度条展示在终端
CURLOPT_PROGRESSFUNCTION
CURLOPT_PROGRESSFUNCTION 指定的函数正常情况下每秒被libcurl调用一次.
CURLOPT_PROGRESSDATA
CURLOPT_PROGRESSDATA指定的参数将作为CURLOPT_PROGRESSFUNCTION指定函数的参数.
整个处理与下载数据回调的处理相同.
3. 其它常用属性.
option:
CURLOPT_URL
设置访问的URI.
CURLOPT_NOSIGNAL
屏蔽其它信号.
CURLOPT_HEADER
取数据时连同HTTP头部一起取回.
CURLOPT_HEADERFUNCTION
CURLOPT_HEADERDATA
只取HTTP头部数据, 处理与下载数据回调的处理相同.
CURLOPT_TIMEOUT
超时时间.
CURLOPT_CONNECTIONTIMEOUT
连接等待时间.
CURLOPT_FOLLOWLOCATION
　　设置支持302重定向
　　 CURLOPT_RANGE
　断点续传, 指定传输分片, 格式:"0-200", 在上面源码的进度条函数中,我们可以看到设置range中,我们得到的TotalToDownload只是一个该分片的长度,而非整个文件的长度, 同时NowDownloaded会每秒刷新, 长度不断增大, 最后直至达到该分片的长度.

参考文献：

[1].http://blog.chinaunix.net/uid-20692625-id-3203258.html

秒客网

libcurl 编程进度条和range请求源码示例

相关文章