moel@1: /* moel@1: moel@1: Version: MPL 1.1/GPL 2.0/LGPL 2.1 moel@1: moel@1: The contents of this file are subject to the Mozilla Public License Version moel@1: 1.1 (the "License"); you may not use this file except in compliance with moel@1: the License. You may obtain a copy of the License at moel@1: moel@1: http://www.mozilla.org/MPL/ moel@1: moel@1: Software distributed under the License is distributed on an "AS IS" basis, moel@1: WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License moel@1: for the specific language governing rights and limitations under the License. moel@1: moel@1: The Original Code is the Open Hardware Monitor code. moel@1: moel@1: The Initial Developer of the Original Code is moel@1: Michael Möller . moel@1: Portions created by the Initial Developer are Copyright (C) 2009-2010 moel@1: the Initial Developer. All Rights Reserved. moel@1: moel@1: Contributor(s): moel@1: moel@1: Alternatively, the contents of this file may be used under the terms of moel@1: either the GNU General Public License Version 2 or later (the "GPL"), or moel@1: the GNU Lesser General Public License Version 2.1 or later (the "LGPL"), moel@1: in which case the provisions of the GPL or the LGPL are applicable instead moel@1: of those above. If you wish to allow use of your version of this file only moel@1: under the terms of either the GPL or the LGPL, and not to allow others to moel@1: use your version of this file under the terms of the MPL, indicate your moel@1: decision by deleting the provisions above and replace them with the notice moel@1: and other provisions required by the GPL or the LGPL. If you do not delete moel@1: the provisions above, a recipient may use your version of this file under moel@1: the terms of any one of the MPL, the GPL or the LGPL. moel@1: moel@1: */ moel@1: moel@1: using System; moel@1: using System.Collections.Generic; moel@1: using System.IO; moel@1: using System.Runtime.InteropServices; moel@1: using System.Text; moel@1: moel@1: namespace OpenHardwareMonitor.Hardware { moel@1: moel@1: public class WinRing0 { moel@1: moel@1: public enum OlsDllStatus{ moel@1: OLS_DLL_NO_ERROR = 0, moel@1: OLS_DLL_UNSUPPORTED_PLATFORM = 1, moel@1: OLS_DLL_DRIVER_NOT_LOADED = 2, moel@1: OLS_DLL_DRIVER_NOT_FOUND = 3, moel@1: OLS_DLL_DRIVER_UNLOADED = 4, moel@1: OLS_DLL_DRIVER_NOT_LOADED_ON_NETWORK = 5, moel@1: OLS_DLL_UNKNOWN_ERROR = 9 moel@1: } moel@1: moel@1: private static bool available = false; moel@1: moel@1: private static string GetDllName() { moel@1: int p = (int)System.Environment.OSVersion.Platform; moel@1: if ((p == 4) || (p == 128)) { moel@1: if (IntPtr.Size == 4) { moel@1: return "libring0.so"; moel@1: } else { moel@1: return "libring0x64.so"; moel@1: } moel@1: } else { moel@1: if (IntPtr.Size == 4) { moel@1: return "WinRing0.dll"; moel@1: } else { moel@1: return "WinRing0x64.dll"; moel@1: } moel@1: } moel@1: } moel@1: moel@1: private delegate bool InitializeOlsDelegate(); moel@1: private delegate void DeinitializeOlsDelegate(); moel@1: moel@1: public delegate uint GetDllStatusDelegate(); moel@1: public delegate bool IsCpuidDelegate(); moel@22: public delegate bool CpuidDelegate(uint index, moel@22: out uint eax, out uint ebx, out uint ecx, out uint edx); moel@22: public delegate bool CpuidExDelegate(uint index, uint ecxValue, moel@22: out uint eax, out uint ebx, out uint ecx, out uint edx); moel@46: public delegate bool RdmsrDelegate(uint index, out uint eax, out uint edx); moel@46: public delegate bool RdmsrTxDelegate(uint index, out uint eax, out uint edx, moel@46: UIntPtr threadAffinityMask); moel@1: public delegate byte ReadIoPortByteDelegate(ushort port); moel@1: public delegate void WriteIoPortByteDelegate(ushort port, byte value); moel@1: public delegate void SetPciMaxBusIndexDelegate(byte max); moel@1: public delegate uint FindPciDeviceByIdDelegate(ushort vendorId, moel@1: ushort deviceId, byte index); moel@1: public delegate bool ReadPciConfigDwordExDelegate(uint pciAddress, moel@1: uint regAddress, out uint value); moel@14: public delegate bool WritePciConfigDwordExDelegate(uint pciAddress, moel@14: uint regAddress, uint value); moel@46: public delegate bool RdtscTxDelegate(out uint eax, out uint edx, moel@46: UIntPtr threadAffinityMask); moel@1: moel@1: private static InitializeOlsDelegate InitializeOls; moel@1: private static DeinitializeOlsDelegate DeinitializeOls; moel@1: moel@1: public static GetDllStatusDelegate GetDllStatus; moel@1: public static IsCpuidDelegate IsCpuid; moel@1: public static CpuidDelegate Cpuid; moel@22: public static CpuidExDelegate CpuidEx; moel@46: public static RdmsrDelegate Rdmsr; moel@46: public static RdmsrTxDelegate RdmsrTx; moel@1: public static ReadIoPortByteDelegate ReadIoPortByte; moel@1: public static WriteIoPortByteDelegate WriteIoPortByte; moel@1: public static SetPciMaxBusIndexDelegate SetPciMaxBusIndex; moel@1: public static FindPciDeviceByIdDelegate FindPciDeviceById; moel@1: public static ReadPciConfigDwordExDelegate ReadPciConfigDwordEx; moel@44: public static WritePciConfigDwordExDelegate WritePciConfigDwordEx; moel@46: public static RdtscTxDelegate RdtscTx; moel@1: moel@1: private static void GetDelegate(string entryPoint, out T newDelegate) moel@1: where T : class moel@1: { moel@1: DllImportAttribute attribute = new DllImportAttribute(GetDllName()); moel@1: attribute.CallingConvention = CallingConvention.Winapi; moel@1: attribute.PreserveSig = true; moel@1: attribute.EntryPoint = entryPoint; moel@1: attribute.CharSet = CharSet.Auto; moel@1: PInvokeDelegateFactory.CreateDelegate(attribute, out newDelegate); moel@1: } moel@1: moel@1: static WinRing0() { moel@1: GetDelegate("InitializeOls", out InitializeOls); moel@1: GetDelegate("DeinitializeOls", out DeinitializeOls); moel@1: GetDelegate("GetDllStatus", out GetDllStatus); moel@1: GetDelegate("IsCpuid", out IsCpuid); moel@1: GetDelegate("Cpuid", out Cpuid); moel@22: GetDelegate("CpuidEx", out CpuidEx); moel@46: GetDelegate("Rdmsr", out Rdmsr); moel@46: GetDelegate("RdmsrTx", out RdmsrTx); moel@1: GetDelegate("ReadIoPortByte", out ReadIoPortByte); moel@1: GetDelegate("WriteIoPortByte", out WriteIoPortByte); moel@1: GetDelegate("SetPciMaxBusIndex", out SetPciMaxBusIndex); moel@1: GetDelegate("FindPciDeviceById", out FindPciDeviceById); moel@1: GetDelegate("ReadPciConfigDwordEx", out ReadPciConfigDwordEx); moel@14: GetDelegate("WritePciConfigDwordEx", out WritePciConfigDwordEx); moel@46: GetDelegate("RdtscTx", out RdtscTx); moel@1: moel@1: try { moel@1: if (InitializeOls != null && InitializeOls()) moel@1: available = true; moel@1: } catch (DllNotFoundException) { } moel@1: } moel@1: moel@1: public static bool IsAvailable { moel@1: get { return available; } moel@1: } moel@1: moel@1: private static Deinitializer deinitializer = new Deinitializer(); moel@1: private class Deinitializer { moel@1: ~Deinitializer() { moel@1: if (available) moel@1: DeinitializeOls(); moel@1: } moel@1: } moel@1: } moel@1: }