TheCloneArmy/mcrcon
1
0
Fork 0
mirror of https://github.com/Tiiffi/mcrcon.git synced 2025-10-27 19:31:07 -04:00
Code Issues Packages Projects Releases Wiki Activity

Comment cleanups and munged output bug (https://github.com/Tiiffi/mcrcon/issues/2) fixed.

Browse Source
This commit is contained in:
Tiiffi
2016-12-14 00:10:49 +02:00
parent 29f1f9c400
commit b878ff1255
1 changed files with 49 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
mcrcon.c
View File

@ -56,7 +56,7 @@
#define RCON_AUTH_RESPONSE 2 #define RCON_AUTH_RESPONSE 2
#define RCON_PID 0xBADC0DE #define RCON_PID 0xBADC0DE
/* Safe value I think. This should me made dynamic for more stable performance! */ // a bit too big perhaps?
#define DATA_BUFFSIZE 10240 #define DATA_BUFFSIZE 10240
// rcon packet structure // rcon packet structure
@ -68,9 +68,9 @@ typedef struct _rc_packet {
// ignoring string2 atm. // ignoring string2 atm.
} rc_packet; } rc_packet;
/* =================================== */ // ===================================
/* FUNCTION DEFINITIONS */ // FUNCTION DEFINITIONS
/* =================================== */ // ===================================
// endianness related functions // endianness related functions
bool is_bigendian(void); bool is_bigendian(void);
@ -281,7 +281,7 @@ void net_init_WSA(void)
} }
#endif #endif
/* socket close and cleanup */ // socket close and cleanup
void net_close(int sd) void net_close(int sd)
{ {
#ifdef _WIN32 #ifdef _WIN32
@ -292,9 +292,9 @@ void net_close(int sd)
#endif #endif
} }
/* Opens and connects socket */ // Opens and connects socket
/* http://man7.org/linux/man-pages/man3/getaddrinfo.3.html */ // http://man7.org/linux/man-pages/man3/getaddrinfo.3.html
/* https://bugs.chromium.org/p/chromium/issues/detail?id=44489 */ // https://bugs.chromium.org/p/chromium/issues/detail?id=44489
int net_connect(const char *host, const char *port) int net_connect(const char *host, const char *port)
{ {
int sd; int sd;
@ -310,7 +310,6 @@ int net_connect(const char *host, const char *port)
net_init_WSA(); net_init_WSA();
#endif #endif
// Get host address info
int ret = getaddrinfo(host, port, &hints, &server_info); int ret = getaddrinfo(host, port, &hints, &server_info);
if (ret != 0) if (ret != 0)
{ {
@ -377,8 +376,8 @@ int net_send(int sd, const uint8_t *buff, size_t size)
int net_send_packet(int sd, rc_packet *packet) int net_send_packet(int sd, rc_packet *packet)
{ {
int len; int len;
int total = 0; /* how many bytes we've sent */ int total = 0; // bytes we've sent
int bytesleft; /* how many we have left to send */ int bytesleft; // bytes left to send
int ret = -1; int ret = -1;
bytesleft = len = packet->size + sizeof(int); bytesleft = len = packet->size + sizeof(int);
@ -391,7 +390,6 @@ int net_send_packet(int sd, rc_packet *packet)
bytesleft -= ret; bytesleft -= ret;
} }
/* return -1 on failure, 0 on success */
return ret == -1 ? -1 : 1; return ret == -1 ? -1 : 1;
} }
@ -400,7 +398,7 @@ rc_packet *net_recv_packet(int sd)
int psize; int psize;
static rc_packet packet = {0, 0, 0, { 0x00 }}; static rc_packet packet = {0, 0, 0, { 0x00 }};
/* packet.size = packet.id = packet.cmd = 0; */ // packet.size = packet.id = packet.cmd = 0;
int ret = recv(sd, (char *) &psize, sizeof(int), 0); int ret = recv(sd, (char *) &psize, sizeof(int), 0);
@ -465,7 +463,7 @@ int net_clean_incoming(int sd, int size)
void print_color(int color) void print_color(int color)
{ {
/* sh compatible color array */ // sh compatible color array
#ifndef _WIN32 #ifndef _WIN32
char *colors[] = char *colors[] =
{ {
@ -506,7 +504,7 @@ void print_color(int color)
} }
} }
/* this hacky mess might use some optimizing */ // this hacky mess might use some optimizing
void packet_print(rc_packet *packet) void packet_print(rc_packet *packet)
{ {
if (raw_output == 1) if (raw_output == 1)
@ -520,16 +518,22 @@ void packet_print(rc_packet *packet)
#ifdef _WIN32 #ifdef _WIN32
CONSOLE_SCREEN_BUFFER_INFO console_info; CONSOLE_SCREEN_BUFFER_INFO console_info;
if (GetConsoleScreenBufferInfo(console_handle, &console_info) != 0) if (GetConsoleScreenBufferInfo(console_handle, &console_info) != 0)
def_color = console_info.wAttributes + 0x30; def_color = console_info.wAttributes + 0x30;
else def_color = 0x37; else
def_color = 0x37;
#endif #endif
/* colors enabled so try to handle the bukkit colors for terminal */ // colors enabled so try to handle the bukkit colors for terminal
if (print_colors == 1) { if (print_colors == 1)
{
for (i = 0; (unsigned char) packet->data[i] != 0; ++i) for (i = 0; (unsigned char) packet->data[i] != 0; ++i)
{ {
if ((unsigned char) packet->data[i] == 0xc2)
continue;
if ((unsigned char) packet->data[i] == 0xa7) if ((unsigned char) packet->data[i] == 0xa7)
{ {
++i; ++i;
@ -540,14 +544,16 @@ void packet_print(rc_packet *packet)
putchar(packet->data[i]); putchar(packet->data[i]);
} }
print_color(def_color); /* cancel coloring */ print_color(def_color); // cancel coloring
} }
/* strip colors */ // strip colors
else else
{ {
for (i = 0; (unsigned char) packet->data[i] != 0; ++i) for (i = 0; (unsigned char) packet->data[i] != 0; ++i)
{ {
if ((unsigned char) packet->data[i] == 0xc2)
continue;
if ((unsigned char) packet->data[i] == 0xa7) if ((unsigned char) packet->data[i] == 0xa7)
{ {
++i; ++i;
@ -558,16 +564,16 @@ void packet_print(rc_packet *packet)
} }
} }
/* print newline if string has no newline */ // print newline if string has no newline
if (packet->data[i-1] != 10 && packet->data[i-1] != 13) if (packet->data[i-1] != 10 && packet->data[i-1] != 13)
putchar('\n'); putchar('\n');
} }
rc_packet *packet_build(int id, int cmd, char *s1) rc_packet *packet_build(int id, int cmd, char *s1)
{ /* hacky function */ {
static rc_packet packet = {0, 0, 0, { 0x00 }}; static rc_packet packet = {0, 0, 0, { 0x00 }};
/* size + id + cmd + s1 + s2 NULL terminator */ // size + id + cmd + s1 + s2 NULL terminator
int s1_len = strlen(s1); int s1_len = strlen(s1);
if (s1_len > DATA_BUFFSIZE) if (s1_len > DATA_BUFFSIZE)
{ {
@ -605,7 +611,7 @@ uint8_t *packet_build_malloc(size_t *size, int32_t id, int32_t cmd, char *string
return packet; return packet;
} }
/* rcon packet structure */ // rcon packet structure
#define MAX_PACKET_SIZE (size_t) 1460 // including size member #define MAX_PACKET_SIZE (size_t) 1460 // including size member
#define MIN_PACKET_SIZE (size_t) 10 #define MIN_PACKET_SIZE (size_t) 10
#define MAX_STRING_SIZE (size_t) (MAX_PACKET_SIZE - 2 - 3 * sizeof(int32_t)) #define MAX_STRING_SIZE (size_t) (MAX_PACKET_SIZE - 2 - 3 * sizeof(int32_t))
@ -653,13 +659,13 @@ int rcon_auth(int rsock, char *passwd)
ret = net_send_packet(rsock, packet); ret = net_send_packet(rsock, packet);
if (!ret) if (!ret)
return 0; /* send failed */ return 0; // send failed
packet = net_recv_packet(rsock); packet = net_recv_packet(rsock);
if (packet == NULL) if (packet == NULL)
return 0; return 0;
/* return 1 if authentication OK */ // return 1 if authentication OK
return packet->id == -1 ? 0 : 1; return packet->id == -1 ? 0 : 1;
} }
@ -688,7 +694,7 @@ int rcon_command(int rsock, char *command)
return 0; return 0;
if (packet->id != RCON_PID) if (packet->id != RCON_PID)
return 0; /* wrong packet id */ return 0;
if (!silent_mode) if (!silent_mode)
{ {
@ -702,7 +708,6 @@ int rcon_command(int rsock, char *command)
packet_print(packet); packet_print(packet);
} }
/* return 1 if world was saved */
return 1; return 1;
} }
@ -719,7 +724,7 @@ int run_commands(int argc, char *argv[])
return ret; return ret;
} }
/* interactive terminal mode */ // interactive terminal mode
int run_terminal_mode(int rsock) int run_terminal_mode(int rsock)
{ {
int ret = 0; int ret = 0;