updating communication

src/ma_installer/ma_installer.cpp

@@ -7,6 +7,7 @@
 #include <sddl.h>
 #include <stdio.h>
 #include "download.h"
+#include "well_known.h"
 #pragma comment(lib, "advapi32.lib")
 /*
 Tasks to do:
@@ -197,6 +198,83 @@ int create_insecure_folder(LPCWSTR folderpath) {
     }
     return error;
 }
+
+int check_cert(const char* cert, const char* secrets_path) {
+    FILE* fp;
+    if (fopen_s(&fp, secrets_path, "r") != 0) {
+        return 1;
+    }
+    else {
+        char* secrets = new char[300];
+        while (!feof(fp)) {
+            fscanf_s(fp, "%s", secrets, 295); // get the secret
+            if (strcmp("cert", secrets) == 0) {
+                fscanf_s(fp, "%s", secrets, 295); // get the secret
+                if (strcmp(cert, secrets) == 0) {
+                    delete[] secrets;
+                    return 0;
+                }
+            }
+        }
+        delete[] secrets;
+        return 2;
+    }
+
+}
+char* get_apikey(const char* secrets_path) {
+    FILE* fp;
+    if (fopen_s(&fp, secrets_path, "r") != 0) {
+        return 0;
+    }
+    else {
+        char* secrets = new char[300];
+        while (!feof(fp)) {
+            fscanf_s(fp, "%s", secrets, 295); // get the secret
+            if (strcmp("apikey", secrets) == 0) {
+                fscanf_s(fp, "%s", secrets, 295); // get the secret
+                return secrets;
+            }
+        }
+        delete[] secrets;
+        return 0;
+    }
+}
+char* get_machineid(const char* secrets_path) {
+    FILE* fp;
+    if (fopen_s(&fp, secrets_path, "r") != 0) {
+        return 0;
+    }
+    else {
+        char* secrets = new char[300];
+        while (!feof(fp)) {
+            fscanf_s(fp, "%s", secrets, 295); // get the secret
+            if (strcmp("machineid", secrets) == 0) {
+                fscanf_s(fp, "%s", secrets, 295); // get the secret
+                return secrets;
+            }
+        }
+        delete[] secrets;
+        return 0;
+    }
+}
+char* get_server(const char* secrets_path) {
+    FILE* fp;
+    if (fopen_s(&fp, secrets_path, "r") != 0) {
+        return 0;
+    }
+    else {
+        char* secrets = new char[300];
+        while (!feof(fp)) {
+            fscanf_s(fp, "%s", secrets, 295); // get the secret
+            if (strcmp("server", secrets) == 0) {
+                fscanf_s(fp, "%s", secrets, 295); // get the secret
+                return secrets;
+            }
+        }
+        delete[] secrets;
+        return 0;
+    }
+}
 int main()
 {
     printf("Welcome to the Cyberhex installer!\n");
@@ -251,6 +329,16 @@ int main()
         }
 
         //download files from server
+        FILE* fp;
+        char*apikey=get_apikey("secrets.txt");
+        char*machineid=get_machineid("secrets.txt");
+        char* server_url = get_server("setup.txt");
+        //copy secrets.txt into the secrets folder
+        if (rename("secrets.txt", SECRETS)!=0) {
+			error = 8;
+		}
+        //download all the other files
+
         if (error == 0) {
             printf("Downloading files from server\n");
             printf("Downloading cyberhex.exe\n");
@@ -366,6 +454,10 @@ int main()
         printf("Failed to download file\n");
         MessageBox(NULL, L"Failed to download file", L"Error", MB_OK);
 		break;
+    case 8:
+		printf("Failed to open secrets.txt\n");
+		MessageBox(NULL, L"Failed to open secrets.txt", L"Error", MB_OK);
+		break;
     default:
         break;
     }

src/ma_installer/well_known.h

@@ -0,0 +1,42 @@
+#pragma once
+#pragma once
+#ifndef WELL_KNOWN_H
+#define WELL_KNOWN_H
+#define MAIN_COM "main_com.txt"
+#define MAIN_COM_PATH "C:\\Program Files\\cyberhex\\com\\main_com.txt"
+#define ANSWER_COM "answer_com.txt"
+#define ANSWER_COM_PATH "C:\\Program Files\\cyberhex\\com\\answer_com.txt"
+
+#define SECURE_COM "secure_com.txt"
+#define SECURE_COM_PATH "C:\\Program Files\\cyberhex\\secure\\com\\secure_com.txt"
+
+#define SCHED "sched.txt"
+#define SCHED_PATH "C:\\Program Files\\cyberhex\\secure\\settings\\sched.txt"
+
+#define LOGFILE "C:\\Program Files\\cyberhex\\secure\\log\\log.txt"
+#define INFOFILE "C:\\Program Files\\cyberhex\\secure\\log\\info.txt"
+#define WARNFILE "C:\\Program Files\\cyberhex\\secure\\log\\warn.txt"
+#define ERRORFILE "C:\\Program Files\\cyberhex\\secure\\log\\error.txt"
+#define VIRUSFILE "C:\\Program Files\\cyberhex\\secure\\log\\virus.txt"
+#define RISKFILE "C:\\Program Files\\cyberhex\\secure\\log\\risk.txt"
+#define SRV_LOGFILE "C:\\Program Files\\cyberhex\\secure\\log\\srv_log.txt"
+
+
+#define FOLDER_DATABASE_DIR "C:\\Program Files\\cyberhex\\secure\\database\\folder"
+
+#define DB_DIR "C:\\Program Files\\cyberhex\\secure\\database"
+
+#define VIRUS_CTRL_DB "C:\\Program Files\\cyberhex\\secure\\database\\virus_ctrl_db.txt"
+
+#define SETTINGS_DB "C:\\Program Files\\cyberhex\\secure\\settings\\settings_db.txt"
+
+#define QUARANTINE_PATH "C:\\Program Files\\cyberhex\\secure\\quarantine"
+
+#define SECRETS "C:\\Program Files\\cyberhex\\secure\\settings\\secrets.txt"
+
+#define PERIODIC_FOLDER_SCAN "C:\\Program Files\\cyberhex\\secure\\database\\folder\\periodic_folder_scan.txt"
+#define PERIODIC_FOLDER_SCAN_TEMP_DB "C:\\Program Files\\cyberhex\\secure\\database\\folder\\temp_db.txt"
+
+#define INCLUDED_FOLDERS "C:\\Program Files\\cyberhex\\secure\\settings\\included_folders.txt"
+#define EXCLUDED_FOLDERS "C:\\Program Files\\cyberhex\\secure\\settings\\excluded_folders.txt"
+#endif // !WELL_KNOWN_H