Blame src/thread-utils.c

Packit ae9e2a
/*
Packit ae9e2a
 * Copyright (C) the libgit2 contributors. All rights reserved.
Packit ae9e2a
 *
Packit ae9e2a
 * This file is part of libgit2, distributed under the GNU GPL v2 with
Packit ae9e2a
 * a Linking Exception. For full terms see the included COPYING file.
Packit ae9e2a
 */
Packit ae9e2a
#include "common.h"
Packit ae9e2a
#include "thread-utils.h"
Packit ae9e2a
Packit ae9e2a
#ifdef _WIN32
Packit ae9e2a
#ifndef WIN32_LEAN_AND_MEAN
Packit ae9e2a
#	define WIN32_LEAN_AND_MEAN
Packit ae9e2a
#endif
Packit ae9e2a
#	include <windows.h>
Packit ae9e2a
#elif defined(hpux) || defined(__hpux) || defined(_hpux)
Packit ae9e2a
#	include <sys/pstat.h>
Packit ae9e2a
#endif
Packit ae9e2a
Packit ae9e2a
/*
Packit ae9e2a
 * By doing this in two steps we can at least get
Packit ae9e2a
 * the function to be somewhat coherent, even
Packit ae9e2a
 * with this disgusting nest of #ifdefs.
Packit ae9e2a
 */
Packit ae9e2a
#ifndef _SC_NPROCESSORS_ONLN
Packit ae9e2a
#	ifdef _SC_NPROC_ONLN
Packit ae9e2a
#		define _SC_NPROCESSORS_ONLN _SC_NPROC_ONLN
Packit ae9e2a
#	elif defined _SC_CRAY_NCPU
Packit ae9e2a
#		define _SC_NPROCESSORS_ONLN _SC_CRAY_NCPU
Packit ae9e2a
#	endif
Packit ae9e2a
#endif
Packit ae9e2a
Packit ae9e2a
int git_online_cpus(void)
Packit ae9e2a
{
Packit ae9e2a
#ifdef _SC_NPROCESSORS_ONLN
Packit ae9e2a
	long ncpus;
Packit ae9e2a
#endif
Packit ae9e2a
Packit ae9e2a
#ifdef _WIN32
Packit ae9e2a
	SYSTEM_INFO info;
Packit ae9e2a
	GetSystemInfo(&info;;
Packit ae9e2a
Packit ae9e2a
	if ((int)info.dwNumberOfProcessors > 0)
Packit ae9e2a
		return (int)info.dwNumberOfProcessors;
Packit ae9e2a
#elif defined(hpux) || defined(__hpux) || defined(_hpux)
Packit ae9e2a
	struct pst_dynamic psd;
Packit ae9e2a
Packit ae9e2a
	if (!pstat_getdynamic(&psd, sizeof(psd), (size_t)1, 0))
Packit ae9e2a
		return (int)psd.psd_proc_cnt;
Packit ae9e2a
#endif
Packit ae9e2a
Packit ae9e2a
#ifdef _SC_NPROCESSORS_ONLN
Packit ae9e2a
	if ((ncpus = (long)sysconf(_SC_NPROCESSORS_ONLN)) > 0)
Packit ae9e2a
		return (int)ncpus;
Packit ae9e2a
#endif
Packit ae9e2a
Packit ae9e2a
	return 1;
Packit ae9e2a
}