← Back to branch summary

~ubuntu-branches/ubuntu/maverick/python3.1/maverick

« back to all changes in this revision

Viewing changes to Demo/pysvr/pysvr.py

  • Committer: Bazaar Package Importer
  • Author(s): Matthias Klose
  • Date: 2009-03-23 00:01:27 UTC
  • Revision ID: james.westby@ubuntu.com-20090323000127-5fstfxju4ufrhthq
Tags: upstream-3.1~a1+20090322
Import upstream version 3.1~a1+20090322

Show diffs side-by-side

added added

removed removed

Lines of Context:
Demo/pysvr/pysvr.py
 
1
#! /usr/bin/env python
 
2
 
 
3
"""A multi-threaded telnet-like server that gives a Python prompt.
 
4
 
 
5
This is really a prototype for the same thing in C.
 
6
 
 
7
Usage: pysvr.py [port]
 
8
 
 
9
For security reasons, it only accepts requests from the current host.
 
10
This can still be insecure, but restricts violations from people who
 
11
can log in on your machine.  Use with caution!
 
12
 
 
13
"""
 
14
 
 
15
import sys, os, string, getopt, _thread, socket, traceback
 
16
 
 
17
PORT = 4000                             # Default port
 
18
 
 
19
def main():
 
20
    try:
 
21
        opts, args = getopt.getopt(sys.argv[1:], "")
 
22
        if len(args) > 1:
 
23
            raise getopt.error("Too many arguments.")
 
24
    except getopt.error as msg:
 
25
        usage(msg)
 
26
    for o, a in opts:
 
27
        pass
 
28
    if args:
 
29
        try:
 
30
            port = string.atoi(args[0])
 
31
        except ValueError as msg:
 
32
            usage(msg)
 
33
    else:
 
34
        port = PORT
 
35
    main_thread(port)
 
36
 
 
37
def usage(msg=None):
 
38
    sys.stdout = sys.stderr
 
39
    if msg:
 
40
        print(msg)
 
41
    print("\n", __doc__, end=' ')
 
42
    sys.exit(2)
 
43
 
 
44
def main_thread(port):
 
45
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
 
46
    sock.bind(("", port))
 
47
    sock.listen(5)
 
48
    print("Listening on port", port, "...")
 
49
    while 1:
 
50
        (conn, addr) = sock.accept()
 
51
        if addr[0] != conn.getsockname()[0]:
 
52
            conn.close()
 
53
            print("Refusing connection from non-local host", addr[0], ".")
 
54
            continue
 
55
        _thread.start_new_thread(service_thread, (conn, addr))
 
56
        del conn, addr
 
57
 
 
58
def service_thread(conn, addr):
 
59
    (caddr, cport) = addr
 
60
    print("Thread %s has connection from %s.\n" % (str(_thread.get_ident()),
 
61
                                                   caddr), end=' ')
 
62
    stdin = conn.makefile("r")
 
63
    stdout = conn.makefile("w", 0)
 
64
    run_interpreter(stdin, stdout)
 
65
    print("Thread %s is done.\n" % str(_thread.get_ident()), end=' ')
 
66
 
 
67
def run_interpreter(stdin, stdout):
 
68
    globals = {}
 
69
    try:
 
70
        str(sys.ps1)
 
71
    except:
 
72
        sys.ps1 = ">>> "
 
73
    source = ""
 
74
    while 1:
 
75
        stdout.write(sys.ps1)
 
76
        line = stdin.readline()
 
77
        if line[:2] == '\377\354':
 
78
            line = ""
 
79
        if not line and not source:
 
80
            break
 
81
        if line[-2:] == '\r\n':
 
82
            line = line[:-2] + '\n'
 
83
        source = source + line
 
84
        try:
 
85
            code = compile_command(source)
 
86
        except SyntaxError as err:
 
87
            source = ""
 
88
            traceback.print_exception(SyntaxError, err, None, file=stdout)
 
89
            continue
 
90
        if not code:
 
91
            continue
 
92
        source = ""
 
93
        try:
 
94
            run_command(code, stdin, stdout, globals)
 
95
        except SystemExit as how:
 
96
            if how:
 
97
                try:
 
98
                    how = str(how)
 
99
                except:
 
100
                    how = ""
 
101
                stdout.write("Exit %s\n" % how)
 
102
            break
 
103
    stdout.write("\nGoodbye.\n")
 
104
 
 
105
def run_command(code, stdin, stdout, globals):
 
106
    save = sys.stdin, sys.stdout, sys.stderr
 
107
    try:
 
108
        sys.stdout = sys.stderr = stdout
 
109
        sys.stdin = stdin
 
110
        try:
 
111
            exec(code, globals)
 
112
        except SystemExit as how:
 
113
            raise SystemExit(how).with_traceback(sys.exc_info()[2])
 
114
        except:
 
115
            type, value, tb = sys.exc_info()
 
116
            if tb: tb = tb.tb_next
 
117
            traceback.print_exception(type, value, tb)
 
118
            del tb
 
119
    finally:
 
120
        sys.stdin, sys.stdout, sys.stderr = save
 
121
 
 
122
from code import compile_command
 
123
 
 
124
main()