监控USB设备插拔

最近做了UKey加密中设计到USB设备. 因UKEy是用来加密和执行PC与项目间通信加密的介质.从作用范围来讲不是传统意义上U盘作为存储介质来使用.其实熟悉网银驱动DR应该了解.在网银系统安全上一个最基本需求就是动态即时监控通信PC驱动以及设备列表通信变化.当然包括我们加密存储介质在PC上USB插拔.

思路一.在WinFrom中通过拦截Windows 消息机制来实现. 类似定义MEssageForm窗体. 假如用鼠标左击一下窗体, 系统会收到一条 WM_LBUTTONDOWN 消息;当鼠标抬起, 系统又会收到 WM_LBUTTONUP 消息.系统收到消息后, 会告诉窗体发生的事情, 然后窗体再做出反应; 当然窗体能否做出反应要看窗体是否有相应的响应代码. 同样也可以把USB设备插拔事件通过重写窗体WndProc(Ref Message m)方法拦截并处理.

首先引入命名空间:

using System.Management;
using System.Threading;
using System.Security.Permissions;

定义Device Management Event的枚举:

  public enum DeviceEvent : int
        {
            DBT_CONFIGCHANGECANCELED = 0x0019,
            DBT_CONFIGCHANGED=0x0018,
            DBT_CUSTOMEVENT=0x8006,
            DBT_DEVICEARRIVAL=0x8000,//USB Insert DEvice Statu
            DBT_DEVICEQUERYREMOVE=0x8001,
            DBT_DEVICEQUERYREMOVEFAILED=0x8002,
            DBT_DEVICEREMOVEPENDING=0x8003,//USB Revoing.
            DBT_DEVICEREMOVECOMPLETE=0x8004,//USB Remove Completed
            DBT_DEVICETYPESPECIFIC=0x8005,
            DBT_DEVNODES_CHANGED=0x0007,//Device List _Changed
            DBT_QUERYCHANGECONFIG=0x0017,
            DBT_USERDEFINED=0xFFFF
        }

其中涉及到USB设备插拔的是DEVICEREMOVEPENDING/DEVICEREMOVECOMPLETE[删除]  DEVICEARRIVAL[插入设备]  重写WinFProc实现窗体上对Windows MEssage进行拦截并重新处理:

      [PermissionSet(SecurityAction.Demand, Name = "FullTrust")]
        protected override void WndProc(ref Message m)
        {
            base.WndProc(ref m);
            DeviceEvent lEvent;

            lEvent = (DeviceEvent)m.WParam.ToInt32();
            switch (lEvent)
            {
                case DeviceEvent.DBT_DEVICEARRIVAL://[Insert]
                    this.CheckDeviceStatus_Lable.BackColor = Color.Green;
                    this.CheckDeviceStatus_Lable.Text = "----Connection Device!----";
                    MessageBox.Show("Just Insert At Moment !", "Insert");
                    break;
                case DeviceEvent.DBT_DEVICEREMOVECOMPLETE://[REmove]
                    this.CheckDeviceStatus_Lable.BackColor = Color.Red;
                    this.CheckDeviceStatus_Lable.Text = "------No Connection------";
                    MessageBox.Show("Remove Complete At Moment!", "Remove");
                    break;
                case DeviceEvent.DBT_DEVNODES_CHANGED://[Device List Have Changed]
                    MessageBox.Show("Device List have been Changed!");
                    break;
                default:
                    break;
            }
        }

首先USB在进行插拔即时如果没有存储介质即Disk或是无驱动的方式可能不能触发DeviceEvent.DBT_DEVICEREMOVECOMPLETE 和 DeviceEvent.DBT_DEVICEARRIVAL插把事件. 但是一点是可以确认的是.PC端只要接受USB设备.PC识别之后设备列表肯定会发生变化. 如果不是加密Key. u盘的 盘符方式存在Disk存储介质.则没有问题.

当然需要移植这种基于From窗体系统Message拦截方法时发现这种可重用性就不高.换一种思路采用Windows底层方式WMI实现.

WMI以CIMOM为基础，CIMOM即公共信息模型对象管理器[Common Information Model Object Manager]，是一个描述操作系统构成单元的对象数据库，为MMC和脚本程序提供了一个访问操作系统构成单元的公共接口。有了WMI，工具软件和脚本程序访问操作系统的不同部分时不需要使用不同的API；相反，操作系统的不同部分都可以插入WMI，工具软件和WMI可以方便地读写WMI.

WMI 可以产生的系统级事件的一些更有用。 每当创建 WMI 类的新实例 ； 例如对于激发称为 __instancecreationevent 事件__instancedeletionevent 时将触发一个实例将被删除.  当然USB插拔时同样获得这个系统事件：

  public void ControlUSBConnectionStatu()
        {
            ManagementEventWatcher getEventWatcher = null;
            WqlEventQuery getEventQuery = null;

            ManagementOperationObserver getObserver = new ManagementOperationObserver();

            //Bind to Loacl Machine and Watch the PortConnection
            ManagementScope getScope = new ManagementScope("root\\CIMV2");
            getScope.Options.EnablePrivileges = true;//set requeired

            try
            {
                getEventQuery = new WqlEventQuery();
                getEventQuery.EventClassName = "__InstanceOperationEvent";
                getEventQuery.WithinInterval = new TimeSpan(0, 0, 0, 1);
                getEventQuery.Condition = @"TargetInstance ISA 'Win32_DiskDrive' ";
                       //[Disk must have DiskDrive fuck ]

                //Event Watcher [Test Event and semd informatio to this message and create new informtion .]
                getEventWatcher = new ManagementEventWatcher(getEventQuery);
                getEventWatcher.EventArrived += new EventArrivedEventHandler(getEventWatcher_EventArrived);
                getEventWatcher.Start();//Start Watch Event

            }
            catch (Exception se)
            { }
            finally
            {
                // getEventWatcher.Stop();
            }
        }

当发生USB插拔并成功监听到事件时处理方法:

void getEventWatcher_EventArrived(object sender, EventArrivedEventArgs e)
       {
           ManagementBaseObject getBaseObject = (ManagementBaseObject)e.NewEvent;
           if ((getBaseObject.ClassPath.ClassName == "__InstanceCreationEvent"))
           {
               //Usb Inserted
               MessageBox.Show("USB Disk Inserted!");
           }
           else
           {
               //Usb Removed
               MessageBox.Show("USB Device Removed!");
           }
       }

如上在定义时设置一个Condition[条件]:存在Disk Driver 磁盘驱动.如果类似某些UKey不存在磁盘驱动同时也无存储介质 经过测试你会发现.加密的UKey插入 如上的WMI监听事件并不能扑捉到USB插拔. 但是针对这种方式我们还有一种更为彻底的方式就是USB插拔唯一可以确定在PC识别必然变化的因素是系统设备列表发生更新.

WMI处理监听:

       /// <summary>
        /// 监听USB Device设备插拔事件 完整操作.
        /// WMI Handle Event Change Device List chenkai
        /// </summary>
        public void RegisterDeviceWMIEventStatu()
        {
            try
            {
                //Device List HAve Changed  And Send Message
                WqlEventQuery query = new WqlEventQuery("SELECT * FROM Win32_DeviceChangeEvent");
                ManagementEventWatcher watcher = new ManagementEventWatcher(query);
                watcher.EventArrived += new EventArrivedEventHandler(watcher_EventArrived);
                watcher.Start();  // Start listening for events
            }
            catch (Exception se)
            { }
        }

WMI监听处理函数:

void watcher_EventArrived(object sender, EventArrivedEventArgs e)
       {
           string geteventtype = e.NewEvent.GetPropertyValue("EventType").ToString();
           ManagementBaseObject getEventObject = (ManagementBaseObject)e.NewEvent;

           if (getEventObject != null)
           {
               //close Operator
               this.CloseDeviceEqument();
           }
       }

其实这依然还是一种折中办法.因为导致PC端设备列表发生变化的因素有很多.如何判定是USB口发生设备变化.WMI依然没有让我们失望:

  //WMI Control The USB Device Change Statu And Send Message to When IT‘s Changed/
        public void ControlUSBDeviceSTatu()
        {
            try
            {
                WqlEventQuery query = new WqlEventQuery("select * from Win32_VolumeChangeEvent");
                ManagementEventWatcher getwatcher = new ManagementEventWatcher(query);
                getwatcher.EventArrived += new EventArrivedEventHandler(getwatcher_EventArrived);
                getwatcher.Start();
            }
            catch (Exception se)
            { }
        }

WMI在USB事情处理函数:

        void getwatcher_EventArrived(object sender, EventArrivedEventArgs e)
        {
            MessageBox.Show(e.NewEvent.GetText(TextFormat.Mof).ToString());
        }

这样来一来就可以清晰判定USB在插拔时所发生在系统PC端的变化通知给应用程序来进行处理.如何获得当前PC端 USB Driver列表.经过几番测试找到一种很好的方式获取全部的USB Driver信息:

//Get ALL USB DRiver And Driver Property Fuck this shit。chenkai
       public static string[] AllInformation()
       {
           StringCollection propNames = new StringCollection();
           ManagementClass driveClass = new ManagementClass("Win32_USBController");
           PropertyDataCollection props = driveClass.Properties;
           foreach (PropertyData driveProperty in props)
               propNames.Add(driveProperty.Name);

           int idx = 0;
           ManagementObjectCollection drives = driveClass.GetInstances();
           string _s = string.Empty;
           List<string> harddisk = new List<string>();

           foreach (ManagementObject drv in drives)
           {
               idx++;
               _s = string.Format(" USB Driver({0}) Properties ", idx);
               harddisk.Add(_s);
               foreach (string strProp in propNames)
               {
                   _s = string.Format("Property: {0}, Value: {1}", strProp, drv[strProp]);
                   harddisk.Add(_s);
               }
           }
           string[] _ss = harddisk.ToArray();
           return _ss;
       }

WMI的更多体现是对Windows 交互中进一步封装和管理.

Can考资料:

Device ManageMent Events

System.Form.control.WinProc[]Method

System.Form.Control.Message[]Method