Implement platform specific input buffer flushing

Minecraft servers have a longstanding bug that omits newlines
when sending messages via RCON. This patch manually inserts
newlines to address the issue.

Reference: https://bugs.mojang.com/browse/MC-7569

Fixes #1

.travis.yml

@@ -1,20 +0,0 @@
-language: c
-sudo: false
-dist: trusty
-cache:
-  ccache: true
-compiler:
-- gcc
-- clang
-env:
-  global:
-  - EXTRAFLAGS='-v'
-  - PREFIX="${HOME}"
-install:
-- 'true'
-script:
-- make
-- ./mcrcon -h
-- ./mcrcon -v
-- make install
-- make uninstall

BUILDING.md

@@ -0,0 +1,29 @@
+Building and installing
+-----------------------
+
+#### Requirements & dependencies:
+- GCC compatible compiler
+- POSIX getopt support from **<unistd.h>**
+
+---
+
+#### 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.

CHANGELOG.md

@@ -1,11 +1,23 @@
 #### Version history:
 
+###### 0.8.0
+ - 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)
+ - 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
  - Quit gracefully when Ctrl-D or Ctrl+C is pressed
  - 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)
- - fix erroneous string length in packet building function
+ - Fix erroneous string length in packet building function
  - Fix typo in ANSI escape sequence for LCYAN
  - Make stdout and stderr unbuffered
 

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 -Os -s -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.

LICENSE

@@ -1,4 +1,4 @@
-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
 warranty. In no event will the authors be held liable for any damages
@@ -18,4 +18,3 @@ freely, subject to the following restrictions:
 
   3. This notice may not be removed or altered from any source
   distribution.
-  

Makefile

@@ -4,7 +4,7 @@
 #   make
 #
 # Windows cross compile:
-#   i686-w64-mingw32-gcc -std=gnu99 -Wall -Wextra -Wpedantic -Os -s -o mcrcon.exe mcrcon.c -lws2_32
+#   x86_64-w64-mingw32-gcc -std=gnu99 -Wall -Wextra -Wpedantic -O2 -fstack-protector-all -o mcrcon.exe mcrcon.c -lws2_32
 
 EXENAME = mcrcon
 PREFIX ?= /usr/local
@@ -13,9 +13,9 @@ INSTALL = install
 LINKER =
 RM = rm -v -f
 
-CC = gcc
+CC ?= gcc
-CFLAGS = -std=gnu99 -Wall -Wextra -Wpedantic -Os -s
+CFLAGS = -std=gnu99 -Wall -Wextra -Wpedantic -Wno-gnu-zero-variadic-macro-arguments -O2
-EXTRAFLAGS ?= -fstack-protector-strong
+EXTRAFLAGS ?= -fstack-protector-strong -D_FORTIFY_SOURCE=2 -fPIE -pie -Wl,-z,relro -Wl,-z,now -fno-common
 
 ifeq ($(OS), Windows_NT)
 	LINKER = -lws2_32
@@ -23,11 +23,6 @@ ifeq ($(OS), Windows_NT)
 	RM = cmd /C del /F
 endif
 
-ifeq ($(shell uname), Darwin)
-	INSTALL = ginstall
-	CFLAGS = -std=gnu99 -Wall -Wextra -Wpedantic -Os
-endif
-
 .PHONY: all
 all: $(EXENAME)
 
@@ -37,8 +32,8 @@ $(EXENAME): mcrcon.c
 ifneq ($(OS), Windows_NT)
 .PHONY: install
 install:
-	$(INSTALL) -vD $(EXENAME) $(DESTDIR)$(PREFIX)/bin/$(EXENAME)
+	$(INSTALL) -v $(EXENAME) $(DESTDIR)$(PREFIX)/bin/$(EXENAME)
-	$(INSTALL) -vD -m 0644 mcrcon.1 $(DESTDIR)$(PREFIX)/share/man/man1/mcrcon.1
+	$(INSTALL) -v -m 0644 mcrcon.1 $(DESTDIR)$(PREFIX)/share/man/man1/mcrcon.1
 	@echo "\nmcrcon installed. Run 'make uninstall' if you want to uninstall.\n"
 
 .PHONY: uninstall
@@ -49,4 +44,4 @@ endif
 
 .PHONY: clean
 clean:
-	$(RM) $(EXENAME)
+	$(RM) $(EXENAME) $(EXENAME).exe

README.md

@@ -1,33 +1,42 @@
 # 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.
+
 
 ---
 
-### Installing:
+### Installing
 
-##### via packet manager:
+##### Binary releases
-See https://pkgs.org/download/mcrcon for available packages in various Linux distros (note that available packages might be outdated).
 
-- Gentoo Linux: https://packages.gentoo.org/packages/games-util/mcrcon
+Pre-built binaries are provided for Linux and Windows: https://github.com/Tiiffi/mcrcon/releases/latest
-- Arch Linux: https://aur.archlinux.org/packages/mcrcon/
 
-##### building from sources:
+##### Via package manager
+See https://pkgs.org/download/mcrcon and https://repology.org/project/mcrcon/packages for available packages in various Linux distros (note that some packages might be outdated).
+
+- Fedora: https://packages.fedoraproject.org/pkgs/mcrcon/mcrcon/
+- Gentoo: https://packages.gentoo.org/packages/games-util/mcrcon
+- Arch: https://aur.archlinux.org/packages/mcrcon/
+- NixOS: https://search.nixos.org/packages?show=mcrcon
+- Snapcraft: https://snapcraft.io/mcrcon-nsg
+- Scoop: https://scoop.sh/#/apps?q=mcrcon
+
+##### Building from sources
 ```sh
 git clone https://github.com/Tiiffi/mcrcon.git
 cd mcrcon
 make
+
+# install is optional
 sudo make install
 ```
-Check [INSTALL.md](INSTALL.md) for more details.
+_Check [BUILDING.md](BUILDING.md) for more details._
-
-Precompiled binaries (if provided)*: https://github.com/Tiiffi/mcrcon/releases/latest
-
-<sub>*At the moment binaries are provided for Linux and Windows.</sub>
 
 ---
 
-### Usage:
+### Usage
 mcrcon [OPTIONS] [COMMANDS]
 
 Sends rcon commands to Minecraft server.
@@ -56,36 +65,44 @@ MCRCON_PASS
 - Command-line options will override environment variables
 - Rcon commands with spaces must be enclosed in quotes
 
-Example:
+###### Example:
-> Send three commands ("say", "save-all", "stop") and wait five seconds between the commands.
+Send three commands ("say", "save-all", "stop") and wait five seconds between the commands:
 
-  ```mcrcon -H my.minecraft.server -p password -w 5 "say Server is restarting!" save-all stop```
+```sh
+mcrcon -H my.minecraft.server -p password -w 5 "say Server is restarting!" save-all stop
+```
+---
+
+> [!TIP]
+>Enable RCON on Minecraft server by adding following lines to [```server.properties```](https://minecraft.gamepedia.com/Server.properties) configuration file.
+>```
+>enable-rcon=true
+>rcon.port=25575
+>rcon.password=your_rcon_pasword
+>```
 
 ---
 
-##### Enable rcon on server
+### Contact
-Remember to enable rcon by adding following lines to [```server.properties```](https://minecraft.gamepedia.com/Server.properties) file.
-```
-enable-rcon=true
-rcon.port=25575
-rcon.password=your_rcon_pasword
-```
-
----
-
-##### Contact:
 
 * WWW:            https://github.com/Tiiffi/mcrcon/
 * MAIL:           tiiffi+mcrcon at gmail
-* BUG REPORTS:    https://github.com/Tiiffi/mcrcon/issues/
+* ISSUES:         https://github.com/Tiiffi/mcrcon/issues/
+
+> [!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, ...)
+>- Server version: Provide the exact version of the game server
+>- Mods and Extensions: List all mods and extensions used, including their versions
+>- 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.
 
 ---
 
 ### License
 
 This project is licensed under the zlib License - see the [LICENSE](LICENSE) file for details.
-
----
-
-<sub>Master:</sub> ![Master build](https://api.travis-ci.org/Tiiffi/mcrcon.svg?branch=master)
-<sub>Develop:</sub> ![Develop build](https://api.travis-ci.org/Tiiffi/mcrcon.svg?branch=develop)

create_shortcut.bat

@@ -0,0 +1,31 @@
+@echo off
+@cls
+
+@set /p host="Enter host (default: "127.0.0.1"): "
+@if "%host%"=="" set host=127.0.0.1
+
+@set /p port="Enter port (default: 25575): "
+@if "%port%"=="" set port=25575
+
+@set /p passwd="Enter password: "
+@if "%passwd%"=="" set passwd=
+
+set name=connect_%host%-%port%
+
+@set /p name="Enter shortcut name (default: "%name%.bat"): "
+@if "%name%"=="" set name=connect_%host%-%port%
+
+set command=@mcrcon.exe -t -H %host% -P %port% -p %passwd%
+
+@echo %command% >> %name%.bat
+@echo.
+@echo Command: "%command%"
+@echo.
+@echo Shortcut "%name%.bat" created!
+@echo.
+
+@set "host="
+@set "port="
+@set "passwd="
+
+@pause

launch.bat

@@ -0,0 +1,29 @@
+@echo off
+@cls
+
+@if not exist mcrcon.exe (
+  @echo ERROR: Cannot find "mcrcon.exe". Bailing out!
+  @echo.
+  @pause
+  @exit
+)
+
+@set /p host="Enter host (default: 127.0.0.1): "
+@if "%host%"=="" set host=127.0.0.1
+
+@set /p port="Enter port (default: 25575): "
+@if "%port%"=="" set port=25575
+
+@set /p passwd="Enter password: "
+@if "%passwd%"=="" set passwd=
+
+@echo.
+mcrcon.exe -t -H %host% -P %port% -p %passwd%
+@echo.
+
+@set "host="
+@set "port="
+@set "passwd="
+
+@pause
+

mcrcon.1

@@ -1,7 +1,7 @@
 .\" Process this file with
 .\" groff -man -Tascii mcrcon.1
 .\"
-.TH MCRCON 1 "October 2021" "Version 0.7.2"
+.TH MCRCON 1 "December 2024" "Version 0.8.0"
 .SH NAME 
 mcrcon \- send rcon commands to a Minecraft server
 .SH SYNOPSIS
@@ -11,7 +11,7 @@ options
 commands
 .B ]
 .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
 .IP -H
 Server address (default: localhost)

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
  * warranty. In no event will the authors be held liable for any damages
@@ -24,30 +24,27 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <stdint.h>
+#include <stdbool.h>
 #include <string.h>
 #include <strings.h>
 #include <signal.h>
 #include <errno.h>
 #include <unistd.h>
-#include <limits.h>
 
 #ifdef _WIN32
-    // for name resolving on windows
+	#include <winsock2.h>
-    // enable this if you get compiler whine about getaddrinfo() on windows
+	#include <windows.h>
-    //#define _WIN32_WINNT 0x0501
+	#include <ws2tcpip.h>
-
+	#include <fcntl.h>
-    #include <ws2tcpip.h>
+	#include <wchar.h>
-    #include <winsock2.h>
-    #include <windows.h>
 #else
-    #include <sys/types.h>
+	#include <sys/socket.h>
-    #include <sys/socket.h>
+	#include <sys/select.h>
-    #include <netinet/in.h>
+	#include <netdb.h>
-    #include <arpa/inet.h>
+	#include <termios.h>
-    #include <netdb.h>
 #endif
 
-#define VERSION "0.7.2"
+#define VERSION "0.8.0"
 #define IN_NAME "mcrcon"
 #define VER_STR IN_NAME" "VERSION" (built: "__DATE__" "__TIME__")"
 
@@ -57,20 +54,26 @@
 #define RCON_AUTH_RESPONSE      2
 #define RCON_PID                0xBADC0DE
 
-#define DATA_BUFFSIZE 4096
+#define MAX_COMMAND_LENGTH 	4096
+#define DATA_BUFFSIZE		MAX_COMMAND_LENGTH + 2 // plus two null terminators
+#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
-typedef struct _rc_packet {
+typedef struct {
-    int size;
+	int32_t size;
-    int id;
+	int32_t id;
-    int cmd;
+	int32_t cmd;
-    char data[DATA_BUFFSIZE];
+	uint8_t data[DATA_BUFFSIZE];
-    // ignoring string2 for now
+	// ignoring string2 for now
 } rc_packet;
 
-
 // ===================================
-//  FUNCTION DEFINITIONS              
+//  FUNCTION DEFINITIONS
 // ===================================
 
 // Network related functions
@@ -79,10 +82,8 @@ void        net_init_WSA(void);
 #endif
 void        net_close(int sd);
 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);
-int         net_clean_incoming(int sd, int size);
 
 // Misc stuff
 void        usage(void);
@@ -94,25 +95,30 @@ int         run_terminal_mode(int sock);
 int         run_commands(int argc, char *argv[]);
 
 // 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);
-int         rcon_auth(int sock, char *passwd);
+bool        rcon_auth(int sock, char *passwd);
 int         rcon_command(int sock, char *command);
 
-
 // =============================================
 //  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 global_connection_alive = 1;
+static int  flag_wait_seconds = 0;
-static int global_rsock;
+static int  global_connection_alive = 1;
-static int global_wait_seconds = 0;
+static bool global_valve_protocol = false;
+static bool global_minecraft_newline_fix = false;
+static int  global_rsock;
 
 #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
 
 // safety stuff (windows is still misbehaving)
@@ -120,39 +126,44 @@ void exit_proc(void)
 {
 	if (global_rsock != -1)
 		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)
 {
 	if (sig == SIGINT)
 		putchar('\n');
 
 	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)
 {
 	char *end;
 	long result = strtol(str, &end, 10);
 
 	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);
 	}
 
 	if (end == str) {
-		fprintf(stderr, "-w invalid value (not a number?)\n");
+		log_error("-w invalid value (not a number?)\n");
 		exit(EXIT_FAILURE);
 	}
 
 	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);
 	}
 
@@ -166,14 +177,10 @@ int main(int argc, char *argv[])
 	char *host = getenv("MCRCON_HOST");
 	char *pass = getenv("MCRCON_PASS");
 	char *port = getenv("MCRCON_PORT");
-	
+
 	if (!port) port = "25575";
 	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();
 
 	// default getopt error handler enabled
@@ -182,20 +189,20 @@ int main(int argc, char *argv[])
 	while ((opt = getopt(argc, argv, "vrtcshw:H:p:P:")) != -1)
 	{
 		switch (opt) {
-			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':
-				global_wait_seconds = mcrcon_parse_seconds(optarg);
+				flag_wait_seconds = mcrcon_parse_seconds(optarg);
-			break;
+				break;
 
 			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/");
 				exit(EXIT_SUCCESS);
 
@@ -209,11 +216,12 @@ int main(int argc, char *argv[])
 
 	if (pass == NULL) {
 		puts("You must give password (-p password).\nTry 'mcrcon -h' or 'man mcrcon' for help.");
-		return 0;
+		exit(EXIT_FAILURE);
 	}
 
-	if(optind == argc && terminal_mode == 0)
+	if(optind == argc && terminal_mode == 0) {
 		terminal_mode = 1;
+	}
 
 	// safety features to prevent "IO: Connection reset" bug on the server side
 	atexit(&exit_proc);
@@ -222,10 +230,20 @@ int main(int argc, char *argv[])
 	signal(SIGINT, &sighandler);
 
 	#ifdef _WIN32
-		net_init_WSA();
+	net_init_WSA();
-		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
 
 	// open socket
@@ -235,13 +253,11 @@ int main(int argc, char *argv[])
 
 	// auth & commands
 	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
-		fprintf(stdout, "Authentication failed!\n");
+		log_error("Authentication failed!\n");
 		exit_code = EXIT_FAILURE;
 	}
 
@@ -278,8 +294,8 @@ void usage(void)
 	puts("Example:\n\t"IN_NAME" -H my.minecraft.server -p password -w 5 \"say Server is restarting!\" save-all stop\n");
 
 	#ifdef _WIN32
-	    puts("Press enter to exit.");
+	puts("Press enter to exit.");
-	    getchar();
+	getchar();
 	#endif
 
 	exit(EXIT_SUCCESS);
@@ -296,7 +312,7 @@ void net_init_WSA(void)
 
 	int err = WSAStartup(version, &wsadata);
 	if (err != 0) {
-		fprintf(stderr, "WSAStartup failed. Error: %d.\n", err);
+		log_error("WSAStartup failed. Error: %d.\n", err);
 		exit(EXIT_FAILURE);
 	}
 }
@@ -306,14 +322,13 @@ void net_init_WSA(void)
 void net_close(int sd)
 {
 	#ifdef _WIN32
-		closesocket(sd);
+	closesocket(sd);
-		WSACleanup();
+	WSACleanup();
 	#else
-		close(sd);
+	close(sd);
 	#endif
 }
 
-// Opens and connects socket
 // http://man7.org/linux/man-pages/man3/getaddrinfo.3.html
 // https://bugs.chromium.org/p/chromium/issues/detail?id=44489
 int net_connect(const char *host, const char *port)
@@ -329,16 +344,16 @@ int net_connect(const char *host, const char *port)
 	hints.ai_protocol = IPPROTO_TCP;
 
 	#ifdef _WIN32
-	  net_init_WSA();
+	net_init_WSA();
 	#endif
 
 	int ret = getaddrinfo(host, port, &hints, &server_info);
 	if (ret != 0) {
-		fprintf(stderr, "Name resolution failed.\n");
+		log_error("Name resolution failed.\n");
 		#ifdef _WIN32
-			fprintf(stderr, "Error %d: %s", ret, gai_strerror(ret));
+		log_error("Error %d: %s", ret, gai_strerror(ret));
 		#else
-			fprintf(stderr, "Error %d: %s\n", ret, gai_strerror(ret));
+		log_error("Error %d: %s\n", ret, gai_strerror(ret));
 		#endif
 
 		exit(EXIT_FAILURE);
@@ -362,9 +377,9 @@ int net_connect(const char *host, const char *port)
 
 	if (p == NULL) {
 		/* TODO (Tiiffi): Check why windows does not report errors */
-		fprintf(stderr, "Connection failed.\n");
+		log_error("Connection failed.\n");
 		#ifndef _WIN32
-			fprintf(stderr, "Error %d: %s\n", errno, strerror(errno));
+		log_error("Error %d: %s\n", errno, strerror(errno));
 		#endif
 
 		freeaddrinfo(server_info);
@@ -375,81 +390,62 @@ int net_connect(const char *host, const char *port)
 	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 size = packet->size + sizeof(int32_t);
 	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;
 		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(int);
-
-	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)
 {
-	int 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(int), 0);
 
 	if (ret == 0) {
-		fprintf(stderr, "Connection lost.\n");
+		log_error("Connection lost.\n");
 		global_connection_alive = 0;
 		return NULL;
 	}
 
-	if (ret != sizeof(int)) {
+	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;
 		return NULL;
 	}
 
-	// NOTE(Tiiffi): This should fail if size is out of spec!
+	if (psize < MIN_PACKET_SIZE || psize > MAX_PACKET_SIZE) {
-	if (psize < 10 || psize > DATA_BUFFSIZE) {
+		log_error("Error: Invalid packet size (%d).\n", psize);
-		fprintf(stderr, "Warning: invalid packet size (%d). Must over 10 and less than %d.\n", psize, DATA_BUFFSIZE);
+		global_connection_alive = 0;
-
-		if(psize > DATA_BUFFSIZE  || psize < 0) psize = DATA_BUFFSIZE;
-		net_clean_incoming(sd, psize);
-
 		return NULL;
 	}
 
 	packet.size = psize;
+	char *p = (char *) &packet;
 
 	int received = 0;
 	while (received < psize) {
-		ret = recv(sd, (char *) &packet + sizeof(int) + received, psize - received, 0);
+		ret = recv(sd, p + sizeof(int32_t) + received, psize - received, 0);
-		if (ret == 0) { /* connection closed before completing receving */
+		if (ret == 0) {
-			fprintf(stderr, "Connection lost.\n");
+			log_error("Connection lost.\n");
 			global_connection_alive = 0;
 			return NULL;
 		}
@@ -460,19 +456,6 @@ rc_packet *net_recv_packet(int sd)
 	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)
 {
 	// sh compatible color array
@@ -497,8 +480,9 @@ void print_color(int color)
 		"\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
 	#endif
 	{
@@ -506,13 +490,13 @@ void print_color(int color)
 		else if (color >= 0x30 && color <= 0x39)
 			color -= 0x30;
 		else if (color == 0x6e)
-			color = 16; /* 0x6e: 'n' */
+			color = 16;
 		else return;
 
 		#ifndef _WIN32
-			fputs(colors[color], stdout);
+		fputs(colors[color], stdout);
 		#else
-			SetConsoleTextAttribute(console_handle, color);
+		SetConsoleTextAttribute(console_handle, color);
 		#endif
 	}
 }
@@ -520,111 +504,193 @@ void print_color(int color)
 // this hacky mess might use some optimizing
 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;
 	}
 
-	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
-		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
 
-	// 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) {
-				continue;
+			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]);
 			}
-			putchar(packet->data[i]);
+			continue;
 		}
-		print_color(def_color); // cancel coloring
+
-	}
+		// Add missing newlines
-	// strip colors
+		if (colors_detected == false && global_minecraft_newline_fix && data[i] == '/') {
-	else {
+			slash ? putchar('\n') : (slash = true);
-		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
-	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};
 
-	// size + id + cmd + s1 + s2 NULL terminator
+	// NOTE(Tiiffi): Issue report states that maximum command packet size is 1460 bytes:
-	int len = strlen(s1);
+	//				 https://github.com/Tiiffi/mcrcon/issues/45#issuecomment-1000940814
-	if (len >= DATA_BUFFSIZE) {
+	//				 https://mctools.readthedocs.io/en/master/rcon.html
-		fprintf(stderr, "Warning: Command string too long (%d). Maximum allowed: %d.\n", len, DATA_BUFFSIZE - 1);
+	//				 Have to do some testing to confirm!
+
+	int len = strlen(s);
+	if (len > MAX_COMMAND_LENGTH) {
+		log_error("Warning: Command string too long (%d). Maximum allowed: %d.\n", len, MAX_COMMAND_LENGTH);
 		return NULL;
 	}
 
-	packet.size = sizeof(int) * 2 + len + 2;
+	packet.size = sizeof packet.id + sizeof packet.cmd + len + 2;
 	packet.id = id;
 	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;
 }
 
-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);
 	if (packet == NULL)
 		return 0;
 
-	ret = net_send_packet(sock, packet);
+	if (!net_send_packet(sock, packet)) {
-	if (!ret)
 		return 0; // send failed
+	}
 
+	receive:
 	packet = net_recv_packet(sock);
 	if (packet == NULL)
 		return 0;
 
-	// return 1 if authentication OK
+	/* Valve rcon sends empty "RCON_RESPONSEVALUE" packet before real auth response
-	return packet->id == -1 ? 0 : 1;
+	 * so we have to check packet type and try again if necessary.
+	 */
+	if (packet->cmd != RCON_AUTH_RESPONSE) {
+		global_valve_protocol = true;
+		goto receive;
+	}
+
+	// return true if authentication OK
+	return packet->id == -1 ? false : true;
 }
 
 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);
 	if (packet == NULL) {
-		global_connection_alive = 0;
+		log_error("Error: packet build() failed!\n");
 		return 0;
 	}
 
-	net_send_packet(sock, packet);
+	if (!net_send_packet(sock, packet)) {
-
+		log_error("Error: net_send_packet() failed!\n");
-	packet = net_recv_packet(sock);
-	if (packet == NULL)
 		return 0;
-
-	if (packet->id != RCON_PID)
-		return 0;
-
-	if (!global_silent_mode) {
-		if (packet->size > 10)
-		packet_print(packet);
 	}
 
+	// 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);
+		if (packet == NULL) {
+			log_error("Error: net_recv_packet() failed!\n");
+			return 0;
+		}
+
+		// 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;
+		}
+
+		if (packet->id == 0xBADA55) break;
+
+		if (flag_silent_mode == false) {
+			if (packet->size > 10)
+				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;
 }
 
@@ -636,63 +702,121 @@ int run_commands(int argc, char *argv[])
 		if (!rcon_command(global_rsock, argv[i]))
 			return EXIT_FAILURE;
 
-		if (++i >= argc)
+		i++;
+
+		if (i >= argc)
 			return EXIT_SUCCESS;
 
-		if (global_wait_seconds > 0) {
+		if (flag_wait_seconds > 0) {
 			#ifdef _WIN32
-				Sleep(global_wait_seconds * 1000);
+			Sleep(flag_wait_seconds * 1000);
 			#else
-				sleep(global_wait_seconds);
+			sleep(flag_wait_seconds);
 			#endif
 		}
 	}
 }
 
-// interactive terminal mode
+// terminal mode
 int run_terminal_mode(int sock)
 {
-	int ret = 0;
+	char command[MAX_COMMAND_LENGTH] = {0};
-	char command[DATA_BUFFSIZE] = {0x00};
 
-	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) {
 		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)
-			break;
 
-		if (len > 0 && global_connection_alive)
+		if (len > 0 && global_connection_alive) {
-			ret += rcon_command(sock, command);
+			if (!rcon_command(sock, command)) {
+				return EXIT_FAILURE;
+			}
+		}
 
 		/* Special case for "stop" command to prevent server-side bug.
 		 * 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
 
-		//command[0] = len = 0;
+			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!
+		}
 	}
 
-	return ret;
+	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;
+	}
+
+	// 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
 int get_line(char *buffer, int bsize)
 {
+	#ifdef _WIN32
+	char *ret = utf8_getline(buffer, bsize, stdin);
+	#else
 	char *ret = fgets(buffer, bsize, stdin);
+	#endif
+
+	flush_input();
+
 	if (ret == NULL) {
 		if (ferror(stdin)) {
-			fprintf(stderr, "Error %d: %s\n", errno, strerror(errno));
+			log_error("Error %d: %s\n", errno, strerror(errno));
 			exit(EXIT_FAILURE);
 		}
 		putchar('\n');
@@ -701,14 +825,17 @@ int get_line(char *buffer, int bsize)
 
 	// remove unwanted characters from the buffer
 	buffer[strcspn(buffer, "\r\n")] = '\0';
-
 	int len = strlen(buffer);
 
-	// clean input buffer if needed 
+#if 0
+	// clean input buffer if needed
+	#ifndef _WIN32
 	if (len == bsize - 1) {
 		int ch;
 		while ((ch = getchar()) != '\n' && ch != EOF);
 	}
+	#endif
+#endif
 
 	return len;
 }