From 4936c6bb20496df5bdace659d50dbb7ee770cb69 Mon Sep 17 00:00:00 2001
From: sowle <crypto.sowle@gmail.com>
Date: Tue, 27 Aug 2019 03:25:35 +0300
Subject: [PATCH] core: check for free space at launch as well

---
 src/currency_core/currency_core.cpp | 23 ++++++++++++++++-------
 src/currency_core/currency_core.h   |  3 ++-
 2 files changed, 18 insertions(+), 8 deletions(-)

diff --git a/src/currency_core/currency_core.cpp b/src/currency_core/currency_core.cpp
index 982b3311..54b93443 100644
--- a/src/currency_core/currency_core.cpp
+++ b/src/currency_core/currency_core.cpp
@@ -19,6 +19,8 @@ using namespace epee;
 #include "currency_format_utils.h"
 #include "misc_language.h"
 
+#define MINIMUM_REQUIRED_FREE_SPACE_BYTES (1024 * 1024 * 100)
+
 DISABLE_VS_WARNINGS(4355)
 #undef LOG_DEFAULT_CHANNEL 
 #define LOG_DEFAULT_CHANNEL "core"
@@ -136,6 +138,9 @@ namespace currency
   {
     bool r = handle_command_line(vm);
 
+    uint64_t available_space = 0;
+    CHECK_AND_ASSERT_MES(!check_if_free_space_critically_low(&available_space), false, "free space in data folder is critically low: " << std::fixed << available_space / (1024 * 1024) << " MB");
+
     r = m_mempool.init(m_config_folder);
     CHECK_AND_ASSERT_MES(r, false, "Failed to initialize memory pool");
 
@@ -710,18 +715,22 @@ namespace currency
       l->on_blockchain_update();
   }
   //-----------------------------------------------------------------------------------------------
-#define MINIMUM_REQUIRED_FREE_SPACE_BYTES (1024 * 1024 * 100)
+  bool core::check_if_free_space_critically_low(uint64_t* p_available_space /* = nullptr */)
+  {
+    boost::filesystem::space_info si = boost::filesystem::space(m_config_folder);
+    if (p_available_space != nullptr)
+      *p_available_space = si.available;
+    return si.available < MINIMUM_REQUIRED_FREE_SPACE_BYTES;
+  }
+
   void core::check_free_space()
   {
     if (!m_critical_error_handler)
       return;
 
-    boost::filesystem::space_info si = boost::filesystem::space(m_config_folder);
-
-    if (si.available < MINIMUM_REQUIRED_FREE_SPACE_BYTES)
-    {
-      m_critical_error_handler->on_critical_low_free_space(si.available, MINIMUM_REQUIRED_FREE_SPACE_BYTES);
-    }
+    uint64_t available_space = 0;
+    if (check_if_free_space_critically_low(&available_space))
+      m_critical_error_handler->on_critical_low_free_space(available_space, MINIMUM_REQUIRED_FREE_SPACE_BYTES);
   }
   //-----------------------------------------------------------------------------------------------
 
diff --git a/src/currency_core/currency_core.h b/src/currency_core/currency_core.h
index 2f7f7485..57fc1241 100644
--- a/src/currency_core/currency_core.h
+++ b/src/currency_core/currency_core.h
@@ -136,8 +136,9 @@ namespace currency
 
      void notify_blockchain_update_listeners();
 
+     bool check_if_free_space_critically_low(uint64_t* p_available_space = nullptr);
      void check_free_space();
-
+     
 
      blockchain_storage m_blockchain_storage;
      tx_memory_pool m_mempool;