Added check_openvpn

Imanol-Mikel Barba Sabariego
1 parent 94ddb039
Showing 8 changed files with 436 additions and 0 deletions
CMakeLists.txt
check_openvpn/README.md
check_openvpn/auxiliar.cpp
check_openvpn/auxiliar.h
check_openvpn/check_openvpn.cpp
check_openvpn/check_openvpn.h
check_openvpn/tcp.cpp
check_openvpn/tcp.h
@@ -21,3 +21,6 @@ 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})
  
+set(SOURCE_FILES_OPENVPN check_openvpn/check_openvpn.cpp  check_openvpn/tcp.cpp check_openvpn/auxiliar.cpp)
+add_executable(check_openvpn ${SOURCE_FILES_OPENVPN})
+
+```
+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;
+}
+
+void split(const string &s, const char* delim, vector<string> & v)
+{
+    char *dup = strdup(s.c_str());
+    char *token = strtok(dup, delim);
+    while(token != NULL)
+    {
+        v.push_back(string(token));
+        token = strtok(NULL, delim);
+    }
+    free(dup);
+}
+
+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;
+}
+//
+// Created by Imanol on 28-may-16.
+//
+
+#ifndef NAGIOS_PLUGINS_AUXILIAR_H
+#define NAGIOS_PLUGINS_AUXILIAR_H
+
+#include <sstream>
+#include <iostream>
+#include <exception>
+#include <vector>
+
+#include <string.h>
+#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);
+void split(const string &s, const char* delim, vector<string> & v);
+
+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_openvpn.h"
+
+using namespace std;
+
+char *servicename = (char*)"OpenVPN";
+
+void printVersion()
+{
+	cout << "check_openvpn v" << VERSION << endl << endl;
+}
+
+void printHelp(bool longVersion)
+{
+	if(longVersion)
+	{
+		printVersion();
+		cout << "Check OpenVPN server" << 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 UPnP server is running" << endl;
+                cout << " -p PORT" << endl;
+                cout << "    OpenVPN Administration port" << endl;
+                cout << " -P PASSWORD" << endl;
+                cout << "    OpenVPN Administration password" << endl;
+		return;
+	}
+	cout << "Usage: " << endl << "check_openvpn [-hV] -H HOSTADDRESS -p PORT -P PASSWORD" << endl << endl;
+}
+
+int check_openvpn(uint16_t port, char *hostname, char *password, string *serverinfo)
+{
+    size_t numClients = 0;
+    string ipList = "";
+    char buffer[MAX_TCP+1];
+    int s;
+    int timeout = 10;
+    int bytes = 0;
+    string output = "";
+
+    s = createSocket();
+    connect(s,port,hostname,timeout);
+    while(output != "ENTER PASSWORD:")
+    {
+        recvMsg(s,buffer,MAX_TCP);
+	output += buffer;
+        memset(buffer,0x00,MAX_TCP+1);
+    }
+    sendMsg(s,strcat(password,"\n"),strlen(password)+1);
+    output = "";
+    do
+    {
+        bytes = recvMsg(s,buffer,MAX_TCP);
+        output += buffer;
+        memset(buffer,0x00,MAX_TCP+1);
+    }while(bytes);
+    if(output == "ENTER PASSWORD:")
+    {
+        *serverinfo = "Incorrect password";
+        close(s);
+        return 3;
+    }
+    output = "";
+    sendMsg(s,(char*)"status 2\n",9);
+    do
+    {
+        bytes = recvMsg(s,buffer,MAX_TCP);
+        output += buffer;
+        memset(buffer,0x00,MAX_TCP+1);
+    }while(output.rfind("END") != output.size()-5);
+    vector<string> splitstr;
+    vector<string> fields;
+    split(output,"\n",splitstr);
+    for(vector<string>::iterator it = splitstr.begin() ; it != splitstr.end(); ++it)
+    {
+        if(it->substr(0,11) == "CLIENT_LIST")
+        {
+            split(*it,",",fields);
+            ipList += fields[3] + ",";
+            numClients++;
+        }
+    }
+    if(numClients)
+    {
+        ipList.pop_back();
+    }
+    *serverinfo = int2str(numClients) + " clients connected (" + ipList + ")" ;
+    close(s);
+    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);
+
+	uint16_t port = 1195;
+	char *hostname = NULL;
+        char *password = NULL;
+	int c;
+
+	while ((c = getopt (argc, argv, "H:p:P:Vh")) != -1)
+	{
+		switch(c)
+		{
+			case 'H':
+        			hostname = optarg;
+        			break;
+			case 'p':
+       		 		port = (uint16_t) str2int(optarg);
+       		 		break;
+			case 'P':
+       		 		password = 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(password == NULL)
+        {
+            cout << "No PASSWORD specified. Exiting." << endl;
+            return 3;
+        }
+
+	string serverinfo = "";
+	int returnCode = check_openvpn(port,hostname,password,&serverinfo);
+
+	cout << servicename;
+	switch(returnCode)
+	{
+		case 0:
+			cout << " OK - " << serverinfo << endl ;
+			break;
+
+		case 2:
+			cout << " CRITICAL - No response" << endl;
+			break;
+
+        case 3:
+            cout << " UNKNOWN - " << serverinfo << endl;
+	}
+
+	return returnCode;
+}
+#ifndef CHECK_OPENVPN_H
+#define CHECK_OPENVPN_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 "tcp.h"
+
+#define VERSION "1.0"
+
+int check_openvpn(char *hostname, string *serverinfo);
+void printVersion();
+void printHelp(bool longVersion);
+
+#endif
+#include "tcp.h"
+
+int createSocket()
+{
+    int sockfd;
+    if((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
+    {
+        cout << "Could not create socket" << endl;
+        exit(3);
+    }
+    return sockfd;
+}
+
+void connect(int s, uint16_t port, char *hostname, int timeout)
+{
+    struct in_addr *addr_ptr;
+    struct hostent *hostPtr;
+    string add;
+    hostPtr = gethostbyname(hostname);
+    if(hostPtr == NULL)
+    {
+        cout << "Could not resolve hostname" << endl;
+        exit(3);
+    }
+    addr_ptr = (struct in_addr *)*hostPtr->h_addr_list;
+    add = inet_ntoa(*addr_ptr);
+    if(add == "")
+    {
+        cout << "Invalid address" << endl;
+        exit(3);
+    }
+    struct sockaddr_in newSockAddr;
+    newSockAddr.sin_family = AF_INET;
+    newSockAddr.sin_port = htons(port);
+    newSockAddr.sin_addr.s_addr = inet_addr(add.c_str());
+    if(connect(s, (struct sockaddr *)&newSockAddr, sizeof(struct sockaddr)) != 0)
+    {
+        cout << "Could not connect to " << hostname << " on port " << port << 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);
+    }
+    fcntl(s, F_SETFL, O_NONBLOCK);
+}
+
+int sendMsg(int s, const char *msg, size_t msgLength)
+{
+    int bytes, total = 0;
+    while(total != msgLength)
+    {
+        bytes = send(s,msg+total,msgLength-total,0);
+        if(bytes == -1)
+        {
+            cout << "TCP: Could not write to socket." << endl;
+            exit(3);
+        }
+        total += bytes;
+    }
+    return total;
+}
+
+int recvMsg(int s, char *msg, size_t msgLength)
+{
+    int bytes, total = 0;
+    while(total != msgLength)
+    {
+        bytes = recv(s, msg+total, msgLength-total,0);
+        if ( bytes <= 0 )
+        {
+            if(errno == EWOULDBLOCK || errno == EAGAIN)
+            {
+                return 0;
+            }
+            cout << "TCP: Could not read from socket." << endl;
+            exit(3);
+        }
+        total += bytes;
+    }
+    return total;
+}
+#ifndef TCP_H
+#define TCP_H
+
+#define MAX_TCP 65535
+
+#include <iostream>
+
+#include <string.h>
+#include <cstdlib>
+
+#include <errno.h>
+#include <arpa/inet.h>
+#include <sys/socket.h>
+#include <netdb.h>
+#include <fcntl.h>
+using namespace std;
+
+int createSocket();
+void connect(int s, uint16_t port, char *hostname, int timeout);
+int sendMsg(int s, const char *msg, size_t msgLength);
+int recvMsg(int s, char *msg, size_t msgLength);
+
+#endif