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_PARAM_H
  33 # include <sys/param.h>
  34 #endif
  35
  36 #if HAVE_SYS_SYSCTL_H
  37 # include <sys/sysctl.h>
  38 #endif
  39
  40 #define ARRAY_SIZE(a) (sizeof (a) / sizeof ((a)[0]))
  41
  42 /* Return the total number of processors.  The result is guaranteed to
  43    be at least 1.  */
  44 unsigned long int
  45 num_processors (void)
  46 {
  47 #if defined _SC_NPROCESSORS_ONLN
  48   { /* This works on glibc, MacOS X 10.5, FreeBSD, AIX, OSF/1, Solaris, Cygwin,
  49        Haiku.  */
  50     long int nprocs = sysconf (_SC_NPROCESSORS_ONLN);
  51     if (0 < nprocs)
  52       return nprocs;
  53   }
  54 #endif
  55
  56 #if HAVE_PSTAT_GETDYNAMIC
  57   { /* This works on HP-UX.  */
  58     struct pst_dynamic psd;
  59     if (0 <= pstat_getdynamic (&psd, sizeof psd, 1, 0)
  60         && 0 < psd.psd_proc_cnt)
  61       return psd.psd_proc_cnt;
  62   }
  63 #endif
  64
  65 #if HAVE_SYSCTL && defined HW_NCPU
  66   { /* This works on MacOS X, FreeBSD, NetBSD, OpenBSD.  */
  67     int nprocs;
  68     size_t len = sizeof (nprocs);
  69     static int mib[2] = { CTL_HW, HW_NCPU };
  70
  71     if (sysctl (mib, ARRAY_SIZE (mib), &nprocs, &len, NULL, 0) == 0
  72         && len == sizeof (nprocs)
  73         && 0 < nprocs)
  74       return nprocs;
  75   }
  76 #endif
  77
  78   return 1;
  79 }