| 7,10 → 7,14 |
|---|
| using System.Text; |
| using System.Windows.Forms; |
| using System.IO; |
| using Dolinay; // Imported class for drive insertion event handling |
| // Imported class for drive insertion event handling |
| using Dolinay; |
| |
| namespace DriveLogger |
| { |
| /// <summary> |
| /// Struct used for storing detailed information for each drive that is detected |
| /// </summary> |
| public struct DriveEntry |
| { |
| public DateTime time; |
| 21,27 → 25,39 |
|---|
| } |
| public partial class MainForm : Form |
| { |
| // List of drives that are displayed in the listview |
| private static List<DriveEntry> driveList = new List<DriveEntry>(); |
| // Log location for the debug and logging texts |
| private static string logLocation = "C:\\DriveLog.txt"; |
| private static int deviceRemovalRefreshInterval = 1000; |
| |
| public MainForm() |
| { |
| // Disable thread safe checking. |
| // !! -- NEED TO IMPLEMENT THREAD SAFE UPDATING OF FORM -- !! |
| CheckForIllegalCrossThreadCalls = false; |
| |
| // Clears the list of DriveEntry structs |
| driveList.Clear(); |
| InitializeComponent(); |
| // Initializes a new DriveDetector class used to detect new drive events |
| DriveDetector driveDetector = new DriveDetector(); |
| driveDetector.DeviceArrived += new DriveDetectorEventHandler(driveDetector_DeviceArrived); |
| driveDetector.DeviceRemoved += new DriveDetectorEventHandler(driveDetector_DeviceRemoved); |
| //driveDetector.DeviceRemoved += new DriveDetectorEventHandler(driveDetector_DeviceRemoved); |
| this.listView_Drives.AfterLabelEdit += new LabelEditEventHandler(listView_Drives_AfterLabelEdit); |
| this.listView_Drives.Validating += new CancelEventHandler(listView_Drives_Validating); |
| |
| // KeyPreview events for keyboard shortcuts |
| this.KeyPreview = true; |
| this.KeyPress += new KeyPressEventHandler(MainForm_KeyPress); |
| |
| // Appends new session start text to log file |
| using (StreamWriter sw = File.AppendText(logLocation)) |
| { |
| sw.WriteLine("-- New Session Started --"); |
| sw.WriteLine("Initializing form details"); |
| } |
| |
| // Adds columns to the listview |
| this.listView_Drives.Columns.Add("Owner", 145, HorizontalAlignment.Left); |
| this.listView_Drives.Columns.Add("Time", 130, HorizontalAlignment.Center); |
| this.listView_Drives.Columns.Add("Drive", 40, HorizontalAlignment.Center); |
| 49,15 → 65,96 |
|---|
| this.listView_Drives.Columns.Add("Size", 60, HorizontalAlignment.Center); |
| |
| paintDriveListbox(); |
| |
| // Creates and runs a background worker for detecting if drives have been |
| // removed and refreshing the listview accordingly |
| BackgroundWorker bgWorker = new BackgroundWorker(); |
| bgWorker.DoWork += new DoWorkEventHandler(bgWorker_DoWork); |
| |
| // Starts the background worker thread |
| bgWorker.RunWorkerAsync(); |
| } |
| void bgWorker_DoWork(object sender, DoWorkEventArgs e) |
| { |
| // currentDrives holds a list of 'previous' drives to compare to |
| List<string> currentDrives = new List<string>(); |
| // newDrives holds a list of the most current drives detected |
| DriveInfo[] newDrives = DriveInfo.GetDrives(); |
| // Sets the two lists of drives to hold the same drives |
| foreach (DriveInfo drive in newDrives) |
| { |
| currentDrives.Add(drive.Name); |
| } |
| // Loop that checks for drives that are removed |
| while (true) |
| { |
| // Updates the list of current drives |
| newDrives = DriveInfo.GetDrives(); |
| // If the count of current drives is more than the count of previous drives |
| if (newDrives.Length < currentDrives.Count) |
| { |
| // Goes through each list and finds the drive that was recently removed |
| bool removedDriveFound = false; |
| foreach (string str in currentDrives) |
| { |
| removedDriveFound = false; |
| // Loop here checks for non-matching entries in the two lists |
| foreach (DriveInfo drive in newDrives) |
| { |
| // If entries match, drive was not removed |
| if (str == drive.Name) |
| { |
| removedDriveFound = true; |
| break; |
| } |
| } |
| // If list mismatch is detected, remove from driveList |
| if (removedDriveFound == false) |
| { |
| driveRemoved(str); |
| } |
| } |
| // Clears and refreshes the currentDrives list |
| currentDrives.Clear(); |
| foreach (DriveInfo drive in newDrives) |
| { |
| currentDrives.Add(drive.Name); |
| } |
| } |
| else |
| { |
| // Sets the two lists to hold the same drives |
| currentDrives.Clear(); |
| foreach (DriveInfo drive in newDrives) |
| { |
| currentDrives.Add(drive.Name); |
| } |
| } |
| // Sleeps the thread for a second |
| System.Threading.Thread.Sleep(deviceRemovalRefreshInterval); |
| } |
| } |
| private void driveRemoved(string str) |
| { |
| // Removes the passed drive from driveList |
| foreach (DriveEntry entry in driveList) |
| { |
| if (str == entry.drive) |
| { |
| driveList.Remove(entry); |
| using (StreamWriter sw = File.AppendText(logLocation)) |
| { |
| sw.WriteLine("Drive Removed -- [" + entry.time.ToString() + "]\t" + entry.drive + "\t\"" + entry.label + "\"\t" + entry.size); |
| } |
| break; |
| } |
| } |
| |
| void listView_Drives_Validating(object sender, CancelEventArgs e) |
| { |
| paintDriveListbox(); |
| } |
| |
| void listView_Drives_AfterLabelEdit(object sender, LabelEditEventArgs e) |
| { |
| // Logs the new label if it is changed |
| if (e.Label != null) |
| { |
| using (StreamWriter sw = File.AppendText(logLocation)) |
| 68,7 → 165,6 |
|---|
| } |
| } |
| } |
| |
| void MainForm_KeyPress(object sender, KeyPressEventArgs e) |
| { |
| switch (e.KeyChar) |
| 81,6 → 177,7 |
|---|
| } |
| private void paintDriveListbox() |
| { |
| // Updates the listview |
| this.listView_Drives.BeginUpdate(); |
| |
| this.listView_Drives.Items.Clear(); |
| 88,22 → 185,25 |
|---|
| // Adds each entry from the driveList into the listView |
| foreach (DriveEntry entry in driveList) |
| { |
| ListViewItem item = new ListViewItem(); |
| item.Text = entry.owner; |
| ListViewItem.ListViewSubItem subTime = new ListViewItem.ListViewSubItem(); |
| subTime.Text = entry.time.ToString(); |
| item.SubItems.Add(subTime); |
| ListViewItem.ListViewSubItem subDrive = new ListViewItem.ListViewSubItem(); |
| subDrive.Text = entry.drive; |
| item.SubItems.Add(subDrive); |
| ListViewItem.ListViewSubItem subLabel = new ListViewItem.ListViewSubItem(); |
| subLabel.Text = entry.label; |
| item.SubItems.Add(subLabel); |
| ListViewItem.ListViewSubItem subSize = new ListViewItem.ListViewSubItem(); |
| subSize.Text = entry.size; |
| item.SubItems.Add(subSize); |
| if (entry.owner != "System Reserved") |
| { |
| ListViewItem item = new ListViewItem(); |
| item.Text = entry.owner; |
| ListViewItem.ListViewSubItem subTime = new ListViewItem.ListViewSubItem(); |
| subTime.Text = entry.time.ToString(); |
| item.SubItems.Add(subTime); |
| ListViewItem.ListViewSubItem subDrive = new ListViewItem.ListViewSubItem(); |
| subDrive.Text = entry.drive; |
| item.SubItems.Add(subDrive); |
| ListViewItem.ListViewSubItem subLabel = new ListViewItem.ListViewSubItem(); |
| subLabel.Text = entry.label; |
| item.SubItems.Add(subLabel); |
| ListViewItem.ListViewSubItem subSize = new ListViewItem.ListViewSubItem(); |
| subSize.Text = entry.size; |
| item.SubItems.Add(subSize); |
| |
| this.listView_Drives.Items.Add(item); |
| this.listView_Drives.Items.Add(item); |
| } |
| } |
| |
| this.listView_Drives.EndUpdate(); |
| 110,8 → 210,12 |
|---|
| } |
| void driveDetector_DeviceArrived(object sender, DriveDetectorEventArgs e) |
| { |
| e.HookQueryRemove = true; |
| // Event call for when a new drive is detected by DriveDetector |
| |
| // Disable e.HookQueryRemoved to prevent a hook being attached to the volume |
| //e.HookQueryRemove = true; |
| |
| // Creates and populates a new DriveEntry |
| DriveEntry newEntry = new DriveEntry(); |
| newEntry.time = DateTime.Now; |
| newEntry.drive = e.Drive; |
| 132,14 → 236,19 |
|---|
| newEntry.label = tempDrive.VolumeLabel; |
| newEntry.size = (tempDrive.TotalSize / 1073741824).ToString() + " GB"; |
| |
| // Pops up a dialog asking for a new label unless the partition is a system partition |
| if (newEntry.label != "System Reserved") |
| { |
| LabelPrompt label = new LabelPrompt(); |
| label.ShowDialog(); |
| newEntry.owner = label.driveLabel; |
| driveList.Add(newEntry); |
| } |
| else |
| newEntry.owner = "System Reserved"; |
| |
| // Adds the new DriveEntry into driveList |
| driveList.Add(newEntry); |
| |
| using (StreamWriter sw = File.AppendText(logLocation)) |
| { |
| sw.WriteLine("Drive Attached -- [" + newEntry.time.ToString() + "]\t\"" + newEntry.owner + "\"\t" + newEntry.drive + "\t\"" + newEntry.label + "\"\t" + newEntry.size); |
| 147,26 → 256,26 |
|---|
| |
| paintDriveListbox(); |
| } |
| void driveDetector_DeviceRemoved(object sender, DriveDetectorEventArgs e) |
| { |
| //DriveEntry entryToRemove = new DriveEntry(); |
| foreach (DriveEntry entry in driveList) |
| { |
| if (e.Drive == entry.drive) |
| { |
| //entryToRemove = entry; |
| driveList.Remove(entry); |
| break; |
| } |
| } |
| //driveList.Remove(entryToRemove); |
| //void driveDetector_DeviceRemoved(object sender, DriveDetectorEventArgs e) |
| //{ |
| // //DriveEntry entryToRemove = new DriveEntry(); |
| // foreach (DriveEntry entry in driveList) |
| // { |
| // if (e.Drive == entry.drive) |
| // { |
| // //entryToRemove = entry; |
| // driveList.Remove(entry); |
| // break; |
| // } |
| // } |
| // //driveList.Remove(entryToRemove); |
| |
| using (StreamWriter sw = File.AppendText(logLocation)) |
| { |
| //sw.WriteLine("Drive Removed -- [" + entryToRemove.time.ToString() + "]\t" + entryToRemove.drive + "\t\"" + entryToRemove.label + "\"\t" + entryToRemove.size); |
| } |
| // using (StreamWriter sw = File.AppendText(logLocation)) |
| // { |
| // //sw.WriteLine("Drive Removed -- [" + entryToRemove.time.ToString() + "]\t" + entryToRemove.drive + "\t\"" + entryToRemove.label + "\"\t" + entryToRemove.size); |
| // } |
| |
| paintDriveListbox(); |
| } |
| // paintDriveListbox(); |
| //} |
| } |
| } |