add portServices to database

read services from nmap-services-tcp.csv
add portService to ip_ports table
store portService for each used port

code_boost/src/cxx/statistics_db.cpp

@@ -2,6 +2,9 @@
 #include <math.h>
 #include <iostream>
 #include <sstream>
+#include <fstream>
+#include <unistd.h>
+#include <stdio.h>
 
 /**
  * Creates a new statistics_db object. Opens an existing database located at database_path. If not existing, creates
@@ -15,6 +18,9 @@ statistics_db::statistics_db(std::string database_path) {
     }
     // creates the DB if not existing, opens the DB for read+write access
     db.reset(new SQLite::Database(database_path, SQLite::OPEN_CREATE | SQLite::OPEN_READWRITE));
+
+    // Read ports and services into portServices vector
+    readPortServicesFromNmap();
 }
 
 /**
@@ -223,16 +229,21 @@ void statistics_db::writeStatisticsPorts(std::unordered_map<ipAddress_inOut_port
                 "portNumber INTEGER,"
                 "portCount INTEGER,"
                 "byteCount REAL,"
+                "portService TEXT COLLATE NOCASE,"
                 "PRIMARY KEY(ipAddress,portDirection,portNumber));";
         db->exec(createTable);
-        SQLite::Statement query(*db, "INSERT INTO ip_ports VALUES (?, ?, ?, ?, ?)");
+        SQLite::Statement query(*db, "INSERT INTO ip_ports VALUES (?, ?, ?, ?, ?, ?)");
         for (auto it = portsStatistics.begin(); it != portsStatistics.end(); ++it) {
             ipAddress_inOut_port e = it->first;
+
+            std::string portService = getPortService(e.portNumber);
+
             query.bind(1, e.ipAddress);
             query.bind(2, e.trafficDirection);
             query.bind(3, e.portNumber);
             query.bind(4, it->second.count);
             query.bind(5, it->second.byteCount);
+            query.bind(6, portService);
             query.exec();
             query.reset();
         }
@@ -453,3 +464,77 @@ void statistics_db::writeDbVersion(){
         std::cout << "Exception in statistics_db: " << e.what() << std::endl;
     }
 }
+
+/**
+ * Reads all ports and their corresponding services from nmap-services-tcp.csv and stores them into portServices vector.
+ */
+void statistics_db::readPortServicesFromNmap()
+{
+    std::string portnumber;
+	std::string service;
+	std::string dump;
+	std::ifstream reader;
+
+	reader.open(getNmapPath(), std::ios::in);
+
+    if(reader.is_open())
+    {
+        getline(reader, dump);
+
+        while(!reader.eof())
+        {
+            getline(reader, portnumber, ',');
+            getline(reader, service, ',');
+            getline(reader, dump);
+            if(!service.empty() && !portnumber.empty())
+            {
+                port_service ps = {std::stoi(portnumber), service};
+                portServices.push_back(ps);
+            }
+        }
+
+        reader.close();
+	}
+
+	else
+	{
+	    port_service ps = {0, "unknown"};
+	    portServices.push_back(ps);
+	}
+}
+
+/**
+ * Gets the service of a given port from portServices.
+ * @param port The port of which the service should be returned.
+ */
+std::string statistics_db::getPortService(int port)
+{
+    for(unsigned int x = 0; x < portServices.size(); x++)
+    {
+        port_service ps = portServices[x];
+        if(ps.p == port) {return ps.s;}
+    }
+
+    return "unknown";
+}
+
+/**
+ * Gets the path to nmap-services-tcp.csv and makes sure the file is reached from any working directory within "/code"
+ * because the working directory can be different when running tests. Checks if the file/path exists and warns the user.
+ */
+std::string statistics_db::getNmapPath()
+{
+    char buff[FILENAME_MAX];
+	std::string dir(getcwd(buff, FILENAME_MAX));
+	dir = dir.substr(0, dir.rfind("/code")) + "/resources/nmap-services-tcp.csv";
+
+	std::ifstream reader;
+	reader.open(dir, std::ios::in);
+	if(!reader.is_open())
+	{
+	    std::cerr << "WARNING: " << dir << " could not be opened! PortServices can't be read!" << std::endl;
+	}
+
+	else {reader.close();}
+	return dir;
+}

code_boost/src/cxx/statistics_db.h

@@ -10,6 +10,10 @@
 #include <string>
 #include "statistics.h"
 #include <SQLiteCpp/SQLiteCpp.h>
+#include <vector>
+
+// Struct which contains a port and it's corresponding service
+struct port_service {int p; std::string s;};
 
 class statistics_db {
 public:
@@ -21,7 +25,7 @@ public:
     /*
      * Database version: Increment number on every change in the C++ code!
      */
-    static const int DB_VERSION = 2;
+    static const int DB_VERSION = 3;
 
     /*
      * Methods for writing values into database
@@ -52,11 +56,18 @@ public:
 
     void writeDbVersion();
 
+    std::string getPortService(int port);
+
+    void readPortServicesFromNmap();
+
+    std::string getNmapPath();
+
 private:
     // Pointer to the SQLite database
     std::unique_ptr<SQLite::Database> db;
 
-
+    // Vector which contains all ports and their corresponding services
+    std::vector<port_service> portServices;
 };