Mirrors/PowerToys
1
0
Fork 0
mirror of https://github.com/microsoft/PowerToys.git synced 2026-04-06 19:26:39 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add search monitor

Browse Source
This commit is contained in:
qianlifeng
2014-11-03 18:49:18 +08:00
parent 9481d04a19
commit 3c4a9a05a0
4 changed files with 42 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
Plugins/Wox.Plugin.FindFile/MFTSearch/MFTSearcher.cs
View File

@@ -15,11 +15,13 @@ namespace Wox.Plugin.FindFile.MFTSearch
public class MFTSearcher
{
private static MFTSearcherCache cache = new MFTSearcherCache();
private static VolumeMonitor monitor = new VolumeMonitor();
private static void IndexVolume(string volume)
{
cache.CheckHashTableKey(volume);
EnumerateVolume(volume,cache.VolumeRecords[volume]);
cache.EnsureVolumeExistInHashTable(volume);
EnumerateVolume(volume, cache.VolumeRecords[volume]);
monitor.Monitor(volume,cache);
}
public static void IndexAllVolumes()
@@ -39,7 +41,7 @@ namespace Wox.Plugin.FindFile.MFTSearch
{
if (string.IsNullOrEmpty(item)) return new List<MFTSearchRecord>();
List<USNRecord> found = cache.FindByName(item,100);
List<USNRecord> found = cache.FindByName(item, 100);
found.ForEach(x => FillPath(x.VolumeName, x, cache));
return found.ConvertAll(o => new MFTSearchRecord(o));
}
@@ -65,7 +67,7 @@ namespace Wox.Plugin.FindFile.MFTSearch
UInt64 fileIndexHigh = (UInt64)fi.FileIndexHigh;
UInt64 indexRoot = (fileIndexHigh << 32) | fi.FileIndexLow;
files.Add(indexRoot,new USNRecord
files.Add(indexRoot, new USNRecord
{
FRN = indexRoot,
Name = volumeName,
@@ -94,7 +96,7 @@ namespace Wox.Plugin.FindFile.MFTSearch
IntPtr pVolume = IntPtr.Zero;
try
{
AddVolumeRootRecord(volumeName,files);
AddVolumeRootRecord(volumeName, files);
pVolume = GetVolumeJournalHandle(volumeName);
EnableVomuleJournal(pVolume);
@@ -217,7 +219,7 @@ namespace Wox.Plugin.FindFile.MFTSearch
{
PInvokeWin32.USN_RECORD usn = new PInvokeWin32.USN_RECORD(pUsnRecord);
files.Add(usn.FRN,new USNRecord
files.Add(usn.FRN, new USNRecord
{
Name = usn.FileName,
ParentFrn = usn.ParentFRN,

11
Plugins/Wox.Plugin.FindFile/MFTSearch/MFTSearcherCache.cs
View File

@@ -17,17 +17,20 @@ namespace Wox.Plugin.FindFile.MFTSearch
public void AddRecord(string volume, List<USNRecord> r)
{
CheckHashTableKey(volume);
EnsureVolumeExistInHashTable(volume);
r.ForEach(x => VolumeRecords[volume].Add(x.FRN, x));
}
public void AddRecord(string volume, USNRecord record)
{
CheckHashTableKey(volume);
VolumeRecords[volume].Add(record.FRN, record);
EnsureVolumeExistInHashTable(volume);
if (!VolumeRecords[volume].ContainsKey(record.FRN))
{
VolumeRecords[volume].Add(record.FRN, record);
}
}
public void CheckHashTableKey(string volume)
public void EnsureVolumeExistInHashTable(string volume)
{
if (!VolumeRecords.ContainsKey(volume))
VolumeRecords.Add(volume, new Dictionary<ulong, USNRecord>());

59
Plugins/Wox.Plugin.FindFile/MFTSearch/VolumeMonitor.cs
View File

@@ -16,12 +16,18 @@ namespace Wox.Plugin.FindFile.MFTSearch
{
foreach (var volume in volumes)
{
if (string.IsNullOrEmpty(volume)) throw new InvalidOperationException("Volume cant't be null or empty string.");
if (!db.ContainsVolume(volume)) throw new InvalidOperationException(string.Format("Volume {0} must be scaned first."));
Thread th = new Thread(new ParameterizedThreadStart(MonitorThread));
th.Start(new Dictionary<string, object> { { "Volume", volume }, { "MFTSearcherCache", db } });
Monitor(volume, db);
}
}
public void Monitor(string volume, MFTSearcherCache db)
{
if (string.IsNullOrEmpty(volume)) throw new InvalidOperationException("Volume cant't be null or empty string.");
if (!db.ContainsVolume(volume)) throw new InvalidOperationException(string.Format("Volume {0} must be scaned first."));
ThreadPool.QueueUserWorkItem(o => MonitorThread(volume, db));
}
private PInvokeWin32.READ_USN_JOURNAL_DATA SetupInputData4JournalRead(string volume, uint reason)
{
IntPtr pMonitorVolume = MFTSearcher.GetVolumeJournalHandle(volume);
@@ -40,11 +46,9 @@ namespace Wox.Plugin.FindFile.MFTSearch
return rujd;
}
private void MonitorThread(object param)
{
MFTSearcherCache db = (param as Dictionary<string, object>)["MFTSearcherCache"] as MFTSearcherCache;
string volume = (param as Dictionary<string, object>)["Volume"] as string;
private void MonitorThread(string volume, MFTSearcherCache db)
{
IntPtr pbuffer = Marshal.AllocHGlobal(0x1000);
PInvokeWin32.READ_USN_JOURNAL_DATA rujd = SetupInputData4JournalRead(volume, 0xFFFFFFFF);
UInt32 cbRead;
@@ -56,7 +60,6 @@ namespace Wox.Plugin.FindFile.MFTSearch
PInvokeWin32.ZeroMemory(prujd, Marshal.SizeOf(rujd));
Marshal.StructureToPtr(rujd, prujd, true);
Debug.WriteLine(string.Format("\nMoniting on {0}......", volume));
IntPtr pVolume = MFTSearcher.GetVolumeJournalHandle(volume);
bool fok = PInvokeWin32.DeviceIoControl(pVolume,
@@ -81,15 +84,9 @@ namespace Wox.Plugin.FindFile.MFTSearch
rujd.StartUsn = Marshal.ReadInt64(pbuffer);
}
}
private void ProcessUSN(PInvokeWin32.USN_RECORD usn, string volume, MFTSearcherCache db)
{
var dbCached = db.FindByFrn(volume, usn.FRN);
MFTSearcher.FillPath(volume, dbCached, db);
Debug.WriteLine(string.Format("------USN[frn={0}]------", usn.FRN));
Debug.WriteLine(string.Format("FileName={0}, USNChangeReason={1}", usn.FileName, USNChangeReason.ReasonPrettyFormat(usn.Reason)));
Debug.WriteLine(string.Format("FileName[Cached]={0}", dbCached == null ? "NoCache" : dbCached.FullPath));
Debug.WriteLine("--------------------------------------");
if (MaskEqual(usn.Reason, USNChangeReason.USN_REASONS["USN_REASON_RENAME_NEW_NAME"]))
ProcessRenameNewName(usn, volume, db);
if ((usn.Reason & USNChangeReason.USN_REASONS["USN_REASON_FILE_CREATE"]) != 0)
@@ -97,47 +94,37 @@ namespace Wox.Plugin.FindFile.MFTSearch
if (MaskEqual(usn.Reason, USNChangeReason.USN_REASONS["USN_REASON_FILE_DELETE"]))
ProcessFileDelete(usn, volume, db);
}
private void ProcessFileDelete(PInvokeWin32.USN_RECORD usn, string volume, MFTSearcherCache db)
{
var cached = db.FindByFrn(volume, usn.FRN);
if (cached == null)
{
return;
}
else
if (cached != null)
{
MFTSearcher.FillPath(volume, cached, db);
var deleteok = db.DeleteRecord(volume, usn.FRN);
Debug.WriteLine(string.Format(">>>> File {0} deleted {1}.", cached.FullPath, deleteok ? "successful" : "fail"));
Debug.WriteLine(string.Format(">>>> DeleteFIle {0} {1}.", cached.FullPath, deleteok ? "successful" : "fail"));
if (RecordDeletedEvent != null)
RecordDeletedEvent(cached);
}
}
private void ProcessRenameNewName(PInvokeWin32.USN_RECORD usn, string volume, MFTSearcherCache db)
{
USNRecord newRecord = USNRecord.ParseUSN(volume, usn);
//string fullpath = newRecord.Name;
//db.FindRecordPath(newRecord, ref fullpath, db.GetVolumeRecords(volume));
//newRecord.FullPath = fullpath;
var oldRecord = db.FindByFrn(volume, usn.FRN);
MFTSearcher.FillPath(volume, oldRecord, db);
MFTSearcher.FillPath(volume, newRecord, db);
Debug.WriteLine(string.Format(">>>> RenameFile {0} to {1}", oldRecord.FullPath, newRecord.FullPath));
db.UpdateRecord(volume, newRecord);
if (RecordRenameEvent != null) RecordRenameEvent(oldRecord, newRecord);
if (newRecord.FullPath.Contains("$RECYCLE.BIN"))
var oldRecord = db.FindByFrn(volume, usn.FRN);
if (oldRecord != null)
{
Debug.WriteLine(string.Format(">>>> Means {0} moved to recycle.", oldRecord.FullPath));
Debug.WriteLine(string.Format(">>>> RenameFile {0} to {1}", oldRecord.FullPath, newRecord.FullPath));
if (RecordRenameEvent != null) RecordRenameEvent(oldRecord, newRecord);
}
}
private void ProcessFileCreate(PInvokeWin32.USN_RECORD usn, string volume, MFTSearcherCache db)
{
USNRecord record = USNRecord.ParseUSN(volume, usn);
//string fullpath = record.Name;
//db.FindRecordPath(record, ref fullpath, db.GetVolumeRecords(volume));
//record.FullPath = fullpath;
db.AddRecord(volume, record);
MFTSearcher.FillPath(volume, record, db);
Debug.WriteLine(string.Format(">>>> NewFile: {0}", record.FullPath));
if (RecordAddedEvent != null)
RecordAddedEvent(record);