1 /**
2 * collectd - src/teamspeak2.c
3 * Copyright (C) 2008 Stefan Hacker
4 * Copyright (C) 2008 Florian Forster
5 *
6 * This program is free software; you can redistribute it and/or modify it
7 * under the terms of the GNU General Public License as published by the
8 * Free Software Foundation; only version 2 of the License is 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 * General Public License for more details.
14 *
15 * You should have received a copy of the GNU General Public License along
16 * with this program; if not, write to the Free Software Foundation, Inc.,
17 * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
18 *
19 * Authors:
20 * Stefan Hacker <d0t at dbclan dot de>
21 * Florian Forster <octo at verplant.org>
22 **/
24 #include "collectd.h"
25 #include "common.h"
26 #include "plugin.h"
28 #include <arpa/inet.h>
29 #include <netinet/in.h>
30 #include <sys/types.h>
31 #include <sys/socket.h>
32 #include <netdb.h>
34 /*
35 * Defines
36 */
37 /* Default host and port */
38 #define DEFAULT_HOST "127.0.0.1"
39 #define DEFAULT_PORT "51234"
41 /*
42 * Variables
43 */
44 /* Server linked list structure */
45 typedef struct vserver_list_s
46 {
47 int port;
48 struct vserver_list_s *next;
49 } vserver_list_t;
50 static vserver_list_t *server_list = NULL;
52 /* Host data */
53 static char *config_host = NULL;
54 static char *config_port = NULL;
56 static FILE *global_read_fh = NULL;
57 static FILE *global_write_fh = NULL;
59 /* Config data */
60 static const char *config_keys[] =
61 {
62 "Host",
63 "Port",
64 "Server"
65 };
66 static int config_keys_num = STATIC_ARRAY_SIZE (config_keys);
68 /*
69 * Functions
70 */
71 static int tss2_add_vserver (int vserver_port)
72 {
73 /*
74 * Adds a new vserver to the linked list
75 */
76 vserver_list_t *entry;
78 /* Check port range */
79 if ((vserver_port <= 0) || (vserver_port > 65535))
80 {
81 ERROR ("teamspeak2 plugin: VServer port is invalid: %i",
82 vserver_port);
83 return (-1);
84 }
86 /* Allocate memory */
87 entry = (vserver_list_t *) malloc (sizeof (vserver_list_t));
88 if (entry == NULL)
89 {
90 ERROR ("teamspeak2 plugin: malloc failed.");
91 return (-1);
92 }
93 memset (entry, 0, sizeof (vserver_list_t));
95 /* Save data */
96 entry->port = vserver_port;
98 /* Insert to list */
99 if(server_list == NULL) {
100 /* Add the server as the first element */
101 server_list = entry;
102 }
103 else {
104 vserver_list_t *prev;
106 /* Add the server to the end of the list */
107 prev = server_list;
108 while (prev->next != NULL)
109 prev = prev->next;
110 prev->next = entry;
111 }
113 INFO ("teamspeak2 plugin: Registered new vserver: %i", vserver_port);
115 return (0);
116 } /* int tss2_add_vserver */
118 static void tss2_submit_gauge (const char *plugin_instance,
119 const char *type, const char *type_instance,
120 gauge_t value)
121 {
122 /*
123 * Submits a gauge value to the collectd daemon
124 */
125 value_t values[1];
126 value_list_t vl = VALUE_LIST_INIT;
128 values[0].gauge = value;
130 vl.values = values;
131 vl.values_len = 1;
132 vl.time = time (NULL);
133 sstrncpy (vl.host, hostname_g, sizeof (vl.host));
134 sstrncpy (vl.plugin, "teamspeak2", sizeof (vl.plugin));
136 if (plugin_instance != NULL)
137 sstrncpy (vl.plugin_instance, plugin_instance,
138 sizeof (vl.plugin_instance));
140 if (type_instance != NULL)
141 sstrncpy (vl.type_instance, type_instance,
142 sizeof (vl.type_instance));
144 plugin_dispatch_values (type, &vl);
145 } /* void tss2_submit_gauge */
147 static void tss2_submit_io (const char *plugin_instance, const char *type,
148 counter_t rx, counter_t tx)
149 {
150 /*
151 * Submits the io rx/tx tuple to the collectd daemon
152 */
153 value_t values[2];
154 value_list_t vl = VALUE_LIST_INIT;
156 values[0].counter = rx;
157 values[1].counter = tx;
159 vl.values = values;
160 vl.values_len = 2;
161 vl.time = time (NULL);
162 sstrncpy (vl.host, hostname_g, sizeof (vl.host));
163 sstrncpy (vl.plugin, "teamspeak2", sizeof (vl.plugin));
165 if (plugin_instance != NULL)
166 sstrncpy (vl.plugin_instance, plugin_instance,
167 sizeof (vl.plugin_instance));
169 plugin_dispatch_values (type, &vl);
170 } /* void tss2_submit_gauge */
172 static void tss2_close_socket (void)
173 {
174 /*
175 * Closes all sockets
176 */
177 if (global_write_fh != NULL)
178 {
179 fputs ("quit\r\n", global_write_fh);
180 }
182 if (global_read_fh != NULL)
183 {
184 fclose (global_read_fh);
185 global_read_fh = NULL;
186 }
188 if (global_write_fh != NULL)
189 {
190 fclose (global_write_fh);
191 global_write_fh = NULL;
192 }
193 } /* void tss2_close_socket */
195 static int tss2_get_socket (FILE **ret_read_fh, FILE **ret_write_fh)
196 {
197 /*
198 * Returns connected file objects or establishes the connection
199 * if it's not already present
200 */
201 struct addrinfo ai_hints;
202 struct addrinfo *ai_head;
203 struct addrinfo *ai_ptr;
204 int sd = -1;
205 int status;
207 /* Check if we already got opened connections */
208 if ((global_read_fh != NULL) && (global_write_fh != NULL))
209 {
210 /* If so, use them */
211 if (ret_read_fh != NULL)
212 *ret_read_fh = global_read_fh;
213 if (ret_write_fh != NULL)
214 *ret_write_fh = global_write_fh;
215 return (0);
216 }
218 /* Get all addrs for this hostname */
219 memset (&ai_hints, 0, sizeof (ai_hints));
220 #ifdef AI_ADDRCONFIG
221 ai_hints.ai_flags |= AI_ADDRCONFIG;
222 #endif
223 ai_hints.ai_family = AF_UNSPEC;
224 ai_hints.ai_socktype = SOCK_STREAM;
226 status = getaddrinfo ((config_host != NULL) ? config_host : DEFAULT_HOST,
227 (config_port != NULL) ? config_port : DEFAULT_PORT,
228 &ai_hints,
229 &ai_head);
230 if (status != 0)
231 {
232 ERROR ("teamspeak2 plugin: getaddrinfo failed: %s",
233 gai_strerror (status));
234 return (-1);
235 }
237 /* Try all given hosts until we can connect to one */
238 for (ai_ptr = ai_head; ai_ptr != NULL; ai_ptr = ai_ptr->ai_next)
239 {
240 /* Create socket */
241 sd = socket (ai_ptr->ai_family, ai_ptr->ai_socktype,
242 ai_ptr->ai_protocol);
243 if (sd < 0)
244 {
245 char errbuf[1024];
246 WARNING ("teamspeak2 plugin: socket failed: %s",
247 sstrerror (errno, errbuf, sizeof (errbuf)));
248 continue;
249 }
251 /* Try to connect */
252 status = connect (sd, ai_ptr->ai_addr, ai_ptr->ai_addrlen);
253 if (status != 0)
254 {
255 char errbuf[1024];
256 WARNING ("teamspeak2 plugin: connect failed: %s",
257 sstrerror (errno, errbuf, sizeof (errbuf)));
258 close (sd);
259 continue;
260 }
262 /*
263 * Success, we can break. Don't need more than one connection
264 */
265 break;
266 } /* for (ai_ptr) */
268 freeaddrinfo (ai_head);
270 /* Check if we really got connected */
271 if (sd < 0)
272 return (-1);
274 /* Create file objects from sockets */
275 global_read_fh = fdopen (sd, "r");
276 if (global_read_fh == NULL)
277 {
278 char errbuf[1024];
279 ERROR ("teamspeak2 plugin: fdopen failed: %s",
280 sstrerror (errno, errbuf, sizeof (errbuf)));
281 close (sd);
282 return (-1);
283 }
285 global_write_fh = fdopen (sd, "w");
286 if (global_write_fh == NULL)
287 {
288 char errbuf[1024];
289 ERROR ("teamspeak2 plugin: fdopen failed: %s",
290 sstrerror (errno, errbuf, sizeof (errbuf)));
291 tss2_close_socket ();
292 return (-1);
293 }
295 { /* Check that the server correctly identifies itself. */
296 char buffer[4096];
297 char *buffer_ptr;
299 buffer_ptr = fgets (buffer, sizeof (buffer), global_read_fh);
300 buffer[sizeof (buffer) - 1] = 0;
302 if (memcmp ("[TS]\r\n", buffer, 6) != 0)
303 {
304 ERROR ("teamspeak2 plugin: Unexpected response when connecting "
305 "to server. Expected ``[TS]'', got ``%s''.",
306 buffer);
307 tss2_close_socket ();
308 return (-1);
309 }
310 DEBUG ("teamspeak2 plugin: Server send correct banner, connected!");
311 }
313 /* Copy the new filehandles to the given pointers */
314 if (ret_read_fh != NULL)
315 *ret_read_fh = global_read_fh;
316 if (ret_write_fh != NULL)
317 *ret_write_fh = global_write_fh;
318 return (0);
319 } /* int tss2_get_socket */
321 static int tss2_send_request (FILE *fh, const char *request)
322 {
323 /*
324 * This function puts a request to the server socket
325 */
326 int status;
328 status = fputs (request, fh);
329 if (status < 0)
330 {
331 ERROR ("teamspeak2 plugin: fputs failed.");
332 tss2_close_socket ();
333 return (-1);
334 }
335 fflush (fh);
337 return (0);
338 } /* int tss2_send_request */
340 static int tss2_receive_line (FILE *fh, char *buffer, int buffer_size)
341 {
342 /*
343 * Receive a single line from the given file object
344 */
345 char *temp;
347 /*
348 * fgets is blocking but much easier then doing anything else
349 * TODO: Non-blocking Version would be safer
350 */
351 temp = fgets (buffer, buffer_size, fh);
352 if (temp == NULL)
353 {
354 char errbuf[1024];
355 ERROR ("teamspeak2 plugin: fgets failed: %s",
356 sstrerror (errno, errbuf, sizeof(errbuf)));
357 tss2_close_socket ();
358 return (-1);
359 }
361 buffer[buffer_size - 1] = 0;
362 return (0);
363 } /* int tss2_receive_line */
365 static int tss2_select_vserver (FILE *read_fh, FILE *write_fh, vserver_list_t *vserver)
366 {
367 /*
368 * Tell the server to select the given vserver
369 */
370 char command[128];
371 char response[128];
372 int status;
374 /* Send request */
375 snprintf (command, sizeof (command), "sel %i\r\n", vserver->port);
376 command[sizeof (command) - 1] = 0;
378 status = tss2_send_request (write_fh, command);
379 if (status != 0)
380 {
381 ERROR ("teamspeak2 plugin: tss2_send_request (%s) failed.", command);
382 return (-1);
383 }
385 /* Get answer */
386 status = tss2_receive_line (read_fh, response, sizeof (response));
387 if (status != 0)
388 {
389 ERROR ("teamspeak2 plugin: tss2_receive_line failed.");
390 return (-1);
391 }
392 response[sizeof (response) - 1] = 0;
394 /* Check answer */
395 if ((strncasecmp ("OK", response, 2) == 0)
396 && ((response[2] == 0)
397 || (response[2] == '\n')
398 || (response[2] == '\r')))
399 return (0);
401 ERROR ("teamspeak2 plugin: Command ``%s'' failed. "
402 "Response received from server was: ``%s''.",
403 command, response);
404 return (-1);
405 } /* int tss2_select_vserver */
407 static int tss2_vserver_gapl (FILE *read_fh, FILE *write_fh,
408 vserver_list_t *vserver, gauge_t *ret_value)
409 {
410 /*
411 * Reads the vserver's average packet loss and submits it to collectd.
412 * Be sure to run the tss2_read_vserver function before calling this so
413 * the vserver is selected correctly.
414 */
415 gauge_t packet_loss = NAN;
416 int status;
418 status = tss2_send_request (write_fh, "gapl\r\n");
419 if (status != 0)
420 {
421 ERROR("teamspeak2 plugin: tss2_send_request (gapl) failed.");
422 return (-1);
423 }
425 while (42)
426 {
427 char buffer[4096];
428 char *value;
429 char *endptr = NULL;
431 status = tss2_receive_line (read_fh, buffer, sizeof (buffer));
432 if (status != 0)
433 {
434 /* Set to NULL just to make sure noone uses these FHs anymore. */
435 read_fh = NULL;
436 write_fh = NULL;
437 ERROR ("teamspeak2 plugin: tss2_receive_line failed.");
438 return (-1);
439 }
440 buffer[sizeof (buffer) - 1] = 0;
442 if (strncmp ("average_packet_loss=", buffer,
443 strlen ("average_packet_loss=")) == 0)
444 {
445 /* Got average packet loss, now interpret it */
446 value = &buffer[20];
447 /* Replace , with . */
448 while (*value != 0)
449 {
450 if (*value == ',')
451 {
452 *value = '.';
453 break;
454 }
455 value++;
456 }
458 value = &buffer[20];
460 packet_loss = strtod (value, &endptr);
461 if (value == endptr)
462 {
463 /* Failed */
464 WARNING ("teamspeak2 plugin: Could not read average package "
465 "loss from string: %s", buffer);
466 continue;
467 }
468 }
469 else if (strncasecmp ("OK", buffer, 2) == 0)
470 {
471 break;
472 }
473 else if (strncasecmp ("ERROR", buffer, 5) == 0)
474 {
475 ERROR ("teamspeak2 plugin: Server returned an error: %s", buffer);
476 return (-1);
477 }
478 else
479 {
480 WARNING ("teamspeak2 plugin: Server returned unexpected string: %s",
481 buffer);
482 }
483 }
485 *ret_value = packet_loss;
486 return (0);
487 } /* int tss2_vserver_gapl */
489 static int tss2_read_vserver (vserver_list_t *vserver)
490 {
491 /*
492 * Poll information for the given vserver and submit it to collect.
493 * If vserver is NULL the global server information will be queried.
494 */
495 int status;
497 gauge_t users = NAN;
498 gauge_t channels = NAN;
499 gauge_t servers = NAN;
500 counter_t rx_octets = 0;
501 counter_t tx_octets = 0;
502 counter_t rx_packets = 0;
503 counter_t tx_packets = 0;
504 gauge_t packet_loss = NAN;
505 int valid = 0;
507 char plugin_instance[DATA_MAX_NAME_LEN];
509 FILE *read_fh;
510 FILE *write_fh;
512 /* Get the send/receive sockets */
513 status = tss2_get_socket (&read_fh, &write_fh);
514 if (status != 0)
515 {
516 ERROR ("teamspeak2 plugin: tss2_get_socket failed.");
517 return (-1);
518 }
520 if (vserver == NULL)
521 {
522 /* Request global information */
523 memset (plugin_instance, 0, sizeof (plugin_instance));
525 status = tss2_send_request (write_fh, "gi\r\n");
526 }
527 else
528 {
529 /* Request server information */
530 snprintf (plugin_instance, sizeof (plugin_instance), "vserver%i",
531 vserver->port);
532 plugin_instance[sizeof (plugin_instance) - 1] = 0;
534 /* Select the server */
535 status = tss2_select_vserver (read_fh, write_fh, vserver);
536 if (status != 0)
537 return (status);
539 status = tss2_send_request (write_fh, "si\r\n");
540 }
542 if (status != 0)
543 {
544 ERROR ("teamspeak2 plugin: tss2_send_request failed.");
545 return (-1);
546 }
548 /* Loop until break */
549 while (42)
550 {
551 char buffer[4096];
552 char *key;
553 char *value;
554 char *endptr = NULL;
556 /* Read one line of the server's answer */
557 status = tss2_receive_line (read_fh, buffer, sizeof (buffer));
558 if (status != 0)
559 {
560 /* Set to NULL just to make sure noone uses these FHs anymore. */
561 read_fh = NULL;
562 write_fh = NULL;
563 ERROR ("teamspeak2 plugin: tss2_receive_line failed.");
564 break;
565 }
567 if (strncasecmp ("ERROR", buffer, 5) == 0)
568 {
569 ERROR ("teamspeak2 plugin: Server returned an error: %s",
570 buffer);
571 break;
572 }
573 else if (strncasecmp ("OK", buffer, 2) == 0)
574 {
575 break;
576 }
578 /* Split line into key and value */
579 key = strchr (buffer, '_');
580 if (key == NULL)
581 {
582 DEBUG ("teamspeak2 plugin: Cannot parse line: %s", buffer);
583 continue;
584 }
585 key++;
587 /* Evaluate assignment */
588 value = strchr (key, '=');
589 if (value == NULL)
590 {
591 DEBUG ("teamspeak2 plugin: Cannot parse line: %s", buffer);
592 continue;
593 }
594 *value = 0;
595 value++;
597 /* Check for known key and save the given value */
598 /* global info: users_online,
599 * server info: currentusers. */
600 if ((strcmp ("currentusers", key) == 0)
601 || (strcmp ("users_online", key) == 0))
602 {
603 users = strtod (value, &endptr);
604 if (value != endptr)
605 valid |= 0x01;
606 }
607 /* global info: channels,
608 * server info: currentchannels. */
609 else if ((strcmp ("currentchannels", key) == 0)
610 || (strcmp ("channels", key) == 0))
611 {
612 channels = strtod (value, &endptr);
613 if (value != endptr)
614 valid |= 0x40;
615 }
616 /* global only */
617 else if (strcmp ("servers", key) == 0)
618 {
619 servers = strtod (value, &endptr);
620 if (value != endptr)
621 valid |= 0x80;
622 }
623 else if (strcmp ("bytesreceived", key) == 0)
624 {
625 rx_octets = strtoll (value, &endptr, 0);
626 if (value != endptr)
627 valid |= 0x02;
628 }
629 else if (strcmp ("bytessend", key) == 0)
630 {
631 tx_octets = strtoll (value, &endptr, 0);
632 if (value != endptr)
633 valid |= 0x04;
634 }
635 else if (strcmp ("packetsreceived", key) == 0)
636 {
637 rx_packets = strtoll (value, &endptr, 0);
638 if (value != endptr)
639 valid |= 0x08;
640 }
641 else if (strcmp ("packetssend", key) == 0)
642 {
643 tx_packets = strtoll (value, &endptr, 0);
644 if (value != endptr)
645 valid |= 0x10;
646 }
647 else if ((strncmp ("allow_codec_", key, strlen ("allow_codec_")) == 0)
648 || (strncmp ("bwinlast", key, strlen ("bwinlast")) == 0)
649 || (strncmp ("bwoutlast", key, strlen ("bwoutlast")) == 0)
650 || (strncmp ("webpost_", key, strlen ("webpost_")) == 0)
651 || (strcmp ("adminemail", key) == 0)
652 || (strcmp ("clan_server", key) == 0)
653 || (strcmp ("countrynumber", key) == 0)
654 || (strcmp ("id", key) == 0)
655 || (strcmp ("ispname", key) == 0)
656 || (strcmp ("linkurl", key) == 0)
657 || (strcmp ("maxusers", key) == 0)
658 || (strcmp ("name", key) == 0)
659 || (strcmp ("password", key) == 0)
660 || (strcmp ("platform", key) == 0)
661 || (strcmp ("server_platform", key) == 0)
662 || (strcmp ("server_uptime", key) == 0)
663 || (strcmp ("server_version", key) == 0)
664 || (strcmp ("udpport", key) == 0)
665 || (strcmp ("uptime", key) == 0)
666 || (strcmp ("users_maximal", key) == 0)
667 || (strcmp ("welcomemessage", key) == 0))
668 /* ignore */;
669 else
670 {
671 INFO ("teamspeak2 plugin: Unknown key-value-pair: "
672 "key = %s; value = %s;", key, value);
673 }
674 } /* while (42) */
676 /* Collect vserver packet loss rates only if the loop above did not exit
677 * with an error. */
678 if ((status == 0) && (vserver != NULL))
679 {
680 status = tss2_vserver_gapl (read_fh, write_fh, vserver, &packet_loss);
681 if (status == 0)
682 {
683 valid |= 0x20;
684 }
685 else
686 {
687 WARNING ("teamspeak2 plugin: Reading package loss "
688 "for vserver %i failed.", vserver->port);
689 }
690 }
692 if ((valid & 0x01) == 0x01)
693 tss2_submit_gauge (plugin_instance, "users", NULL, users);
695 if ((valid & 0x06) == 0x06)
696 tss2_submit_io (plugin_instance, "io_octets", rx_octets, tx_octets);
698 if ((valid & 0x18) == 0x18)
699 tss2_submit_io (plugin_instance, "io_packets", rx_packets, tx_packets);
701 if ((valid & 0x20) == 0x20)
702 tss2_submit_gauge (plugin_instance, "percent", "packet_loss", packet_loss);
704 if ((valid & 0x40) == 0x40)
705 tss2_submit_gauge (plugin_instance, "gauge", "channels", channels);
707 if ((valid & 0x80) == 0x80)
708 tss2_submit_gauge (plugin_instance, "gauge", "servers", servers);
710 if (valid == 0)
711 return (-1);
712 return (0);
713 } /* int tss2_read_vserver */
715 static int tss2_config (const char *key, const char *value)
716 {
717 /*
718 * Interpret configuration values
719 */
720 if (strcasecmp ("Host", key) == 0)
721 {
722 char *temp;
724 temp = strdup (value);
725 if (temp == NULL)
726 {
727 ERROR("teamspeak2 plugin: strdup failed.");
728 return (1);
729 }
730 sfree (config_host);
731 config_host = temp;
732 }
733 else if (strcasecmp ("Port", key) == 0)
734 {
735 char *temp;
737 temp = strdup (value);
738 if (temp == NULL)
739 {
740 ERROR("teamspeak2 plugin: strdup failed.");
741 return (1);
742 }
743 sfree (config_port);
744 config_port = temp;
745 }
746 else if (strcasecmp ("Server", key) == 0)
747 {
748 /* Server variable found */
749 int status;
751 status = tss2_add_vserver (atoi (value));
752 if (status != 0)
753 return (1);
754 }
755 else
756 {
757 /* Unknown variable found */
758 return (-1);
759 }
761 return 0;
762 } /* int tss2_config */
764 static int tss2_read (void)
765 {
766 /*
767 * Poll function which collects global and vserver information
768 * and submits it to collectd
769 */
770 vserver_list_t *vserver;
771 int success = 0;
772 int status;
774 /* Handle global server variables */
775 status = tss2_read_vserver (NULL);
776 if (status == 0)
777 {
778 success++;
779 }
780 else
781 {
782 WARNING ("teamspeak2 plugin: Reading global server variables failed.");
783 }
785 /* Handle vservers */
786 for (vserver = server_list; vserver != NULL; vserver = vserver->next)
787 {
788 status = tss2_read_vserver (vserver);
789 if (status == 0)
790 {
791 success++;
792 }
793 else
794 {
795 WARNING ("teamspeak2 plugin: Reading statistics "
796 "for vserver %i failed.", vserver->port);
797 continue;
798 }
799 }
801 if (success == 0)
802 return (-1);
803 return (0);
804 } /* int tss2_read */
806 static int tss2_shutdown(void)
807 {
808 /*
809 * Shutdown handler
810 */
811 vserver_list_t *entry;
813 tss2_close_socket ();
815 entry = server_list;
816 server_list = NULL;
817 while (entry != NULL)
818 {
819 vserver_list_t *next;
821 next = entry->next;
822 sfree (entry);
823 entry = next;
824 }
826 /* Get rid of the configuration */
827 sfree (config_host);
828 sfree (config_port);
830 return (0);
831 } /* int tss2_shutdown */
833 void module_register(void)
834 {
835 /*
836 * Mandatory module_register function
837 */
838 plugin_register_config ("teamspeak2", tss2_config,
839 config_keys, config_keys_num);
840 plugin_register_read ("teamspeak2", tss2_read);
841 plugin_register_shutdown ("teamspeak2", tss2_shutdown);
842 } /* void module_register */
844 /* vim: set sw=4 ts=4 : */