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