[WTF] Add environment variable helpers
[WebKit-https.git] / Source / WTF / wtf / NumberOfCores.cpp
1 /*
2  * Copyright (C) 2012 University of Szeged. 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 UNIVERSITY OF SZEGED ``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 UNIVERSITY OF SZEGED 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 "config.h"
27 #include <wtf/NumberOfCores.h>
28
29 #include <cstdio>
30 #include <wtf/Environment.h>
31
32 #if OS(DARWIN)
33 #include <sys/param.h>
34 // sys/types.h must come before sys/sysctl.h because the latter uses
35 // data types defined in the former. See sysctl(3) and style(9).
36 #include <sys/types.h>
37 #include <sys/sysctl.h>
38 #elif OS(LINUX) || OS(AIX) || OS(OPENBSD) || OS(NETBSD) || OS(FREEBSD)
39 #include <unistd.h>
40 #elif OS(WINDOWS)
41 #include <windows.h>
42 #endif
43
44 namespace WTF {
45
46 int numberOfProcessorCores()
47 {
48     const int defaultIfUnavailable = 1;
49     static int s_numberOfCores = -1;
50
51     if (s_numberOfCores > 0)
52         return s_numberOfCores;
53
54     if (auto coresEnv = Environment::get("WTF_numberOfProcessorCores")) {
55         bool ok;
56         auto numberOfCores = coresEnv->toUIntStrict(&ok);
57         if (ok) {
58             s_numberOfCores = numberOfCores;
59             return s_numberOfCores;
60         }
61
62         fprintf(stderr, "WARNING: failed to parse WTF_numberOfProcessorCores=%s\n", coresEnv->utf8().data());
63     }
64
65 #if OS(DARWIN)
66     unsigned result;
67     size_t length = sizeof(result);
68     int name[] = {
69             CTL_HW,
70             HW_AVAILCPU
71     };
72     int sysctlResult = sysctl(name, sizeof(name) / sizeof(int), &result, &length, 0, 0);
73
74     s_numberOfCores = sysctlResult < 0 ? defaultIfUnavailable : result;
75 #elif OS(LINUX) || OS(AIX) || OS(OPENBSD) || OS(NETBSD) || OS(FREEBSD)
76     long sysconfResult = sysconf(_SC_NPROCESSORS_ONLN);
77
78     s_numberOfCores = sysconfResult < 0 ? defaultIfUnavailable : static_cast<int>(sysconfResult);
79 #elif OS(WINDOWS)
80     UNUSED_PARAM(defaultIfUnavailable);
81     SYSTEM_INFO sysInfo;
82     GetSystemInfo(&sysInfo);
83
84     s_numberOfCores = sysInfo.dwNumberOfProcessors;
85 #else
86     s_numberOfCores = defaultIfUnavailable;
87 #endif
88     return s_numberOfCores;
89 }
90
91 }