1/*
2 * Copyright (C) 2014 Apple Inc. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
6 * are met:
7 * 1. Redistributions of source code must retain the above copyright
8 * notice, this list of conditions and the following disclaimer.
9 * 2. Redistributions in binary form must reproduce the above copyright
10 * notice, this list of conditions and the following disclaimer in the
11 * documentation and/or other materials provided with the distribution.
12 *
13 * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
14 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR
17 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24 */
25
26#include "CPUCount.h"
27#include <stdlib.h>
28#include <sys/param.h>
29#include <sys/types.h>
30#include <unistd.h>
31
32#ifdef __APPLE__
33#include <sys/sysctl.h>
34#endif
35
36static size_t count;
37
38size_t cpuCount()
39{
40 if (count)
41 return count;
42
43#ifdef __APPLE__
44 size_t length = sizeof(count);
45 int name[] = {
46 CTL_HW,
47 HW_NCPU
48 };
49 int sysctlResult = sysctl(name, sizeof(name) / sizeof(int), &count, &length, 0, 0);
50 if (sysctlResult < 0)
51 abort();
52#else
53 long sysconfResult = sysconf(_SC_NPROCESSORS_ONLN);
54 if (sysconfResult < 0)
55 abort();
56 count = sysconfResult;
57#endif
58
59 return count;
60}
61