← Back to branch summary

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

« back to all changes in this revision

Viewing changes to services/usrmgt-server

  • Committer: David Coles
  • Date: 2009-08-06 04:04:37 UTC
  • Revision ID: coles.david@gmail.com-20090806040437-a8k5jhkkf2ixud5a
Add a rather lenient RLIMIT_NPROC that will prevent simple fork bombs (hopefully accidental...) from taking down a server.

Show diffs side-by-side

added added

removed removed

Lines of Context:
services/usrmgt-server
4
4
import sys
5
5
import logging
6
6
 
7
 
import conf
8
 
import common.db
9
 
import common.chat
10
 
import common.makeuser
11
 
import common.studpath
 
7
import ivle.config
 
8
import ivle.database
 
9
import ivle.chat
 
10
import ivle.makeuser
 
11
 
 
12
config = ivle.config.Config()
12
13
 
13
14
# usage:
14
15
#   usrmgt-server <port> <magic>
28
29
#   - Rebuild svn auth file
29
30
#   - Rebuild passwd + push to nodes.
30
31
 
31
 
def activate_user(props):
 
32
def activate_user(store, props, config):
32
33
    """Create the on-disk stuff for the given user.
33
34
       Sets the state of the user in the db from pending to enabled.
 
35
    @param config: An ivle.config.Config object.
34
36
       Expected properties:
35
37
        login       - the user name for the jail
36
38
                      STRING REQUIRED
37
 
       Return Value: None
 
39
    @return: None
38
40
    """
39
41
 
40
42
    os.umask(0022) # Bad, but start_server sets it worse.
41
43
 
42
44
    login = props['login']
43
45
 
44
 
    db = common.db.DB()
45
 
 
46
 
    try:
47
 
 
48
 
        # FIXME: check we're pending
49
 
 
50
 
        details = db.get_user(login)
51
 
 
52
 
        # make svn config/auth
53
 
 
54
 
        repopath = os.path.join(conf.svn_repo_path, 'users', login)
55
 
        logging.debug("Creating user's Subversion repository")
56
 
        common.makeuser.make_svn_repo(repopath, throw_on_error=True)
57
 
 
58
 
        rebuild_svn_config(props)
59
 
 
60
 
        logging.debug("Adding Subversion authentication")
61
 
        passwd = common.makeuser.make_svn_auth(login, throw_on_error=True)
62
 
        logging.debug("passwd: %s" % passwd)
63
 
 
64
 
        logging.debug("Creating jail")
65
 
        common.makeuser.make_jail(login, details.unixid, svn_pass=passwd)
66
 
 
67
 
        logging.info("Enabling user")
68
 
        db.update_user(login, state='enabled')
69
 
 
70
 
        return {"response": "okay"}
71
 
 
72
 
    finally:
73
 
        db.close()
74
 
 
75
 
def rebuild_svn_config(props):
 
46
    # FIXME: check we're pending
 
47
 
 
48
    # Get the full User object from the db associated with this
 
49
    user = ivle.database.User.get_by_login(store, login)
 
50
 
 
51
    # make svn config/auth
 
52
    repopath = os.path.join(config['paths']['svn']['repo_path'],
 
53
                            'users', login)
 
54
    logging.debug("Creating user's Subversion repository")
 
55
    ivle.makeuser.make_svn_repo(repopath, throw_on_error=True)
 
56
 
 
57
    rebuild_svn_config(store, props, config)
 
58
 
 
59
    logging.debug("Adding Subversion authentication")
 
60
    passwd = ivle.makeuser.make_svn_auth(store, login, config,
 
61
                                         throw_on_error=True)
 
62
 
 
63
    logging.debug("Creating jail")
 
64
    ivle.makeuser.make_jail(user, config)
 
65
 
 
66
    logging.info("Enabling user")
 
67
    user.state = u'enabled'
 
68
 
 
69
    return {"response": "okay"}
 
70
 
 
71
def rebuild_svn_config(store, props, config):
76
72
    """Rebuilds the svn config file
77
 
    Return value:
78
 
        response (okay, failure)
 
73
    @param config: An ivle.config.Config object.
 
74
    @return: response (okay, failure)
79
75
    """
80
76
    try:
81
 
        common.makeuser.rebuild_svn_config()
 
77
        ivle.makeuser.rebuild_svn_config(store, config)
82
78
    except Exception, e:
83
79
        logging.warning('Rebuild of Subversion authorization config failed!')
84
80
        return{'response': 'failure', 'msg': repr(e)}
85
81
 
86
82
    return {'response': 'okay'}
87
83
 
88
 
def rebuild_svn_group_config(props):
 
84
def rebuild_svn_group_config(store, props, config):
89
85
    """Rebuilds the svn group config file
90
 
    Return value:
91
 
        response (okay, failure)
 
86
    @param config: An ivle.config.Config object.
 
87
    @return: response (okay, failure)
92
88
    """
93
89
    try:
94
 
        common.makeuser.rebuild_svn_group_config()
 
90
        ivle.makeuser.rebuild_svn_group_config(store, config)
95
91
    except Exception, e:
96
92
        logging.warning(
97
93
            'Rebuild of Subversion group authorization config failed!')
99
95
 
100
96
    return {'response': 'okay'}
101
97
 
102
 
def create_group_repository(props):
 
98
def create_group_repository(store, props, config):
103
99
    """Creates on disk repository for the given group
 
100
    @param config: An ivle.config.Config object.
104
101
    Expected properties:
105
102
        subj_short_name, year, semester, groupnm
106
 
    Return value:
107
 
        response (okay, failure)
 
103
    @return: response (okay, failure)
108
104
    """
109
105
 
110
106
    subj_short_name = props['subj_short_name']
113
109
    groupnm = props['groupnm']
114
110
 
115
111
    namespace = "_".join([subj_short_name, year, semester, groupnm])
116
 
    repopath = os.path.join(conf.svn_repo_path, 'groups', namespace)
 
112
    repopath = os.path.join(config['paths']['svn']['repo_path'],
 
113
                            'groups', namespace)
117
114
    logging.debug("Creating Subversion repository %s"%repopath)
118
115
    try:
119
 
        common.makeuser.make_svn_repo(repopath)
 
116
        ivle.makeuser.make_svn_repo(repopath)
120
117
    except Exception, e:
121
118
        logging.error("Failed to create Subversion repository %s: %s"%
122
119
            (repopath,repr(e)))
132
129
    }
133
130
 
134
131
def initializer():
 
132
    logging.basicConfig(filename="/var/log/usrmgt.log", level=logging.INFO)
 
133
    logging.info("Starting usrmgt server on port %d (pid = %d)" %
 
134
                 (config['usrmgt']['port'], pid))
 
135
 
135
136
    try:
136
137
        pidfile = open('/var/run/usrmgt-server.pid', 'w')
137
138
        pidfile.write('%d\n' % os.getpid())
142
143
 
143
144
def dispatch(props):
144
145
    logging.debug(repr(props))
 
146
 
 
147
    store = ivle.database.get_store(config)
145
148
    action = props.keys()[0]
146
 
    return actions[action](props[action])
 
149
    res = actions[action](store, props[action], config)
 
150
 
 
151
    if res['response'] == 'okay':
 
152
        store.commit()
 
153
    else:
 
154
        store.rollback()
 
155
    store.close()
 
156
    return res
147
157
 
148
158
if __name__ == "__main__":
149
159
    pid = os.getpid()
150
160
 
151
 
    logging.basicConfig(filename="/var/log/usrmgt.log", level=logging.INFO)
152
 
    logging.info("Starting usrmgt server on port %d (pid = %d)" %
153
 
                 (conf.usrmgt_port, pid))
154
 
 
155
 
    common.chat.start_server(conf.usrmgt_port, conf.usrmgt_magic, True, dispatch, initializer)
 
161
    ivle.chat.start_server(config['usrmgt']['port'],config['usrmgt']['magic'],
 
162
                           True, dispatch, initializer)