使用libcurl实现的上传器

头文件

/**********************************************************************

* Copyright (C) 2014 -  - All Rights Reserved

*

* 文件名称:        Uploader_LibCurl.h

* 摘    要:        上传器 - LibCurl实现

*     

* 作    者:        yanglinbo,

* 修    改:        查看文件最下方.

* 

***********************************************************************/

#ifndef __Uploader_LibCurl_H__

#define __Uploader_LibCurl_H__

#include <curl/curl.h>

#include <fstream>

#include <string>

class CUploader

{

public:

    CUploader(void);

    virtual ~CUploader(void);

    /// 线程入口函数

    virtual bool run();

    /// 启动上传

    virtual bool start(const std::string& strUrl, const std::string& strLocalFile);

    /// 停止上传

    virtual bool stop();

    /// 是否运行状态

    bool isRunning() const;

protected:

    /// 读取回调

    static size_t handleRead(void *buffer, size_t size, size_t nmemb, void *userp);

    /// 进度回调

    static size_t handleProgress(void *buffer, double dltotal, double dlnow, double ultotal, double ulnow);

    size_t onUpload(void *buffer, size_t size, size_t nmemb);

    size_t onProgress(const double& ultotal, const double& ulnow);

    /// 上传回调

    void onUpload();

    /// 设置libcurl选项

    bool setOption();

    /// 清除数据

    void clear();

    CURL*            m_pCurl;                ///< libcurl句柄

    FILE*            m_pFile;                ///< 文件指针

    bool            m_bRunning;                ///< 运行标志

    std::string        m_strRemoteUrl;            ///< 远程链接

    std::string        m_strLocalFilePath;        ///< 本地文件路径

    size_t            m_nLocalFileSize;        ///< 本地文件大小

};

#endif

实现文件

* 文件名称:        Uploader_LibCurl.cpp

#include "StdAfx.h"

#include "Uploader.h"

#include <sys/stat.h>

#include <fcntl.h>

CUploader::CUploader(void)

: m_pCurl(NULL)

, m_pFile(NULL)

, m_bRunning(false)

, m_nLocalFileSize(0)

}

CUploader::~CUploader(void)

    stop();

bool CUploader::run()

    onUpload();

    return true;

bool CUploader::isRunning() const

    return m_bRunning;

void CUploader::clear()

    if (m_pFile)

    {

        fclose(m_pFile);

        m_pFile = NULL;

    }

    if (m_pCurl)

        curl_easy_cleanup(m_pCurl);

        m_pCurl = NULL;

        curl_global_cleanup();

    m_strRemoteUrl.clear();

    m_strLocalFilePath.clear();

    m_nLocalFileSize = 0;

bool CUploader::setOption()

    // 远程URL，支持 http, https, ftp

    curl_easy_setopt(m_pCurl, CURLOPT_URL, m_strRemoteUrl.c_str());

    curl_easy_setopt(m_pCurl, CURLOPT_INFILESIZE_LARGE, (curl_off_t)m_nLocalFileSize);

    // 设置User-Agent

    std::string useragent = _T("Mozilla/5.0 (Windows NT 6.1; WOW64; rv:13.0) Gecko/20100101 Firefox/13.0.1");

    curl_easy_setopt(m_pCurl, CURLOPT_USERAGENT, useragent.c_str());

    // 设置重定向的最大次数

    curl_easy_setopt(m_pCurl, CURLOPT_MAXREDIRS, 5);

    // 设置301、302跳转跟随location

    curl_easy_setopt(m_pCurl, CURLOPT_FOLLOWLOCATION, 1);

    curl_easy_setopt(m_pCurl, CURLOPT_UPLOAD, TRUE);

    curl_easy_setopt(m_pCurl, CURLOPT_NOSIGNAL, 1L);

    curl_easy_setopt(m_pCurl, CURLOPT_POST, TRUE);

    curl_easy_setopt(m_pCurl, CURLOPT_FORBID_REUSE, TRUE);

    // 上传内容回调函数

    curl_easy_setopt(m_pCurl, CURLOPT_READFUNCTION, handleRead);

    curl_easy_setopt(m_pCurl, CURLOPT_READDATA, this);

    // 进度回调函数

    curl_easy_setopt(m_pCurl, CURLOPT_NOPROGRESS, 0);

    curl_easy_setopt(m_pCurl, CURLOPT_PROGRESSDATA, this);

    curl_easy_setopt(m_pCurl, CURLOPT_PROGRESSFUNCTION, handleProgress);

    // 跳过服务器SSL验证，不使用CA证书

    curl_easy_setopt(m_pCurl, CURLOPT_SSL_VERIFYPEER, 0L);

    // 验证服务器端发送的证书，默认是 2(高)，1（中），0（禁用）

    curl_easy_setopt(m_pCurl, CURLOPT_SSL_VERIFYHOST, 0L);

bool CUploader::start(const std::string& strUrl, const std::string& strLocalFile)

    if (strUrl.empty()) return false;

    if (m_bRunning == true) return true;

    clear();

    m_strRemoteUrl    = strUrl;

    m_strLocalFilePath    = strLocalFile;

    // 打开文件

    m_pFile = fopen(m_strLocalFilePath.c_str(), "rb");

    if (m_pFile == NULL)

        return false;

    // 获取文件大小

    struct stat file_info;

    if (fstat(fileno(m_pFile), &file_info) != 0) 

    m_nLocalFileSize = file_info.st_size;

    // 初始化libcurl

    m_pCurl = curl_easy_init();

    if (m_pCurl == NULL)

    // 设置libcurl的选项

    setOption();

    m_bRunning = true;

bool CUploader::stop()

    m_bRunning = false;

size_t CUploader::handleRead( void *buffer, size_t size, size_t nmemb, void *userp )

    CUploader* pDownloader = (CUploader*) userp;

    if (pDownloader)

        return pDownloader->onUpload(buffer, size, nmemb);

    return 0;

size_t CUploader::handleProgress( void *buffer, double dltotal, double dlnow, double ultotal, double ulnow )

    CUploader* pDownloader = (CUploader*) buffer;

        pDownloader->onProgress(ultotal, ulnow);

size_t CUploader::onProgress( const double& ultotal, const double& ulnow )

    TRACE("%.2f / %.2f (%.2g %%)\n", ulnow, ultotal, ulnow*100.0/ultotal);

size_t CUploader::onUpload( void *buffer, size_t size, size_t nmemb )

    size_t return_size = fread(buffer, size, nmemb, m_pFile);

    return return_size;

void CUploader::onUpload()

    // 执行上传

    CURLcode return_code = CURLE_OK;

    return_code = curl_easy_perform(m_pCurl);

    // 关闭文件

    // 上传失败

    if (return_code != CURLE_OK)

        return;

    // 获取状态码

    int response_code = 0;

    curl_easy_getinfo(m_pCurl, CURLINFO_RESPONSE_CODE, &response_code);

示例代码：

CUploader uploader

uploader.start("ftp://upload:[email protected]/hello.exe", "C:\\fly.exe");

uploader.run();