]> granicus.if.org Git - sysstat/blobdiff - count.c
sar/sadc: Dereference nr array pointer in struct activity
[sysstat] / count.c
diff --git a/count.c b/count.c
index 116ca996ab60fd3d436c1c1202cb2b3a917333a8..0bc57121c330fa3983ed2220ff08e0b28ceb5177 100644 (file)
--- a/count.c
+++ b/count.c
@@ -1,6 +1,6 @@
 /*
  * count.c: Count items for which statistics will be collected.
- * (C) 1999-2013 by Sebastien GODARD (sysstat <at> orange.fr)
+ * (C) 1999-2020 by Sebastien GODARD (sysstat <at> orange.fr)
  *
  ***************************************************************************
  * This program is free software; you can redistribute it and/or modify it *
@@ -15,7 +15,7 @@
  *                                                                         *
  * You should have received a copy of the GNU General Public License along *
  * with this program; if not, write to the Free Software Foundation, Inc., *
- * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA                   *
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1335 USA              *
  ***************************************************************************
  */
 
@@ -27,7 +27,7 @@
 #include <ctype.h>
 #include <sys/types.h>
 #include <sys/stat.h>
-#include <sys/vfs.h>
+#include <sys/statvfs.h>
 #include <unistd.h>
 
 #include "common.h"
@@ -50,7 +50,7 @@
  * @highest    If set to TRUE, then look for the highest processor number.
  *             This is used when eg. the machine has 4 CPU numbered 0, 1, 4
  *             and 5. In this case, this procedure will return 6.
- * 
+ *
  * RETURNS:
  * Number of processors (online and offline).
  * A value of 0 means that /sys was not mounted.
@@ -73,8 +73,8 @@ int get_sys_cpu_nr(int highest)
        while ((drd = readdir(dir)) != NULL) {
 
                if (!strncmp(drd->d_name, "cpu", 3) && isdigit(drd->d_name[3])) {
-                       snprintf(line, MAX_PF_NAME, "%s/%s", SYSFS_DEVCPU, drd->d_name);
-                       line[MAX_PF_NAME - 1] = '\0';
+                       snprintf(line, sizeof(line), "%s/%s", SYSFS_DEVCPU, drd->d_name);
+                       line[sizeof(line) - 1] = '\0';
                        if (stat(line, &buf) < 0)
                                continue;
                        if (S_ISDIR(buf.st_mode)) {
@@ -119,7 +119,7 @@ int get_proc_cpu_nr(void)
                exit(1);
        }
 
-       while (fgets(line, 16, fp) != NULL) {
+       while (fgets(line, sizeof(line), fp) != NULL) {
 
                if (strncmp(line, "cpu ", 4) && !strncmp(line, "cpu", 3)) {
                        sscanf(line + 3, "%d", &num_proc);
@@ -131,7 +131,8 @@ int get_proc_cpu_nr(void)
 
        fclose(fp);
 
-       return (proc_nr + 1);
+       proc_nr++;
+       return proc_nr;
 }
 
 /*
@@ -155,9 +156,9 @@ int get_proc_cpu_nr(void)
  * 2: two proc...
  ***************************************************************************
  */
-int get_cpu_nr(unsigned int max_nr_cpus, int highest)
+__nr_t get_cpu_nr(unsigned int max_nr_cpus, int highest)
 {
-       int cpu_nr;
+       __nr_t cpu_nr;
 
        if ((cpu_nr = get_sys_cpu_nr(highest)) == 0) {
                /* /sys may be not mounted. Use /proc/stat instead */
@@ -184,14 +185,14 @@ int get_cpu_nr(unsigned int max_nr_cpus, int highest)
  * @cpu_nr             Number of processors.
  *
  * RETURNS:
- * Number of interrupts per processor + a pre-allocation constant.
+ * Number of interrupts per processor.
  ***************************************************************************
  */
-int get_irqcpu_nr(char *file, int max_nr_irqcpu, int cpu_nr)
+__nr_t get_irqcpu_nr(char *file, int max_nr_irqcpu, int cpu_nr)
 {
        FILE *fp;
        char *line = NULL;
-       unsigned int irq = 0;
+       __nr_t irq = 0;
        int p;
 
        if ((fp = fopen(file, "r")) == NULL)
@@ -228,12 +229,13 @@ int get_irqcpu_nr(char *file, int max_nr_irqcpu, int cpu_nr)
  * Number of devices (and partitions).
  ***************************************************************************
  */
-int get_diskstats_dev_nr(int count_part, int only_used_dev)
+__nr_t get_diskstats_dev_nr(int count_part, int only_used_dev)
 {
        FILE *fp;
        char line[256];
        char dev_name[MAX_NAME_LEN];
-       int dev = 0, i;
+       __nr_t dev = 0;
+       int i;
        unsigned long rd_ios, wr_ios;
 
        if ((fp = fopen(DISKSTATS, "r")) == NULL)
@@ -244,11 +246,11 @@ int get_diskstats_dev_nr(int count_part, int only_used_dev)
         * Counting devices and partitions is simply a matter of counting
         * the number of lines...
         */
-       while (fgets(line, 256, fp) != NULL) {
+       while (fgets(line, sizeof(line), fp) != NULL) {
                if (!count_part) {
                        i = sscanf(line, "%*d %*d %s %lu %*u %*u %*u %lu",
                                   dev_name, &rd_ios, &wr_ios);
-                       if ((i == 2) || !is_device(dev_name, ACCEPT_VIRTUAL_DEVICES))
+                       if ((i == 2) || !is_device(SLASH_SYS, dev_name, ACCEPT_VIRTUAL_DEVICES))
                                /* It was a partition and not a device */
                                continue;
                        if (only_used_dev && !rd_ios && !wr_ios)
@@ -274,17 +276,17 @@ int get_diskstats_dev_nr(int count_part, int only_used_dev)
  * Number of interrupts, including total number of interrupts.
  ***************************************************************************
  */
-int get_irq_nr(void)
+__nr_t get_irq_nr(void)
 {
        FILE *fp;
        char line[8192];
-       int in = 0;
+       __nr_t in = 0;
        int pos = 4;
 
        if ((fp = fopen(STAT, "r")) == NULL)
                return 0;
 
-       while (fgets(line, 8192, fp) != NULL) {
+       while (fgets(line, sizeof(line), fp) != NULL) {
 
                if (!strncmp(line, "intr ", 5)) {
 
@@ -309,16 +311,16 @@ int get_irq_nr(void)
  * Number of serial lines supporting tx/rx accouting.
  ***************************************************************************
  */
-int get_serial_nr(void)
+__nr_t get_serial_nr(void)
 {
        FILE *fp;
        char line[256];
-       int sl = 0;
+       __nr_t sl = 0;
 
        if ((fp = fopen(SERIAL, "r")) == NULL)
                return 0;       /* No SERIAL file */
 
-       while (fgets(line, 256, fp) != NULL) {
+       while (fgets(line, sizeof(line), fp) != NULL) {
                /*
                 * tx/rx statistics are always present,
                 * except when serial line is unknown.
@@ -342,16 +344,16 @@ int get_serial_nr(void)
  * Number of network interfaces.
  ***************************************************************************
  */
-int get_iface_nr(void)
+__nr_t get_iface_nr(void)
 {
        FILE *fp;
        char line[128];
-       int iface = 0;
+       __nr_t iface = 0;
 
        if ((fp = fopen(NET_DEV, "r")) == NULL)
                return 0;       /* No network device file */
 
-       while (fgets(line, 128, fp) != NULL) {
+       while (fgets(line, sizeof(line), fp) != NULL) {
                if (strchr(line, ':')) {
                        iface++;
                }
@@ -374,9 +376,9 @@ int get_iface_nr(void)
  * Number of devices.
  ***************************************************************************
  */
-int get_disk_nr(unsigned int f)
+__nr_t get_disk_nr(unsigned int f)
 {
-       int disk_nr;
+       __nr_t disk_nr;
 
        /*
         * Partitions are taken into account by sar -d only with
@@ -395,19 +397,19 @@ int get_disk_nr(unsigned int f)
  * Number of frequencies.
  ***************************************************************************
  */
-int get_freq_nr(void)
+__nr_t get_freq_nr(void)
 {
        FILE *fp;
        char filename[MAX_PF_NAME];
        char line[128];
-       int freq = 0;
+       __nr_t freq = 0;
 
        snprintf(filename, MAX_PF_NAME, "%s/cpu0/%s",
                 SYSFS_DEVCPU, SYSFS_TIME_IN_STATE);
        if ((fp = fopen(filename, "r")) == NULL)
                return 0;       /* No time_in_state file for CPU#0 */
 
-       while (fgets(line, 128, fp) != NULL) {
+       while (fgets(line, sizeof(line), fp) != NULL) {
                freq++;
        }
 
@@ -426,11 +428,11 @@ int get_freq_nr(void)
  * Return -1 if directory doesn't exist in sysfs.
  ***************************************************************************
  */
-int get_usb_nr(void)
+__nr_t get_usb_nr(void)
 {
        DIR *dir;
        struct dirent *drd;
-       int usb = 0;
+       __nr_t usb = 0;
 
        /* Open relevant /sys directory */
        if ((dir = opendir(SYSFS_USBDEV)) == NULL)
@@ -458,31 +460,61 @@ int get_usb_nr(void)
  * Number of filesystems.
  ***************************************************************************
  */
-int get_filesystem_nr(void)
+__nr_t get_filesystem_nr(void)
 {
        FILE *fp;
-       char line[256], fs_name[MAX_FS_LEN], mountp[128];
-       int fs = 0;
-       struct statfs buf;
+       char line[512], fs_name[MAX_FS_LEN], mountp[256], type[128];
+       char *pos = 0, *pos2 = 0;
+       __nr_t fs = 0;
+       int skip = 0, skip_next = 0;
+       struct statvfs buf;
 
        if ((fp = fopen(MTAB, "r")) == NULL)
                /* File non-existent */
                return 0;
 
        /* Get current filesystem */
-       while (fgets(line, 256, fp) != NULL) {
+       while (fgets(line, sizeof(line), fp) != NULL) {
+               /*
+                * Ignore line if the preceding line did not contain '\n'.
+                * (Some very long lines may be found for instance when
+                * overlay2 filesystem with docker is used).
+                */
+               skip = skip_next;
+               skip_next = (strchr(line, '\n') == NULL);
+               if (skip)
+                       continue;
+
                if (line[0] == '/') {
-                       
+                       /* Find field separator position */
+                       pos = strchr(line, ' ');
+                       if (pos == NULL)
+                               continue;
+
+                       /*
+                        * Find second field separator position,
+                        * read filesystem type,
+                        * if filesystem type is autofs, skip it
+                       */
+                       pos2 = strchr(pos + 1, ' ');
+                       if (pos2 == NULL)
+                               continue;
+
+                       sscanf(pos2 + 1, "%127s", type);
+                       if(strcmp(type, "autofs") == 0)
+                               continue;
+
                        /* Read filesystem name and mount point */
-                       sscanf(line, "%71s %127s", fs_name, mountp);
-                       
+                       sscanf(line, "%127s", fs_name);
+                       sscanf(pos + 1, "%255s", mountp);
+
                        /* Replace octal codes */
                        oct2chr(mountp);
-                       
-                       /* Check that total size is not null */
-                       if (statfs(mountp, &buf) < 0)
+
+                       /* Check that total size is not zero */
+                       if (__statvfs(mountp, &buf) < 0)
                                continue;
-                       
+
                        if (buf.f_blocks) {
                                fs++;
                        }
@@ -494,5 +526,38 @@ int get_filesystem_nr(void)
        return fs;
 }
 
+/*
+ ***************************************************************************
+ * Find number of fibre channel hosts in /sys/class/fc_host/.
+ *
+ * RETURNS:
+ * Number of FC hosts.
+ * Return -1 if directory doesn't exist in sysfs.
+ ***************************************************************************
+ */
+__nr_t get_fchost_nr(void)
+{
+       DIR *dir;
+       struct dirent *drd;
+       __nr_t fc = 0;
+
+       if ((dir = opendir(SYSFS_FCHOST)) == NULL) {
+               /* Directory non-existent */
+               return -1;
+       }
+
+       while ((drd = readdir(dir)) != NULL) {
+
+               if (!strncmp(drd->d_name, "host", 4)) {
+                       fc++;
+               }
+       }
+
+       /* Close directory */
+       closedir(dir);
+
+       return fc;
+}
+
 /*------------------ END: FUNCTIONS USED BY SADC ONLY ---------------------*/
 #endif /* SOURCE_SADC */