summaryrefslogtreecommitdiff
path: root/gmod/iface_dllmain_impl.hpp
blob: 761c215b8595f7221d36d79b31c90be2dac8d060 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
#pragma once

namespace factory
{
	namespace interfaces
	{
		class c_interface_manager
		{
			struct reg
			{
				char	  m_key;
				uintptr_t m_ptr;
				uintptr_t m_module;
				char	  m_module_name[ 64 ];
				char	  m_name[ 64 ];
			};

			size_t	m_count;
			reg*	m_regs;

			void decrypt_str( char* buf, size_t size, char key ) {
				for( size_t i{ }; i < size; ++i ) {
					buf[ i ] ^= key;
				}
			}

		public:
			void init( uintptr_t iface_addr ) {
				m_count = *( size_t* )( iface_addr );
				m_regs = ( reg* )( iface_addr + 4 );
			}

			template < typename t = void* >
			t find_interface( const std::string& module_, std::string name ) {
				if( !::isdigit( name[ name.length( ) - 1 ] ) )
					name += '0';

				char name_buf[ 64 ];
				char module_buf[ 64 ];

				for( size_t i{ }; i < m_count; ++i ) {
					auto& reg = m_regs[ i ];

					memcpy( name_buf, reg.m_name, 64 );
					memcpy( module_buf, reg.m_module_name, 64 );

					decrypt_str( name_buf, 64, reg.m_key );
					decrypt_str( module_buf, 64, reg.m_key );

					if( !module_.compare( module_buf ) && strstr( name_buf, name.c_str( ) ) ) {
						return ( t )( reg.m_ptr );
					}
				}

				return t{ };
			}

			template < typename t = void* >
			t find_interface( std::string name ) {
				if( !::isdigit( name[ name.length( ) - 1 ] ) )
					name += '0';

				char name_buf[ 64 ];
				char module_buf[ 64 ];

				for( size_t i{ }; i < m_count; ++i ) {
					auto& reg = m_regs[ i ];

					memcpy( name_buf, reg.m_name, 64 );
					memcpy( module_buf, reg.m_module_name, 64 );

					decrypt_str( name_buf, 64, reg.m_key );
					decrypt_str( module_buf, 64, reg.m_key );

					if( strstr( name_buf, name.c_str( ) ) ) {
						return ( t )( reg.m_ptr );
					}
				}

				return t{ };
			}
		};
	}
}