src/apple_sensors.c

   1 /**
   2  * collectd - src/apple_sensors.c
   3  * Copyright (C) 2006,2007  Florian octo 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 octo Forster <octo at collectd.org>
  25  **/
  26
  27 #include "collectd.h"
  28
  29 #include "common.h"
  30 #include "plugin.h"
  31
  32 #if HAVE_CTYPE_H
  33 #include <ctype.h>
  34 #endif
  35
  36 #if HAVE_MACH_MACH_TYPES_H
  37 #include <mach/mach_types.h>
  38 #endif
  39 #if HAVE_MACH_MACH_INIT_H
  40 #include <mach/mach_init.h>
  41 #endif
  42 #if HAVE_MACH_MACH_ERROR_H
  43 #include <mach/mach_error.h>
  44 #endif
  45 #if HAVE_MACH_MACH_PORT_H
  46 #include <mach/mach_port.h>
  47 #endif
  48 #if HAVE_COREFOUNDATION_COREFOUNDATION_H
  49 #include <CoreFoundation/CoreFoundation.h>
  50 #endif
  51 #if HAVE_IOKIT_IOKITLIB_H
  52 #include <IOKit/IOKitLib.h>
  53 #endif
  54 #if HAVE_IOKIT_IOTYPES_H
  55 #include <IOKit/IOTypes.h>
  56 #endif
  57
  58 static mach_port_t io_master_port = MACH_PORT_NULL;
  59
  60 static int as_init(void) {
  61   kern_return_t status;
  62
  63   if (io_master_port != MACH_PORT_NULL) {
  64     mach_port_deallocate(mach_task_self(), io_master_port);
  65     io_master_port = MACH_PORT_NULL;
  66   }
  67
  68   status = IOMasterPort(MACH_PORT_NULL, &io_master_port);
  69   if (status != kIOReturnSuccess) {
  70     ERROR("IOMasterPort failed: %s", mach_error_string(status));
  71     io_master_port = MACH_PORT_NULL;
  72     return (-1);
  73   }
  74
  75   return (0);
  76 }
  77
  78 static void as_submit(const char *type, const char *type_instance, double val) {
  79   value_list_t vl = VALUE_LIST_INIT;
  80
  81   vl.values = &(value_t){.gauge = val};
  82   vl.values_len = 1;
  83   sstrncpy(vl.plugin, "apple_sensors", sizeof(vl.plugin));
  84   sstrncpy(vl.type, type, sizeof(vl.type));
  85   sstrncpy(vl.type_instance, type_instance, sizeof(vl.type_instance));
  86
  87   plugin_dispatch_values(&vl);
  88 }
  89
  90 static int as_read(void) {
  91   kern_return_t status;
  92   io_iterator_t iterator;
  93   io_object_t io_obj;
  94   CFMutableDictionaryRef prop_dict;
  95   CFTypeRef property;
  96
  97   char type[128];
  98   char inst[128];
  99   int value_int;
 100   double value_double;
 101   if (!io_master_port || (io_master_port == MACH_PORT_NULL))
 102     return (-1);
 103
 104   status = IOServiceGetMatchingServices(
 105       io_master_port, IOServiceNameMatching("IOHWSensor"), &iterator);
 106   if (status != kIOReturnSuccess) {
 107     ERROR("IOServiceGetMatchingServices failed: %s", mach_error_string(status));
 108     return (-1);
 109   }
 110
 111   while ((io_obj = IOIteratorNext(iterator))) {
 112     prop_dict = NULL;
 113     status = IORegistryEntryCreateCFProperties(
 114         io_obj, &prop_dict, kCFAllocatorDefault, kNilOptions);
 115     if (status != kIOReturnSuccess) {
 116       DEBUG("IORegistryEntryCreateCFProperties failed: %s",
 117             mach_error_string(status));
 118       continue;
 119     }
 120
 121     /* Copy the sensor type. */
 122     property = NULL;
 123     if (!CFDictionaryGetValueIfPresent(prop_dict, CFSTR("type"), &property))
 124       continue;
 125     if (CFGetTypeID(property) != CFStringGetTypeID())
 126       continue;
 127     if (!CFStringGetCString(property, type, sizeof(type),
 128                             kCFStringEncodingASCII))
 129       continue;
 130     type[sizeof(type) - 1] = '\0';
 131
 132     /* Copy the sensor location. This will be used as `instance'. */
 133     property = NULL;
 134     if (!CFDictionaryGetValueIfPresent(prop_dict, CFSTR("location"), &property))
 135       continue;
 136     if (CFGetTypeID(property) != CFStringGetTypeID())
 137       continue;
 138     if (!CFStringGetCString(property, inst, sizeof(inst),
 139                             kCFStringEncodingASCII))
 140       continue;
 141     inst[sizeof(inst) - 1] = '\0';
 142     for (int i = 0; i < 128; i++) {
 143       if (inst[i] == '\0')
 144         break;
 145       else if (isalnum(inst[i]))
 146         inst[i] = (char)tolower(inst[i]);
 147       else
 148         inst[i] = '_';
 149     }
 150
 151     /* Get the actual value. Some computation, based on the `type'
 152      * is neccessary. */
 153     property = NULL;
 154     if (!CFDictionaryGetValueIfPresent(prop_dict, CFSTR("current-value"),
 155                                        &property))
 156       continue;
 157     if (CFGetTypeID(property) != CFNumberGetTypeID())
 158       continue;
 159     if (!CFNumberGetValue(property, kCFNumberIntType, &value_int))
 160       continue;
 161
 162     /* Found e.g. in the 1.5GHz PowerBooks */
 163     if (strcmp(type, "temperature") == 0) {
 164       value_double = ((double)value_int) / 65536.0;
 165       sstrncpy(type, "temperature", sizeof(type));
 166     } else if (strcmp(type, "temp") == 0) {
 167       value_double = ((double)value_int) / 10.0;
 168       sstrncpy(type, "temperature", sizeof(type));
 169     } else if (strcmp(type, "fanspeed") == 0) {
 170       value_double = ((double)value_int) / 65536.0;
 171       sstrncpy(type, "fanspeed", sizeof(type));
 172     } else if (strcmp(type, "voltage") == 0) {
 173       /* Leave this to the battery plugin. */
 174       continue;
 175     } else if (strcmp(type, "adc") == 0) {
 176       value_double = ((double)value_int) / 10.0;
 177       sstrncpy(type, "fanspeed", sizeof(type));
 178     } else {
 179       DEBUG("apple_sensors: Read unknown sensor type: %s", type);
 180       value_double = (double)value_int;
 181     }
 182
 183     as_submit(type, inst, value_double);
 184
 185     CFRelease(prop_dict);
 186     IOObjectRelease(io_obj);
 187   } /* while (iterator) */
 188
 189   IOObjectRelease(iterator);
 190
 191   return (0);
 192 } /* int as_read */
 193
 194 void module_register(void) {
 195   plugin_register_init("apple_sensors", as_init);
 196   plugin_register_read("apple_sensors", as_read);
 197 } /* void module_register */