← Back to branch summary

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

« back to all changes in this revision

Viewing changes to ivle/chat.py

  • Committer: William Grant
  • Date: 2009-05-18 12:13:44 UTC
  • Revision ID: grantw@unimelb.edu.au-20090518121344-m69gxcfqwoeirtru
Fix make_path_segments' docstring.

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