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 : */