using System; using System.Runtime.InteropServices; namespace Wox.Plugin.FindFile.MFTSearch { public class PInvokeWin32 { public const UInt32 GENERIC_READ = 0x80000000; public const UInt32 GENERIC_WRITE = 0x40000000; public const UInt32 FILE_SHARE_READ = 0x00000001; public const UInt32 FILE_SHARE_WRITE = 0x00000002; public const UInt32 FILE_ATTRIBUTE_DIRECTORY = 0x00000010; public const UInt32 OPEN_EXISTING = 3; public const UInt32 FILE_FLAG_BACKUP_SEMANTICS = 0x02000000; public const Int32 INVALID_HANDLE_VALUE = -1; public const UInt32 FSCTL_QUERY_USN_JOURNAL = 0x000900f4; public const UInt32 FSCTL_ENUM_USN_DATA = 0x000900b3; public const UInt32 FSCTL_CREATE_USN_JOURNAL = 0x000900e7; public const UInt32 FSCTL_READ_USN_JOURNAL = 0x000900bb; [DllImport("kernel32.dll", SetLastError = true)] public static extern IntPtr CreateFile(string lpFileName, uint dwDesiredAccess, uint dwShareMode, IntPtr lpSecurityAttributes, uint dwCreationDisposition, uint dwFlagsAndAttributes, IntPtr hTemplateFile); [DllImport("kernel32.dll", SetLastError = true)] [return: MarshalAs(UnmanagedType.Bool)] public static extern bool GetFileInformationByHandle(IntPtr hFile, out BY_HANDLE_FILE_INFORMATION lpFileInformation); [DllImport("kernel32.dll", SetLastError = true)] [return: MarshalAs(UnmanagedType.Bool)] public static extern bool CloseHandle(IntPtr hObject); [DllImport("kernel32.dll", ExactSpelling = true, SetLastError = true, CharSet = CharSet.Auto)] [return: MarshalAs(UnmanagedType.Bool)] public static extern bool DeviceIoControl(IntPtr hDevice, UInt32 dwIoControlCode, IntPtr lpInBuffer, Int32 nInBufferSize, out USN_JOURNAL_DATA lpOutBuffer, Int32 nOutBufferSize, out uint lpBytesReturned, IntPtr lpOverlapped); [DllImport("kernel32.dll", ExactSpelling = true, SetLastError = true, CharSet = CharSet.Auto)] [return: MarshalAs(UnmanagedType.Bool)] public static extern bool DeviceIoControl(IntPtr hDevice, UInt32 dwIoControlCode, IntPtr lpInBuffer, Int32 nInBufferSize, IntPtr lpOutBuffer, Int32 nOutBufferSize, out uint lpBytesReturned, IntPtr lpOverlapped); [DllImport("kernel32.dll")] public static extern void ZeroMemory(IntPtr ptr, Int32 size); [StructLayout(LayoutKind.Sequential, Pack = 1)] public struct BY_HANDLE_FILE_INFORMATION { public uint FileAttributes; public FILETIME CreationTime; public FILETIME LastAccessTime; public FILETIME LastWriteTime; public uint VolumeSerialNumber; public uint FileSizeHigh; public uint FileSizeLow; public uint NumberOfLinks; public uint FileIndexHigh; public uint FileIndexLow; } [StructLayout(LayoutKind.Sequential, Pack = 1)] public struct FILETIME { public uint DateTimeLow; public uint DateTimeHigh; } [StructLayout(LayoutKind.Sequential, Pack = 1)] public struct USN_JOURNAL_DATA { public UInt64 UsnJournalID; public Int64 FirstUsn; public Int64 NextUsn; public Int64 LowestValidUsn; public Int64 MaxUsn; public UInt64 MaximumSize; public UInt64 AllocationDelta; } [StructLayout(LayoutKind.Sequential, Pack = 1)] public struct MFT_ENUM_DATA { public UInt64 StartFileReferenceNumber; public Int64 LowUsn; public Int64 HighUsn; } [StructLayout(LayoutKind.Sequential, Pack = 1)] public struct CREATE_USN_JOURNAL_DATA { public UInt64 MaximumSize; public UInt64 AllocationDelta; } [StructLayout(LayoutKind.Sequential, Pack = 1)] public struct READ_USN_JOURNAL_DATA { public Int64 StartUsn; public UInt32 ReasonMask; public UInt32 ReturnOnlyOnClose; public UInt64 Timeout; public UInt64 BytesToWaitFor; public UInt64 UsnJournalID; } public class USN_RECORD { public UInt32 RecordLength; public UInt16 MajorVersion; public UInt16 MinorVersion; public UInt64 FRN; // 8 public UInt64 ParentFRN; // 16 public Int64 Usn; // Need be care public UInt64 TimeStamp; // Need Be care public UInt32 Reason; public UInt32 SourceInfo; public UInt32 SecurityId; public UInt32 FileAttributes; // 52 public UInt16 FileNameLength; public UInt16 FileNameOffset; public string FileName = string.Empty; private const int RecordLength_OFFSET = 0; private const int MajorVersion_OFFSET = 4; private const int MinorVersion_OFFSET = 6; private const int FileReferenceNumber_OFFSET = 8; private const int ParentFileReferenceNumber_OFFSET = 16; private const int Usn_OFFSET = 24; private const int TimeStamp_OFFSET = 32; private const int Reason_OFFSET = 40; private const int SourceInfo_OFFSET = 44; private const int SecurityId_OFFSET = 48; private const int FileAttributes_OFFSET = 52; private const int FileNameLength_OFFSET = 56; private const int FileNameOffset_OFFSET = 58; private const int FileName_OFFSET = 60; public USN_RECORD(IntPtr p) { this.RecordLength = (UInt32)Marshal.ReadInt32(p, RecordLength_OFFSET); this.MajorVersion = (UInt16)Marshal.ReadInt16(p, MajorVersion_OFFSET); this.MinorVersion = (UInt16)Marshal.ReadInt16(p, MinorVersion_OFFSET); this.FRN = (UInt64)Marshal.ReadInt64(p, FileReferenceNumber_OFFSET); this.ParentFRN = (UInt64)Marshal.ReadInt64(p, ParentFileReferenceNumber_OFFSET); this.Usn = Marshal.ReadInt64(p, Usn_OFFSET); this.TimeStamp = (UInt64)Marshal.ReadInt64(p, TimeStamp_OFFSET); this.Reason = (UInt32)Marshal.ReadInt32(p, Reason_OFFSET); this.SourceInfo = (UInt32)Marshal.ReadInt32(p, SourceInfo_OFFSET); this.SecurityId = (UInt32)Marshal.ReadInt32(p, SecurityId_OFFSET); this.FileAttributes = (UInt32)Marshal.ReadInt32(p, FileAttributes_OFFSET); this.FileNameLength = (UInt16)Marshal.ReadInt16(p, FileNameLength_OFFSET); this.FileNameOffset = (UInt16)Marshal.ReadInt16(p, FileNameOffset_OFFSET); this.FileName = Marshal.PtrToStringUni(new IntPtr(p.ToInt32() + this.FileNameOffset), this.FileNameLength / sizeof(char)); } public bool IsFolder { get { return 0 != (FileAttributes & PInvokeWin32.FILE_ATTRIBUTE_DIRECTORY); } } } } }