mirror of
https://github.com/openvswitch/ovs
synced 2025-10-23 14:57:06 +00:00
As part of retrieving system statistics, process status APIs along with helper functions were implemented. Some of them are very generic and can be reused by other subsystems. Move the APIs in system-stats.c to process.c and util.c and make them available. This patch doesn't change any functionality. Signed-off-by: Bhanuprakash Bodireddy <bhanuprakash.bodireddy@intel.com> Signed-off-by: Ben Pfaff <blp@ovn.org>
400 lines
11 KiB
C
400 lines
11 KiB
C
/* Copyright (c) 2010, 2012, 2013, 2014 Nicira, Inc.
|
||
*
|
||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||
* you may not use this file except in compliance with the License.
|
||
* You may obtain a copy of the License at:
|
||
*
|
||
* http://www.apache.org/licenses/LICENSE-2.0
|
||
*
|
||
* Unless required by applicable law or agreed to in writing, software
|
||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
* See the License for the specific language governing permissions and
|
||
* limitations under the License.
|
||
*/
|
||
|
||
#include <config.h>
|
||
|
||
#include "system-stats.h"
|
||
|
||
#include <ctype.h>
|
||
#include <dirent.h>
|
||
#include <errno.h>
|
||
#if HAVE_MNTENT_H
|
||
#include <mntent.h>
|
||
#endif
|
||
#include <stdint.h>
|
||
#include <stdio.h>
|
||
#include <stdlib.h>
|
||
#if HAVE_SYS_STATVFS_H
|
||
#include <sys/statvfs.h>
|
||
#endif
|
||
#include <unistd.h>
|
||
|
||
#include "daemon.h"
|
||
#include "dirs.h"
|
||
#include "openvswitch/dynamic-string.h"
|
||
#include "openvswitch/json.h"
|
||
#include "latch.h"
|
||
#include "openvswitch/ofpbuf.h"
|
||
#include "ovs-rcu.h"
|
||
#include "ovs-thread.h"
|
||
#include "poll-loop.h"
|
||
#include "openvswitch/shash.h"
|
||
#include "process.h"
|
||
#include "smap.h"
|
||
#include "timeval.h"
|
||
#include "openvswitch/vlog.h"
|
||
|
||
VLOG_DEFINE_THIS_MODULE(system_stats);
|
||
|
||
/* #ifdefs make it a pain to maintain code: you have to try to build both ways.
|
||
* Thus, this file tries to compile as much of the code as possible regardless
|
||
* of the target, by writing "if (LINUX)" instead of "#ifdef __linux__" where
|
||
* this is possible. */
|
||
#ifdef __linux__
|
||
#define LINUX 1
|
||
#include <asm/param.h>
|
||
#else
|
||
#define LINUX 0
|
||
#endif
|
||
|
||
static void
|
||
get_cpu_cores(struct smap *stats)
|
||
{
|
||
long int n_cores = count_cpu_cores();
|
||
if (n_cores > 0) {
|
||
smap_add_format(stats, "cpu", "%ld", n_cores);
|
||
}
|
||
}
|
||
|
||
static void
|
||
get_load_average(struct smap *stats OVS_UNUSED)
|
||
{
|
||
#if HAVE_GETLOADAVG
|
||
double loadavg[3];
|
||
|
||
if (getloadavg(loadavg, 3) == 3) {
|
||
smap_add_format(stats, "load_average", "%.2f,%.2f,%.2f",
|
||
loadavg[0], loadavg[1], loadavg[2]);
|
||
}
|
||
#endif
|
||
}
|
||
|
||
static void
|
||
get_memory_stats(struct smap *stats)
|
||
{
|
||
if (!LINUX) {
|
||
unsigned int pagesize = get_page_size();
|
||
#ifdef _SC_PHYS_PAGES
|
||
long int phys_pages = sysconf(_SC_PHYS_PAGES);
|
||
#else
|
||
long int phys_pages = 0;
|
||
#endif
|
||
#ifdef _SC_AVPHYS_PAGES
|
||
long int avphys_pages = sysconf(_SC_AVPHYS_PAGES);
|
||
#else
|
||
long int avphys_pages = 0;
|
||
#endif
|
||
int mem_total, mem_used;
|
||
|
||
#ifndef _WIN32
|
||
if (pagesize <= 0 || phys_pages <= 0 || avphys_pages <= 0) {
|
||
return;
|
||
}
|
||
|
||
mem_total = phys_pages * (pagesize / 1024);
|
||
mem_used = (phys_pages - avphys_pages) * (pagesize / 1024);
|
||
#else
|
||
MEMORYSTATUS memory_status;
|
||
GlobalMemoryStatus(&memory_status);
|
||
|
||
mem_total = memory_status.dwTotalPhys;
|
||
mem_used = memory_status.dwTotalPhys - memory_status.dwAvailPhys;
|
||
#endif
|
||
smap_add_format(stats, "memory", "%d,%d", mem_total, mem_used);
|
||
} else {
|
||
static const char file_name[] = "/proc/meminfo";
|
||
int mem_used, mem_cache, swap_used;
|
||
int mem_free = 0;
|
||
int buffers = 0;
|
||
int cached = 0;
|
||
int swap_free = 0;
|
||
int mem_total = 0;
|
||
int swap_total = 0;
|
||
struct shash dict;
|
||
char line[128];
|
||
FILE *stream;
|
||
|
||
stream = fopen(file_name, "r");
|
||
if (!stream) {
|
||
VLOG_WARN_ONCE("%s: open failed (%s)",
|
||
file_name, ovs_strerror(errno));
|
||
return;
|
||
}
|
||
|
||
shash_init(&dict);
|
||
shash_add(&dict, "MemTotal", &mem_total);
|
||
shash_add(&dict, "MemFree", &mem_free);
|
||
shash_add(&dict, "Buffers", &buffers);
|
||
shash_add(&dict, "Cached", &cached);
|
||
shash_add(&dict, "SwapTotal", &swap_total);
|
||
shash_add(&dict, "SwapFree", &swap_free);
|
||
while (fgets(line, sizeof line, stream)) {
|
||
char key[16];
|
||
int value;
|
||
|
||
if (ovs_scan(line, "%15[^:]: %u", key, &value)) {
|
||
int *valuep = shash_find_data(&dict, key);
|
||
if (valuep) {
|
||
*valuep = value;
|
||
}
|
||
}
|
||
}
|
||
fclose(stream);
|
||
shash_destroy(&dict);
|
||
|
||
mem_used = mem_total - mem_free;
|
||
mem_cache = buffers + cached;
|
||
swap_used = swap_total - swap_free;
|
||
smap_add_format(stats, "memory", "%d,%d,%d,%d,%d",
|
||
mem_total, mem_used, mem_cache, swap_total, swap_used);
|
||
}
|
||
}
|
||
|
||
static void
|
||
get_process_stats(struct smap *stats)
|
||
{
|
||
#ifndef _WIN32
|
||
struct dirent *de;
|
||
DIR *dir;
|
||
|
||
dir = opendir(ovs_rundir());
|
||
if (!dir) {
|
||
VLOG_ERR_ONCE("%s: open failed (%s)",
|
||
ovs_rundir(), ovs_strerror(errno));
|
||
return;
|
||
}
|
||
|
||
while ((de = readdir(dir)) != NULL) {
|
||
struct process_info pinfo;
|
||
char *file_name;
|
||
char *extension;
|
||
char *key;
|
||
pid_t pid;
|
||
|
||
#ifdef _DIRENT_HAVE_D_TYPE
|
||
if (de->d_type != DT_UNKNOWN && de->d_type != DT_REG) {
|
||
continue;
|
||
}
|
||
#endif
|
||
|
||
extension = strrchr(de->d_name, '.');
|
||
if (!extension || strcmp(extension, ".pid")) {
|
||
continue;
|
||
}
|
||
|
||
file_name = xasprintf("%s/%s", ovs_rundir(), de->d_name);
|
||
pid = read_pidfile(file_name);
|
||
free(file_name);
|
||
if (pid < 0) {
|
||
continue;
|
||
}
|
||
|
||
key = xasprintf("process_%.*s",
|
||
(int) (extension - de->d_name), de->d_name);
|
||
if (!smap_get(stats, key)) {
|
||
if (LINUX && get_process_info(pid, &pinfo)) {
|
||
smap_add_format(stats, key, "%lu,%lu,%lld,%d,%lld,%lld",
|
||
pinfo.vsz, pinfo.rss, pinfo.cputime,
|
||
pinfo.crashes, pinfo.booted, pinfo.uptime);
|
||
} else {
|
||
smap_add(stats, key, "");
|
||
}
|
||
}
|
||
free(key);
|
||
}
|
||
|
||
closedir(dir);
|
||
#endif /* _WIN32 */
|
||
}
|
||
|
||
static void
|
||
get_filesys_stats(struct smap *stats OVS_UNUSED)
|
||
{
|
||
#if HAVE_GETMNTENT_R && HAVE_STATVFS
|
||
static const char file_name[] = "/etc/mtab";
|
||
struct mntent mntent;
|
||
struct mntent *me;
|
||
char buf[4096];
|
||
FILE *stream;
|
||
struct ds s;
|
||
|
||
stream = setmntent(file_name, "r");
|
||
if (!stream) {
|
||
VLOG_ERR_ONCE("%s: open failed (%s)", file_name, ovs_strerror(errno));
|
||
return;
|
||
}
|
||
|
||
ds_init(&s);
|
||
while ((me = getmntent_r(stream, &mntent, buf, sizeof buf)) != NULL) {
|
||
unsigned long long int total, free;
|
||
struct statvfs vfs;
|
||
char *p;
|
||
|
||
/* Skip non-local and read-only filesystems. */
|
||
if (strncmp(me->mnt_fsname, "/dev", 4)
|
||
|| !strstr(me->mnt_opts, "rw")) {
|
||
continue;
|
||
}
|
||
|
||
/* Given the mount point we can stat the file system. */
|
||
if (statvfs(me->mnt_dir, &vfs) && vfs.f_flag & ST_RDONLY) {
|
||
/* That's odd... */
|
||
continue;
|
||
}
|
||
|
||
/* Now format the data. */
|
||
if (s.length) {
|
||
ds_put_char(&s, ' ');
|
||
}
|
||
for (p = me->mnt_dir; *p != '\0'; p++) {
|
||
ds_put_char(&s, *p == ' ' || *p == ',' ? '_' : *p);
|
||
}
|
||
total = (unsigned long long int) vfs.f_frsize * vfs.f_blocks / 1024;
|
||
free = (unsigned long long int) vfs.f_frsize * vfs.f_bfree / 1024;
|
||
ds_put_format(&s, ",%llu,%llu", total, total - free);
|
||
}
|
||
endmntent(stream);
|
||
|
||
if (s.length) {
|
||
smap_add(stats, "file_systems", ds_cstr(&s));
|
||
}
|
||
ds_destroy(&s);
|
||
#endif /* HAVE_GETMNTENT_R && HAVE_STATVFS */
|
||
}
|
||
|
||
#define SYSTEM_STATS_INTERVAL (5 * 1000) /* In milliseconds. */
|
||
|
||
static struct ovs_mutex mutex = OVS_MUTEX_INITIALIZER;
|
||
static pthread_cond_t cond = PTHREAD_COND_INITIALIZER;
|
||
static struct latch latch OVS_GUARDED_BY(mutex);
|
||
static bool enabled;
|
||
static bool started OVS_GUARDED_BY(mutex);
|
||
static struct smap *system_stats OVS_GUARDED_BY(mutex);
|
||
|
||
OVS_NO_RETURN static void *system_stats_thread_func(void *);
|
||
static void discard_stats(void);
|
||
|
||
/* Enables or disables system stats collection, according to 'enable'. */
|
||
void
|
||
system_stats_enable(bool enable)
|
||
{
|
||
if (enabled != enable) {
|
||
ovs_mutex_lock(&mutex);
|
||
if (enable) {
|
||
if (!started) {
|
||
ovs_thread_create("system_stats",
|
||
system_stats_thread_func, NULL);
|
||
latch_init(&latch);
|
||
started = true;
|
||
}
|
||
discard_stats();
|
||
xpthread_cond_signal(&cond);
|
||
}
|
||
enabled = enable;
|
||
ovs_mutex_unlock(&mutex);
|
||
}
|
||
}
|
||
|
||
/* Tries to obtain a new snapshot of system stats every SYSTEM_STATS_INTERVAL
|
||
* milliseconds.
|
||
*
|
||
* When a new snapshot is available (which only occurs if system stats are
|
||
* enabled), returns it as an smap owned by the caller. The caller must use
|
||
* both smap_destroy() and free() to completely free the returned data.
|
||
*
|
||
* When no new snapshot is available, returns NULL. */
|
||
struct smap *
|
||
system_stats_run(void)
|
||
{
|
||
struct smap *stats = NULL;
|
||
|
||
ovs_mutex_lock(&mutex);
|
||
if (system_stats) {
|
||
latch_poll(&latch);
|
||
|
||
if (enabled) {
|
||
stats = system_stats;
|
||
system_stats = NULL;
|
||
} else {
|
||
discard_stats();
|
||
}
|
||
}
|
||
ovs_mutex_unlock(&mutex);
|
||
|
||
return stats;
|
||
}
|
||
|
||
/* Causes poll_block() to wake up when system_stats_run() needs to be
|
||
* called. */
|
||
void
|
||
system_stats_wait(void)
|
||
{
|
||
if (enabled) {
|
||
latch_wait(&latch);
|
||
}
|
||
}
|
||
|
||
static void
|
||
discard_stats(void) OVS_REQUIRES(mutex)
|
||
{
|
||
if (system_stats) {
|
||
smap_destroy(system_stats);
|
||
free(system_stats);
|
||
system_stats = NULL;
|
||
}
|
||
}
|
||
|
||
static void *
|
||
system_stats_thread_func(void *arg OVS_UNUSED)
|
||
{
|
||
pthread_detach(pthread_self());
|
||
|
||
for (;;) {
|
||
long long int next_refresh;
|
||
struct smap *stats;
|
||
|
||
ovs_mutex_lock(&mutex);
|
||
while (!enabled) {
|
||
/* The thread is sleeping, potentially for a long time, and it's
|
||
* not holding RCU protected references, so it makes sense to
|
||
* quiesce */
|
||
ovsrcu_quiesce_start();
|
||
ovs_mutex_cond_wait(&cond, &mutex);
|
||
ovsrcu_quiesce_end();
|
||
}
|
||
ovs_mutex_unlock(&mutex);
|
||
|
||
stats = xmalloc(sizeof *stats);
|
||
smap_init(stats);
|
||
get_cpu_cores(stats);
|
||
get_load_average(stats);
|
||
get_memory_stats(stats);
|
||
get_process_stats(stats);
|
||
get_filesys_stats(stats);
|
||
|
||
ovs_mutex_lock(&mutex);
|
||
discard_stats();
|
||
system_stats = stats;
|
||
latch_set(&latch);
|
||
ovs_mutex_unlock(&mutex);
|
||
|
||
next_refresh = time_msec() + SYSTEM_STATS_INTERVAL;
|
||
do {
|
||
poll_timer_wait_until(next_refresh);
|
||
poll_block();
|
||
} while (time_msec() < next_refresh);
|
||
}
|
||
}
|