Skip to content

Instantly share code, notes, and snippets.

@mlaily
Last active November 28, 2023 06:05
Show Gist options
  • Star 7 You must be signed in to star a gist
  • Fork 5 You must be signed in to fork a gist
  • Save mlaily/9423f1855bb176d52a327f5874915a97 to your computer and use it in GitHub Desktop.
Save mlaily/9423f1855bb176d52a327f5874915a97 to your computer and use it in GitHub Desktop.
Find which process have a file opened using the Restart Manager API
using System;
using System.Collections.Generic;
using System.Diagnostics.CodeAnalysis;
using System.Linq;
using System.Runtime.InteropServices;
using System.Text;
using System.Threading.Tasks;
// ReSharper disable InconsistentNaming
namespace RestartManager
{
public enum RM_APP_TYPE
{
/// <summary>
/// The application cannot be classified as any other type.
/// An application of this type can only be shut down by a forced shutdown.
/// </summary>
RmUnknownApp = 0,
/// <summary>
/// A Windows application run as a stand-alone process that displays a top-level window.
/// </summary>
RmMainWindow = 1,
/// <summary>
/// A Windows application that does not run as a stand-alone process and does not display a top-level window.
/// </summary>
RmOtherWindow = 2,
/// <summary>
/// The application is a Windows service.
/// </summary>
RmService = 3,
/// <summary>
/// The application is Windows Explorer.
/// </summary>
RmExplorer = 4,
/// <summary>
/// The application is a stand-alone console application.
/// </summary>
RmConsole = 5,
/// <summary>
/// A system restart is required to complete the installation because a process cannot be shut down.
/// The process cannot be shut down because of the following reasons.
/// The process may be a critical process.
/// The current user may not have permission to shut down the process.
/// The process may belong to the primary installer that started the Restart Manager.
/// </summary>
RmCritical = 1000,
}
[Flags]
public enum RM_APP_STATUS
{
/// <summary>
/// The application is in a state that is not described by any other enumerated state.
/// </summary>
RmStatusUnknown = 0x0,
/// <summary>
/// The application is currently running.
/// </summary>
RmStatusRunning = 0x1,
/// <summary>
/// The Restart Manager has stopped the application.
/// </summary>
RmStatusStopped = 0x2,
/// <summary>
/// An action outside the Restart Manager has stopped the application.
/// </summary>
RmStatusStoppedOther = 0x4,
/// <summary>
/// The Restart Manager has restarted the application.
/// </summary>
RmStatusRestarted = 0x8,
/// <summary>
/// The Restart Manager encountered an error when stopping the application.
/// </summary>
RmStatusErrorOnStop = 0x10,
/// <summary>
/// The Restart Manager encountered an error when restarting the application.
/// </summary>
RmStatusErrorOnRestart = 0x20,
/// <summary>
/// Shutdown is masked by a filter.
/// </summary>
RmStatusShutdownMasked = 0x40,
/// <summary>
/// Restart is masked by a filter.
/// </summary>
RmStatusRestartMasked = 0x80,
}
[Flags]
public enum RM_REBOOT_REASON
{
/// <summary>
/// A system restart is not required.
/// </summary>
RmRebootReasonNone = 0x0,
/// <summary>
/// The current user does not have sufficient privileges to shut down one or more processes.
/// </summary>
RmRebootReasonPermissionDenied = 0x1,
/// <summary>
/// One or more processes are running in another Terminal Services session.
/// </summary>
RmRebootReasonSessionMismatch = 0x2,
/// <summary>
/// A system restart is needed because one or more processes to be shut down are critical processes.
/// </summary>
RmRebootReasonCriticalProcess = 0x4,
/// <summary>
/// A system restart is needed because one or more services to be shut down are critical services.
/// </summary>
RmRebootReasonCriticalService = 0x8,
/// <summary>
/// A system restart is needed because the current process must be shut down.
/// </summary>
RmRebootReasonDetectedSelf = 0x10,
}
public enum RmResult
{
/// <summary>
/// The resources specified have been registered.
/// </summary>
ERROR_SUCCESS = 0,
/// <summary>
/// A Restart Manager function could not obtain a Registry write mutex in the allotted time.
/// A system restart is recommended because further use of the Restart Manager is likely to fail.
/// </summary>
ERROR_SEM_TIMEOUT = 121,
/// <summary>
/// One or more arguments are not correct.
/// This error value is returned by Restart Manager function if a NULL pointer or 0 is passed in a parameter that requires a non-null and non-zero value.
/// </summary>
ERROR_BAD_ARGUMENTS = 160,
/// <summary>
/// An operation was unable to read or write to the registry.
/// </summary>
ERROR_WRITE_FAULT = 29,
/// <summary>
/// A Restart Manager operation could not complete because not enough memory was available.
/// </summary>
ERROR_OUTOFMEMORY = 14,
/// <summary>
/// An invalid handle was passed to the function.
/// No Restart Manager session exists for the handle supplied.
/// </summary>
ERROR_INVALID_HANDLE = 6,
/// <summary>
/// The maximum number of sessions has been reached.
/// </summary>
ERROR_MAX_SESSIONS_REACHED = 353,
/// <summary>
/// This error value is returned by the RmGetList function if the rgAffectedApps buffer is too small to hold all application information in the list.
/// </summary>
ERROR_MORE_DATA = 234,
/// <summary>
/// The current operation is canceled by user.
/// </summary>
ERROR_CANCELLED = 1223
}
/// <summary>
/// Uniquely identifies a process by its PID and the time the process began.
/// An array of RM_UNIQUE_PROCESS structures can be passed to the RmRegisterResources function.
/// </summary>
[StructLayout(LayoutKind.Sequential)]
public struct RM_UNIQUE_PROCESS
{
public int dwProcessId;
public System.Runtime.InteropServices.ComTypes.FILETIME ProcessStartTime;
}
/// <summary>
/// Describes an application that is to be registered with the Restart Manager.
/// </summary>
[StructLayout(LayoutKind.Sequential, CharSet = CharSet.Unicode)]
public struct RM_PROCESS_INFO
{
public const int CCH_RM_MAX_APP_NAME = 255;
public const int CCH_RM_MAX_SVC_NAME = 63;
public const int RM_INVALID_SESSION = -1;
/// <summary>
/// Contains an RM_UNIQUE_PROCESS structure that uniquely identifies the application by its PID and the time the process began.
/// </summary>
public RM_UNIQUE_PROCESS Process;
/// <summary>
/// If the process is a service, this parameter returns the long name for the service.
/// If the process is not a service, this parameter returns the user-friendly name for the application.
/// If the process is a critical process, and the installer is run with elevated privileges,
/// this parameter returns the name of the executable file of the critical process.
/// If the process is a critical process, and the installer is run as a service, this parameter returns the long name of the critical process.
/// </summary>
[MarshalAs(UnmanagedType.ByValTStr, SizeConst = CCH_RM_MAX_APP_NAME + 1)]
public string strAppName;
/// <summary>
/// If the process is a service, this is the short name for the service.
/// This member is not used if the process is not a service.
/// </summary>
[MarshalAs(UnmanagedType.ByValTStr, SizeConst = CCH_RM_MAX_SVC_NAME + 1)]
public string strServiceShortName;
/// <summary>
/// Contains an RM_APP_TYPE enumeration value that specifies the type of application as RmUnknownApp, RmMainWindow, RmOtherWindow, RmService, RmExplorer or RmCritical.
/// </summary>
public RM_APP_TYPE ApplicationType;
/// <summary>
/// Contains a bit mask that describes the current status of the application.
/// See the RM_APP_STATUS enumeration.
/// </summary>
public RM_APP_STATUS AppStatus;
/// <summary>
/// Contains the Terminal Services session ID of the process.
/// If the terminal session of the process cannot be determined, the value of this member is set to RM_INVALID_SESSION (-1).
/// This member is not used if the process is a service or a system critical process.
/// </summary>
public int TSSessionId;
/// <summary>
/// TRUE if the application can be restarted by the Restart Manager; otherwise, FALSE.
/// This member is always TRUE if the process is a service.
/// This member is always FALSE if the process is a critical system process.
/// </summary>
[MarshalAs(UnmanagedType.Bool)]
public bool bRestartable;
}
}
using System;
using System.Collections.Generic;
using System.Linq;
using System.Runtime.InteropServices;
using System.Text;
using System.Threading.Tasks;
namespace RestartManager
{
public static class NativeMethods
{
/// <summary>
/// Registers resources to a Restart Manager session.
/// The Restart Manager uses the list of resources registered with the session to determine which applications
/// and services must be shut down and restarted. Resources can be identified by filenames, service short names,
/// or RM_UNIQUE_PROCESS structures that describe running applications.
/// The RmRegisterResources function can be used by a primary or secondary installer.
/// </summary>
/// <param name="dwSessionHandle">A handle to an existing Restart Manager session.</param>
/// <param name="nFiles">The number of files being registered.</param>
/// <param name="rgsFilenames">An array of null-terminated strings of full filename paths. This parameter can be NULL if nFiles is 0.</param>
/// <param name="nApplications">The number of processes being registered.</param>
/// <param name="rgApplications">An array of RM_UNIQUE_PROCESS structures. This parameter can be NULL if nApplications is 0.</param>
/// <param name="nServices">The number of services to be registered.</param>
/// <param name="rgsServiceNames">An array of null-terminated strings of service short names. This parameter can be NULL if nServices is 0.</param>
/// <returns>This is the most recent error received. The function can return one of the system error codes that are defined in Winerror.h.</returns>
[DllImport("rstrtmgr.dll", CharSet = CharSet.Unicode)]
public static extern RmResult RmRegisterResources(
int dwSessionHandle,
uint nFiles,
string[] rgsFilenames,
uint nApplications,
[In] RM_UNIQUE_PROCESS[] rgApplications,
uint nServices,
string[] rgsServiceNames);
/// <summary>
/// Starts a new Restart Manager session.
/// A maximum of 64 Restart Manager sessions per user session can be open on the system at the same time.
/// When this function starts a session, it returns a session handle and session key that can be used in subsequent calls to the Restart Manager API.
/// </summary>
/// <param name="pSessionHandle">
/// A pointer to the handle of a Restart Manager session.
/// The session handle can be passed in subsequent calls to the Restart Manager API.
/// </param>
/// <param name="dwSessionFlags">Reserved. This parameter should be 0.</param>
/// <param name="strSessionKey">
/// A null-terminated string that contains the session key to the new session.
/// The string must be allocated before calling the RmStartSession function.
/// </param>
/// <returns>This is the most recent error received. The function can return one of the system error codes that are defined in Winerror.h.</returns>
[DllImport("rstrtmgr.dll", CharSet = CharSet.Auto)]
public static extern RmResult RmStartSession(out int pSessionHandle, int dwSessionFlags, string strSessionKey);
/// <summary>
/// Ends the Restart Manager session.
/// This function should be called by the primary installer that has previously started the session by calling the RmStartSession function.
/// The RmEndSession function can be called by a secondary installer that is joined to the session once no more resources need to be registered
/// by the secondary installer.
/// </summary>
/// <param name="dwSessionHandle">A handle to an existing Restart Manager session.</param>
/// <returns>This is the most recent error received. The function can return one of the system error codes that are defined in Winerror.h.</returns>
[DllImport("rstrtmgr.dll")]
public static extern RmResult RmEndSession(int dwSessionHandle);
/// <summary>
/// Gets a list of all applications and services that are currently using resources that have been registered with the Restart Manager session.
/// </summary>
/// <param name="dwSessionHandle">A handle to an existing Restart Manager session.</param>
/// <param name="pnProcInfoNeeded">
/// A pointer to an array size necessary to receive RM_PROCESS_INFO structures required to return information for all affected applications and services.
/// </param>
/// <param name="pnProcInfo">A pointer to the total number of RM_PROCESS_INFO structures in an array and number of structures filled.</param>
/// <param name="rgAffectedApps">
/// An array of RM_PROCESS_INFO structures that list the applications and services using resources that have been registered with the session.
/// </param>
/// <param name="lpdwRebootReasons">
/// Pointer to location that receives a value of the RM_REBOOT_REASON enumeration that describes the reason a system restart is needed.
/// </param>
/// <returns>This is the most recent error received. The function can return one of the system error codes that are defined in Winerror.h.</returns>
[DllImport("rstrtmgr.dll")]
public static extern RmResult RmGetList(
int dwSessionHandle,
out uint pnProcInfoNeeded,
ref uint pnProcInfo,
[In, Out] RM_PROCESS_INFO[] rgAffectedApps,
out RM_REBOOT_REASON lpdwRebootReasons);
}
}
public static RM_PROCESS_INFO[] FindLockerProcesses(string path)
{
int handle;
if (NativeMethods.RmStartSession(out handle, 0, strSessionKey: Guid.NewGuid().ToString()) != RmResult.ERROR_SUCCESS)
throw new Exception("Could not begin session. Unable to determine file lockers.");
try
{
string[] resources = { path }; // Just checking on one resource.
if (NativeMethods.RmRegisterResources(handle, (uint)resources.LongLength, resources, 0, null, 0, null) != RmResult.ERROR_SUCCESS)
throw new Exception("Could not register resource.");
// The first try is done expecting at most ten processes to lock the file.
uint arraySize = 10;
RmResult result;
do
{
var array = new RM_PROCESS_INFO[arraySize];
uint arrayCount;
RM_REBOOT_REASON lpdwRebootReasons;
result = NativeMethods.RmGetList(handle, out arrayCount, ref arraySize, array, out lpdwRebootReasons);
if (result == RmResult.ERROR_SUCCESS)
{
// Adjust the array length to fit the actual count.
Array.Resize(ref array, (int)arrayCount);
return array;
}
else if (result == RmResult.ERROR_MORE_DATA)
{
// We need to initialize a bigger array. We only set the size, and do another iteration.
// (the out parameter arrayCount contains the expected value for the next try)
arraySize = arrayCount;
}
else
{
throw new Exception("Could not list processes locking resource. Failed to get size of result.");
}
} while (result != RmResult.ERROR_SUCCESS);
}
finally
{
NativeMethods.RmEndSession(handle);
}
return new RM_PROCESS_INFO[0];
}
@RenniePet
Copy link

You can change the do-while statement into while (true), and remove the final return statement.

@arvobowen
Copy link

arvobowen commented Sep 18, 2019

Just a passing developer's opinion, You can also remove quite a few of the listed includes in the first two files.

@silentdth
Copy link

This wasn't working for me in an elevated visual studio Unit Testing environment, returned a zero handle to RmStartSession. I assume this is because I'm not in a desktop app.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment