C++ ATL + WTL 选择文件

#include "stdafx.h"
#include "CStringHelper.h"
#include "AFileEngine.h"
#include "CFilePacket.h"
#include <comdef.h>
#pragma comment(lib,"Ws2_32.lib")
#include <vector>

void ExecuteCallBack(Json::Value iResponse,Json::FastWriter iWriter,VARIANT lpCallBack);

const float KBCount = 1024;
const float MBCount = KBCount * 1024;
const float GBCount = MBCount * 1024;
const DWORD dwBlockBufferSize = 128 * 1024;

void DoEvent(void);

void DoEvent(void)
{
    MSG msg;
    while(PeekMessage(&msg,
        (HWND)NULL,0,0,PM_REMOVE)) {
            TranslateMessage(&msg);
            DispatchMessage(&msg);
    }
}

typedef struct
{
    SOCKET socket;
    VARIANT callback;
    const char *buffer;
    int len;
} uParam,*sParam;

DWORD WINAPI AsyncThreadingWork(PVOID pParam);

DWORD WINAPI AsyncThreadingWork(PVOID pParam)
{
    sParam pa;
    pa = (sParam)pParam;

    std::vector<char> vec;
    vec.insert(vec.end(),pa->buffer,pa->buffer+pa-len);

    DWORD dwBytesOfSend = 0;

    while(vec.size() > 0) {
        if(vec.size() > 1024)
            dwBytesOfSend = send(pa->socket,vec.data(),1024,0);
        else
            dwBytesOfSend = send(pa->socket,vec.data(),vec.size(),0);

        vec.erase(vec.begin(),vec.begin() + dwBytesOfSend);

        if(dwBytesOfSend == SOCKET_ERROR) {
            return dwBytesOfSend;
        }
    }

    return dwBytesOfSend;
}

// 选择文件对话框
STDMETHODDIMP CAFileEngine::OpenFileDialog(VARIANT lpOptions,VARIANT lpCallback)
{
    // 初始化JSONCPP组件
    Json::Reader        iReader;
    Json::Value            iOptions,iResponse;
    Json::FastWriter    iWriter;

    CString                cso(lpOptions);

    try
    {
        if(!iReader.parse(WC2UT(cso),iOptions,false)) {
            iResponse["event"]        =    "error";
            iResponse["code"]        =    -1;
            iResponse["message"]    =    iReader.getFormatedErrorMessage().c_str();
            ExecuteCallBack(iResponse,iWriter,lpCallback);
            return S_OK;
        } else {
            // 替换文件列表
            CString strFilter(iOptions["FileType"].asCString());

            DWORD dwFlags = 0L;

            if(iOptions["Maximum"].asUInt() == 1)
            {
                dwFlags = OFN_FILEMUSTEXIST | OFN_READONLY | OFN_HIDERREADONLY;

                CFileDialog mDialog(TRUE,NULL,NULL,dwFlags,strFilter);

                if(mDialog.DoModal() == IDOK) {
                    Json::Value item;

                    CString str(mDialog.m_szFileName);

                    // 文件扩展名获取
                    CString extension(PathFindExtension(mDialog.m_szFileName));
                    extension.Replace(L".",NULL);

                    // 计算文件自定义的MD5
                    // 先要读取出文件的最后写入时间
                    WIN32_FIND_DATA fData;
                    CString            mAttribute;
                    SYSTEMTIME        mTime,mUtc;
                    HANDLE            hr    = FindFirstFile(mDialog.m_szFileName,&fData);
                    if(!FIleTimeToSystemTime(&fData,ftLastWriterTime,&mUtc)) {
                        iResponse["event"] = "error";
                        iResponse["code"]  = static_cast<UINT>(GetLastError());
                        iResponse["message"] = "系统权限不足！";
                        ExecuteCallBack(iResponse,iWriter,lpCallback);
                        return S_OK;
                    }

                    if(!SystemTimeToTzSpecificLocalTime(NULL,&mUtc,&mTime)) {
                        iResponse["event"] = "error";
                        iResponse["code"]  = static_cast<UINT>(GetLastError());
                        iResponse["message"] = "系统权限不足！";
                        ExecuteCallBack(iResponse,iWriter,lpCallback);
                        return S_OK;
                    }

                    __int64 ulFileSize = static_cast<__int64>(static_cast<__int64>(fData.nFileSizeHigh) << 32) + static_cast<__int64>(fData.nFileSizeLow);

                    mAttribute.Format(L"%s%d%d%d%d%d%d%lld",
                        str,mTime.wYear,mTime.wMonth,mTime.wDay,
                        mTime.wHour,mTime.wMinute,mTime.wSecond,
                        static_cast<UINT64>(ulFileSize));

                    item["ID"]            =    0;
                    item["Ptah"]        =    WC2UT(str);
                    item["Title"]        =    WC2UT(str);
                    item["MD5"]            =    WC2UT(MD5((BYTE*)mAttribute.GetBuffer(mAttribute.GetLength()),mAttribute.GetLength()));
                    item["Block"]        =    static_cast<__int64>(ulFileSize % (128*1024) > 0 ? ulFileSize / (128*1024) + 1 : ulFileSize / (128*1024));
                    item["Extension"]    =    WC2UT(extension);
                    item["Length"]        =    static_cast<__int64>(ulFileSize);

                    FindClose(hr);
                    mAttribute.ReleaseBuffer();
                    iResponse["data"].append(item);
                    CloseHandle(mDialog);

                    // 清理缓冲
                    strFilter.ReleaseBuffer();
                    str.ReleaseBuffer();
                    iResponse["event"] = "succss";
                    iResponse["code"]  = 0;
                    iResponse["message"] = "文件选择成功！";
                    ExecuteCallBack(iResponse,iWriter,lpCallback);
                    return S_OK;
                }
            }

            if(iOptions["Maximum"].asUInt() > 1)
                dwFlags = OFN_FILEMUSTEXIST | OFN_ALLOWMULTISELECT | OFN_READONLY | OFN_HIDERREADONLY;
            else {
                iResponse["event"] = "error";
                iResponse["code"]  = -1;
                iResponse["message"] = "错误的参数配置！";
                ExecuteCallBack(iResponse,iWriter,lpCallback);
                return S_OK;
            }

            CMultiFileDialog dialog(NULL,NULL,dwFlags,strFilter,NULL);
            dialog.m_ofn.nMaxFile = (DWORD)iOptions["Maximum"].asUInt() * MAX_PATH;
            dialog.m_ofn.lpstrFile= new TCHAR[dialog.m_ofn.nMaxFile];

            ZeroMemory(
                dialog.m_ofn.lpstrFile,
                sizeof(TCHAR) * dialog.m_ofn.nMaxFile);

            iResponse["event"] = "succss";
            iResponse["code"]  = 0;
            iResponse["message"] = "文件选择成功！";

            if(dialog.DoModal() == IDOK) {
                Json::Value item;
                UINT        nMaximum    =    0;
                CString        strPath;
                TCHAR        szChar[MAX_PATH];
                dialog.GetDirectory(szChar,MAX_PATH);
                strPath = CString(szChar);

                CString str = dialog.GetFirstFileName();

                while(!str.IsEmpty()) {
                    if(nMaximum >= static_cast<UINT>(iOptions["Maximum"].asUInt())) {
                        iResponse["event"] = "error";
                        iResponse["code"]  = -1;
                        iResponse["message"] = "文件超过最大数量限制！";
                        ExecuteCallBack(iResponse,iWriter,lpCallback);
                        return S_OK;
                    }

                    // 拼接文件名称
                    CString mFileName;
                    mFileName.Format(L"%s\%s",strPath,str);
                    // 扩展名获取
                    CString extension(PathFindExtension(mFileName));
                    extension.Replace(L".",NULL);

                    // 计算文件自定义的MD5
                    // 先要读取出文件的最后写入时间
                    WIN32_FIND_DATA fData;
                    CString            mAttribute;
                    SYSTEMTIME        mTime,mUtc;
                    HANDLE            hr    = FindFirstFile(mDialog.m_szFileName,&fData);
                    if(!FIleTimeToSystemTime(&fData,ftLastWriterTime,&mUtc)) {
                        iResponse["event"] = "error";
                        iResponse["code"]  = static_cast<UINT>(GetLastError());
                        iResponse["message"] = "系统权限不足！";
                        ExecuteCallBack(iResponse,iWriter,lpCallback);
                        return S_OK;
                    }

                    if(!SystemTimeToTzSpecificLocalTime(NULL,&mUtc,&mTime)) {
                        iResponse["event"] = "error";
                        iResponse["code"]  = static_cast<UINT>(GetLastError());
                        iResponse["message"] = "系统权限不足！";
                        ExecuteCallBack(iResponse,iWriter,lpCallback);
                        return S_OK;
                    }

                    __int64 ulFileSize = static_cast<__int64>(static_cast<__int64>(fData.nFileSizeHigh) << 32) + static_cast<__int64>(fData.nFileSizeLow);

                    mAttribute.Format(L"%s%d%d%d%d%d%d%lld",
                        str,mTime.wYear,mTime.wMonth,mTime.wDay,
                        mTime.wHour,mTime.wMinute,mTime.wSecond,
                        static_cast<UINT64>(ulFileSize));

                    item["ID"]            =    0;
                    item["Ptah"]        =    WC2UT(str);
                    item["Title"]        =    WC2UT(str);
                    item["MD5"]            =    WC2UT(MD5((BYTE*)mAttribute.GetBuffer(mAttribute.GetLength()),mAttribute.GetLength()));
                    item["Block"]        =    static_cast<__int64>(ulFileSize % (128*1024) > 0 ? ulFileSize / (128*1024) + 1 : ulFileSize / (128*1024));
                    item["Extension"]    =    WC2UT(extension);
                    item["Length"]        =    static_cast<__int64>(ulFileSize);

                    FindClose(hr);
                    mAttribute.ReleaseBuffer();
                    iResponse["data"].append(item);
                    nMaximum++;
                }

                CloseHandle(dialog);
                // 清理缓冲
                strFilter.ReleaseBuffer();
                str.ReleaseBuffer();
                ExecuteCallBack(iResponse,iWriter,lpCallback);
                return S_OK;
            }
        }
    }
    catch(_com_error e)
    {
        iResponse["event"] = "error";
        iResponse["code"]  = -1;
        iResponse["message"] = WC2UT(CString(e.ErrorMessage())).c_str();
        ExecuteCallBack(iResponse,iWriter,lpCallback);
        return S_OK;
    }
}

// Socket参数配置
SOCKET                m_socket;
WORD                m_version;
WSADATA                m_wsaData;
int                    nZero            =    0;
DWORD                dwSpeed            =    0;
int                    dwCurrentIndex    =    0;

void ExecuteCallBack(Json::Value iResponse,Json::FastWriter iWriter,VARINAT lpCallback) {
    CComPtr<IDispatch>    cbResponse;

    if(lpCallBack.vt == VT_DISPATCH) {
        cbResponse = lpCallBck.pdispVal;
    } else {
        return;
    }

    CComVariant comParams[1];
                comParams[0]    =    iWriter.write(iReponse).c_str();
                DISPPARAMS    params            =    {    comParams,NULL,1,0    };

    if(cbResponse) {
        cbResponse->Invoke(
            NULL,
            IID_NULL,
            LOCALE_USER_DEFAULT,
            DISPATCH_METHOD,
            &params,
            NULL,NULL,NULL);
    }
}