1. 程式人生 > >C#.NET U盤插拔監控

C#.NET U盤插拔監控

【1】涉及的知識點

1) windows訊息處理函式

protected override void WndProc(ref Message m)

捕獲Message的系統硬體改變發出的系統訊息

2) 硬體資訊類
DriveInfo

【2】核心函式

訊息常量:

/// <summary>
/// windows訊息常量
/// </summary>
class CWndProMsgConst
{
    public const int WM_DEVICECHANGE = 0x219; // 系統硬體改變發出的系統訊息
    public const int DBT_DEVICEARRIVAL = 0x8000;// 裝置檢測結束,並且可以使用
    public const int DBT_CONFIGCHANGECANCELED = 0x0019;
    public const int DBT_CONFIGCHANGED = 0x0018;
    public const int DBT_CUSTOMEVENT = 0x8006;
    public const int DBT_DEVICEQUERYREMOVE = 0x8001;
    public const int DBT_DEVICEQUERYREMOVEFAILED = 0x8002;
    public const int DBT_DEVICEREMOVECOMPLETE = 0x8004;// 裝置解除安裝或者拔出
    public const int DBT_DEVICEREMOVEPENDING = 0x8003;
    public const int DBT_DEVICETYPEHANGED = 0x0007;
    public const int DBT_QUERYCHANGSPECIFIC = 0x8005;
    public const int DBT_DEVNODES_CECONFIG = 0x0017;
    public const int DBT_USERDEFINED = 0xFFFF;
}    

掃描函式:
        /// <summary>
        /// 掃描U口裝置
        /// </summary>
        private void ScanUSBDisk()
        {
            _usbdiskList.Clear();
            DriveInfo[] drives = DriveInfo.GetDrives();

            foreach (DriveInfo drive in drives)
            {
                if ((drive.DriveType == DriveType.Removable) && !drive.Name.Substring(0, 1).Equals("A"))
                {
                    try
                    {
                        _usbdiskList.Add(drive.Name);
                    }
                    catch
                    {
                        MessageBox.Show("當前盤不能正確識別,請重新嘗試!", "提示", MessageBoxButtons.OK, MessageBoxIcon.Information);
                    }
                }
            }
        }

訊息處理函式:
        public void FillData(Form form, Message m, ListBox listbox)
        {
            _listbox = listbox;
            _form    = form;

            try
            {
                if (m.Msg == CWndProMsgConst.WM_DEVICECHANGE) // 系統硬體改變發出的系統訊息
                {
                    switch (m.WParam.ToInt32())
                    {
                        case CWndProMsgConst.WM_DEVICECHANGE:
                            break;
                        //裝置檢測結束,並且可以使用
                        case CWndProMsgConst.DBT_DEVICEARRIVAL:
                            {
                                ScanUSBDisk();
                                _listbox.Items.Clear();
                                foreach (string str in _usbdiskList)
                                {
                                    _listbox.Items.Add(str);
                                }                                
                            }
                            break;
                        // 裝置解除安裝或者拔出
                        case CWndProMsgConst.DBT_DEVICEREMOVECOMPLETE:
                            {
                                ScanUSBDisk();
                                _listbox.Items.Clear();
                                foreach (string str in _usbdiskList)
                                {
                                    _listbox.Items.Add(str);
                                }
                            }                           
                            break;
                        default:
                            break;
                    }
                }
            }
            catch (Exception ex)
            {
                MessageBox.Show("當前盤不能正確識別,請重新嘗試!", "提示", MessageBoxButtons.OK, MessageBoxIcon.Information);
            }
        }           

完整的CS封裝檔案:
using System;
using System.Collections.Generic;
using System.Text;
using System.Windows.Forms;
using System.Threading;
using System.IO;

namespace USBMonitor
{
    /// <summary>
    /// USB插拔監控類
    /// </summary>
    public class CUSBMonitor
    {
        private delegate void SetTextCallback(string s);
        private IList<string> _usbdiskList = new List<string>();
        private ListBox _listbox = null;
        private Form    _form  = null;

        public CUSBMonitor()
        {
            System.Timers.Timer timer = new System.Timers.Timer(1000);
            timer.Enabled = true;

            // 達到間隔時發生
            timer.Elapsed += new System.Timers.ElapsedEventHandler(TimerList);
            timer.AutoReset = false; // 僅在間隔第一次結束後引發一次          
        }

        public void FillData(Form form, Message m, ListBox listbox)
        {
            _listbox = listbox;
            _form    = form;

            try
            {
                if (m.Msg == CWndProMsgConst.WM_DEVICECHANGE) // 系統硬體改變發出的系統訊息
                {
                    switch (m.WParam.ToInt32())
                    {
                        case CWndProMsgConst.WM_DEVICECHANGE:
                            break;
                        //裝置檢測結束,並且可以使用
                        case CWndProMsgConst.DBT_DEVICEARRIVAL:
                            {
                                ScanUSBDisk();
                                _listbox.Items.Clear();
                                foreach (string str in _usbdiskList)
                                {
                                    _listbox.Items.Add(str);
                                }                                
                            }
                            break;
                        // 裝置解除安裝或者拔出
                        case CWndProMsgConst.DBT_DEVICEREMOVECOMPLETE:
                            {
                                ScanUSBDisk();
                                _listbox.Items.Clear();
                                foreach (string str in _usbdiskList)
                                {
                                    _listbox.Items.Add(str);
                                }
                            }                           
                            break;
                        default:
                            break;
                    }
                }
            }
            catch (Exception ex)
            {
                MessageBox.Show("當前盤不能正確識別,請重新嘗試!", "提示", MessageBoxButtons.OK, MessageBoxIcon.Information);
            }
        }           

        /// <summary>
        /// 設定USB列表
        /// </summary>
        void TimerList(object sender, System.Timers.ElapsedEventArgs e)
        {
            ScanUSBDisk();
            foreach (string str in _usbdiskList)
            {
                SetText(str);
            }
        }

        /// <summary>
        /// 掃描U口裝置
        /// </summary>
        private void ScanUSBDisk()
        {
            _usbdiskList.Clear();
            DriveInfo[] drives = DriveInfo.GetDrives();

            foreach (DriveInfo drive in drives)
            {
                if ((drive.DriveType == DriveType.Removable) && !drive.Name.Substring(0, 1).Equals("A"))
                {
                    try
                    {
                        _usbdiskList.Add(drive.Name);
                    }
                    catch
                    {
                        MessageBox.Show("當前盤不能正確識別,請重新嘗試!", "提示", MessageBoxButtons.OK, MessageBoxIcon.Information);
                    }
                }
            }
        }

        /// <summary>
        /// 設定List列表
        /// </summary>
        /// <param name="text">名稱</param>
        public void SetText(string text)
        {
            if (_listbox == null)
                return;

            if (this._listbox.InvokeRequired) // 呼叫方位於建立控制元件所在的執行緒以外的執行緒中
            {
                if (_listbox.Items.Contains(text))
                    return;

                SetTextCallback d = new SetTextCallback(SetText);
                _form.Invoke(d, new object[] { text });
            }
            else
            {
                if (_listbox.Items.Contains(text))
                    return;

                this._listbox.Items.Add(text);
            }
        }
    }

    /// <summary>
    /// windows訊息常量
    /// </summary>
    class CWndProMsgConst
    {
        public const int WM_DEVICECHANGE = 0x219; // 系統硬體改變發出的系統訊息
        public const int DBT_DEVICEARRIVAL = 0x8000;// 裝置檢測結束,並且可以使用
        public const int DBT_CONFIGCHANGECANCELED = 0x0019;
        public const int DBT_CONFIGCHANGED = 0x0018;
        public const int DBT_CUSTOMEVENT = 0x8006;
        public const int DBT_DEVICEQUERYREMOVE = 0x8001;
        public const int DBT_DEVICEQUERYREMOVEFAILED = 0x8002;
        public const int DBT_DEVICEREMOVECOMPLETE = 0x8004;// 裝置解除安裝或者拔出
        public const int DBT_DEVICEREMOVEPENDING = 0x8003;
        public const int DBT_DEVICETYPEHANGED = 0x0007;
        public const int DBT_QUERYCHANGSPECIFIC = 0x8005;
        public const int DBT_DEVNODES_CECONFIG = 0x0017;
        public const int DBT_USERDEFINED = 0xFFFF;
    }    
}

測試窗體(重寫訊息函式):
using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Drawing;
using System.Linq;
using System.Text;
using System.Windows.Forms;

namespace USBMonitor
{
    public partial class Main : Form
    {
        public Main()
        {
            InitializeComponent();
        }

        CUSBMonitor usbMonitor = new CUSBMonitor();

        protected override void WndProc(ref Message m)
        {
            usbMonitor.FillData(this, m, _listBox);

            base.WndProc(ref m);
        }
    }
}