ReadDirectoryChangesW 监控文件夹 (一个简单的监控示例程序)（文件被修改了，也可以探测到）

1. // .h文件
2. #pragma once
3. typedef void (*PFN_NotifyAction)(DWORD dwAction, LPWSTR szFile, DWORD dwLength);
4. class CDirectoryWatch
5. {
6. public:
7. CDirectoryWatch(void);
8. virtual ~CDirectoryWatch(void);
9. public:
10. BOOL StartDirectoryWatch(LPCTSTR lpszDirectory, PFN_NotifyAction pFn_NotifyAction);
11. BOOL StopDirectoryWatch(void);
12. private:
13. static UINT __cdecl ThreadProc(LPVOID lParam);
14. static UINT __cdecl DirectoryWatch(LPVOID lParam);
15. private:
16. HANDLE m_hFile;
17. CWinThread* m_pThread;
18. TCHAR m_szDirectory[MAX_PATH];
19. };

1. // .cpp文件
2. #include "StdAfx.h"
3. #include "DirectoryWatch.h"
4. #include <strsafe.h>
5. typedef enum
6. {
7. MSG_STARTWATCH = (WM_USER + 0x11),
8. MSG_STOPWATCH,
9. MSG_EXITTHREAD
10. };
11. #define MAX_BUFFER_SIZE (1024)
12. typedef struct _tagWATCHPARAMETERS
13. {
14. _tagWATCHPARAMETERS()
15. {
16. hFile = INVALID_HANDLE_VALUE;
17. hEvent = NULL;
18. memset(&ol, 0, sizeof(OVERLAPPED));
19. pBuffer = NULL;
20. dwBufferSize = 0;
21. bExit = FALSE;
22. pFn_NotifyAction = NULL;
23. }
24. HANDLE hFile;
25. HANDLE hEvent;
26. OVERLAPPED ol;
27. BYTE* pBuffer;
28. DWORD dwBufferSize;
29. BOOL bExit;
30. PFN_NotifyAction pFn_NotifyAction;
31. }WATCH_PARAMETERS, *PWATCH_PARAMETERS;
32. CDirectoryWatch::CDirectoryWatch() : m_hFile(INVALID_HANDLE_VALUE), m_pThread(NULL)
33. {
34. memset(m_szDirectory, 0, sizeof(m_szDirectory));
35. m_pThread = AfxBeginThread(ThreadProc, NULL, 0, CREATE_SUSPENDED, 0, NULL);
36. if(NULL == m_pThread)
37. {
38. TRACE("Error Code : %d\n", GetLastError());
39. return ;
40. }
41. m_pThread->m_bAutoDelete = FALSE;
42. m_pThread->ResumeThread();
43. }
44. CDirectoryWatch::~CDirectoryWatch()
45. {
46. if(INVALID_HANDLE_VALUE != m_hFile)
47. {
48. CloseHandle(m_hFile);
49. m_hFile = INVALID_HANDLE_VALUE;
50. }
51. if((NULL != m_pThread) && (NULL != m_pThread->m_hThread))
52. {
53. m_pThread->PostThreadMessage(MSG_EXITTHREAD, 0, 0);
54. WaitForSingleObject(m_pThread->m_hThread, INFINITE);
55. delete m_pThread;
56. m_pThread = NULL;
57. }
58. }
59. BOOL CDirectoryWatch::StartDirectoryWatch(LPCTSTR lpszDirectory, PFN_NotifyAction pFn_NotifyAction)
60. {
61. if(NULL == m_pThread)
62. {
63. return FALSE;
64. }
65. if(NULL == lpszDirectory)
66. {
67. return FALSE;
68. }
69. if(NULL == pFn_NotifyAction)
70. {
71. return FALSE;
72. }
73. if(!PathFileExists(lpszDirectory))
74. {
75. TRACE("Error Code : %d\n", GetLastError());
76. return FALSE;
77. }
78. if(!PathIsDirectory(lpszDirectory))
79. {
80. TRACE("Error Code : %d\n", GetLastError());
81. return FALSE;
82. }
83. if(0 == _tcslen(m_szDirectory))
84. {
85. StringCchPrintf(m_szDirectory, _countof(m_szDirectory), _T("%s"), lpszDirectory);
86. }
87. else if(CSTR_EQUAL != CompareStringOrdinal(m_szDirectory, -1, lpszDirectory, -1, TRUE))
88. {
89. TRACE("Not Change Directory.\n");
90. return FALSE;
91. }
92. if(INVALID_HANDLE_VALUE == m_hFile)
93. {
94. m_hFile = CreateFile(lpszDirectory, FILE_LIST_DIRECTORY, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE,
95. NULL, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS | FILE_FLAG_OVERLAPPED, NULL);
96. if(INVALID_HANDLE_VALUE == m_hFile)
97. {
98. TRACE("Error Code : %d\n", GetLastError());
99. return FALSE;
100. }
101. }
102. return m_pThread->PostThreadMessage(MSG_STARTWATCH, (WPARAM)m_hFile, (LPARAM)pFn_NotifyAction);
103. }
104. BOOL CDirectoryWatch::StopDirectoryWatch()
105. {
106. if(NULL != m_pThread)
107. {
108. return m_pThread->PostThreadMessage(MSG_STOPWATCH, 0, 0);
109. }
110. return FALSE;
111. }
112. UINT __cdecl CDirectoryWatch::DirectoryWatch(LPVOID lParam)
113. {
114. WATCH_PARAMETERS* pParam = (WATCH_PARAMETERS*)lParam;
115. if(NULL == pParam)
116. {
117. return 0;
118. }
119. HANDLE& hFile = pParam->hFile;
120. BYTE* pBuffer = pParam->pBuffer;
121. DWORD dwBufferSize = pParam->dwBufferSize;
122. OVERLAPPED& ol = pParam->ol;
123. HANDLE& hEvent = pParam->hEvent;
124. BOOL& bExit = pParam->bExit;
125. PFN_NotifyAction pFn_NotifyAction = pParam->pFn_NotifyAction;
126. DWORD dwBytesReturn = 0;
127. DWORD dwRet = WAIT_FAILED;
128. DWORD dwOffSet = 0;
129. TCHAR szFile[MAX_PATH] = {0};
130. while(TRUE)
131. {
132. if(WAIT_OBJECT_0 != WaitForSingleObject(hEvent, INFINITE))
133. {
134. TRACE("Error Code : %d\n", GetLastError());
135. break;
136. }
137. if(bExit)
138. {
139. break;
140. }
141. if(!ReadDirectoryChangesW(hFile, pBuffer, dwBufferSize, TRUE,
142. FILE_NOTIFY_CHANGE_FILE_NAME | FILE_NOTIFY_CHANGE_DIR_NAME | FILE_NOTIFY_CHANGE_ATTRIBUTES
143. | FILE_NOTIFY_CHANGE_SIZE | FILE_NOTIFY_CHANGE_LAST_WRITE | FILE_NOTIFY_CHANGE_LAST_ACCESS
144. | FILE_NOTIFY_CHANGE_CREATION | FILE_NOTIFY_CHANGE_SECURITY, &dwBytesReturn, &ol, NULL))
145. {
146. TRACE("Error Code : %d\n", GetLastError());
147. break;
148. }
149. if(!GetOverlappedResult(hFile, &ol, &dwBytesReturn, TRUE))
150. {
151. TRACE("Error Code : %d\n", GetLastError());
152. break;
153. }
154. FILE_NOTIFY_INFORMATION* pFileNotify = (FILE_NOTIFY_INFORMATION*)pBuffer;
155. do
156. {
157. if(pFn_NotifyAction && (WAIT_OBJECT_0 == WaitForSingleObject(hEvent, 0)))
158. {
159. pFn_NotifyAction(pFileNotify->Action, pFileNotify->FileName, (pFileNotify->FileNameLength) / sizeof(WCHAR));
160. }
161. dwOffSet = pFileNotify->NextEntryOffset;
162. pFileNotify = (FILE_NOTIFY_INFORMATION*)((BYTE*)pFileNotify + dwOffSet);
163. } while (dwOffSet);
164. }
165. TRACE0("DirectoryWatch Thread Exit ... \n");
166. return 0;
167. }
168. UINT __cdecl CDirectoryWatch::ThreadProc(LPVOID lParam)
169. {
170. WATCH_PARAMETERS* pParam = new WATCH_PARAMETERS;
171. if(NULL == pParam)
172. {
173. goto __CLEANUP__;
174. }
175. BYTE* pBuffer = new BYTE[MAX_BUFFER_SIZE];
176. if(NULL == pBuffer)
177. {
178. goto __CLEANUP__;
179. }
180. memset(pBuffer, 0, MAX_BUFFER_SIZE);
181. pParam->pBuffer = pBuffer;
182. pParam->dwBufferSize = MAX_BUFFER_SIZE;
183. HANDLE hWatchEvent  = CreateEvent(NULL, TRUE, FALSE, NULL);
184. if(NULL == hWatchEvent)
185. {
186. goto __CLEANUP__;
187. }
188. pParam->ol.hEvent = hWatchEvent;
189. CWinThread* pThread = NULL;
190. HANDLE hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
191. if(NULL == hEvent)
192. {
193. goto __CLEANUP__;
194. }
195. pParam->hEvent = hEvent;
196. MSG msg;
197. while(GetMessage(&msg, NULL, 0, 0))
198. {
199. switch(msg.message)
200. {
201. case MSG_STARTWATCH:
202. {
203. HANDLE hFile = (HANDLE)(msg.wParam);
204. PFN_NotifyAction pFn_NotifyAction = (PFN_NotifyAction)(msg.lParam);
205. if((INVALID_HANDLE_VALUE == hFile) && (NULL == pFn_NotifyAction))
206. {
207. break;
208. }
209. if(NULL == pThread)
210. {
211. pParam->hFile = hFile;
212. pParam->pFn_NotifyAction = pFn_NotifyAction;
213. pThread = AfxBeginThread(DirectoryWatch, (LPVOID)pParam, 0, CREATE_SUSPENDED, NULL);
214. if(NULL == pThread)
215. {
216. goto __CLEANUP__;
217. }
218. pThread->m_bAutoDelete = FALSE;
219. pThread->ResumeThread();
220. }
221. SetEvent(hEvent);
222. }
223. break;
224. case MSG_STOPWATCH:
225. {
226. ResetEvent(hEvent);
227. }
228. break;
229. case MSG_EXITTHREAD:
230. {
231. SetEvent(hEvent);
232. pParam->bExit = FALSE;
233. if((NULL != pThread) && (NULL != pThread->m_hThread))
234. {
235. WaitForSingleObject(pThread->m_hThread, INFINITE);
236. delete pThread;
237. pThread = NULL;
238. }
239. goto __CLEANUP__;
240. }
241. default:
242. break;
243. }
244. TranslateMessage(&msg);
245. DispatchMessage(&msg);
246. }
247. __CLEANUP__:
248. if(NULL != hWatchEvent)
249. {
250. CloseHandle(hWatchEvent);
251. hWatchEvent = NULL;
252. }
253. if(NULL != pBuffer)
254. {
255. delete[] pBuffer;
256. pBuffer = NULL;
257. }
258. if(NULL != pParam)
259. {
260. delete pParam;
261. pParam = NULL;
262. }
263. TRACE0("ThreadProc Thread Exit ...\n");
264. return 0;
265. }

1. // 测试代码
2. #include "stdafx.h"
3. #include "DirectoryWatch.h"
4. void NotifyAction(DWORD dwAction, LPWSTR szFile, DWORD dwLength)
5. {
6. switch(dwAction)
7. {
8. case FILE_ACTION_ADDED:
9. wprintf(L"FILE_ACTION_ADDED: \n\t");
10. break;
11. case FILE_ACTION_REMOVED:
12. wprintf(L"FILE_ACTION_REMOVED: \n\t");
13. break;
14. case FILE_ACTION_MODIFIED:
15. wprintf(L"FILE_ACTION_MODIFIED: \n\t");
16. break;
17. case FILE_ACTION_RENAMED_OLD_NAME:
18. wprintf(L"FILE_ACTION_RENAMED_OLD_NAME: \n\t");
19. break;
20. case FILE_ACTION_RENAMED_NEW_NAME:
21. wprintf(L"FILE_ACTION_RENAMED_NEW_NAME: \n\t");
22. break;
23. default:
24. break;
25. }
26. WCHAR szPath[MAX_PATH] = {0};
27. wmemcpy(szPath, szFile, min(dwLength, MAX_PATH));
28. wprintf(L"%s\n", szPath);
29. }
30. int _tmain(int argc, TCHAR* argv[], TCHAR* envp[])
31. {
32. CDirectoryWatch watch;
33. wprintf(L"Start Directory Watch ...\n");
34. watch.StartDirectoryWatch(_T("F:\\11"), NotifyAction);
35. Sleep(30 * 1000);
36. watch.StopDirectoryWatch();
37. wprintf(L"Stop Directory Watch ...\n");
38. Sleep(10 * 1000);
39. wprintf(L"Start Directory Watch ...\n");
40. watch.StartDirectoryWatch(_T("F:\\11"), NotifyAction);
41. Sleep(30 * 1000);
42. watch.StopDirectoryWatch();
43. wprintf(L"Stop Directory Watch ...\n");
44. Sleep(30 * 1000);
45. wprintf(L"Process Exit ...\n");
46. return 0;
47. }

效果如下图所示：

http://blog.csdn.net/visualeleven/article/details/7562014