源码： Windows下HTTP方式单线程下载

针对昨天的试验结果，书写了一个HTTP方式单线程下载的小程序，目前尚不支持断点续传。
希望各位看客使劲拍砖～～

原理：套接字发送HTTP GET方式的请求，然后根据HTTP响应，循环接收信息。

进行下载调用函数为：DownByHttpFun
传入参数为：下载的url，保存位置、错误信息等

下载过程中，获得进度函数为： GetDownloadRate
返回float型的进度，输出参数下载文件总大小和已下载文件大小
传入参数

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <WinSock2.h>
#include <windows.h>
#include "HttpDown.h"

#pragma comment(lib, "Ws2_32.lib")

/************************************************************************/
/* 全局变量 */
/************************************************************************/
unsigned long g_FiltTotleLen = 0; //下载文件总大小
unsigned long g_DownedFileLen = 0; //已下载文件大小

/************************************************************************/
/* 函数描述：判断某一指定字符串是否为ip地址 */
/* 返回值： 是则返回0,否则返回-1 */
/* 作者： liuwp */
/* 创建时间：2008-03-27 */
/* 其他 */
/************************************************************************/
int IsIpAddrV4Str(char *pstr)
{
char a[4],b[4],c[4],d[4];
int num =0;

if ( !pstr)
return -1;

num = sscanf(pstr,"%3[0-9].%3[0-9].%3[0-9].%3[0-9]",a,b,c,d);

if ( num != 4 )
return -1;

if ( atoi(a)>255 || atoi(b)>255 || atoi(c) > 255 || atoi(d) >255 )
return -1;

return 0;
}

/************************************************************************/
/* 函数描述：根据下载地址获得HOST、IP等信息 */
/* 返回值： 成功返回0，失败返回－1 */
/* 作者： liuwp */
/* 创建时间：2008-03-27 */
/* 其他 */
/************************************************************************/
int GetDownInfoByUlr(const char *pDownUrl, char *pIpAddr, int *Port
, char *pGetInfo, int GetLen, char *RetErrorBuf, int ErrorLen)
{
int RetCode = 0;
char pUrlTempBuf[1000];
char pHostInf[500];

memset(pUrlTempBuf, 0, sizeof(pUrlTempBuf));
memset(pHostInf, 0, sizeof(pHostInf));

//参数判断
if (!pDownUrl || !pIpAddr || !Port || !pGetInfo || !RetErrorBuf)
return -1;

//拷贝字符串
strncpy(pUrlTempBuf, pDownUrl, sizeof(pUrlTempBuf)-1);

//获得HOST起始位置
char *pHostStart = strstr(pUrlTempBuf, "://");
if (!pHostStart)
{
_snprintf(RetErrorBuf, ErrorLen-1, "给定的下载URL格式错误，应该类似 http://HOST/GET ");
return -1;
}

//根据第一个 “/”获得HOST信息
pHostStart = pHostStart + strlen("://");

char *pHostEnd = strchr(pHostStart, '/');
if (!pHostEnd)
{
_snprintf(RetErrorBuf, ErrorLen-1, "给定的下载URL格式错误，应该类似 http://HOST/GET ");
return -1;
}

*pHostEnd = '/0';
_snprintf(pHostInf, sizeof(pHostInf)-1, "%s", pHostStart);

//获得GET信息
_snprintf(pGetInfo, GetLen-1, "/%s", pHostEnd+1);

//根据HOST信息获得对应的IP和端口
char *pPortStart = strchr(pHostInf, ':');
if (!pPortStart)
{
*Port = 80; //默认为80
}
else
{
*pPortStart = '/0';
sscanf(pPortStart+1, "%d", Port);
}

//判断获得地址是否为IP,不是则转换为IP
RetCode = IsIpAddrV4Str(pHostInf);
if (RetCode != 0)
{
//初始化环境
WSADATA WSAData;
WSAStartup(MAKEWORD(2,2),&WSAData);

struct hostent* lpHostEnt = gethostbyname(pHostInf);
if(lpHostEnt == NULL)
{
_snprintf(RetErrorBuf, ErrorLen-1, "获得域名对应的IP地址失败, GetLastError()=%d", GetLastError());
return -1;
}
else
{
LPSTR lpaddr = lpHostEnt->h_addr_list[0];
if(lpaddr)
{
struct in_addr inAddr;
memcpy (&inAddr, lpaddr, 4);

sprintf(pIpAddr, "%s", inet_ntoa (inAddr) );
}
}

//清除环境
WSACleanup();

}
else
{
sprintf(pIpAddr, "%s", pHostInf );
}

return 0;
}

/************************************************************************/
/* 函数描述：通过HTTP协议进行下载 */
/* 返回值： 下载成功返回0，失败返回－1 */
/* 作者： liuwp */
/* 创建时间：2008-03-27 */
/* 其他 */
/************************************************************************/
int DownByHttpFun(const char *pDownUrl, const char *pDownFile, unsigned long *pTotleLen
, char *RetErrorBuf, int ErrorLen)
{
WSADATA WSAData;
sockaddr_in ServerAddr;
fd_set fdset;
struct timeval tv;
HANDLE hFileHandle = NULL;
bool FirstFlag = true;
int RetCode = 0;
int WriteLen = 0;
char SendBuf[1000];
char RecvBuf[4096];
char IpAddr[30];
char GetInfo[600];
int Port = 0;
char *pConLen = NULL;
DWORD dwFileSize = 0;

memset(IpAddr, 0, sizeof(IpAddr));
memset(GetInfo, 0, sizeof(GetInfo));
memset(RecvBuf, 0, sizeof(RecvBuf));
memset(SendBuf, 0, sizeof(SendBuf));

if (!pDownUrl || !pDownFile || !pTotleLen || !RetErrorBuf)
{
_snprintf(RetErrorBuf, ErrorLen-1, "给定参数不正确");
return -1;
}

//根据下载地址获得HOST, GET, IP，端口等信息
RetCode = GetDownInfoByUlr(pDownUrl, IpAddr, &Port
, GetInfo, sizeof(GetInfo)-1, RetErrorBuf, ErrorLen);
if (RetCode != 0)
return -1;

//初始化环境
WSAStartup(MAKEWORD(2,2),&WSAData);

//创建套接字
SOCKET Sock = socket(AF_INET, SOCK_STREAM, 0);
if ( SOCKET_ERROR == Sock )
{
_snprintf(RetErrorBuf, ErrorLen-1, "创建套接字失败, GetLastError()= %d !", GetLastError());
goto LEXIT;
}

//给定服务器地址
ServerAddr.sin_family = AF_INET;
ServerAddr.sin_addr.s_addr = inet_addr(IpAddr);
ServerAddr.sin_port = htons(Port);

//连接服务器
RetCode = connect(Sock, (struct sockaddr *)&ServerAddr, sizeof(ServerAddr));
if (RetCode == SOCKET_ERROR)
{
_snprintf(RetErrorBuf, ErrorLen-1, "连接服务器失败, GetLastError()= %d !", GetLastError());
goto LEXIT;
}

//发送信息
_snprintf(SendBuf, sizeof(SendBuf)-1
, "GET %s HTTP/1.1/r/nHost: %s:%d/r/n/r/n"
,GetInfo, IpAddr, Port);

RetCode = send(Sock, SendBuf, strlen(SendBuf), 0);
if (RetCode < 0)
{
_snprintf(RetErrorBuf, ErrorLen-1, "发送数据失败, GetLastError()= %d !", GetLastError());

closesocket(Sock);
WSACleanup();

goto LEXIT;

}

//接收信息
FD_ZERO(&fdset);
FD_SET(Sock, &fdset);
tv.tv_sec = 2*60; //秒
tv.tv_usec = 0;

//打开文件
hFileHandle = CreateFile(pDownFile, GENERIC_WRITE, 0
, (LPSECURITY_ATTRIBUTES) NULL,CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL,(HANDLE) NULL);
if (hFileHandle == INVALID_HANDLE_VALUE)
{
_snprintf(RetErrorBuf, ErrorLen-1, "打开文件%s失败, GetLastError()= %d !"
, pDownFile, GetLastError());

goto LEXIT;
}

//置空全局变量
g_FiltTotleLen = 0; //下载文件总大小
g_DownedFileLen = 0; //已下载文件大小

while(1)
{
memset(RecvBuf, 0, sizeof(RecvBuf));
int len = select(0, &fdset, NULL, NULL, &tv);
if (len < 1)
{
break;
}

//套接字接收信息
len = recv(Sock, RecvBuf, sizeof(RecvBuf)-1 ,0);
if (len <= 0)
{
break;
}

if (!FirstFlag) //不是第一次，追加写
{
dwFileSize = GetFileSize(hFileHandle, NULL);
SetFilePointer(hFileHandle, dwFileSize, NULL, FILE_BEGIN);
RetCode = WriteFile(hFileHandle, RecvBuf, len , (DWORD*)&WriteLen, NULL);
if (RetCode == 0)
{
_snprintf(RetErrorBuf, ErrorLen-1, "写入文件失败 ! GetLastError()= %d !", GetLastError());
goto LEXIT;
}

//获得此时已经下载的文件大小
g_DownedFileLen = dwFileSize + len;

//如果已经下载完毕，则跳出
if (g_DownedFileLen == *pTotleLen)
break;

continue;
}

//判断返回值
int RetCode = 0;
sscanf(RecvBuf+strlen("HTTP/1.1 "), "%d", &RetCode);
if (RetCode < 200 || RetCode > 300)
{
_snprintf(RetErrorBuf, ErrorLen-1, "下载失败, 服务器返回码为%d !", RetCode);
goto LEXIT;
}

//获得总大小
pConLen = strstr(RecvBuf, "/r/nContent-Length:");
if (!pConLen)
{
_snprintf(RetErrorBuf, ErrorLen-1, "HTTP协议中未给定文件的大小!");
goto LEXIT;
}

sscanf(pConLen, "/r/nContent-Length: %ld/r/n", pTotleLen);
g_FiltTotleLen = *pTotleLen;

char *pEnd = strstr(RecvBuf, "/r/n/r/n");
if (!pEnd)
{
break;
}

pEnd = pEnd + strlen("/r/n/r/n");

//移动到文件尾部,写文件
dwFileSize = GetFileSize(hFileHandle, NULL);
SetFilePointer(hFileHandle, dwFileSize, NULL, FILE_BEGIN);
RetCode = WriteFile(hFileHandle, pEnd, len-(pEnd-RecvBuf) , (DWORD*)&WriteLen, NULL);
if (RetCode == 0)
{
_snprintf(RetErrorBuf, ErrorLen-1, "写入文件失败 ! GetLastError()= %d !", GetLastError());
goto LEXIT;
}

//获得此时已经下载的文件大小
g_DownedFileLen = dwFileSize + len - (pEnd - RecvBuf);

//更改第一次接收标识
FirstFlag = false;

}

//下载完毕,判断下载大小是否相同
dwFileSize = GetFileSize(hFileHandle, NULL);
if (pTotleLen && *pTotleLen != 0 && *pTotleLen != dwFileSize)
{
_snprintf(RetErrorBuf, ErrorLen-1, "下载文件不完整, 请重新下载!");
goto LEXIT;
}

//关闭文件
if (hFileHandle)
{
CloseHandle(hFileHandle);
hFileHandle = NULL;
}

//关闭套接字
if (Sock != 0)
{
closesocket(Sock);
Sock = 0;
}

//清除环境
WSACleanup();

return 0;

LEXIT:

//关闭文件
if (hFileHandle)
{
CloseHandle(hFileHandle);
hFileHandle = NULL;
}

//关闭套接字
if (Sock != 0)
{
closesocket(Sock);
Sock = 0;
}

//清除环境
WSACleanup();

return -1;

}

/************************************************************************/
/* 函数描述：获得文件下载的相关信息 */
/* 返回值： */
/* 作者： liuwp */
/* 创建时间：2008-03-28 */
/* 其他 */
/************************************************************************/
float GetDownloadRate(unsigned long *pTotleLen , unsigned long *pDownLen)
{
float DownRate = 0;

if (!pTotleLen || !pTotleLen )
return 0;

//获得文件总大小和已下载的大小
*pTotleLen = g_FiltTotleLen;
*pDownLen = g_DownedFileLen;

//计算速率
if (g_FiltTotleLen == 0)
return 0;

if (g_DownedFileLen == 0)
return 0;

DownRate = (float)g_DownedFileLen/(float)g_FiltTotleLen;

return DownRate;
}