Hardware/WinRing0.cs
author moel.mich
Sat, 18 Sep 2010 16:56:52 +0000
changeset 189 69b90bbffb97
parent 167 b7cc9d09aefe
child 195 0ee888c485d5
permissions -rw-r--r--
Fixed Issue 112.
     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.Runtime.InteropServices;
    40 using System.Threading;
    41 
    42 namespace OpenHardwareMonitor.Hardware {
    43 
    44   internal class WinRing0 {
    45 
    46     private 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     private static Mutex isaBusMutex;
    60 
    61     private static string GetDllName() {   
    62       int p = (int)System.Environment.OSVersion.Platform;
    63       if ((p == 4) || (p == 128)) {
    64         if (IntPtr.Size == 4) {
    65           return "libring0.so";
    66         } else {
    67           return "libring0x64.so";
    68         }
    69       } else {
    70         if (IntPtr.Size == 4) {
    71           return "WinRing0.dll";
    72         } else {
    73           return "WinRing0x64.dll";
    74         }
    75       }                       
    76     }
    77     
    78     private delegate bool InitializeOlsDelegate();
    79     private delegate void DeinitializeOlsDelegate();
    80     
    81     public delegate bool IsCpuidDelegate();
    82     public delegate bool CpuidTxDelegate(uint index, uint ecxValue,
    83       out uint eax, out uint ebx, out uint ecx, out uint edx,
    84       UIntPtr threadAffinityMask);
    85     public delegate bool RdmsrDelegate(uint index, out uint eax, out uint edx);
    86     public delegate bool RdmsrTxDelegate(uint index, out uint eax, out uint edx,
    87       UIntPtr threadAffinityMask);
    88     public delegate byte ReadIoPortByteDelegate(ushort port);
    89     public delegate void WriteIoPortByteDelegate(ushort port, byte value);
    90     public delegate uint FindPciDeviceByIdDelegate(ushort vendorId, 
    91       ushort deviceId, byte index);
    92     public delegate bool ReadPciConfigDwordExDelegate(uint pciAddress, 
    93       uint regAddress, out uint value);
    94     public delegate bool WritePciConfigDwordExDelegate(uint pciAddress, 
    95       uint regAddress, uint value);
    96     public delegate bool RdtscTxDelegate(out uint eax, out uint edx,
    97       UIntPtr threadAffinityMask);
    98     public delegate bool RdtscDelegate(out uint eax, out uint edx);
    99 
   100     private static InitializeOlsDelegate InitializeOls = 
   101       CreateDelegate<InitializeOlsDelegate>("InitializeOls");
   102     private static DeinitializeOlsDelegate DeinitializeOls =
   103       CreateDelegate<DeinitializeOlsDelegate>("DeinitializeOls");
   104 
   105     public static readonly IsCpuidDelegate IsCpuid =
   106       CreateDelegate<IsCpuidDelegate>("IsCpuid");
   107     public static readonly CpuidTxDelegate CpuidTx =
   108       CreateDelegate<CpuidTxDelegate>("CpuidTx");
   109     public static readonly RdmsrDelegate Rdmsr =
   110       CreateDelegate<RdmsrDelegate>("Rdmsr");
   111     public static readonly RdmsrTxDelegate RdmsrTx =
   112       CreateDelegate<RdmsrTxDelegate>("RdmsrTx");
   113     public static readonly ReadIoPortByteDelegate ReadIoPortByte =
   114       CreateDelegate<ReadIoPortByteDelegate>("ReadIoPortByte");
   115     public static readonly WriteIoPortByteDelegate WriteIoPortByte =
   116       CreateDelegate<WriteIoPortByteDelegate>("WriteIoPortByte");
   117     public static readonly FindPciDeviceByIdDelegate FindPciDeviceById =
   118       CreateDelegate<FindPciDeviceByIdDelegate>("FindPciDeviceById");
   119     public static readonly ReadPciConfigDwordExDelegate ReadPciConfigDwordEx =
   120       CreateDelegate<ReadPciConfigDwordExDelegate>("ReadPciConfigDwordEx");
   121     public static readonly WritePciConfigDwordExDelegate WritePciConfigDwordEx =
   122       CreateDelegate<WritePciConfigDwordExDelegate>("WritePciConfigDwordEx");
   123     public static readonly RdtscTxDelegate RdtscTx =
   124       CreateDelegate<RdtscTxDelegate>("RdtscTx");
   125     public static readonly RdtscDelegate Rdtsc =
   126       CreateDelegate<RdtscDelegate>("Rdtsc");
   127  
   128     private static T CreateDelegate<T>(string entryPoint) where T : class {
   129       DllImportAttribute attribute = new DllImportAttribute(GetDllName());
   130       attribute.CallingConvention = CallingConvention.Winapi;
   131       attribute.PreserveSig = true;
   132       attribute.EntryPoint = entryPoint;
   133       attribute.CharSet = CharSet.Auto;
   134       T result;
   135       PInvokeDelegateFactory.CreateDelegate(attribute, out result);
   136       return result;
   137     }
   138 
   139     public static void Open() {
   140       try {
   141         if (InitializeOls != null && InitializeOls())
   142           available = true;
   143       } catch (DllNotFoundException) { }   
   144       
   145       isaBusMutex = new Mutex(false, "Access_ISABUS.HTP.Method");
   146     }
   147 
   148     public static bool IsAvailable {
   149       get { return available; }
   150     }
   151 
   152     public static void Close() {
   153       if (available)
   154         DeinitializeOls();        
   155       isaBusMutex.Close();      
   156     }    
   157 
   158     public static bool WaitIsaBusMutex(int millisecondsTimeout) {
   159       try {
   160         return isaBusMutex.WaitOne(millisecondsTimeout, false);
   161       } catch (AbandonedMutexException) { return false; } 
   162         catch (InvalidOperationException) { return false; }     
   163     }
   164 
   165     public static void ReleaseIsaBusMutex() {
   166       isaBusMutex.ReleaseMutex();
   167     }    
   168   }
   169 }