使用 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^)/~