Home | History | Annotate | Download | only in wtf 
      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 "NumberOfCores.h"
     28 
     29 #if OS(DARWIN) || OS(OPENBSD) || OS(NETBSD) || OS(FREEBSD)
     30 #include <sys/param.h>
     31 // sys/types.h must come before sys/sysctl.h because the latter uses
     32 // data types defined in the former. See sysctl(3) and style(9).
     33 #include <sys/types.h>
     34 #include <sys/sysctl.h>
     35 #elif OS(LINUX) || OS(SOLARIS)
     36 #include <unistd.h>
     37 #elif OS(WINDOWS)
     38 #include "wtf/UnusedParam.h"
     39 #include <windows.h>
     40 #endif
     41 
     42 namespace WTF {
     43 
     44 int numberOfProcessorCores()
     45 {
     46     const int defaultIfUnavailable = 1;
     47     static int s_numberOfCores = -1;
     48 
     49     if (s_numberOfCores > 0)
     50         return s_numberOfCores;
     51 
     52 #if OS(DARWIN) || OS(OPENBSD) || OS(NETBSD) || OS(FREEBSD)
     53     unsigned result;
     54     size_t length = sizeof(result);
     55     int name[] = {
     56             CTL_HW,
     57             HW_NCPU
     58     };
     59     int sysctlResult = sysctl(name, sizeof(name) / sizeof(int), &result, &length, 0, 0);
     60 
     61     s_numberOfCores = sysctlResult < 0 ? defaultIfUnavailable : result;
     62 #elif OS(LINUX) || OS(SOLARIS)
     63     long sysconfResult = sysconf(_SC_NPROCESSORS_ONLN);
     64 
     65     s_numberOfCores = sysconfResult < 0 ? defaultIfUnavailable : static_cast<int>(sysconfResult);
     66 #elif OS(WINDOWS)
     67     UNUSED_PARAM(defaultIfUnavailable);
     68     SYSTEM_INFO sysInfo;
     69     GetSystemInfo(&sysInfo);
     70 
     71     s_numberOfCores = sysInfo.dwNumberOfProcessors;
     72 #else
     73     s_numberOfCores = defaultIfUnavailable;
     74 #endif
     75     return s_numberOfCores;
     76 }
     77 
     78 }
     79