Prepare support for serialization of non-default-constructible objects

src/windows-emulator/emulator_utils.hpp

@@ -7,7 +7,7 @@ class emulator_object
 public:
 	using value_type = T;
 
-	emulator_object() = default;
+	//emulator_object() = default;
 
 	emulator_object(emulator& emu, const uint64_t address = 0)
 		: emu_(&emu)
@@ -25,7 +25,7 @@ public:
 		return this->address_;
 	}
 
-	uint64_t size() const
+	constexpr uint64_t size() const
 	{
 		return sizeof(T);
 	}
@@ -174,6 +174,11 @@ public:
 		return this->active_address_;
 	}
 
+	emulator& get_emulator() const
+	{
+		return *this->emu_;
+	}
+
 	void serialize(utils::buffer_serializer& buffer) const
 	{
 		buffer.write(this->address_);

src/windows-emulator/handles.hpp

@@ -105,7 +105,7 @@ public:
 	handle store(T value)
 	{
 		auto index = this->find_free_index();
-		this->store_[index] = std::move(value);
+		this->store_.emplace(index, std::move(value));
 
 		return make_handle(index);
 	}

src/windows-emulator/process_context.hpp

@@ -192,7 +192,10 @@ private:
 class emulator_thread : ref_counted_object
 {
 public:
-	emulator_thread() = default;
+	emulator_thread(x64_emulator& emu)
+		: emu_ptr(&emu)
+	{
+	}
 
 	emulator_thread(x64_emulator& emu, const process_context& context, uint64_t start_address, uint64_t argument,
 	                uint64_t stack_size, uint32_t id);
@@ -205,20 +208,7 @@ public:
 
 	~emulator_thread()
 	{
-		if (marker.was_moved())
-		{
-			return;
-		}
-
-		if (this->stack_base)
-		{
-			this->emu_ptr->release_memory(this->stack_base, this->stack_size);
-		}
-
-		if (this->gs_segment)
-		{
-			this->gs_segment->release();
-		}
+		this->release();
 	}
 
 	moved_marker marker{};
@@ -285,6 +275,11 @@ public:
 
 	void serialize(utils::buffer_serializer& buffer) const
 	{
+		if (this->marker.was_moved())
+		{
+			throw std::runtime_error("Object was moved!");
+		}
+
 		buffer.write(this->stack_base);
 		buffer.write(this->stack_size);
 		buffer.write(this->start_address);
@@ -310,6 +305,13 @@ public:
 
 	void deserialize(utils::buffer_deserializer& buffer)
 	{
+		if (this->marker.was_moved())
+		{
+			throw std::runtime_error("Object was moved!");
+		}
+
+		this->release();
+
 		buffer.read(this->stack_base);
 		buffer.read(this->stack_size);
 		buffer.read(this->start_address);
@@ -335,6 +337,31 @@ public:
 
 private:
 	void setup_registers(x64_emulator& emu, const process_context& context) const;
+
+	void release()
+	{
+		if (this->marker.was_moved())
+		{
+			return;
+		}
+
+		if (this->stack_base)
+		{
+			if (!this->emu_ptr)
+			{
+				throw std::runtime_error("Emulator was never assigned!");
+			}
+
+			this->emu_ptr->release_memory(this->stack_base, this->stack_size);
+			this->stack_base = 0;
+		}
+
+		if (this->gs_segment)
+		{
+			this->gs_segment->release();
+			this->gs_segment = {};
+		}
+	}
 };
 
 struct process_context
@@ -454,6 +481,7 @@ struct process_context
 
 		buffer.read_vector(this->default_register_set);
 		buffer.read(this->current_thread_id);
+
 		buffer.read(this->threads);
 
 		this->active_thread = this->threads.get(buffer.read<uint64_t>());

src/windows-emulator/syscalls.cpp

@@ -1848,7 +1848,7 @@ namespace
 	                           const ULONG share_access,
 	                           const ULONG open_options)
 	{
-		return handle_NtCreateFile(c, file_handle, desired_access, object_attributes, io_status_block, {}, 0,
+		return handle_NtCreateFile(c, file_handle, desired_access, object_attributes, io_status_block, {c.emu}, 0,
 		                           share_access, FILE_OPEN, open_options, 0, 0);
 	}
 

src/windows-emulator/windows_emulator.cpp

@@ -279,9 +279,9 @@ namespace
 
 		if (record.ExceptionRecord)
 		{
-			record_mapping[&record] = record_obj;
+			record_mapping.emplace(&record, record_obj);
 
-			emulator_object<EXCEPTION_RECORD> nested_record_obj{};
+			emulator_object<EXCEPTION_RECORD> nested_record_obj{allocator.get_emulator()};
 			const auto nested_record = record_mapping.find(record.ExceptionRecord);
 
 			if (nested_record != record_mapping.end())
@@ -584,7 +584,7 @@ void emulator_thread::mark_as_ready(const NTSTATUS status)
 
 bool emulator_thread::is_thread_ready(process_context& context)
 {
-	if(this->exit_status.has_value())
+	if (this->exit_status.has_value())
 	{
 		return false;
 	}
@@ -903,7 +903,7 @@ void windows_emulator::start(std::chrono::nanoseconds timeout, size_t count)
 			const auto current_instructions = this->process().executed_instructions;
 			const auto diff = current_instructions - start_instructions;
 
-			if(diff >= count)
+			if (diff >= count)
 			{
 				break;
 			}
@@ -923,6 +923,11 @@ void windows_emulator::serialize(utils::buffer_serializer& buffer) const
 
 void windows_emulator::deserialize(utils::buffer_deserializer& buffer)
 {
+	buffer.register_factory<emulator_thread>([this]
+	{
+		return emulator_thread(this->emu());
+	});
+
 	this->emu().deserialize(buffer);
 	this->process_.deserialize(buffer);
 	this->dispatcher_.deserialize(buffer);