使用 Visual Studio 编译 wget 为库文件

添加代码与预编译指令与上一篇使用 Visual Studio 编译 wget 为可执行文件一致，区别在于这回建的是静态库工程（编译为动态库过程类似:)）

从wget的main函数开始读下来，发现问题不少，程序可能基于效率或者编码方便的因素，大量使用静态变量，导致在将其修改为静态库之后存在潜在危险。

需要修改的部分如下：

1、main.c文件，no_prefix函数：

static char *
no_prefix (const char *s)
{
static char buffer[1024];
static char *p = buffer;
 ...
}

由于原程序使用指针p来指向尚未处理的数据，no_prefix函数调用一次p指针向后移动一段距离，直到buffer的末尾。

因为原有wget程序一次运行完成之后即退出，这段代码不会出问题，但在做为函数调用进行多次下载的时候，指针p没有移动到buffer的开头，就导致了缓冲区溢出问题。

只需在一次下载结束之后对其设置重置指针p的标志即可在下次执行下载任务开始前对指针p进行归位处理。

在no_prefix函数变量定义之后添加如下代码：

if (g_iIsClearBufferPointer == 1) //g_iIsClearBufferPointer为全局变量作为Flag
{
g_iIsClearBufferPointer = 0;
p = buffer;
}

2、添加回调函数及其参数定义

typedef enum{
Loading,
Complete,
Fail = -1
} DownloadInfoType;

typedef struct DownloadInfo_tag
{
float Percent;
char *pInfoStr;
void *pTag;
DownloadInfoType InfoType;
} DownloadInfo;

typedef void (* DOWNLOAD_CALLBACK)(DownloadInfo);
DOWNLOAD_CALLBACK g_download_process_callback = NULL;

3、外部调用下载所用函数：

//************************************
// Method: downloadFile
// FullName: downloadFile
// Access: public
// Returns: void
// Qualifier: 下载所用函数
// Parameter: char * pCmdStr：下载参数字符串，与wget命令行下载方式相同
// Parameter: DOWNLOAD_CALLBACK callbackFun：传入的回调函数指针
//************************************
void downloadFile(char *pCmdStr, DOWNLOAD_CALLBACK callbackFun)
{
char **cmd;
int iItemNum = 0;
int i;
g_download_process_callback = callbackFun;
cmd = (char**) malloc(sizeof(char*) * MAX_CMD_NUM);
for (i = 0; i < MAX_CMD_NUM; i++)
{
cmd[i] = ( char* ) malloc(sizeof(char) * MAX_PATH);
ZeroMemory(cmd[i], MAX_PATH);
}
divideCmdString(pCmdStr, cmd, MAX_CMD_NUM, &iItemNum);

wget_main(iItemNum, cmd);　　//原有wget的main函数

for (i = 0; i < MAX_CMD_NUM; i++)
{
free(cmd[i]);
cmd[i] = NULL;
}
free(cmd);
cmd = NULL;
}

4、模拟系统分割参数字符串供wget的原有main函数使用

//************************************
// Method: divideCmdString
// FullName: divideCmdString
// Access: public
// Returns: void
// Qualifier:
// Parameter: char * pCmdStr：输入参数串
// Parameter: char * * pItems：输出分割好的参数
// Parameter: int iMaxItemNum：最大参数数目
// Parameter: int * pIItemNum：分割得到的参数数目
//************************************
void divideCmdString(char *pCmdStr, char **pItems, int iMaxItemNum, int *pIItemNum)
{
if (pCmdStr != NULL && pItems != NULL && pIItemNum != NULL)
{
int iCmdLen = strlen(pCmdStr);
int iCounter = 0;
int iInnerCounter = 0;
int i;
for (i = 0; i < iCmdLen; i++)
{
if (*(pCmdStr + i) == ' ') //是空格
{
if (iInnerCounter > 0)
{
pItems[iCounter][iInnerCounter] = '\0';
iInnerCounter = 0;
iCounter++;
if (iCounter >= iMaxItemNum)
{
*pIItemNum = --iCounter;
return;
}
}
}
else
{
pItems[iCounter][iInnerCounter] = *(pCmdStr + i);
iInnerCounter++;
}
}
pItems[iCounter][iInnerCounter] = '\0';
if (iInnerCounter == 0)
{
*pIItemNum = iCounter;
}
else
{
*pIItemNum = ++iCounter;
}

}
}

5、wget的main函数

第一个步骤是重置会引发错误的静态变量

g_iIsClearBufferPointer = 1; //清除no_prefix函数中静态变量指针
optind = 1; //清除getopt.c文件中的同名变量
total_downloaded_bytes = 0; //清除总下载量

第二个步骤是在函数中if (opt.recursive && opt.spider)的判断之后添加进度提示

/* Print broken links. */
if (opt.recursive && opt.spider)
{
print_broken_links();
}
//进度提示，此处为下载失败消息
if (total_downloaded_bytes == 0)
{
DownloadInfo info;
ZeroMemory(&info, sizeof(DownloadInfo));
info.InfoType = Fail;
g_download_process_callback(info);
}
/* Print the downloaded sum. */

6、在ftp.c的ftp_retrieve_list函数中添加下载进度提示

if (file_exists_p (con->target))
{
int iLen = strlen(con->target);
char *pSubStr = ".listing";
int iSubStrLen = strlen(pSubStr);
//避免在检测listing文件时错误发送complete消息 //linsy
if (strcmp(con->target + (iLen - iSubStrLen), pSubStr) != 0)
{
DownloadInfo info;
info.InfoType = Complete;
info.Percent = 100;
g_download_process_callback(info);
}
}

7、mswindows.c，添加下载进度提示

float g_fOldPercent = -1;

void send_download_process_info(float percent)
{
DownloadInfo info;
if(g_fOldPercent == floor(percent))
{
return ;
}
g_fOldPercent = floor(percent);
info.Percent = floor(percent);

info.pInfoStr = NULL;
if (info.Percent == 100)
{
info.InfoType = Complete;
}
else
{
info.InfoType = Loading;
}

if (g_download_process_callback != NULL)
{
g_download_process_callback(info);
}

}

在void ws_percenttitle (double percentage_float)函数中添加调用：send_download_process_info(percentage_float);

接下来就可以开始使用wget的函数直接进行下载了。\(^o^)/~