Merge pull request #4 from Mosder/2-simple-capturing-mode

Simple capturing mode

7 files changed, 128 insertions, 0 deletions

diff --git a/analyzer/capture.c b/analyzer/capture.c
new file mode 100644
index 0000000..0a75e09
--- /dev/null
+++ b/analyzer/capture.c
@@ -0,0 +1,28 @@
+#include "capture.h"
+
+#include <stdio.h>
+#include <stdlib.h>
+
+#include "headers/interface.h"
+#include "headers/server.h"
+#include "utils/fs.h"
+#include "utils/time.h"
+
+void capture_to_file(char *path) {
+    char command[128];
+    sprintf(command, "tshark -i %s -f \"src host %s\" -w %s", INTERFACE, SERVER_IP, path);
+    system(command);
+}
+
+void capturing_mode(char *path) {
+    if (path != NULL) {
+        capture_to_file(path);
+        return;
+    }
+
+    char capture_dir[256], new_path[256], date_time[128];
+    get_capture_dir(capture_dir);
+    get_date_time_string(date_time);
+    sprintf(new_path, "%s/capture-%s.pcap", capture_dir, date_time);
+    capture_to_file(new_path);
+}
diff --git a/analyzer/capture.h b/analyzer/capture.h
new file mode 100644
index 0000000..648e5ad
--- /dev/null
+++ b/analyzer/capture.h
@@ -0,0 +1,8 @@
+// capture.h
+// functions for capturing packets using tshark
+
+// starts capturing mode
+// params:
+//      - path - file path to capture the .pcap file to,
+//               if NULL, saves it to default capture directory as capture-<date>_<time>.pcap
+extern void capturing_mode(char *path);
diff --git a/analyzer/headers/interface.h b/analyzer/headers/interface.h
new file mode 100644
index 0000000..8b407c4
--- /dev/null
+++ b/analyzer/headers/interface.h
@@ -0,0 +1 @@
+#define INTERFACE "enp8s0"
diff --git a/analyzer/utils/fs.c b/analyzer/utils/fs.c
new file mode 100644
index 0000000..76fb581
--- /dev/null
+++ b/analyzer/utils/fs.c
@@ -0,0 +1,43 @@
+#include "utils.h"
+
+#include <fcntl.h>
+#include <sys/stat.h>
+#include <unistd.h>
+
+void get_home_dir(char *home_dir) {
+    strcpy(home_dir, getenv("HOME"));
+}
+
+void get_main_dir(char *main_dir) {
+    char home_dir[256];
+    get_home_dir(home_dir);
+    sprintf(main_dir, "%s/.local/share/branal", home_dir);
+}
+
+void get_capture_dir(char *capture_dir) {
+    char main_dir[256];
+    get_main_dir(main_dir);
+    sprintf(capture_dir, "%s/captures", main_dir);
+}
+
+void get_saves_dir(char *saves_dir) {
+    char main_dir[256];
+    get_main_dir(main_dir);
+    sprintf(saves_dir, "%s/saves", main_dir);
+}
+
+void create_necessary_directories() {
+    char main_dir[256], capture_dir[256], saves_dir[256];
+    get_main_dir(main_dir);
+    get_capture_dir(capture_dir);
+    get_saves_dir(saves_dir);
+    mkdir(main_dir, 0755);
+    mkdir(capture_dir, 0755);
+    mkdir(saves_dir, 0755);
+}
+
+void write_file(char *path, void *contents, int contents_len, int flags, mode_t permissions) {
+    int fd = open(path, flags, permissions);
+    write(fd, contents, contents_len);
+    close(fd);
+}
diff --git a/analyzer/utils/fs.h b/analyzer/utils/fs.h
new file mode 100644
index 0000000..6335799
--- /dev/null
+++ b/analyzer/utils/fs.h
@@ -0,0 +1,31 @@
+// utils/fs.h
+// functions for operating on the file system, i.e. creating directories, writing files, etc.
+
+#include <sys/types.h>
+
+// get main directory used by the program
+// params:
+//      - main_dir - buffer to save the directory to
+extern void get_main_dir(char *main_dir);
+
+// get directory to save the .pcap files to
+// params:
+//      - capture_dir - buffer to save the directory to
+extern void get_capture_dir(char *capture_dir);
+
+// get directory of save data of the program
+// params:
+//      - saves_dir - buffer to save the directory to
+extern void get_saves_dir(char *saves_dir);
+
+// create directories that will be used by the program
+extern void create_necessary_directories();
+
+// writes file to the specified path
+// params:
+//      - path - path to write the file to
+//      - contents - contents of the file
+//      - contents_len - amount of bytes of contents to write
+//      - flags - flags to open the file with
+//      - permissions - permissions of the written file
+extern void write_file(char *path, void *contents, int contents_len, int flags, mode_t permissions);
diff --git a/analyzer/utils/time.c b/analyzer/utils/time.c
new file mode 100644
index 0000000..72c0bc7
--- /dev/null
+++ b/analyzer/utils/time.c
@@ -0,0 +1,10 @@
+#include "time.h"
+
+#include <stdio.h>
+#include <time.h>
+
+void get_date_time_string(char *date_time) {
+    time_t now = time(NULL);
+    struct tm *t = localtime(&now);
+    sprintf(date_time, "%02d%02d%02d_%02d%02d%02d", t->tm_year - 100, t->tm_mon + 1, t->tm_mday, t->tm_hour, t->tm_min, t->tm_sec);
+}
diff --git a/analyzer/utils/time.h b/analyzer/utils/time.h
new file mode 100644
index 0000000..0bd0df1
--- /dev/null
+++ b/analyzer/utils/time.h
@@ -0,0 +1,7 @@
+// utils/time.h
+// functions regarding date and time
+
+// gets a string <date>_<time> for current localtime
+// params:
+//      date_time - buffer to save the datetime string to
+extern void get_date_time_string(char *date_time);