From c04f773222bb7118ae58f4710f3db43985bd76df Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ben=20Gr=C3=A4f?= <ben@gra3f.de>
Date: Wed, 18 Jul 2018 21:58:00 +0200
Subject: [PATCH] Fix hugepage status in popup

---
 index.html              |  3 ++-
 src/cc/ClientStatus.cpp | 20 +++++++++++++++-----
 src/cc/Service.cpp      |  4 ++++
 3 files changed, 21 insertions(+), 6 deletions(-)

diff --git a/index.html b/index.html
index 49d82dd1..2c6d5613 100644
--- a/index.html
+++ b/index.html
@@ -567,7 +567,8 @@
             tooltip += '\n';
             tooltip += "CPU Cache L2/L3: " + (row.client_status.cpu_l2 / 1024) + " MB/"+ (row.client_status.cpu_l3 / 1024) + " MB";
             tooltip += '\n';
-            tooltip += "Huge Pages: " + (row.client_status.hugepages_available ? " available" : " unavailable");
+            tooltip += "Huge Pages: " + (row.client_status.hugepages_available ? " available, " : " unavailable, ");
+            tooltip += (row.client_status.hugepages_enabled ? "enabled (" + row.client_status.total_hugepages + "/" + row.client_status.total_pages + ")"  : "disabled");
             tooltip += '\n';
             tooltip += "Used Threads: " + row.client_status.current_threads;
             tooltip += (row.client_status.hash_factor > 1 ? " [" + row.client_status.hash_factor + "x multi hash mode]" :"");
diff --git a/src/cc/ClientStatus.cpp b/src/cc/ClientStatus.cpp
index ceefd778..1a040907 100644
--- a/src/cc/ClientStatus.cpp
+++ b/src/cc/ClientStatus.cpp
@@ -426,10 +426,6 @@ bool ClientStatus::parseFromJson(const rapidjson::Document& document)
             m_isHugepagesEnabled = clientStatus["hugepages_enabled"].GetBool();
         }
 
-        if (clientStatus.HasMember("hash_factor")) {
-            m_hashFactor = clientStatus["hash_factor"].GetInt();
-        }
-
         if (clientStatus.HasMember("cpu_is_x64")) {
             m_isCpuX64 = clientStatus["cpu_is_x64"].GetBool();
         }
@@ -454,6 +450,18 @@ bool ClientStatus::parseFromJson(const rapidjson::Document& document)
             m_hashrateHighest = clientStatus["hashrate_highest"].GetDouble();
         }
 
+        if (clientStatus.HasMember("hash_factor")) {
+            m_hashFactor = clientStatus["hash_factor"].GetInt();
+        }
+
+        if (clientStatus.HasMember("total_pages")) {
+            m_totalPages = clientStatus["total_pages"].GetInt();
+        }
+
+        if (clientStatus.HasMember("total_hugepages")) {
+            m_totalHugepages = clientStatus["total_hugepages"].GetInt();
+        }
+
         if (clientStatus.HasMember("current_threads")) {
             m_currentThreads = clientStatus["current_threads"].GetInt();
         }
@@ -525,7 +533,6 @@ rapidjson::Value ClientStatus::toJson(rapidjson::MemoryPoolAllocator<rapidjson::
 
     clientStatus.AddMember("hugepages_available", m_hasHugepages, allocator);
     clientStatus.AddMember("hugepages_enabled", m_isHugepagesEnabled, allocator);
-    clientStatus.AddMember("hash_factor", m_hashFactor, allocator);
     clientStatus.AddMember("cpu_is_x64", m_isCpuX64, allocator);
     clientStatus.AddMember("cpu_has_aes", m_hasCpuAES, allocator);
 
@@ -534,6 +541,9 @@ rapidjson::Value ClientStatus::toJson(rapidjson::MemoryPoolAllocator<rapidjson::
     clientStatus.AddMember("hashrate_long", m_hashrateLong, allocator);
     clientStatus.AddMember("hashrate_highest", m_hashrateHighest, allocator);
 
+    clientStatus.AddMember("hash_factor", m_hashFactor, allocator);
+    clientStatus.AddMember("total_pages", m_totalPages, allocator);
+    clientStatus.AddMember("total_hugepages", m_totalHugepages, allocator);
     clientStatus.AddMember("current_threads", m_currentThreads, allocator);
     clientStatus.AddMember("cpu_sockets", m_cpuSockets, allocator);
     clientStatus.AddMember("cpu_cores", m_cpuCores, allocator);
diff --git a/src/cc/Service.cpp b/src/cc/Service.cpp
index 05e1542b..b649cc7f 100644
--- a/src/cc/Service.cpp
+++ b/src/cc/Service.cpp
@@ -228,6 +228,10 @@ unsigned Service::setClientStatus(const std::string& clientIp, const std::string
         clientStatus.parseFromJson(document);
         clientStatus.setExternalIp(clientIp);
 
+        if (m_clientStatus.find(clientId) == m_clientStatus.end()) {
+            m_clientCommand[clientId] = ControlCommand(ControlCommand::RESTART);
+        }
+
         m_clientStatus[clientId] = clientStatus;
 
         resultCode = getClientCommand(clientId, resp);