← Back to branch summary

~azzar1/unity/add-show-desktop-key

« back to all changes in this revision

Viewing changes to ivle/chat.py

  • Committer: drtomc
  • Date: 2007-12-05 22:06:57 UTC
  • Revision ID: svn-v3-trunk0:2b9c9e99-6f39-0410-b283-7f802c844ae2:trunk:6
Make a bunch of small changes suggested by Ivo.

Show diffs side-by-side

added added

removed removed

Lines of Context:
ivle/chat.py
1
 
# IVLE - Informatics Virtual Learning Environment
2
 
# Copyright (C) 2007-2008 The University of Melbourne
3
 
#
4
 
# This program is free software; you can redistribute it and/or modify
5
 
# it under the terms of the GNU General Public License as published by
6
 
# the Free Software Foundation; either version 2 of the License, or
7
 
# (at your option) any later version.
8
 
#
9
 
# This program is distributed in the hope that it will be useful,
10
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
11
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
 
# GNU General Public License for more details.
13
 
#
14
 
# You should have received a copy of the GNU General Public License
15
 
# along with this program; if not, write to the Free Software
16
 
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
17
 
 
18
 
# Module: Chat
19
 
# Author: Thomas Conway
20
 
# Date:   5/2/2008
21
 
 
22
 
import cjson
23
 
import cStringIO
24
 
import hashlib
25
 
import sys
26
 
import os
27
 
import socket
28
 
import traceback
29
 
 
30
 
class Terminate(Exception):
31
 
    """Exception thrown when server is to be shut down. It will attempt to sned 
32
 
    the final_response to the client and then exits"""
33
 
    def __init__(self, final_response=None):
34
 
        self.final_response = final_response
35
 
 
36
 
    def __str__(self):
37
 
        return repr(self.final_response)
38
 
 
39
 
 
40
 
def start_server(port, magic, daemon_mode, handler, initializer = None):
41
 
    # Attempt to open the socket.
42
 
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
43
 
    s.bind(('', port))
44
 
    s.listen(1)
45
 
 
46
 
    # Excellent! It worked. Let's turn ourself into a daemon,
47
 
    # then get on with the job of being a python interpreter.
48
 
    if daemon_mode:
49
 
        if os.fork():   # launch child and...
50
 
            os._exit(0) # kill off parent
51
 
        os.setsid()
52
 
        if os.fork():   # launch child and...
53
 
            os._exit(0) # kill off parent again.
54
 
        os.umask(077)
55
 
 
56
 
        try:
57
 
            MAXFD = os.sysconf("SC_OPEN_MAX")
58
 
        except:
59
 
            MAXFD = 256
60
 
 
61
 
        # Close all file descriptors, except the socket.
62
 
        for i in xrange(MAXFD):
63
 
            if i == s.fileno():
64
 
                continue
65
 
            try:
66
 
                os.close(i)
67
 
            except OSError:
68
 
                pass
69
 
 
70
 
        si = os.open(os.devnull, os.O_RDONLY)
71
 
        os.dup2(si, sys.stdin.fileno())
72
 
 
73
 
        so = os.open(os.devnull, os.O_WRONLY)
74
 
        os.dup2(so, sys.stdout.fileno())
75
 
 
76
 
        se = os.open(os.devnull, os.O_WRONLY)
77
 
        os.dup2(se, sys.stderr.fileno())
78
 
 
79
 
    if initializer:
80
 
        initializer()
81
 
 
82
 
    while True:
83
 
        (conn, addr) = s.accept()
84
 
        try:
85
 
            # Grab the input
86
 
            buf = cStringIO.StringIO()
87
 
            blk = conn.recv(1024)
88
 
            while blk:
89
 
                buf.write(blk)
90
 
                try:
91
 
                    blk = conn.recv(1024, socket.MSG_DONTWAIT)
92
 
                except:
93
 
                    # Exception thrown if it WOULD block (but we
94
 
                    # told it not to wait) - ie. we are done
95
 
                    blk = None
96
 
            inp = buf.getvalue()
97
 
            env = cjson.decode(inp)
98
 
            
99
 
            # Check that the message is 
100
 
            digest = hashlib.md5(env['content'] + magic).hexdigest()
101
 
            if env['digest'] != digest:
102
 
                conn.close()
103
 
                continue
104
 
 
105
 
            content = cjson.decode(env['content'])
106
 
 
107
 
            response = handler(content)
108
 
 
109
 
            conn.sendall(cjson.encode(response))
110
 
 
111
 
            conn.close()
112
 
 
113
 
        except Terminate, t:
114
 
            # Try and send final response and then terminate
115
 
            if t.final_response:
116
 
                conn.sendall(cjson.encode(t.final_response))
117
 
            conn.close()
118
 
            sys.exit(0)
119
 
        except Exception:
120
 
            # Make a JSON object full of exceptional goodness
121
 
            tb_dump = cStringIO.StringIO()
122
 
            e_type, e_val, e_tb = sys.exc_info()
123
 
            traceback.print_tb(e_tb, file=tb_dump)
124
 
            json_exc = {
125
 
                "type": e_type.__name__,
126
 
                "value": str(e_val),
127
 
                "traceback": tb_dump.getvalue()
128
 
            }
129
 
            conn.sendall(cjson.encode(json_exc))
130
 
            conn.close()
131
 
 
132
 
 
133
 
def chat(host, port, msg, magic, decode = True):
134
 
    sok = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
135
 
    sok.connect((host, port))
136
 
    content = cjson.encode(msg)
137
 
    digest = hashlib.md5(content + magic).hexdigest()
138
 
    env = {'digest':digest,'content':content}
139
 
    sok.send(cjson.encode(env))
140
 
 
141
 
    buf = cStringIO.StringIO()
142
 
    blk = sok.recv(1024)
143
 
    while blk:
144
 
        buf.write(blk)
145
 
        try:
146
 
            blk = sok.recv(1024, socket.MSG_DONTWAIT)
147
 
        except socket.error, e:
148
 
            if e[0] != 11:
149
 
                raise
150
 
            # Exception thrown if it WOULD block (but we
151
 
            # told it not to wait) - ie. we are done
152
 
            blk = None
153
 
    inp = buf.getvalue()
154
 
    sok.close()
155
 
 
156
 
    if decode:
157
 
        return cjson.decode(inp)
158
 
    else:
159
 
        return inp
160