Merge branch 'collectd-5.7'
[collectd.git] / src / numa.c
1 /**
2  * collectd - src/numa.c
3  * Copyright (C) 2012       Florian Forster
4  *
5  * Permission is hereby granted, free of charge, to any person obtaining a
6  * copy of this software and associated documentation files (the "Software"),
7  * to deal in the Software without restriction, including without limitation
8  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
9  * and/or sell copies of the Software, and to permit persons to whom the
10  * Software is furnished to do so, subject to the following conditions:
11  *
12  * The above copyright notice and this permission notice shall be included in
13  * all copies or substantial portions of the Software.
14  *
15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
20  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
21  * DEALINGS IN THE SOFTWARE.
22  *
23  * Authors:
24  *   Florian Forster <octo at collectd.org>
25  **/
26
27 #include "collectd.h"
28
29 #include "common.h"
30 #include "plugin.h"
31
32 #if !KERNEL_LINUX
33 #error "No applicable input method."
34 #endif
35
36 #ifndef NUMA_ROOT_DIR
37 #define NUMA_ROOT_DIR "/sys/devices/system/node"
38 #endif
39
40 static int max_node = -1;
41
42 static void numa_dispatch_value(int node, /* {{{ */
43                                 const char *type_instance, value_t v) {
44   value_list_t vl = VALUE_LIST_INIT;
45
46   vl.values = &v;
47   vl.values_len = 1;
48
49   sstrncpy(vl.plugin, "numa", sizeof(vl.plugin));
50   snprintf(vl.plugin_instance, sizeof(vl.plugin_instance), "node%i", node);
51   sstrncpy(vl.type, "vmpage_action", sizeof(vl.type));
52   sstrncpy(vl.type_instance, type_instance, sizeof(vl.type_instance));
53
54   plugin_dispatch_values(&vl);
55 } /* }}} void numa_dispatch_value */
56
57 static int numa_read_node(int node) /* {{{ */
58 {
59   char path[PATH_MAX];
60   FILE *fh;
61   char buffer[128];
62   int status;
63   int success;
64
65   snprintf(path, sizeof(path), NUMA_ROOT_DIR "/node%i/numastat", node);
66
67   fh = fopen(path, "r");
68   if (fh == NULL) {
69     char errbuf[1024];
70     ERROR("numa plugin: Reading node %i failed: open(%s): %s", node, path,
71           sstrerror(errno, errbuf, sizeof(errbuf)));
72     return -1;
73   }
74
75   success = 0;
76   while (fgets(buffer, sizeof(buffer), fh) != NULL) {
77     char *fields[4];
78     value_t v;
79
80     status = strsplit(buffer, fields, STATIC_ARRAY_SIZE(fields));
81     if (status != 2) {
82       WARNING("numa plugin: Ignoring line with unexpected "
83               "number of fields (node %i).",
84               node);
85       continue;
86     }
87
88     v.derive = 0;
89     status = parse_value(fields[1], &v, DS_TYPE_DERIVE);
90     if (status != 0)
91       continue;
92
93     numa_dispatch_value(node, fields[0], v);
94     success++;
95   }
96
97   fclose(fh);
98   return success ? 0 : -1;
99 } /* }}} int numa_read_node */
100
101 static int numa_read(void) /* {{{ */
102 {
103   int i;
104   int status;
105   int success;
106
107   if (max_node < 0) {
108     WARNING("numa plugin: No NUMA nodes were detected.");
109     return -1;
110   }
111
112   success = 0;
113   for (i = 0; i <= max_node; i++) {
114     status = numa_read_node(i);
115     if (status == 0)
116       success++;
117   }
118
119   return success ? 0 : -1;
120 } /* }}} int numa_read */
121
122 static int numa_init(void) /* {{{ */
123 {
124   /* Determine the number of nodes on this machine. */
125   while (42) {
126     char path[PATH_MAX];
127     struct stat statbuf = {0};
128     int status;
129
130     snprintf(path, sizeof(path), NUMA_ROOT_DIR "/node%i", max_node + 1);
131
132     status = stat(path, &statbuf);
133     if (status == 0) {
134       max_node++;
135       continue;
136     } else if (errno == ENOENT) {
137       break;
138     } else /* ((status != 0) && (errno != ENOENT)) */
139     {
140       char errbuf[1024];
141       ERROR("numa plugin: stat(%s) failed: %s", path,
142             sstrerror(errno, errbuf, sizeof(errbuf)));
143       return -1;
144     }
145   }
146
147   DEBUG("numa plugin: Found %i nodes.", max_node + 1);
148   return 0;
149 } /* }}} int numa_init */
150
151 void module_register(void) {
152   plugin_register_init("numa", numa_init);
153   plugin_register_read("numa", numa_read);
154 } /* void module_register */