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_PSTAT_H
  29 # include <sys/pstat.h>
  30 #endif
  31
  32 #if HAVE_SYS_SYSMP_H
  33 # include <sys/sysmp.h>
  34 #endif
  35
  36 #if HAVE_SYS_PARAM_H
  37 # include <sys/param.h>
  38 #endif
  39
  40 #if HAVE_SYS_SYSCTL_H
  41 # include <sys/sysctl.h>
  42 #endif
  43
  44 #define ARRAY_SIZE(a) (sizeof (a) / sizeof ((a)[0]))
  45
  46 /* Return the total number of processors.  The result is guaranteed to
  47    be at least 1.  */
  48 unsigned long int
  49 num_processors (void)
  50 {
  51 #if defined _SC_NPROCESSORS_ONLN
  52   { /* This works on glibc, MacOS X 10.5, FreeBSD, AIX, OSF/1, Solaris, Cygwin,
  53        Haiku.  */
  54     long int nprocs = sysconf (_SC_NPROCESSORS_ONLN);
  55     if (0 < nprocs)
  56       return nprocs;
  57   }
  58 #endif
  59
  60 #if HAVE_PSTAT_GETDYNAMIC
  61   { /* This works on HP-UX.  */
  62     struct pst_dynamic psd;
  63     if (0 <= pstat_getdynamic (&psd, sizeof psd, 1, 0)
  64         && 0 < psd.psd_proc_cnt)
  65       return psd.psd_proc_cnt;
  66   }
  67 #endif
  68
  69 #if HAVE_SYSMP && defined MP_NAPROCS
  70   { /* This works on IRIX.  */
  71     /* MP_NPROCS yields the number of installed processors.
  72        MP_NAPROCS yields the number of processors available to unprivileged
  73        processes.  We need the latter.  */
  74     int nprocs = sysmp (MP_NAPROCS);
  75     if (0 < nprocs)
  76       return nprocs;
  77   }
  78 #endif
  79
  80 #if HAVE_SYSCTL && defined HW_NCPU
  81   { /* This works on MacOS X, FreeBSD, NetBSD, OpenBSD.  */
  82     int nprocs;
  83     size_t len = sizeof (nprocs);
  84     static int mib[2] = { CTL_HW, HW_NCPU };
  85
  86     if (sysctl (mib, ARRAY_SIZE (mib), &nprocs, &len, NULL, 0) == 0
  87         && len == sizeof (nprocs)
  88         && 0 < nprocs)
  89       return nprocs;
  90   }
  91 #endif
  92
  93   return 1;
  94 }