使用libcurl实现的上传器-阿里云开发者社区

开发者社区> 杨粼波> 正文

使用libcurl实现的上传器

简介:
+关注继续查看
头文件
ExpandedBlockStart.gif/**********************************************************************
* 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
ExpandedBlockStart.gif{
public:
    CUploader(void);
    virtual ~CUploader(void);

public:
ExpandedSubBlockStart.gif    /// 线程入口函数
    virtual bool run();

ExpandedSubBlockStart.gif    /// 启动上传
    virtual bool start(const std::string& strUrl, const std::string& strLocalFile);

ExpandedSubBlockStart.gif    /// 停止上传
    virtual bool stop();

ExpandedSubBlockStart.gif    /// 是否运行状态
    bool isRunning() const;

protected:
ExpandedSubBlockStart.gif    /// 读取回调
    static size_t handleRead(void *buffer, size_t size, size_t nmemb, void *userp);

ExpandedSubBlockStart.gif    /// 进度回调
    static size_t handleProgress(void *buffer, double dltotal, double dlnow, double ultotal, double ulnow);

protected:
ExpandedSubBlockStart.gif    /// 读取回调
    size_t onUpload(void *buffer, size_t size, size_t nmemb);

ExpandedSubBlockStart.gif    /// 进度回调
    size_t onProgress(const double& ultotal, const double& ulnow);

ExpandedSubBlockStart.gif    /// 上传回调
    void onUpload();

protected:
ExpandedSubBlockStart.gif    /// 设置libcurl选项
    bool setOption();

ExpandedSubBlockStart.gif    /// 清除数据
    void clear();

protected:
ExpandedSubBlockStart.gif    CURL*            m_pCurl;                ///< libcurl句柄
    FILE*            m_pFile;                ///< 文件指针

ExpandedSubBlockStart.gif    bool            m_bRunning;                ///< 运行标志

ExpandedSubBlockStart.gif    std::string        m_strRemoteUrl;            ///< 远程链接
    std::string        m_strLocalFilePath;        ///< 本地文件路径

ExpandedSubBlockStart.gif    size_t            m_nLocalFileSize;        ///< 本地文件大小
};



#endif

实现文件
ExpandedBlockStart.gif/**********************************************************************
* Copyright (C) 2014 -  - All Rights Reserved
*
* 文件名称:        Uploader_LibCurl.cpp
* 摘    要:        上传器 - LibCurl实现
*     
* 作    者:        yanglinbo,
* 修    改:        查看文件最下方.

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


#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)
ExpandedBlockStart.gif{
}


CUploader::~CUploader(void)
ExpandedBlockStart.gif{
    stop();
}


bool CUploader::run()
ExpandedBlockStart.gif{
    onUpload();
    return true;
}


bool CUploader::isRunning() const
ExpandedBlockStart.gif{
    return m_bRunning;
}


void CUploader::clear()
ExpandedBlockStart.gif{
    if (m_pFile)
ExpandedSubBlockStart.gif    {
        fclose(m_pFile);
        m_pFile = NULL;
    }


    if (m_pCurl)
ExpandedSubBlockStart.gif    {
        curl_easy_cleanup(m_pCurl);
        m_pCurl = NULL;
        curl_global_cleanup();
    }


    m_strRemoteUrl.clear();
    m_strLocalFilePath.clear();

    m_nLocalFileSize = 0;
}


bool CUploader::setOption()
ExpandedBlockStart.gif{
    // 远程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);

    return true;
}


bool CUploader::start(const std::string& strUrl, const std::string& strLocalFile)
ExpandedBlockStart.gif{
    if (strUrl.empty()) return false;

    if (m_bRunning == truereturn true;

    clear();

    m_strRemoteUrl    = strUrl;
    m_strLocalFilePath    = strLocalFile;

    // 打开文件
    m_pFile = fopen(m_strLocalFilePath.c_str(), "rb");
    if (m_pFile == NULL)
ExpandedSubBlockStart.gif    {
        return false;
    }


    // 获取文件大小
    struct stat file_info;
    if (fstat(fileno(m_pFile), &file_info) != 0) 
ExpandedSubBlockStart.gif    {
        return false;
    }

    m_nLocalFileSize = file_info.st_size;

    // 初始化libcurl
    m_pCurl = curl_easy_init();
    if (m_pCurl == NULL)
ExpandedSubBlockStart.gif    {
        return false;
    }


    // 设置libcurl的选项
    setOption();

    m_bRunning = true;

    return true;
}


bool CUploader::stop()
ExpandedBlockStart.gif{
    clear();

    m_bRunning = false;

    return true;
}


size_t CUploader::handleRead( void *buffer, size_t size, size_t nmemb, void *userp )
ExpandedBlockStart.gif{
    CUploader* pDownloader = (CUploader*) userp;
    if (pDownloader)
ExpandedSubBlockStart.gif    {
        return pDownloader->onUpload(buffer, size, nmemb);
    }

    return 0;
}


size_t CUploader::handleProgress( void *buffer, double dltotal, double dlnow, double ultotal, double ulnow )
ExpandedBlockStart.gif{
    CUploader* pDownloader = (CUploader*) buffer;
    if (pDownloader)
ExpandedSubBlockStart.gif    {
        pDownloader->onProgress(ultotal, ulnow);
    }

    return 0;
}


size_t CUploader::onProgress( const double& ultotal, const double& ulnow )
ExpandedBlockStart.gif{
    TRACE("%.2f / %.2f (%.2g %%)\n", ulnow, ultotal, ulnow*100.0/ultotal);
    return 0;
}


size_t CUploader::onUpload( void *buffer, size_t size, size_t nmemb )
ExpandedBlockStart.gif{
    size_t return_size = fread(buffer, size, nmemb, m_pFile);
    return return_size;
}


void CUploader::onUpload()
ExpandedBlockStart.gif{
    // 执行上传
    CURLcode return_code = CURLE_OK;
    return_code = curl_easy_perform(m_pCurl);

    // 关闭文件
    if (m_pFile)
ExpandedSubBlockStart.gif    {
        fclose(m_pFile);
        m_pFile = NULL;
    }


    // 上传失败
    if (return_code != CURLE_OK)
ExpandedSubBlockStart.gif    {
        return;
    }


    // 获取状态码
    int response_code = 0;
    curl_easy_getinfo(m_pCurl, CURLINFO_RESPONSE_CODE, &response_code);
}


示例代码:
CUploader uploader
uploader.start("ftp://upload:upload@127.0.0.1/hello.exe", "C:\\fly.exe");
uploader.run();

版权声明:本文内容由阿里云实名注册用户自发贡献,版权归原作者所有,阿里云开发者社区不拥有其著作权,亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容,填写侵权投诉表单进行举报,一经查实,本社区将立刻删除涉嫌侵权内容。

相关文章
CentOS 7.9 基于LAMP架构搭建DISCUZ论坛
使用阿里云服务器搭建Linux服务器,并搭建基于LAMP架构的DISCUZ论坛服务,并通过内网正常的访问到自己的DISCUZ论坛界面。
6 0
Pycharm远程ECS部署Django小项目
本文完成了一个简易的Django项目的部署,ECS上选用的系统是Ubuntu,所用的Django项目是笔者本科期间的课程设计,详见https://github.com/ztingz/NTCS
15 0
如何科学的使用无影云电脑
我当时整个人都高兴的跳起来了,由于坐下的时候太着急,一屁股坐在了地上,导致我之后的一段时间都只能半只屁股坐在椅子上,带薪摸鱼的难度得到大幅度提升。 作为一名网络安全领域的博主,经常用Python做一些奇怪的操作,比如爬CSDN的热榜信息,爬b站舞蹈区的小姐姐,一边爬一边感慨中国文化的博大精深
20 0
阿里云飞天加速计划ecs服务器试用体验
关于阿里云飞天加速计划ecs服务器的试用体验
9 0
你真的了解CSS3硬件加速吗?
你真的了解CSS3硬件加速吗?
24 0
跟云服务器相比,使用高防服务器有什么好处
随着互联网的普及发展,行业建站非常普遍,无论是正规企业网站、游戏网站、购物网站,还是一些娱乐网站。网站给用户带来方便的同时,也带来了各种网络威胁,近年来网络攻击频发,网络攻击事件层出不穷,致使企业安全问题日益突出。
6 0
MQTT知识大全【科普贴】
MQTT(Message Queuing Telemetry Transport,消息队列遥测传输协议),是一种基于客户端与服务器之间发布/订阅(publish/subscribe)模式的"轻量级"通讯协议。为了保证通讯的安全性和可靠性,该协议基于TCP/IP协议构建。由IBM在1999年发布,最初被用于石油管道传感器与卫星的通讯连接,2014年,它正式成为一个OASIS开放标准并得到了大量开源语言的支持。
9 0
如何搭建一台供自己永久使用的服务器,永久运行自己的网站【再见 云服务器】
如何搭建一台供自己永久使用的服务器,永久运行自己的网站【再见 云服务器】
18 0
前后端分离 -- 深入浅出 Spring Boot + Vue 实现工程项目进度管理系统 Vue不过如此~
前后端分离 -- 深入浅出 Spring Boot + Vue 实现工程项目进度管理系统 Vue不过如此~
9 0
+关注
杨粼波
网游的老兵
1135
文章
0
问答
文章排行榜
最热
最新
相关电子书
更多
《2021云上架构与运维峰会演讲合集》
立即下载
《零基础CSS入门教程》
立即下载
《零基础HTML入门教程》
立即下载