2f685e6e96c6733647f42875cd3bd47d55c85170
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 **/
23 #include "collectd.h"
25 #include "common.h"
26 #include "plugin.h"
27 #include "configfile.h"
29 #include <netdb.h>
31 #include <modbus.h>
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
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
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 */
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;
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;
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];
111 mb_data_t *next;
112 }; /* }}} */
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;
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;
132 mb_slave_t *slaves;
133 size_t slaves_num;
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;
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;
151 mb_data_group_t *next;
152 }; /* }}} */
154 /*
155 * Global variables
156 */
157 static mb_data_t *data_definitions = NULL;
159 /*
160 * Functions
161 */
162 static mb_data_t *data_get_by_name (mb_data_t *src, /* {{{ */
163 const char *name)
164 {
165 mb_data_t *ptr;
167 if (name == NULL)
168 return (NULL);
170 for (ptr = src; ptr != NULL; ptr = ptr->next)
171 if (strcasecmp (ptr->name, name) == 0)
172 return (ptr);
174 return (NULL);
175 } /* }}} mb_data_t *data_get_by_name */
177 static int data_append (mb_data_t **dst, mb_data_t *src) /* {{{ */
178 {
179 mb_data_t *ptr;
181 if ((dst == NULL) || (src == NULL))
182 return (EINVAL);
184 ptr = *dst;
186 if (ptr == NULL)
187 {
188 *dst = src;
189 return (0);
190 }
192 while (ptr->next != NULL)
193 ptr = ptr->next;
195 ptr->next = src;
197 return (0);
198 } /* }}} int data_append */
200 /* Copy a single mb_data_t and append it to another list. */
201 static int data_copy (mb_data_t **dst, const mb_data_t *src) /* {{{ */
202 {
203 mb_data_t *tmp;
204 int status;
206 if ((dst == NULL) || (src == NULL))
207 return (EINVAL);
209 tmp = malloc (sizeof (*tmp));
210 if (tmp == NULL)
211 return (ENOMEM);
212 memcpy (tmp, src, sizeof (*tmp));
213 tmp->name = NULL;
214 tmp->next = NULL;
216 tmp->name = strdup (src->name);
217 if (tmp->name == NULL)
218 {
219 sfree (tmp);
220 return (ENOMEM);
221 }
223 status = data_append (dst, tmp);
224 if (status != 0)
225 {
226 sfree (tmp->name);
227 sfree (tmp);
228 return (status);
229 }
231 return (0);
232 } /* }}} int data_copy */
234 /* Lookup a single mb_data_t instance, copy it and append the copy to another
235 * list. */
236 static int data_copy_by_name (mb_data_t **dst, mb_data_t *src, /* {{{ */
237 const char *name)
238 {
239 mb_data_t *ptr;
241 if ((dst == NULL) || (src == NULL) || (name == NULL))
242 return (EINVAL);
244 ptr = data_get_by_name (src, name);
245 if (ptr == NULL)
246 return (ENOENT);
248 return (data_copy (dst, ptr));
249 } /* }}} int data_copy_by_name */
251 /* Read functions */
253 static int mb_submit (mb_host_t *host, mb_slave_t *slave, /* {{{ */
254 mb_data_t *data, value_t value)
255 {
256 value_list_t vl = VALUE_LIST_INIT;
258 if ((host == NULL) || (slave == NULL) || (data == NULL))
259 return (EINVAL);
261 if (host->interval == 0)
262 host->interval = plugin_get_interval ();
264 if (slave->instance[0] == 0)
265 ssnprintf (slave->instance, sizeof (slave->instance), "slave_%i",
266 slave->id);
268 vl.values = &value;
269 vl.values_len = 1;
270 vl.interval = host->interval;
271 sstrncpy (vl.host, host->host, sizeof (vl.host));
272 sstrncpy (vl.plugin, "modbus", sizeof (vl.plugin));
273 sstrncpy (vl.plugin_instance, slave->instance, sizeof (vl.plugin_instance));
274 sstrncpy (vl.type, data->type, sizeof (vl.type));
275 sstrncpy (vl.type_instance, data->instance, sizeof (vl.type_instance));
277 return (plugin_dispatch_values (&vl));
278 } /* }}} int mb_submit */
280 static float mb_register_to_float (uint16_t hi, uint16_t lo) /* {{{ */
281 {
282 union
283 {
284 uint8_t b[4];
285 uint16_t s[2];
286 float f;
287 } conv;
289 #if BYTE_ORDER == LITTLE_ENDIAN
290 /* little endian */
291 conv.b[0] = lo & 0x00ff;
292 conv.b[1] = (lo >> 8) & 0x00ff;
293 conv.b[2] = hi & 0x00ff;
294 conv.b[3] = (hi >> 8) & 0x00ff;
295 #else
296 conv.b[3] = lo & 0x00ff;
297 conv.b[2] = (lo >> 8) & 0x00ff;
298 conv.b[1] = hi & 0x00ff;
299 conv.b[0] = (hi >> 8) & 0x00ff;
300 #endif
302 return (conv.f);
303 } /* }}} float mb_register_to_float */
305 #if LEGACY_LIBMODBUS
306 /* Version 2.0.3 */
307 static int mb_init_connection (mb_host_t *host) /* {{{ */
308 {
309 int status;
311 if (host == NULL)
312 return (EINVAL);
314 #if COLLECT_DEBUG
315 modbus_set_debug (&host->connection, 1);
316 #endif
318 /* We'll do the error handling ourselves. */
319 modbus_set_error_handling (&host->connection, NOP_ON_ERROR);
321 if (host->conntype == MBCONN_TCP)
322 {
323 if ((host->port < 1) || (host->port > 65535))
324 host->port = MODBUS_TCP_DEFAULT_PORT;
326 DEBUG ("Modbus plugin: Trying to connect to \"%s\", port %i.",
327 host->node, host->port);
329 modbus_init_tcp (&host->connection,
330 /* host = */ host->node,
331 /* port = */ host->port);
332 }
333 else /* MBCONN_RTU */
334 {
335 DEBUG ("Modbus plugin: Trying to connect to \"%s\".", host->node);
337 modbus_init_rtu (&host->connection,
338 /* device = */ host->node,
339 /* baudrate = */ host->baudrate,
340 'N', 8, 1, 0);
341 }
343 status = modbus_connect (&host->connection);
344 if (status != 0)
345 {
346 ERROR ("Modbus plugin: modbus_connect (%s, %i) failed with status %i.",
347 host->node, host->port ? host->port : host->baudrate, status);
348 return (status);
349 }
351 host->is_connected = 1;
352 return (0);
353 } /* }}} int mb_init_connection */
354 /* #endif LEGACY_LIBMODBUS */
356 #else /* if !LEGACY_LIBMODBUS */
357 /* Version 2.9.2 */
358 static int mb_init_connection (mb_host_t *host) /* {{{ */
359 {
360 int status;
362 if (host == NULL)
363 return (EINVAL);
365 if (host->connection != NULL)
366 return (0);
368 if (host->conntype == MBCONN_TCP)
369 {
370 if ((host->port < 1) || (host->port > 65535))
371 host->port = MODBUS_TCP_DEFAULT_PORT;
373 DEBUG ("Modbus plugin: Trying to connect to \"%s\", port %i.",
374 host->node, host->port);
376 host->connection = modbus_new_tcp (host->node, host->port);
377 if (host->connection == NULL)
378 {
379 ERROR ("Modbus plugin: Creating new Modbus/TCP object failed.");
380 return (-1);
381 }
382 }
383 else
384 {
385 DEBUG ("Modbus plugin: Trying to connect to \"%s\", baudrate %i.",
386 host->node, host->baudrate);
388 host->connection = modbus_new_rtu (host->node, host->baudrate, 'N', 8, 1);
389 if (host->connection == NULL)
390 {
391 ERROR ("Modbus plugin: Creating new Modbus/RTU object failed.");
392 return (-1);
393 }
394 }
396 #if COLLECT_DEBUG
397 modbus_set_debug (host->connection, 1);
398 #endif
400 /* We'll do the error handling ourselves. */
401 modbus_set_error_recovery (host->connection, 0);
403 status = modbus_connect (host->connection);
404 if (status != 0)
405 {
406 ERROR ("Modbus plugin: modbus_connect (%s, %i) failed with status %i.",
407 host->node, host->port ? host->port : host->baudrate, status);
408 modbus_free (host->connection);
409 host->connection = NULL;
410 return (status);
411 }
413 return (0);
414 } /* }}} int mb_init_connection */
415 #endif /* !LEGACY_LIBMODBUS */
417 #define CAST_TO_VALUE_T(ds,vt,raw) do { \
418 if ((ds)->ds[0].type == DS_TYPE_COUNTER) \
419 (vt).counter = (counter_t) (raw); \
420 else if ((ds)->ds[0].type == DS_TYPE_GAUGE) \
421 (vt).gauge = (gauge_t) (raw); \
422 else if ((ds)->ds[0].type == DS_TYPE_DERIVE) \
423 (vt).derive = (derive_t) (raw); \
424 else /* if (ds->ds[0].type == DS_TYPE_ABSOLUTE) */ \
425 (vt).absolute = (absolute_t) (raw); \
426 } while (0)
428 static int mb_read_data (mb_host_t *host, mb_slave_t *slave, /* {{{ */
429 mb_data_t *data)
430 {
431 uint16_t values[2] = { 0 };
432 int values_num;
433 const data_set_t *ds;
434 int status = 0;
436 if ((host == NULL) || (slave == NULL) || (data == NULL))
437 return (EINVAL);
439 ds = plugin_get_ds (data->type);
440 if (ds == NULL)
441 {
442 ERROR ("Modbus plugin: Type \"%s\" is not defined.", data->type);
443 return (-1);
444 }
446 if (ds->ds_num != 1)
447 {
448 ERROR ("Modbus plugin: The type \"%s\" has %zu data sources. "
449 "I can only handle data sets with only one data source.",
450 data->type, ds->ds_num);
451 return (-1);
452 }
454 if ((ds->ds[0].type != DS_TYPE_GAUGE)
455 && (data->register_type != REG_TYPE_INT32)
456 && (data->register_type != REG_TYPE_UINT32))
457 {
458 NOTICE ("Modbus plugin: The data source of type \"%s\" is %s, not gauge. "
459 "This will most likely result in problems, because the register type "
460 "is not UINT32.", data->type, DS_TYPE_TO_STRING (ds->ds[0].type));
461 }
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;
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);
479 status = getpeername (modbus_get_socket (host->connection),
480 &sockaddr, &saddrlen);
481 if (status != 0)
482 status = errno;
483 }
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 }
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 }
547 DEBUG ("Modbus plugin: mb_read_data: Success! "
548 "modbus_read_registers returned with status %i.", status);
550 if (data->register_type == REG_TYPE_FLOAT)
551 {
552 float float_value;
553 value_t vt;
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);
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;
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);
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;
588 v.u16 = values[0];
590 DEBUG ("Modbus plugin: mb_read_data: "
591 "Returned int16 value is %"PRIi16, v.i16);
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;
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);
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;
613 DEBUG ("Modbus plugin: mb_read_data: "
614 "Returned uint16 value is %"PRIu16, values[0]);
616 CAST_TO_VALUE_T (ds, vt, values[0]);
617 mb_submit (host, slave, data, vt);
618 }
620 return (0);
621 } /* }}} int mb_read_data */
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;
629 if ((host == NULL) || (slave == NULL))
630 return (EINVAL);
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 }
640 if (success == 0)
641 return (-1);
642 else
643 return (0);
644 } /* }}} int mb_read_slave */
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;
653 if ((user_data == NULL) || (user_data->data == NULL))
654 return (EINVAL);
656 host = user_data->data;
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 }
666 if (success == 0)
667 return (-1);
668 else
669 return (0);
670 } /* }}} int mb_read */
672 /* Free functions */
674 static void data_free_one (mb_data_t *data) /* {{{ */
675 {
676 if (data == NULL)
677 return;
679 sfree (data->name);
680 sfree (data);
681 } /* }}} void data_free_one */
683 static void data_free_all (mb_data_t *data) /* {{{ */
684 {
685 mb_data_t *next;
687 if (data == NULL)
688 return;
690 next = data->next;
691 data_free_one (data);
693 data_free_all (next);
694 } /* }}} void data_free_all */
696 static void slaves_free_all (mb_slave_t *slaves, size_t slaves_num) /* {{{ */
697 {
698 size_t i;
700 if (slaves == NULL)
701 return;
703 for (i = 0; i < slaves_num; i++)
704 data_free_all (slaves[i].collect);
705 sfree (slaves);
706 } /* }}} void slaves_free_all */
708 static void host_free (void *void_host) /* {{{ */
709 {
710 mb_host_t *host = void_host;
712 if (host == NULL)
713 return;
715 slaves_free_all (host->slaves, host->slaves_num);
716 sfree (host);
717 } /* }}} void host_free */
719 /* Config functions */
721 static int mb_config_add_data (oconfig_item_t *ci) /* {{{ */
722 {
723 mb_data_t data = { 0 };
724 int status;
725 int i;
727 data.name = NULL;
728 data.register_type = REG_TYPE_UINT16;
729 data.next = NULL;
731 status = cf_util_get_string (ci, &data.name);
732 if (status != 0)
733 return (status);
735 for (i = 0; i < ci->children_num; i++)
736 {
737 oconfig_item_t *child = ci->children + i;
739 if (strcasecmp ("Type", child->key) == 0)
740 status = cf_util_get_string_buffer (child,
741 data.type, sizeof (data.type));
742 else if (strcasecmp ("Instance", child->key) == 0)
743 status = cf_util_get_string_buffer (child,
744 data.instance, sizeof (data.instance));
745 else if (strcasecmp ("RegisterBase", child->key) == 0)
746 status = cf_util_get_int (child, &data.register_base);
747 else if (strcasecmp ("RegisterType", child->key) == 0)
748 {
749 char tmp[16];
750 status = cf_util_get_string_buffer (child, tmp, sizeof (tmp));
751 if (status != 0)
752 /* do nothing */;
753 else if (strcasecmp ("Int16", tmp) == 0)
754 data.register_type = REG_TYPE_INT16;
755 else if (strcasecmp ("Int32", tmp) == 0)
756 data.register_type = REG_TYPE_INT32;
757 else if (strcasecmp ("Uint16", tmp) == 0)
758 data.register_type = REG_TYPE_UINT16;
759 else if (strcasecmp ("Uint32", tmp) == 0)
760 data.register_type = REG_TYPE_UINT32;
761 else if (strcasecmp ("Float", tmp) == 0)
762 data.register_type = REG_TYPE_FLOAT;
763 else
764 {
765 ERROR ("Modbus plugin: The register type \"%s\" is unknown.", tmp);
766 status = -1;
767 }
768 }
769 else if (strcasecmp ("RegisterCmd", child->key) == 0)
770 {
771 #if LEGACY_LIBMODBUS
772 ERROR("Modbus plugin: RegisterCmd parameter can not be used "
773 "with your libmodbus version");
774 #else
775 char tmp[16];
776 status = cf_util_get_string_buffer (child, tmp, sizeof (tmp));
777 if (status != 0)
778 /* do nothing */;
779 else if (strcasecmp ("ReadHolding", tmp) == 0)
780 data.modbus_register_type = MREG_HOLDING;
781 else if (strcasecmp ("ReadInput", tmp) == 0)
782 data.modbus_register_type = MREG_INPUT;
783 else
784 {
785 ERROR ("Modbus plugin: The modbus_register_type \"%s\" is unknown.",
786 tmp);
787 status = -1;
788 }
789 #endif
790 }
791 else
792 {
793 ERROR ("Modbus plugin: Unknown configuration option: %s", child->key);
794 status = -1;
795 }
797 if (status != 0)
798 break;
799 } /* for (i = 0; i < ci->children_num; i++) */
801 assert (data.name != NULL);
802 if (data.type[0] == 0)
803 {
804 ERROR ("Modbus plugin: Data block \"%s\": No type has been specified.",
805 data.name);
806 status = -1;
807 }
809 if (status == 0)
810 data_copy (&data_definitions, &data);
812 sfree (data.name);
814 return (status);
815 } /* }}} int mb_config_add_data */
817 static int mb_config_set_host_address (mb_host_t *host, /* {{{ */
818 const char *address)
819 {
820 struct addrinfo *ai_list;
821 struct addrinfo *ai_ptr;
822 int status;
824 if ((host == NULL) || (address == NULL))
825 return (EINVAL);
827 struct addrinfo ai_hints = {
828 /* XXX: libmodbus can only handle IPv4 addresses. */
829 .ai_family = AF_INET,
830 .ai_flags = AI_ADDRCONFIG
831 };
833 status = getaddrinfo (address, /* service = */ NULL,
834 &ai_hints, &ai_list);
835 if (status != 0)
836 {
837 char errbuf[1024];
838 ERROR ("Modbus plugin: getaddrinfo failed: %s",
839 (status == EAI_SYSTEM)
840 ? sstrerror (errno, errbuf, sizeof (errbuf))
841 : gai_strerror (status));
842 return (status);
843 }
845 for (ai_ptr = ai_list; ai_ptr != NULL; ai_ptr = ai_ptr->ai_next)
846 {
847 status = getnameinfo (ai_ptr->ai_addr, ai_ptr->ai_addrlen,
848 host->node, sizeof (host->node),
849 /* service = */ NULL, /* length = */ 0,
850 /* flags = */ NI_NUMERICHOST);
851 if (status == 0)
852 break;
853 } /* for (ai_ptr) */
855 freeaddrinfo (ai_list);
857 if (status != 0)
858 ERROR ("Modbus plugin: Unable to translate node name: \"%s\"", address);
859 else /* if (status == 0) */
860 {
861 DEBUG ("Modbus plugin: mb_config_set_host_address: %s -> %s",
862 address, host->node);
863 }
865 return (status);
866 } /* }}} int mb_config_set_host_address */
868 static int mb_config_add_slave (mb_host_t *host, oconfig_item_t *ci) /* {{{ */
869 {
870 mb_slave_t *slave;
871 int status;
872 int i;
874 if ((host == NULL) || (ci == NULL))
875 return (EINVAL);
877 slave = realloc (host->slaves, sizeof (*slave) * (host->slaves_num + 1));
878 if (slave == NULL)
879 return (ENOMEM);
880 host->slaves = slave;
881 slave = host->slaves + host->slaves_num;
882 memset (slave, 0, sizeof (*slave));
883 slave->collect = NULL;
885 status = cf_util_get_int (ci, &slave->id);
886 if (status != 0)
887 return (status);
889 for (i = 0; i < ci->children_num; i++)
890 {
891 oconfig_item_t *child = ci->children + i;
893 if (strcasecmp ("Instance", child->key) == 0)
894 status = cf_util_get_string_buffer (child,
895 slave->instance, sizeof (slave->instance));
896 else if (strcasecmp ("Collect", child->key) == 0)
897 {
898 char buffer[1024];
899 status = cf_util_get_string_buffer (child, buffer, sizeof (buffer));
900 if (status == 0)
901 data_copy_by_name (&slave->collect, data_definitions, buffer);
902 status = 0; /* continue after failure. */
903 }
904 else
905 {
906 ERROR ("Modbus plugin: Unknown configuration option: %s", child->key);
907 status = -1;
908 }
910 if (status != 0)
911 break;
912 }
914 if ((status == 0) && (slave->collect == NULL))
915 status = EINVAL;
917 if (slave->id < 0)
918 status = EINVAL;
920 if (status == 0)
921 host->slaves_num++;
922 else /* if (status != 0) */
923 data_free_all (slave->collect);
925 return (status);
926 } /* }}} int mb_config_add_slave */
928 static int mb_config_add_host (oconfig_item_t *ci) /* {{{ */
929 {
930 mb_host_t *host;
931 int status;
932 int i;
934 host = calloc (1, sizeof (*host));
935 if (host == NULL)
936 return (ENOMEM);
937 host->slaves = NULL;
939 status = cf_util_get_string_buffer (ci, host->host, sizeof (host->host));
940 if (status != 0)
941 {
942 sfree (host);
943 return (status);
944 }
945 if (host->host[0] == 0)
946 {
947 sfree (host);
948 return (EINVAL);
949 }
951 for (i = 0; i < ci->children_num; i++)
952 {
953 oconfig_item_t *child = ci->children + i;
954 status = 0;
956 if (strcasecmp ("Address", child->key) == 0)
957 {
958 char buffer[NI_MAXHOST];
959 status = cf_util_get_string_buffer (child, buffer, sizeof (buffer));
960 if (status == 0)
961 status = mb_config_set_host_address (host, buffer);
962 if (status == 0)
963 host->conntype = MBCONN_TCP;
964 }
965 else if (strcasecmp ("Port", child->key) == 0)
966 {
967 host->port = cf_util_get_port_number (child);
968 if (host->port <= 0)
969 status = -1;
970 }
971 else if (strcasecmp ("Device", child->key) == 0)
972 {
973 status = cf_util_get_string_buffer (child, host->node, sizeof (host->node));
974 if (status == 0)
975 host->conntype = MBCONN_RTU;
976 }
977 else if (strcasecmp ("Baudrate", child->key) == 0)
978 status = cf_util_get_int(child, &host->baudrate);
979 else if (strcasecmp ("Interval", child->key) == 0)
980 status = cf_util_get_cdtime (child, &host->interval);
981 else if (strcasecmp ("Slave", child->key) == 0)
982 /* Don't set status: Gracefully continue if a slave fails. */
983 mb_config_add_slave (host, child);
984 else
985 {
986 ERROR ("Modbus plugin: Unknown configuration option: %s", child->key);
987 status = -1;
988 }
990 if (status != 0)
991 break;
992 } /* for (i = 0; i < ci->children_num; i++) */
994 assert (host->host[0] != 0);
995 if (host->node[0] == 0)
996 {
997 ERROR ("Modbus plugin: Data block \"%s\": No address or device has been specified.",
998 host->host);
999 status = -1;
1000 }
1001 if (host->conntype == MBCONN_RTU && !host->baudrate)
1002 {
1003 ERROR ("Modbus plugin: Data block \"%s\": No serial baudrate has been specified.",
1004 host->host);
1005 status = -1;
1006 }
1007 if ((host->conntype == MBCONN_TCP && host->baudrate) ||
1008 (host->conntype == MBCONN_RTU && host->port))
1009 {
1010 ERROR ("Modbus plugin: Data block \"%s\": You've mixed up RTU and TCP options.",
1011 host->host);
1012 status = -1;
1013 }
1015 if (status == 0)
1016 {
1017 user_data_t ud;
1018 char name[1024];
1020 ud.data = host;
1021 ud.free_func = host_free;
1023 ssnprintf (name, sizeof (name), "modbus-%s", host->host);
1025 plugin_register_complex_read (/* group = */ NULL, name,
1026 /* callback = */ mb_read,
1027 /* interval = */ host->interval,
1028 &ud);
1029 }
1030 else
1031 {
1032 host_free (host);
1033 }
1035 return (status);
1036 } /* }}} int mb_config_add_host */
1038 static int mb_config (oconfig_item_t *ci) /* {{{ */
1039 {
1040 int i;
1042 if (ci == NULL)
1043 return (EINVAL);
1045 for (i = 0; i < ci->children_num; i++)
1046 {
1047 oconfig_item_t *child = ci->children + i;
1049 if (strcasecmp ("Data", child->key) == 0)
1050 mb_config_add_data (child);
1051 else if (strcasecmp ("Host", child->key) == 0)
1052 mb_config_add_host (child);
1053 else
1054 ERROR ("Modbus plugin: Unknown configuration option: %s", child->key);
1055 }
1057 return (0);
1058 } /* }}} int mb_config */
1060 /* ========= */
1062 static int mb_shutdown (void) /* {{{ */
1063 {
1064 data_free_all (data_definitions);
1065 data_definitions = NULL;
1067 return (0);
1068 } /* }}} int mb_shutdown */
1070 void module_register (void)
1071 {
1072 plugin_register_complex_config ("modbus", mb_config);
1073 plugin_register_shutdown ("modbus", mb_shutdown);
1074 } /* void module_register */
1076 /* vim: set sw=2 sts=2 et fdm=marker : */