从USB设备获取驱动器名称

Question

I have a wpf applicatin that shall detect the addition and removing of an usb stick and give me the drive name. at the moment I have this:

       protected override void OnSourceInitialized(EventArgs e)
       {
        base.OnSourceInitialized(e);

        // Adds the windows message processing hook and registers USB device add/removal notification.
        HwndSource source = HwndSource.FromHwnd(new WindowInteropHelper(this).Handle);
        if (source != null)
        {
            IntPtr windowHandle = source.Handle;
            source.AddHook(HwndHandler);
            UsbNotification.RegisterUsbDeviceNotification(windowHandle);
        }
    }

    // Convert to the Drive name (”D:”, “F:”, etc)
    private string ToDriveName(int mask)
    {
        char letter;
        string drives = "ABCDEFGHIJKLMNOPQRSTUVWXYZ";
        // 1 = A
        // 2 = B
        // 4 = C...
        int cnt = 0;
        int pom = mask / 2;
        while (pom != 0)
        {
            // while there is any bit set in the mask
            // shift it to the righ...        
            pom = pom / 2;
            cnt++;
        }

        if (cnt < drives.Length)
            letter = drives[cnt];
        else
            letter = '?';

        string strReturn;

        strReturn= letter + ":\\";
        return strReturn;
    }


    /// <summary>
    /// Method that receives window messages.
    /// </summary>
    private IntPtr HwndHandler(IntPtr hwnd, int msg, IntPtr wparam, IntPtr lparam, ref bool handled)
    {
        if (msg == UsbNotification.WmDevicechange)
        {

            switch ((int)wparam)
            {
                case UsbNotification.DbtDeviceremovecomplete:
                    Usb_DeviceRemoved(); // this is where you do your magic
                    break;
                case UsbNotification.DbtDevicearrival:
                    DEV_BROADCAST_VOLUME volume = (DEV_BROADCAST_VOLUME)Marshal.PtrToStructure(lparam, typeof(DEV_BROADCAST_VOLUME)); 
                    Usb_DeviceAdded(ToDriveName(volume.dbcv_unitmask)); // this is where you do your magic
                    break;
            }
        }

        handled = false;
        return IntPtr.Zero;
    }

    // Contains information about a logical volume.
    [StructLayout(LayoutKind.Sequential)]
    private struct DEV_BROADCAST_VOLUME
    {
        public int dbcv_size;
        public int dbcv_devicetype;
        public int dbcv_reserved;
        public int dbcv_unitmask;
    }


    private void Usb_DeviceRemoved()
    {
        //todo something
    }
    private void Usb_DeviceAdded(string strDrive)
    {
        //todo something
    }

so far this works fine, at least the detection of usb insert and remove.

But after I have insert the stick I need to know the drive name so that I can copy my files to the usb stick.

Unfortunately ToDriveName returns and '?' as drive letter.

I also tried this:

    private string ToDriveName(int Mask)
    {
        int i = 0;
        for (i = 0; i < 26; ++i)
        {
            if ((Mask & 0x1) == 0x1) break;
            Mask = Mask >> 1;
        }
        char cLetter= (char)(Convert.ToChar(i) + 'A');

        string strReturn;

        strReturn= cLetter + ":\\";
        return strReturn;
    }

then I get an E:\\ instead of the G:\\ G: is my USB stick and E is my DVD drive

In the debugger I have following values in volume: dbch_Size 0x000000d2
dbch_Devicetype 0x00000005
dbch_Unitmask 0xa5dcbf10
dbch_Reserved 0x00000000

Answer 1

Casting to DEV_BROADCAST_VOLUME only makes sense when dbch_Devicetype is DBT_DEVTYP_VOLUME (2).

You need

if (volume.dbch_Devicetype == 2) Usb_DeviceAdded(ToDriveName(volume.dbcv_unitmask));

Strictly speaking, you should first use DEV_BROADCAST_HDR , test the devicetype, and only when it is 2, use DEV_BROADCAST_VOLUME .