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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
#include "client.hpp"
#include "server.hpp"
#include "injection.hpp"
/*
std::vector< byte > server::c_client::receive_message( ) {
std::vector< uint8_t > ret;
char buffer[ BUFFER_SIZE ]{ };
int received = 0;
while( true ) {
received = recv( m_socket, buffer, BUFFER_SIZE, 0 );
if( received < 0 )
break;
for( int i{ }; i < received; ++i )
ret.push_back( buffer[ i ] );
if( received < BUFFER_SIZE )
break;
}
if( ret.size( ) ) {
decode_buffer( ret.data( ), ret.size( ) );
ret.push_back( 0 );
ret.erase( ret.begin( ) );
}
return ret;
}
*/
void server::c_client::kill() {
closesocket(m_socket);
printf("%s disconnected\n", get_ip());
}
std::string server::c_client::get_msg() {
std::string ret{ };
char buffer[BUFFER_SIZE]{ };
int received = 0;
while(true) {
received = recv(m_socket, buffer, BUFFER_SIZE, 0);
if(received < 0)
break;
for(int i{ }; i < received; ++i)
ret.push_back(buffer[i]);
if(received < BUFFER_SIZE)
break;
}
if(ret.size()) {
decode_buffer((uint8_t*)ret.data(), ret.size());
// ret.push_back( 0 ); somehow broke things :/
ret.erase(ret.begin());
}
return ret;
}
bool server::c_client::send_msg(byte* msg, size_t length) {
auto buffer = std::make_unique< uint8_t[] >(length + 1);
uint8_t key = util::random_number(0, 255) & 0xff;
buffer[0] = key;
memcpy(buffer.get() + 1,
msg,
length);
for(size_t i = 1; i <= length; ++i) {
buffer[i] ^= key;
}
int result = send(m_socket, (char*)buffer.get(), (int)length + 1, 0);
if(result == -1) {
#if WIN64
printf("error sending message to %s: %d\n",
get_ip(), WSAGetLastError());
#else
printf("error sending message to %s\n",
get_ip());
#endif
return false;
}
return true;
}
bool server::c_client::send_msg(const char* msg) {
auto length = strlen(msg);
auto buffer = std::make_unique< uint8_t[] >(length + 1);
uint8_t key = util::random_number(0, 255) & 0xff;
memset(buffer.get(), 0, length + 1);
buffer[0] = key;
memcpy(buffer.get() + 1,
msg,
length);
for(size_t i = 1; i <= length; ++i) {
buffer[i] ^= key;
}
int result = send(m_socket, (char*)buffer.get(), (int)length + 1, 0);
if(result == -1) {
#if WIN64
printf("error sending message to %s: %d\n",
get_ip(), WSAGetLastError());
#else
printf("error sending message to %s\n",
get_ip());
#endif
return false;
}
return true;
}
bool server::c_client::handle() {
if(get_msg() != "hello")
return false;
inject::c_pe_file pe_file;
if(!pe_file.get("test.dll"))
return false;
printf("file read\n");
inject::c_inject_transaction transaction;
if(!transaction.get(pe_file))
return false;
printf("starting transaction\n");
transaction.process_reloc(0x10000000);
printf("reloc processed\n");
return true;
}
|
