Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- using System;
- using System.Collections.Generic;
- using System.Runtime.InteropServices;
- using System.Diagnostics;
- using System.Text;
- using System.Threading;
- namespace FileLockInfo
- {
- public class Win32Processes
- {
- /// <summary>
- /// Return a list of processes that hold on the given file.
- /// </summary>
- public static List<Process> GetProcessesLockingFile(string filePath)
- {
- var procs = new List<Process>();
- var processListSnapshot = Process.GetProcesses();
- foreach (var process in processListSnapshot)
- {
- if (process.Id <= 4) { continue; } // system processes
- var files = GetFilesLockedBy(process);
- if (files.Contains(filePath)) procs.Add(process);
- }
- return procs;
- }
- /// <summary>
- /// Return a list of file locks held by the process.
- /// </summary>
- public static List<string> GetFilesLockedBy(Process process)
- {
- var outp = new List<string>();
- ThreadStart ts = delegate
- {
- try
- {
- outp = UnsafeGetFilesLockedBy(process);
- }
- catch { Ignore(); }
- };
- try
- {
- var t = new Thread(ts);
- t.IsBackground = true;
- t.Start();
- if (!t.Join(250))
- {
- try
- {
- t.Interrupt();
- t.Abort();
- }
- catch { Ignore(); }
- }
- }
- catch { Ignore(); }
- return outp;
- }
- #region Inner Workings
- private static void Ignore() { }
- private static List<string> UnsafeGetFilesLockedBy(Process process)
- {
- try
- {
- var handles = GetHandles(process);
- var files = new List<string>();
- foreach (var handle in handles)
- {
- var file = GetFilePath(handle, process);
- if (file != null) files.Add(file);
- }
- return files;
- }
- catch
- {
- return new List<string>();
- }
- }
- const int CNST_SYSTEM_HANDLE_INFORMATION = 16;
- private static string GetFilePath(Win32API.SYSTEM_HANDLE_INFORMATION systemHandleInformation, Process process)
- {
- var ipProcessHwnd = Win32API.OpenProcess(Win32API.ProcessAccessFlags.All, false, process.Id);
- var objBasic = new Win32API.OBJECT_BASIC_INFORMATION();
- var objObjectType = new Win32API.OBJECT_TYPE_INFORMATION();
- var objObjectName = new Win32API.OBJECT_NAME_INFORMATION();
- var strObjectName = "";
- var nLength = 0;
- IntPtr ipTemp, ipHandle;
- if (!Win32API.DuplicateHandle(ipProcessHwnd, systemHandleInformation.Handle, Win32API.GetCurrentProcess(), out ipHandle, 0, false, Win32API.DUPLICATE_SAME_ACCESS))
- return null;
- IntPtr ipBasic = Marshal.AllocHGlobal(Marshal.SizeOf(objBasic));
- Win32API.NtQueryObject(ipHandle, (int)Win32API.ObjectInformationClass.ObjectBasicInformation, ipBasic, Marshal.SizeOf(objBasic), ref nLength);
- objBasic = (Win32API.OBJECT_BASIC_INFORMATION)Marshal.PtrToStructure(ipBasic, objBasic.GetType());
- Marshal.FreeHGlobal(ipBasic);
- IntPtr ipObjectType = Marshal.AllocHGlobal(objBasic.TypeInformationLength);
- nLength = objBasic.TypeInformationLength;
- // this one never locks...
- while ((uint)(Win32API.NtQueryObject(ipHandle, (int)Win32API.ObjectInformationClass.ObjectTypeInformation, ipObjectType, nLength, ref nLength)) == Win32API.STATUS_INFO_LENGTH_MISMATCH)
- {
- if (nLength == 0)
- {
- Console.WriteLine("nLength returned at zero! ");
- return null;
- }
- Marshal.FreeHGlobal(ipObjectType);
- ipObjectType = Marshal.AllocHGlobal(nLength);
- }
- objObjectType = (Win32API.OBJECT_TYPE_INFORMATION)Marshal.PtrToStructure(ipObjectType, objObjectType.GetType());
- if (Is64Bits())
- {
- ipTemp = new IntPtr(Convert.ToInt64(objObjectType.Name.Buffer.ToString(), 10) >> 32);
- }
- else
- {
- ipTemp = objObjectType.Name.Buffer;
- }
- var strObjectTypeName = Marshal.PtrToStringUni(ipTemp, objObjectType.Name.Length >> 1);
- Marshal.FreeHGlobal(ipObjectType);
- if (strObjectTypeName != "File")
- return null;
- nLength = objBasic.NameInformationLength;
- var ipObjectName = Marshal.AllocHGlobal(nLength);
- // ...this call sometimes hangs. Is a Windows error.
- while ((uint)(Win32API.NtQueryObject(ipHandle, (int)Win32API.ObjectInformationClass.ObjectNameInformation, ipObjectName, nLength, ref nLength)) == Win32API.STATUS_INFO_LENGTH_MISMATCH)
- {
- Marshal.FreeHGlobal(ipObjectName);
- if (nLength == 0)
- {
- Console.WriteLine("nLength returned at zero! " + strObjectTypeName);
- return null;
- }
- ipObjectName = Marshal.AllocHGlobal(nLength);
- }
- objObjectName = (Win32API.OBJECT_NAME_INFORMATION)Marshal.PtrToStructure(ipObjectName, objObjectName.GetType());
- if (Is64Bits())
- {
- ipTemp = new IntPtr(Convert.ToInt64(objObjectName.Name.Buffer.ToString(), 10) >> 32);
- }
- else
- {
- ipTemp = objObjectName.Name.Buffer;
- }
- if (ipTemp != IntPtr.Zero)
- {
- var baTemp = new byte[nLength];
- try
- {
- Marshal.Copy(ipTemp, baTemp, 0, nLength);
- strObjectName = Marshal.PtrToStringUni(Is64Bits() ? new IntPtr(ipTemp.ToInt64()) : new IntPtr(ipTemp.ToInt32()));
- }
- catch (AccessViolationException)
- {
- return null;
- }
- finally
- {
- Marshal.FreeHGlobal(ipObjectName);
- Win32API.CloseHandle(ipHandle);
- }
- }
- string path = GetRegularFileNameFromDevice(strObjectName);
- try
- {
- return path;
- }
- catch
- {
- return null;
- }
- }
- private static string GetRegularFileNameFromDevice(string strRawName)
- {
- string strFileName = strRawName;
- foreach (string strDrivePath in Environment.GetLogicalDrives())
- {
- var sbTargetPath = new StringBuilder(Win32API.MAX_PATH);
- if (Win32API.QueryDosDevice(strDrivePath.Substring(0, 2), sbTargetPath, Win32API.MAX_PATH) == 0)
- {
- return strRawName;
- }
- string strTargetPath = sbTargetPath.ToString();
- if (strFileName.StartsWith(strTargetPath))
- {
- strFileName = strFileName.Replace(strTargetPath, strDrivePath.Substring(0, 2));
- break;
- }
- }
- return strFileName;
- }
- private static IEnumerable<Win32API.SYSTEM_HANDLE_INFORMATION> GetHandles(Process process)
- {
- var nHandleInfoSize = 0x10000;
- var ipHandlePointer = Marshal.AllocHGlobal(nHandleInfoSize);
- var nLength = 0;
- IntPtr ipHandle;
- while (Win32API.NtQuerySystemInformation(CNST_SYSTEM_HANDLE_INFORMATION, ipHandlePointer, nHandleInfoSize, ref nLength) == Win32API.STATUS_INFO_LENGTH_MISMATCH)
- {
- nHandleInfoSize = nLength;
- Marshal.FreeHGlobal(ipHandlePointer);
- ipHandlePointer = Marshal.AllocHGlobal(nLength);
- }
- var baTemp = new byte[nLength];
- Marshal.Copy(ipHandlePointer, baTemp, 0, nLength);
- long lHandleCount;
- if (Is64Bits())
- {
- lHandleCount = Marshal.ReadInt64(ipHandlePointer);
- ipHandle = new IntPtr(ipHandlePointer.ToInt64() + 8);
- }
- else
- {
- lHandleCount = Marshal.ReadInt32(ipHandlePointer);
- ipHandle = new IntPtr(ipHandlePointer.ToInt32() + 4);
- }
- var lstHandles = new List<Win32API.SYSTEM_HANDLE_INFORMATION>();
- for (long lIndex = 0; lIndex < lHandleCount; lIndex++)
- {
- var shHandle = new Win32API.SYSTEM_HANDLE_INFORMATION();
- if (Is64Bits())
- {
- shHandle = (Win32API.SYSTEM_HANDLE_INFORMATION)Marshal.PtrToStructure(ipHandle, shHandle.GetType());
- ipHandle = new IntPtr(ipHandle.ToInt64() + Marshal.SizeOf(shHandle) + 8);
- }
- else
- {
- ipHandle = new IntPtr(ipHandle.ToInt64() + Marshal.SizeOf(shHandle));
- shHandle = (Win32API.SYSTEM_HANDLE_INFORMATION)Marshal.PtrToStructure(ipHandle, shHandle.GetType());
- }
- if (shHandle.ProcessID != process.Id) continue;
- lstHandles.Add(shHandle);
- }
- return lstHandles;
- }
- private static bool Is64Bits()
- {
- return Marshal.SizeOf(typeof(IntPtr)) == 8;
- }
- internal class Win32API
- {
- [DllImport("ntdll.dll")]
- public static extern int NtQueryObject(IntPtr ObjectHandle, int
- ObjectInformationClass, IntPtr ObjectInformation, int ObjectInformationLength,
- ref int returnLength);
- [DllImport("kernel32.dll", SetLastError = true)]
- public static extern uint QueryDosDevice(string lpDeviceName, StringBuilder lpTargetPath, int ucchMax);
- [DllImport("ntdll.dll")]
- public static extern uint NtQuerySystemInformation(int
- SystemInformationClass, IntPtr SystemInformation, int SystemInformationLength,
- ref int returnLength);
- [DllImport("kernel32.dll")]
- public static extern IntPtr OpenProcess(ProcessAccessFlags dwDesiredAccess, [MarshalAs(UnmanagedType.Bool)] bool bInheritHandle, int dwProcessId);
- [DllImport("kernel32.dll")]
- public static extern int CloseHandle(IntPtr hObject);
- [DllImport("kernel32.dll", SetLastError = true)]
- [return: MarshalAs(UnmanagedType.Bool)]
- public static extern bool DuplicateHandle(IntPtr hSourceProcessHandle,
- ushort hSourceHandle, IntPtr hTargetProcessHandle, out IntPtr lpTargetHandle,
- uint dwDesiredAccess, [MarshalAs(UnmanagedType.Bool)] bool bInheritHandle, uint dwOptions);
- [DllImport("kernel32.dll")]
- public static extern IntPtr GetCurrentProcess();
- public enum ObjectInformationClass
- {
- ObjectBasicInformation = 0,
- ObjectNameInformation = 1,
- ObjectTypeInformation = 2,
- ObjectAllTypesInformation = 3,
- ObjectHandleInformation = 4
- }
- [Flags]
- public enum ProcessAccessFlags : uint
- {
- All = 0x001F0FFF,
- Terminate = 0x00000001,
- CreateThread = 0x00000002,
- VMOperation = 0x00000008,
- VMRead = 0x00000010,
- VMWrite = 0x00000020,
- DupHandle = 0x00000040,
- SetInformation = 0x00000200,
- QueryInformation = 0x00000400,
- Synchronize = 0x00100000
- }
- [StructLayout(LayoutKind.Sequential)]
- public struct OBJECT_BASIC_INFORMATION
- { // Information Class 0
- public int Attributes;
- public int GrantedAccess;
- public int HandleCount;
- public int PointerCount;
- public int PagedPoolUsage;
- public int NonPagedPoolUsage;
- public int Reserved1;
- public int Reserved2;
- public int Reserved3;
- public int NameInformationLength;
- public int TypeInformationLength;
- public int SecurityDescriptorLength;
- public System.Runtime.InteropServices.ComTypes.FILETIME CreateTime;
- }
- [StructLayout(LayoutKind.Sequential)]
- public struct OBJECT_TYPE_INFORMATION
- { // Information Class 2
- public UNICODE_STRING Name;
- public int ObjectCount;
- public int HandleCount;
- public int Reserved1;
- public int Reserved2;
- public int Reserved3;
- public int Reserved4;
- public int PeakObjectCount;
- public int PeakHandleCount;
- public int Reserved5;
- public int Reserved6;
- public int Reserved7;
- public int Reserved8;
- public int InvalidAttributes;
- public GENERIC_MAPPING GenericMapping;
- public int ValidAccess;
- public byte Unknown;
- public byte MaintainHandleDatabase;
- public int PoolType;
- public int PagedPoolUsage;
- public int NonPagedPoolUsage;
- }
- [StructLayout(LayoutKind.Sequential)]
- public struct OBJECT_NAME_INFORMATION
- { // Information Class 1
- public UNICODE_STRING Name;
- }
- [StructLayout(LayoutKind.Sequential, Pack = 1)]
- public struct UNICODE_STRING
- {
- public ushort Length;
- public ushort MaximumLength;
- public IntPtr Buffer;
- }
- [StructLayout(LayoutKind.Sequential)]
- public struct GENERIC_MAPPING
- {
- public int GenericRead;
- public int GenericWrite;
- public int GenericExecute;
- public int GenericAll;
- }
- [StructLayout(LayoutKind.Sequential, Pack = 1)]
- public struct SYSTEM_HANDLE_INFORMATION
- { // Information Class 16
- public int ProcessID;
- public byte ObjectTypeNumber;
- public byte Flags; // 0x01 = PROTECT_FROM_CLOSE, 0x02 = INHERIT
- public ushort Handle;
- public int Object_Pointer;
- public UInt32 GrantedAccess;
- }
- public const int MAX_PATH = 260;
- public const uint STATUS_INFO_LENGTH_MISMATCH = 0xC0000004;
- public const int DUPLICATE_SAME_ACCESS = 0x2;
- public const uint FILE_SEQUENTIAL_ONLY = 0x00000004;
- }
- #endregion
- }
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement