windows文件监控

来源：互联网发布：传奇登录器源码编辑：程序博客网时间：2024/04/30 01:30
最近打算做一个监控文件的项目，决定采用FindFirstChangeNotification()和ReadDirectoryChangesW()实现监控。网上有很多现成的例子，但我找到的代码运行起来都或多或少存在不满意的地方，所以在他们的基础上进行了小的改动。
主要函数讲解在windows下文件的监控--ReadDirectoryChangesW函数的使用，主要代码来源在vc监控。
下面放上我的代码：
（注意的是监控目录名是从主函数传入的）
<span style="font-size:18px;">// MonitorTest.cpp : Defines the entry point for the console application.//#include <windows.h>#include<iostream>#include<stdio.h>#include<stdlib.h>#include<fstream>#include<iomanip>#include<io.h>#include<time.h>#include <sstream>using namespace std;DWORD WINAPI ThreadProc(LPVOID lpParam){LPCTSTR pDirtory = (char*)lpParam;    BOOL bRet = FALSE;    BYTE Buffer[1024] = { 0 };    FILE_NOTIFY_INFORMATION *pBuffer = (FILE_NOTIFY_INFORMATION *)Buffer;    DWORD BytesReturned = 0;    HANDLE hFile = CreateFile(pDirtory,                FILE_LIST_DIRECTORY,                FILE_SHARE_READ|FILE_SHARE_DELETE|FILE_SHARE_WRITE,                NULL,                OPEN_EXISTING,                FILE_FLAG_BACKUP_SEMANTICS,                NULL);    if ( INVALID_HANDLE_VALUE == hFile )    {        return 1;    }    printf("monitor... \r\n");    fflush(stdout);    while ( TRUE )    {        ZeroMemory(Buffer, 1024);        //子线程一直在读取目录的改变        //调用系统ReadDirectoryChangesW API        bRet = ReadDirectoryChangesW(hFile,                        &Buffer,                        sizeof(Buffer),                        TRUE,                        FILE_NOTIFY_CHANGE_FILE_NAME |  // 修改文件名                        FILE_NOTIFY_CHANGE_ATTRIBUTES | // 修改文件属性                        FILE_NOTIFY_CHANGE_LAST_WRITE |                        FILE_NOTIFY_CHANGE_DIR_NAME, // 最后一次写入                        &BytesReturned,                        NULL, NULL);        if (GetLastError()==ERROR_INVALID_FUNCTION)        {        cout<<"系统不支持!"<<flush;        return 1;        }        else if(bRet == 0)        {        cout<<"找不到监控文件夹!"<<flush;        return 1;        }        else if (GetLastError()==ERROR_NOTIFY_ENUM_DIR)        {        cout<<"内存溢出!"<<flush;        return 1;        }else        {        time_t ChangeTime;        time(&ChangeTime);            char szFileName[MAX_PATH] = { 0 };            // 宽字符转换多字节            WideCharToMultiByte(CP_ACP,                                0,                                pBuffer->FileName,                                pBuffer->FileNameLength / 2,                                szFileName,                                MAX_PATH,                                NULL,                                NULL);            switch(pBuffer->Action)            {                // 添加            case FILE_ACTION_ADDED:                {                    //printf("添加 : %s\r\n", szFileName);                    cout<<ctime(&ChangeTime)<<"添加:"<<szFileName<<"\n"<<flush;                    break;                }                // 删除            case FILE_ACTION_REMOVED:                {                    //printf("删除 : %s\r\n", szFileName);                    cout<<ctime(&ChangeTime)<<"删除:"<<szFileName<<"\n"<<flush;                    break;                }                // 修改            case FILE_ACTION_MODIFIED:                {                //修改为文件夹                _finddata_t file;                string sdirfile;                sdirfile = string(pDirtory);                if(sdirfile.find_last_of("\\") != sdirfile.length()){                sdirfile = string(pDirtory)+"\\"+szFileName;                }else{                sdirfile = string(pDirtory)+szFileName;                }                long lf;                if((lf = _findfirst(sdirfile.c_str(), &file))== -1l){                cout<<"目录不存在:"<<sdirfile<<"\n"<<flush;                return 1;                }                if((file.attrib & _A_SUBDIR) == 0){                //printf("修改 : %s\r\n", szFileName);                cout<<ctime(&ChangeTime)<<"修改:"<<szFileName<<"\n"<<flush;                }                    break;                }                // 重命名            case FILE_ACTION_RENAMED_OLD_NAME:                {                    //printf("重命名 : %s", szFileName);                    cout<<ctime(&ChangeTime)<<"重命名:"<<szFileName;                    if ( pBuffer->NextEntryOffset != 0 )                    {                        FILE_NOTIFY_INFORMATION *tmpBuffer = (FILE_NOTIFY_INFORMATION *)((DWORD)pBuffer + pBuffer->NextEntryOffset);                        switch ( tmpBuffer->Action )                        {                        case FILE_ACTION_RENAMED_NEW_NAME:                            {                                ZeroMemory(szFileName, MAX_PATH);                                WideCharToMultiByte(CP_ACP,                                    0,                                    tmpBuffer->FileName,                                    tmpBuffer->FileNameLength / 2,                                    szFileName,                                    MAX_PATH,                                    NULL,                                    NULL);                                //printf(" ->  : %s \r\n", szFileName);                                cout<<"->"<<szFileName<<"\n"<<flush;                                break;                            }                        }                    }                    break;                }            case FILE_ACTION_RENAMED_NEW_NAME:                {                    //printf("重命名(new) : %s\r\n", szFileName);                    cout<<ctime(&ChangeTime)<<"重命名(new):"<<szFileName<<"\n"<<flush;                }            }        }        fflush(stdout);    }    CloseHandle(hFile);    return 0;}int main(int argc, char* argv[]){if(argc == 1){return 0;}    HANDLE hThread = CreateThread(NULL, 0, ThreadProc, argv[1], 0, NULL);    if ( hThread == NULL )    {        return -1;    }    //等待线程结束    WaitForSingleObject(hThread, INFINITE);    CloseHandle(hThread);    return 0;}</span><span style="font-size: 24px;"></span>
0 0