Georg Brandl | 856898b | 2010-12-30 22:11:50 +0000 | [diff] [blame^] | 1 | #!/usr/bin/env python3 |
Guido van Rossum | 4e62037 | 1995-04-10 11:46:03 +0000 | [diff] [blame] | 2 | |
Georg Brandl | 856898b | 2010-12-30 22:11:50 +0000 | [diff] [blame^] | 3 | """ |
| 4 | Remote python server. |
| 5 | Execute Python commands remotely and send output back. |
| 6 | |
| 7 | WARNING: This version has a gaping security hole -- it accepts requests |
| 8 | from any host on the Internet! |
| 9 | """ |
Guido van Rossum | 4e62037 | 1995-04-10 11:46:03 +0000 | [diff] [blame] | 10 | |
| 11 | import sys |
Georg Brandl | 7fafbc9 | 2010-12-30 21:33:07 +0000 | [diff] [blame] | 12 | from socket import socket, AF_INET, SOCK_STREAM |
Collin Winter | 6f2df4d | 2007-07-17 20:59:35 +0000 | [diff] [blame] | 13 | import io |
Guido van Rossum | 4e62037 | 1995-04-10 11:46:03 +0000 | [diff] [blame] | 14 | import traceback |
| 15 | |
| 16 | PORT = 4127 |
| 17 | BUFSIZE = 1024 |
| 18 | |
| 19 | def main(): |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 20 | if len(sys.argv) > 1: |
Georg Brandl | 7fafbc9 | 2010-12-30 21:33:07 +0000 | [diff] [blame] | 21 | port = int(sys.argv[1]) |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 22 | else: |
| 23 | port = PORT |
| 24 | s = socket(AF_INET, SOCK_STREAM) |
| 25 | s.bind(('', port)) |
| 26 | s.listen(1) |
Georg Brandl | 7fafbc9 | 2010-12-30 21:33:07 +0000 | [diff] [blame] | 27 | while True: |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 28 | conn, (remotehost, remoteport) = s.accept() |
Georg Brandl | 7fafbc9 | 2010-12-30 21:33:07 +0000 | [diff] [blame] | 29 | print('connection from', remotehost, remoteport) |
| 30 | request = b'' |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 31 | while 1: |
| 32 | data = conn.recv(BUFSIZE) |
| 33 | if not data: |
| 34 | break |
Georg Brandl | 7fafbc9 | 2010-12-30 21:33:07 +0000 | [diff] [blame] | 35 | request += data |
| 36 | reply = execute(request.decode()) |
| 37 | conn.send(reply.encode()) |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 38 | conn.close() |
Guido van Rossum | 4e62037 | 1995-04-10 11:46:03 +0000 | [diff] [blame] | 39 | |
| 40 | def execute(request): |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 41 | stdout = sys.stdout |
| 42 | stderr = sys.stderr |
Collin Winter | 6f2df4d | 2007-07-17 20:59:35 +0000 | [diff] [blame] | 43 | sys.stdout = sys.stderr = fakefile = io.StringIO() |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 44 | try: |
| 45 | try: |
Georg Brandl | 7cae87c | 2006-09-06 06:51:57 +0000 | [diff] [blame] | 46 | exec(request, {}, {}) |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 47 | except: |
Collin Winter | 6f2df4d | 2007-07-17 20:59:35 +0000 | [diff] [blame] | 48 | print() |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 49 | traceback.print_exc(100) |
| 50 | finally: |
| 51 | sys.stderr = stderr |
| 52 | sys.stdout = stdout |
| 53 | return fakefile.getvalue() |
Guido van Rossum | 4e62037 | 1995-04-10 11:46:03 +0000 | [diff] [blame] | 54 | |
Georg Brandl | 7fafbc9 | 2010-12-30 21:33:07 +0000 | [diff] [blame] | 55 | try: |
| 56 | main() |
| 57 | except KeyboardInterrupt: |
| 58 | pass |