Small fix

src/component/gsc.cpp

@@ -21,7 +21,7 @@ namespace gsc
 	{
 		std::string method_name(unsigned int id)
 		{
-			const auto map = *game::plutonium::method_map_rev;
+			const auto& map = *game::plutonium::method_map_rev;
 
 			for (const auto& function : map)
 			{
@@ -36,7 +36,7 @@ namespace gsc
 
 		std::string function_name(unsigned int id)
 		{
-			const auto map = *game::plutonium::function_map_rev;
+			const auto& map = *game::plutonium::function_map_rev;
 
 			for (const auto& function : map)
 			{
@@ -192,7 +192,7 @@ namespace gsc
 				return scr_get_object_field_hook.invoke<void>(classnum, entnum, offset);
 			}
 
-			const auto field = custom_fields[classnum][offset];
+			const auto& field = custom_fields[classnum][offset];
 
 			try
 			{
@@ -217,7 +217,7 @@ namespace gsc
 			}
 
 			const auto args = get_arguments();
-			const auto field = custom_fields[classnum][offset];
+			const auto& field = custom_fields[classnum][offset];
 
 			try
 			{
@@ -272,11 +272,20 @@ namespace gsc
 			const std::function<scripting::script_value(unsigned int entnum)>& getter,
 			const std::function<void(unsigned int entnum, const scripting::script_value&)>& setter)
 		{
-			const auto token_id = token_map_start++;
+			uint16_t token_id{};
-			const auto offset = field_offset_start++;
+			auto& token_map = *game::plutonium::token_map_rev;
+			if (token_map.find(name) != token_map.end())
+			{
+				token_id = token_map.at(name);
+			}
+			else
+			{
+				token_id = token_map_start++;
+				token_map.insert(std::make_pair(name, token_id));
+			}
 
+			const auto offset = field_offset_start++;
 			custom_fields[classnum][offset] = {name, getter, setter};
-			(*game::plutonium::token_map_rev)[name] = token_id;
 
 			post_load_callbacks.push_back([classnum, name, token_id, offset]()
 			{
@@ -320,7 +329,7 @@ namespace gsc
 			scr_set_object_field_hook.create(0x52BCC0, scr_set_object_field_stub);
 			scr_post_load_scripts_hook.create(0x628B50, scr_post_load_scripts_stub);
 
-			field::add(classid::entity, "flags",
+			field::add(classid::entity, "entityflags",
 				[](unsigned int entnum) -> scripting::script_value
 				{
 					const auto entity = &game::g_entities[entnum];

src/component/signatures.cpp

@@ -2,8 +2,6 @@
 #include "signatures.hpp"
 #include <utils/hook.hpp>
 
-#define PAYLOAD_SIZE 0x20000000
-
 namespace signatures
 {
 	size_t load_image_size()
@@ -34,7 +32,8 @@ namespace signatures
 	{
 		const char* string_ptr = nullptr;
 		std::string mask(string.size(), 'x');
-		utils::hook::signature signature(PAYLOAD_SIZE, get_image_size() - PAYLOAD_SIZE);
+		const auto base = reinterpret_cast<size_t>(GetModuleHandle("plutonium-bootstrapper-win32.exe"));
+		utils::hook::signature signature(base, get_image_size() - base);
 
 		signature.add({
 			string,
@@ -53,8 +52,8 @@ namespace signatures
 	{
 		char bytes[4] = {0};
 		const auto string_ptr = find_string_ptr(string);
-		memcpy(bytes, &string_ptr, sizeof(size_t));
+		memcpy(bytes, &string_ptr, sizeof(bytes));
-		return find_string_ptr(bytes);
+		return find_string_ptr({bytes, 4});
 	}
 
 	bool process_maps()
@@ -70,19 +69,6 @@ namespace signatures
 		game::plutonium::method_map_rev.set(map_ptr + 0x20);
 		game::plutonium::file_map_rev.set(map_ptr + 0x40);
 		game::plutonium::token_map_rev.set(map_ptr + 0x60);
-
-		for (const auto& function : *game::plutonium::function_map_rev)
-		{
-			utils::io::write_file("functions.txt", function.first, true);
-			utils::io::write_file("functions.txt", "\n", true);
-		}
-
-		for (const auto& function : *game::plutonium::method_map_rev)
-		{
-			utils::io::write_file("methods.txt", function.first, true);
-			utils::io::write_file("methods.txt", "\n", true);
-		}
-
 		return true;
 	}
 
@@ -102,7 +88,6 @@ namespace signatures
 	bool process()
 	{
 		load_function_tables();
-		process_printf();
+		return process_printf() && process_maps();
-		return process_maps();
 	}
 }