且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

浅谈一个线程通信代码的内存泄露及解决方案

更新时间:2022-09-02 10:44:53

//线程参数
typedef struct _MyData 
{
public:
    CString szFromPath;//源目录
    CString szToPath;//目标目录
} MYDATA, *PMYDATA;

//拷贝线程回送消息
typedef struct _CopyResponse
{
    DWORD _copiedFiles;
    CString msg;//消息体
}COPYRESPONSE,*PCOPYRESPONSE;


#define    WM_MYCOPYTHREAD (WM_USER+100)  //来自拷贝线程的消息

//保存创建的线程
DWORD dwThreadId[MAX_THREADS];
HANDLE hThread[MAX_THREADS];


    hThread[0] = CreateThread(NULL,0, CopyFiles,&data,0,&dwThreadId[0]);    //创建文件复制线程

DWORD CopyFiles(LPVOID lParam)
{//复制文件线程运行函数
    PMYDATA pData;
    pData = (PMYDATA)lParam;
    RealCopyFiles(pData->szFromPath,pData->szToPath); //做实际的文件复制工作,这里面也有很多传给主线程的消息
    //构造传给主线程的消息体
    PCOPYRESPONSE pInfo = new COPYRESPONSE;
    pInfo->_copiedFiles = dwCopiedFilesNum;
    pInfo->msg = _T("END");
    ::PostMessage(hMyMainWnd,WM_MYCOPYTHREAD,(WPARAM)pInfo,NULL);//发出结束消息
    return 0;
}

LRESULT  CMultiThreadBackDlg::OnCopyThreadMsg(WPARAM wParam,LPARAM lParam)
{//处理来自拷贝线程的消息
    PCOPYRESPONSE pInfo = (PCOPYRESPONSE)wParam;
//…这里是具体的处理工作,比如更新界面上的进度条等等
    delete pInfo;
    return 0;
}


这个方法思路比较清楚,启动子线程去负责复制文件,在子线程中不断地向主线程发出特定的消息,请求主线程去更新界面。分工合作,不错的实现,不是吗?遗憾的是,这个代码有一个致命的缺陷--内存泄露。你试着在子线程正在复制文件时,突然关闭对话框界面,看看你的VS中报告了什么,呵呵,眼熟的内存泄露来了。
2008061601.jpg
先别着急,google下看看高手们如何探讨这个问题的,个人推荐CSDN上的这个帖子http://topic.csdn.net/t/20040803/17/3239143.html,比较细致地分析了这种情况的原因和危害。问题的原因很简单,是因为在子线程里new了对象,post给父线程后希望由后者来delete掉这个对象,但当你强制关掉了对话框,此时父线程就挂掉了,那么自然没有人来帮你delete掉还没干掉的对象了,自然就泄露了。
      那么如何解决这个问题呢,帖子中的方法我没有去尝试,这里给出我的一个解决方案。
   我的解决方案主体代码:
//线程参数
typedef struct _MyData 
{
public:
    CString szFromPath;//源目录
    CString szToPath;//目标目录
    CStatic* pCurFile;//显示当前复制的文件
} MYDATA, *PMYDATA;

创建子线程代码:
        data.szFromPath = this->m_strFromPath;//源目录
        data.szToPath = this->m_strFullToPath;//目标目录
        data.pCurFile = &m_static_info;//把主界面的控件指针传给子线程
        hThread[0] = CreateThread(NULL,0, CopyFiles,&data,0,&dwThreadId[0]);    //创建文件复制线程

DWORD CMultiThreadBackDlg::CopyFiles(LPVOID lParam)
{//复制文件线程运行函数
    PMYDATA pData;
    pData = (PMYDATA)lParam;
    if(bChildExit==TRUE)
    {
        return 0;
    }
    RealCopyFiles(pData->szFromPath,pData->szToPath,pData);//做实际的文件复制工作
    pData->pCurFile->SetWindowTextW(_T("完成复制"));
    return 0;
}

void CMultiThreadBackDlg::OnCancel()
{
    this->ShowWindow(SW_HIDE);
    bChildExit = TRUE;//设置子线程退出标志
    if(hThread[0]!=INVALID_HANDLE_VALUE)
    {//等待子线程退出
        WaitForThreadToTerminate(hThread[0]);
        hThread[0] = INVALID_HANDLE_VALUE;
    }
    CDialog::OnCancel();
}

void CMultiThreadBackDlg::PeekMsgLoop()
{//自定义消息处理,防止主窗体阻塞消息处理
    MSG msg;
    while(::PeekMessage(&msg,this->GetSafeHwnd(),NULL,NULL,PM_REMOVE))
    {
        ::TranslateMessage(&msg);
        ::DispatchMessage(&msg);
    }
}

void CMultiThreadBackDlg::WaitForThreadToTerminate(HANDLE  hThread)
{//等待子线程终结
    DWORD dwRet;
    do
    {
        dwRet = ::MsgWaitForMultipleObjects(1,&hThread,FALSE,INFINITE,QS_ALLINPUT);
        if(dwRet!=WAIT_OBJECT_0)
        {
            PeekMsgLoop();
        }
    }while((dwRet!=WAIT_OBJECT_0)&&(dwRet!=WAIT_FAILED));
}


      将界面主线程上控件的指针传入子线程,这避免了在堆中new对象然后post给主线程去delete这种不好的操作,但它本身并不能避免内存泄露,试想你突然强制关闭了对话框,而子线程还在运行中,那么子线程中引用它的指针就失效了,还是会出问题的。
      这里我采用的技巧是重载窗口关闭的处理函数,在这里提供自己的窗口消息处理函数去处理窗口的消息,防止窗口阻塞在这种无聊的事情上,同时设置子线程退出标志,并等待子线程完成,等这一切都处理完毕后我们才真正离开,为了避免产生“时差”误导用户,我们可以起先用this->ShowWindow(SW_HIDE);这样的代码来“糊弄”下用户。



本文转自Phinecos(洞庭散人)博客园博客,原文链接:http://www.cnblogs.com/phinecos/archive/2008/06/16/1223326.html,如需转载请自行联系原作者