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

277 lines
6.9 KiB
Python
Executable file

#!/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.20130424
# protocol version: 0.2.20130423
#
#
import sys, os, socket, socketserver, threading, time
def motd():
return MOTD
#def content(client_address, data):
#ret = ""
#ret = ret +\
#"{BEGIN}\n"\
#"asrcp" + ProtocolVersion + "\n"
## Look if the received message is an
## valid alias or a predefined command
## if it's 'version', return the server and protocol version
#if data == "version":
#if VERBOSITY >= 2: print("Got valid service command from"
#+ str(client_address) + ": ", data)
#ret = ret +\
#"202 Valid Service Command\n"\
#"002 Version\n"\
#"ServerVersion:" + ServerVersion + "\n"\
#"ProtocolVersion:" + ProtocolVersion + "\n"
## if it's 'help', give a little help
#elif data == 'help':
#if VERBOSITY >= 2: print("Got valid command from"
#+ str(client_address) + ": ", data)
## send status code
#ret = ret + "202 Valid Service Command\n\n"
## send the list of aliases
#ret = ret + "Aviable aliases:\n"
#for i in aliases.keys():
#ret = ret + str(i) + "\n"
## if it's a valid userdefined command
#elif data in aliases:
## send status code
#ret = ret + "201 Valid Command\n\n"
## ohmagawd! a debug message!!1!
#if VERBOSITY >= 2: print("Got valid command from"
#+ str(client_address) + ": ", data)
## execute the aliased command
#g_dict, l_dict = {}, {}
#exec(str(aliases[data]), g_dict, l_dict)
## send may contain data to send to the client
#if l_dict["send"]:
#content = str(l_dict["send"]).replace('{', '\{')
#content = content.replace('}', '\}')
#ret = ret + content + "\n"
## ALL IS LOST!!1! this has to be invalid!
#else:
## send status code
#ret = ret + "203 Invalid Command\n"
#if VERBOSITY >= 2: print("Got invalid command from",
#str(client_address), ": ", data)
#ret = ret + "{END}\n"
#return ret
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 VERBOSITY >=3:
print("Client connected: " + str(self.client_address))
# send header line 1
self.request.sendall(bytes
("asrpc " + ProtocolVersion + "\n", ENCODING))
# send motd
self.request.sendall(bytes(motd() + "\n", ENCODING))
# Receive data
self.data = str(self.rfile.readline().strip(), ENCODING)
# content handler
self.request.sendall(bytes(
content.handler(str(self.client_address), self.data, aliases,
ServerVersion, ProtocolVersion, VERBOSITY), ENCODING))
class ThreadedTCPServer(socketserver.ThreadingMixIn, socketserver.TCPServer):
pass
def main():
global VERBOSITY, HOST, PORT, TIMEOUT, ENCODING
from include import argparser, content
parser = argparser
args = parser.parse(ServerVersion, ProtocolVersion)
if os.path.exists("pid"):
if args.delete_pid_file: os.remove("pid")
elif args.allow_multiple_instances: pass
else:
print(
"\npid file already exists\n"\
"If the server didn't shut down correctly, just delete this file "\
"or pass -n\n"\
"If you want to start multiple instances, pass -m\n")
return 1
print(args)
if args.host: HOST = args.host
if args.port: PORT = args.port
if args.timeout: TIMEOUT = args.timeout
if args.encoding: ENCODING = args.encoding
if args.verbosity: VERBOSITY = args.verbosity
# write pid file
pidf = open("pid", 'w')
pidf.write(str(os.getpid()))
pidf.close()
print("aSRC Server\n"\
"Server version: " + ServerVersion + "\n"\
"Protocol version: " + ProtocolVersion + "\n\n"\
"To stop the server, press Ctrl-C\n")
try:
if VERBOSITY >= 1: print("Initializing server...")
# Create server
server = ThreadedTCPServer((HOST, PORT), ThreadedRequestHandler)
if VERBOSITY >= 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("\nGot Ctrl-C, shutting down server...")
try:
server.shutdown()
os.remove("pid")
except: print("Failed to shutdown server correctly, "\
"socket may be still in use or wasn't even started:",
sys.exc_info())
#except:
# print("\nAn error occured:\n", sys.exc_info(), "\n")
if VERBOSITY >= 3: input("Press Enter to continue\n")
return 0
if __name__ == '__main__':
ServerVersion = "0.0.0"
ProtocolVersion = "0.2"
##############
# 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 - still doesn't work obviously...
TIMEOUT = 15
# 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.\n"\
"Here (may) be more text..."
# Verbosity of logging.
# Can be from 0 (only default output) to 3 (debug messages)
VERBOSITY = 3
##############
main()