USB DETECTION

Commentaire de saidbouch le 24/08/2009 13:37:32

#include<windows.h>
#include<stdio.h>
#include<dbt.h>
#define mbox(tst) MessageBox(0,tst,"USB detection",16)
HINSTANCE hinst=0;
DWORD init=0;
LRESULT CALLBACK WinProc(HWND,UINT uMsg,WPARAM,LPARAM);

char wclass[]="MyClass_Ex";
int WINAPI WinMain(HINSTANCE hInstance,HINSTANCE phInstance,LPSTR cmd,int shcmd)
{
    HWND hwnd;
    MSG msg;
    WNDCLASSEX wndClass;
    hinst=hInstance;
    wndClass.cbSize=sizeof(WNDCLASSEX);
    wndClass.style=0;
    wndClass.lpfnWndProc=WinProc;
    wndClass.cbClsExtra=wndClass.cbWndExtra=0;
    wndClass.hInstance=hInstance;
    wndClass.hIcon=LoadIcon (NULL, IDI_APPLICATION);
    wndClass.hCursor=LoadCursor(NULL,IDC_ARROW);
    wndClass.hbrBackground=(HBRUSH) COLOR_BACKGROUND;
    wndClass.lpszMenuName=0;
    wndClass.lpszClassName=wclass;
    wndClass.hIconSm=LoadIcon (NULL, IDI_APPLICATION);
FreeConsole();
    if(!RegisterClassEx(&wndClass))
    {
    MessageBoxA(0,"class","l",16);
    return 0;
    }
    hwnd=CreateWindowExA(0,wclass,"USB Detection [Said Bouchkaren Maroc]",WS_OVERLAPPEDWINDOW,0,0,450,400,0,NULL,hInstance,NULL);
    ShowWindow(hwnd,shcmd);
    while(GetMessage(&msg,NULL,0,0))
    {
              TranslateMessage(&msg);
        DispatchMessage(&msg);
    }
    return msg.wParam;
}

LRESULT CALLBACK WinProc(HWND hWnd,UINT uMsg,WPARAM wParam,LPARAM lParam)
{
        switch(uMsg)
        {
        case WM_DESTROY:
             PostQuitMessage(0);
             break;
        case WM_DEVICECHANGE:
                 {
                                 char ts[20];
                 DWORD drv=0;
                    PDEV_BROADCAST_HDR lpdb = (PDEV_BROADCAST_HDR)lParam;
                    PDEV_BROADCAST_VOLUME lpdbv;
                    int m=0;
                    DWORD hlp;
                    ts[0]=0;
                    switch(wParam)
                    {
                       case DBT_DEVICEARRIVAL:
                            drv=GetLogicalDrives();
                            for(m=0;m<29&&(drv>>m)!=(init>>m);m++);
                              lpdbv=(PDEV_BROADCAST_VOLUME)lpdb;
                                 sprintf (ts, "Drive %c:\\ was inserted\n",m-1+'A');
                                 mbox(ts);
                                 init=drv;
                            break;
                       case DBT_DEVICEREMOVECOMPLETE:
                            drv=GetLogicalDrives();
                            for(m=0;m<29&&(drv>>m)!=(init>>m);m++);
                              lpdbv=(PDEV_BROADCAST_VOLUME)lpdb;
                                 sprintf (ts, "Drive %c:\\ was removed\n",m-1+'A');
                                 mbox(ts);
                                 init=drv;
                            break;
                    }
                
                 }
                 break;
        
        default : return DefWindowProc(hWnd,uMsg,wParam,lParam);
        }
        return 0;
}