1 files changed, 37 insertions, 1 deletions

diff --git a/src/main.c b/src/main.c
index 0a80bf6..fbbdd09 100644
--- a/src/main.c
+++ b/src/main.c
@@ -28,6 +28,8 @@ program_params_t parse_args ( int argc, char* argv[] );
 
 int connect_socket ( char* _host, char* _port );
 
+void print_help ( char* _progname );
+
 int main( int argc, char* argv[] ) {
 	program_params_t args = parse_args( argc, argv );
 
@@ -51,7 +53,33 @@ int main( int argc, char* argv[] ) {
 }
 
 program_params_t parse_args ( int argc, char* argv[] ) {
-	program_params_t ret = { "127.0.0.1", "25575", "1234", "list" };
+	program_params_t ret;
+	memset(&ret, 0, sizeof ret);
+
+	if ( argc < 5 )
+		print_help ( argv[0] );
+
+	ret.host = argv[1];
+	ret.port = argv[2];
+	ret.password = argv[3];
+
+	unsigned int len = 0;
+	for ( int i = 4; i < argc; i++ ) {
+		len += strlen ( argv[i] ) + 1;
+	}
+
+	ret.command = malloc ( len + 1);
+	ret.command[len] = 0;
+
+	unsigned int cnt = 0;
+	for ( int i = 4; i < argc; i++ ) {
+		int arglen = strlen(argv[i]) + 1;
+		memcpy ( ret.command + cnt, argv[i], arglen);
+		(ret.command + cnt)[arglen - 1] = ' ';
+		cnt += arglen;
+	}
+	ret.command[len-1] = 0; // Hacky AF
+
 	return ret;
 }
 
@@ -89,3 +117,11 @@ int connect_socket ( char* _host, char* _port ) {
 	return sock;
 }
 
+void print_help ( char* _progname ) {
+	printf(
+			"Minecraft RCON client\n"
+			"Usage:\n"
+			"	%s HOST PORT PASSWORD COMMAND",
+			_progname
+	      );
+}