mirror of
https://github.com/openhardwaremonitor/openhardwaremonitor
synced 2025-08-22 18:07:09 +00:00
61 lines
1.8 KiB
C#
Executable File
61 lines
1.8 KiB
C#
Executable File
/*
|
|
|
|
This Source Code Form is subject to the terms of the Mozilla Public
|
|
License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
Copyright (C) 2010 Michael Möller <mmoeller@openhardwaremonitor.org>
|
|
|
|
*/
|
|
|
|
using System;
|
|
using System.Runtime.InteropServices;
|
|
|
|
namespace OpenHardwareMonitor.Hardware {
|
|
|
|
internal static class ThreadAffinity {
|
|
|
|
public static ulong Set(ulong mask) {
|
|
if (mask == 0)
|
|
return 0;
|
|
|
|
int p = (int)Environment.OSVersion.Platform;
|
|
if ((p == 4) || (p == 128)) { // Unix
|
|
ulong result = 0;
|
|
if (NativeMethods.sched_getaffinity(0, (IntPtr)Marshal.SizeOf(result),
|
|
ref result) != 0)
|
|
return 0;
|
|
if (NativeMethods.sched_setaffinity(0, (IntPtr)Marshal.SizeOf(mask),
|
|
ref mask) != 0)
|
|
return 0;
|
|
return result;
|
|
} else { // Windows
|
|
return (ulong)NativeMethods.SetThreadAffinityMask(
|
|
NativeMethods.GetCurrentThread(), (UIntPtr)mask);
|
|
}
|
|
}
|
|
|
|
private static class NativeMethods {
|
|
private const string KERNEL = "kernel32.dll";
|
|
|
|
[DllImport(KERNEL, CallingConvention = CallingConvention.Winapi)]
|
|
public static extern UIntPtr
|
|
SetThreadAffinityMask(IntPtr handle, UIntPtr mask);
|
|
|
|
[DllImport(KERNEL, CallingConvention = CallingConvention.Winapi)]
|
|
public static extern IntPtr GetCurrentThread();
|
|
|
|
private const string LIBC = "libc";
|
|
|
|
[DllImport(LIBC)]
|
|
public static extern int sched_getaffinity(int pid, IntPtr maskSize,
|
|
ref ulong mask);
|
|
|
|
[DllImport(LIBC)]
|
|
public static extern int sched_setaffinity(int pid, IntPtr maskSize,
|
|
ref ulong mask);
|
|
}
|
|
}
|
|
}
|
|
|