/// <summary>
/// when starting, get a list of all existing logical disk
/// </summary>
/// <returns></returns>
private List <LogicalDiskInfo> GetExistingDevices()
{
if (LogHelper.CanDebug())
{
LogHelper.Begin("WMIEventWatcher.GetExistingDevices");
}
try
{
List <LogicalDiskInfo> devices = new List <LogicalDiskInfo>();
ObjectQuery diskQuery = new ObjectQuery("Select * from Win32_DiskDrive where InterfaceType='USB'");
foreach (ManagementObject drive in new ManagementObjectSearcher(diskQuery).Get())
{
ObjectQuery partQuery = new ObjectQuery(
String.Format("associators of {{Win32_DiskDrive.DeviceID='{0}'}} where AssocClass = Win32_DiskDriveToDiskPartition", drive["DeviceID"])
);
#if DEBUG
DebugPrint(drive);
#endif
foreach (ManagementObject partition in new ManagementObjectSearcher(partQuery).Get())
{
// associate partitions with logical disks (drive letter volumes)
ObjectQuery logicalQuery = new ObjectQuery(
String.Format("associators of {{Win32_DiskPartition.DeviceID='{0}'}} where AssocClass = Win32_LogicalDiskToPartition", partition["DeviceID"])
);
#if DEBUG
DebugPrint(partition);
#endif
foreach (ManagementObject logical in new ManagementObjectSearcher(logicalQuery).Get())
{
#if DEBUG
DebugPrint(logical);
#endif
LogicalDiskInfo disk = new LogicalDiskInfo();
ParseDiskDriveInfo(disk, drive);
ParseDiskLogicalInfo(disk, logical);
devices.Add(disk);
}
}
}
return(devices);
}
catch (Exception err)
{
LogHelper.Manage("WMIEventWatcher.GetExistingDevices", err);
return(null);
}
finally
{
LogHelper.End("WMIEventWatcher.GetExistingDevices");
}
}
private ManagementObjectSearcher GetDiskInformation(LogicalDiskInfo device, string device_name)
{
ManagementObjectSearcher searcher2 = null;
if (LogHelper.CanDebug())
{
LogHelper.Begin("WMIEventWatcher.GetDiskInformation");
}
try
{
ObjectQuery query1 = new ObjectQuery("SELECT * FROM Win32_DiskDrive where PNPDeviceID like '%" + device_name + "%'");
searcher2 = new ManagementObjectSearcher(query1);
ManagementObjectCollection disks = searcher2.Get();
foreach (var disk in disks)
{
#if DEBUG
DebugPrint(disk);
#endif
//Use the disk drive device id to find associated partition
ObjectQuery query2 = new ObjectQuery("ASSOCIATORS OF {Win32_DiskDrive.DeviceID='" + Convert.ToString(disk["DeviceID"]).Replace("\"", "\\") + "'} WHERE AssocClass=Win32_DiskDriveToDiskPartition");
searcher2 = new ManagementObjectSearcher(query2);
ManagementObjectCollection partitions = searcher2.Get();
foreach (var part in partitions)
{
#if DEBUG
DebugPrint(part);
#endif
//Use partition device id to find logical disk
ObjectQuery query3 = new ObjectQuery("ASSOCIATORS OF {Win32_DiskPartition.DeviceID='" + Convert.ToString(part["DeviceID"]).Replace("\"", "\\") + "'} WHERE AssocClass=Win32_LogicalDiskToPartition");
searcher2 = new ManagementObjectSearcher(query3);
ManagementObjectCollection logic = searcher2.Get();
foreach (var disklogic in logic)
{
#if DEBUG
DebugPrint(disklogic);
#endif
ParseDiskDriveInfo(device, disk);
ParseDiskLogicalInfo(device, disklogic);
}
}
}
}
catch (Exception err)
{
LogHelper.Manage("WMIEventWatcher.GetDiskInformation", err);
}
finally
{
LogHelper.End("WMIEventWatcher.GetDiskInformation");
}
return(searcher2);
}
private void ParseDiskDriveInfo(LogicalDiskInfo disk, ManagementBaseObject drive)
{
if (LogHelper.CanDebug())
{
LogHelper.Begin("WMIEventWatcher.ParseDiskDriveInfo");
}
try
{
disk.PNPDeviceID = drive["PNPDeviceID"].ToString().Replace("\\", "");
disk.Model = drive["Model"].ToString();
disk.Caption = drive["Caption"].ToString();
}
catch (Exception err)
{
LogHelper.Manage("WMIEventWatcher.ParseDiskDriveInfo", err);
}
finally
{
LogHelper.End("WMIEventWatcher.ParseDiskDriveInfo");
}
}
private void ParseDiskLogicalInfo(LogicalDiskInfo disk, ManagementBaseObject logical)
{
if (LogHelper.CanDebug())
{
LogHelper.Begin("WMIEventWatcher.ParseDiskLogicalInfo");
}
try
{
disk.Name = logical["Name"].ToString();
disk.Path = logical["Name"].ToString();
disk.VolumeLabel = logical["VolumeName"].ToString();
disk.AvailableFreeSpace = (long)logical["FreeSpace"];
disk.TotalSize = (long)logical["Size"];
}
catch (Exception err)
{
LogHelper.Manage("WMIEventWatcher.ParseDiskLogicalInfo", err);
}
finally
{
LogHelper.End("WMIEventWatcher.ParseDiskLogicalInfo");
}
}
/// <summary>
/// handle the remove event
/// </summary>
/// <param name="sender"></param>
/// <param name="e"></param>
private void HandleRemovedEvent(object sender, EventArrivedEventArgs e)
{
if (LogHelper.CanDebug())
{
LogHelper.Begin("WMIEventWatcher.HandleRemovedEvent");
}
try
{
ManagementBaseObject targetInstance = e.NewEvent["TargetInstance"] as ManagementBaseObject;
#if DEBUG
DebugPrint(targetInstance);
#endif
string PNP_deviceID = Convert.ToString(targetInstance["Dependent"]).Split('=').Last().Replace("\"", "").Replace("\\", "");
LogicalDiskInfo device = Devices.Find(x => x.PNPDeviceID == PNP_deviceID);
if (device != null)
{
Devices.Remove(device);
if (EventArrived != null)
{
EventArrived(this, new WMIEventArgs()
{
Disk = device, EventType = WMIActions.Removed
});
}
}
}
catch (Exception err)
{
LogHelper.Manage("WMIEventWatcher.HandleRemovedEvent", err);
}
finally
{
LogHelper.End("WMIEventWatcher.HandleRemovedEvent");
}
}
/// <summary>
/// handle the add event
/// </summary>
/// <param name="sender"></param>
/// <param name="e"></param>
private void HandleAddedEvent(object sender, EventArrivedEventArgs e)
{
LogicalDiskInfo device = null;
if (LogHelper.CanDebug())
{
LogHelper.Begin("WMIEventWatcher.HandleAddedEvent");
}
try
{
ManagementBaseObject targetInstance = e.NewEvent["TargetInstance"] as ManagementBaseObject;
#if DEBUG
DebugPrint(targetInstance);
#endif
// get the device name in the dependent property Dependent: extract the last part
// \\FR-L25676\root\cimv2:Win32_PnPEntity.DeviceID="USBSTOR\\DISK&VEN_USB&PROD_FLASH_DISK&REV_1100\\AA04012700076941&0"
string PNP_deviceID = Convert.ToString(targetInstance["Dependent"]).Split('=').Last().Replace("\"", "").Replace("\\", "");
string device_name = Convert.ToString(targetInstance["Dependent"]).Split('\\').Last().Replace("\"", "");
// query that device entity
ObjectQuery query = new ObjectQuery(string.Format("Select * from Win32_PnPEntity Where DeviceID like \"%{0}%\"", device_name));
// check if match usb removable disk
using (ManagementObjectSearcher searcher = new ManagementObjectSearcher(query))
{
ManagementObjectCollection entities = searcher.Get();
//first loop to check USBSTOR
foreach (var entity in entities)
{
string service = Convert.ToString(entity["Service"]);
if (service == "USBSTOR")
{
device = new LogicalDiskInfo();
}
}
if (device != null)
{
foreach (var entity in entities)
{
string service = Convert.ToString(entity["Service"]);
if (service == "disk")
{
GetDiskInformation(device, device_name);
Devices.Add(device);
if (EventArrived != null)
{
EventArrived(this, new WMIEventArgs()
{
Disk = device, EventType = WMIActions.Added
});
}
}
}
}
}
}
catch (Exception err)
{
LogHelper.Manage("WMIEventWatcher.HandleAddedEvent", err);
}
finally
{
LogHelper.End("WMIEventWatcher.HandleAddedEvent");
}
}
