Serialize registry manager

momo5502 · Nov 3, 2024 · c96aaaa · c96aaaa
1 parent 898299d
commit c96aaaa
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 13 deletions.
diff --git a/src/emulator/serialization.hpp b/src/emulator/serialization.hpp
@@ -39,7 +39,8 @@ namespace utils
 
 		template <typename T>
 		struct has_serialize_function<T, std::void_t<decltype(::serialize(std::declval<buffer_serializer&>(),
-		                                                                std::declval<const std::remove_cvref_t<T>&>()))>>
+		                                                                  std::declval<const std::remove_cvref_t<T>&>())
+		                              )>>
 			: std::true_type
 		{
 		};
@@ -51,7 +52,8 @@ namespace utils
 
 		template <typename T>
 		struct has_deserialize_function<T, std::void_t<decltype(::deserialize(
-			                                std::declval<buffer_deserializer&>(), std::declval<std::remove_cvref_t<T>&>()))>>
+			                                std::declval<buffer_deserializer&>(),
+			                                std::declval<std::remove_cvref_t<T>&>()))>>
 			: std::true_type
 		{
 		};

diff --git a/src/windows-emulator/registry_manager.cpp b/src/windows-emulator/registry_manager.cpp
@@ -1,5 +1,7 @@
 #include "registry_manager.hpp"
+
 #include <hive_parser.hh>
+#include <serialization_helper.hpp>
 
 namespace
 {
@@ -39,31 +41,41 @@ namespace
 
 registry_manager::~registry_manager() = default;
 
-registry_manager::registry_manager(const std::filesystem::path& hive_path)
+registry_manager::registry_manager(std::filesystem::path hive_path)
+	: hive_path_(std::move(hive_path))
 {
+	this->setup();
+}
+
+void registry_manager::setup()
+{
+	this->path_mapping_.clear();
+	this->hives_.clear();
+
 	const std::filesystem::path root = R"(\registry)";
 	const std::filesystem::path machine = root / "machine";
 
-	register_hive(this->hives_, machine / "system", hive_path / "SYSTEM");
-	register_hive(this->hives_, machine / "security", hive_path / "SECURITY");
-	register_hive(this->hives_, machine / "sam", hive_path / "SAM");
-	register_hive(this->hives_, machine / "software", hive_path / "SOFTWARE");
-	register_hive(this->hives_, machine / "system", hive_path / "SYSTEM");
-	register_hive(this->hives_, machine / "hardware", hive_path / "HARDWARE");
+	register_hive(this->hives_, machine / "system", this->hive_path_ / "SYSTEM");
+	register_hive(this->hives_, machine / "security", this->hive_path_ / "SECURITY");
+	register_hive(this->hives_, machine / "sam", this->hive_path_ / "SAM");
+	register_hive(this->hives_, machine / "software", this->hive_path_ / "SOFTWARE");
+	register_hive(this->hives_, machine / "system", this->hive_path_ / "SYSTEM");
+	register_hive(this->hives_, machine / "hardware", this->hive_path_ / "HARDWARE");
 
-	register_hive(this->hives_, root / "user", hive_path / "NTUSER.dat");
+	register_hive(this->hives_, root / "user", this->hive_path_ / "NTUSER.dat");
 
 	this->add_path_mapping(machine / "system" / "CurrentControlSet", machine / "system" / "ControlSet001");
 }
 
 void registry_manager::serialize(utils::buffer_serializer& buffer) const
 {
-	(void)buffer;
+	buffer.write(this->hive_path_);
 }
 
 void registry_manager::deserialize(utils::buffer_deserializer& buffer)
 {
-	(void)buffer;
+	buffer.read(this->hive_path_);
+	this->setup();
 }
 
 std::filesystem::path registry_manager::normalize_path(const std::filesystem::path& path) const

diff --git a/src/windows-emulator/registry_manager.hpp b/src/windows-emulator/registry_manager.hpp
@@ -35,7 +35,8 @@ class registry_manager
 	using hive_ptr = std::unique_ptr<hive_parser>;
 	using hive_map = std::unordered_map<std::filesystem::path, hive_ptr>;
 
-	registry_manager(const std::filesystem::path& hive_path);
+	registry_manager() = default;
+	registry_manager(std::filesystem::path hive_path);
 	~registry_manager();
 
 	void serialize(utils::buffer_serializer& buffer) const;
@@ -45,11 +46,14 @@ class registry_manager
 	std::optional<registry_value> get_value(const registry_key& key, const std::string_view name);
 
 private:
+	std::filesystem::path hive_path_{};
 	hive_map hives_{};
 	std::unordered_map<std::filesystem::path, std::filesystem::path> path_mapping_{};
 
 	std::filesystem::path normalize_path(const std::filesystem::path& path) const;
 	void add_path_mapping(const std::filesystem::path& key, const std::filesystem::path& value);
 
 	hive_map::iterator find_hive(const std::filesystem::path& key);
+
+	void setup();
 };