349 lines
9.6 KiB
C
349 lines
9.6 KiB
C
|
/******************************************************************************
|
||
|
|
*
|
||
|
|
* This program is free software; you can redistribute it and/or modify
|
||
|
|
* it under the terms of the GNU General Public License as published by
|
||
|
|
* the Free Software Foundation; either version 2 of the License, or
|
||
|
|
* (at your option) any later version.
|
||
|
|
*
|
||
|
|
* This program is distributed in the hope that it will be useful,
|
||
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
|
* GNU General Public License for more details.
|
||
|
|
*
|
||
|
|
* You should have received a copy of the GNU General Public License
|
||
|
|
* along with this program; if not, write to the Free Software
|
||
|
|
* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
||
|
|
*
|
||
|
|
* $Id: check_game.c,v 1.22 2004/12/25 23:17:44 opensides Exp $
|
||
|
|
*****************************************************************************/
|
||
|
|
|
||
|
|
const char *progname = "check_game";
|
||
|
|
const char *revision = "$Revision: 1.22 $";
|
||
|
|
const char *copyright = "2002-2004";
|
||
|
|
const char *email = "nagiosplug-devel@lists.sourceforge.net";
|
||
|
|
|
||
|
|
#include "common.h"
|
||
|
|
#include "popen.h"
|
||
|
|
#include "utils.h"
|
||
|
|
|
||
|
|
int process_arguments (int, char **);
|
||
|
|
int validate_arguments (void);
|
||
|
|
void print_help (void);
|
||
|
|
void print_usage (void);
|
||
|
|
|
||
|
|
#define QSTAT_DATA_DELIMITER ","
|
||
|
|
|
||
|
|
#define QSTAT_HOST_ERROR "ERROR"
|
||
|
|
#define QSTAT_HOST_DOWN "DOWN"
|
||
|
|
#define QSTAT_HOST_TIMEOUT "TIMEOUT"
|
||
|
|
#define QSTAT_MAX_RETURN_ARGS 12
|
||
|
|
|
||
|
|
char *server_ip;
|
||
|
|
char *game_type;
|
||
|
|
int port = 0;
|
||
|
|
|
||
|
|
int verbose;
|
||
|
|
|
||
|
|
int qstat_game_players_max = -1;
|
||
|
|
int qstat_game_players = -1;
|
||
|
|
int qstat_game_field = -1;
|
||
|
|
int qstat_map_field = -1;
|
||
|
|
int qstat_ping_field = -1;
|
||
|
|
|
||
|
|
|
||
|
|
int
|
||
|
|
main (int argc, char **argv)
|
||
|
|
{
|
||
|
|
char *command_line;
|
||
|
|
int result = STATE_UNKNOWN;
|
||
|
|
FILE *fp;
|
||
|
|
char input_buffer[MAX_INPUT_BUFFER];
|
||
|
|
char *p, *ret[QSTAT_MAX_RETURN_ARGS];
|
||
|
|
int i;
|
||
|
|
|
||
|
|
setlocale (LC_ALL, "");
|
||
|
|
bindtextdomain (PACKAGE, LOCALEDIR);
|
||
|
|
textdomain (PACKAGE);
|
||
|
|
|
||
|
|
if (process_arguments (argc, argv) == ERROR)
|
||
|
|
usage4 (_("Could not parse arguments"));
|
||
|
|
|
||
|
|
result = STATE_OK;
|
||
|
|
|
||
|
|
/* create the command line to execute */
|
||
|
|
asprintf (&command_line, "%s -raw %s -%s %s",
|
||
|
|
PATH_TO_QSTAT, QSTAT_DATA_DELIMITER, game_type, server_ip);
|
||
|
|
|
||
|
|
if (port)
|
||
|
|
asprintf (&command_line, "%s:%-d", command_line, port);
|
||
|
|
|
||
|
|
if (verbose > 0)
|
||
|
|
printf ("%s\n", command_line);
|
||
|
|
|
||
|
|
/* run the command */
|
||
|
|
fp = spopen (command_line);
|
||
|
|
if (fp == NULL) {
|
||
|
|
printf (_("Could not open pipe: %s\n"), command_line);
|
||
|
|
return STATE_UNKNOWN;
|
||
|
|
}
|
||
|
|
|
||
|
|
fgets (input_buffer, MAX_INPUT_BUFFER - 1, fp); /* Only interested in the first line */
|
||
|
|
|
||
|
|
/* strip the newline character from the end of the input */
|
||
|
|
input_buffer[strlen (input_buffer) - 1] = 0;
|
||
|
|
|
||
|
|
/* sanity check */
|
||
|
|
/* was thinking about running qstat without any options, capturing the
|
||
|
|
-default line, parsing it & making an array of all know server types
|
||
|
|
but thought this would be too much hassle considering this is a tool
|
||
|
|
for intelligent sysadmins (ha). Could put a static array of known
|
||
|
|
server types in a header file but then we'd be limiting ourselves
|
||
|
|
|
||
|
|
In the end, I figured I'd simply let an error occur & then trap it
|
||
|
|
*/
|
||
|
|
|
||
|
|
if (!strncmp (input_buffer, "unknown option", 14)) {
|
||
|
|
printf (_("CRITICAL - Host type parameter incorrect!\n"));
|
||
|
|
result = STATE_CRITICAL;
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* initialize the returned data buffer */
|
||
|
|
for (i = 0; i < QSTAT_MAX_RETURN_ARGS; i++)
|
||
|
|
ret[i] = strdup("");
|
||
|
|
|
||
|
|
i = 0;
|
||
|
|
p = (char *) strtok (input_buffer, QSTAT_DATA_DELIMITER);
|
||
|
|
while (p != NULL) {
|
||
|
|
ret[i] = p;
|
||
|
|
p = (char *) strtok (NULL, QSTAT_DATA_DELIMITER);
|
||
|
|
i++;
|
||
|
|
if (i >= QSTAT_MAX_RETURN_ARGS)
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (strstr (ret[2], QSTAT_HOST_ERROR)) {
|
||
|
|
printf (_("CRITICAL - Host not found\n"));
|
||
|
|
result = STATE_CRITICAL;
|
||
|
|
}
|
||
|
|
else if (strstr (ret[2], QSTAT_HOST_DOWN)) {
|
||
|
|
printf (_("CRITICAL - Game server down or unavailable\n"));
|
||
|
|
result = STATE_CRITICAL;
|
||
|
|
}
|
||
|
|
else if (strstr (ret[2], QSTAT_HOST_TIMEOUT)) {
|
||
|
|
printf (_("CRITICAL - Game server timeout\n"));
|
||
|
|
result = STATE_CRITICAL;
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
printf ("OK: %s/%s %s (%s), Ping: %s ms|%s %s\n",
|
||
|
|
ret[qstat_game_players],
|
||
|
|
ret[qstat_game_players_max],
|
||
|
|
ret[qstat_game_field],
|
||
|
|
ret[qstat_map_field],
|
||
|
|
ret[qstat_ping_field],
|
||
|
|
perfdata ("players", atol(ret[qstat_game_players]), "",
|
||
|
|
FALSE, 0, FALSE, 0,
|
||
|
|
TRUE, 0, TRUE, atol(ret[qstat_game_players_max])),
|
||
|
|
fperfdata ("ping", strtod(ret[qstat_ping_field], NULL), "",
|
||
|
|
FALSE, 0, FALSE, 0,
|
||
|
|
TRUE, 0, FALSE, 0));
|
||
|
|
}
|
||
|
|
|
||
|
|
/* close the pipe */
|
||
|
|
spclose (fp);
|
||
|
|
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
int
|
||
|
|
process_arguments (int argc, char **argv)
|
||
|
|
{
|
||
|
|
int c;
|
||
|
|
|
||
|
|
int opt_index = 0;
|
||
|
|
static struct option long_opts[] = {
|
||
|
|
{"help", no_argument, 0, 'h'},
|
||
|
|
{"version", no_argument, 0, 'V'},
|
||
|
|
{"verbose", no_argument, 0, 'v'},
|
||
|
|
{"timeout", required_argument, 0, 't'},
|
||
|
|
{"hostname", required_argument, 0, 'H'},
|
||
|
|
{"port", required_argument, 0, 'P'},
|
||
|
|
{"game-type", required_argument, 0, 'G'},
|
||
|
|
{"map-field", required_argument, 0, 'm'},
|
||
|
|
{"ping-field", required_argument, 0, 'p'},
|
||
|
|
{"game-field", required_argument, 0, 'g'},
|
||
|
|
{"players-field", required_argument, 0, 129},
|
||
|
|
{"max-players-field", required_argument, 0, 130},
|
||
|
|
{0, 0, 0, 0}
|
||
|
|
};
|
||
|
|
|
||
|
|
if (argc < 2)
|
||
|
|
return ERROR;
|
||
|
|
|
||
|
|
for (c = 1; c < argc; c++) {
|
||
|
|
if (strcmp ("-mf", argv[c]) == 0)
|
||
|
|
strcpy (argv[c], "-m");
|
||
|
|
else if (strcmp ("-pf", argv[c]) == 0)
|
||
|
|
strcpy (argv[c], "-p");
|
||
|
|
else if (strcmp ("-gf", argv[c]) == 0)
|
||
|
|
strcpy (argv[c], "-g");
|
||
|
|
}
|
||
|
|
|
||
|
|
while (1) {
|
||
|
|
c = getopt_long (argc, argv, "hVvt:H:P:G:g:p:m:", long_opts, &opt_index);
|
||
|
|
|
||
|
|
if (c == -1 || c == EOF)
|
||
|
|
break;
|
||
|
|
|
||
|
|
switch (c) {
|
||
|
|
case '?': /* args not parsable */
|
||
|
|
usage2 (_("Unknown argument"), optarg);
|
||
|
|
case 'h': /* help */
|
||
|
|
print_help ();
|
||
|
|
exit (STATE_OK);
|
||
|
|
case 'V': /* version */
|
||
|
|
print_revision (progname, revision);
|
||
|
|
exit (STATE_OK);
|
||
|
|
case 'v': /* version */
|
||
|
|
verbose = TRUE;
|
||
|
|
break;
|
||
|
|
case 't': /* timeout period */
|
||
|
|
timeout_interval = atoi (optarg);
|
||
|
|
break;
|
||
|
|
case 'H': /* hostname */
|
||
|
|
if (strlen (optarg) >= MAX_HOST_ADDRESS_LENGTH)
|
||
|
|
die (STATE_UNKNOWN, _("Input buffer overflow\n"));
|
||
|
|
server_ip = optarg;
|
||
|
|
break;
|
||
|
|
case 'P': /* port */
|
||
|
|
port = atoi (optarg);
|
||
|
|
break;
|
||
|
|
case 'G': /* hostname */
|
||
|
|
if (strlen (optarg) >= MAX_INPUT_BUFFER)
|
||
|
|
die (STATE_UNKNOWN, _("Input buffer overflow\n"));
|
||
|
|
game_type = optarg;
|
||
|
|
break;
|
||
|
|
case 'p': /* index of ping field */
|
||
|
|
qstat_ping_field = atoi (optarg);
|
||
|
|
if (qstat_ping_field < 0 || qstat_ping_field > QSTAT_MAX_RETURN_ARGS)
|
||
|
|
return ERROR;
|
||
|
|
break;
|
||
|
|
case 'm': /* index on map field */
|
||
|
|
qstat_map_field = atoi (optarg);
|
||
|
|
if (qstat_map_field < 0 || qstat_map_field > QSTAT_MAX_RETURN_ARGS)
|
||
|
|
return ERROR;
|
||
|
|
break;
|
||
|
|
case 'g': /* index of game field */
|
||
|
|
qstat_game_field = atoi (optarg);
|
||
|
|
if (qstat_game_field < 0 || qstat_game_field > QSTAT_MAX_RETURN_ARGS)
|
||
|
|
return ERROR;
|
||
|
|
break;
|
||
|
|
case 129: /* index of player count field */
|
||
|
|
qstat_game_players = atoi (optarg);
|
||
|
|
if (qstat_game_players_max == 0)
|
||
|
|
qstat_game_players_max = qstat_game_players - 1;
|
||
|
|
if (qstat_game_players < 0 || qstat_game_players > QSTAT_MAX_RETURN_ARGS)
|
||
|
|
return ERROR;
|
||
|
|
break;
|
||
|
|
case 130: /* index of max players field */
|
||
|
|
qstat_game_players_max = atoi (optarg);
|
||
|
|
if (qstat_game_players_max < 0 || qstat_game_players_max > QSTAT_MAX_RETURN_ARGS)
|
||
|
|
return ERROR;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
c = optind;
|
||
|
|
/* first option is the game type */
|
||
|
|
if (!game_type && c<argc)
|
||
|
|
game_type = strdup (argv[c++]);
|
||
|
|
|
||
|
|
/* Second option is the server name */
|
||
|
|
if (!server_ip && c<argc)
|
||
|
|
server_ip = strdup (argv[c++]);
|
||
|
|
|
||
|
|
return validate_arguments ();
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
int
|
||
|
|
validate_arguments (void)
|
||
|
|
{
|
||
|
|
if (qstat_game_players_max < 0)
|
||
|
|
qstat_game_players_max = 4;
|
||
|
|
|
||
|
|
if (qstat_game_players < 0)
|
||
|
|
qstat_game_players = 5;
|
||
|
|
|
||
|
|
if (qstat_game_field < 0)
|
||
|
|
qstat_game_field = 2;
|
||
|
|
|
||
|
|
if (qstat_map_field < 0)
|
||
|
|
qstat_map_field = 3;
|
||
|
|
|
||
|
|
if (qstat_ping_field < 0)
|
||
|
|
qstat_ping_field = 5;
|
||
|
|
|
||
|
|
return OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
void
|
||
|
|
print_help (void)
|
||
|
|
{
|
||
|
|
print_revision (progname, revision);
|
||
|
|
|
||
|
|
printf ("Copyright (c) 1999 Ian Cass, Knowledge Matters Limited\n");
|
||
|
|
printf (COPYRIGHT, copyright, email);
|
||
|
|
|
||
|
|
printf (_("This plugin tests game server connections with the specified host."));
|
||
|
|
|
||
|
|
print_usage ();
|
||
|
|
|
||
|
|
printf (_(UT_HELP_VRSN));
|
||
|
|
|
||
|
|
printf (_("\
|
||
|
|
<game> = Game type that is recognised by qstat (without the leading dash)\n\
|
||
|
|
<ip_address> = The IP address of the device you wish to query\n\
|
||
|
|
[port] = Optional port of which to connect\n\
|
||
|
|
[game_field] = Field number in raw qstat output that contains game name\n\
|
||
|
|
[map_field] = Field number in raw qstat output that contains map name\n\
|
||
|
|
[ping_field] = Field number in raw qstat output that contains ping time\n"));
|
||
|
|
|
||
|
|
printf (_(UT_TIMEOUT), DEFAULT_SOCKET_TIMEOUT);
|
||
|
|
|
||
|
|
printf (_("\n\
|
||
|
|
Notes:\n\
|
||
|
|
- This plugin uses the 'qstat' command, the popular game server status query tool .\n\
|
||
|
|
If you don't have the package installed, you will need to download it from\n\
|
||
|
|
http://www.activesw.com/people/steve/qstat.html before you can use this plugin.\n"));
|
||
|
|
|
||
|
|
printf (_(UT_SUPPORT));
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
void
|
||
|
|
print_usage (void)
|
||
|
|
{
|
||
|
|
printf ("\
|
||
|
|
Usage: %s <game> <ip_address> [-p port] [-gf game_field] [-mf map_field]\n\
|
||
|
|
[-pf ping_field]\n", progname);
|
||
|
|
}
|
||
|
|
|
||
|
|
/******************************************************************************
|
||
|
|
*
|
||
|
|
* Test Cases:
|
||
|
|
*
|
||
|
|
* ./check_game --players 7 -p 8 --map 5 qs 67.20.190.61 26000
|
||
|
|
*
|
||
|
|
* qstat -raw , -qs 67.20.190.61
|
||
|
|
* ==> QS,67.20.190.61,Nightmare.fintek.ca,67.20.190.61:26000,3,e2m1,6,0,83,0
|
||
|
|
*
|
||
|
|
* qstat -qs 67.20.190.61
|
||
|
|
* ==> ADDRESS PLAYERS MAP RESPONSE TIME NAME
|
||
|
|
* ==> 67.20.190.61 0/ 6 e2m1 79 / 0 Nightmare.fintek.ca
|
||
|
|
*
|
||
|
|
******************************************************************************/
|