1
0
Fork 0
asrc-dev/asrc-server.py

226 lines
6.1 KiB
Python
Raw Normal View History

2013-04-23 00:25:16 +02:00
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
#
#
# asrc-server.py
#
# Copyright 2012 - 2013 Fanir-Thuban <projects@mail.fanir.de>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
# MA 02110-1301, USA.
#
#
# aSRC (Aliased Server Remote Control)
# - SERVER -
#
# program version: 0.0.0.20130422
# protocol version: 0.2.20130422
#
#
import sys, os, socket, socketserver, threading, time
class ThreadedRequestHandler(socketserver.StreamRequestHandler):
def handle(self):
# Set time for timeout in seconds
self.timeout = TIMEOUT
# Print a line with the adress of the connected client
if DEBUGLEVEL >=3:
print("Client connected: " + str(self.client_address))
# Send MOTD
self.request.sendall(bytes(motd + "\r\n", ENCODING))
# Receive data
self.data = str(self.rfile.readline().strip(), ENCODING)
# Look if the received message is an
# valid alias or a predefined command
# if it's 'version', return the server and protocol version
if self.data == "version":
if DEBUGLEVEL >= 2: print("Got valid service command from"
+ str(self.client_address) + ": ", self.data)
# send status code
self.request.sendall(bytes("000 Received 'version'", ENCODING))
# send response
self.request.sendall(bytes(
"Server: " + ServerVersion + "\r\n"\
"Protocol: " + ProtocolVersion + "\r\n", ENCODING))
# if it's 'help', give a little help
elif self.data == 'help':
if DEBUGLEVEL >= 2: print("Got valid command from"
+ str(self.client_address) + ": ", self.data)
# send status code
self.request.sendall(bytes("000 Received 'help'", ENCODING))
# send the list of aliases
self.request.sendall(bytes("Aviable aliases:\r\n", ENCODING))
for i in aliases.keys():
self.request.sendall(bytes(str(i) + "\r\n", ENCODING))
self.request.sendall(bytes("\r\n", ENCODING))
# if it's a valid userdefined command
elif self.data in aliases:
# send status code
self.request.sendall(bytes(
"<<:0x0130:Received valid command\r\n", ENCODING))
# ohmagawd! a debug message!!1!
if DEBUGLEVEL >= 2: print("Got valid command from"
+ str(self.client_address) + ": ", self.data)
# execute the aliased command
g_dict, l_dict = {}, {}
exec(str(aliases[self.data]), g_dict, l_dict)
# send may contain data to send to the client
if l_dict["send"]:
self.request.sendall(bytes(str(
l_dict["send"]) + "\r\n", ENCODING))
else:
self.request.sendall(bytes("\r\n", ENCODING))
# ALL IS LOST!!1! this have to be invalid!
else:
# send status code
self.request.sendall(bytes(
"<<:0x0230:Received invalid command\r\n", ENCODING))
if DEBUGLEVEL >= 2: print("Got invalid command from",
str(self.client_address), ": ", self.data)
class ThreadedTCPServer(socketserver.ThreadingMixIn, socketserver.TCPServer):
pass
def main():
print("aSRC Server\r\n"\
"Server version: " + ServerVersion + "\r\n"\
"Protocol version: " + ProtocolVersion + "\r\n\r\n"\
"To stop the server, press Ctrl-C\r\n")
try:
if DEBUGLEVEL >= 1: print("Initializing server...")
# Create server
server = ThreadedTCPServer((HOST, PORT), ThreadedRequestHandler)
if DEBUGLEVEL >= 1: print("Starting server on",
server.server_address[0], "port", server.server_address[1], "...")
# Start a thread with the server -- that thread will then start one
# more thread for each request
ServerThread = threading.Thread(target = server.serve_forever)
# Exit the server thread when the main thread terminates
ServerThread.daemon = True
ServerThread.start()
while True:
time.sleep(10)
except KeyboardInterrupt:
print("\r\nGot Ctrl-C, shutting down server...")
try: server.shutdown()
except: print("Failed to shutdown server correctly, "\
"socket may be still in use or wasn't even started:",
sys.exc_info())
except:
print("\r\nAn error occured:\r\n", sys.exc_info(), "\r\n")
if DEBUGLEVEL >= 3: input("Press Enter to continue\r\n")
return 0
if __name__ == '__main__':
ServerVersion = "0.0.0"
ProtocolVersion = "0.0.0"
##############
# SETTINGS #
##############
# IP or hostname (use 0.0.0.0 for all interfaces) and port, on which
# the server should listen
HOST = '127.0.0.1'
PORT = 24642
# Timeout of a connection in seconds - obviously still doesn't work...
TIMEOUT = 4
# Encoding to be used when communicating with a client
ENCODING = 'utf-8'
# Dictionary of aliases. Use python syntax. You can use
# the variable send for text to send to the client.
#
# Shell commands can be executed with:
# import subprocess
# send = subprocess.check_output(["command", "arg", "somemorearg"])
#
# You don't have to include sys, os, socket, socketserver,
# threading and time, through they are included already.
aliases = dict(
who = '''
import subprocess
send = subprocess.check_output(["whoami"])
''',
where = '''
import subprocess
send = subprocess.check_output(["pwd"])
''',
uname = '''
import os
send = os.uname()
''',
date = '''
import subprocess
send = subprocess.check_output(["date"])
''',
ping_fanir = '''
import subprocess
send = subprocess.check_output(["ping", "-c 2", "fanir.de"])
''')
# This is sent to the client after the connection is established
motd = "Welcome! This is only a test server, for developing purposes.\r\n"\
"Here (may) be more text..."
# Verbosity of logging.
# Can be from 0 (only default output) to 3 (debug messages)
DEBUGLEVEL = 3
##############
main()