利用curl下载文件(进度条显示) 代码片段

版权声明：原创作品，允许转载，转载时请务必以超链接形式标明文章 原始出处 、作者信息和本声明。否则将追究法律责任。http://axiii.blog.51cto.com/396236/112836

在
项目中需要用到程序更新的功能,同事介绍说是curl中的开发库很牛x,又是跨平台(他们总是这么喜欢跨平台的东西
*_*),于是下载这个包测试了一下，确实不错。准备正式用到项目中，以下一个例子用于从互联网上抓取一个文件下载到本地，并加上进度条显示，做得挺简
陋，不过功能差不多就这样了。
程序运行预览.




首先需要加入多线程的机制，因为程序一边在下载文件，一边在显示进度条，单线程的方式肯定不行，所以我用到了wxTimer来实现,在downloadMain.h 中定义了一个wxTimer，并做了事件申明.
DECLARE_EVENT_TABLE()





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


* Name: downloadMain.h


* Purpose: Defines Application Frame


* Author: (alan)


* Created: 2008-11-14


* Copyright: (谦泰通讯)


* License:


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





#ifndef DOWNLOADMAIN_H


#define DOWNLOADMAIN_H











#include "downloadApp.h"








#include <wx/timer.h>


#include "GUIDialog.h"





class downloadDialog: public GUIDialog


{


public:


downloadDialog(wxDialog *dlg);


~downloadDialog();


void OnTimer(wxTimerEvent& event);


private:


virtual void OnClose(wxCloseEvent& event);


virtual void OnQuit(wxCommandEvent& event);


virtual void OnAbout(wxCommandEvent& event);


void downloadfile();


wxTimer* m_timerdown;


DECLARE_EVENT_TABLE()


};


#endif // DOWNLOADMAIN_H


下面是主程序的代码.


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


* Name: downloadMain.cpp


* Purpose: Code for Application Frame


* Author: (alan)


* Created: 2008-11-14


* Copyright: (谦泰通讯)


* License:


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





#ifdef WX_PRECOMP


#include "wx_pch.h"


#endif





#ifdef __BORLANDC__


#pragma hdrstop


#endif //__BORLANDC__





#include "downloadMain.h"


#include <curl/curl.h>


#include <curl/types.h>


#include <curl/easy.h>


#include "update.h"


#include <wx/msgdlg.h>


#include <wx/utils.h>


#define TIMER_ID 22222


//事件监听声明


BEGIN_EVENT_TABLE(downloadDialog, GUIDialog)


EVT_TIMER(TIMER_ID, downloadDialog::OnTimer)


END_EVENT_TABLE()





enum wxbuildinfoformat


{


short_f, long_f


};





wxString wxbuildinfo(wxbuildinfoformat format)


{


wxString wxbuild(wxVERSION_STRING);





if (format == long_f )


{


#if defined(__WXMSW__)


wxbuild << _T("-Windows");


#elif defined(__WXMAC__)


wxbuild << _T("-Mac");


#elif defined(__UNIX__)


wxbuild << _T("-Linux");


#endif





#if wxUSE_UNICODE


wxbuild << _T("-Unicode build");


#else


wxbuild << _T("-ANSI build");


#endif // wxUSE_UNICODE


}





return wxbuild;


}





//声明一个文件结构体


struct FtpFile


{


char *filename;


FILE *stream;


};





downloadDialog::downloadDialog(wxDialog *dlg)


: GUIDialog(dlg)


{





//创建一个定时器,制定TIMER_ID


m_timerdown = new wxTimer(this, TIMER_ID);


//定时器开始运行,这里会自动执行OnTimer函数


m_timerdown->Start(100);


}





downloadDialog::~downloadDialog()


{





}


//定时器操作


void downloadDialog::OnTimer(wxTimerEvent &event)


{


downloadfile();


}


//文件写入流


int my_fwrite(void *buffer, size_t size, size_t nmemb, void *stream)


{


struct FtpFile *out=(struct FtpFile *)stream;


if (out && !out->stream)


{


out->stream=fopen(out->filename, "wb");


if (!out->stream)


{


return -1;


}


}


return fwrite(buffer, size, nmemb, out->stream);


}


//进度条显示函数


int wxcurldav_dl_progress_func(void* ptr, double rDlTotal, double rDlNow, double rUlTotal, double rUlNow)


{


wxGauge* pGauge = (wxGauge*) ptr;


if(pGauge)


//设置进度条的值


pGauge->SetValue(100.0 * (rDlNow/rDlTotal));


return 0;


}


//下载文件函数


void downloadDialog::downloadfile()


{


//创建curl对象


CURL *curl;


CURLcode res;


m_staticText2->SetLabel(wxT("请耐心等待程序下载更新包..."));


struct FtpFile ftpfile=


{


//定义下载到本地的文件位置和路径


"tmp.exe",NULL


};


curl_global_init(CURL_GLOBAL_DEFAULT);





//curl初始化


curl = curl_easy_init();





//curl对象存在的情况下执行操作


if (curl)


{


//设置远端地址


curl_easy_setopt(curl, CURLOPT_URL,"http://dl_dir.qq.com/minigamefile/QQGame2008ReleaseP2_web_setup.EXE");


//执行写入文件流操作


curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, my_fwrite);


curl_easy_setopt(curl, CURLOPT_WRITEDATA, &ftpfile);


//curl的进度条声明


curl_easy_setopt(curl, CURLOPT_NOPROGRESS, FALSE);


//回调进度条函数


curl_easy_setopt(curl, CURLOPT_PROGRESSFUNCTION, wxcurldav_dl_progress_func);


//设置进度条名称


curl_easy_setopt(curl, CURLOPT_PROGRESSDATA, m_gauge1);





//进度条


m_gauge1->SetValue(0);


//写入文件


res = curl_easy_perform(curl);


m_gauge1->SetValue(100);


//释放curl对象


curl_easy_cleanup(curl);





if (CURLE_OK != res)


;


}


if (ftpfile.stream)


{


//关闭文件流


fclose(ftpfile.stream);


}


//释放全局curl对象


curl_global_cleanup();


//这一步很重要，停止定时器,不然程序会无休止的运行下去


m_timerdown->Stop();


//执行刚下载完毕的程序,进行程序更新


int pid = ::wxExecute(_T("tmp.exe"));


wxMessageBox(wxT("下载完毕，程序开始执行更新操作......"));


}


void downloadDialog::OnClose(wxCloseEvent &event)


{


Destroy();


}





void downloadDialog::OnQuit(wxCommandEvent &event)


{


Destroy();


}





void downloadDialog::OnAbout(wxCommandEvent &event)


{





}