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