-
Notifications
You must be signed in to change notification settings - Fork 0
/
yarcon.h
248 lines (221 loc) · 6.79 KB
/
yarcon.h
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
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
#ifndef YARCON_H
#define YARCON_H
#include <stdlib.h>
#include "color_utils.h"
#define PROGRAM_NAME "yarcon"
#define VERSION "0.1.0"
#define MAX_BUFFER_SIZE 4096
#define MAX_LINES_SIZE 128
#define MAX_LINE_SIZE 64
// Globals
struct sockaddr_in si_other;
unsigned int sockaddr_len;
typedef enum rcon_protocol_implementation_type
{
RCON_SOURCE_PROTOCOL = 0,
RCON_BATTLEYE_PROTOCOL = 1
} RconProtocolType;
typedef enum source_packet_type
{
SERVERDATA_RESPONSE_VALUE = 0,
SERVERDATA_AUTH_RESPONSE = 2,
SERVERDATA_EXECCOMMAND = 2,
SERVERDATA_AUTH = 3,
} PacketSourceType;
typedef enum be_packet_type
{
BE_PACKET_LOGIN = 0x00,
BE_PACKET_COMMAND = 0x01,
BE_PACKET_MESSAGE = 0x02,
} PacketBattleyeType;
// Their payload follows the following basic structure:
// You can find out more information at
typedef struct packet_source_struct
{
int32_t size; // 4 bytes
int32_t id; // 4 bytes
PacketSourceType type; // 1 byte
char body[MAX_BUFFER_SIZE]; // len of body
size_t len;
} Pckt_Src_Struct;
// BE Packet Structure
// Source: https://www.battleye.com/downloads/BERConProtocol.txt
typedef struct packet_battleye_struct
{
unsigned char start_header[2]; // 'B'(0x42) | 'E'(0x45)
uint32_t checksum; // 4-byte CRC32 checksum of the subsequent bytes
unsigned char payload[MAX_BUFFER_SIZE]; // 0xff + packet type + command
} Pckt_BE_Struct;
static void print_usage()
{
puts("");
puts(RED "yarcon " VERSION RESET " - " GREEN "https://github.com/xbelanch/yarcon" RESET);
puts("Send rcon commands to game servers with rcon support.");
puts("Usage: yarcon [OPTIONS] [COMMANDS]");
puts("Options:");
puts("-H\t\tHost address (example: 0.0.0.0)");
puts("-p\t\tPort (example: 2301)");
puts("-P\t\tpassword");
puts("-h\t\tPrint usage");
puts("-c\t\tCommand");
puts("-b\t\tUse Battleye instead of Source Protocol");
puts("-d\t\tGive us some debug info");
// puts("-f\t\tOpen config file");
puts("Examples: ");
puts(BLUE "Project Zomboid:\t" YELLOW "yarcon -d -H 0.0.0.0 -p 16261 -P password -c players" RESET);
puts(BLUE "DayZ:\t\t\t" YELLOW "yarcon -b -d -H 0.0.0.0 -p 2301 -P password -c players" RESET);
}
// Simple function for removing a character
static char *strremove(char *s, char chr) {
char *e = malloc(sizeof(char) * 1024);
memset(e, 0, 1024);
char *d = s;
char *ptr = e;
while (*d != '\0') {
if (*d == chr) {
++d;
} else {
*e++ = *d++;
}
}
e = ptr;
return (e);
}
// Simple append string function
void append_str(char *src, unsigned char *dst)
{
while (*src != '\0') {
*dst++ = *src++;
}
}
// Stolen from: https://gist.github.com/MultiMote/169265fd74fe94b44941c1b05b296f0d
uint32_t crc32(unsigned char *begin, unsigned char *end) {
int j;
uint32_t byte, crc, mask;
static uint32_t table[256];
/* Set up the table, if necessary. */
if (table[1] == 0) {
for (byte = 0; byte <= 255; byte++) {
crc = byte;
for (j = 7; j >= 0; j--) { // Do eight times.
mask = -(crc & 1);
crc = (crc >> 1) ^ (0xEDB88320 & mask);
}
table[byte] = crc;
}
}
/* Through with table setup, now calculate the CRC. */
crc = 0xFFFFFFFF;
while (begin != end) {
byte = *begin;
crc = (crc >> 8) ^ table[(crc ^ byte) & 0xFF];
++begin;
}
return ~crc;
}
int serialize_int32_t(int32_t val, char *buffer)
{
// val = htonl(val);
memcpy(buffer, &val, sizeof(int32_t));
return (0);
}
int rcon_serialize_data(Pckt_Src_Struct *pckt, char *buffer)
{
char *ptr = buffer;
serialize_int32_t(pckt->size, ptr);
ptr += sizeof(int32_t);
serialize_int32_t(pckt->id, ptr);
ptr += sizeof(int32_t);
serialize_int32_t(pckt->type, ptr);
ptr += sizeof(int32_t);
memcpy(ptr, pckt->body, strlen(pckt->body));
return (0);
}
void rcon_populate_source_packet(Pckt_Src_Struct *pckt, PacketSourceType type, char *body)
{
pckt->size = (sizeof(uint32_t) * 2) + strlen(body) + 2;
pckt->id = abs(rand());
pckt->type = type;
pckt->len = (sizeof(uint32_t) * 3) + strlen(body) + 2;
memcpy(pckt->body, body, strlen(body));
}
int rcon_serialize_be_data(Pckt_BE_Struct *pckt, char *buffer)
{
char *ptr = buffer;
memcpy(ptr, pckt->start_header, sizeof(unsigned char) * 2);
ptr += sizeof(unsigned char) * 2;
memcpy(ptr, &pckt->checksum, sizeof(uint32_t));
ptr += sizeof(uint32_t);
memcpy(ptr, &pckt->payload, 1024);
return (0);
}
void rcon_populate_be_packet(Pckt_BE_Struct *pckt, PacketBattleyeType type, char *body)
{
// For checksum we need to populate payload first
unsigned char payload[1024];
memset(payload, '\0', 1024);
unsigned char *ptr = payload;
memset(ptr++, 0xFF, sizeof(unsigned char));
memset(ptr++, type, sizeof(unsigned char));
int offset = 2;
if (type == BE_PACKET_COMMAND) {
memset(ptr++, 0x0, sizeof(unsigned char));
offset++;
}
append_str(body, ptr);
memcpy(pckt->payload, payload, offset + strlen(body));
unsigned char *begin = payload;
unsigned char *end = payload + offset + strlen(body);
pckt->checksum = crc32(begin, end);
append_str("BE", pckt->start_header);
}
int rcon_server_connect(char *host,
char *port,
RconProtocolType type)
{
int sckfd;
sockaddr_len = sizeof(si_other);
struct hostent *hostname = gethostbyname(host);
si_other.sin_port = htons(atoi(port));
si_other.sin_family = AF_INET;
if( hostname != NULL) {
memcpy(&si_other.sin_addr, hostname->h_addr_list[0], hostname->h_length);
} else {
si_other.sin_addr.s_addr = inet_addr(host);
}
// SOURCE
if (type == RCON_SOURCE_PROTOCOL) {
sckfd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if (sckfd < 0) {
perror(RED "[!] " RESET "Error");
exit(1);
}
int err = connect(sckfd, (struct sockaddr *) &si_other, sockaddr_len);
if (err < 0) {
perror(RED "[!] " RESET "Error");
exit(1);
}
} // BE
else if (type == RCON_BATTLEYE_PROTOCOL) {
if ((sckfd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) < 0 ) {
perror(RED "[!] " RESET "Error");
exit(1);
}
}
return (sckfd);
}
int rcon_send(int sckfd, char *buffer, int buffer_size, bool battleye) {
int ret;
if (battleye) {
ret = sendto(sckfd, buffer, buffer_size, 0, (struct sockaddr *) &si_other, sockaddr_len);
} else {
ret = send(sckfd, buffer, buffer_size, 0);
}
if (ret == -1) {
perror(RED "[!] " RESET "Error");
return (1);
} else {
return (ret);
}
}
#endif // YARCON_H