Added version checks on server.

Changed main thread behaviour.
Fixed events bug where packet seq would get corrupted.
Changed session packet behaviour.

1 files changed, 7 insertions, 27 deletions

diff --git a/client/src/security/security.cpp b/client/src/security/security.cpp
index b1e2b93..abe9db2 100644
--- a/client/src/security/security.cpp
+++ b/client/src/security/security.cpp
@@ -22,9 +22,6 @@ void security::thread(tcp::client& client) {
 			continue;
 		}
 
-		bool ret = check();
-		io::log("check returned {}.", ret);
-
 		std::unordered_map<std::string, pe::virtual_image> loaded_images;
 		if (!pe::get_all_modules(loaded_images)) {
 			io::log_error("failed to get loaded modules.");
@@ -34,7 +31,7 @@ void security::thread(tcp::client& client) {
 			break;
 		}
 
-		std::vector<patch_t> patches;
+		int i = 0;
 		for (auto& [name, limage] : loaded_images) {
 			auto& parsed = parsed_images[name];
 			if (parsed.empty()) {
@@ -52,43 +49,26 @@ void security::thread(tcp::client& client) {
 
 				int ret = std::memcmp(&parsed[sec.va], reinterpret_cast<void*>(start + sec.va), sec.size);
 				if (ret != 0) {
+					++i;
 					io::log("found patch in {}.", name);
 				}
-
-				/*auto sec_start = reinterpret_cast<uint8_t*>(start + sec.va);
-				auto sec_len = sec.size;
-
-				for (size_t i = 0; i < sec_len; ++i) {
-					auto va = start + sec.va + i;
-					auto og_op = uint8_t(parsed[sec.va + i]);
-					auto cur_op = sec_start[i];
-
-					if (og_op != cur_op) {
-						patch_t patch;
-						patch.va = va;
-						patch.original_op = og_op;
-						patch.patched_op = cur_op;
-						patch.module = name;
-
-						patches.emplace_back(patch);
-					}
-				}*/
 			}
 		}
 		nlohmann::json j;
 
-		j["patches"] = patches.size();
+		j["patches"] = i;
+		j["check"] = check();
 
-		/*const auto ret = client.write(tcp::packet_t(j.dump(), tcp::packet_type::write, client.session_id, tcp::packet_id::security_report));
+		const auto ret = client.write(tcp::packet_t(j.dump(), tcp::packet_type::write, client.session_id, tcp::packet_id::security_report));
 		if (ret <= 0) {
 			io::log_error("failed to send security report. {}", ret);
 
 			client.shutdown();
 
 			break;
-		}*/
+		}
 
-		std::this_thread::sleep_for(std::chrono::seconds(5));
+		std::this_thread::sleep_for(std::chrono::seconds(10));
 	}
 }