← Back to branch summary

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

« back to all changes in this revision

Viewing changes to lib/common/chat.py

  • Committer: mattgiuca
  • Date: 2008-07-15 07:19:34 UTC
  • Revision ID: svn-v3-trunk0:2b9c9e99-6f39-0410-b283-7f802c844ae2:trunk:875
Added "migrations" directory, which contains incremental database update
    scripts.
Updated users.sql, uniqueness key on offering table.
Added migration matching this update to the migrations directory. Mm handy!

Show diffs side-by-side

added added

removed removed

Lines of Context:
lib/common/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 md5
 
25
import sys
 
26
import os
 
27
import socket
 
28
import traceback
 
29
 
 
30
def start_server(port, magic, daemon_mode, handler, initializer = None):
 
31
    # Attempt to open the socket.
 
32
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
 
33
    s.bind(('', port))
 
34
    s.listen(1)
 
35
 
 
36
    # Excellent! It worked. Let's turn ourself into a daemon,
 
37
    # then get on with the job of being a python interpreter.
 
38
    if daemon_mode:
 
39
        if os.fork():   # launch child and...
 
40
            os._exit(0) # kill off parent
 
41
        os.setsid()
 
42
        if os.fork():   # launch child and...
 
43
            os._exit(0) # kill off parent again.
 
44
        os.umask(077)
 
45
 
 
46
    if initializer:
 
47
        initializer()
 
48
 
 
49
    while True:
 
50
        (conn, addr) = s.accept()
 
51
        try:
 
52
            # Grab the input
 
53
            buf = cStringIO.StringIO()
 
54
            blk = conn.recv(1024)
 
55
            while blk:
 
56
                buf.write(blk)
 
57
                try:
 
58
                    blk = conn.recv(1024, socket.MSG_DONTWAIT)
 
59
                except:
 
60
                    # Exception thrown if it WOULD block (but we
 
61
                    # told it not to wait) - ie. we are done
 
62
                    blk = None
 
63
            inp = buf.getvalue()
 
64
            env = cjson.decode(inp)
 
65
            
 
66
            # Check that the message is 
 
67
            digest = md5.new(env['content'] + magic).digest().encode('hex')
 
68
            if env['digest'] != digest:
 
69
                conn.close()
 
70
                continue
 
71
 
 
72
            content = cjson.decode(env['content'])
 
73
 
 
74
            response = handler(content)
 
75
 
 
76
            conn.sendall(cjson.encode(response))
 
77
 
 
78
            conn.close()
 
79
        except Exception:
 
80
            # Make a JSON object full of exceptional goodness
 
81
            tb_dump = cStringIO.StringIO()
 
82
            e_type, e_val, e_tb = sys.exc_info()
 
83
            traceback.print_tb(e_tb, file=tb_dump)
 
84
            json_exc = {
 
85
                "type": e_type.__name__,
 
86
                "value": str(e_val),
 
87
                "traceback": tb_dump.getvalue()
 
88
            }
 
89
            conn.sendall(cjson.encode(json_exc))
 
90
            conn.close()
 
91
 
 
92
 
 
93
def chat(host, port, msg, magic, decode = True):
 
94
    sok = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
 
95
    sok.connect((host, port))
 
96
    content = cjson.encode(msg)
 
97
    digest = md5.new(content + magic).digest().encode("hex")
 
98
    env = {'digest':digest,'content':content}
 
99
    sok.send(cjson.encode(env))
 
100
 
 
101
    buf = cStringIO.StringIO()
 
102
    blk = sok.recv(1024)
 
103
    while blk:
 
104
        buf.write(blk)
 
105
        try:
 
106
            blk = conn.recv(1024, socket.MSG_DONTWAIT)
 
107
        except:
 
108
            # Exception thrown if it WOULD block (but we
 
109
            # told it not to wait) - ie. we are done
 
110
            blk = None
 
111
    inp = buf.getvalue()
 
112
    sok.close()
 
113
 
 
114
    if decode:
 
115
        return cjson.decode(inp)
 
116
    else:
 
117
        return inp
 
118