Hardware/WinRing0.cs
author moel.mich
Thu, 04 Mar 2010 20:26:56 +0000
changeset 72 c4dfd596a6c3
parent 44 c150de283ca0
child 79 9cdbe1d8d12a
permissions -rw-r--r--
Changed the system tray font.
     1 /*
     2   
     3   Version: MPL 1.1/GPL 2.0/LGPL 2.1
     4 
     5   The contents of this file are subject to the Mozilla Public License Version
     6   1.1 (the "License"); you may not use this file except in compliance with
     7   the License. You may obtain a copy of the License at
     8  
     9   http://www.mozilla.org/MPL/
    10 
    11   Software distributed under the License is distributed on an "AS IS" basis,
    12   WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
    13   for the specific language governing rights and limitations under the License.
    14 
    15   The Original Code is the Open Hardware Monitor code.
    16 
    17   The Initial Developer of the Original Code is 
    18   Michael Möller <m.moeller@gmx.ch>.
    19   Portions created by the Initial Developer are Copyright (C) 2009-2010
    20   the Initial Developer. All Rights Reserved.
    21 
    22   Contributor(s):
    23 
    24   Alternatively, the contents of this file may be used under the terms of
    25   either the GNU General Public License Version 2 or later (the "GPL"), or
    26   the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
    27   in which case the provisions of the GPL or the LGPL are applicable instead
    28   of those above. If you wish to allow use of your version of this file only
    29   under the terms of either the GPL or the LGPL, and not to allow others to
    30   use your version of this file under the terms of the MPL, indicate your
    31   decision by deleting the provisions above and replace them with the notice
    32   and other provisions required by the GPL or the LGPL. If you do not delete
    33   the provisions above, a recipient may use your version of this file under
    34   the terms of any one of the MPL, the GPL or the LGPL.
    35  
    36 */
    37 
    38 using System;
    39 using System.Collections.Generic;
    40 using System.IO;
    41 using System.Runtime.InteropServices;
    42 using System.Text;
    43 
    44 namespace OpenHardwareMonitor.Hardware {
    45 
    46   public class WinRing0 {
    47     
    48     public enum OlsDllStatus{
    49       OLS_DLL_NO_ERROR                        = 0,
    50       OLS_DLL_UNSUPPORTED_PLATFORM            = 1,
    51       OLS_DLL_DRIVER_NOT_LOADED               = 2,
    52       OLS_DLL_DRIVER_NOT_FOUND                = 3,
    53       OLS_DLL_DRIVER_UNLOADED                 = 4,
    54       OLS_DLL_DRIVER_NOT_LOADED_ON_NETWORK    = 5,
    55       OLS_DLL_UNKNOWN_ERROR                   = 9
    56     }
    57 
    58     private static bool available = false;
    59 
    60     private static string GetDllName() {   
    61       int p = (int)System.Environment.OSVersion.Platform;
    62       if ((p == 4) || (p == 128)) {
    63         if (IntPtr.Size == 4) {
    64           return "libring0.so";
    65         } else {
    66           return "libring0x64.so";
    67         }
    68       } else {
    69         if (IntPtr.Size == 4) {
    70           return "WinRing0.dll";
    71         } else {
    72           return "WinRing0x64.dll";
    73         }
    74       }                       
    75     }
    76     
    77     private delegate bool InitializeOlsDelegate();
    78     private delegate void DeinitializeOlsDelegate();
    79     
    80     public delegate uint GetDllStatusDelegate();
    81     public delegate bool IsCpuidDelegate();
    82     public delegate bool CpuidDelegate(uint index, 
    83       out uint eax, out uint ebx, out uint ecx, out uint edx);
    84     public delegate bool CpuidExDelegate(uint index, uint ecxValue, 
    85       out uint eax, out uint ebx, out uint ecx, out uint edx);
    86     public delegate bool RdmsrDelegate(uint index, out uint eax, out uint edx);
    87     public delegate bool RdmsrTxDelegate(uint index, out uint eax, out uint edx,
    88       UIntPtr threadAffinityMask);
    89     public delegate byte ReadIoPortByteDelegate(ushort port);
    90     public delegate void WriteIoPortByteDelegate(ushort port, byte value);
    91     public delegate void SetPciMaxBusIndexDelegate(byte max);
    92     public delegate uint FindPciDeviceByIdDelegate(ushort vendorId, 
    93       ushort deviceId, byte index);
    94     public delegate bool ReadPciConfigDwordExDelegate(uint pciAddress, 
    95       uint regAddress, out uint value);
    96     public delegate bool WritePciConfigDwordExDelegate(uint pciAddress, 
    97       uint regAddress, uint value);
    98     public delegate bool RdtscTxDelegate(out uint eax, out uint edx,
    99       UIntPtr threadAffinityMask);
   100 
   101     private static InitializeOlsDelegate InitializeOls;
   102     private static DeinitializeOlsDelegate DeinitializeOls;
   103 
   104     public static GetDllStatusDelegate GetDllStatus;
   105     public static IsCpuidDelegate IsCpuid;
   106     public static CpuidDelegate Cpuid;
   107     public static CpuidExDelegate CpuidEx;
   108     public static RdmsrDelegate Rdmsr;
   109     public static RdmsrTxDelegate RdmsrTx;
   110     public static ReadIoPortByteDelegate ReadIoPortByte;
   111     public static WriteIoPortByteDelegate WriteIoPortByte;
   112     public static SetPciMaxBusIndexDelegate SetPciMaxBusIndex;
   113     public static FindPciDeviceByIdDelegate FindPciDeviceById;
   114     public static ReadPciConfigDwordExDelegate ReadPciConfigDwordEx;
   115     public static WritePciConfigDwordExDelegate WritePciConfigDwordEx;
   116     public static RdtscTxDelegate RdtscTx;
   117 
   118     private static void GetDelegate<T>(string entryPoint, out T newDelegate) 
   119       where T : class 
   120     {
   121       DllImportAttribute attribute = new DllImportAttribute(GetDllName());
   122       attribute.CallingConvention = CallingConvention.Winapi;
   123       attribute.PreserveSig = true;
   124       attribute.EntryPoint = entryPoint;
   125       attribute.CharSet = CharSet.Auto;
   126       PInvokeDelegateFactory.CreateDelegate(attribute, out newDelegate);
   127     }
   128 
   129     static WinRing0() {
   130       GetDelegate("InitializeOls", out InitializeOls);
   131       GetDelegate("DeinitializeOls", out DeinitializeOls);
   132       GetDelegate("GetDllStatus", out GetDllStatus);
   133       GetDelegate("IsCpuid", out IsCpuid);
   134       GetDelegate("Cpuid", out Cpuid);
   135       GetDelegate("CpuidEx", out CpuidEx);
   136       GetDelegate("Rdmsr", out  Rdmsr);
   137       GetDelegate("RdmsrTx", out  RdmsrTx);
   138       GetDelegate("ReadIoPortByte", out ReadIoPortByte);
   139       GetDelegate("WriteIoPortByte", out WriteIoPortByte);
   140       GetDelegate("SetPciMaxBusIndex", out SetPciMaxBusIndex);
   141       GetDelegate("FindPciDeviceById", out FindPciDeviceById);
   142       GetDelegate("ReadPciConfigDwordEx", out ReadPciConfigDwordEx);
   143       GetDelegate("WritePciConfigDwordEx", out WritePciConfigDwordEx);
   144       GetDelegate("RdtscTx", out RdtscTx);
   145 
   146       try {
   147         if (InitializeOls != null && InitializeOls())
   148           available = true;
   149       } catch (DllNotFoundException) { }       
   150     }
   151     
   152     public static bool IsAvailable {
   153       get { return available; }
   154     }
   155 
   156     private static Deinitializer deinitializer = new Deinitializer();
   157     private class Deinitializer {
   158       ~Deinitializer() {
   159         if (available)
   160           DeinitializeOls();
   161       }
   162     }
   163   }
   164 }