Hardware/WinRing0.cs
author moel.mich
Wed, 03 Feb 2010 20:35:10 +0000
changeset 24 09ab31bee6bd
parent 14 51c2f209da6d
child 44 c150de283ca0
permissions -rw-r--r--
Release version 0.1.11. Added support for load percentage sensor type. Added load sensors for CPUs and ATI GPUs.
     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 RdmsrPxDelegate(uint index, ref uint eax, ref uint edx, 
    87       UIntPtr processAffinityMask);
    88     public delegate byte ReadIoPortByteDelegate(ushort port);
    89     public delegate void WriteIoPortByteDelegate(ushort port, byte value);
    90     public delegate void SetPciMaxBusIndexDelegate(byte max);
    91     public delegate uint FindPciDeviceByIdDelegate(ushort vendorId, 
    92       ushort deviceId, byte index);
    93     public delegate bool ReadPciConfigDwordExDelegate(uint pciAddress, 
    94       uint regAddress, out uint value);
    95     public delegate bool WritePciConfigDwordExDelegate(uint pciAddress, 
    96       uint regAddress, uint value);
    97 
    98     private static InitializeOlsDelegate InitializeOls;
    99     private static DeinitializeOlsDelegate DeinitializeOls;
   100 
   101     public static GetDllStatusDelegate GetDllStatus;
   102     public static IsCpuidDelegate IsCpuid;
   103     public static CpuidDelegate Cpuid;
   104     public static CpuidExDelegate CpuidEx;
   105     public static RdmsrPxDelegate RdmsrPx;
   106     public static ReadIoPortByteDelegate ReadIoPortByte;
   107     public static WriteIoPortByteDelegate WriteIoPortByte;
   108     public static SetPciMaxBusIndexDelegate SetPciMaxBusIndex;
   109     public static FindPciDeviceByIdDelegate FindPciDeviceById;
   110     public static ReadPciConfigDwordExDelegate ReadPciConfigDwordEx;
   111     public static WritePciConfigDwordExDelegate WritePciConfigDwordEx;    
   112 
   113     private static void GetDelegate<T>(string entryPoint, out T newDelegate) 
   114       where T : class 
   115     {
   116       DllImportAttribute attribute = new DllImportAttribute(GetDllName());
   117       attribute.CallingConvention = CallingConvention.Winapi;
   118       attribute.PreserveSig = true;
   119       attribute.EntryPoint = entryPoint;
   120       attribute.CharSet = CharSet.Auto;
   121       PInvokeDelegateFactory.CreateDelegate(attribute, out newDelegate);
   122     }
   123 
   124     static WinRing0() {
   125       GetDelegate("InitializeOls", out InitializeOls);
   126       GetDelegate("DeinitializeOls", out DeinitializeOls);
   127       GetDelegate("GetDllStatus", out GetDllStatus);
   128       GetDelegate("IsCpuid", out IsCpuid);
   129       GetDelegate("Cpuid", out Cpuid);
   130       GetDelegate("CpuidEx", out CpuidEx);
   131       GetDelegate("RdmsrPx", out  RdmsrPx);
   132       GetDelegate("ReadIoPortByte", out ReadIoPortByte);
   133       GetDelegate("WriteIoPortByte", out WriteIoPortByte);
   134       GetDelegate("SetPciMaxBusIndex", out SetPciMaxBusIndex);
   135       GetDelegate("FindPciDeviceById", out FindPciDeviceById);
   136       GetDelegate("ReadPciConfigDwordEx", out ReadPciConfigDwordEx);
   137       GetDelegate("WritePciConfigDwordEx", out WritePciConfigDwordEx);
   138 
   139       try {
   140         if (InitializeOls != null && InitializeOls())
   141           available = true;
   142       } catch (DllNotFoundException) { }       
   143     }
   144     
   145     public static bool IsAvailable {
   146       get { return available; }
   147     }
   148 
   149     private static Deinitializer deinitializer = new Deinitializer();
   150     private class Deinitializer {
   151       ~Deinitializer() {
   152         if (available)
   153           DeinitializeOls();
   154       }
   155     }
   156   }
   157 }