Test DyKnow folder size on program start

eschan145 · Jan 22, 2025 · f10486e · f10486e
1 parent 727cfe8
commit f10486e
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 0 deletions.
diff --git a/src/api.cpp b/src/api.cpp
@@ -87,6 +87,41 @@ extern "C" {
     }
 }
 
+DK_API uint64_t get_dyknow_size(const std::string& directory) {
+    uint64_t total = 0;
+    WIN32_FIND_DATAA data;
+    HANDLE find = FindFirstFile((directory + "\\*").c_str(), &data);
+
+    if (find = INVALID_HANDLE_VALUE) {
+        error("Failed to access DyKnow folder! Validating...\n");
+        validate();
+        return 0;
+    }
+
+    do {
+        std::string filename = data.cFileName;
+
+        if ((filename == ".") || (filename == "..")) {
+            continue;
+        }
+
+        std::string full_path = FOLDER_PATH + "\\" + filename;
+
+        if (data.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
+            // If it's a directory, recursively calculate its size
+            total += get_dyknow_size(directory);
+        } else {
+            LARGE_INTEGER file_size;
+
+            file_size.LowPart = data.nFileSizeLow;
+            file_size.HighPart = data.nFileSizeHigh;
+            total += file_size.QuadPart;
+        }
+    } while (FindNextFileA(find, &data) != 0);
+
+    FindClose(find);
+}
+
 DK_API void validate() {
     /*
     Check for the validity of the DyKnow installation. If the DyKnow
@@ -133,6 +168,9 @@ DK_API void validate() {
         std::cout << "Successfully validated DyKnow installation and file "
                   << "integrity.\n";
     }
+
+    uint64_t size = get_dyknow_size();
+    std::cout << "DyKnow folder size: " << size << "\n";
 }
 
 bool exists(const char* path) {

diff --git a/src/api.h b/src/api.h
@@ -100,8 +100,11 @@ extern "C" {
     DK_API int get_kill_method();
     DK_API void set_kill_method(int value);
 
+    DK_API uint64_t get_dyknow_size(const std::string& directory = FOLDER_PATH);
+
     DK_API void validate();
     DK_API const char* get_folder_path();
+
     DK_API void start_monitoring(const char* folder_path = FOLDER_PATH);
     DK_API void stop_monitoring();
     DK_API int monitor_executables(int interval);