Http文件下载类源代码

// HttpClient.h: HTTP协议客户端的实现类

//

//////////////////////////////////////////////////////////////////////

#if !defined(_HTTPCLIENT_H__)

#define _HTTPCLIENT_H__

#include "../comm/tcp.h"

#pragma warning(disable : 4786)

#include <map>

#include <string>

using namespace std;

#define BUFFER_SIZE 8192

#define HEADS_BUFFER_SIZE 2048

#define PEEK_SIZE 500

#define TCP_TIMEOUT 5

class CHttpClient

{

public:

void Cancel();

const char* GetStatusString();

CHttpClient();

virtual ~CHttpClient();

bool Download(char* url,char* fname); //直接下载到文件

bool Download(char* url,HWND hParent,long msg = WM_USER+1); //消息方式下载数据流

inline long Instr(char* from,char* find);

virtual void OnDataRecived(char* data,long datasize,int first = false );

long GetHostName(char *url, char *buf);

int GetStatus();

long GetHeader(char* headname,char* buf);

long GetHeader(char* from,char* searchfor,char* buf);

bool Reset();

private:

long m_Message;

HWND m_hParent;

bool m_noCancel;

char m_fname[BUFFER_SIZE];

int m_status;

char m_heads_buffer[HEADS_BUFFER_SIZE];

char m_buffer[BUFFER_SIZE+1];

CTcp tcp;

map<const int,string> m_map;

};

#endif

//----------------------------------------------------------------

// HttpClient.cpp: implementation of the CHttpClient class.

//

//////////////////////////////////////////////////////////////////////

#include "stdafx.h"

#include "HttpClient.h"

#ifdef _DEBUG

#undef THIS_FILE

static char THIS_FILE[]=__FILE__;

#define new DEBUG_NEW

#endif

//////////////////////////////////////////////////////////////////////

// Construction/Destruction

//////////////////////////////////////////////////////////////////////

CHttpClient::CHttpClient()

{

Reset();

memset(m_fname,0,BUFFER_SIZE);

tcp.init();

m_map[100]= "Continue";

m_map[101]= "Switching Protocols";

m_map[200]= "OK";

m_map[201]= "Created";

m_map[202]= "Accepted";

m_map[203]= "Non-Authoritative Information";

m_map[204]= "No Content";

m_map[205]= "Reset Content";

m_map[206]= "Partial Content";

m_map[300]= "Multiple Choices";

m_map[301]= "Moved Permanently";

m_map[302]= "Found";

m_map[303]= "See Other";

m_map[304]= "Not Modified";

m_map[305]= "Use Proxy";

m_map[307]= "Temporary Redirect";

m_map[400]= "Bad Request";

m_map[401]= "Unauthorized";

m_map[402]= "Payment Required";

m_map[403]= "Forbidden";

m_map[404]= "Not Found";

m_map[405]= "Method Not Allowed";

m_map[406]= "Not Acceptable";

m_map[407]= "Proxy Authentication Required";

m_map[408]= "Request Time-out";

m_map[409]= "Conflict";

m_map[410]= "Gone ";

m_map[411]= "Length Required";

m_map[412]= "Precondition Failed";

m_map[413]= "Request Entity Too Large";

m_map[414]= "Request-URI Too Large";

m_map[415]= "Unsupported Media Type";

m_map[416]= "Requested range not satisfiabl";

m_map[417]= "Expectation Failed";

m_map[500]= "Internal Server Error";

m_map[501]= "Not Implemented";

m_map[502]= "Bad Gateway";

m_map[503]= "Service Unavailable";

m_map[504]= "Gateway Time-out";

m_map[505]= "HTTP Version not supported";

}

CHttpClient::~CHttpClient()

{

}

bool CHttpClient::Reset()

{

m_noCancel=true;

m_hParent=NULL;

m_Message=0;

memset(m_buffer,0,BUFFER_SIZE);

memset(m_heads_buffer,0,HEADS_BUFFER_SIZE);

return 1;

}

//HTTP下载主程序

bool CHttpClient::Download(char *url, HWND hParent,long msg)

{

m_Message=m_Message;

m_hParent=hParent;

//状态为忙

Reset();

char hostname[256];

memset(hostname,0,256);

GetHostName(url,hostname);

long AllSize=0;

long ReadSize=0;

bool headFinished=false;

int s;

s=tcp.connect(hostname,80,TCP_TIMEOUT,1);

if(s<0)

return false;

sprintf(m_buffer,"GET %s HTTP/1.0 \r\nHost:%s\r\nAccept:*/*\r\nUser-Agent: AutoUpdate/1.0 \r\n\r\n",url,hostname);

tcp.send(s,m_buffer,strlen(m_buffer),TCP_TIMEOUT);

//先取PEEK_SIZE个字节，这样可以确保取到Content-Length

ReadSize=tcp.recv(s,m_buffer,PEEK_SIZE,TCP_TIMEOUT);

if(ReadSize==-1)

return false;

m_buffer[ReadSize]=0;

memset(hostname,0,256);

int m,n;

m=Instr(m_buffer," ");

n=Instr(m_buffer+m+1," ");

memcpy(hostname,m_buffer+m+1,n);

m_status=atoi(hostname);

if(m_status!=200)return false;

memset(hostname,0,256);

if(GetHeader(m_buffer,"Content-Length",hostname)<0)

return false;

//取得内容总长度

AllSize=atol(hostname);

memset(hostname,0,256);

//寻找HTTP数据区表示

int i;

i=Instr(m_buffer,"\r\n\r\n");

if(i>0)

{

memcpy(m_heads_buffer,m_buffer,i+4);

headFinished=true;

ReadSize=ReadSize-i-4;

OnDataRecived(m_buffer+i+4,ReadSize,true);

}

while(ReadSize<AllSize && m_noCancel)

{

int nSize;

if(headFinished)

{

nSize=(AllSize-ReadSize)>BUFFER_SIZE?BUFFER_SIZE:(AllSize-ReadSize);

memset(m_buffer,0,BUFFER_SIZE);

if(tcp.recv(s,m_buffer,nSize,TCP_TIMEOUT)!=nSize)

return false;

ReadSize+=nSize;

m_buffer[nSize]=0;

OnDataRecived(m_buffer,nSize);

}

else

{

nSize=(AllSize-ReadSize)>(BUFFER_SIZE-ReadSize)?(BUFFER_SIZE-ReadSize):(AllSize-ReadSize);

if(tcp.recv(s,m_buffer+ReadSize,nSize,TCP_TIMEOUT)!=nSize)

return false;

ReadSize+=nSize;

m_buffer[nSize]=0;

//寻找HTTP数据区表示

i=Instr(m_buffer,"\r\n\r\n");

if(i>0)

{

memcpy(m_heads_buffer,m_buffer,i+4);

headFinished=true;

ReadSize=ReadSize-i-4;

OnDataRecived(m_buffer+i+4,ReadSize,true);

}

}

}

if(!m_noCancel)remove(m_fname);

tcp.close(s);

return true;

}

int CHttpClient::GetStatus()

{

return m_status;

}

//从URL中取出主机名

long CHttpClient::GetHostName(char *url, char *buf)

{

int i,j;

char* lpsz = _tcsstr(url, "//");

if(!lpsz)return -1;

i=lpsz-url+2;

lpsz= _tcsstr(url+i, "/");

if(!lpsz)return -1;

j=lpsz-url;

memcpy(buf,url+i,j-i);

return j-i;

}

long CHttpClient::GetHeader(char *headname, char *buf)

{

return GetHeader(m_heads_buffer,headname,buf);

}

long CHttpClient::GetHeader(char *from, char *headname, char *buf)

{

char *p;

char* lpsz = _tcsstr(from, headname);

if(!lpsz)return -1;

//lpsz++;

p=lpsz+strlen(headname)+2;

lpsz = _tcsstr(p, "\r\n");

if(!lpsz)return -1;

memcpy(buf,p,lpsz-p);

return strlen(buf);

}

void CHttpClient::OnDataRecived(char* data,long datasize ,int first)

{

if(m_hParent)

::SendMessage(m_hParent,m_Message,(unsigned int)data,datasize);

else

{

if(first)remove(m_fname);

FILE* f=fopen(m_fname,"ab");

if(f){

fwrite(data,1,datasize,f);

fclose(f);

}

}

//处理数据，一定要拷贝走哦，缓冲区可是循环使用的

}

inline long CHttpClient::Instr(char *from, char *find)

{

char* lpsz = _tcsstr(from, find);

return (!lpsz)?-1:lpsz-from;

}

bool CHttpClient::Download(char *url, char *fname)

{

strcpy(m_fname,fname);

return Download(url,NULL,NULL);

}

const char* CHttpClient::GetStatusString()

{

return m_map[m_status].c_str();

}

void CHttpClient::Cancel()

{

m_noCancel=false;

}

• ·在Tomcat中安装JSTL
• ·抓屏技巧二三例
• ·原来对下载文件进行速度限制这么简单（ISAPI
• ·网络故障的技术一些东东
• ·Visual Studio 2005 产品线总
• ·穷人搭建j2ee开发平台_eclipse3.0
• ·用Media Player设计个性化媒体播放器
• ·VC6.0下利用消息实现内部进程通讯
• ·select 补集 not in
• ·Java中的排序