]> sourceware.org Git - glibc.git/commitdiff
linux: Use sched_getaffinity for __get_nprocs (BZ #27645)
authorAdhemerval Zanella <adhemerval.zanella@linaro.org>
Thu, 25 Mar 2021 12:30:07 +0000 (09:30 -0300)
committerAdhemerval Zanella <adhemerval.zanella@linaro.org>
Fri, 7 May 2021 16:54:09 +0000 (13:54 -0300)
Both the sysfs and procfs parsing (through GET_NPROCS_PARSER) are
removed in favor the syscall.  The initial scratch buffer should
fit to most of the common usage (1024 bytes with maps to 8192 CPUs).

Checked on x86_64-linux-gnu and aarch64-linux-gnu.

include/bits/cpu-set.h
posix/sched_cpucount.c
sysdeps/unix/sysv/linux/alpha/getsysstats.c
sysdeps/unix/sysv/linux/getsysstats.c
sysdeps/unix/sysv/linux/m68k/getsysstats.c [deleted file]
sysdeps/unix/sysv/linux/microblaze/getsysstats.c [deleted file]
sysdeps/unix/sysv/linux/mips/getsysstats.c [deleted file]
sysdeps/unix/sysv/linux/sparc/getsysstats.c

index 388f03cfbd16fc749f2b65a561c254cbc3f925e7..05aa0a8cf90fb398073190abc6c8a4b816ba47c5 100644 (file)
@@ -1 +1,9 @@
+#ifndef _BITS_CPU_SET_H
 #include <posix/bits/cpu-set.h>
+
+#ifndef _ISOMAC
+int __sched_cpucount (size_t __setsize, const cpu_set_t *__setp);
+libc_hidden_proto (__sched_cpucount)
+#endif
+
+#endif /* _BITS_CPU_SET_H  */
index 63d0e99590b3ecc02b45d9f7805911ab9ce4278d..95c125f8a4cbc5461f1ec56034dd5d8246ca69f4 100644 (file)
@@ -38,3 +38,4 @@ __sched_cpucount (size_t setsize, const cpu_set_t *setp)
     s += countbits (setp->__bits[i]);
   return s;
 }
+libc_hidden_def (__sched_cpucount)
index bdf794703a8bf9f2576084bdeea30627c6de3fc9..098575faef4b96143fce6ea5ba9ac3307d1fa145 100644 (file)
    <https://www.gnu.org/licenses/>.  */
 
 
-/* We need to define a special parser for /proc/cpuinfo.  */
-#define GET_NPROCS_PARSER(FD, BUFFER, CP, RE, BUFFER_END, RESULT)         \
-  do                                                                      \
-    {                                                                     \
-      /* Find the line that contains the information about the number of   \
-        active cpus.  We don't have to fear extremely long lines since    \
-        the kernel will not generate them.  8192 bytes are really enough. \
-        If there is no "CPUs ..." line then we are on a UP system.  */    \
-      char *l;                                                            \
-      (RESULT) = 1;                                                       \
-      while ((l = next_line (FD, BUFFER, &CP, &RE, BUFFER_END)) != NULL)  \
-       if ((sscanf (BUFFER, "cpus active : %d", &(RESULT)) == 1)          \
-           || (sscanf (BUFFER, "CPUs probed %*d active %d",               \
-                       &(RESULT)) == 1))                                  \
-         break;                                                           \
-    }                                                                     \
-  while (0)
-
-
 /* On the Alpha we can distinguish between the number of configured and
    active cpus.  */
 #define GET_NPROCS_CONF_PARSER(FP, BUFFER, RESULT)                        \
index 2c7f40998a57c8900f0996b0221f09a893698476..39e447b78d030ea556997bf9ffa6341586eeb9da 100644 (file)
    License along with the GNU C Library; if not, see
    <https://www.gnu.org/licenses/>.  */
 
-#include <alloca.h>
-#include <assert.h>
-#include <ctype.h>
 #include <dirent.h>
-#include <errno.h>
-#include <fcntl.h>
-#include <mntent.h>
-#include <paths.h>
+#include <not-cancel.h>
+#include <scratch_buffer.h>
 #include <stdio.h>
 #include <stdio_ext.h>
-#include <stdlib.h>
-#include <string.h>
-#include <unistd.h>
 #include <sys/sysinfo.h>
-
-#include <atomic.h>
-#include <not-cancel.h>
-
-
-/* How we can determine the number of available processors depends on
-   the configuration.  There is currently (as of version 2.0.21) no
-   system call to determine the number.  It is planned for the 2.1.x
-   series to add this, though.
-
-   One possibility to implement it for systems using Linux 2.0 is to
-   examine the pseudo file /proc/cpuinfo.  Here we have one entry for
-   each processor.
-
-   But not all systems have support for the /proc filesystem.  If it
-   is not available we simply return 1 since there is no way.  */
-
-
-/* Other architectures use different formats for /proc/cpuinfo.  This
-   provides a hook for alternative parsers.  */
-#ifndef GET_NPROCS_PARSER
-# define GET_NPROCS_PARSER(FD, BUFFER, CP, RE, BUFFER_END, RESULT) \
-  do                                                                   \
-    {                                                                  \
-      (RESULT) = 0;                                                    \
-      /* Read all lines and count the lines starting with the string   \
-        "processor".  We don't have to fear extremely long lines since \
-        the kernel will not generate them.  8192 bytes are really      \
-        enough.  */                                                    \
-      char *l;                                                         \
-      while ((l = next_line (FD, BUFFER, &CP, &RE, BUFFER_END)) != NULL) \
-       if (strncmp (l, "processor", 9) == 0)                           \
-         ++(RESULT);                                                   \
-    }                                                                  \
-  while (0)
-#endif
-
-
-static char *
-next_line (int fd, char *const buffer, char **cp, char **re,
-          char *const buffer_end)
-{
-  char *res = *cp;
-  char *nl = memchr (*cp, '\n', *re - *cp);
-  if (nl == NULL)
-    {
-      if (*cp != buffer)
-       {
-         if (*re == buffer_end)
-           {
-             memmove (buffer, *cp, *re - *cp);
-             *re = buffer + (*re - *cp);
-             *cp = buffer;
-
-             ssize_t n = __read_nocancel (fd, *re, buffer_end - *re);
-             if (n < 0)
-               return NULL;
-
-             *re += n;
-
-             nl = memchr (*cp, '\n', *re - *cp);
-             while (nl == NULL && *re == buffer_end)
-               {
-                 /* Truncate too long lines.  */
-                 *re = buffer + 3 * (buffer_end - buffer) / 4;
-                 n = __read_nocancel (fd, *re, buffer_end - *re);
-                 if (n < 0)
-                   return NULL;
-
-                 nl = memchr (*re, '\n', n);
-                 **re = '\n';
-                 *re += n;
-               }
-           }
-         else
-           nl = memchr (*cp, '\n', *re - *cp);
-
-         res = *cp;
-       }
-
-      if (nl == NULL)
-       nl = *re - 1;
-    }
-
-  *cp = nl + 1;
-  assert (*cp <= *re);
-
-  return res == *re ? NULL : res;
-}
-
+#include <sysdep.h>
 
 int
 __get_nprocs (void)
 {
-  static int cached_result = -1;
-  static time_t timestamp;
-
-  time_t now = time_now ();
-  time_t prev = timestamp;
-  atomic_read_barrier ();
-  if (now == prev && cached_result > -1)
-    return cached_result;
-
-  /* XXX Here will come a test for the new system call.  */
-
-  const size_t buffer_size = __libc_use_alloca (8192) ? 8192 : 512;
-  char *buffer = alloca (buffer_size);
-  char *buffer_end = buffer + buffer_size;
-  char *cp = buffer_end;
-  char *re = buffer_end;
+  struct scratch_buffer set;
+  scratch_buffer_init (&set);
 
-  const int flags = O_RDONLY | O_CLOEXEC;
-  /* This file contains comma-separated ranges.  */
-  int fd = __open_nocancel ("/sys/devices/system/cpu/online", flags);
-  char *l;
-  int result = 0;
-  if (fd != -1)
+  int r;
+  while (true)
     {
-      l = next_line (fd, buffer, &cp, &re, buffer_end);
-      if (l != NULL)
-       do
-         {
-           char *endp;
-           unsigned long int n = strtoul (l, &endp, 10);
-           if (l == endp)
-             {
-               result = 0;
-               break;
-             }
-
-           unsigned long int m = n;
-           if (*endp == '-')
-             {
-               l = endp + 1;
-               m = strtoul (l, &endp, 10);
-               if (l == endp)
-                 {
-                   result = 0;
-                   break;
-                 }
-             }
-
-           result += m - n + 1;
-
-           l = endp;
-           if (l < re && *l == ',')
-             ++l;
-         }
-       while (l < re && *l != '\n');
-
-      __close_nocancel_nostatus (fd);
-
-      if (result > 0)
-       goto out;
+      /* The possible error are EFAULT for an invalid buffer or ESRCH for
+        invalid pid, none could happen.  */
+      r = INTERNAL_SYSCALL_CALL (sched_getaffinity, 0, set.length,
+                                 set.data);
+      if (r > 0)
+        break;
+
+      if (!scratch_buffer_grow (&set))
+        /* Default to an SMP system in case we cannot obtain an accurate
+           number.  */
+        return 2;
     }
 
-  cp = buffer_end;
-  re = buffer_end;
-
-  /* Default to an SMP system in case we cannot obtain an accurate
-     number.  */
-  result = 2;
+  /* The scratch_buffer is aligned to max_align_t.  */
+  r = __sched_cpucount (r, (const cpu_set_t *) set.data);
 
-  /* The /proc/stat format is more uniform, use it by default.  */
-  fd = __open_nocancel ("/proc/stat", flags);
-  if (fd != -1)
-    {
-      result = 0;
-
-      while ((l = next_line (fd, buffer, &cp, &re, buffer_end)) != NULL)
-       /* The current format of /proc/stat has all the cpu* entries
-          at the front.  We assume here that stays this way.  */
-       if (strncmp (l, "cpu", 3) != 0)
-         break;
-       else if (isdigit (l[3]))
-         ++result;
+  scratch_buffer_free (&set);
 
-      __close_nocancel_nostatus (fd);
-    }
-  else
-    {
-      fd = __open_nocancel ("/proc/cpuinfo", flags);
-      if (fd != -1)
-       {
-         GET_NPROCS_PARSER (fd, buffer, cp, re, buffer_end, result);
-         __close_nocancel_nostatus (fd);
-       }
-    }
-
- out:
-  cached_result = result;
-  atomic_write_barrier ();
-  timestamp = now;
-
-  return result;
+  return r;
 }
 libc_hidden_def (__get_nprocs)
 weak_alias (__get_nprocs, get_nprocs)
diff --git a/sysdeps/unix/sysv/linux/m68k/getsysstats.c b/sysdeps/unix/sysv/linux/m68k/getsysstats.c
deleted file mode 100644 (file)
index 6915f6d..0000000
+++ /dev/null
@@ -1,37 +0,0 @@
-/* Determine various system internal values, Linux/m68k version.
-   Copyright (C) 2003-2021 Free Software Foundation, Inc.
-   This file is part of the GNU C Library.
-   Contributed by Andreas Schwab <schwab@suse.de>
-
-   The GNU C Library is free software; you can redistribute it and/or
-   modify it under the terms of the GNU Lesser General Public
-   License as published by the Free Software Foundation; either
-   version 2.1 of the License, or (at your option) any later version.
-
-   The GNU C Library 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
-   Lesser General Public License for more details.
-
-   You should have received a copy of the GNU Lesser General Public
-   License along with the GNU C Library.  If not, see
-   <https://www.gnu.org/licenses/>.  */
-
-
-/* We need to define a special parser for /proc/cpuinfo.  */
-#define GET_NPROCS_PARSER(FD, BUFFER, CP, RE, BUFFER_END, RESULT)        \
-  do                                                                     \
-    {                                                                    \
-      (RESULT) = 0;                                                      \
-      /* Read all lines and count the lines starting with the string     \
-        "CPU:".  We don't have to fear extremely long lines since        \
-        the kernel will not generate them.  8192 bytes are really        \
-        enough.  */                                                      \
-      char *l;                                                           \
-      while ((l = next_line (FD, BUFFER, &CP, &RE, BUFFER_END)) != NULL)  \
-       if (strncmp (l, "CPU:", 4) == 0)                                  \
-         ++(RESULT);                                                     \
-    }                                                                    \
-  while (0)
-
-#include <sysdeps/unix/sysv/linux/getsysstats.c>
diff --git a/sysdeps/unix/sysv/linux/microblaze/getsysstats.c b/sysdeps/unix/sysv/linux/microblaze/getsysstats.c
deleted file mode 100644 (file)
index 46403e7..0000000
+++ /dev/null
@@ -1,34 +0,0 @@
-/* Copyright (C) 1997-2021 Free Software Foundation, Inc.
-
-   This file is part of the GNU C Library.
-
-   The GNU C Library is free software; you can redistribute it and/or
-   modify it under the terms of the GNU Lesser General Public License as
-   published by the Free Software Foundation; either version 2.1 of the
-   License, or (at your option) any later version.
-
-   The GNU C Library 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
-   Lesser General Public License for more details.
-
-   You should have received a copy of the GNU Lesser General Public
-   License along with the GNU C Library; if not, see
-   <https://www.gnu.org/licenses/>.  */
-
-/* We need to define a special parser for /proc/cpuinfo.  */
-#define GET_NPROCS_PARSER(FD, BUFFER, CP, RE, BUFFER_END, RESULT)              \
-  do                                                                           \
-    {                                                                          \
-      (RESULT) = 0;                                                            \
-      /* Read all lines and count the lines starting with the string           \
-         "CPU-Family:".  We don't have to fear extremely long lines since      \
-         the kernel will not generate them.  8192 bytes are really enough.  */ \
-      char *l;                                                                 \
-      while ((l = next_line (FD, BUFFER, &CP, &RE, BUFFER_END)) != NULL)       \
-      if (strncmp (l, "CPU-Family:", 11) == 0)                                 \
-        ++(RESULT);                                                            \
-    }                                                                          \
-  while (0)
-
-#include <sysdeps/unix/sysv/linux/getsysstats.c>
diff --git a/sysdeps/unix/sysv/linux/mips/getsysstats.c b/sysdeps/unix/sysv/linux/mips/getsysstats.c
deleted file mode 100644 (file)
index 34f352e..0000000
+++ /dev/null
@@ -1,36 +0,0 @@
-/* Determine various system internal values, Linux/MIPS version.
-   Copyright (C) 2001-2021 Free Software Foundation, Inc.
-   This file is part of the GNU C Library.
-
-   The GNU C Library is free software; you can redistribute it and/or
-   modify it under the terms of the GNU Lesser General Public
-   License as published by the Free Software Foundation; either
-   version 2.1 of the License, or (at your option) any later version.
-
-   The GNU C Library 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
-   Lesser General Public License for more details.
-
-   You should have received a copy of the GNU Lesser General Public
-   License along with the GNU C Library.  If not, see
-   <https://www.gnu.org/licenses/>.  */
-
-
-/* We need to define a special parser for /proc/cpuinfo.  */
-#define GET_NPROCS_PARSER(FD, BUFFER, CP, RE, BUFFER_END, RESULT)        \
-  do                                                                     \
-    {                                                                    \
-      (RESULT) = 0;                                                      \
-      /* Read all lines and count the lines starting with the string     \
-        "cpu model".  We don't have to fear extremely long lines since   \
-        the kernel will not generate them.  8192 bytes are really        \
-        enough.  */                                                      \
-      char *l;                                                           \
-      while ((l = next_line (FD, BUFFER, &CP, &RE, BUFFER_END)) != NULL)  \
-       if (strncmp (l, "cpu model", 9) == 0)                             \
-         ++(RESULT);                                                     \
-    }                                                                    \
-  while (0)
-
-#include <sysdeps/unix/sysv/linux/getsysstats.c>
index 8e1760802f2a49bfe2ac57f23d7cb076c63dde1a..a8adba8321a18423ddc5b0d4783089e246b70299 100644 (file)
    <https://www.gnu.org/licenses/>.  */
 
 
-/* We need to define a special parser for /proc/cpuinfo.  */
-#define GET_NPROCS_PARSER(FD, BUFFER, CP, RE, BUFFER_END, RESULT)        \
-  do                                                                     \
-    {                                                                    \
-      (RESULT) = 0;                                                      \
-      /* Find the line that contains the information about the number of  \
-        active cpus.  We don't have to fear extremely long lines since   \
-        the kernel will not generate them.  8192 bytes are really        \
-        enough.  */                                                      \
-      char *l;                                                           \
-      while ((l = next_line (FD, BUFFER, &CP, &RE, BUFFER_END)) != NULL)  \
-       if (sscanf (l, "ncpus active : %d", &(RESULT)) == 1)              \
-         break;                                                          \
-    }                                                                    \
-  while (0)
-
-
 /* On the Sparc we can distinguish between the number of configured and
    active cpus.  */
 #define GET_NPROCS_CONF_PARSER(FP, BUFFER, RESULT)                      \
This page took 0.077616 seconds and 5 git commands to generate.