Colors component

src/client/component/colors.cpp

@@ -0,0 +1,184 @@
+#include <std_include.hpp>
+#include "loader/component_loader.hpp"
+
+#include "game/game.hpp"
+#include "game/dvars.hpp"
+
+#include <utils/hook.hpp>
+#include <utils/string.hpp>
+
+namespace colors
+{
+	struct HsvColor
+	{
+		unsigned char h;
+		unsigned char s;
+		unsigned char v;
+	};
+
+	namespace
+	{
+		std::vector<DWORD> color_table;
+
+		DWORD hsv_to_rgb(const HsvColor hsv)
+		{
+			DWORD rgb;
+
+			if (hsv.s == 0)
+			{
+				return RGB(hsv.v, hsv.v, hsv.v);
+			}
+
+			// converting to 16 bit to prevent overflow
+			const unsigned int h = hsv.h;
+			const unsigned int s = hsv.s;
+			const unsigned int v = hsv.v;
+
+			const auto region = static_cast<uint8_t>(h / 43);
+			const auto remainder = (h - (region * 43)) * 6;
+
+			const auto p = static_cast<uint8_t>((v * (255 - s)) >> 8);
+			const auto q = static_cast<uint8_t>(
+				(v * (255 - ((s * remainder) >> 8))) >> 8);
+			const auto t = static_cast<uint8_t>(
+				(v * (255 - ((s * (255 - remainder)) >> 8))) >> 8);
+
+			switch (region)
+			{
+			case 0:
+				rgb = RGB(v, t, p);
+				break;
+			case 1:
+				rgb = RGB(q, v, p);
+				break;
+			case 2:
+				rgb = RGB(p, v, t);
+				break;
+			case 3:
+				rgb = RGB(p, q, v);
+				break;
+			case 4:
+				rgb = RGB(t, p, v);
+				break;
+			default:
+				rgb = RGB(v, p, q);
+				break;
+			}
+
+			return rgb;
+		}
+
+		int color_index(const char c)
+		{
+			const auto index = c - 48;
+			return index >= 0xC ? 7 : index;
+		}
+
+		char add(const uint8_t r, const uint8_t g, const uint8_t b)
+		{
+			const char index = '0' + static_cast<char>(color_table.size());
+			color_table.push_back(RGB(r, g, b));
+			return index;
+		}
+
+		void com_clean_name_stub(const char* in, char* out, const int out_size)
+		{
+			strncpy_s(out, out_size, in, _TRUNCATE);
+		}
+
+		char* i_clean_str_stub(char* string)
+		{
+			utils::string::strip(string, string, static_cast<int>(strlen(string)) + 1);
+
+			return string;
+		}
+
+		size_t get_client_name_stub(const int local_client_num, const int index, char* buf, const int size,
+		                            const size_t unk, const size_t unk2)
+		{
+			// CL_GetClientName (CL_GetClientNameAndClantag?)
+			const auto result = reinterpret_cast<size_t(*)(const int, int, char*, int, size_t, size_t)>(0x140213E60)(
+				local_client_num, index, buf, size, unk, unk2);
+
+			utils::string::strip(buf, buf, size);
+
+			return result;
+		}
+
+		void rb_lookup_color_stub(const char index, DWORD* color)
+		{
+			*color = RGB(255, 255, 255);
+
+			if (index == '8')
+			{
+				*color = *reinterpret_cast<DWORD*>(SELECT_VALUE(0x14CEDD978, 0x14D8594C4));
+			}
+			else if (index == '9')
+			{
+				*color = *reinterpret_cast<DWORD*>(SELECT_VALUE(0x14CEDD97C, 0x14D8594C8));
+			}
+			else if (index == ':')
+			{
+				*color = hsv_to_rgb({static_cast<uint8_t>((game::Sys_Milliseconds() / 100) % 256), 255, 255});
+			}
+			else if (index == ';')
+			{
+				*color = *reinterpret_cast<DWORD*>(SELECT_VALUE(0x14CEDD984, 0x14D8594D0));
+			}
+			else if (index == '<')
+			{
+				*color = 0xFFFCFF80;
+			}
+			else
+			{
+				*color = color_table[color_index(index)];
+			}
+		}
+	}
+
+	class component final : public component_interface
+	{
+	public:
+		void post_unpack() override
+		{
+			if (game::environment::is_dedi())
+			{
+				return;
+			}
+
+			if (!game::environment::is_sp())
+			{
+				// allows colored name in-game
+				utils::hook::jump(0x1404C9510, com_clean_name_stub);
+
+				// don't apply colors to overhead names
+				utils::hook::call(0x1401891B0, get_client_name_stub);
+
+				// patch I_CleanStr
+				utils::hook::jump(0x1404C99A0, i_clean_str_stub);
+			}
+
+			// force new colors
+			utils::hook::jump(SELECT_VALUE(0x1404D3A20, 0x1405F2B90), rb_lookup_color_stub);
+
+			// add colors
+			add(0, 0, 0); // 0  - Black
+			add(255, 49, 49); // 1  - Red
+			add(134, 192, 0); // 2  - Green
+			add(255, 173, 34); // 3  - Yellow
+			add(0, 135, 193); // 4  - Blue
+			add(32, 197, 255); // 5  - Light Blue
+			add(151, 80, 221); // 6  - Pink
+			add(255, 255, 255); // 7  - White
+
+			add(0, 0, 0); // 8  - Team color (axis?)
+			add(0, 0, 0); // 9  - Team color (allies?)
+
+			add(0, 0, 0); // 10 - Rainbow (:)
+			add(0, 0, 0);
+			// 11 - Server color (;) - using that color in infostrings (e.g. your name) fails, ';' is an illegal character!
+		}
+	};
+}
+
+REGISTER_COMPONENT(colors::component)

src/client/game/symbols.hpp

@@ -132,6 +132,7 @@ namespace game
 	WEAK symbol<void(const char* error, ...)> Sys_Error{ 0x14038C770, 0x1404D6260 };
 	WEAK symbol<HANDLE(int folder, const char* baseFileName)> Sys_CreateFile{ 0x14037BCA0, 0x1404CC8A0 };
 	WEAK symbol<bool()> Sys_IsDatabaseReady2{ 0x1402FF980, 0x1403E1840 };
+	WEAK symbol<int()> Sys_Milliseconds{ 0x14038E9F0, 0x1404D8730 };
 	WEAK symbol<bool(int, void const*, const netadr_s*)> Sys_SendPacket{ 0x14038E720, 0x1404D8460 };
 	WEAK symbol<void()> Sys_ShowConsole{ 0x14038FA90, 0x1404D98B0 };
 

src/client/loader/loader.cpp

@@ -194,13 +194,6 @@ void loader::load_tls(const utils::nt::library& target, const utils::nt::library
 		const auto tls_index = *reinterpret_cast<DWORD*>(target_tls->AddressOfIndex);
 		*reinterpret_cast<DWORD*>(source_tls->AddressOfIndex) = tls_index;
 
-		// I made sure it's large enough for IW6.
-		/*if (tls_size > TLS_PAYLOAD_SIZE)
-		{
-			throw std::runtime_error(utils::string::va(
-				"TLS data is of size 0x%X, but we have only reserved 0x%X bytes!", tls_size, TLS_PAYLOAD_SIZE));
-		}*/
-
 		DWORD old_protect;
 		VirtualProtect(PVOID(target_tls->StartAddressOfRawData),
 		               source_tls->EndAddressOfRawData - source_tls->StartAddressOfRawData, PAGE_READWRITE,