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
|
#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;
return true;
}
|
