使用 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函数变量定义之后添加如下代码:
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
if (g_iIsClearBufferPointer == 1) //g_iIsClearBufferPointer为全局变量作为Flag
{
g_iIsClearBufferPointer = 0;
p = buffer;
}
2、添加回调函数及其参数定义
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
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、外部调用下载所用函数:
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
//************************************
// 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函数使用
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
//************************************
// 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函数
第一个步骤是重置会引发错误的静态变量
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
g_iIsClearBufferPointer = 1; //清除no_prefix函数中静态变量指针
optind = 1; //清除getopt.c文件中的同名变量
total_downloaded_bytes = 0; //清除总下载量
第二个步骤是在函数中if (opt.recursive && opt.spider)的判断之后添加进度提示
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
/* 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函数中添加下载进度提示
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
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,添加下载进度提示
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
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^)/~