1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
|
/* Copyright (C) 2007-2013 Open Information Security Foundation
*
* You can copy, redistribute or modify this Program under the terms of
* the GNU General Public License version 2 as published by the Free
* Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* version 2 along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
* 02110-1301, USA.
*/
/**
* \file
*
* \author Pablo Rincon Crespo <pablo.rincon.crespo@gmail.com>
*
* Retrieve CPU information (configured CPUs, online CPUs)
*/
#include "suricata-common.h"
#include "util-error.h"
#include "util-debug.h"
#include "util-cpu.h"
#include "util-byte.h"
/**
* Ok, if they should use sysconf, check that they have the macro's
* (syscalls) defined;
*
* Note: For windows it's different; Check the following:
* SYSTEM_INFO info;
* GetSystemInfo(&info);
* -> info.dwNumberOfProcessors;
*/
#ifdef _SC_NPROCESSORS_CONF
#define SYSCONF_NPROCESSORS_CONF_COMPAT
#endif
#ifdef _SC_NPROCESSORS_ONLN
#define SYSCONF_NPROCESSORS_ONLN_COMPAT
#endif
/* This one is available on Solaris 10 */
#ifdef _SC_NPROCESSORS_MAX
#define SYSCONF_NPROCESSORS_MAX_COMPAT
#endif
/**
* \brief Get the number of cpus configured in the system
* \retval 0 if the syscall is not available or we have an error;
* otherwise it will return the number of cpus configured
*/
uint16_t UtilCpuGetNumProcessorsConfigured(void)
{
#ifdef SYSCONF_NPROCESSORS_CONF_COMPAT
long nprocs = -1;
nprocs = sysconf(_SC_NPROCESSORS_CONF);
if (nprocs < 1) {
SCLogError("Couldn't retrieve the number of cpus "
"configured (%s)",
strerror(errno));
return 0;
}
if (nprocs > UINT16_MAX) {
SCLogDebug("It seems that there are more than %d CPUs "
"configured on this system. You can modify util-cpu.{c,h} "
"to use uint32_t to support it", UINT16_MAX);
return UINT16_MAX;
}
return (uint16_t)nprocs;
#elif OS_WIN32
int64_t nprocs = 0;
const char* envvar = getenv("NUMBER_OF_PROCESSORS");
if (envvar != NULL) {
if (StringParseInt64(&nprocs, 10, 0, envvar) < 0) {
SCLogWarning("Invalid value for number of "
"processors: %s",
envvar);
return 0;
}
}
if (nprocs < 1) {
SCLogError("Couldn't retrieve the number of cpus "
"configured from the NUMBER_OF_PROCESSORS environment variable");
return 0;
}
return (uint16_t)nprocs;
#else
SCLogError("Couldn't retrieve the number of cpus "
"configured, sysconf macro unavailable");
return 0;
#endif
}
/**
* \brief Get the number of cpus online in the system
* \retval 0 if the syscall is not available or we have an error;
* otherwise it will return the number of cpus online
*/
uint16_t UtilCpuGetNumProcessorsOnline(void)
{
#ifdef SYSCONF_NPROCESSORS_ONLN_COMPAT
long nprocs = -1;
nprocs = sysconf(_SC_NPROCESSORS_ONLN);
if (nprocs < 1) {
SCLogError("Couldn't retrieve the number of cpus "
"online (%s)",
strerror(errno));
return 0;
}
if (nprocs > UINT16_MAX) {
SCLogDebug("It seems that there are more than %d CPUs online. "
"You can modify util-cpu.{c,h} to use uint32_t to "
"support it", UINT16_MAX);
return UINT16_MAX;
}
return (uint16_t)nprocs;
#elif OS_WIN32
return UtilCpuGetNumProcessorsConfigured();
#else
SCLogError("Couldn't retrieve the number of cpus online, "
"synconf macro unavailable");
return 0;
#endif
}
/**
* \brief Get the maximum number of cpus allowed in the system
* This syscall is present on Solaris, but it's not on linux
* or macosx. Maybe you should look at UtilCpuGetNumProcessorsConfigured()
* \retval 0 if the syscall is not available or we have an error;
* otherwise it will return the number of cpus allowed
*/
uint16_t UtilCpuGetNumProcessorsMax(void)
{
#ifdef SYSCONF_NPROCESSORS_MAX_COMPAT
long nprocs = -1;
nprocs = sysconf(_SC_NPROCESSORS_MAX);
if (nprocs < 1) {
SCLogError("Couldn't retrieve the maximum number of cpus "
"allowed by the system (%s)",
strerror(errno));
return 0;
}
if (nprocs > UINT16_MAX) {
SCLogDebug("It seems that the system support more that %"PRIu16" CPUs. You "
"can modify util-cpu.{c,h} to use uint32_t to support it", UINT16_MAX);
return UINT16_MAX;
}
return (uint16_t)nprocs;
#else
SCLogError("Couldn't retrieve the maximum number of cpus allowed by "
"the system, synconf macro unavailable");
return 0;
#endif
}
/**
* \brief Print a summary of CPUs detected (configured and online)
*/
void UtilCpuPrintSummary(void)
{
uint16_t cpus_conf = UtilCpuGetNumProcessorsConfigured();
uint16_t cpus_online = UtilCpuGetNumProcessorsOnline();
SCLogDebug("CPUs Summary: ");
if (cpus_conf > 0)
SCLogDebug("CPUs configured: %"PRIu16, cpus_conf);
if (cpus_online > 0)
SCLogInfo("CPUs/cores online: %"PRIu16, cpus_online);
if (cpus_online == 0 && cpus_conf == 0)
SCLogInfo("Couldn't retrieve any information of CPU's, please, send your operating "
"system info and check util-cpu.{c,h}");
}
/**
* Get the current number of ticks from the CPU.
*
* \todo We'll have to deal with removing ticks from the extra cpuids in between
* 2 calls.
*/
uint64_t UtilCpuGetTicks(void)
{
uint64_t val;
#if defined(__GNUC__) && (defined(__x86_64) || defined(_X86_64_) || defined(ia_64) || defined(__i386__))
#if defined(__x86_64) || defined(_X86_64_) || defined(ia_64)
__asm__ __volatile__ (
"xorl %%eax,%%eax\n\t"
"cpuid\n\t"
::: "%rax", "%rbx", "%rcx", "%rdx");
#else
__asm__ __volatile__ (
"xorl %%eax,%%eax\n\t"
"pushl %%ebx\n\t"
"cpuid\n\t"
"popl %%ebx\n\t"
::: "%eax", "%ecx", "%edx");
#endif
uint32_t a, d;
__asm__ __volatile__ ("rdtsc" : "=a" (a), "=d" (d));
val = ((uint64_t)a) | (((uint64_t)d) << 32);
#if defined(__x86_64) || defined(_X86_64_) || defined(ia_64)
__asm__ __volatile__ (
"xorl %%eax,%%eax\n\t"
"cpuid\n\t"
::: "%rax", "%rbx", "%rcx", "%rdx");
#else
__asm__ __volatile__ (
"xorl %%eax,%%eax\n\t"
"pushl %%ebx\n\t"
"cpuid\n\t"
"popl %%ebx\n\t"
::: "%eax", "%ecx", "%edx");
#endif
#else /* #if defined(__GNU__) */
//#warning Using inferior version of UtilCpuGetTicks
struct timeval now;
gettimeofday(&now, NULL);
val = (now.tv_sec * 1000000) + now.tv_usec;
#endif
return val;
}
|
