src/thermal.c

   1 /**
   2  * collectd - src/thermal.c
   3  * Copyright (C) 2008  Michał Mirosław
   4  *
   5  * This program is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License as published by the
   7  * Free Software Foundation; only version 2 of the License is applicable.
   8  *
   9  * This program is distributed in the hope that it will be useful, but
  10  * WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along
  15  * with this program; if not, write to the Free Software Foundation, Inc.,
  16  * 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  17  *
  18  * Authors:
  19  *   Michał Mirosław <mirq-linux at rere.qmqm.pl>
  20  **/
  21
  22 #include "collectd.h"
  23 #include "common.h"
  24 #include "plugin.h"
  25 #include "configfile.h"
  26 #include "utils_ignorelist.h"
  27
  28 #if !KERNEL_LINUX
  29 # error "This module is for Linux only."
  30 #endif
  31
  32 const char *const dirname_sysfs = "/sys/class/thermal";
  33 const char *const dirname_procfs = "/proc/acpi/thermal_zone";
  34
  35 static char force_procfs = 0;
  36 static ignorelist_t *device_list;
  37 static value_list_t vl_temp_template = VALUE_LIST_STATIC;
  38 static value_list_t vl_state_template = VALUE_LIST_STATIC;
  39
  40 enum dev_type {
  41         TEMP = 0,
  42         COOLING_DEV
  43 };
  44
  45 static void thermal_submit (const char *plugin_instance, enum dev_type dt,
  46                 gauge_t value)
  47 {
  48         value_list_t vl = (dt == TEMP) ? vl_temp_template : vl_state_template;
  49         value_t vt;
  50
  51         vt.gauge = value;
  52
  53         vl.values = &vt;
  54         vl.values_len = 1;
  55
  56         sstrncpy (vl.plugin, "thermal", sizeof(vl.plugin));
  57         sstrncpy (vl.plugin_instance, plugin_instance,
  58                         sizeof(vl.plugin_instance));
  59         sstrncpy (vl.type, (dt == TEMP) ? "temperature" : "gauge",
  60                         sizeof (vl.type));
  61
  62         plugin_dispatch_values (&vl);
  63 }
  64
  65 static int thermal_sysfs_device_read (const char __attribute__((unused)) *dir,
  66                 const char *name, void __attribute__((unused)) *user_data)
  67 {
  68         char filename[256];
  69         char data[1024];
  70         int len;
  71         int ok = 0;
  72
  73         if (device_list && ignorelist_match (device_list, name))
  74                 return -1;
  75
  76         len = snprintf (filename, sizeof (filename),
  77                         "%s/%s/temp", dirname_sysfs, name);
  78         if ((len < 0) || ((size_t) len >= sizeof (filename)))
  79                 return -1;
  80
  81         len = read_file_contents (filename, data, sizeof(data));
  82         if (len > 1 && data[--len] == '\n') {
  83                 char *endptr = NULL;
  84                 double temp;
  85
  86                 data[len] = 0;
  87                 errno = 0;
  88                 temp = strtod (data, &endptr) / 1000.0;
  89
  90                 if (endptr == data + len && errno == 0) {
  91                         thermal_submit(name, TEMP, temp);
  92                         ++ok;
  93                 }
  94         }
  95
  96         len = snprintf (filename, sizeof (filename),
  97                         "%s/%s/cur_state", dirname_sysfs, name);
  98         if ((len < 0) || ((size_t) len >= sizeof (filename)))
  99                 return -1;
 100
 101         len = read_file_contents (filename, data, sizeof(data));
 102         if (len > 1 && data[--len] == '\n') {
 103                 char *endptr = NULL;
 104                 double state;
 105
 106                 data[len] = 0;
 107                 errno = 0;
 108                 state = strtod (data, &endptr);
 109
 110                 if (endptr == data + len && errno == 0) {
 111                         thermal_submit(name, COOLING_DEV, state);
 112                         ++ok;
 113                 }
 114         }
 115
 116         return ok ? 0 : -1;
 117 }
 118
 119 static int thermal_procfs_device_read (const char __attribute__((unused)) *dir,
 120                 const char *name, void __attribute__((unused)) *user_data)
 121 {
 122         const char str_temp[] = "temperature:";
 123         char filename[256];
 124         char data[1024];
 125         int len;
 126
 127         if (device_list && ignorelist_match (device_list, name))
 128                 return -1;
 129
 130         /**
 131          * rechot ~ # cat /proc/acpi/thermal_zone/THRM/temperature
 132          * temperature:             55 C
 133          */
 134
 135         len = snprintf (filename, sizeof (filename),
 136                         "%s/%s/temperature", dirname_procfs, name);
 137         if ((len < 0) || ((size_t) len >= sizeof (filename)))
 138                 return -1;
 139
 140         len = read_file_contents (filename, data, sizeof(data));
 141         if ((len > 0) && ((size_t) len > sizeof(str_temp))
 142                         && (data[--len] == '\n')
 143                         && (! strncmp(data, str_temp, sizeof(str_temp)-1))) {
 144                 char *endptr = NULL;
 145                 double temp;
 146                 double celsius, add;
 147
 148                 if (data[--len] == 'C') {
 149                         add = 0;
 150                         celsius = 1;
 151                 } else if (data[len] == 'F') {
 152                         add = -32;
 153                         celsius = 5/9;
 154                 } else if (data[len] == 'K') {
 155                         add = -273.15;
 156                         celsius = 1;
 157                 } else
 158                         return -1;
 159
 160                 while (len > 0 && data[--len] == ' ')
 161                         ;
 162                 data[len + 1] = 0;
 163
 164                 while (len > 0 && data[--len] != ' ')
 165                         ;
 166                 ++len;
 167
 168                 errno = 0;
 169                 temp = (strtod (data + len, &endptr) + add) * celsius;
 170
 171                 if (endptr != data + len && errno == 0) {
 172                         thermal_submit(name, TEMP, temp);
 173                         return 0;
 174                 }
 175         }
 176
 177         return -1;
 178 }
 179
 180 static const char *config_keys[] = {
 181         "Device",
 182         "IgnoreSelected",
 183         "ForceUseProcfs"
 184 };
 185
 186 static int thermal_config (const char *key, const char *value)
 187 {
 188         if (device_list == NULL)
 189                 device_list = ignorelist_create (1);
 190
 191         if (strcasecmp (key, "Device") == 0)
 192         {
 193                 if (ignorelist_add (device_list, value))
 194                 {
 195                         ERROR ("thermal plugin: "
 196                                         "Cannot add value to ignorelist.");
 197                         return 1;
 198                 }
 199         }
 200         else if (strcasecmp (key, "IgnoreSelected") == 0)
 201         {
 202                 ignorelist_set_invert (device_list, 1);
 203                 if (IS_TRUE (value))
 204                         ignorelist_set_invert (device_list, 0);
 205         }
 206         else if (strcasecmp (key, "ForceUseProcfs") == 0)
 207         {
 208                 force_procfs = 0;
 209                 if (IS_TRUE (value))
 210                         force_procfs = 1;
 211         }
 212         else
 213         {
 214                 return -1;
 215         }
 216
 217         return 0;
 218 }
 219
 220 static int thermal_sysfs_read (void)
 221 {
 222         return walk_directory (dirname_sysfs, thermal_sysfs_device_read,
 223                         /* user_data = */ NULL, /* include hidden */ 0);
 224 }
 225
 226 static int thermal_procfs_read (void)
 227 {
 228         return walk_directory (dirname_procfs, thermal_procfs_device_read,
 229                         /* user_data = */ NULL, /* include hidden */ 0);
 230 }
 231
 232 static int thermal_init (void)
 233 {
 234         int ret = -1;
 235
 236         if (!force_procfs && access (dirname_sysfs, R_OK | X_OK) == 0) {
 237                 ret = plugin_register_read ("thermal", thermal_sysfs_read);
 238         } else if (access (dirname_procfs, R_OK | X_OK) == 0) {
 239                 ret = plugin_register_read ("thermal", thermal_procfs_read);
 240         }
 241
 242         if (!ret) {
 243                 vl_temp_template.values_len = 1;
 244                 vl_temp_template.interval = interval_g;
 245                 sstrncpy (vl_temp_template.host, hostname_g,
 246                         sizeof(vl_temp_template.host));
 247                 sstrncpy (vl_temp_template.plugin, "thermal",
 248                         sizeof(vl_temp_template.plugin));
 249                 sstrncpy (vl_temp_template.type_instance, "temperature",
 250                         sizeof(vl_temp_template.type_instance));
 251
 252                 vl_state_template = vl_temp_template;
 253                 sstrncpy (vl_state_template.type_instance, "cooling_state",
 254                         sizeof(vl_state_template.type_instance));
 255         }
 256
 257         return ret;
 258 }
 259
 260 static int thermal_shutdown (void)
 261 {
 262         ignorelist_free (device_list);
 263
 264         return 0;
 265 }
 266
 267 void module_register (void)
 268 {
 269         plugin_register_config ("thermal", thermal_config,
 270                         config_keys, STATIC_ARRAY_SIZE(config_keys));
 271         plugin_register_init ("thermal", thermal_init);
 272         plugin_register_shutdown ("thermal", thermal_shutdown);
 273 }
 274