cpu-affinity 0.1.1

// This file is part of cpu-affinity. It is subject to the license terms in the COPYRIGHT file found in the top-level directory of this distribution and at https://raw.githubusercontent.com/lemonrock/cpu-affinity/master/COPYRIGHT. No part of cpu-affinity, including this file, may be copied, modified, propagated, or distributed except according to the terms contained in the COPYRIGHT file.
// Copyright © 2018 The developers of cpu-affinity. See the COPYRIGHT file in the top-level directory of this distribution and at https://raw.githubusercontent.com/lemonrock/cpu-affinity/master/COPYRIGHT.


#[cfg(any(target_os = "android", target_os = "fuschia", target_os = "linux", target_env = "uclibc"))] use ::libc::sched_setaffinity;
#[cfg(target_os = "emscripten")] use self::emscripten::sched_setaffinity;
#[cfg(target_os = "emscripten")] use self::emscripten::pthread_setaffinity_np;
#[cfg(target_os = "fuschia")] use self::fuschia::pthread_setaffinity_np;
#[cfg(target_os = "linux")] use self::linux::pthread_setaffinity_np;
#[cfg(target_env = "uclibc")] use self::uclibc::pthread_setaffinity_np;
use ::libc::CPU_SET;
use ::libc::cpu_set_t;
use ::std::mem::size_of;
use ::std::mem::zeroed;


impl LogicalCores
{
	const _IsSettingProcessAffinitySupported: bool = true;

	#[inline(always)]
	fn _set_process_affinity(&self, process_identifier: ProcessIdentifier) -> io::Result<()>
	{
		let cpu_set = self.as_cpu_set_t();
		let result = unsafe { sched_setaffinity(process_identifier, Self::SizeOfCpuSetT, &cpu_set) };
		if result == 0
		{
			Ok(())
		}
		else
		{
			Err(Self::last_os_error())
		}
	}

	#[inline(always)]
	fn as_cpu_set_t(&self) -> cpu_set_t
	{
		let mut cpu_set_t = Self::empty_cpu_set_t();
		for logical_core in self.0.iter()
		{
			unsafe { CPU_SET(*logical_core, &mut cpu_set_t) };
		}
		cpu_set_t
	}

	#[inline(always)]
	fn empty_cpu_set_t() -> cpu_set_t
	{
		unsafe { zeroed() }
	}

	const SizeOfCpuSetT: usize = size_of::<cpu_set_t>();
}

#[cfg(any(target_os = "android"))]
impl LogicalCores
{
	const _IsSettingThreadAffinitySupported: bool = false;

	#[inline(always)]
	fn _set_thread_affinity(&self, _thread_identifier: ThreadIdentifier) -> io::Result<()>
	{
		Ok(())
	}
}

#[cfg(any(target_os = "emscripten", target_os = "fuschia", target_os = "linux", target_env = "uclibc"))]
impl LogicalCores
{
	const _IsSettingThreadAffinitySupported: bool = true;

	#[inline(always)]
	fn _set_thread_affinity(&self, thread_identifier: ThreadIdentifier) -> io::Result<()>
	{
		let cpu_set = self.as_cpu_set_t();
		let result = unsafe { pthread_setaffinity_np(thread_identifier, Self::SizeOfCpuSetT, &cpu_set) };
		if result == 0
		{
			Ok(())
		}
		else
		{
			Err(io::Error::from_raw_os_error(result))
		}
	}
}