aboutsummaryrefslogtreecommitdiff
path: root/src/main.c
blob: fbbdd09c0a7a869ec0a2e997a5adf81dcccd3cbd (plain)
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
/*
 * src/main.c
 * (c) 2020 Jonas Gunz <himself@jonasgunz.de>
 * License: MIT
*/

#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <string.h>
#include <errno.h>

#include <sys/types.h>
#include <sys/socket.h>
#include <netdb.h>
#include <unistd.h>

#include "rcon.h"

typedef struct program_params_s {
	char* 	host;
	char*	port;
	char*	password;
	char*	command;
} program_params_t;

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 );

	int sock = connect_socket ( args.host, args.port );

	fprintf ( stderr, "Connection successful.\n" );

	if ( rcon_login ( sock, args.password ) ) {
		fprintf ( stderr, "Authentification failed.\n" );
		exit ( EXIT_FAILURE );
	}

	char* result = NULL;
        rcon_command ( &result, sock, args.command );

	printf("%s\n", result);

	close (sock);

	return 0;
}

program_params_t parse_args ( int argc, char* argv[] ) {
	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;
}

int connect_socket ( char* _host, char* _port ) {
	int sock = -1;
	int ret = 0;
	struct addrinfo hints, *result, *iter;

	memset ( &hints, 0, sizeof( struct addrinfo ) );
	hints.ai_family = AF_UNSPEC;
	hints.ai_socktype = SOCK_STREAM;

	ret = getaddrinfo ( _host, _port, &hints, &result );
	if ( ret ) {
		fprintf ( stderr, "getaddrinfo: %s\n", gai_strerror(ret) );
		exit ( EXIT_FAILURE );
	}

	for ( iter = result; iter != NULL; iter = iter->ai_next ) {
		sock = socket( result->ai_family, result->ai_socktype, result->ai_protocol );
		if ( sock < 0 )
			continue;

		if ( connect ( sock, iter->ai_addr, iter->ai_addrlen ) != -1 )
			break;

		close (sock);
	}

	if ( ! iter ) {
		fprintf ( stderr, "Connection failed.\n" );
		exit ( EXIT_FAILURE );
	}

	return sock;
}

void print_help ( char* _progname ) {
	printf(
			"Minecraft RCON client\n"
			"Usage:\n"
			"	%s HOST PORT PASSWORD COMMAND",
			_progname
	      );
}