mirror of
https://github.com/Tiiffi/mcrcon.git
synced 2025-10-28 11:51:08 -04:00
Compare commits
35 Commits
4488127350
...
Develop
| Author | SHA1 | Date | |
|---|---|---|---|
| dfd840de1f | |||
| d5515f43ce | |||
| 4a1da8d373 | |||
| b1b46ca08c | |||
| 2bb1fafdaa | |||
| 0fb17971c0 | |||
| ec11d77e89 | |||
| 2d29741691 | |||
| cc77044df1 | |||
| dba07aacf7 | |||
| bd76b897de | |||
| 1106f27700 | |||
| 00fc3b5bcb | |||
| a0fe9e1645 | |||
| 5f460e8912 | |||
| 6fed74ba74 | |||
| 5d1c87b26f | |||
| fd77ad7fc0 | |||
| bc5617407d | |||
| 5ae06f4d37 | |||
| 233031cdcb | |||
| ebf5172cd0 | |||
| 0b72e4d17d | |||
| af70f87bc1 | |||
| 53fd703911 | |||
| c0d28bcb17 | |||
| edf8344983 | |||
| a502204e26 | |||
| 7162bc6fe6 | |||
| f270a485b5 | |||
| df66816bee | |||
| 5c7ab407d7 | |||
| c83d96cc91 | |||
| 427fd206ca | |||
| 1a4010cbba |
34
BUILDING.md
Normal file
34
BUILDING.md
Normal file
@ -0,0 +1,34 @@
|
|||||||
|
Building and installing
|
||||||
|
-----------------------
|
||||||
|
|
||||||
|
### Prerequisites
|
||||||
|
|
||||||
|
- GCC compatible compiler
|
||||||
|
- make
|
||||||
|
- POSIX.1 support
|
||||||
|
* getopt()
|
||||||
|
* strcasecmp()
|
||||||
|
* tcflush()
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
### Compiling
|
||||||
|
|
||||||
|
cc -std=gnu99 -Wpedantic -Wall -Wextra -Wno-gnu-zero-variadic-macro-arguments -O2 -o mcrcon mcrcon.c
|
||||||
|
|
||||||
|
>[!NOTE]
|
||||||
|
>If you are compiling on Windows remember to link with winsock by adding `-lws2_32` to your compiler command line.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
Or you can run **make**
|
||||||
|
|
||||||
|
make - compiles mcrcon
|
||||||
|
make install - installs compiled binaries and manpage to the system
|
||||||
|
make uninstall - removes binaries and manpage from the system
|
||||||
|
|
||||||
|
file install locations:
|
||||||
|
/usr/local/bin/mcrcon
|
||||||
|
/usr/local/share/man/man1/mcrcon.1
|
||||||
|
|
||||||
|
Makefile **install** and **uninstall** rules are not available on Windows.
|
||||||
14
CHANGELOG.md
14
CHANGELOG.md
@ -1,13 +1,21 @@
|
|||||||
#### Version history:
|
#### Version history:
|
||||||
|
|
||||||
###### 0.7.3
|
###### 0.8.0
|
||||||
- Add support to Valve style rcon authentication
|
- Implement support for multipacket responses
|
||||||
|
- Add support for Valve style rcon authentication
|
||||||
|
- Add experimental UTF-8 support for Windows
|
||||||
- Change maximum packet size to correct value (4096 -> 4106)
|
- Change maximum packet size to correct value (4096 -> 4106)
|
||||||
|
- Attempt to add missing newlines in bugged Minecraft servers
|
||||||
|
* Implemented for responses to the 'help' command and unknown commands
|
||||||
|
- Print auth failed message to stderr instead of stdout
|
||||||
|
- Fail immediately if received packet size is out of spec
|
||||||
|
- Return proper exit code from run_terminal_mode()
|
||||||
|
- Add error messages to rcon_command() function
|
||||||
|
|
||||||
###### 0.7.2
|
###### 0.7.2
|
||||||
- Quit gracefully when Ctrl-D or Ctrl+C is pressed
|
- Quit gracefully when Ctrl-D or Ctrl+C is pressed
|
||||||
- Remove "exit" and "quit" as quitting commands
|
- Remove "exit" and "quit" as quitting commands
|
||||||
* these are actual rcon commands on some servers
|
* These are actual rcon commands on some servers
|
||||||
- Suppress compiler warning (strncpy)
|
- Suppress compiler warning (strncpy)
|
||||||
- Fix erroneous string length in packet building function
|
- Fix erroneous string length in packet building function
|
||||||
- Fix typo in ANSI escape sequence for LCYAN
|
- Fix typo in ANSI escape sequence for LCYAN
|
||||||
|
|||||||
22
INSTALL.md
22
INSTALL.md
@ -1,22 +0,0 @@
|
|||||||
Building and installing
|
|
||||||
-----------------------
|
|
||||||
|
|
||||||
Only dependency is C library and POSIX getopt support.
|
|
||||||
|
|
||||||
Compiling with GCC or CLANG:
|
|
||||||
|
|
||||||
cc -std=gnu99 -Wpedantic -Wall -Wextra -O2 -o mcrcon mcrcon.c
|
|
||||||
|
|
||||||
Note: on Windows remember to link with winsock by adding `-lws2_32` to your compiler command line.
|
|
||||||
|
|
||||||
Or you can just run "**make**":
|
|
||||||
|
|
||||||
make - compiles mcrcon
|
|
||||||
make install - installs compiled binaries and manpage to the system
|
|
||||||
make uninstall - removes binaries and manpage from the system
|
|
||||||
|
|
||||||
file install locations:
|
|
||||||
/usr/local/bin/mcrcon
|
|
||||||
/usr/local/share/man/man1/mcrcon.1
|
|
||||||
|
|
||||||
Makefile "**install**" and "**uninstall**" rules are disabled on windows.
|
|
||||||
8
Makefile
8
Makefile
@ -13,9 +13,9 @@ INSTALL = install
|
|||||||
LINKER =
|
LINKER =
|
||||||
RM = rm -v -f
|
RM = rm -v -f
|
||||||
|
|
||||||
CC = gcc
|
CC ?= gcc
|
||||||
CFLAGS = -std=gnu99 -Wall -Wextra -Wpedantic -O2
|
CFLAGS = -std=gnu99 -Wall -Wextra -Wpedantic -Wno-gnu-zero-variadic-macro-arguments -O2
|
||||||
EXTRAFLAGS ?= -fstack-protector-all
|
EXTRAFLAGS ?= -fstack-protector-strong -D_FORTIFY_SOURCE=2 -fPIE -pie -Wl,-z,relro -Wl,-z,now -fno-common
|
||||||
|
|
||||||
ifeq ($(OS), Windows_NT)
|
ifeq ($(OS), Windows_NT)
|
||||||
LINKER = -lws2_32
|
LINKER = -lws2_32
|
||||||
@ -44,4 +44,4 @@ endif
|
|||||||
|
|
||||||
.PHONY: clean
|
.PHONY: clean
|
||||||
clean:
|
clean:
|
||||||
$(RM) $(EXENAME)
|
$(RM) $(EXENAME) $(EXENAME).exe
|
||||||
|
|||||||
44
README.md
44
README.md
@ -1,6 +1,9 @@
|
|||||||
# mcrcon
|
# mcrcon
|
||||||
|
|
||||||
mcrcon is console based Minecraft [rcon](https://developer.valvesoftware.com/wiki/Source_RCON_Protocol) client for remote administration and server maintenance scripts.
|
mcrcon is a command-line [rcon](https://developer.valvesoftware.com/wiki/Source_RCON_Protocol) client intended for remote server administration and maintenance automation.
|
||||||
|
|
||||||
|
Though originally developed for Minecraft servers, it also works with a variety of other servers using the Valve or Minecraft-style rcon protocol.
|
||||||
|
|
||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
@ -29,7 +32,7 @@ make
|
|||||||
# install is optional
|
# install is optional
|
||||||
sudo make install
|
sudo make install
|
||||||
```
|
```
|
||||||
Check [INSTALL.md](INSTALL.md) for more details.
|
_Check [BUILDING.md](BUILDING.md) for more details._
|
||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
@ -68,17 +71,15 @@ Send three commands ("say", "save-all", "stop") and wait five seconds between th
|
|||||||
```sh
|
```sh
|
||||||
mcrcon -H my.minecraft.server -p password -w 5 "say Server is restarting!" save-all stop
|
mcrcon -H my.minecraft.server -p password -w 5 "say Server is restarting!" save-all stop
|
||||||
```
|
```
|
||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
##### How to enable rcon on a Minecraft Server
|
> [!TIP]
|
||||||
|
>Enable RCON on Minecraft server by adding following lines to [```server.properties```](https://minecraft.gamepedia.com/Server.properties) configuration file.
|
||||||
Enable rcon by adding following lines to [```server.properties```](https://minecraft.gamepedia.com/Server.properties) configuration file.
|
>```
|
||||||
```
|
>enable-rcon=true
|
||||||
enable-rcon=true
|
>rcon.port=25575
|
||||||
rcon.port=25575
|
>rcon.password=your_rcon_pasword
|
||||||
rcon.password=your_rcon_pasword
|
>```
|
||||||
```
|
|
||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
@ -88,16 +89,17 @@ rcon.password=your_rcon_pasword
|
|||||||
* MAIL: tiiffi+mcrcon at gmail
|
* MAIL: tiiffi+mcrcon at gmail
|
||||||
* ISSUES: https://github.com/Tiiffi/mcrcon/issues/
|
* ISSUES: https://github.com/Tiiffi/mcrcon/issues/
|
||||||
|
|
||||||
When reporting issues, please provide the following information:
|
> [!TIP]
|
||||||
|
>When reporting issues, please provide the following information:
|
||||||
- Version of mcrcon: Please specify the precise version number
|
>
|
||||||
- Game: Indicate the specific game server you're using (e.g., Minecraft, Valve Source Engine game, ARK, ...)
|
>- Version of mcrcon: Please specify the precise version number
|
||||||
- Server version: Provide the exact version of the game server
|
>- Game: Indicate the specific game server you're using (e.g., Minecraft, Valve Source Engine game, ARK, ...)
|
||||||
- Mods and Extensions: List all mods and extensions used, including their versions
|
>- Server version: Provide the exact version of the game server
|
||||||
- Issue Description: Clearly describe the problem you're encountering and the expected behavior.
|
>- Mods and Extensions: List all mods and extensions used, including their versions
|
||||||
- Steps to reproduce
|
>- Issue Description: Clearly describe the problem you're encountering and the expected behavior.
|
||||||
|
>- Steps to reproduce
|
||||||
If you're tech-savvy, consider providing a packet capture file (PCAP). Remember to use a fake password.
|
>
|
||||||
|
>If you're tech-savvy, consider providing a packet capture file (PCAP). Remember to use a fake password.
|
||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
|
|||||||
4
mcrcon.1
4
mcrcon.1
@ -1,7 +1,7 @@
|
|||||||
.\" Process this file with
|
.\" Process this file with
|
||||||
.\" groff -man -Tascii mcrcon.1
|
.\" groff -man -Tascii mcrcon.1
|
||||||
.\"
|
.\"
|
||||||
.TH MCRCON 1 "November 2024" "Version 0.7.3"
|
.TH MCRCON 1 "December 2024" "Version 0.8.0"
|
||||||
.SH NAME
|
.SH NAME
|
||||||
mcrcon \- send rcon commands to a Minecraft server
|
mcrcon \- send rcon commands to a Minecraft server
|
||||||
.SH SYNOPSIS
|
.SH SYNOPSIS
|
||||||
@ -11,7 +11,7 @@ options
|
|||||||
commands
|
commands
|
||||||
.B ]
|
.B ]
|
||||||
.SH DESCRIPTION
|
.SH DESCRIPTION
|
||||||
mcrcon is Minecraft rcon client for remote administration and server maintenance scripts.
|
mcrcon is a command-line rcon client intended for remote server administration and maintenance automation.
|
||||||
.SH OPTIONS
|
.SH OPTIONS
|
||||||
.IP -H
|
.IP -H
|
||||||
Server address (default: localhost)
|
Server address (default: localhost)
|
||||||
|
|||||||
485
mcrcon.c
485
mcrcon.c
@ -1,5 +1,5 @@
|
|||||||
/*
|
/*
|
||||||
* Copyright (c) 2012-2021, Tiiffi <tiiffi at gmail>
|
* Copyright (c) 2012-2024, Tiiffi <tiiffi at gmail>
|
||||||
*
|
*
|
||||||
* This software is provided 'as-is', without any express or implied
|
* This software is provided 'as-is', without any express or implied
|
||||||
* warranty. In no event will the authors be held liable for any damages
|
* warranty. In no event will the authors be held liable for any damages
|
||||||
@ -24,30 +24,27 @@
|
|||||||
#include <stdio.h>
|
#include <stdio.h>
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
#include <stdint.h>
|
#include <stdint.h>
|
||||||
|
#include <stdbool.h>
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
#include <strings.h>
|
#include <strings.h>
|
||||||
#include <signal.h>
|
#include <signal.h>
|
||||||
#include <errno.h>
|
#include <errno.h>
|
||||||
#include <unistd.h>
|
#include <unistd.h>
|
||||||
#include <limits.h>
|
|
||||||
|
|
||||||
#ifdef _WIN32
|
#ifdef _WIN32
|
||||||
// for name resolving on windows
|
|
||||||
// enable this if you get compiler whine about getaddrinfo() on windows
|
|
||||||
//#define _WIN32_WINNT 0x0501
|
|
||||||
|
|
||||||
#include <ws2tcpip.h>
|
|
||||||
#include <winsock2.h>
|
#include <winsock2.h>
|
||||||
#include <windows.h>
|
#include <windows.h>
|
||||||
|
#include <ws2tcpip.h>
|
||||||
|
#include <fcntl.h>
|
||||||
|
#include <wchar.h>
|
||||||
#else
|
#else
|
||||||
#include <sys/types.h>
|
|
||||||
#include <sys/socket.h>
|
#include <sys/socket.h>
|
||||||
#include <netinet/in.h>
|
#include <sys/select.h>
|
||||||
#include <arpa/inet.h>
|
|
||||||
#include <netdb.h>
|
#include <netdb.h>
|
||||||
|
#include <termios.h>
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#define VERSION "0.7.3"
|
#define VERSION "0.8.0"
|
||||||
#define IN_NAME "mcrcon"
|
#define IN_NAME "mcrcon"
|
||||||
#define VER_STR IN_NAME" "VERSION" (built: "__DATE__" "__TIME__")"
|
#define VER_STR IN_NAME" "VERSION" (built: "__DATE__" "__TIME__")"
|
||||||
|
|
||||||
@ -57,20 +54,24 @@
|
|||||||
#define RCON_AUTH_RESPONSE 2
|
#define RCON_AUTH_RESPONSE 2
|
||||||
#define RCON_PID 0xBADC0DE
|
#define RCON_PID 0xBADC0DE
|
||||||
|
|
||||||
#define DATA_BUFFSIZE 4096
|
#define MAX_COMMAND_LENGTH 4096
|
||||||
#define MAX_PACKET_SIZE 4106
|
#define DATA_BUFFSIZE MAX_COMMAND_LENGTH + 2 // plus two null terminators
|
||||||
#define MIN_PACKET_SIZE 10
|
#define MAX_PACKET_SIZE 4106 // id (4) + cmd (4) + DATA_BUFFSIZE
|
||||||
|
#define MIN_PACKET_SIZE 10 // id (4) + cmd (4) + two empty strings (2)
|
||||||
|
|
||||||
|
#define MAX_WAIT_TIME 600
|
||||||
|
|
||||||
|
#define log_error(fmt, ...) fprintf(stderr, fmt, ##__VA_ARGS__);
|
||||||
|
|
||||||
// rcon packet structure
|
// rcon packet structure
|
||||||
typedef struct _rc_packet {
|
typedef struct {
|
||||||
int32_t size;
|
int32_t size;
|
||||||
int32_t id;
|
int32_t id;
|
||||||
int32_t cmd;
|
int32_t cmd;
|
||||||
char data[DATA_BUFFSIZE];
|
uint8_t data[DATA_BUFFSIZE];
|
||||||
// ignoring string2 for now
|
// ignoring string2 for now
|
||||||
} rc_packet;
|
} rc_packet;
|
||||||
|
|
||||||
|
|
||||||
// ===================================
|
// ===================================
|
||||||
// FUNCTION DEFINITIONS
|
// FUNCTION DEFINITIONS
|
||||||
// ===================================
|
// ===================================
|
||||||
@ -81,10 +82,8 @@ void net_init_WSA(void);
|
|||||||
#endif
|
#endif
|
||||||
void net_close(int sd);
|
void net_close(int sd);
|
||||||
int net_connect(const char *host, const char *port);
|
int net_connect(const char *host, const char *port);
|
||||||
int net_send(int sd, const uint8_t *buffer, size_t size);
|
bool net_send_packet(int sd, rc_packet *packet);
|
||||||
int net_send_packet(int sd, rc_packet *packet);
|
|
||||||
rc_packet* net_recv_packet(int sd);
|
rc_packet* net_recv_packet(int sd);
|
||||||
int net_clean_incoming(int sd, int size);
|
|
||||||
|
|
||||||
// Misc stuff
|
// Misc stuff
|
||||||
void usage(void);
|
void usage(void);
|
||||||
@ -96,25 +95,30 @@ int run_terminal_mode(int sock);
|
|||||||
int run_commands(int argc, char *argv[]);
|
int run_commands(int argc, char *argv[]);
|
||||||
|
|
||||||
// Rcon protocol related functions
|
// Rcon protocol related functions
|
||||||
rc_packet* packet_build(int id, int cmd, char *s1);
|
rc_packet* packet_build(int id, int cmd, char s[static 1]);
|
||||||
void packet_print(rc_packet *packet);
|
void packet_print(rc_packet *packet);
|
||||||
int rcon_auth(int sock, char *passwd);
|
bool rcon_auth(int sock, char *passwd);
|
||||||
int rcon_command(int sock, char *command);
|
int rcon_command(int sock, char *command);
|
||||||
|
|
||||||
|
|
||||||
// =============================================
|
// =============================================
|
||||||
// GLOBAL VARIABLES
|
// GLOBAL VARIABLES
|
||||||
// =============================================
|
// =============================================
|
||||||
static int global_raw_output = 0;
|
static int flag_raw_output = 0;
|
||||||
static int global_silent_mode = 0;
|
static int flag_silent_mode = 0;
|
||||||
static int global_disable_colors = 0;
|
static int flag_disable_colors = 0;
|
||||||
|
static int flag_wait_seconds = 0;
|
||||||
static int global_connection_alive = 1;
|
static int global_connection_alive = 1;
|
||||||
|
static bool global_valve_protocol = false;
|
||||||
|
static bool global_minecraft_newline_fix = false;
|
||||||
static int global_rsock;
|
static int global_rsock;
|
||||||
static int global_wait_seconds = 0;
|
|
||||||
|
|
||||||
#ifdef _WIN32
|
#ifdef _WIN32
|
||||||
// console coloring on windows
|
// console coloring on windows
|
||||||
HANDLE console_handle;
|
HANDLE console_handle;
|
||||||
|
|
||||||
|
// console code pages
|
||||||
|
UINT old_output_codepage;
|
||||||
|
UINT old_input_codepage;
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
// safety stuff (windows is still misbehaving)
|
// safety stuff (windows is still misbehaving)
|
||||||
@ -122,39 +126,44 @@ void exit_proc(void)
|
|||||||
{
|
{
|
||||||
if (global_rsock != -1)
|
if (global_rsock != -1)
|
||||||
net_close(global_rsock);
|
net_close(global_rsock);
|
||||||
|
|
||||||
|
#ifdef _WIN32
|
||||||
|
// Restore previous code pages
|
||||||
|
SetConsoleOutputCP(old_output_codepage);
|
||||||
|
SetConsoleCP(old_input_codepage);
|
||||||
|
|
||||||
|
// Set back to binary mode
|
||||||
|
_setmode(_fileno(stdin), _O_BINARY);
|
||||||
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check windows & linux behaviour !!!
|
// TODO: check exact windows and linux behaviour
|
||||||
void sighandler(int sig)
|
void sighandler(int sig)
|
||||||
{
|
{
|
||||||
if (sig == SIGINT)
|
if (sig == SIGINT)
|
||||||
putchar('\n');
|
putchar('\n');
|
||||||
|
|
||||||
global_connection_alive = 0;
|
global_connection_alive = 0;
|
||||||
#ifndef _WIN32
|
|
||||||
exit(EXIT_SUCCESS);
|
exit(EXIT_SUCCESS);
|
||||||
#endif
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#define MAX_WAIT_TIME 600
|
|
||||||
|
|
||||||
unsigned int mcrcon_parse_seconds(char *str)
|
unsigned int mcrcon_parse_seconds(char *str)
|
||||||
{
|
{
|
||||||
char *end;
|
char *end;
|
||||||
long result = strtol(str, &end, 10);
|
long result = strtol(str, &end, 10);
|
||||||
|
|
||||||
if (errno != 0) {
|
if (errno != 0) {
|
||||||
fprintf(stderr, "-w invalid value.\nerror %d: %s\n", errno, strerror(errno));
|
log_error("-w invalid value.\nerror %d: %s\n", errno, strerror(errno));
|
||||||
exit(EXIT_FAILURE);
|
exit(EXIT_FAILURE);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (end == str) {
|
if (end == str) {
|
||||||
fprintf(stderr, "-w invalid value (not a number?)\n");
|
log_error("-w invalid value (not a number?)\n");
|
||||||
exit(EXIT_FAILURE);
|
exit(EXIT_FAILURE);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (result <= 0 || result > MAX_WAIT_TIME) {
|
if (result <= 0 || result > MAX_WAIT_TIME) {
|
||||||
fprintf(stderr, "-w value out of range.\nAcceptable value is 1 - %d (seconds).\n", MAX_WAIT_TIME);
|
log_error("-w value out of range.\nAcceptable value is 1 - %d (seconds).\n", MAX_WAIT_TIME);
|
||||||
exit(EXIT_FAILURE);
|
exit(EXIT_FAILURE);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -172,10 +181,6 @@ int main(int argc, char *argv[])
|
|||||||
if (!port) port = "25575";
|
if (!port) port = "25575";
|
||||||
if (!host) host = "localhost";
|
if (!host) host = "localhost";
|
||||||
|
|
||||||
// disable output buffering (https://github.com/Tiiffi/mcrcon/pull/39)
|
|
||||||
setvbuf(stdout, NULL, _IONBF, 0);
|
|
||||||
setvbuf(stderr, NULL, _IONBF, 0);
|
|
||||||
|
|
||||||
if(argc < 1 && pass == NULL) usage();
|
if(argc < 1 && pass == NULL) usage();
|
||||||
|
|
||||||
// default getopt error handler enabled
|
// default getopt error handler enabled
|
||||||
@ -187,17 +192,17 @@ int main(int argc, char *argv[])
|
|||||||
case 'H': host = optarg; break;
|
case 'H': host = optarg; break;
|
||||||
case 'P': port = optarg; break;
|
case 'P': port = optarg; break;
|
||||||
case 'p': pass = optarg; break;
|
case 'p': pass = optarg; break;
|
||||||
case 'c': global_disable_colors = 1; break;
|
case 'c': flag_disable_colors = 1; break;
|
||||||
case 's': global_silent_mode = 1; break;
|
case 's': flag_silent_mode = 1; break;
|
||||||
case 'i': /* reserved for interp mode */break;
|
case 'i': /* reserved for interp mode */ break;
|
||||||
case 't': terminal_mode = 1; break;
|
case 't': terminal_mode = 1; break;
|
||||||
case 'r': global_raw_output = 1; break;
|
case 'r': flag_raw_output = 1; break;
|
||||||
case 'w':
|
case 'w':
|
||||||
global_wait_seconds = mcrcon_parse_seconds(optarg);
|
flag_wait_seconds = mcrcon_parse_seconds(optarg);
|
||||||
break;
|
break;
|
||||||
|
|
||||||
case 'v':
|
case 'v':
|
||||||
puts(VER_STR" - https://github.com/Tiiffi/mcrcon");
|
puts(VER_STR);
|
||||||
puts("Bug reports:\n\ttiiffi+mcrcon at gmail\n\thttps://github.com/Tiiffi/mcrcon/issues/");
|
puts("Bug reports:\n\ttiiffi+mcrcon at gmail\n\thttps://github.com/Tiiffi/mcrcon/issues/");
|
||||||
exit(EXIT_SUCCESS);
|
exit(EXIT_SUCCESS);
|
||||||
|
|
||||||
@ -214,8 +219,9 @@ int main(int argc, char *argv[])
|
|||||||
exit(EXIT_FAILURE);
|
exit(EXIT_FAILURE);
|
||||||
}
|
}
|
||||||
|
|
||||||
if(optind == argc && terminal_mode == 0)
|
if(optind == argc && terminal_mode == 0) {
|
||||||
terminal_mode = 1;
|
terminal_mode = 1;
|
||||||
|
}
|
||||||
|
|
||||||
// safety features to prevent "IO: Connection reset" bug on the server side
|
// safety features to prevent "IO: Connection reset" bug on the server side
|
||||||
atexit(&exit_proc);
|
atexit(&exit_proc);
|
||||||
@ -228,6 +234,16 @@ int main(int argc, char *argv[])
|
|||||||
console_handle = GetStdHandle(STD_OUTPUT_HANDLE);
|
console_handle = GetStdHandle(STD_OUTPUT_HANDLE);
|
||||||
if (console_handle == INVALID_HANDLE_VALUE)
|
if (console_handle == INVALID_HANDLE_VALUE)
|
||||||
console_handle = NULL;
|
console_handle = NULL;
|
||||||
|
|
||||||
|
// Set the output and input code pages to utf-8
|
||||||
|
old_output_codepage = GetConsoleOutputCP();
|
||||||
|
old_input_codepage = GetConsoleCP();
|
||||||
|
|
||||||
|
SetConsoleOutputCP(CP_UTF8);
|
||||||
|
SetConsoleCP(CP_UTF8);
|
||||||
|
|
||||||
|
// Set the file translation mode to UTF16
|
||||||
|
_setmode(_fileno(stdin), _O_U16TEXT);
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
// open socket
|
// open socket
|
||||||
@ -237,13 +253,11 @@ int main(int argc, char *argv[])
|
|||||||
|
|
||||||
// auth & commands
|
// auth & commands
|
||||||
if (rcon_auth(global_rsock, pass)) {
|
if (rcon_auth(global_rsock, pass)) {
|
||||||
if (terminal_mode)
|
if (terminal_mode) exit_code = run_terminal_mode(global_rsock);
|
||||||
run_terminal_mode(global_rsock);
|
else exit_code = run_commands(argc, argv);
|
||||||
else
|
|
||||||
exit_code = run_commands(argc, argv);
|
|
||||||
}
|
}
|
||||||
else { // auth failed
|
else { // auth failed
|
||||||
fprintf(stdout, "Authentication failed!\n");
|
log_error("Authentication failed!\n");
|
||||||
exit_code = EXIT_FAILURE;
|
exit_code = EXIT_FAILURE;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -298,7 +312,7 @@ void net_init_WSA(void)
|
|||||||
|
|
||||||
int err = WSAStartup(version, &wsadata);
|
int err = WSAStartup(version, &wsadata);
|
||||||
if (err != 0) {
|
if (err != 0) {
|
||||||
fprintf(stderr, "WSAStartup failed. Error: %d.\n", err);
|
log_error("WSAStartup failed. Error: %d.\n", err);
|
||||||
exit(EXIT_FAILURE);
|
exit(EXIT_FAILURE);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -315,7 +329,6 @@ void net_close(int sd)
|
|||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
// 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)
|
||||||
@ -336,11 +349,11 @@ int net_connect(const char *host, const char *port)
|
|||||||
|
|
||||||
int ret = getaddrinfo(host, port, &hints, &server_info);
|
int ret = getaddrinfo(host, port, &hints, &server_info);
|
||||||
if (ret != 0) {
|
if (ret != 0) {
|
||||||
fprintf(stderr, "Name resolution failed.\n");
|
log_error("Name resolution failed.\n");
|
||||||
#ifdef _WIN32
|
#ifdef _WIN32
|
||||||
fprintf(stderr, "Error %d: %s", ret, gai_strerror(ret));
|
log_error("Error %d: %s", ret, gai_strerror(ret));
|
||||||
#else
|
#else
|
||||||
fprintf(stderr, "Error %d: %s\n", ret, gai_strerror(ret));
|
log_error("Error %d: %s\n", ret, gai_strerror(ret));
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
exit(EXIT_FAILURE);
|
exit(EXIT_FAILURE);
|
||||||
@ -364,9 +377,9 @@ int net_connect(const char *host, const char *port)
|
|||||||
|
|
||||||
if (p == NULL) {
|
if (p == NULL) {
|
||||||
/* TODO (Tiiffi): Check why windows does not report errors */
|
/* TODO (Tiiffi): Check why windows does not report errors */
|
||||||
fprintf(stderr, "Connection failed.\n");
|
log_error("Connection failed.\n");
|
||||||
#ifndef _WIN32
|
#ifndef _WIN32
|
||||||
fprintf(stderr, "Error %d: %s\n", errno, strerror(errno));
|
log_error("Error %d: %s\n", errno, strerror(errno));
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
freeaddrinfo(server_info);
|
freeaddrinfo(server_info);
|
||||||
@ -377,82 +390,62 @@ int net_connect(const char *host, const char *port)
|
|||||||
return sd;
|
return sd;
|
||||||
}
|
}
|
||||||
|
|
||||||
int net_send(int sd, const uint8_t *buff, size_t size)
|
bool net_send_packet(int sd, rc_packet *packet)
|
||||||
{
|
{
|
||||||
size_t sent = 0;
|
size_t sent = 0;
|
||||||
|
size_t size = packet->size + sizeof(int32_t);
|
||||||
size_t left = size;
|
size_t left = size;
|
||||||
|
|
||||||
while (sent < size) {
|
char *p = (char *) packet;
|
||||||
int result = send(sd, (const char *) buff + sent, left, 0);
|
|
||||||
|
|
||||||
if (result == -1)
|
while (sent < size) {
|
||||||
return -1;
|
ssize_t result = send(sd, p + sent, left, 0);
|
||||||
|
|
||||||
|
if (result == -1) return false;
|
||||||
|
|
||||||
sent += result;
|
sent += result;
|
||||||
left -= sent;
|
left -= sent;
|
||||||
}
|
}
|
||||||
|
|
||||||
return 0;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
int net_send_packet(int sd, rc_packet *packet)
|
|
||||||
{
|
|
||||||
int len;
|
|
||||||
int total = 0; // bytes we've sent
|
|
||||||
int bytesleft; // bytes left to send
|
|
||||||
int ret = -1;
|
|
||||||
|
|
||||||
bytesleft = len = packet->size + sizeof(int32_t);
|
|
||||||
|
|
||||||
while (total < len) {
|
|
||||||
ret = send(sd, (char *) packet + total, bytesleft, 0);
|
|
||||||
if(ret == -1) break;
|
|
||||||
total += ret;
|
|
||||||
bytesleft -= ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
return ret == -1 ? -1 : 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
// TODO: Fix the wonky behaviour when server quit/exit/stop/close
|
||||||
|
// command is issued. Client should close gracefully without errors.
|
||||||
rc_packet *net_recv_packet(int sd)
|
rc_packet *net_recv_packet(int sd)
|
||||||
{
|
{
|
||||||
int32_t psize;
|
int32_t psize;
|
||||||
static rc_packet packet = {0, 0, 0, { 0x00 }};
|
static rc_packet packet = {0};
|
||||||
|
|
||||||
// packet.size = packet.id = packet.cmd = 0;
|
ssize_t ret = recv(sd, (char *) &psize, sizeof(psize), 0);
|
||||||
|
|
||||||
int ret = recv(sd, (char *) &psize, sizeof(psize), 0);
|
|
||||||
|
|
||||||
if (ret == 0) {
|
if (ret == 0) {
|
||||||
fprintf(stderr, "Connection lost.\n");
|
log_error("Connection lost.\n");
|
||||||
global_connection_alive = 0;
|
global_connection_alive = 0;
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (ret != sizeof(psize)) {
|
if (ret != sizeof(psize)) {
|
||||||
fprintf(stderr, "Error: recv() failed. Invalid packet size (%d).\n", ret);
|
log_error("Error: recv() failed.\n");
|
||||||
global_connection_alive = 0;
|
global_connection_alive = 0;
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
// NOTE(Tiiffi): This should fail if size is out of spec!
|
|
||||||
if (psize < MIN_PACKET_SIZE || psize > MAX_PACKET_SIZE) {
|
if (psize < MIN_PACKET_SIZE || psize > MAX_PACKET_SIZE) {
|
||||||
fprintf(stderr, "Warning: invalid packet size (%d). Must over 10 and less than %d.\n", psize, MAX_PACKET_SIZE);
|
log_error("Error: Invalid packet size (%d).\n", psize);
|
||||||
|
global_connection_alive = 0;
|
||||||
// WARNING(Tiiffi): This is probably not the way to go. Probably should just fail and exit.
|
|
||||||
if(psize > MAX_PACKET_SIZE || psize < 0) psize = MAX_PACKET_SIZE;
|
|
||||||
net_clean_incoming(sd, psize);
|
|
||||||
|
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
packet.size = psize;
|
packet.size = psize;
|
||||||
|
char *p = (char *) &packet;
|
||||||
|
|
||||||
int received = 0;
|
int received = 0;
|
||||||
while (received < psize) {
|
while (received < psize) {
|
||||||
ret = recv(sd, (char *) &packet + sizeof(int32_t) + received, psize - received, 0);
|
ret = recv(sd, p + sizeof(int32_t) + received, psize - received, 0);
|
||||||
if (ret == 0) {
|
if (ret == 0) {
|
||||||
fprintf(stderr, "Connection lost.\n");
|
log_error("Connection lost.\n");
|
||||||
global_connection_alive = 0;
|
global_connection_alive = 0;
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
@ -463,19 +456,6 @@ rc_packet *net_recv_packet(int sd)
|
|||||||
return &packet;
|
return &packet;
|
||||||
}
|
}
|
||||||
|
|
||||||
int net_clean_incoming(int sd, int size)
|
|
||||||
{
|
|
||||||
char tmp[size];
|
|
||||||
int ret = recv(sd, tmp, size, 0);
|
|
||||||
|
|
||||||
if(ret == 0) {
|
|
||||||
fprintf(stderr, "Connection lost.\n");
|
|
||||||
global_connection_alive = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
void print_color(int color)
|
void print_color(int color)
|
||||||
{
|
{
|
||||||
// sh compatible color array
|
// sh compatible color array
|
||||||
@ -500,8 +480,9 @@ void print_color(int color)
|
|||||||
"\033[4m" /* 16 UNDERLINE 0x6e */
|
"\033[4m" /* 16 UNDERLINE 0x6e */
|
||||||
};
|
};
|
||||||
|
|
||||||
/* 0x72: 'r' */
|
if (color == 0 || color == 0x72) {
|
||||||
if (color == 0 || color == 0x72) fputs("\033[0m", stdout); /* CANCEL COLOR */
|
fputs("\033[0m", stdout); // cancel color
|
||||||
|
}
|
||||||
else
|
else
|
||||||
#endif
|
#endif
|
||||||
{
|
{
|
||||||
@ -509,7 +490,7 @@ void print_color(int color)
|
|||||||
else if (color >= 0x30 && color <= 0x39)
|
else if (color >= 0x30 && color <= 0x39)
|
||||||
color -= 0x30;
|
color -= 0x30;
|
||||||
else if (color == 0x6e)
|
else if (color == 0x6e)
|
||||||
color = 16; /* 0x6e: 'n' */
|
color = 16;
|
||||||
else return;
|
else return;
|
||||||
|
|
||||||
#ifndef _WIN32
|
#ifndef _WIN32
|
||||||
@ -523,87 +504,106 @@ 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 (global_raw_output == 1) {
|
uint8_t *data = packet->data;
|
||||||
for (int i = 0; packet->data[i] != 0; ++i)
|
int i;
|
||||||
putchar(packet->data[i]);
|
|
||||||
|
|
||||||
|
if (flag_raw_output == 1) {
|
||||||
|
fputs((char *) data, stdout);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
int i;
|
// Newline fix for Minecraft
|
||||||
int def_color = 0;
|
if (global_valve_protocol == false) {
|
||||||
|
const char test[] = "Unknown or incomplete command, see below for error";
|
||||||
|
size_t test_size = sizeof test - 1;
|
||||||
|
if (strncmp((char *) data, test, test_size) == 0) {
|
||||||
|
fwrite(data, test_size, 1, stdout);
|
||||||
|
putchar('\n');
|
||||||
|
data = &data[test_size];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
int default_color = 0;
|
||||||
|
|
||||||
#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;
|
default_color = console_info.wAttributes + 0x30;
|
||||||
} else def_color = 0x37;
|
}
|
||||||
|
else default_color = 0x37;
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
// colors enabled so try to handle the bukkit colors for terminal
|
bool slash = false;
|
||||||
if (global_disable_colors == 0) {
|
bool colors_detected = false;
|
||||||
for (i = 0; (unsigned char) packet->data[i] != 0; ++i) {
|
|
||||||
if (packet->data[i] == 0x0A) print_color(def_color);
|
for (i = 0; data[i] != 0; ++i)
|
||||||
else if((unsigned char) packet->data[i] == 0xc2 && (unsigned char) packet->data[i+1] == 0xa7) {
|
{
|
||||||
print_color(packet->data[i+=2]);
|
if (data[i] == 0x0A) {
|
||||||
|
print_color(default_color);
|
||||||
|
}
|
||||||
|
else if(data[i] == 0xc2 && data[i + 1] == 0xa7) {
|
||||||
|
// Disable new line fixes if Bukkit colors are detected
|
||||||
|
colors_detected = true;
|
||||||
|
i += 2;
|
||||||
|
if (flag_disable_colors == 0) {
|
||||||
|
print_color(data[i]);
|
||||||
|
}
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
putchar(packet->data[i]);
|
|
||||||
}
|
// Add missing newlines
|
||||||
print_color(def_color); // cancel coloring
|
if (colors_detected == false && global_minecraft_newline_fix && data[i] == '/') {
|
||||||
}
|
slash ? putchar('\n') : (slash = true);
|
||||||
// strip colors
|
|
||||||
else {
|
|
||||||
for (i = 0; (unsigned char) packet->data[i] != 0; ++i) {
|
|
||||||
if ((unsigned char) packet->data[i] == 0xc2 && (unsigned char) packet->data[i+1] == 0xa7) {
|
|
||||||
i+=2;
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
putchar(packet->data[i]);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
putchar(data[i]);
|
||||||
}
|
}
|
||||||
|
print_color(default_color); // cancel coloring
|
||||||
|
|
||||||
// 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) putchar('\n');
|
if (data[i - 1] != '\n') {
|
||||||
|
putchar('\n');
|
||||||
|
}
|
||||||
|
|
||||||
|
fflush(stdout);
|
||||||
}
|
}
|
||||||
|
|
||||||
rc_packet *packet_build(int id, int cmd, char *s1)
|
rc_packet *packet_build(int id, int cmd, char s[static 1])
|
||||||
{
|
{
|
||||||
static rc_packet packet = {0, 0, 0, { 0x00 }};
|
static rc_packet packet = {0};
|
||||||
|
|
||||||
// NOTE(Tiiffi): Issue report states that outgoing payload has max size of 1460 bytes:
|
// NOTE(Tiiffi): Issue report states that maximum command packet size is 1460 bytes:
|
||||||
// https://github.com/Tiiffi/mcrcon/issues/45#issuecomment-1000940814
|
// https://github.com/Tiiffi/mcrcon/issues/45#issuecomment-1000940814
|
||||||
// https://mctools.readthedocs.io/en/master/rcon.html
|
// https://mctools.readthedocs.io/en/master/rcon.html
|
||||||
// Have to do some testing to confirm!
|
// Have to do some testing to confirm!
|
||||||
|
|
||||||
// size + id + cmd + s1 + s2 NULL terminator
|
int len = strlen(s);
|
||||||
int len = strlen(s1);
|
if (len > MAX_COMMAND_LENGTH) {
|
||||||
if (len >= DATA_BUFFSIZE) {
|
log_error("Warning: Command string too long (%d). Maximum allowed: %d.\n", len, MAX_COMMAND_LENGTH);
|
||||||
fprintf(stderr, "Warning: Command string too long (%d). Maximum allowed: %d.\n", len, DATA_BUFFSIZE - 1);
|
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
packet.size = sizeof packet.size * 2 + len + 2;
|
packet.size = sizeof packet.id + sizeof packet.cmd + len + 2;
|
||||||
packet.id = id;
|
packet.id = id;
|
||||||
packet.cmd = cmd;
|
packet.cmd = cmd;
|
||||||
strncpy(packet.data, s1, DATA_BUFFSIZE - 1);
|
if (packet.size > 0) memcpy(packet.data, s, len);
|
||||||
|
packet.data[len] = 0;
|
||||||
|
packet.data[len + 1] = 0;
|
||||||
|
|
||||||
return &packet;
|
return &packet;
|
||||||
}
|
}
|
||||||
|
|
||||||
int rcon_auth(int sock, char *passwd)
|
bool rcon_auth(int sock, char *passwd)
|
||||||
{
|
{
|
||||||
int ret;
|
|
||||||
|
|
||||||
rc_packet *packet = packet_build(RCON_PID, RCON_AUTHENTICATE, passwd);
|
rc_packet *packet = packet_build(RCON_PID, RCON_AUTHENTICATE, passwd);
|
||||||
if (packet == NULL)
|
if (packet == NULL)
|
||||||
return 0;
|
return 0;
|
||||||
|
|
||||||
ret = net_send_packet(sock, packet);
|
if (!net_send_packet(sock, packet)) {
|
||||||
if (!ret)
|
|
||||||
return 0; // send failed
|
return 0; // send failed
|
||||||
|
}
|
||||||
|
|
||||||
receive:
|
receive:
|
||||||
packet = net_recv_packet(sock);
|
packet = net_recv_packet(sock);
|
||||||
if (packet == NULL)
|
if (packet == NULL)
|
||||||
return 0;
|
return 0;
|
||||||
@ -612,35 +612,85 @@ receive:
|
|||||||
* so we have to check packet type and try again if necessary.
|
* so we have to check packet type and try again if necessary.
|
||||||
*/
|
*/
|
||||||
if (packet->cmd != RCON_AUTH_RESPONSE) {
|
if (packet->cmd != RCON_AUTH_RESPONSE) {
|
||||||
|
global_valve_protocol = true;
|
||||||
goto receive;
|
goto receive;
|
||||||
}
|
}
|
||||||
|
|
||||||
// return 1 if authentication OK
|
// return true if authentication OK
|
||||||
return packet->id == -1 ? 0 : 1;
|
return packet->id == -1 ? false : true;
|
||||||
}
|
}
|
||||||
|
|
||||||
int rcon_command(int sock, char *command)
|
int rcon_command(int sock, char *command)
|
||||||
{
|
{
|
||||||
|
if (global_valve_protocol == false && strcasecmp(command, "help") == 0)
|
||||||
|
global_minecraft_newline_fix = true;
|
||||||
|
|
||||||
rc_packet *packet = packet_build(RCON_PID, RCON_EXEC_COMMAND, command);
|
rc_packet *packet = packet_build(RCON_PID, RCON_EXEC_COMMAND, command);
|
||||||
if (packet == NULL) {
|
if (packet == NULL) {
|
||||||
global_connection_alive = 0;
|
log_error("Error: packet build() failed!\n");
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
net_send_packet(sock, packet);
|
if (!net_send_packet(sock, packet)) {
|
||||||
|
log_error("Error: net_send_packet() failed!\n");
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Workaround to handle valve multipacket responses
|
||||||
|
// This one does not require using select()
|
||||||
|
packet = packet_build(0xBADA55, 0xBADA55, "");
|
||||||
|
if (packet == NULL) {
|
||||||
|
log_error("Error: packet build() failed!\n");
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!net_send_packet(sock, packet)) {
|
||||||
|
log_error("Error: net_send_packet() failed!\n");
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
// initialize stuff for select()
|
||||||
|
fd_set read_fds;
|
||||||
|
FD_ZERO(&read_fds);
|
||||||
|
FD_SET(sock, &read_fds);
|
||||||
|
|
||||||
|
// Set 5 second timeout
|
||||||
|
struct timeval timeout = {0};
|
||||||
|
timeout.tv_sec = 5;
|
||||||
|
timeout.tv_usec = 0;
|
||||||
|
|
||||||
|
int incoming = 0;
|
||||||
|
|
||||||
|
do {
|
||||||
packet = net_recv_packet(sock);
|
packet = net_recv_packet(sock);
|
||||||
if (packet == NULL)
|
if (packet == NULL) {
|
||||||
|
log_error("Error: net_recv_packet() failed!\n");
|
||||||
return 0;
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
if (packet->id != RCON_PID)
|
// Check for packet id and multipacket guard id
|
||||||
|
if (packet->id != RCON_PID && packet->id != 0xBADA55) {
|
||||||
|
log_error("Error: invalid packet id!\n");
|
||||||
return 0;
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
if (!global_silent_mode) {
|
if (packet->id == 0xBADA55) break;
|
||||||
|
|
||||||
|
if (flag_silent_mode == false) {
|
||||||
if (packet->size > 10)
|
if (packet->size > 10)
|
||||||
packet_print(packet);
|
packet_print(packet);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
int result = select(sock + 1, &read_fds, NULL, NULL, &timeout);
|
||||||
|
if (result == -1) {
|
||||||
|
log_error("Error: select() failed!\n");
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
incoming = (result > 0 && FD_ISSET(sock, &read_fds));
|
||||||
|
}
|
||||||
|
while(incoming);
|
||||||
|
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -652,63 +702,121 @@ int run_commands(int argc, char *argv[])
|
|||||||
if (!rcon_command(global_rsock, argv[i]))
|
if (!rcon_command(global_rsock, argv[i]))
|
||||||
return EXIT_FAILURE;
|
return EXIT_FAILURE;
|
||||||
|
|
||||||
if (++i >= argc)
|
i++;
|
||||||
|
|
||||||
|
if (i >= argc)
|
||||||
return EXIT_SUCCESS;
|
return EXIT_SUCCESS;
|
||||||
|
|
||||||
if (global_wait_seconds > 0) {
|
if (flag_wait_seconds > 0) {
|
||||||
#ifdef _WIN32
|
#ifdef _WIN32
|
||||||
Sleep(global_wait_seconds * 1000);
|
Sleep(flag_wait_seconds * 1000);
|
||||||
#else
|
#else
|
||||||
sleep(global_wait_seconds);
|
sleep(flag_wait_seconds);
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// interactive terminal mode
|
// terminal mode
|
||||||
int run_terminal_mode(int sock)
|
int run_terminal_mode(int sock)
|
||||||
{
|
{
|
||||||
int ret = 0;
|
char command[MAX_COMMAND_LENGTH] = {0};
|
||||||
char command[DATA_BUFFSIZE] = {0};
|
|
||||||
|
|
||||||
puts("Logged in.\nType 'Q' or press Ctrl-D / Ctrl-C to disconnect.");
|
puts("Logged in. Press Ctrl-D or Ctrl-C to disconnect.");
|
||||||
|
|
||||||
while (global_connection_alive) {
|
while (global_connection_alive) {
|
||||||
putchar('>');
|
putchar('>');
|
||||||
|
fflush(stdout);
|
||||||
|
|
||||||
int len = get_line(command, DATA_BUFFSIZE);
|
int len = get_line(command, MAX_COMMAND_LENGTH);
|
||||||
if (len < 1) continue;
|
if (len < 1) continue;
|
||||||
|
|
||||||
if (strcasecmp(command, "Q") == 0)
|
if (len > 0 && global_connection_alive) {
|
||||||
break;
|
if (!rcon_command(sock, command)) {
|
||||||
|
return EXIT_FAILURE;
|
||||||
if (len > 0 && global_connection_alive)
|
}
|
||||||
ret += rcon_command(sock, command);
|
}
|
||||||
|
|
||||||
/* Special case for "stop" command to prevent server-side bug.
|
/* Special case for "stop" command to prevent server-side bug.
|
||||||
* https://bugs.mojang.com/browse/MC-154617
|
* https://bugs.mojang.com/browse/MC-154617
|
||||||
*
|
*
|
||||||
* NOTE: This is hacky workaround which should be handled better to
|
* NOTE: Not sure if this still a problem?!
|
||||||
* ensure compatibility with other servers using source RCON.
|
|
||||||
* NOTE: strcasecmp() is POSIX function.
|
|
||||||
*/
|
*/
|
||||||
if (strcasecmp(command, "stop") == 0) {
|
if (global_valve_protocol == false && strcasecmp(command, "stop") == 0) {
|
||||||
break;
|
// Timeout to before checking if connection was closed
|
||||||
|
#ifdef _WIN32
|
||||||
|
Sleep(2 * 1000);
|
||||||
|
#else
|
||||||
|
sleep(2);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
char tmp[1];
|
||||||
|
#ifdef _WIN32
|
||||||
|
// TODO: More Windows side testing!
|
||||||
|
int result = recv(sock, tmp, sizeof(tmp), MSG_PEEK | 0);
|
||||||
|
#else
|
||||||
|
int result = recv(sock, tmp, sizeof(tmp), MSG_PEEK | MSG_DONTWAIT);
|
||||||
|
#endif
|
||||||
|
if (result == 0) {
|
||||||
|
break; // Connection closed
|
||||||
|
}
|
||||||
|
// TODO: Check for return values and errors!
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
//command[0] = len = 0;
|
return EXIT_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
#ifdef _WIN32
|
||||||
|
char *utf8_getline(char *buf, int size, FILE *stream)
|
||||||
|
{
|
||||||
|
// Widechar fgets
|
||||||
|
wchar_t in[MAX_COMMAND_LENGTH] = {0};
|
||||||
|
|
||||||
|
wchar_t *result = fgetws(in, MAX_COMMAND_LENGTH, stream);
|
||||||
|
if (result == NULL) {
|
||||||
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
return ret;
|
// Calculates UTF-8 buffer size
|
||||||
|
int required_size = WideCharToMultiByte(CP_UTF8, 0, in, -1, NULL, 0, NULL, NULL);
|
||||||
|
if (size < required_size) {
|
||||||
|
// TODO: Proper error handling & reporting
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (WideCharToMultiByte(CP_UTF8, 0, in, -1, buf, size, NULL, NULL) == 0) {
|
||||||
|
// TODO: Proper error handling & reporting
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
return buf;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
void flush_input(void)
|
||||||
|
{
|
||||||
|
#ifdef _WIN32
|
||||||
|
FlushConsoleInputBuffer(console_handle);
|
||||||
|
#else
|
||||||
|
tcflush(STDIN_FILENO, TCIFLUSH);
|
||||||
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
// gets line from stdin and deals with rubbish left in the input buffer
|
// gets line from stdin and deals with rubbish left in the input buffer
|
||||||
int get_line(char *buffer, int bsize)
|
int get_line(char *buffer, int bsize)
|
||||||
{
|
{
|
||||||
|
#ifdef _WIN32
|
||||||
|
char *ret = utf8_getline(buffer, bsize, stdin);
|
||||||
|
#else
|
||||||
char *ret = fgets(buffer, bsize, stdin);
|
char *ret = fgets(buffer, bsize, stdin);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
flush_input();
|
||||||
|
|
||||||
if (ret == NULL) {
|
if (ret == NULL) {
|
||||||
if (ferror(stdin)) {
|
if (ferror(stdin)) {
|
||||||
fprintf(stderr, "Error %d: %s\n", errno, strerror(errno));
|
log_error("Error %d: %s\n", errno, strerror(errno));
|
||||||
exit(EXIT_FAILURE);
|
exit(EXIT_FAILURE);
|
||||||
}
|
}
|
||||||
putchar('\n');
|
putchar('\n');
|
||||||
@ -717,14 +825,7 @@ int get_line(char *buffer, int bsize)
|
|||||||
|
|
||||||
// remove unwanted characters from the buffer
|
// remove unwanted characters from the buffer
|
||||||
buffer[strcspn(buffer, "\r\n")] = '\0';
|
buffer[strcspn(buffer, "\r\n")] = '\0';
|
||||||
|
|
||||||
int len = strlen(buffer);
|
int len = strlen(buffer);
|
||||||
|
|
||||||
// clean input buffer if needed
|
|
||||||
if (len == bsize - 1) {
|
|
||||||
int ch;
|
|
||||||
while ((ch = getchar()) != '\n' && ch != EOF);
|
|
||||||
}
|
|
||||||
|
|
||||||
return len;
|
return len;
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user