penguin/tinyproxy

src/compat_cpu.c

raw · 706 bytes

#include "compat_cpu.h"

#include <limits.h>

#ifdef _WIN32
#include <windows.h>
#else
#include <unistd.h>

#include <time.h>
#include <errno.h>
#endif

int compat_cpu_count(void)
{
#ifdef _WIN32
	SYSTEM_INFO si;

	GetSystemInfo(&si);

	if (si.dwNumberOfProcessors == 0 ||
		si.dwNumberOfProcessors > INT_MAX) {
		return 1;
	}

	return (int)si.dwNumberOfProcessors;
#else
	long n = sysconf(_SC_NPROCESSORS_ONLN);

	if (n < 1 || n > INT_MAX) {
		return 1;
	}

	return (int)n;
#endif
}

void sleep_ms(unsigned int ms)
{
#ifdef _WIN32
	Sleep(ms);
#else
	struct timespec ts;

	ts.tv_sec = ms / 1000;
	ts.tv_nsec = (long)(ms % 1000) * 1000000L;

	while (nanosleep(&ts, &ts) < 0 && errno == EINTR) {
	}
#endif
}