/*

   Version: MPL 1.1/GPL 2.0/LGPL 2.1

   The contents of this file are subject to the Mozilla Public License Version

   1.1 (the "License"); you may not use this file except in compliance with

   the License. You may obtain a copy of the License at

   http://www.mozilla.org/MPL/

   Software distributed under the License is distributed on an "AS IS" basis,

   WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License

   for the specific language governing rights and limitations under the License.

   The Original Code is the Open Hardware Monitor code.

   The Initial Developer of the Original Code is 

   Michael Möller <m.moeller@gmx.ch>.

   Portions created by the Initial Developer are Copyright (C) 2010-2011

   the Initial Developer. All Rights Reserved.

   Contributor(s):

   Alternatively, the contents of this file may be used under the terms of

   either the GNU General Public License Version 2 or later (the "GPL"), or

   the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),

   in which case the provisions of the GPL or the LGPL are applicable instead

   of those above. If you wish to allow use of your version of this file only

   under the terms of either the GPL or the LGPL, and not to allow others to

   use your version of this file under the terms of the MPL, indicate your

   decision by deleting the provisions above and replace them with the notice

   and other provisions required by the GPL or the LGPL. If you do not delete

   the provisions above, a recipient may use your version of this file under

   the terms of any one of the MPL, the GPL or the LGPL.

 */

 using System;

 using System.IO;

 using System.Reflection;

 using System.Runtime.InteropServices;

 using System.Threading;

 using System.Text;

 namespace OpenHardwareMonitor.Hardware {

   internal static class Ring0 {

     private static KernelDriver driver;

     private static string fileName;

     private static Mutex isaBusMutex;

     private static readonly StringBuilder report = new StringBuilder();

     private const uint OLS_TYPE = 40000;

     private static IOControlCode

       IOCTL_OLS_GET_REFCOUNT = new IOControlCode(OLS_TYPE, 0x801,

         IOControlCode.Access.Any),

       IOCTL_OLS_GET_DRIVER_VERSION = new IOControlCode(OLS_TYPE, 0x800,

         IOControlCode.Access.Any),

       IOCTL_OLS_READ_MSR = new IOControlCode(OLS_TYPE, 0x821,

         IOControlCode.Access.Any),

       IOCTL_OLS_WRITE_MSR = new IOControlCode(OLS_TYPE, 0x822, 

         IOControlCode.Access.Any),

       IOCTL_OLS_READ_IO_PORT_BYTE = new IOControlCode(OLS_TYPE, 0x833,

         IOControlCode.Access.Read),

       IOCTL_OLS_WRITE_IO_PORT_BYTE = new IOControlCode(OLS_TYPE, 0x836, 

         IOControlCode.Access.Write),

       IOCTL_OLS_READ_PCI_CONFIG = new IOControlCode(OLS_TYPE, 0x851, 

         IOControlCode.Access.Read),

       IOCTL_OLS_WRITE_PCI_CONFIG = new IOControlCode(OLS_TYPE, 0x852,

         IOControlCode.Access.Write),

       IOCTL_OLS_READ_MEMORY = new IOControlCode(OLS_TYPE, 0x841,

         IOControlCode.Access.Read);

     private static string GetTempFileName() {

       // try to get a file in the temporary folder

       try {

         return Path.GetTempFileName();        

       } catch (IOException) { } 

         catch (UnauthorizedAccessException) { }

       // if this failed, we try to create one in the application folder

       string fileName = Path.ChangeExtension(

         Assembly.GetExecutingAssembly().Location, ".sys");

       try {

         using (FileStream stream = File.Create(fileName)) {

           return fileName;

         }        

       } catch (IOException) { } 

         catch (UnauthorizedAccessException) { }

       return null;

     }

     private static bool ExtractDriver(string fileName) {

       string resourceName = "OpenHardwareMonitor.Hardware." +

         (IntPtr.Size == 4 ? "WinRing0.sys" : "WinRing0x64.sys");

       string[] names =

         Assembly.GetExecutingAssembly().GetManifestResourceNames();

       byte[] buffer = null;

       for (int i = 0; i < names.Length; i++) {

         if (names[i].Replace('\\', '.') == resourceName) {

           using (Stream stream = Assembly.GetExecutingAssembly().

             GetManifestResourceStream(names[i])) 

           {

               buffer = new byte[stream.Length];

               stream.Read(buffer, 0, buffer.Length);

           }

         }

       }

       if (buffer == null)

         return false;

       using (FileStream target = new FileStream(fileName, FileMode.Create)) {

         target.Write(buffer, 0, buffer.Length);

       }

       return true;

     }

     public static void Open() {

       // no implementation for unix systems

       int p = (int)Environment.OSVersion.Platform;

       if ((p == 4) || (p == 128))

         return;  

       if (driver != null)

         return;

       // clear the current report

       report.Length = 0;

       driver = new KernelDriver("WinRing0_1_2_0");

       driver.Open();

       if (!driver.IsOpen) {

         // driver is not loaded, try to reinstall and open

         driver.Delete();

         fileName = GetTempFileName();

         if (fileName != null && ExtractDriver(fileName)) {

           if (driver.Install(fileName)) {

             driver.Open();

             if (!driver.IsOpen) {

               driver.Delete();

               report.AppendLine("Status: Opening driver failed");

             }

           } else {

             report.AppendLine("Status: Installing driver \"" +

               fileName + "\" failed" +

               (File.Exists(fileName) ? " and file exists" : ""));

             report.AppendLine();

             report.Append("Exception: " + Marshal.GetExceptionForHR(

               Marshal.GetHRForLastWin32Error()).Message);

           }

         } else {

           report.AppendLine("Status: Extracting driver failed");

         }

         try {

           // try to delte the driver file

           if (File.Exists(fileName))

             File.Delete(fileName);

           fileName = null;

         } catch (IOException) { } 

           catch (UnauthorizedAccessException) { }

       }

       if (!driver.IsOpen) 

         driver = null;

       isaBusMutex = new Mutex(false, "Global\\Access_ISABUS.HTP.Method");

     }

     public static bool IsOpen {

       get { return driver != null; }

     }

     public static void Close() {

       if (driver == null)

         return;

       uint refCount = 0;

       driver.DeviceIOControl(IOCTL_OLS_GET_REFCOUNT, null, ref refCount);

       driver.Close();

       if (refCount <= 1)

         driver.Delete();

       driver = null;

       isaBusMutex.Close();

       // try to delete temporary driver file again if failed during open

       if (fileName != null && File.Exists(fileName)) {

         try {

           File.Delete(fileName);

           fileName = null;

         } catch (IOException) { } 

           catch (UnauthorizedAccessException) { }

       }

     }

     public static string GetReport() {

       if (report.Length > 0) {

         StringBuilder r = new StringBuilder();

         r.AppendLine("Ring0");

         r.AppendLine();

         r.Append(report);

         r.AppendLine();

         return r.ToString();

       } else

         return null;

     }

     public static bool WaitIsaBusMutex(int millisecondsTimeout) {

       try {

         return isaBusMutex.WaitOne(millisecondsTimeout, false);

       } catch (AbandonedMutexException) { return false; } 

         catch (InvalidOperationException) { return false; }

     }

     public static void ReleaseIsaBusMutex() {

       isaBusMutex.ReleaseMutex();

     }

     public static bool Rdmsr(uint index, out uint eax, out uint edx) {

       if (driver == null) {

         eax = 0;

         edx = 0;

         return false;

       }

       ulong buffer = 0;

       bool result = driver.DeviceIOControl(IOCTL_OLS_READ_MSR, index,

         ref buffer);

       edx = (uint)((buffer >> 32) & 0xFFFFFFFF);

       eax = (uint)(buffer & 0xFFFFFFFF);

       return result;

     }

     public static bool RdmsrTx(uint index, out uint eax, out uint edx,

       ulong threadAffinityMask) 

     {

       ulong mask = ThreadAffinity.Set(threadAffinityMask);

       bool result = Rdmsr(index, out eax, out edx);

       ThreadAffinity.Set(mask);

       return result;

     }

     [StructLayout(LayoutKind.Sequential, Pack = 1)]

     private struct WrmsrInput {

       public uint Register;

       public ulong Value;

     }

     public static bool Wrmsr(uint index, uint eax, uint edx) {

       if (driver == null)

         return false;

       WrmsrInput input = new WrmsrInput();

       input.Register = index;

       input.Value = ((ulong)edx << 32) | eax;

       return driver.DeviceIOControl(IOCTL_OLS_WRITE_MSR, input);

     }

     public static byte ReadIoPort(uint port) {

       if (driver == null)

         return 0;

       uint value = 0;

       driver.DeviceIOControl(IOCTL_OLS_READ_IO_PORT_BYTE, port, ref value);

       return (byte)(value & 0xFF);

     }

     [StructLayout(LayoutKind.Sequential, Pack = 1)]

     private struct WriteIoPortInput {

       public uint PortNumber;

       public byte Value;

     }

     public static void WriteIoPort(uint port, byte value) {

       if (driver == null)

         return;

       WriteIoPortInput input = new WriteIoPortInput();

       input.PortNumber = port;

       input.Value = value;

       driver.DeviceIOControl(IOCTL_OLS_WRITE_IO_PORT_BYTE, input);

     }

     public const uint InvalidPciAddress = 0xFFFFFFFF;

     public static uint GetPciAddress(byte bus, byte device, byte function) {

       return

         (uint)(((bus & 0xFF) << 8) | ((device & 0x1F) << 3) | (function & 7));

     }

     [StructLayout(LayoutKind.Sequential, Pack = 1)]

     private struct ReadPciConfigInput {

       public uint PciAddress;

       public uint RegAddress;

     }

     public static bool ReadPciConfig(uint pciAddress, uint regAddress, 

       out uint value) 

     {

       if (driver == null || (regAddress & 3) != 0) {

         value = 0;

         return false;

       }

       ReadPciConfigInput input = new ReadPciConfigInput();

       input.PciAddress = pciAddress;

       input.RegAddress = regAddress;

       value = 0;

       return driver.DeviceIOControl(IOCTL_OLS_READ_PCI_CONFIG, input, 

         ref value);

     }

     [StructLayout(LayoutKind.Sequential, Pack = 1)]

     private struct WritePciConfigInput {

       public uint PciAddress;

       public uint RegAddress;

       public uint Value;

     }

     public static bool WritePciConfig(uint pciAddress, uint regAddress, 

       uint value) 

     {

       if (driver == null || (regAddress & 3) != 0)

         return false;

       WritePciConfigInput input = new WritePciConfigInput();

       input.PciAddress = pciAddress;

       input.RegAddress = regAddress;

       input.Value = value;

       return driver.DeviceIOControl(IOCTL_OLS_WRITE_PCI_CONFIG, input);

     }

     [StructLayout(LayoutKind.Sequential, Pack = 1)]

     private struct ReadMemoryInput {

       public ulong address;

       public uint unitSize;

       public uint count;

     }

     public static bool ReadMemory<T>(ulong address, ref T buffer) {

       if (driver == null) {

         return false;

       }

       ReadMemoryInput input = new ReadMemoryInput();

       input.address = address;

       input.unitSize = 1;

       input.count = (uint)Marshal.SizeOf(buffer);

       return driver.DeviceIOControl(IOCTL_OLS_READ_MEMORY, input,

         ref buffer);

     }

   }

 }