src/onewire.c

   1 /**
   2  * collectd - src/onewire.c
   3  * Copyright (C) 2008  noris network AG
   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  *   Florian octo Forster <octo at noris.net>
  20  **/
  21
  22 #include "collectd.h"
  23 #include "common.h"
  24 #include "plugin.h"
  25 #include "utils_ignorelist.h"
  26
  27 #include <sys/time.h>
  28 #include <sys/types.h>
  29 #include <regex.h>
  30 #include <owcapi.h>
  31
  32 #define OW_FAMILY_LENGTH 8
  33 #define OW_FAMILY_MAX_FEATURES 2
  34 struct ow_family_features_s
  35 {
  36   char family[OW_FAMILY_LENGTH];
  37   struct
  38   {
  39     char filename[DATA_MAX_NAME_LEN];
  40     char type[DATA_MAX_NAME_LEN];
  41     char type_instance[DATA_MAX_NAME_LEN];
  42   } features[OW_FAMILY_MAX_FEATURES];
  43   size_t features_num;
  44 };
  45 typedef struct ow_family_features_s ow_family_features_t;
  46
  47 /* internal timing info collected in debug version only */
  48 #if COLLECT_DEBUG
  49 static struct timeval tv_begin, tv_end, tv_diff;
  50 #endif /* COLLECT_DEBUG */
  51
  52 /* regexp to extract address (without family) and file from the owfs path */
  53 static const char *regexp_to_match = "[A-Fa-f0-9]{2}\\.([A-Fa-f0-9]{12})/([[:alnum:]]+)$";
  54
  55 /* see http://owfs.sourceforge.net/ow_table.html for a list of families */
  56 static ow_family_features_t ow_family_features[] =
  57 {
  58   { /* DS18S20 Precision Thermometer and DS1920 ibutton */
  59     /* family = */ "10.",
  60     {
  61       {
  62         /* filename = */ "temperature",
  63         /* type = */ "temperature",
  64         /* type_instance = */ ""
  65       }
  66     },
  67     /* features_num = */ 1
  68   },
  69   { /* DS1822 Econo Thermometer */
  70     /* family = */ "22.",
  71     {
  72       {
  73         /* filename = */ "temperature",
  74         /* type = */ "temperature",
  75         /* type_instance = */ ""
  76       }
  77     },
  78     /* features_num = */ 1
  79   },
  80   { /* DS18B20 Programmable Resolution Thermometer */
  81     /* family = */ "28.",
  82     {
  83       {
  84         /* filename = */ "temperature",
  85         /* type = */ "temperature",
  86         /* type_instance = */ ""
  87       }
  88     },
  89     /* features_num = */ 1
  90   },
  91   { /* DS2436 Volts/Temp */
  92     /* family = */ "1B.",
  93     {
  94       {
  95         /* filename = */ "temperature",
  96         /* type = */ "temperature",
  97         /* type_instance = */ ""
  98       }
  99     },
 100     /* features_num = */ 1
 101   },
 102   { /* DS2438 Volts/Temp */
 103     /* family = */ "26.",
 104     {
 105       {
 106         /* filename = */ "temperature",
 107         /* type = */ "temperature",
 108         /* type_instance = */ ""
 109       }
 110     },
 111     /* features_num = */ 1
 112   }
 113 };
 114 static int ow_family_features_num = STATIC_ARRAY_SIZE (ow_family_features);
 115
 116 static char *device_g = NULL;
 117 static cdtime_t ow_interval = 0;
 118 static _Bool direct_access = 0;
 119
 120 static const char *config_keys[] =
 121 {
 122   "Device",
 123   "IgnoreSelected",
 124   "Sensor",
 125   "Interval"
 126 };
 127 static int config_keys_num = STATIC_ARRAY_SIZE (config_keys);
 128
 129 static ignorelist_t *sensor_list;
 130
 131 static _Bool   regex_direct_initialized = 0;
 132 static regex_t regex_direct;
 133
 134 /**
 135  * List of onewire owfs "files" to be directly read
 136  */
 137 typedef struct direct_access_element_s
 138 {
 139         char *path;                 /**< The whole owfs path */
 140     char *address;              /**< 1-wire address without family */
 141     char *file;                 /**< owfs file - e.g. temperature */
 142         struct direct_access_element_s *next; /**< Next in the list */
 143 } direct_access_element_t;
 144
 145 static direct_access_element_t * direct_list = NULL;
 146
 147 /* =================================================================================== */
 148
 149 #if COLLECT_DEBUG
 150 /* Return 1 if the difference is negative, otherwise 0.  */
 151 static int timeval_subtract(struct timeval *result, struct timeval *t2, struct timeval *t1)
 152 {
 153     long int diff = (t2->tv_usec + 1000000 * t2->tv_sec) - (t1->tv_usec + 1000000 * t1->tv_sec);
 154     result->tv_sec = diff / 1000000;
 155     result->tv_usec = diff % 1000000;
 156
 157     return (diff<0);
 158 }
 159 #endif /* COLLECT_DEBUG */
 160
 161 /* =================================================================================== */
 162
 163 static void direct_list_element_free(direct_access_element_t *el)
 164 {
 165     if (el != NULL)
 166     {
 167         DEBUG ("onewire plugin: direct_list_element_free - deleting <%s>", el->path);
 168         sfree (el->path);
 169         sfree (el->address);
 170         sfree (el->file);
 171         free (el);
 172     }
 173 }
 174
 175 static int direct_list_insert(const char * config)
 176 {
 177     regmatch_t               pmatch[3];
 178     size_t                   nmatch = 3;
 179     direct_access_element_t *element = NULL;
 180
 181     DEBUG ("onewire plugin: direct_list_insert <%s>", config);
 182
 183     element = (direct_access_element_t *) malloc (sizeof(*element));
 184     if (element == NULL)
 185     {
 186         ERROR ("onewire plugin: direct_list_insert - cannot allocate element");
 187         return 1;
 188     }
 189     element->path    = NULL;
 190     element->address = NULL;
 191     element->file    = NULL;
 192
 193     element->path = strdup (config);
 194     if (element->path == NULL)
 195     {
 196         ERROR ("onewire plugin: direct_list_insert - cannot allocate path");
 197         direct_list_element_free (element);
 198         return 1;
 199     }
 200
 201     DEBUG ("onewire plugin: direct_list_insert - about to match %s", config);
 202
 203     if (!regex_direct_initialized)
 204     {
 205         if (regcomp (&regex_direct, regexp_to_match, REG_EXTENDED))
 206         {
 207             ERROR ("onewire plugin: Cannot compile regex");
 208             direct_list_element_free (element);
 209             return (1);
 210         }
 211         regex_direct_initialized = 1;
 212         DEBUG ("onewire plugin: Compiled regex!!");
 213     }
 214
 215     if (regexec (&regex_direct, config, nmatch, pmatch, 0))
 216     {
 217         ERROR ("onewire plugin: direct_list_insert - no regex  match");
 218         direct_list_element_free (element);
 219         return 1;
 220     }
 221
 222     if (pmatch[1].rm_so<0)
 223     {
 224         ERROR ("onewire plugin: direct_list_insert - no address regex match");
 225         direct_list_element_free (element);
 226         return 1;
 227     }
 228     element->address = strndup (config+pmatch[1].rm_so,
 229                                 pmatch[1].rm_eo - pmatch[1].rm_so);
 230     if (element->address == NULL)
 231     {
 232         ERROR ("onewire plugin: direct_list_insert - cannot allocate address");
 233         direct_list_element_free (element);
 234         return 1;
 235     }
 236     DEBUG ("onewire plugin: direct_list_insert - found address <%s>",
 237            element->address);
 238
 239     if (pmatch[2].rm_so<0)
 240     {
 241         ERROR ("onewire plugin: direct_list_insert - no file regex match");
 242         direct_list_element_free (element);
 243         return 1;
 244     }
 245     element->file = strndup (config+pmatch[2].rm_so,
 246                              pmatch[2].rm_eo - pmatch[2].rm_so);
 247     if (element->file == NULL)
 248     {
 249         ERROR ("onewire plugin: direct_list_insert - cannot allocate file");
 250         direct_list_element_free (element);
 251         return 1;
 252     }
 253     DEBUG ("onewire plugin: direct_list_insert - found file <%s>", element->file);
 254
 255     element->next = direct_list;
 256     direct_list = element;
 257
 258     return 0;
 259 }
 260
 261 static void direct_list_free(void)
 262 {
 263     direct_access_element_t *traverse = direct_list;
 264     direct_access_element_t *tmp = NULL;;
 265
 266     while(traverse != NULL)
 267     {
 268         tmp = traverse;
 269         traverse = traverse->next;
 270         direct_list_element_free (tmp);
 271         tmp = NULL;
 272     }
 273 }
 274
 275 /* =================================================================================== */
 276
 277 static int cow_load_config (const char *key, const char *value)
 278 {
 279   if (sensor_list == NULL)
 280     sensor_list = ignorelist_create (1);
 281
 282   if (strcasecmp (key, "Sensor") == 0)
 283   {
 284     if (direct_list_insert (value))
 285     {
 286         DEBUG ("onewire plugin: Cannot add %s to direct_list_insert.", value);
 287
 288         if (ignorelist_add (sensor_list, value))
 289         {
 290             ERROR ("onewire plugin: Cannot add value to ignorelist.");
 291             return (1);
 292         }
 293     }
 294     else
 295     {
 296         DEBUG ("onewire plugin: %s is a direct access", value);
 297         direct_access = 1;
 298     }
 299   }
 300   else if (strcasecmp (key, "IgnoreSelected") == 0)
 301   {
 302     ignorelist_set_invert (sensor_list, 1);
 303     if (IS_TRUE (value))
 304       ignorelist_set_invert (sensor_list, 0);
 305   }
 306   else if (strcasecmp (key, "Device") == 0)
 307   {
 308     char *temp;
 309     temp = strdup (value);
 310     if (temp == NULL)
 311     {
 312       ERROR ("onewire plugin: strdup failed.");
 313       return (1);
 314     }
 315     sfree (device_g);
 316     device_g = temp;
 317   }
 318   else if (strcasecmp ("Interval", key) == 0)
 319   {
 320     double tmp;
 321     tmp = atof (value);
 322     if (tmp > 0.0)
 323       ow_interval = DOUBLE_TO_CDTIME_T (tmp);
 324     else
 325       ERROR ("onewire plugin: Invalid `Interval' setting: %s", value);
 326   }
 327   else
 328   {
 329     return (-1);
 330   }
 331
 332   return (0);
 333 }
 334
 335 static int cow_read_values (const char *path, const char *name,
 336     const ow_family_features_t *family_info)
 337 {
 338   value_t values[1];
 339   value_list_t vl = VALUE_LIST_INIT;
 340   int success = 0;
 341   size_t i;
 342
 343   if (sensor_list != NULL)
 344   {
 345     DEBUG ("onewire plugin: Checking ignorelist for `%s'", name);
 346     if (ignorelist_match (sensor_list, name) != 0)
 347       return 0;
 348   }
 349
 350   vl.values = values;
 351   vl.values_len = 1;
 352
 353   sstrncpy (vl.host, hostname_g, sizeof (vl.host));
 354   sstrncpy (vl.plugin, "onewire", sizeof (vl.plugin));
 355   sstrncpy (vl.plugin_instance, name, sizeof (vl.plugin_instance));
 356
 357   for (i = 0; i < family_info->features_num; i++)
 358   {
 359     char *buffer;
 360     size_t buffer_size;
 361     int status;
 362
 363     char file[4096];
 364     char *endptr;
 365
 366     snprintf (file, sizeof (file), "%s/%s",
 367         path, family_info->features[i].filename);
 368     file[sizeof (file) - 1] = 0;
 369
 370     buffer = NULL;
 371     buffer_size = 0;
 372     DEBUG ("Start reading onewire device %s", file);
 373     status = OW_get (file, &buffer, &buffer_size);
 374     if (status < 0)
 375     {
 376       ERROR ("onewire plugin: OW_get (%s/%s) failed. status = %#x;",
 377           path, family_info->features[i].filename, status);
 378       return (-1);
 379     }
 380     DEBUG ("Read onewire device %s as %s", file, buffer);
 381
 382     endptr = NULL;
 383     values[0].gauge = strtod (buffer, &endptr);
 384     if (endptr == NULL)
 385     {
 386       ERROR ("onewire plugin: Buffer is not a number: %s", buffer);
 387       continue;
 388     }
 389
 390     sstrncpy (vl.type, family_info->features[i].type, sizeof (vl.type));
 391     sstrncpy (vl.type_instance, family_info->features[i].type_instance,
 392         sizeof (vl.type_instance));
 393
 394     plugin_dispatch_values (&vl);
 395     success++;
 396
 397     free (buffer);
 398   } /* for (i = 0; i < features_num; i++) */
 399
 400   return ((success > 0) ? 0 : -1);
 401 } /* int cow_read_values */
 402
 403 /* Forward declaration so the recursion below works */
 404 static int cow_read_bus (const char *path);
 405
 406 /*
 407  * cow_read_ds2409
 408  *
 409  * Handles:
 410  * - DS2409 - MicroLAN Coupler
 411  */
 412 static int cow_read_ds2409 (const char *path)
 413 {
 414   char subpath[4096];
 415   int status;
 416
 417   status = ssnprintf (subpath, sizeof (subpath), "%s/main", path);
 418   if ((status > 0) && (status < sizeof (subpath)))
 419     cow_read_bus (subpath);
 420
 421   status = ssnprintf (subpath, sizeof (subpath), "%s/aux", path);
 422   if ((status > 0) && (status < sizeof (subpath)))
 423     cow_read_bus (subpath);
 424
 425   return (0);
 426 } /* int cow_read_ds2409 */
 427
 428 static int cow_read_bus (const char *path)
 429 {
 430   char *buffer;
 431   size_t buffer_size;
 432   int status;
 433
 434   char *buffer_ptr;
 435   char *dummy;
 436   char *saveptr;
 437   char subpath[4096];
 438
 439   status = OW_get (path, &buffer, &buffer_size);
 440   if (status < 0)
 441   {
 442     ERROR ("onewire plugin: OW_get (%s) failed. status = %#x;",
 443         path, status);
 444     return (-1);
 445   }
 446   DEBUG ("onewire plugin: OW_get (%s) returned: %s",
 447       path, buffer);
 448
 449   dummy = buffer;
 450   saveptr = NULL;
 451   while ((buffer_ptr = strtok_r (dummy, ",/", &saveptr)) != NULL)
 452   {
 453     int i;
 454
 455     dummy = NULL;
 456
 457     if (strcmp ("/", path) == 0)
 458       status = ssnprintf (subpath, sizeof (subpath), "/%s", buffer_ptr);
 459     else
 460       status = ssnprintf (subpath, sizeof (subpath), "%s/%s",
 461           path, buffer_ptr);
 462     if ((status <= 0) || (status >= sizeof (subpath)))
 463       continue;
 464
 465     for (i = 0; i < ow_family_features_num; i++)
 466     {
 467       if (strncmp (ow_family_features[i].family, buffer_ptr,
 468             strlen (ow_family_features[i].family)) != 0)
 469         continue;
 470
 471       cow_read_values (subpath,
 472           buffer_ptr + strlen (ow_family_features[i].family),
 473           ow_family_features + i);
 474       break;
 475     }
 476     if (i < ow_family_features_num)
 477       continue;
 478
 479     /* DS2409 */
 480     if (strncmp ("1F.", buffer_ptr, strlen ("1F.")) == 0)
 481     {
 482       cow_read_ds2409 (subpath);
 483       continue;
 484     }
 485   } /* while (strtok_r) */
 486
 487   free (buffer);
 488   return (0);
 489 } /* int cow_read_bus */
 490
 491
 492 /* =================================================================================== */
 493
 494 static int cow_simple_read (void)
 495 {
 496   value_t      values[1];
 497   value_list_t vl = VALUE_LIST_INIT;
 498   char        *buffer;
 499   size_t       buffer_size;
 500   int          status;
 501   char        *endptr;
 502   direct_access_element_t *traverse;
 503
 504   /* traverse list and check entries */
 505   for (traverse = direct_list; traverse != NULL; traverse = traverse->next)
 506   {
 507       vl.values = values;
 508       vl.values_len = 1;
 509
 510       sstrncpy (vl.host, hostname_g, sizeof (vl.host));
 511       sstrncpy (vl.plugin, "onewire", sizeof (vl.plugin));
 512       sstrncpy (vl.plugin_instance, traverse->address, sizeof (vl.plugin_instance));
 513
 514       status = OW_get (traverse->path, &buffer, &buffer_size);
 515       if (status < 0)
 516       {
 517           ERROR ("onewire plugin: OW_get (%s) failed. status = %#x;",
 518                  traverse->path,
 519                  status);
 520           return (-1);
 521       }
 522       DEBUG ("onewire plugin: Read onewire device %s as %s", traverse->path, buffer);
 523
 524
 525       endptr = NULL;
 526       values[0].gauge = strtod (buffer, &endptr);
 527       if (endptr == NULL)
 528       {
 529           ERROR ("onewire plugin: Buffer is not a number: %s", buffer);
 530           continue;
 531       }
 532
 533       sstrncpy (vl.type, traverse->file, sizeof (vl.type));
 534       sstrncpy (vl.type_instance, "",   sizeof (""));
 535
 536       plugin_dispatch_values (&vl);
 537       free (buffer);
 538   } /* for (traverse) */
 539
 540   return 0;
 541 } /* int cow_simple_read */
 542
 543 /* =================================================================================== */
 544
 545 static int cow_read (user_data_t *ud __attribute__((unused)))
 546 {
 547     int result=0;
 548
 549 #if COLLECT_DEBUG
 550     gettimeofday (&tv_begin, NULL);
 551 #endif /* COLLECT_DEBUG */
 552
 553     if (direct_access)
 554     {
 555         DEBUG ("onewire plugin: Direct access read");
 556         result = cow_simple_read ();
 557     }
 558     else
 559     {
 560         DEBUG ("onewire plugin: Standard access read");
 561         result = cow_read_bus ("/");
 562     }
 563
 564 #if COLLECT_DEBUG
 565     gettimeofday (&tv_end, NULL);
 566     timeval_subtract (&tv_diff, &tv_end, &tv_begin);
 567     DEBUG ("onewire plugin: Onewire read took us %ld.%06ld s",
 568            tv_diff.tv_sec,
 569            tv_diff.tv_usec);
 570 #endif /* COLLECT_DEBUG */
 571
 572     return result;
 573 } /* int cow_read */
 574
 575 static int cow_shutdown (void)
 576 {
 577     OW_finish ();
 578     ignorelist_free (sensor_list);
 579
 580     direct_list_free ();
 581
 582     if (regex_direct_initialized)
 583     {
 584         regfree(&regex_direct);
 585     }
 586
 587     return (0);
 588 } /* int cow_shutdown */
 589
 590 static int cow_init (void)
 591 {
 592   int status;
 593   struct timespec cb_interval;
 594
 595   if (device_g == NULL)
 596   {
 597     ERROR ("onewire plugin: cow_init: No device configured.");
 598     return (-1);
 599   }
 600
 601   DEBUG ("onewire plugin: about to init device <%s>.", device_g);
 602   status = (int) OW_init (device_g);
 603   if (status != 0)
 604   {
 605     ERROR ("onewire plugin: OW_init(%s) failed: %i.", device_g, status);
 606     return (1);
 607   }
 608
 609   CDTIME_T_TO_TIMESPEC (ow_interval, &cb_interval);
 610
 611   plugin_register_complex_read (/* group = */ NULL, "onewire", cow_read,
 612       (ow_interval != 0) ? &cb_interval : NULL,
 613       /* user data = */ NULL);
 614   plugin_register_shutdown ("onewire", cow_shutdown);
 615
 616   return (0);
 617 } /* int cow_init */
 618
 619 void module_register (void)
 620 {
 621   plugin_register_init ("onewire", cow_init);
 622   plugin_register_config ("onewire", cow_load_config,
 623                           config_keys, config_keys_num);
 624 }
 625
 626 /* vim: set sw=2 sts=2 ts=8 et fdm=marker cindent : */