MicLieg
1 year ago
committed by
MicLieg
MicLieg
5 changed files with 180 additions and 4 deletions
@ -0,0 +1,44 @@ |
|||
#!/bin/bash |
|||
# LinuxGSM command_rcon.sh module |
|||
# Author: Daniel Gibbs |
|||
# Contributors: http://linuxgsm.com/contrib |
|||
# Website: https://linuxgsm.com |
|||
# Description: Send rcon commands to different gameservers. |
|||
|
|||
commandname="RCON" |
|||
commandaction="Rcon" |
|||
moduleselfname="$(basename "$(readlink -f "${BASH_SOURCE[0]}")")" |
|||
fn_firstcommand_set |
|||
|
|||
check.sh |
|||
if [ "${status}" == "0" ]; then |
|||
fn_print_error_nl "Server not running" |
|||
fn_script_log_error "Failed to access: Server not running" |
|||
if fn_prompt_yn "Do you want to start the server?" Y; then |
|||
exitbypass=1 |
|||
command_start.sh |
|||
fi |
|||
fi |
|||
|
|||
|
|||
if [ -n "${userinput2}" ]; then |
|||
rconcommandtosend="${userinput2}" |
|||
else |
|||
fn_print_header |
|||
fn_print_information_nl "Send a RCON command to the server." |
|||
echo "" |
|||
rconcommandtosend=$(fn_prompt_message "RCON command: ") |
|||
fi |
|||
|
|||
fn_print_dots "Sending RCON command to server: \"${rconcommandtosend}\"" |
|||
|
|||
if [ ! -f "${modulesdir}/rcon.py" ]; then |
|||
fn_fetch_file_github "lgsm/modules" "rcon.py" "${modulesdir}" "chmodx" "norun" "noforce" "nohash" |
|||
fi |
|||
|
|||
"${modulesdir}"/rcon.py -a "${telnetip}" -p "${rconport}" -P "${rconpassword}" -c "${rconcommandtosend}" > /dev/null 2>&1 |
|||
|
|||
fn_print_ok_nl "Sending RCON command to server: \"${rconcommandtosend}\"" |
|||
fn_script_log_pass "RCON command \"${rconcommandtosend}\" sent to server" |
|||
|
|||
core_exit.sh |
|||
@ -0,0 +1,116 @@ |
|||
#! /usr/bin/env python3 |
|||
# -*- coding: utf-8 -*- |
|||
# LinuxGSM rcon.py module |
|||
# Author: MicLieg |
|||
# Contributors: http://linuxgsm.com/contrib |
|||
# Website: https://linuxgsm.com |
|||
# Description: Allows sending RCON commands to different gameservers. |
|||
|
|||
import argparse |
|||
import socket |
|||
import struct |
|||
import sys |
|||
|
|||
|
|||
class PacketTypes: |
|||
LOGIN = 3 |
|||
COMMAND = 2 |
|||
|
|||
|
|||
class Rcon: |
|||
|
|||
def __init__(self, arguments): |
|||
self.arguments = arguments |
|||
self.connection = None |
|||
|
|||
def __enter__(self): |
|||
self.connect_to_server() |
|||
return self |
|||
|
|||
def __exit__(self, exc_type, exc_val, exc_tb): |
|||
if self.connection: |
|||
self.connection.close() |
|||
|
|||
@staticmethod |
|||
def fatal_error(error_message, error_code): |
|||
sys.stderr.write(f'ERROR: {error_code} {error_message}\n') |
|||
sys.exit(error_code) |
|||
|
|||
@staticmethod |
|||
def exit_success(success_message=''): |
|||
sys.stdout.write(f'OK: {success_message}\n') |
|||
sys.exit(0) |
|||
|
|||
def connect_to_server(self): |
|||
self.connection = socket.socket(socket.AF_INET, socket.SOCK_STREAM) |
|||
self.connection.settimeout(self.arguments.timeout) |
|||
|
|||
try: |
|||
self.connection.connect((self.arguments.address, self.arguments.port)) |
|||
except socket.timeout: |
|||
self.fatal_error('Request timed out', 1) |
|||
except Exception as e: |
|||
self.fatal_error(f'Unable to connect: {e}', 1) |
|||
|
|||
def send_packet(self, request_id, packet_type, data): |
|||
# Packet structure follows the Source RCON Protocol: size, request ID, type, data, two null bytes |
|||
packet = ( |
|||
struct.pack('<l', request_id) |
|||
+ struct.pack('<l', packet_type) |
|||
+ data.encode('utf8') + b'\x00\x00' |
|||
) |
|||
try: |
|||
self.connection.send(struct.pack('<l', len(packet)) + packet) |
|||
except socket.error as e: |
|||
self.fatal_error(f'Failed to send packet: {e}', 2) |
|||
|
|||
def receive_packet(self): |
|||
try: |
|||
response = self.connection.recv(self.arguments.buffer) |
|||
return response |
|||
except socket.error as e: |
|||
self.fatal_error(f'Failed to receive response: {e}', 3) |
|||
|
|||
def login(self): |
|||
self.send_packet(1, PacketTypes.LOGIN, self.arguments.password) |
|||
response = self.receive_packet() |
|||
if response: |
|||
size, id_response, type_response = struct.unpack('<l', response[:4]), struct.unpack('<l', response[ |
|||
4:8]), struct.unpack( |
|||
'<l', response[8:12]) |
|||
|
|||
if id_response[0] == -1: |
|||
self.fatal_error('Login to RCON failed', 4) |
|||
else: |
|||
self.fatal_error('No response received for login', 4) |
|||
|
|||
def send_command(self): |
|||
self.send_packet(2, PacketTypes.COMMAND, self.arguments.command) |
|||
response = self.receive_packet() |
|||
if response: |
|||
response_message = response[12:-2].decode('utf-8') # Stripping trailing null bytes |
|||
self.exit_success(str(response_message)) |
|||
else: |
|||
self.fatal_error('No response received for command', 5) |
|||
|
|||
|
|||
def parse_args(): |
|||
parser = argparse.ArgumentParser(description='Sends RCON commands to Minecraft servers.') |
|||
parser.add_argument('-a', '--address', type=str, required=True, help='The server IP address.') |
|||
parser.add_argument('-p', '--port', type=int, required=True, help='The server port.') |
|||
parser.add_argument('-P', '--password', type=str, required=True, help='The RCON password.') |
|||
parser.add_argument('-c', '--command', type=str, required=True, help='The RCON command to send.') |
|||
parser.add_argument('-t', '--timeout', type=int, default=5, help='The timeout for server response.') |
|||
parser.add_argument('-b', '--buffer', type=int, default=4096, help='The buffer length for server response.') |
|||
return parser.parse_args() |
|||
|
|||
|
|||
def main(): |
|||
arguments = parse_args() |
|||
with Rcon(arguments) as rcon: |
|||
rcon.login() |
|||
rcon.send_command() |
|||
|
|||
|
|||
if __name__ == '__main__': |
|||
main() |
|||
Loading…
Reference in new issue