← 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-02-15 04:45:12 UTC
  • Revision ID: svn-v3-trunk0:2b9c9e99-6f39-0410-b283-7f802c844ae2:trunk:472
db.py: No longer exceptions if password is not supplied.
makeuser: Converted explicit args into kwargs which are simply forwarded along
    to db. Makeuser db NO LONGER accepts "force" argument (which used to
    delete and replace login rows). This is dangerous, and we have removed the
    delete user functionality altogether - because we never delete users.
makeuser.py (script): Major change. Replaced the command-line interface so it
    now only takes 3 required arguments, and a bunch of optional arguments
    getopts style.
    All the arguments are specified at the top; the rest of the code works
    with these data structures so they are only specified in one place
    (including all the work of putting them into a dict and passing it to the
    database).
    Note that makeuser.py is a *second* point of control for the list of the
    login fields (the other is in db.DB.login_fields_list). This is by design,
    because there is a semantic difference between "what fields are in the db"
    and "what arguments should we accept on the command line".

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