自己写的搜索文件的程序 点停止后卡死 不知道怎么回事
飞翔的薄荷 2011-02-25 09:55:48 // FindDlg.cpp : 实现文件//
#include "stdafx.h"
#include "Find.h"
#include "FindDlg.h"
#include "afxdialogex.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#endif
// 用于应用程序“关于”菜单项的 CAboutDlg 对话框
class CAboutDlg : public CDialogEx
{
public:
CAboutDlg();
// 对话框数据
enum { IDD = IDD_ABOUTBOX };
protected:
virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支持
// 实现
protected:
DECLARE_MESSAGE_MAP()
};
CAboutDlg::CAboutDlg() : CDialogEx(CAboutDlg::IDD)
{
}
void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
CDialogEx::DoDataExchange(pDX);
}
BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx)
END_MESSAGE_MAP()
// CFindDlg 对话框
CFindDlg::CFindDlg(CWnd* pParent /*=NULL*/)
: CDialogEx(CFindDlg::IDD, pParent)
,m_bStart(0),m_threadNumber(0)
{
// memset(m_dir,0,sizeof(m_dir));
memset(m_handle,0,sizeof(m_handle));
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
m_event = CreateEvent(0,false,0,0);
}
void CFindDlg::DoDataExchange(CDataExchange* pDX)
{
CDialogEx::DoDataExchange(pDX);
DDX_Control(pDX, IDC_LIST, m_list);
DDX_Control(pDX, IDC_NOW_PATH, m_nowpath);
DDX_Control(pDX, IDC_TEMP, m_temp);
}
BEGIN_MESSAGE_MAP(CFindDlg, CDialogEx)
ON_WM_SYSCOMMAND()
ON_WM_PAINT()
ON_WM_QUERYDRAGICON()
ON_BN_CLICKED(IDOK, &CFindDlg::OnBnClickedOk)
ON_WM_DESTROY()
ON_BN_CLICKED(ID_C, &CFindDlg::OnBnClickedC)
ON_MESSAGE(WM_FIND,FindMsg)
ON_BN_CLICKED(ID_PATH, &CFindDlg::OnBnClickedPath)
ON_COMMAND(IDCANCEL, &CFindDlg::OnIdcancel)
ON_COMMAND(IDCLOSE, &CFindDlg::OnIdclose)
ON_NOTIFY(NM_RCLICK, IDC_LIST, &CFindDlg::OnRclickList)
ON_COMMAND(ID_DIR, &CFindDlg::OnDir)
ON_COMMAND(ID_FULLPATH, &CFindDlg::OnFullpath)
ON_NOTIFY(NM_DBLCLK, IDC_LIST, &CFindDlg::OnDblclkList)
ON_EN_KILLFOCUS(IDC_TEMP, &CFindDlg::OnKillfocusTemp)
ON_NOTIFY(LVN_ITEMCHANGED, IDC_LIST, &CFindDlg::OnLvnItemchangedList)
ON_EN_CHANGE(IDC_NAME, &CFindDlg::OnChangeName)
END_MESSAGE_MAP()
// CFindDlg 消息处理程序
BOOL CFindDlg::OnInitDialog()
{
CDialogEx::OnInitDialog();
// 将“关于...”菜单项添加到系统菜单中。
// IDM_ABOUTBOX 必须在系统命令范围内。
ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
ASSERT(IDM_ABOUTBOX < 0xF000);
CMenu* pSysMenu = GetSystemMenu(FALSE);
if (pSysMenu != NULL)
{
BOOL bNameValid;
CString strAboutMenu;
bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX);
ASSERT(bNameValid);
if (!strAboutMenu.IsEmpty())
{
pSysMenu->AppendMenu(MF_SEPARATOR);
pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
}
}
// 设置此对话框的图标。当应用程序主窗口不是对话框时,框架将自动
// 执行此操作
SetIcon(m_hIcon, TRUE); // 设置大图标
SetIcon(m_hIcon, FALSE); // 设置小图标
// TODO: 在此添加额外的初始化代码
//int style = m_list.GetExtendedStyle();
//m_list.SetExtendedStyle(style|LVS_EX_FULLROWSELECT |LVS_EX_GRIDLINES );
m_list.InsertColumn(0,_T("修改时间"),0,120);
m_list.InsertColumn(1,_T("大小"),0,80);
m_list.InsertColumn(2,_T("路径"),0,500);
m_temp.ShowWindow(SW_HIDE);
return TRUE; // 除非将焦点设置到控件,否则返回 TRUE
}
void CFindDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
if ((nID & 0xFFF0) == IDM_ABOUTBOX)
{
CAboutDlg dlgAbout;
dlgAbout.DoModal();
}
else
{
CDialogEx::OnSysCommand(nID, lParam);
}
}
// 如果向对话框添加最小化按钮,则需要下面的代码
// 来绘制该图标。对于使用文档/视图模型的 MFC 应用程序,
// 这将由框架自动完成。
void CFindDlg::OnPaint()
{
if (IsIconic())
{
CPaintDC dc(this); // 用于绘制的设备上下文
SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);
// 使图标在工作区矩形中居中
int cxIcon = GetSystemMetrics(SM_CXICON);
int cyIcon = GetSystemMetrics(SM_CYICON);
CRect rect;
GetClientRect(&rect);
int x = (rect.Width() - cxIcon + 1) / 2;
int y = (rect.Height() - cyIcon + 1) / 2;
// 绘制图标
dc.DrawIcon(x, y, m_hIcon);
}
else
{
CDialogEx::OnPaint();
}
}
//当用户拖动最小化窗口时系统调用此函数取得光标
//显示。
HCURSOR CFindDlg::OnQueryDragIcon()
{
return static_cast<HCURSOR>(m_hIcon);
}
void TheEnd(void *p)
{
CFindDlg *findDlg = (CFindDlg*) p;
InterlockedExchangeAdd(&(findDlg->m_threadNumber),-1);
if(0<findDlg->m_dirList.size())
{
SetEvent(findDlg->m_event);
}
else
{
if(0==findDlg->m_threadNumber)
{
PostMessage(findDlg->m_hWnd,WM_FIND,fd_over,0);
}
}
}
unsigned __stdcall threadfun(void *p){
CFindDlg *findDlg = (CFindDlg*) p;
TCHAR dir[MAX_PATH] = {0};
TCHAR dir_old[MAX_PATH] = {0};
WIN32_FIND_DATA FindFileData;
HANDLE hFind;
CString str;
while(findDlg->m_bStart)
{
WaitForSingleObject(findDlg->m_event,INFINITE);
if(!findDlg->m_bStart)break;
InterlockedExchangeAdd(&(findDlg->m_threadNumber),1);
EnterCriticalSection(&g_cs[0]);
vector<TCHAR*>::iterator it = findDlg->m_dirList.begin();
if(findDlg->m_dirList.end()==it)
{
LeaveCriticalSection(&g_cs[0]);
TheEnd(p);
continue;
}
if(*it)
{
wcscpy_s(dir,MAX_PATH,*it);
delete *it;
*it = 0;
}
findDlg->m_dirList.erase(it);
findDlg->m_nowpath.SetWindowText(dir);
LeaveCriticalSection(&g_cs[0]);
if( _T('\\')==dir[wcslen(dir)-1])
{
dir[wcslen(dir)-1] = 0;
}
wcscpy_s(dir_old,MAX_PATH,dir);
wcscat_s(dir,MAX_PATH,_T("\\*"));
hFind = FindFirstFile(dir,&FindFileData);
// hFind = FindFirstFile(findDlg->m_findname,&FindFileData);
if(INVALID_HANDLE_VALUE!=hFind)
{
do
{
if(!findDlg->m_bStart)break;
if(FindFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
{
if(0==wcscmp(FindFileData.cFileName,_T("..")))continue;
if(0==wcscmp(FindFileData.cFileName,_T(".")))continue;
TCHAR *dir_find = new TCHAR[MAX_PATH];
wcscpy_s(dir_find,MAX_PATH,dir_old);
wcscat_s(dir_find,MAX_PATH,_T("\\"));
wcscat_s(dir_find,MAX_PATH,FindFileData.cFileName);
EnterCriticalSection(&g_cs[0]);
findDlg->m_dirList.push_back(dir_find);
LeaveCriticalSection(&g_cs[0]);
SetEvent(findDlg->m_event);
}
//else
//{
// str = FindFileData.cFileName;
// if(-1!=str.Find(findDlg->m_findname))
// {
// CString temp;
// CListCtrl &list = findDlg->m_list;
// WIN32_FIND_DATA *fd = &FindFileData;
// CTime lastwt(fd->ftLastWriteTime);
// EnterCriticalSection(&g_cs[1]);
// list.InsertItem(0,lastwt.Format(_T("%Y-%m-%d %H:%M:%S")));
// temp.Format(_T("%d"),fd->nFileSizeLow);
// list.SetItemText(0,1,temp);
// temp = CString(dir_old) + _T("\\") + fd->cFileName;
// list.SetItemText(0,2,temp);
// LeaveCriticalSection(&g_cs[1]);
// }
//}
}
while(FindNextFile(hFind,&FindFileData));
}
FindClose(hFind);
wcscpy_s(dir,MAX_PATH,dir_old);
wcscat_s(dir,MAX_PATH,_T("\\"));
wcscat_s(dir,MAX_PATH,findDlg->m_findname);
hFind = FindFirstFile(dir,&FindFileData);
if(INVALID_HANDLE_VALUE!=hFind)
{
do
{
CString temp;
CListCtrl &list = findDlg->m_list;
WIN32_FIND_DATA *fd = &FindFileData;
CTime lastwt(fd->ftLastWriteTime);
EnterCriticalSection(&g_cs[1]);
list.InsertItem(0,lastwt.Format(_T("%Y-%m-%d %H:%M:%S")));
temp.Format(_T("%d"),fd->nFileSizeLow);
list.SetItemText(0,1,temp);
temp = CString(dir_old) + _T("\\") + fd->cFileName;
list.SetItemText(0,2,temp);
LeaveCriticalSection(&g_cs[1]); if(!findDlg->m_bStart)break;
}
while(FindNextFile(hFind,&FindFileData));
}
FindClose(hFind);
TheEnd(p);
}
SetEvent(findDlg->m_event);
return 0;
}
void CFindDlg::OnBnClickedOk()
{
// TODO: 在此添加控件通知处理程序代码
//CDialogEx::OnOK();
m_list.DeleteAllItems();
CString str;
GetDlgItem(IDC_PATH)->GetWindowText(str);
if(str.IsEmpty())
{
MessageBox(_T("请输入查找路径"));
return;
}
TCHAR *dir = new TCHAR[MAX_PATH];
wcscpy_s(dir,MAX_PATH,str);
EnterCriticalSection(&g_cs[0]);
m_dirList.push_back(dir);
LeaveCriticalSection(&g_cs[0]);
GetDlgItem(IDC_NAME)->GetWindowText(m_findname);
if(m_findname.IsEmpty())
{
MessageBox(_T("请输入要查找的文件"));
return;
}
GetDlgItem(IDC_PATH)->EnableWindow(false);
GetDlgItem(ID_PATH)->EnableWindow(false);
GetDlgItem(IDOK)->EnableWindow(false);
GetDlgItem(IDC_NAME)->EnableWindow(false);
m_bStart = true;
SetEvent(m_event);
for(int i = 0; i!=Max_Thread;++i)
{
m_handle[i] = (HANDLE)_beginthreadex(0,0,threadfun,this,0,0);
}}
void CFindDlg::OnDestroy()
{
CDialogEx::OnDestroy();
// TODO: 在此处添加消息处理程序代码
Stop();
}
BOOL CFindDlg::Stop()
{
if(false==m_bStart)return true;
m_bStart = false;
SetEvent(m_event);
EnterCriticalSection(&g_cs[0]);
vector<TCHAR*>::iterator it = m_dirList.begin();
while(it!=m_dirList.end())
{
if(*it)
{
delete *it;
*it = 0;
}
m_dirList.erase(it);
//++it;
}
//m_dirList.clear();
LeaveCriticalSection(&g_cs[0]);
//int j = 0;
for(int i = 0;i!=Max_Thread;++i)
{
if(m_handle[i]&& -1!=(unsigned)m_handle[i])
{
while(WAIT_TIMEOUT==WaitForSingleObject(m_handle[i],100))
{
//if(20<j)
//{
// _endthreadex((unsigned)m_handle[i]);
//};
SetEvent(m_event);
//++j;
}
CloseHandle(m_handle[i]);
m_handle[i] = 0;
}
} GetDlgItem(IDC_PATH)->EnableWindow(true);
GetDlgItem(ID_PATH)->EnableWindow(true);
GetDlgItem(IDOK)->EnableWindow(true);
GetDlgItem(IDC_NAME)->EnableWindow(true);
CString temp;
temp.Format(_T("共找到:%d个文件"),m_list.GetItemCount());
GetDlgItem(IDC_ALL)->SetWindowText(temp);
return true;
}
void CFindDlg::OnBnClickedC()
{
// TODO: 在此添加控件通知处理程序代码
Stop();
}
afx_msg LRESULT CFindDlg::FindMsg(WPARAM wparam, LPARAM lparam)
{
switch(wparam)
{
case fd_over:
Stop();
break;
case fd_file:
{
}
break;
default:
break;
}
return 0;
}