]> git.proxmox.com Git - mirror_lxcfs.git/blobdiff - src/proc_cpuview.c
tree-wide: set _GNU_SOURCE in meson.build
[mirror_lxcfs.git] / src / proc_cpuview.c
index 03089c0102d5493121986243af8081962ad448e1..472d69dff40792d477d805626452f26b01e15c67 100644 (file)
@@ -1,20 +1,10 @@
 /* SPDX-License-Identifier: LGPL-2.1+ */
 
-#ifndef _GNU_SOURCE
-#define _GNU_SOURCE
-#endif
-
-#ifndef FUSE_USE_VERSION
-#define FUSE_USE_VERSION 26
-#endif
-
-#define _FILE_OFFSET_BITS 64
+#include "config.h"
 
-#define __STDC_FORMAT_MACROS
 #include <dirent.h>
 #include <errno.h>
 #include <fcntl.h>
-#include <fuse.h>
 #include <inttypes.h>
 #include <libgen.h>
 #include <pthread.h>
@@ -39,8 +29,9 @@
 #include <sys/sysinfo.h>
 #include <sys/vfs.h>
 
+#include "proc_cpuview.h"
+
 #include "bindings.h"
-#include "config.h"
 #include "cgroup_fuse.h"
 #include "cpuset_parse.h"
 #include "cgroups/cgroup.h"
@@ -239,7 +230,7 @@ static struct cg_proc_stat *prune_proc_stat_list(struct cg_proc_stat *node)
 
        for (struct cg_proc_stat *prev = NULL; node; ) {
                if (!cgroup_supports("cpu", node->cg, "cpu.shares")) {
-                       call_cleaner(free_proc_stat_node) struct cg_proc_stat *cur = node;
+                       struct cg_proc_stat *cur = node;
 
                        if (prev)
                                prev->next = node->next;
@@ -247,7 +238,9 @@ static struct cg_proc_stat *prune_proc_stat_list(struct cg_proc_stat *node)
                                first = node->next;
 
                        node = node->next;
-                       lxcfs_debug("Removing stat node for %s\n", cur->cg);
+                       lxcfs_debug("Removing stat node for %s\n", cur);
+
+                       free_proc_stat_node(cur);
                } else {
                        if (!first)
                                first = node;
@@ -308,7 +301,8 @@ out:
        return node;
 }
 
-static struct cg_proc_stat *find_or_create_proc_stat_node(struct cpuacct_usage *usage, int cpu_count, const char *cg)
+static struct cg_proc_stat *find_or_create_proc_stat_node(struct cpuacct_usage *usage,
+                                                         int cpu_count, const char *cg)
 {
        int hash = calc_hash(cg) % CPUVIEW_HASH_SIZE;
        struct cg_proc_stat_head *head = proc_stat_history[hash];
@@ -353,7 +347,10 @@ static void add_cpu_usage(uint64_t *surplus, struct cpuacct_usage *usage,
        if (free_space > usage->idle)
                free_space = usage->idle;
 
-       to_add = free_space > *surplus ? *surplus : free_space;
+       if (free_space > *surplus)
+               to_add = *surplus;
+       else
+               to_add = free_space;
 
        *counter += to_add;
        usage->idle -= to_add;
@@ -401,28 +398,28 @@ static uint64_t diff_cpu_usage(struct cpuacct_usage *older,
 /*
  * Read cgroup CPU quota parameters from `cpu.cfs_quota_us` or
  * `cpu.cfs_period_us`, depending on `param`. Parameter value is returned
- * throuh `value`.
+ * through `value`.
  */
 static bool read_cpu_cfs_param(const char *cg, const char *param, int64_t *value)
 {
        __do_free char *str = NULL;
-       char file[11 + 6 + 1]; /* cpu.cfs__us + quota/period + \0 */
+       char file[STRLITERALLEN("cpu.cfs_period_us") + 1];
        bool first = true;
+       int ret;
 
-       if (!pure_unified_layout(cgroup_ops)) {
-               snprintf(file, sizeof(file), "cpu.cfs_%s_us", param);
-        } else {
-               strcpy(file, "cpu.max");
+       if (pure_unified_layout(cgroup_ops)) {
                first = !strcmp(param, "quota");
+               ret = snprintf(file, sizeof(file), "cpu.max");
+       } else {
+               ret = snprintf(file, sizeof(file), "cpu.cfs_%s_us", param);
        }
-
-       if (!cgroup_ops->get(cgroup_ops, "cpu", cg, file, &str))
+       if (ret < 0 || (size_t)ret >= sizeof(file))
                return false;
 
-       if (sscanf(str, first ? "%" PRId64 : "%*d %" PRId64, value) != 1)
+       if (!cgroup_ops->get(cgroup_ops, "cpu", cg, file, &str))
                return false;
 
-       return true;
+       return sscanf(str, first ? "%" PRId64 : "%*d %" PRId64, value) == 1;
 }
 
 /*
@@ -435,8 +432,11 @@ static double exact_cpu_count(const char *cg)
        int nprocs;
        int64_t cfs_quota, cfs_period;
 
-       read_cpu_cfs_param(cg, "quota", &cfs_quota);
-       read_cpu_cfs_param(cg, "period", &cfs_period);
+       if (!read_cpu_cfs_param(cg, "quota", &cfs_quota))
+               return 0;
+
+       if (!read_cpu_cfs_param(cg, "period", &cfs_period))
+               return 0;
 
        if (cfs_quota <= 0 || cfs_period <= 0)
                return 0;
@@ -462,14 +462,17 @@ int max_cpu_count(const char *cg)
        int64_t cfs_quota, cfs_period;
        int nr_cpus_in_cpuset = 0;
 
-       read_cpu_cfs_param(cg, "quota", &cfs_quota);
-       read_cpu_cfs_param(cg, "period", &cfs_period);
+       if (!read_cpu_cfs_param(cg, "quota", &cfs_quota))
+               return 0;
+
+       if (!read_cpu_cfs_param(cg, "period", &cfs_period))
+               return 0;
 
        cpuset = get_cpuset(cg);
        if (cpuset)
                nr_cpus_in_cpuset = cpu_number_in_cpuset(cpuset);
 
-       if (cfs_quota <= 0 || cfs_period <= 0){
+       if (cfs_quota <= 0 || cfs_period <= 0) {
                if (nr_cpus_in_cpuset > 0)
                        return nr_cpus_in_cpuset;
 
@@ -478,7 +481,8 @@ int max_cpu_count(const char *cg)
 
        rv = cfs_quota / cfs_period;
 
-       /* In case quota/period does not yield a whole number, add one CPU for
+       /*
+        * In case quota/period does not yield a whole number, add one CPU for
         * the remainder.
         */
        if ((cfs_quota % cfs_period) > 0)
@@ -488,7 +492,7 @@ int max_cpu_count(const char *cg)
        if (rv > nprocs)
                rv = nprocs;
 
-       /* use min value in cpu quota and cpuset */
+       /* Use min value in cpu quota and cpuset. */
        if (nr_cpus_in_cpuset > 0 && nr_cpus_in_cpuset < rv)
                rv = nr_cpus_in_cpuset;
 
@@ -575,10 +579,9 @@ int cpuview_proc_stat(const char *cg, const char *cpuset,
 
                if (all_used >= cg_used) {
                        cg_cpu_usage[curcpu].idle = idle + (all_used - cg_used);
-
                } else {
-                       lxcfs_error("cpu%d from %s has unexpected cpu time: %" PRIu64 " in /proc/stat, %" PRIu64 " in cpuacct.usage_all; unable to determine idle time",
-                                   curcpu, cg, all_used, cg_used);
+                       lxcfs_v("cpu%d from %s has unexpected cpu time: %" PRIu64 " in /proc/stat, %" PRIu64 " in cpuacct.usage_all; unable to determine idle time",
+                               curcpu, cg, all_used, cg_used);
                        cg_cpu_usage[curcpu].idle = idle;
                }
        }
@@ -588,13 +591,14 @@ int cpuview_proc_stat(const char *cg, const char *cpuset,
        if (max_cpus > cpu_cnt || !max_cpus)
                max_cpus = cpu_cnt;
 
+       /* takes lock pthread_mutex_lock(&node->lock) */
        stat_node = find_or_create_proc_stat_node(cg_cpu_usage, nprocs, cg);
        if (!stat_node)
                return log_error(0, "Failed to find/create stat node for %s", cg);
 
-       diff = malloc(sizeof(struct cpuacct_usage) * nprocs);
+       diff = zalloc(sizeof(struct cpuacct_usage) * nprocs);
        if (!diff)
-               return 0;
+               goto out_pthread_mutex_unlock;
 
        /*
         * If the new values are LOWER than values stored in memory, it means
@@ -620,13 +624,13 @@ int cpuview_proc_stat(const char *cg, const char *cpuset,
 
                i++;
 
-               stat_node->usage[curcpu].user += diff[curcpu].user;
+               stat_node->usage[curcpu].user   += diff[curcpu].user;
                stat_node->usage[curcpu].system += diff[curcpu].system;
-               stat_node->usage[curcpu].idle += diff[curcpu].idle;
+               stat_node->usage[curcpu].idle   += diff[curcpu].idle;
 
                if (max_cpus > 0 && i >= max_cpus) {
-                       user_surplus += diff[curcpu].user;
-                       system_surplus += diff[curcpu].system;
+                       user_surplus    += diff[curcpu].user;
+                       system_surplus  += diff[curcpu].system;
                }
        }
 
@@ -638,7 +642,6 @@ int cpuview_proc_stat(const char *cg, const char *cpuset,
                uint64_t max_diff_idle = 0;
                uint64_t max_diff_idle_index = 0;
                double exact_cpus;
-
                /* threshold = maximum usage per cpu, including idle */
                threshold = total_sum / cpu_cnt * max_cpus;
 
@@ -680,20 +683,20 @@ int cpuview_proc_stat(const char *cg, const char *cpuset,
                        if (i == max_cpus)
                                break;
 
-                       stat_node->view[curcpu].user += diff[curcpu].user;
-                       stat_node->view[curcpu].system += diff[curcpu].system;
-                       stat_node->view[curcpu].idle += diff[curcpu].idle;
+                       stat_node->view[curcpu].user    += diff[curcpu].user;
+                       stat_node->view[curcpu].system  += diff[curcpu].system;
+                       stat_node->view[curcpu].idle    += diff[curcpu].idle;
 
-                       user_sum += stat_node->view[curcpu].user;
-                       system_sum += stat_node->view[curcpu].system;
-                       idle_sum += stat_node->view[curcpu].idle;
+                       user_sum        += stat_node->view[curcpu].user;
+                       system_sum      += stat_node->view[curcpu].system;
+                       idle_sum        += stat_node->view[curcpu].idle;
 
-                       diff_user += diff[curcpu].user;
-                       diff_system += diff[curcpu].system;
-                       diff_idle += diff[curcpu].idle;
+                       diff_user       += diff[curcpu].user;
+                       diff_system     += diff[curcpu].system;
+                       diff_idle       += diff[curcpu].idle;
                        if (diff[curcpu].idle > max_diff_idle) {
-                               max_diff_idle = diff[curcpu].idle;
-                               max_diff_idle_index = curcpu;
+                               max_diff_idle           = diff[curcpu].idle;
+                               max_diff_idle_index     = curcpu;
                        }
 
                        lxcfs_v("curcpu: %d, diff_user: %lu, diff_system: %lu, diff_idle: %lu\n", curcpu, diff[curcpu].user, diff[curcpu].system, diff[curcpu].idle);
@@ -708,12 +711,18 @@ int cpuview_proc_stat(const char *cg, const char *cpuset,
                        lxcfs_v("revising cpu usage view to match the exact cpu count [%f]\n", exact_cpus);
                        lxcfs_v("delta: %lu\n", delta);
                        lxcfs_v("idle_sum before: %lu\n", idle_sum);
-                       idle_sum = idle_sum > delta ? idle_sum - delta : 0;
+                       if (idle_sum > delta)
+                               idle_sum = idle_sum - delta;
+                       else
+                               idle_sum = 0;
                        lxcfs_v("idle_sum after: %lu\n", idle_sum);
 
                        curcpu = max_diff_idle_index;
                        lxcfs_v("curcpu: %d, idle before: %lu\n", curcpu, stat_node->view[curcpu].idle);
-                       stat_node->view[curcpu].idle = stat_node->view[curcpu].idle > delta ? stat_node->view[curcpu].idle - delta : 0;
+                       if (stat_node->view[curcpu].idle > delta)
+                               stat_node->view[curcpu].idle = stat_node->view[curcpu].idle - delta;
+                       else
+                               stat_node->view[curcpu].idle = 0;
                        lxcfs_v("curcpu: %d, idle after: %lu\n", curcpu, stat_node->view[curcpu].idle);
                }
        } else {
@@ -721,13 +730,13 @@ int cpuview_proc_stat(const char *cg, const char *cpuset,
                        if (!stat_node->usage[curcpu].online)
                                continue;
 
-                       stat_node->view[curcpu].user = stat_node->usage[curcpu].user;
-                       stat_node->view[curcpu].system = stat_node->usage[curcpu].system;
-                       stat_node->view[curcpu].idle = stat_node->usage[curcpu].idle;
+                       stat_node->view[curcpu].user    = stat_node->usage[curcpu].user;
+                       stat_node->view[curcpu].system  = stat_node->usage[curcpu].system;
+                       stat_node->view[curcpu].idle    = stat_node->usage[curcpu].idle;
 
-                       user_sum += stat_node->view[curcpu].user;
-                       system_sum += stat_node->view[curcpu].system;
-                       idle_sum += stat_node->view[curcpu].idle;
+                       user_sum        += stat_node->view[curcpu].user;
+                       system_sum      += stat_node->view[curcpu].system;
+                       idle_sum        += stat_node->view[curcpu].idle;
                }
        }
 
@@ -737,10 +746,16 @@ int cpuview_proc_stat(const char *cg, const char *cpuset,
                     "cpu  %" PRIu64 " 0 %" PRIu64 " %" PRIu64 " 0 0 0 0 0 0\n",
                     user_sum, system_sum, idle_sum);
        lxcfs_v("cpu-all: %s\n", buf);
-       if (l < 0)
-               return log_error(0, "Failed to write cache");
-       if (l >= buf_size)
-               return log_error(0, "Write to cache was truncated");
+       if (l < 0) {
+               lxcfs_error("Failed to write cache");
+               total_len = 0;
+               goto out_pthread_mutex_unlock;
+       }
+       if ((size_t)l >= buf_size) {
+               lxcfs_error("Write to cache was truncated");
+               total_len = 0;
+               goto out_pthread_mutex_unlock;
+       }
 
        buf += l;
        buf_size -= l;
@@ -762,10 +777,16 @@ int cpuview_proc_stat(const char *cg, const char *cpuset,
                             stat_node->view[curcpu].system,
                             stat_node->view[curcpu].idle);
                lxcfs_v("cpu: %s\n", buf);
-               if (l < 0)
-                       return log_error(0, "Failed to write cache");
-               if (l >= buf_size)
-                       return log_error(0, "Write to cache was truncated");
+               if (l < 0) {
+                       lxcfs_error("Failed to write cache");
+                       total_len = 0;
+                       goto out_pthread_mutex_unlock;
+               }
+               if ((size_t)l >= buf_size) {
+                       lxcfs_error("Write to cache was truncated");
+                       total_len = 0;
+                       goto out_pthread_mutex_unlock;
+               }
 
                buf += l;
                buf_size -= l;
@@ -774,10 +795,16 @@ int cpuview_proc_stat(const char *cg, const char *cpuset,
 
        /* Pass the rest of /proc/stat, start with the last line read */
        l = snprintf(buf, buf_size, "%s", line);
-       if (l < 0)
-               return log_error(0, "Failed to write cache");
-       if (l >= buf_size)
-               return log_error(0, "Write to cache was truncated");
+       if (l < 0) {
+               lxcfs_error("Failed to write cache");
+               total_len = 0;
+               goto out_pthread_mutex_unlock;
+       }
+       if ((size_t)l >= buf_size) {
+               lxcfs_error("Write to cache was truncated");
+               total_len = 0;
+               goto out_pthread_mutex_unlock;
+       }
 
        buf += l;
        buf_size -= l;
@@ -786,16 +813,23 @@ int cpuview_proc_stat(const char *cg, const char *cpuset,
        /* Pass the rest of the host's /proc/stat */
        while (getline(&line, &linelen, f) != -1) {
                l = snprintf(buf, buf_size, "%s", line);
-               if (l < 0)
-                       return log_error(0, "Failed to write cache");
-               if (l >= buf_size)
-                       return log_error(0, "Write to cache was truncated");
+               if (l < 0) {
+                       lxcfs_error("Failed to write cache");
+                       total_len = 0;
+                       goto out_pthread_mutex_unlock;
+               }
+               if ((size_t)l >= buf_size) {
+                       lxcfs_error("Write to cache was truncated");
+                       total_len = 0;
+                       goto out_pthread_mutex_unlock;
+               }
 
                buf += l;
                buf_size -= l;
                total_len += l;
        }
 
+out_pthread_mutex_unlock:
        if (stat_node)
                pthread_mutex_unlock(&stat_node->lock);
 
@@ -838,7 +872,7 @@ int proc_cpuinfo_read(char *buf, size_t size, off_t offset,
        size_t cache_size = d->buflen;
 
        if (offset) {
-               int left;
+               size_t left;
 
                if (offset > d->size)
                        return -EINVAL;
@@ -901,7 +935,7 @@ int proc_cpuinfo_read(char *buf, size_t size, off_t offset,
                                l = snprintf(cache, cache_size, "processor      : %d\n", curcpu);
                                if (l < 0)
                                        return log_error(0, "Failed to write cache");
-                               if (l >= cache_size)
+                               if ((size_t)l >= cache_size)
                                        return log_error(0, "Write to cache was truncated");
                                cache += l;
                                cache_size -= l;
@@ -926,7 +960,7 @@ int proc_cpuinfo_read(char *buf, size_t size, off_t offset,
                        l = snprintf(cache, cache_size, "processor %d:%s", curcpu, p);
                        if (l < 0)
                                return log_error(0, "Failed to write cache");
-                       if (l >= cache_size)
+                       if ((size_t)l >= cache_size)
                                return log_error(0, "Write to cache was truncated");
 
                        cache += l;
@@ -939,7 +973,7 @@ int proc_cpuinfo_read(char *buf, size_t size, off_t offset,
                        l = snprintf(cache, cache_size, "%s", line);
                        if (l < 0)
                                return log_error(0, "Failed to write cache");
-                       if (l >= cache_size)
+                       if ((size_t)l >= cache_size)
                                return log_error(0, "Write to cache was truncated");
 
                        cache += l;
@@ -962,21 +996,21 @@ int proc_cpuinfo_read(char *buf, size_t size, off_t offset,
                cache_size = d->buflen;
                total_len = 0;
                l = snprintf(cache, cache_size, "vendor_id       : IBM/S390\n");
-               if (l < 0 || l >= cache_size)
+               if (l < 0 || (size_t)l >= cache_size)
                        return 0;
 
                cache_size -= l;
                cache += l;
                total_len += l;
                l = snprintf(cache, cache_size, "# processors    : %d\n", curcpu + 1);
-               if (l < 0 || l >= cache_size)
+               if (l < 0 || (size_t)l >= cache_size)
                        return 0;
 
                cache_size -= l;
                cache += l;
                total_len += l;
                l = snprintf(cache, cache_size, "%s", origcache);
-               if (l < 0 || l >= cache_size)
+               if (l < 0 || (size_t)l >= cache_size)
                        return 0;
                total_len += l;
        }
@@ -1081,18 +1115,18 @@ int read_cpuacct_usage_all(char *cg, char *cpuset,
 
 static bool cpuview_init_head(struct cg_proc_stat_head **head)
 {
-       *head = malloc(sizeof(struct cg_proc_stat_head));
-       if (!(*head))
-               return log_error(false, "%s", strerror(errno));
+       __do_free struct cg_proc_stat_head *h;
 
-       (*head)->lastcheck = time(NULL);
-       (*head)->next = NULL;
+       h = zalloc(sizeof(struct cg_proc_stat_head));
+       if (!h)
+               return false;
 
-       if (pthread_rwlock_init(&(*head)->lock, NULL) != 0) {
-               free_disarm(*head);
-               return log_error(false, "Failed to initialize list lock");
-       }
+       if (pthread_rwlock_init(&h->lock, NULL))
+               return false;
+
+       h->lastcheck = time(NULL);
 
+       *head = move_ptr(h);
        return true;
 }