Created plugin template for nfs4

Imanol-Mikel Barba Sabariego
1 parent 999ee4d0
Showing 8 changed files with 308 additions and 1 deletions
CMakeLists.txt
check_nfs4/README.md
check_nfs4/auxiliar.cpp
check_nfs4/auxiliar.h
check_nfs4/check_nfs4.cpp
check_nfs4/check_nfs4.h
check_nfs4/udp.cpp
check_nfs4/udp.h
@@ -6,6 +6,9 @@ set(CMAKE_CXX_FLAGS &quot;${CMAKE_CXX_FLAGS} -std=c++11  -U__STRICT_ANSI__&quot;)
 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_NFS4 check_nfs4/check_nfs4.cpp check_csgo/auxiliar.cpp check_csgo/udp.cpp)
+add_executable(check_nfs4 ${SOURCE_FILES_NFS4})
+
 set(SOURCE_FILES_MEMFREE check_memfree/check_memfree.cpp  check_memfree/range.cpp check_memfree/auxiliar.cpp)
 add_executable(check_memfree ${SOURCE_FILES_MEMFREE})
  
@@ -16,4 +19,5 @@ set(SOURCE_FILES_SMART check_smart/check_smart.cpp check_smart/auxiliar.cpp)
 add_executable(check_smart ${SOURCE_FILES_SMART})
  
 set(SOURCE_FILES_TFTP check_tftp/check_tftp.cpp  check_tftp/udp.cpp check_tftp/auxiliar.cpp)
-add_executable(check_tftp ${SOURCE_FILES_TFTP})
 \ No newline at end of file
+add_executable(check_tftp ${SOURCE_FILES_TFTP})
+
+```
+check_memfree v1.0
+
+Check free memory space on local machine.
+
+Usage: 
+check_memfree [-hV] -w <percent_free>% -c <percent_free>%
+check_memfree [-hV] -w <bytes_free> -c <bytes_free>
+
+Options:
+ -h
+    Print detailed help screen
+ -V
+    Print version information
+ -w INTEGER
+    Exit with WARNING status if less than INTEGER bytes of memory space are free
+ -w PERCENT%
+    Exit with WARNING status if less than PERCENT of memory space is free
+ -c INTEGER
+    Exit with CRITICAL status if less than INTEGER bytes of memory space are free
+ -c PERCENT%
+    Exit with CRITCAL status if less than PERCENT of memory space is free
+```
+//
+// Created by Imanol on 28-may-16.
+//
+
+#include "auxiliar.h"
+
+void timer_handler (int signum)
+{
+    if(signum == SIGVTALRM)
+    {
+        cout << servicename << " CRITICAL - timeout occurred" << endl;
+        exit(2);
+    }
+}
+
+int str2int(string str)
+{
+    int num;
+    stringstream sstream;
+    sstream << str;
+    if(!(sstream >> num))
+    {
+        throw integerConversionException("Integer conversion error");
+    }
+    return num;
+}
+
+string int2str(int x)
+{
+    string str;
+    stringstream sstream;
+    sstream << x;
+    sstream >> str;
+    return str;
+}
+
+int exec(string cmd, string *output)
+{
+    *output = "";
+    FILE* pipe = popen(cmd.c_str(), "r");
+    if (!pipe)
+    {
+        cout << "Error opening child process" << endl;
+        exit(3);
+    }
+    char buffer[128];
+    while(!feof(pipe))
+    {
+        if(fgets(buffer, 128, pipe) != NULL)
+        {
+            *output += buffer;
+        }
+    }
+    return pclose(pipe)/256;
+}
 \ No newline at end of file
+//
+// Created by Imanol on 28-may-16.
+//
+
+#ifndef NAGIOS_PLUGINS_AUXILIAR_H
+#define NAGIOS_PLUGINS_AUXILIAR_H
+
+#include <sstream>
+#include <iostream>
+#include <exception>
+
+#include <stdlib.h>
+#include <stdio.h>
+#include <signal.h>
+
+using namespace std;
+
+extern char *servicename;
+
+int str2int(string str);
+string int2str(int x);
+int exec(string cmd, string *output);
+void timer_handler (int signum);
+
+class integerConversionException : public exception
+{
+private:
+    string s;
+public:
+    integerConversionException(std::string ss) : s(ss) {}
+    ~integerConversionException() throw () {}
+    const char* what() const throw() { return s.c_str(); }
+};
+
+#endif //NAGIOS_PLUGINS_AUXILIAR_H
+#include "check_nfs4.h"
+
+using namespace std;
+
+char *servicename = (char*)"NFSv4";
+
+void printVersion()
+{
+	cout << "check_csgo v" << VERSION << endl << endl;
+}
+
+void printHelp(bool longVersion)
+{
+	if(longVersion)
+	{
+		printVersion();
+		cout << "Check CS:GO DS instance." << endl << endl;
+		printHelp(false);
+		cout << "Options:" << endl;
+		cout << " -h" << endl;
+		cout << "    Print detailed help screen" << endl;
+		cout << " -V" << endl;
+		cout << "    Print version information" << endl;
+		cout << " -H HOSTADDRESS" << endl;
+		cout << "    Host where the Source DS is running" << endl;
+		cout << " -p" << endl;
+		cout << "    Port where the Source DS is listening. Default is 27015." << endl << endl;
+		return;
+	}
+	cout << "Usage: " << endl << "check_csgo [-hV] -H HOSTADDRESS [-p PORT]" << endl << endl;
+}
+
+int check_nfs4(char *hostname, char *mountpoint, string *detail)
+{
+	return 0;
+}
+
+int main(int argc, char **argv)
+{
+	struct itimerval timer;
+	timer.it_value.tv_sec = 10;
+	timer.it_value.tv_usec = 0;
+	timer.it_interval.tv_sec = 0;
+	timer.it_interval.tv_usec = 0;
+	setitimer (ITIMER_VIRTUAL, &timer, 0);
+
+	struct sigaction sa;
+	memset (&sa, 0, sizeof (sa));
+	sa.sa_handler = &timer_handler;
+	sigaction (SIGVTALRM, &sa, 0);
+
+	char *hostname = NULL;
+	char *mountpoint = NULL;
+	int c;
+
+	while ((c = getopt (argc, argv, "H:m:Vh")) != -1)
+	{
+		switch(c)
+		{
+			case 'H':
+        			hostname = optarg;
+        			break;
+			case 'm':
+       		 		mountpoint = optarg;
+       		 		break;
+			case 'V':
+				printVersion();
+				return 0;
+			case 'h':
+				printHelp(true);
+				return 0;
+			case '?':
+				printHelp(false);
+				return 3;
+		}
+	}
+
+	if(hostname == NULL)
+	{
+		cout << "No HOSTADDRESS specified. Exiting." << endl;
+		return 3;
+	}
+	if(mountpoint == NULL)
+	{
+		cout << "No mountpoint specified. Exiting." << endl;
+		return 3;
+	}
+
+	string details = "";
+	int returnCode = check_nfs4(hostname,mountpoint,&details);
+
+	cout << servicename;
+	switch(returnCode)
+	{
+		case 0:
+			cout << " OK";
+            cout << " - " << mountpoint << " " << details << endl;
+			break;
+
+        case 2:
+			cout << " CRITICAL - " << details << endl;
+			break;
+	}
+
+	return returnCode;
+}
+#ifndef CHECK_CSGO_H
+#define CHECK_CSGO_H
+
+#include <iostream>
+#include <sstream>
+#include <fstream>
+#include <limits>
+
+#include <ctype.h>
+#include <cstdlib>
+#include <unistd.h>
+#include <cstring>
+
+#include <sys/time.h>
+
+#include "auxiliar.h"
+#include "udp.h"
+
+#define VERSION "1.0"
+
+int check_nfs4(char *hostname, char *mountpoint, string *detail);
+void printVersion();
+void printHelp(bool longVersion);
+
+#endif
+#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