lib/nproc.c

   1 /* Detect the number of processors.
   2
   3    Copyright (C) 2009 Free Software Foundation, Inc.
   4
   5    This program is free software; you can redistribute it and/or modify
   6    it under the terms of the GNU General Public License as published by
   7    the Free Software Foundation; either version 2, or (at your option)
   8    any later version.
   9
  10    This program is distributed in the hope that it will be useful,
  11    but WITHOUT ANY WARRANTY; without even the implied warranty of
  12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13    GNU General Public License for more details.
  14
  15    You should have received a copy of the GNU General Public License
  16    along with this program; if not, write to the Free Software Foundation,
  17    Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.  */
  18
  19 /* Written by Glen Lenker.  */
  20
  21 #include <config.h>
  22 #include "nproc.h"
  23
  24 #include <unistd.h>
  25
  26 #include <sys/types.h>
  27
  28 #if HAVE_SYS_PARAM_H
  29 # include <sys/param.h>
  30 #endif
  31
  32 #if HAVE_SYS_SYSCTL_H
  33 # include <sys/sysctl.h>
  34 #endif
  35
  36 #define ARRAY_SIZE(a) (sizeof (a) / sizeof ((a)[0]))
  37
  38 /* Return the total number of processors.  The result is guaranteed to
  39    be at least 1.  */
  40 unsigned long int
  41 num_processors (void)
  42 {
  43 #if defined _SC_NPROCESSORS_ONLN
  44   { /* This works on glibc, MacOS X 10.5, FreeBSD, AIX, OSF/1, Solaris, Cygwin,
  45        Haiku.  */
  46     long int nprocs = sysconf (_SC_NPROCESSORS_ONLN);
  47     if (0 < nprocs)
  48       return nprocs;
  49   }
  50 #endif
  51
  52 #if HAVE_SYSCTL && defined HW_NCPU
  53   { /* This works on MacOS X, FreeBSD, NetBSD, OpenBSD.  */
  54     int nprocs;
  55     size_t len = sizeof (nprocs);
  56     static int mib[2] = { CTL_HW, HW_NCPU };
  57
  58     if (sysctl (mib, ARRAY_SIZE (mib), &nprocs, &len, NULL, 0) == 0
  59         && len == sizeof (nprocs)
  60         && 0 < nprocs)
  61       return nprocs;
  62   }
  63 #endif
  64
  65   return 1;
  66 }