mirror of
https://github.com/openhardwaremonitor/openhardwaremonitor
synced 2025-08-22 09:57:20 +00:00
297 lines
10 KiB
C#
297 lines
10 KiB
C#
/*
|
|
|
|
This Source Code Form is subject to the terms of the Mozilla Public
|
|
License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
Copyright (C) 2010-2012 Michael Möller <mmoeller@openhardwaremonitor.org>
|
|
|
|
*/
|
|
|
|
using System;
|
|
using System.IO;
|
|
using System.Runtime.InteropServices;
|
|
using System.Security.AccessControl;
|
|
using Microsoft.Win32.SafeHandles;
|
|
|
|
namespace OpenHardwareMonitor.Hardware {
|
|
internal class KernelDriver {
|
|
|
|
private string id;
|
|
|
|
private SafeFileHandle device;
|
|
|
|
public KernelDriver(string id) {
|
|
this.id = id;
|
|
}
|
|
|
|
public bool Install(string path, out string errorMessage) {
|
|
IntPtr manager = NativeMethods.OpenSCManager(null, null,
|
|
ServiceControlManagerAccessRights.SC_MANAGER_ALL_ACCESS);
|
|
|
|
if (manager == IntPtr.Zero) {
|
|
errorMessage = "OpenSCManager returned zero.";
|
|
return false;
|
|
}
|
|
|
|
IntPtr service = NativeMethods.CreateService(manager, id, id,
|
|
ServiceAccessRights.SERVICE_ALL_ACCESS,
|
|
ServiceType.SERVICE_KERNEL_DRIVER, StartType.SERVICE_DEMAND_START,
|
|
ErrorControl.SERVICE_ERROR_NORMAL, path, null, null, null, null,
|
|
null);
|
|
|
|
if (service == IntPtr.Zero) {
|
|
if (Marshal.GetHRForLastWin32Error() == ERROR_SERVICE_EXISTS)
|
|
service = NativeMethods.OpenService(manager, id,
|
|
ServiceAccessRights.SERVICE_ALL_ACCESS);
|
|
else {
|
|
errorMessage = "CreateService returned the error: " +
|
|
Marshal.GetExceptionForHR(Marshal.GetHRForLastWin32Error()).Message;
|
|
NativeMethods.CloseServiceHandle(manager);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
if (!NativeMethods.StartService(service, 0, null)) {
|
|
if (Marshal.GetHRForLastWin32Error() != ERROR_SERVICE_ALREADY_RUNNING) {
|
|
errorMessage = "StartService returned the error: " +
|
|
Marshal.GetExceptionForHR(Marshal.GetHRForLastWin32Error()).Message;
|
|
NativeMethods.CloseServiceHandle(service);
|
|
NativeMethods.CloseServiceHandle(manager);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
NativeMethods.CloseServiceHandle(service);
|
|
NativeMethods.CloseServiceHandle(manager);
|
|
|
|
try {
|
|
// restrict the driver access to system (SY) and builtin admins (BA)
|
|
// TODO: replace with a call to IoCreateDeviceSecure in the driver
|
|
FileSecurity fileSecurity = File.GetAccessControl(@"\\.\" + id);
|
|
fileSecurity.SetSecurityDescriptorSddlForm(
|
|
"O:BAG:SYD:(A;;FA;;;SY)(A;;FA;;;BA)");
|
|
File.SetAccessControl(@"\\.\" + id, fileSecurity);
|
|
} catch { }
|
|
|
|
errorMessage = null;
|
|
return true;
|
|
}
|
|
|
|
public bool Open() {
|
|
device = new SafeFileHandle(NativeMethods.CreateFile(@"\\.\" + id,
|
|
FileAccess.GENERIC_READ | FileAccess.GENERIC_WRITE, 0, IntPtr.Zero,
|
|
CreationDisposition.OPEN_EXISTING, FileAttributes.FILE_ATTRIBUTE_NORMAL,
|
|
IntPtr.Zero), true);
|
|
|
|
if (device.IsInvalid) {
|
|
device.Close();
|
|
device.Dispose();
|
|
device = null;
|
|
}
|
|
|
|
return device != null;
|
|
}
|
|
|
|
public bool IsOpen {
|
|
get { return device != null; }
|
|
}
|
|
|
|
public bool DeviceIOControl(IOControlCode ioControlCode, object inBuffer) {
|
|
if (device == null)
|
|
return false;
|
|
|
|
uint bytesReturned;
|
|
bool b = NativeMethods.DeviceIoControl(device, ioControlCode,
|
|
inBuffer, inBuffer == null ? 0 : (uint)Marshal.SizeOf(inBuffer),
|
|
null, 0, out bytesReturned, IntPtr.Zero);
|
|
return b;
|
|
}
|
|
|
|
public bool DeviceIOControl<T>(IOControlCode ioControlCode, object inBuffer,
|
|
ref T outBuffer)
|
|
{
|
|
if (device == null)
|
|
return false;
|
|
|
|
object boxedOutBuffer = outBuffer;
|
|
uint bytesReturned;
|
|
bool b = NativeMethods.DeviceIoControl(device, ioControlCode,
|
|
inBuffer, inBuffer == null ? 0 : (uint)Marshal.SizeOf(inBuffer),
|
|
boxedOutBuffer, (uint)Marshal.SizeOf(boxedOutBuffer),
|
|
out bytesReturned, IntPtr.Zero);
|
|
outBuffer = (T)boxedOutBuffer;
|
|
return b;
|
|
}
|
|
|
|
public void Close() {
|
|
if (device != null) {
|
|
device.Close();
|
|
device.Dispose();
|
|
device = null;
|
|
}
|
|
}
|
|
|
|
public bool Delete() {
|
|
IntPtr manager = NativeMethods.OpenSCManager(null, null,
|
|
ServiceControlManagerAccessRights.SC_MANAGER_ALL_ACCESS);
|
|
|
|
if (manager == IntPtr.Zero)
|
|
return false;
|
|
|
|
IntPtr service = NativeMethods.OpenService(manager, id,
|
|
ServiceAccessRights.SERVICE_ALL_ACCESS);
|
|
|
|
if (service == IntPtr.Zero)
|
|
return true;
|
|
|
|
ServiceStatus status = new ServiceStatus();
|
|
NativeMethods.ControlService(service, ServiceControl.SERVICE_CONTROL_STOP,
|
|
ref status);
|
|
|
|
NativeMethods.DeleteService(service);
|
|
|
|
NativeMethods.CloseServiceHandle(service);
|
|
NativeMethods.CloseServiceHandle(manager);
|
|
|
|
return true;
|
|
}
|
|
|
|
private enum ServiceAccessRights : uint {
|
|
SERVICE_ALL_ACCESS = 0xF01FF
|
|
}
|
|
|
|
private enum ServiceControlManagerAccessRights : uint {
|
|
SC_MANAGER_ALL_ACCESS = 0xF003F
|
|
}
|
|
|
|
private enum ServiceType : uint {
|
|
SERVICE_KERNEL_DRIVER = 1,
|
|
SERVICE_FILE_SYSTEM_DRIVER = 2
|
|
}
|
|
|
|
private enum StartType : uint {
|
|
SERVICE_BOOT_START = 0,
|
|
SERVICE_SYSTEM_START = 1,
|
|
SERVICE_AUTO_START = 2,
|
|
SERVICE_DEMAND_START = 3,
|
|
SERVICE_DISABLED = 4
|
|
}
|
|
|
|
private enum ErrorControl : uint {
|
|
SERVICE_ERROR_IGNORE = 0,
|
|
SERVICE_ERROR_NORMAL = 1,
|
|
SERVICE_ERROR_SEVERE = 2,
|
|
SERVICE_ERROR_CRITICAL = 3
|
|
}
|
|
|
|
private enum ServiceControl : uint {
|
|
SERVICE_CONTROL_STOP = 1,
|
|
SERVICE_CONTROL_PAUSE = 2,
|
|
SERVICE_CONTROL_CONTINUE = 3,
|
|
SERVICE_CONTROL_INTERROGATE = 4,
|
|
SERVICE_CONTROL_SHUTDOWN = 5,
|
|
SERVICE_CONTROL_PARAMCHANGE = 6,
|
|
SERVICE_CONTROL_NETBINDADD = 7,
|
|
SERVICE_CONTROL_NETBINDREMOVE = 8,
|
|
SERVICE_CONTROL_NETBINDENABLE = 9,
|
|
SERVICE_CONTROL_NETBINDDISABLE = 10,
|
|
SERVICE_CONTROL_DEVICEEVENT = 11,
|
|
SERVICE_CONTROL_HARDWAREPROFILECHANGE = 12,
|
|
SERVICE_CONTROL_POWEREVENT = 13,
|
|
SERVICE_CONTROL_SESSIONCHANGE = 14
|
|
}
|
|
|
|
[StructLayout(LayoutKind.Sequential, Pack = 1)]
|
|
private struct ServiceStatus {
|
|
public uint dwServiceType;
|
|
public uint dwCurrentState;
|
|
public uint dwControlsAccepted;
|
|
public uint dwWin32ExitCode;
|
|
public uint dwServiceSpecificExitCode;
|
|
public uint dwCheckPoint;
|
|
public uint dwWaitHint;
|
|
}
|
|
|
|
private enum FileAccess : uint {
|
|
GENERIC_READ = 0x80000000,
|
|
GENERIC_WRITE = 0x40000000
|
|
}
|
|
|
|
private enum CreationDisposition : uint {
|
|
CREATE_NEW = 1,
|
|
CREATE_ALWAYS = 2,
|
|
OPEN_EXISTING = 3,
|
|
OPEN_ALWAYS = 4,
|
|
TRUNCATE_EXISTING = 5
|
|
}
|
|
|
|
private enum FileAttributes : uint {
|
|
FILE_ATTRIBUTE_NORMAL = 0x80
|
|
}
|
|
|
|
private const int
|
|
ERROR_SERVICE_EXISTS = unchecked((int)0x80070431),
|
|
ERROR_SERVICE_ALREADY_RUNNING = unchecked((int)0x80070420);
|
|
|
|
private static class NativeMethods {
|
|
private const string ADVAPI = "advapi32.dll";
|
|
private const string KERNEL = "kernel32.dll";
|
|
|
|
[DllImport(ADVAPI, CallingConvention = CallingConvention.Winapi)]
|
|
public static extern IntPtr OpenSCManager(string machineName,
|
|
string databaseName, ServiceControlManagerAccessRights dwAccess);
|
|
|
|
[DllImport(ADVAPI, CallingConvention = CallingConvention.Winapi)]
|
|
[return: MarshalAs(UnmanagedType.Bool)]
|
|
public static extern bool CloseServiceHandle(IntPtr hSCObject);
|
|
|
|
[DllImport(ADVAPI, CallingConvention = CallingConvention.Winapi,
|
|
SetLastError = true)]
|
|
public static extern IntPtr CreateService(IntPtr hSCManager,
|
|
string lpServiceName, string lpDisplayName,
|
|
ServiceAccessRights dwDesiredAccess, ServiceType dwServiceType,
|
|
StartType dwStartType, ErrorControl dwErrorControl,
|
|
string lpBinaryPathName, string lpLoadOrderGroup, string lpdwTagId,
|
|
string lpDependencies, string lpServiceStartName, string lpPassword);
|
|
|
|
[DllImport(ADVAPI, CallingConvention = CallingConvention.Winapi,
|
|
SetLastError = true)]
|
|
public static extern IntPtr OpenService(IntPtr hSCManager,
|
|
string lpServiceName, ServiceAccessRights dwDesiredAccess);
|
|
|
|
[DllImport(ADVAPI, CallingConvention = CallingConvention.Winapi,
|
|
SetLastError = true)]
|
|
[return: MarshalAs(UnmanagedType.Bool)]
|
|
public static extern bool DeleteService(IntPtr hService);
|
|
|
|
[DllImport(ADVAPI, CallingConvention = CallingConvention.Winapi,
|
|
SetLastError = true)]
|
|
[return: MarshalAs(UnmanagedType.Bool)]
|
|
public static extern bool StartService(IntPtr hService,
|
|
uint dwNumServiceArgs, string[] lpServiceArgVectors);
|
|
|
|
[DllImport(ADVAPI, CallingConvention = CallingConvention.Winapi,
|
|
SetLastError = true)]
|
|
[return: MarshalAs(UnmanagedType.Bool)]
|
|
public static extern bool ControlService(IntPtr hService,
|
|
ServiceControl dwControl, ref ServiceStatus lpServiceStatus);
|
|
|
|
[DllImport(KERNEL, CallingConvention = CallingConvention.Winapi)]
|
|
public static extern bool DeviceIoControl(SafeFileHandle device,
|
|
IOControlCode ioControlCode,
|
|
[MarshalAs(UnmanagedType.AsAny)] [In] object inBuffer,
|
|
uint inBufferSize,
|
|
[MarshalAs(UnmanagedType.AsAny)] [Out] object outBuffer,
|
|
uint nOutBufferSize, out uint bytesReturned, IntPtr overlapped);
|
|
|
|
[DllImport(KERNEL, CallingConvention = CallingConvention.Winapi,
|
|
SetLastError = true)]
|
|
public static extern IntPtr CreateFile(string lpFileName,
|
|
FileAccess dwDesiredAccess, uint dwShareMode,
|
|
IntPtr lpSecurityAttributes, CreationDisposition dwCreationDisposition,
|
|
FileAttributes dwFlagsAndAttributes, IntPtr hTemplateFile);
|
|
}
|
|
}
|
|
}
|