Plugin functional

Imanol-Mikel Barba Sabariego
1 parent f2154f92
Showing 5 changed files with 129 additions and 16 deletions
CMakeLists.txt
check_csgo/check_csgo.cpp
check_csgo/check_csgo.h
check_csgo/udp.cpp
check_csgo/udp.h
@@ -3,7 +3,7 @@ project(nagios_plugins)
 set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11  -U__STRICT_ANSI__")
-set(SOURCE_FILES_CSGO check_csgo/check_csgo.cpp check_memfree/auxiliar.cpp)
+set(SOURCE_FILES_CSGO check_csgo/check_csgo.cpp check_csgo/auxiliar.cpp check_csgo/udp.cpp)
 add_executable(check_csgo ${SOURCE_FILES_CSGO})
 set(SOURCE_FILES_MEMFREE check_memfree/check_memfree.cpp  check_memfree/range.cpp check_memfree/auxiliar.cpp)
@@ -14,7 +14,7 @@ void printHelp(bool longVersion)
 	if(longVersion)
 	{
 		printVersion();
-		cout << "Check Source DS instance." << endl << endl;
+		cout << "Check CS:GO DS instance." << endl << endl;
 		printHelp(false);
 		cout << "Options:" << endl;
 		cout << " -h" << endl;
@@ -30,8 +30,59 @@ void printHelp(bool longVersion)
 	cout << "Usage: " << endl << "check_csgo [-hV] -H HOSTADDRESS [-p PORT]" << endl << endl;
 }
+SOURCEDS_PACKET* getSourceDSResponse(int s, sockaddr_in *server)
+{
+    char *data = new char[DATAGRAM_LENGTH];
+    int dataLength = recvMsg(s, data, DATAGRAM_LENGTH, server);
+    if(dataLength == -1)
+    {
+        return NULL;
+    }
+    char *header = (char*) "\xFF\xFF\xFF\xFF";
+    if(memcmp(data,header,HDR_SIZE))
+    {
+       //Invalid response, or non-supported multi-packet response format
+       return NULL;
+    }
+    SOURCEDS_PACKET *response = new SOURCEDS_PACKET;
+    response->data = data+HDR_SIZE;
+    response->length = dataLength;
+    return response;
+}
+
 int check_csgo(char *hostname, uint16_t port, SERVERINFO *server_info)
 {
+    struct sockaddr_in si;
+    struct hostent *host = NULL;
+    int s;
+    int timeout = 10;
+    setupSocket(port,hostname,host,timeout,&si,&s);
+    string discard;
+
+    char *info_query = (char*) "\xFF\xFF\xFF\xFF\x54Source Engine Query\x00";
+
+    sendMsg(s,info_query, strlen(info_query)+1,&si);
+    SOURCEDS_PACKET *response = getSourceDSResponse(s,&si);
+    if(response == NULL)
+    {
+        return 2;
+    }
+
+    ssize_t offset = 2;
+    server_info->name = string(response->data+offset,strlen(response->data+offset));
+    offset += server_info->name.size()+1;
+    server_info->map = string(response->data+offset,strlen(response->data+offset));
+    offset += server_info->map.size()+1;
+    discard = string(response->data+offset,strlen(response->data+offset));
+    offset += discard.size() + 1;
+    server_info->game = string(response->data+offset,strlen(response->data+offset));
+    offset += server_info->game.size()+1;
+    offset += 2;
+    server_info->players = (uint8_t) *((response->data)+offset);
+
+    delete[] (response->data-HDR_SIZE);
+    delete response;
+
     return 0;
 }
@@ -61,7 +112,7 @@ int main(int argc, char **argv)
         			hostname = optarg;
         			break;
 			case 'p':
-       		 		port = str2int(optarg);
+       		 		port = (uint16_t) str2int(optarg);
        		 		break;
 			case 'V':
 				printVersion();
@@ -89,17 +140,14 @@ int main(int argc, char **argv)
 	{
 		case 0:
 			cout << " OK";
-			break;
-
-		case 1:
-			cout << " WARNING";
+            cout << " - " << hostname << " " << server_info.name << " " << server_info.game << " " << server_info.map << " " << server_info.players << endl;
 			break;
 		case 2:
-			cout << " CRITICAL";
+			cout << " CRITICAL - No response";
 			break;
 	}
-	cout << " - " << server_info.name << " " << server_info.ip << " " << server_info.name << " " << server_info.map << " " << server_info.players << " " << server_info.latency << endl;
+
 	return returnCode;
 }
@@ -14,21 +14,27 @@
 #include <sys/time.h>
 #include "auxiliar.h"
+#include "udp.h"
 #define VERSION "1.0"
+#define DATAGRAM_LENGTH 1400
+#define HDR_SIZE 4
+struct sourcedspacket_struct {
+    ssize_t length;
+    char* data;
+};
+typedef struct sourcedspacket_struct SOURCEDS_PACKET;
 struct serverinfo_struct {
-    char *name;
-    char *ip;
-    char *game;
-    char *map;
-    unsigned int players;
-    unsigned int latency;
+    string name;
+    string game;
+    string map;
+    uint8_t players;
 };
-
 typedef struct serverinfo_struct SERVERINFO;
+SOURCEDS_PACKET* getSourceDSResponse(int s, sockaddr_in *server);
 int check_csgo(char *hostname, uint16_t port, SERVERINFO *server_info);
 void printVersion();
 void printHelp(bool longVersion);
+#include "udp.h"
+
+void setupSocket(uint16_t port, char *hostname, struct hostent *host, int timeout, struct sockaddr_in *si, int *s)
+{
+	if((*s = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) == -1)
+	{
+		cout << "Couldn't create socket" << endl;
+		exit(3);
+	}
+	struct timeval tv;
+	tv.tv_sec = timeout;
+	tv.tv_usec = 0;
+	if(setsockopt(*s, SOL_SOCKET, SO_RCVTIMEO,&tv,sizeof(tv)) < 0) 
+	{
+		cout << "Error setting socket timeout" << endl;
+		exit(3);
+	}
+	if(!(host = gethostbyname(hostname)))
+	{
+		cout << "Client could not get host address information" << endl;
+		exit(3);
+	}
+
+	memset((char *) si, 0, sizeof(*si));
+	si->sin_family = AF_INET;
+	memcpy (&(si->sin_addr), host->h_addr, host->h_length);
+	si->sin_port = htons(port);
+}
+
+int sendMsg(int s, char *msg, size_t msgLength, struct sockaddr_in *si)
+{
+	return sendto(s, msg, msgLength, 0,(struct sockaddr*) si, sizeof(*si));
+}
+
+int recvMsg(int s, char *msg, size_t msgLength, struct sockaddr_in *si)
+{
+	size_t slen = sizeof(*si);
+	return recvfrom(s, msg, msgLength, 0, (struct sockaddr *) si, (socklen_t*)&slen);
+}
+#ifndef UDP_H
+#define UDP_H
+
+#include <iostream>
+
+#include <string.h>
+#include <cstdlib>
+
+#include <arpa/inet.h>
+#include <sys/socket.h>
+#include <netdb.h>
+#include <netinet/in.h>
+
+using namespace std;
+
+void setupSocket(uint16_t port, char *hostname, struct hostent *host, int timeout, struct sockaddr_in *si, int *s);
+int sendMsg(int s, char *msg, size_t msgLength, struct sockaddr_in *si);
+int recvMsg(int s, char *msg, size_t msgLength, struct sockaddr_in *si);
+
+#endif