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