Hardware/WinRing0.cs
author moel.mich
Wed, 22 Sep 2010 19:12:12 +0000
changeset 196 5e9a8595296c
parent 195 0ee888c485d5
child 201 958e9fe8afdf
permissions -rw-r--r--
Rewritten the PCI access for AMD CPUs.
     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;
    59     private static Mutex isaBusMutex;
    60 
    61     private static string GetDllName() {   
    62       int p = (int)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 bool ReadPciConfigDwordExDelegate(uint pciAddress, 
    91       uint regAddress, out uint value);
    92     public delegate bool WritePciConfigDwordExDelegate(uint pciAddress, 
    93       uint regAddress, uint value);
    94     public delegate bool RdtscTxDelegate(out uint eax, out uint edx,
    95       UIntPtr threadAffinityMask);
    96     public delegate bool RdtscDelegate(out uint eax, out uint edx);
    97 
    98     private static readonly InitializeOlsDelegate InitializeOls = 
    99       CreateDelegate<InitializeOlsDelegate>("InitializeOls");
   100     private static readonly DeinitializeOlsDelegate DeinitializeOls =
   101       CreateDelegate<DeinitializeOlsDelegate>("DeinitializeOls");
   102 
   103     public static readonly IsCpuidDelegate IsCpuid =
   104       CreateDelegate<IsCpuidDelegate>("IsCpuid");
   105     public static readonly CpuidTxDelegate CpuidTx =
   106       CreateDelegate<CpuidTxDelegate>("CpuidTx");
   107     public static readonly RdmsrDelegate Rdmsr =
   108       CreateDelegate<RdmsrDelegate>("Rdmsr");
   109     public static readonly RdmsrTxDelegate RdmsrTx =
   110       CreateDelegate<RdmsrTxDelegate>("RdmsrTx");
   111     public static readonly ReadIoPortByteDelegate ReadIoPortByte =
   112       CreateDelegate<ReadIoPortByteDelegate>("ReadIoPortByte");
   113     public static readonly WriteIoPortByteDelegate WriteIoPortByte =
   114       CreateDelegate<WriteIoPortByteDelegate>("WriteIoPortByte");
   115     public static readonly ReadPciConfigDwordExDelegate ReadPciConfigDwordEx =
   116       CreateDelegate<ReadPciConfigDwordExDelegate>("ReadPciConfigDwordEx");
   117     public static readonly WritePciConfigDwordExDelegate WritePciConfigDwordEx =
   118       CreateDelegate<WritePciConfigDwordExDelegate>("WritePciConfigDwordEx");
   119     public static readonly RdtscTxDelegate RdtscTx =
   120       CreateDelegate<RdtscTxDelegate>("RdtscTx");
   121     public static readonly RdtscDelegate Rdtsc =
   122       CreateDelegate<RdtscDelegate>("Rdtsc");
   123  
   124     private static T CreateDelegate<T>(string entryPoint) where T : class {
   125       DllImportAttribute attribute = new DllImportAttribute(GetDllName());
   126       attribute.CallingConvention = CallingConvention.Winapi;
   127       attribute.PreserveSig = true;
   128       attribute.EntryPoint = entryPoint;
   129       attribute.CharSet = CharSet.Auto;
   130       T result;
   131       PInvokeDelegateFactory.CreateDelegate(attribute, out result);
   132       return result;
   133     }
   134 
   135     public static void Open() {
   136       try {
   137         if (InitializeOls != null && InitializeOls())
   138           available = true;
   139       } catch (DllNotFoundException) { }   
   140       
   141       isaBusMutex = new Mutex(false, "Access_ISABUS.HTP.Method");
   142     }
   143 
   144     public static bool IsAvailable {
   145       get { return available; }
   146     }
   147 
   148     public static void Close() {
   149       if (available)
   150         DeinitializeOls();        
   151       isaBusMutex.Close();      
   152     }    
   153 
   154     public static bool WaitIsaBusMutex(int millisecondsTimeout) {
   155       try {
   156         return isaBusMutex.WaitOne(millisecondsTimeout, false);
   157       } catch (AbandonedMutexException) { return false; } 
   158         catch (InvalidOperationException) { return false; }     
   159     }
   160 
   161     public static void ReleaseIsaBusMutex() {
   162       isaBusMutex.ReleaseMutex();
   163     }
   164 
   165     public const uint InvalidPciAddress = 0xFFFFFFFF;
   166 
   167     public static uint GetPciAddress(byte bus, byte device, byte function)	{
   168       return 
   169         (uint)(((bus & 0xFF) << 8) | ((device & 0x1F) << 3) | (function & 7));
   170     }
   171   }
   172 }