Merge branch 'collectd-5.4' into collectd-5.5
[collectd.git] / src / modbus.c
1 /**
2  * collectd - src/modbus.c
3  * Copyright (C) 2010,2011  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 Lesser General Public License as published by
7  * the Free Software Foundation; only version 2.1 of the License is
8  * applicable.
9  *
10  * This program is distributed in the hope that it will be useful, but
11  * WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * Lesser General Public License for more details.
14  *
15  * You should have received a copy of the GNU Lesser General Public License
16  * along with this program; if not, write to the Free Software Foundation,
17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
18  *
19  * Authors:
20  *   Florian Forster <octo at noris.net>
21  **/
22
23 #include "collectd.h"
24 #include "common.h"
25 #include "plugin.h"
26 #include "configfile.h"
27
28 #include <netdb.h>
29
30 #include <modbus.h>
31
32 #ifndef LIBMODBUS_VERSION_CHECK
33 /* Assume version 2.0.3 */
34 # define LEGACY_LIBMODBUS 1
35 #else
36 /* Assume version 2.9.2 */
37 #endif
38
39 #ifndef MODBUS_TCP_DEFAULT_PORT
40 # ifdef MODBUS_TCP_PORT
41 #  define MODBUS_TCP_DEFAULT_PORT MODBUS_TCP_PORT
42 # else
43 #  define MODBUS_TCP_DEFAULT_PORT 502
44 # endif
45 #endif
46
47 /*
48  * <Data "data_name">
49  *   RegisterBase 1234
50  *   RegisterCmd ReadHolding
51  *   RegisterType float
52  *   Type gauge
53  *   Instance "..."
54  * </Data>
55  *
56  * <Host "name">
57  *   Address "addr"
58  *   Port "1234"
59  *   # Or:
60  *   # Device "/dev/ttyUSB0"
61  *   # Baudrate 38400
62  *   # (Assumes 8N1)
63  *   Interval 60
64  *
65  *   <Slave 1>
66  *     Instance "foobar" # optional
67  *     Collect "data_name"
68  *   </Slave>
69  * </Host>
70  */
71
72 /*
73  * Data structures
74  */
75 enum mb_register_type_e /* {{{ */
76 {
77   REG_TYPE_INT16,
78   REG_TYPE_INT32,
79   REG_TYPE_UINT16,
80   REG_TYPE_UINT32,
81   REG_TYPE_FLOAT
82 }; /* }}} */
83 enum mb_mreg_type_e /* {{{ */ 
84 {
85   MREG_HOLDING,
86   MREG_INPUT
87 }; /* }}} */
88 typedef enum mb_register_type_e mb_register_type_t;
89 typedef enum mb_mreg_type_e mb_mreg_type_t;
90
91 /* TCP or RTU depending on what is specified in host config block */
92 enum mb_conntype_e /* {{{ */
93 {
94   MBCONN_TCP,
95   MBCONN_RTU
96 }; /* }}} */
97 typedef enum mb_conntype_e mb_conntype_t;
98
99 struct mb_data_s;
100 typedef struct mb_data_s mb_data_t;
101 struct mb_data_s /* {{{ */
102 {
103   char *name;
104   int register_base;
105   mb_register_type_t register_type;
106   mb_mreg_type_t modbus_register_type;
107   char type[DATA_MAX_NAME_LEN];
108   char instance[DATA_MAX_NAME_LEN];
109
110   mb_data_t *next;
111 }; /* }}} */
112
113 struct mb_slave_s /* {{{ */
114 {
115   int id;
116   char instance[DATA_MAX_NAME_LEN];
117   mb_data_t *collect;
118 }; /* }}} */
119 typedef struct mb_slave_s mb_slave_t;
120
121 struct mb_host_s /* {{{ */
122 {
123   char host[DATA_MAX_NAME_LEN];
124   char node[NI_MAXHOST];        /* TCP hostname or RTU serial device */
125   /* char service[NI_MAXSERV]; */
126   int port;                     /* for Modbus/TCP */
127   int baudrate;                 /* for Modbus/RTU */
128   mb_conntype_t conntype;
129   cdtime_t interval;
130
131   mb_slave_t *slaves;
132   size_t slaves_num;
133
134 #if LEGACY_LIBMODBUS
135   modbus_param_t connection;
136 #else
137   modbus_t *connection;
138 #endif
139   _Bool is_connected;
140 }; /* }}} */
141 typedef struct mb_host_s mb_host_t;
142
143 struct mb_data_group_s;
144 typedef struct mb_data_group_s mb_data_group_t;
145 struct mb_data_group_s /* {{{ */
146 {
147   mb_data_t *registers;
148   size_t registers_num;
149
150   mb_data_group_t *next;
151 }; /* }}} */
152
153 /*
154  * Global variables
155  */
156 static mb_data_t *data_definitions = NULL;
157
158 /*
159  * Functions
160  */
161 static mb_data_t *data_get_by_name (mb_data_t *src, /* {{{ */
162     const char *name)
163 {
164   mb_data_t *ptr;
165
166   if (name == NULL)
167     return (NULL);
168
169   for (ptr = src; ptr != NULL; ptr = ptr->next)
170     if (strcasecmp (ptr->name, name) == 0)
171       return (ptr);
172
173   return (NULL);
174 } /* }}} mb_data_t *data_get_by_name */
175
176 static int data_append (mb_data_t **dst, mb_data_t *src) /* {{{ */
177 {
178   mb_data_t *ptr;
179
180   if ((dst == NULL) || (src == NULL))
181     return (EINVAL);
182
183   ptr = *dst;
184
185   if (ptr == NULL)
186   {
187     *dst = src;
188     return (0);
189   }
190
191   while (ptr->next != NULL)
192     ptr = ptr->next;
193
194   ptr->next = src;
195
196   return (0);
197 } /* }}} int data_append */
198
199 /* Copy a single mb_data_t and append it to another list. */
200 static int data_copy (mb_data_t **dst, const mb_data_t *src) /* {{{ */
201 {
202   mb_data_t *tmp;
203   int status;
204
205   if ((dst == NULL) || (src == NULL))
206     return (EINVAL);
207
208   tmp = malloc (sizeof (*tmp));
209   if (tmp == NULL)
210     return (ENOMEM);
211   memcpy (tmp, src, sizeof (*tmp));
212   tmp->name = NULL;
213   tmp->next = NULL;
214
215   tmp->name = strdup (src->name);
216   if (tmp->name == NULL)
217   {
218     sfree (tmp);
219     return (ENOMEM);
220   }
221
222   status = data_append (dst, tmp);
223   if (status != 0)
224   {
225     sfree (tmp->name);
226     sfree (tmp);
227     return (status);
228   }
229
230   return (0);
231 } /* }}} int data_copy */
232
233 /* Lookup a single mb_data_t instance, copy it and append the copy to another
234  * list. */
235 static int data_copy_by_name (mb_data_t **dst, mb_data_t *src, /* {{{ */
236     const char *name)
237 {
238   mb_data_t *ptr;
239
240   if ((dst == NULL) || (src == NULL) || (name == NULL))
241     return (EINVAL);
242
243   ptr = data_get_by_name (src, name);
244   if (ptr == NULL)
245     return (ENOENT);
246
247   return (data_copy (dst, ptr));
248 } /* }}} int data_copy_by_name */
249
250 /* Read functions */
251
252 static int mb_submit (mb_host_t *host, mb_slave_t *slave, /* {{{ */
253     mb_data_t *data, value_t value)
254 {
255   value_list_t vl = VALUE_LIST_INIT;
256
257   if ((host == NULL) || (slave == NULL) || (data == NULL))
258     return (EINVAL);
259
260   if (host->interval <= 0)
261     host->interval = plugin_get_interval ();
262
263   if (slave->instance[0] == 0)
264     ssnprintf (slave->instance, sizeof (slave->instance), "slave_%i",
265         slave->id);
266
267   vl.values = &value;
268   vl.values_len = 1;
269   vl.interval = host->interval;
270   sstrncpy (vl.host, host->host, sizeof (vl.host));
271   sstrncpy (vl.plugin, "modbus", sizeof (vl.plugin));
272   sstrncpy (vl.plugin_instance, slave->instance, sizeof (vl.plugin_instance));
273   sstrncpy (vl.type, data->type, sizeof (vl.type));
274   sstrncpy (vl.type_instance, data->instance, sizeof (vl.type_instance));
275
276   return (plugin_dispatch_values (&vl));
277 } /* }}} int mb_submit */
278
279 static float mb_register_to_float (uint16_t hi, uint16_t lo) /* {{{ */
280 {
281   union
282   {
283     uint8_t b[4];
284     uint16_t s[2];
285     float f;
286   } conv;
287
288 #if BYTE_ORDER == LITTLE_ENDIAN
289   /* little endian */
290   conv.b[0] = lo & 0x00ff;
291   conv.b[1] = (lo >> 8) & 0x00ff;
292   conv.b[2] = hi & 0x00ff;
293   conv.b[3] = (hi >> 8) & 0x00ff;
294 #else
295   conv.b[3] = lo & 0x00ff;
296   conv.b[2] = (lo >> 8) & 0x00ff;
297   conv.b[1] = hi & 0x00ff;
298   conv.b[0] = (hi >> 8) & 0x00ff;
299 #endif
300
301   return (conv.f);
302 } /* }}} float mb_register_to_float */
303
304 #if LEGACY_LIBMODBUS
305 /* Version 2.0.3 */
306 static int mb_init_connection (mb_host_t *host) /* {{{ */
307 {
308   int status;
309
310   if (host == NULL)
311     return (EINVAL);
312
313 #if COLLECT_DEBUG
314   modbus_set_debug (&host->connection, 1);
315 #endif
316
317   /* We'll do the error handling ourselves. */
318   modbus_set_error_handling (&host->connection, NOP_ON_ERROR);
319
320   if (host->conntype == MBCONN_TCP)
321   {
322     if ((host->port < 1) || (host->port > 65535))
323       host->port = MODBUS_TCP_DEFAULT_PORT;
324
325     DEBUG ("Modbus plugin: Trying to connect to \"%s\", port %i.",
326         host->node, host->port);
327
328     modbus_init_tcp (&host->connection,
329         /* host = */ host->node,
330         /* port = */ host->port);
331   }
332   else  /* MBCONN_RTU */
333   {
334     DEBUG ("Modbus plugin: Trying to connect to \"%s\".", host->node);
335
336     modbus_init_rtu (&host->connection,
337        /* device = */ host->node,
338      /* baudrate = */ host->baudrate,
339                       'N', 8, 1, 0);
340   }
341
342   status = modbus_connect (&host->connection);
343   if (status != 0)
344   {
345     ERROR ("Modbus plugin: modbus_connect (%s, %i) failed with status %i.",
346         host->node, host->port ? host->port : host->baudrate, status);
347     return (status);
348   }
349
350   host->is_connected = 1;
351   return (0);
352 } /* }}} int mb_init_connection */
353 /* #endif LEGACY_LIBMODBUS */
354
355 #else /* if !LEGACY_LIBMODBUS */
356 /* Version 2.9.2 */
357 static int mb_init_connection (mb_host_t *host) /* {{{ */
358 {
359   int status;
360
361   if (host == NULL)
362     return (EINVAL);
363
364   if (host->connection != NULL)
365     return (0);
366
367   if (host->conntype == MBCONN_TCP)
368   {
369     if ((host->port < 1) || (host->port > 65535))
370       host->port = MODBUS_TCP_DEFAULT_PORT;
371
372     DEBUG ("Modbus plugin: Trying to connect to \"%s\", port %i.",
373         host->node, host->port);
374
375     host->connection = modbus_new_tcp (host->node, host->port);
376     if (host->connection == NULL)
377     {
378       ERROR ("Modbus plugin: Creating new Modbus/TCP object failed.");
379       return (-1);
380     }
381   }
382   else
383   {
384     DEBUG ("Modbus plugin: Trying to connect to \"%s\", baudrate %i.",
385         host->node, host->baudrate);
386
387     host->connection = modbus_new_rtu (host->node, host->baudrate, 'N', 8, 1);
388     if (host->connection == NULL)
389     {
390       ERROR ("Modbus plugin: Creating new Modbus/RTU object failed.");
391       return (-1);
392     }
393   }
394
395 #if COLLECT_DEBUG
396   modbus_set_debug (host->connection, 1);
397 #endif
398
399   /* We'll do the error handling ourselves. */
400   modbus_set_error_recovery (host->connection, 0);
401
402   status = modbus_connect (host->connection);
403   if (status != 0)
404   {
405     ERROR ("Modbus plugin: modbus_connect (%s, %i) failed with status %i.",
406         host->node, host->port ? host->port : host->baudrate, status);
407     modbus_free (host->connection);
408     host->connection = NULL;
409     return (status);
410   }
411
412   return (0);
413 } /* }}} int mb_init_connection */
414 #endif /* !LEGACY_LIBMODBUS */
415
416 #define CAST_TO_VALUE_T(ds,vt,raw) do { \
417   if ((ds)->ds[0].type == DS_TYPE_COUNTER) \
418     (vt).counter = (counter_t) (raw); \
419   else if ((ds)->ds[0].type == DS_TYPE_GAUGE) \
420     (vt).gauge = (gauge_t) (raw); \
421   else if ((ds)->ds[0].type == DS_TYPE_DERIVE) \
422     (vt).derive = (derive_t) (raw); \
423   else /* if (ds->ds[0].type == DS_TYPE_ABSOLUTE) */ \
424     (vt).absolute = (absolute_t) (raw); \
425 } while (0)
426
427 static int mb_read_data (mb_host_t *host, mb_slave_t *slave, /* {{{ */
428     mb_data_t *data)
429 {
430   uint16_t values[2];
431   int values_num;
432   const data_set_t *ds;
433   int status = 0;
434
435   if ((host == NULL) || (slave == NULL) || (data == NULL))
436     return (EINVAL);
437
438   ds = plugin_get_ds (data->type);
439   if (ds == NULL)
440   {
441     ERROR ("Modbus plugin: Type \"%s\" is not defined.", data->type);
442     return (-1);
443   }
444
445   if (ds->ds_num != 1)
446   {
447     ERROR ("Modbus plugin: The type \"%s\" has %i data sources. "
448         "I can only handle data sets with only one data source.",
449         data->type, ds->ds_num);
450     return (-1);
451   }
452
453   if ((ds->ds[0].type != DS_TYPE_GAUGE)
454       && (data->register_type != REG_TYPE_INT32)
455       && (data->register_type != REG_TYPE_UINT32))
456   {
457     NOTICE ("Modbus plugin: The data source of type \"%s\" is %s, not gauge. "
458         "This will most likely result in problems, because the register type "
459         "is not UINT32.", data->type, DS_TYPE_TO_STRING (ds->ds[0].type));
460   }
461
462   memset (values, 0, sizeof (values));
463   if ((data->register_type == REG_TYPE_INT32)
464       || (data->register_type == REG_TYPE_UINT32)
465       || (data->register_type == REG_TYPE_FLOAT))
466     values_num = 2;
467   else
468     values_num = 1;
469
470   if (host->connection == NULL)
471   {
472     status = EBADF;
473   }
474   else if (host->conntype == MBCONN_TCP)
475   {
476     struct sockaddr sockaddr;
477     socklen_t saddrlen = sizeof (sockaddr);
478
479     status = getpeername (modbus_get_socket (host->connection),
480         &sockaddr, &saddrlen);
481     if (status != 0)
482       status = errno;
483   }
484
485   if ((status == EBADF) || (status == ENOTSOCK) || (status == ENOTCONN))
486   {
487     status = mb_init_connection (host);
488     if (status != 0)
489     {
490       ERROR ("Modbus plugin: mb_init_connection (%s/%s) failed. ",
491           host->host, host->node);
492       host->is_connected = 0;
493       host->connection = NULL;
494       return (-1);
495     }
496   }
497   else if (status != 0)
498   {
499 #if LEGACY_LIBMODBUS
500     modbus_close (&host->connection);
501 #else
502     modbus_close (host->connection);
503     modbus_free (host->connection);
504 #endif
505   }
506  
507 #if LEGACY_LIBMODBUS
508   /* Version 2.0.3: Pass the connection struct as a pointer and pass the slave
509    * id to each call of "read_holding_registers". */
510 # define modbus_read_registers(ctx, addr, nb, dest) \
511   read_holding_registers (&(ctx), slave->id, (addr), (nb), (dest))
512 #else /* if !LEGACY_LIBMODBUS */
513   /* Version 2.9.2: Set the slave id once before querying the registers. */
514   status = modbus_set_slave (host->connection, slave->id);
515   if (status != 0)
516   {
517     ERROR ("Modbus plugin: modbus_set_slave (%i) failed with status %i.",
518         slave->id, status);
519     return (-1);
520   }
521 #endif
522   if (data->modbus_register_type == MREG_INPUT){
523     status = modbus_read_input_registers (host->connection,
524         /* start_addr = */ data->register_base,
525         /* num_registers = */ values_num, /* buffer = */ values);
526   }
527   else{
528     status = modbus_read_registers (host->connection,
529         /* start_addr = */ data->register_base,
530         /* num_registers = */ values_num, /* buffer = */ values);
531   }
532   if (status != values_num)
533   {
534     ERROR ("Modbus plugin: modbus read function (%s/%s) failed. "
535            " status = %i, values_num = %i. Giving up.",
536            host->host, host->node, status, values_num);
537 #if LEGACY_LIBMODBUS
538     modbus_close (&host->connection);
539 #else
540     modbus_close (host->connection);
541     modbus_free (host->connection);
542 #endif
543     host->connection = NULL;
544     return (-1);
545   }
546
547   DEBUG ("Modbus plugin: mb_read_data: Success! "
548       "modbus_read_registers returned with status %i.", status);
549
550   if (data->register_type == REG_TYPE_FLOAT)
551   {
552     float float_value;
553     value_t vt;
554
555     float_value = mb_register_to_float (values[0], values[1]);
556     DEBUG ("Modbus plugin: mb_read_data: "
557         "Returned float value is %g", (double) float_value);
558
559     CAST_TO_VALUE_T (ds, vt, float_value);
560     mb_submit (host, slave, data, vt);
561   }
562   else if (data->register_type == REG_TYPE_INT32)
563   {
564     union
565     {
566       uint32_t u32;
567       int32_t  i32;
568     } v;
569     value_t vt;
570
571     v.u32 = (((uint32_t) values[0]) << 16)
572       | ((uint32_t) values[1]);
573     DEBUG ("Modbus plugin: mb_read_data: "
574         "Returned int32 value is %"PRIi32, v.i32);
575
576     CAST_TO_VALUE_T (ds, vt, v.i32);
577     mb_submit (host, slave, data, vt);
578   }
579   else if (data->register_type == REG_TYPE_INT16)
580   {
581     union
582     {
583       uint16_t u16;
584       int16_t  i16;
585     } v;
586     value_t vt;
587
588     v.u16 = values[0];
589
590     DEBUG ("Modbus plugin: mb_read_data: "
591         "Returned int16 value is %"PRIi16, v.i16);
592
593     CAST_TO_VALUE_T (ds, vt, v.i16);
594     mb_submit (host, slave, data, vt);
595   }
596   else if (data->register_type == REG_TYPE_UINT32)
597   {
598     uint32_t v32;
599     value_t vt;
600
601     v32 = (((uint32_t) values[0]) << 16)
602       | ((uint32_t) values[1]);
603     DEBUG ("Modbus plugin: mb_read_data: "
604         "Returned uint32 value is %"PRIu32, v32);
605
606     CAST_TO_VALUE_T (ds, vt, v32);
607     mb_submit (host, slave, data, vt);
608   }
609   else /* if (data->register_type == REG_TYPE_UINT16) */
610   {
611     value_t vt;
612
613     DEBUG ("Modbus plugin: mb_read_data: "
614         "Returned uint16 value is %"PRIu16, values[0]);
615
616     CAST_TO_VALUE_T (ds, vt, values[0]);
617     mb_submit (host, slave, data, vt);
618   }
619
620   return (0);
621 } /* }}} int mb_read_data */
622
623 static int mb_read_slave (mb_host_t *host, mb_slave_t *slave) /* {{{ */
624 {
625   mb_data_t *data;
626   int success;
627   int status;
628
629   if ((host == NULL) || (slave == NULL))
630     return (EINVAL);
631
632   success = 0;
633   for (data = slave->collect; data != NULL; data = data->next)
634   {
635     status = mb_read_data (host, slave, data);
636     if (status == 0)
637       success++;
638   }
639
640   if (success == 0)
641     return (-1);
642   else
643     return (0);
644 } /* }}} int mb_read_slave */
645
646 static int mb_read (user_data_t *user_data) /* {{{ */
647 {
648   mb_host_t *host;
649   size_t i;
650   int success;
651   int status;
652
653   if ((user_data == NULL) || (user_data->data == NULL))
654     return (EINVAL);
655
656   host = user_data->data;
657
658   success = 0;
659   for (i = 0; i < host->slaves_num; i++)
660   {
661     status = mb_read_slave (host, host->slaves + i);
662     if (status == 0)
663       success++;
664   }
665
666   if (success == 0)
667     return (-1);
668   else
669     return (0);
670 } /* }}} int mb_read */
671
672 /* Free functions */
673
674 static void data_free_one (mb_data_t *data) /* {{{ */
675 {
676   if (data == NULL)
677     return;
678
679   sfree (data->name);
680   sfree (data);
681 } /* }}} void data_free_one */
682
683 static void data_free_all (mb_data_t *data) /* {{{ */
684 {
685   mb_data_t *next;
686
687   if (data == NULL)
688     return;
689
690   next = data->next;
691   data_free_one (data);
692
693   data_free_all (next);
694 } /* }}} void data_free_all */
695
696 static void slaves_free_all (mb_slave_t *slaves, size_t slaves_num) /* {{{ */
697 {
698   size_t i;
699
700   if (slaves == NULL)
701     return;
702
703   for (i = 0; i < slaves_num; i++)
704     data_free_all (slaves[i].collect);
705   sfree (slaves);
706 } /* }}} void slaves_free_all */
707
708 static void host_free (void *void_host) /* {{{ */
709 {
710   mb_host_t *host = void_host;
711
712   if (host == NULL)
713     return;
714
715   slaves_free_all (host->slaves, host->slaves_num);
716   sfree (host);
717 } /* }}} void host_free */
718
719 /* Config functions */
720
721 static int mb_config_add_data (oconfig_item_t *ci) /* {{{ */
722 {
723   mb_data_t data;
724   int status;
725   int i;
726
727   memset (&data, 0, sizeof (data));
728   data.name = NULL;
729   data.register_type = REG_TYPE_UINT16;
730   data.next = NULL;
731
732   status = cf_util_get_string (ci, &data.name);
733   if (status != 0)
734     return (status);
735
736   for (i = 0; i < ci->children_num; i++)
737   {
738     oconfig_item_t *child = ci->children + i;
739
740     if (strcasecmp ("Type", child->key) == 0)
741       status = cf_util_get_string_buffer (child,
742           data.type, sizeof (data.type));
743     else if (strcasecmp ("Instance", child->key) == 0)
744       status = cf_util_get_string_buffer (child,
745           data.instance, sizeof (data.instance));
746     else if (strcasecmp ("RegisterBase", child->key) == 0)
747       status = cf_util_get_int (child, &data.register_base);
748     else if (strcasecmp ("RegisterType", child->key) == 0)
749     {
750       char tmp[16];
751       status = cf_util_get_string_buffer (child, tmp, sizeof (tmp));
752       if (status != 0)
753         /* do nothing */;
754       else if (strcasecmp ("Int16", tmp) == 0)
755         data.register_type = REG_TYPE_INT16;
756       else if (strcasecmp ("Int32", tmp) == 0)
757         data.register_type = REG_TYPE_INT32;
758       else if (strcasecmp ("Uint16", tmp) == 0)
759         data.register_type = REG_TYPE_UINT16;
760       else if (strcasecmp ("Uint32", tmp) == 0)
761         data.register_type = REG_TYPE_UINT32;
762       else if (strcasecmp ("Float", tmp) == 0)
763         data.register_type = REG_TYPE_FLOAT;
764       else
765       {
766         ERROR ("Modbus plugin: The register type \"%s\" is unknown.", tmp);
767         status = -1;
768       }
769     }
770     else if (strcasecmp ("RegisterCmd", child->key) == 0)
771     {
772 #if LEGACY_LIBMODBUS
773       ERROR("Modbus plugin: RegisterCmd parameter can not be used "
774             "with your libmodbus version");
775 #else
776       char tmp[16];
777       status = cf_util_get_string_buffer (child, tmp, sizeof (tmp));
778       if (status != 0)
779         /* do nothing */;
780       else if (strcasecmp ("ReadHolding", tmp) == 0)
781         data.modbus_register_type = MREG_HOLDING;
782       else if (strcasecmp ("ReadInput", tmp) == 0)
783         data.modbus_register_type = MREG_INPUT;
784       else
785       {
786         ERROR ("Modbus plugin: The modbus_register_type \"%s\" is unknown.",
787                tmp);
788         status = -1;
789       }
790 #endif
791     }
792     else
793     {
794       ERROR ("Modbus plugin: Unknown configuration option: %s", child->key);
795       status = -1;
796     }
797
798     if (status != 0)
799       break;
800   } /* for (i = 0; i < ci->children_num; i++) */
801
802   assert (data.name != NULL);
803   if (data.type[0] == 0)
804   {
805     ERROR ("Modbus plugin: Data block \"%s\": No type has been specified.",
806         data.name);
807     status = -1;
808   }
809
810   if (status == 0)
811     data_copy (&data_definitions, &data);
812
813   sfree (data.name);
814
815   return (status);
816 } /* }}} int mb_config_add_data */
817
818 static int mb_config_set_host_address (mb_host_t *host, /* {{{ */
819     const char *address)
820 {
821   struct addrinfo *ai_list;
822   struct addrinfo *ai_ptr;
823   struct addrinfo  ai_hints;
824   int status;
825
826   if ((host == NULL) || (address == NULL))
827     return (EINVAL);
828
829   memset (&ai_hints, 0, sizeof (ai_hints));
830 #if AI_ADDRCONFIG
831   ai_hints.ai_flags |= AI_ADDRCONFIG;
832 #endif
833   /* XXX: libmodbus can only handle IPv4 addresses. */
834   ai_hints.ai_family = AF_INET;
835   ai_hints.ai_addr = NULL;
836   ai_hints.ai_canonname = NULL;
837   ai_hints.ai_next = NULL;
838
839   ai_list = NULL;
840   status = getaddrinfo (address, /* service = */ NULL,
841       &ai_hints, &ai_list);
842   if (status != 0)
843   {
844     char errbuf[1024];
845     ERROR ("Modbus plugin: getaddrinfo failed: %s",
846         (status == EAI_SYSTEM)
847         ? sstrerror (errno, errbuf, sizeof (errbuf))
848         : gai_strerror (status));
849     return (status);
850   }
851
852   for (ai_ptr = ai_list; ai_ptr != NULL; ai_ptr = ai_ptr->ai_next)
853   {
854     status = getnameinfo (ai_ptr->ai_addr, ai_ptr->ai_addrlen,
855         host->node, sizeof (host->node),
856         /* service = */ NULL, /* length = */ 0,
857         /* flags = */ NI_NUMERICHOST);
858     if (status == 0)
859       break;
860   } /* for (ai_ptr) */
861
862   freeaddrinfo (ai_list);
863
864   if (status != 0)
865     ERROR ("Modbus plugin: Unable to translate node name: \"%s\"", address);
866   else /* if (status == 0) */
867   {
868     DEBUG ("Modbus plugin: mb_config_set_host_address: %s -> %s",
869         address, host->node);
870   }
871
872   return (status);
873 } /* }}} int mb_config_set_host_address */
874
875 static int mb_config_add_slave (mb_host_t *host, oconfig_item_t *ci) /* {{{ */
876 {
877   mb_slave_t *slave;
878   int status;
879   int i;
880
881   if ((host == NULL) || (ci == NULL))
882     return (EINVAL);
883
884   slave = realloc (host->slaves, sizeof (*slave) * (host->slaves_num + 1));
885   if (slave == NULL)
886     return (ENOMEM);
887   host->slaves = slave;
888   slave = host->slaves + host->slaves_num;
889   memset (slave, 0, sizeof (*slave));
890   slave->collect = NULL;
891
892   status = cf_util_get_int (ci, &slave->id);
893   if (status != 0)
894     return (status);
895
896   for (i = 0; i < ci->children_num; i++)
897   {
898     oconfig_item_t *child = ci->children + i;
899
900     if (strcasecmp ("Instance", child->key) == 0)
901       status = cf_util_get_string_buffer (child,
902           slave->instance, sizeof (slave->instance));
903     else if (strcasecmp ("Collect", child->key) == 0)
904     {
905       char buffer[1024];
906       status = cf_util_get_string_buffer (child, buffer, sizeof (buffer));
907       if (status == 0)
908         data_copy_by_name (&slave->collect, data_definitions, buffer);
909       status = 0; /* continue after failure. */
910     }
911     else
912     {
913       ERROR ("Modbus plugin: Unknown configuration option: %s", child->key);
914       status = -1;
915     }
916
917     if (status != 0)
918       break;
919   }
920
921   if ((status == 0) && (slave->collect == NULL))
922     status = EINVAL;
923
924   if (slave->id < 0)
925     status = EINVAL;
926
927   if (status == 0)
928     host->slaves_num++;
929   else /* if (status != 0) */
930     data_free_all (slave->collect);
931
932   return (status);
933 } /* }}} int mb_config_add_slave */
934
935 static int mb_config_add_host (oconfig_item_t *ci) /* {{{ */
936 {
937   mb_host_t *host;
938   int status;
939   int i;
940
941   host = malloc (sizeof (*host));
942   if (host == NULL)
943     return (ENOMEM);
944   memset (host, 0, sizeof (*host));
945   host->slaves = NULL;
946
947   status = cf_util_get_string_buffer (ci, host->host, sizeof (host->host));
948   if (status != 0)
949   {
950     sfree (host);
951     return (status);
952   }
953   if (host->host[0] == 0)
954   {
955     sfree (host);
956     return (EINVAL);
957   }
958
959   for (i = 0; i < ci->children_num; i++)
960   {
961     oconfig_item_t *child = ci->children + i;
962     status = 0;
963
964     if (strcasecmp ("Address", child->key) == 0)
965     {
966       char buffer[NI_MAXHOST];
967       status = cf_util_get_string_buffer (child, buffer, sizeof (buffer));
968       if (status == 0)
969         status = mb_config_set_host_address (host, buffer);
970       if (status == 0)
971         host->conntype = MBCONN_TCP;
972     }
973     else if (strcasecmp ("Port", child->key) == 0)
974     {
975       host->port = cf_util_get_port_number (child);
976       if (host->port <= 0)
977         status = -1;
978     }
979     else if (strcasecmp ("Device", child->key) == 0)
980     {
981       status = cf_util_get_string_buffer (child, host->node, sizeof (host->node));
982       if (status == 0)
983         host->conntype = MBCONN_RTU;
984     }
985     else if (strcasecmp ("Baudrate", child->key) == 0)
986       status = cf_util_get_int(child, &host->baudrate);
987     else if (strcasecmp ("Interval", child->key) == 0)
988       status = cf_util_get_cdtime (child, &host->interval);
989     else if (strcasecmp ("Slave", child->key) == 0)
990       /* Don't set status: Gracefully continue if a slave fails. */
991       mb_config_add_slave (host, child);
992     else
993     {
994       ERROR ("Modbus plugin: Unknown configuration option: %s", child->key);
995       status = -1;
996     }
997
998     if (status != 0)
999       break;
1000   } /* for (i = 0; i < ci->children_num; i++) */
1001
1002   assert (host->host[0] != 0);
1003   if (host->node[0] == 0)
1004   {
1005     ERROR ("Modbus plugin: Data block \"%s\": No address or device has been specified.",
1006         host->host);
1007     status = -1;
1008   }
1009   if (host->conntype == MBCONN_RTU && !host->baudrate)
1010   {
1011     ERROR ("Modbus plugin: Data block \"%s\": No serial baudrate has been specified.",
1012         host->host);
1013     status = -1;
1014   }
1015   if ((host->conntype == MBCONN_TCP && host->baudrate) ||
1016       (host->conntype == MBCONN_RTU && host->port))
1017   {
1018     ERROR ("Modbus plugin: Data block \"%s\": You've mixed up RTU and TCP options.",
1019         host->host);
1020     status = -1;
1021   }
1022
1023   if (status == 0)
1024   {
1025     user_data_t ud;
1026     char name[1024];
1027     struct timespec interval = { 0, 0 };
1028
1029     ud.data = host;
1030     ud.free_func = host_free;
1031
1032     ssnprintf (name, sizeof (name), "modbus-%s", host->host);
1033
1034     CDTIME_T_TO_TIMESPEC (host->interval, &interval);
1035
1036     plugin_register_complex_read (/* group = */ NULL, name,
1037         /* callback = */ mb_read,
1038         /* interval = */ (host->interval > 0) ? &interval : NULL,
1039         &ud);
1040   }
1041   else
1042   {
1043     host_free (host);
1044   }
1045
1046   return (status);
1047 } /* }}} int mb_config_add_host */
1048
1049 static int mb_config (oconfig_item_t *ci) /* {{{ */
1050 {
1051   int i;
1052
1053   if (ci == NULL)
1054     return (EINVAL);
1055
1056   for (i = 0; i < ci->children_num; i++)
1057   {
1058     oconfig_item_t *child = ci->children + i;
1059
1060     if (strcasecmp ("Data", child->key) == 0)
1061       mb_config_add_data (child);
1062     else if (strcasecmp ("Host", child->key) == 0)
1063       mb_config_add_host (child);
1064     else
1065       ERROR ("Modbus plugin: Unknown configuration option: %s", child->key);
1066   }
1067
1068   return (0);
1069 } /* }}} int mb_config */
1070
1071 /* ========= */
1072
1073 static int mb_shutdown (void) /* {{{ */
1074 {
1075   data_free_all (data_definitions);
1076   data_definitions = NULL;
1077
1078   return (0);
1079 } /* }}} int mb_shutdown */
1080
1081 void module_register (void)
1082 {
1083   plugin_register_complex_config ("modbus", mb_config);
1084   plugin_register_shutdown ("modbus", mb_shutdown);
1085 } /* void module_register */
1086
1087 /* vim: set sw=2 sts=2 et fdm=marker : */