1.1 --- /dev/null	Thu Jan 01 00:00:00 1970 +0000
     1.2 +++ b/Hardware/ThreadAffinity.cs	Wed Nov 03 22:07:46 2010 +0000
     1.3 @@ -0,0 +1,87 @@
     1.4 +/*
     1.5 +  
     1.6 +  Version: MPL 1.1/GPL 2.0/LGPL 2.1
     1.7 +
     1.8 +  The contents of this file are subject to the Mozilla Public License Version
     1.9 +  1.1 (the "License"); you may not use this file except in compliance with
    1.10 +  the License. You may obtain a copy of the License at
    1.11 + 
    1.12 +  http://www.mozilla.org/MPL/
    1.13 +
    1.14 +  Software distributed under the License is distributed on an "AS IS" basis,
    1.15 +  WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
    1.16 +  for the specific language governing rights and limitations under the License.
    1.17 +
    1.18 +  The Original Code is the Open Hardware Monitor code.
    1.19 +
    1.20 +  The Initial Developer of the Original Code is 
    1.21 +  Michael Möller <m.moeller@gmx.ch>.
    1.22 +  Portions created by the Initial Developer are Copyright (C) 2010
    1.23 +  the Initial Developer. All Rights Reserved.
    1.24 +
    1.25 +  Contributor(s):
    1.26 +
    1.27 +  Alternatively, the contents of this file may be used under the terms of
    1.28 +  either the GNU General Public License Version 2 or later (the "GPL"), or
    1.29 +  the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
    1.30 +  in which case the provisions of the GPL or the LGPL are applicable instead
    1.31 +  of those above. If you wish to allow use of your version of this file only
    1.32 +  under the terms of either the GPL or the LGPL, and not to allow others to
    1.33 +  use your version of this file under the terms of the MPL, indicate your
    1.34 +  decision by deleting the provisions above and replace them with the notice
    1.35 +  and other provisions required by the GPL or the LGPL. If you do not delete
    1.36 +  the provisions above, a recipient may use your version of this file under
    1.37 +  the terms of any one of the MPL, the GPL or the LGPL.
    1.38 + 
    1.39 +*/
    1.40 +
    1.41 +using System;
    1.42 +using System.Runtime.InteropServices;
    1.43 +
    1.44 +namespace OpenHardwareMonitor.Hardware {
    1.45 +  
    1.46 +  internal static class ThreadAffinity {
    1.47 +  
    1.48 +    public static ulong Set(ulong mask) { 
    1.49 +      if (mask == 0)
    1.50 +        return 0;
    1.51 +        
    1.52 +      int p = (int)Environment.OSVersion.Platform;
    1.53 +      if ((p == 4) || (p == 128)) { // Unix
    1.54 +        ulong result = 0;
    1.55 +        if (NativeMethods.sched_getaffinity(0, (IntPtr)Marshal.SizeOf(result), 
    1.56 +          ref result) != 0)          
    1.57 +          return 0;
    1.58 +        if (NativeMethods.sched_setaffinity(0, (IntPtr)Marshal.SizeOf(mask), 
    1.59 +          ref mask) != 0)
    1.60 +          return 0;
    1.61 +        return result;
    1.62 +      } else { // Windows      
    1.63 +        return (ulong)NativeMethods.SetThreadAffinityMask(
    1.64 +          NativeMethods.GetCurrentThread(), (UIntPtr)mask);
    1.65 +      }
    1.66 +    }
    1.67 +  
    1.68 +    private static class NativeMethods {      
    1.69 +      private const string KERNEL = "kernel32.dll";
    1.70 +
    1.71 +      [DllImport(KERNEL, CallingConvention = CallingConvention.Winapi)]
    1.72 +      public static extern UIntPtr
    1.73 +        SetThreadAffinityMask(IntPtr handle, UIntPtr mask);
    1.74 +
    1.75 +      [DllImport(KERNEL, CallingConvention = CallingConvention.Winapi)]
    1.76 +      public static extern IntPtr GetCurrentThread();       
    1.77 +      
    1.78 +      private const string LIBC = "libc";
    1.79 +      
    1.80 +      [DllImport(LIBC)]
    1.81 +      public static extern int sched_getaffinity(int pid, IntPtr maskSize,
    1.82 +        ref ulong mask);
    1.83 +      
    1.84 +      [DllImport(LIBC)]
    1.85 +      public static extern int sched_setaffinity(int pid, IntPtr maskSize,
    1.86 +        ref ulong mask);  
    1.87 +    }  
    1.88 +  }
    1.89 +}
    1.90 +