23
23
# This is a command-line application, for use by the administrator.
24
# This program is a frontend for the modules in the setup packages that
25
# build and install IVLE in separate steps.
24
# This program configures, builds and installs IVLE in three separate steps.
26
25
# It is called with at least one argument, which specifies which operation to
28
# setup.py listmake (for developer use only)
29
# Recurses through the source tree and builds a list of all files which should
30
# be copied upon installation. This should be run by the developer before
31
# cutting a distribution, and the listfile it generates should be included in
32
# the distribution, avoiding the administrator having to run it.
34
# setup.py config [args]
35
# Configures IVLE with machine-specific details, most notably, various paths.
36
# Either prompts the administrator for these details or accepts them as
38
# Creates lib/conf/conf.py and trampoline/conf.h.
41
# Compiles all files and sets up a jail template in the source directory.
43
# Compiles (GCC) trampoline/trampoline.c to trampoline/trampoline.
45
# Creates standard subdirs inside the jail, eg bin, opt, home, tmp.
46
# Copies console/ to a location within the jail.
47
# Copies OS programs and files to corresponding locations within the jail
48
# (eg. python and Python libs, ld.so, etc).
49
# Generates .pyc files for all the IVLE .py files.
51
# setup.py install [--nojail] [--dry|n]
53
# Create target install directory ($target).
55
# Copy trampoline/trampoline to $target/bin.
56
# chown and chmod the installed trampoline.
57
# Copy www/ to $target.
58
# Copy jail/ to jails template directory (unless --nojail specified).
71
# Import modules from the website is tricky since they're in the www
73
sys.path.append(os.path.join(os.getcwd(), 'lib'))
75
import common.makeuser
77
# Determine which Python version (2.4 or 2.5, for example) we are running,
78
# and use that as the filename to the Python directory.
79
# Just get the first 3 characters of sys.version.
80
PYTHON_VERSION = sys.version[0:3]
82
# Operating system files to copy over into the jail.
83
# These will be copied from the given place on the OS file system into the
84
# same place within the jail.
87
'/lib/tls/i686/cmov/libc.so.6',
88
'/lib/tls/i686/cmov/libdl.so.2',
89
'/lib/tls/i686/cmov/libm.so.6',
90
'/lib/tls/i686/cmov/libpthread.so.0',
91
'/lib/tls/i686/cmov/libutil.so.1',
94
# These 2 files do not exist in Ubuntu
95
#'/etc/ld.so.preload',
96
#'/etc/ld.so.nohwcap',
102
'/usr/bin/python%s' % PYTHON_VERSION,
103
# Needed by fileservice
104
'/lib/libcom_err.so.2',
105
'/lib/libcrypt.so.1',
106
'/lib/libkeyutils.so.1',
107
'/lib/libresolv.so.2',
110
'/usr/lib/libapr-1.so.0',
111
'/usr/lib/libaprutil-1.so.0',
112
'/usr/lib/libdb-4.4.so',
113
'/usr/lib/libexpat.so.1',
114
'/usr/lib/libgcrypt.so.11',
115
'/usr/lib/libgnutls.so.13',
116
'/usr/lib/libgpg-error.so.0',
117
'/usr/lib/libgssapi_krb5.so.2',
118
'/usr/lib/libk5crypto.so.3',
119
'/usr/lib/libkrb5.so.3',
120
'/usr/lib/libkrb5support.so.0',
121
'/usr/lib/liblber.so.2',
122
'/usr/lib/libldap_r.so.2',
123
'/usr/lib/libneon.so.26',
124
'/usr/lib/libpq.so.5',
125
'/usr/lib/libsasl2.so.2',
126
'/usr/lib/libsqlite3.so.0',
127
'/usr/lib/libsvn_client-1.so.1',
128
'/usr/lib/libsvn_delta-1.so.1',
129
'/usr/lib/libsvn_diff-1.so.1',
130
'/usr/lib/libsvn_fs-1.so.1',
131
'/usr/lib/libsvn_fs_base-1.so.1',
132
'/usr/lib/libsvn_fs_fs-1.so.1',
133
'/usr/lib/libsvn_ra-1.so.1',
134
'/usr/lib/libsvn_ra_dav-1.so.1',
135
'/usr/lib/libsvn_ra_local-1.so.1',
136
'/usr/lib/libsvn_ra_svn-1.so.1',
137
'/usr/lib/libsvn_repos-1.so.1',
138
'/usr/lib/libsvn_subr-1.so.1',
139
'/usr/lib/libsvn_wc-1.so.1',
140
'/usr/lib/libtasn1.so.3',
141
'/usr/lib/libxml2.so.2',
142
# Needed by matplotlib
143
'/usr/lib/i686/cmov/libssl.so.0.9.8',
144
'/usr/lib/i686/cmov/libcrypto.so.0.9.8',
145
'/lib/tls/i686/cmov/libnsl.so.1',
146
'/usr/lib/libz.so.1',
147
'/usr/lib/atlas/liblapack.so.3',
148
'/usr/lib/atlas/libblas.so.3',
149
'/usr/lib/libg2c.so.0',
150
'/usr/lib/libstdc++.so.6',
151
'/usr/lib/libfreetype.so.6',
152
'/usr/lib/libpng12.so.0',
153
'/usr/lib/libBLT.2.4.so.8.4',
154
'/usr/lib/libtk8.4.so.0',
155
'/usr/lib/libtcl8.4.so.0',
156
'/usr/lib/tcl8.4/init.tcl',
157
'/usr/lib/libX11.so.6',
158
'/usr/lib/libXau.so.6',
159
'/usr/lib/libXdmcp.so.6',
160
'/lib/libgcc_s.so.1',
163
# Symlinks to make within the jail. Src mapped to dst.
165
'python%s' % PYTHON_VERSION: 'jail/usr/bin/python',
167
# Trees to copy. Src mapped to dst (these will be passed to action_copytree).
169
'/usr/lib/python%s' % PYTHON_VERSION:
170
'jail/usr/lib/python%s' % PYTHON_VERSION,
171
'/usr/share/matplotlib': 'jail/usr/share/matplotlib',
172
'/etc/ld.so.conf.d': 'jail/etc/ld.so.conf.d',
176
"""A configuration option; one of the things written to conf.py."""
177
def __init__(self, option_name, default, prompt, comment):
178
"""Creates a configuration option.
179
option_name: Name of the variable in conf.py. Also name of the
180
command-line argument to setup.py conf.
181
default: Default value for this variable.
182
prompt: (Short) string presented during the interactive prompt in
184
comment: (Long) comment string stored in conf.py. Each line of this
185
string should begin with a '#'.
187
self.option_name = option_name
188
self.default = default
190
self.comment = comment
192
# Configuration options, defaults and descriptions
194
config_options.append(ConfigOption("root_dir", "/",
195
"""Root directory where IVLE is located (in URL space):""",
197
# In URL space, where in the site is IVLE located. (All URLs will be prefixed
199
# eg. "/" or "/ivle"."""))
200
config_options.append(ConfigOption("ivle_install_dir", "/opt/ivle",
201
'Root directory where IVLE will be installed (on the local file '
204
# In the local file system, where IVLE is actually installed.
205
# This directory should contain the "www" and "bin" directories."""))
206
config_options.append(ConfigOption("jail_base", "/home/informatics/jails",
207
"""Root directory where the jails (containing user files) are stored
208
(on the local file system):""",
210
# In the local file system, where are the student/user file spaces located.
211
# The user jails are expected to be located immediately in subdirectories of
212
# this location."""))
213
config_options.append(ConfigOption("subjects_base",
214
"/home/informatics/subjects",
215
"""Root directory where the subject directories (containing worksheets
216
and other per-subject files) are stored (on the local file system):""",
218
# In the local file system, where are the per-subject file spaces located.
219
# The individual subject directories are expected to be located immediately
220
# in subdirectories of this location."""))
221
config_options.append(ConfigOption("exercises_base",
222
"/home/informatics/exercises",
223
"""Root directory where the exercise directories (containing
224
subject-independent exercise sheets) are stored (on the local file
227
# In the local file system, where are the subject-independent exercise sheet
228
# file spaces located."""))
229
config_options.append(ConfigOption("public_host", "public.localhost",
230
"""Hostname which will cause the server to go into "public mode",
231
providing login-free access to student's published work:""",
233
# The server goes into "public mode" if the browser sends a request with this
234
# host. This is for security reasons - we only serve public student files on a
235
# separate domain to the main IVLE site.
236
# Public mode does not use cookies, and serves only public content.
237
# Private mode (normal mode) requires login, and only serves files relevant to
238
# the logged-in user."""))
239
config_options.append(ConfigOption("allowed_uids", "33",
240
"""UID of the web server process which will run IVLE.
241
Only this user may execute the trampoline. May specify multiple users as
242
a comma-separated list.
245
# The User-ID of the web server process which will run IVLE, and any other
246
# users who are allowed to run the trampoline. This is stores as a string of
247
# comma-separated integers, simply because it is not used within Python, only
248
# used by the setup program to write to conf.h (see setup.py config)."""))
249
config_options.append(ConfigOption("db_host", "localhost",
250
"""PostgreSQL Database config
251
==========================
252
Hostname of the DB server:""",
254
### PostgreSQL Database config ###
255
# Database server hostname"""))
256
config_options.append(ConfigOption("db_port", "5432",
257
"""Port of the DB server:""",
259
# Database server port"""))
260
config_options.append(ConfigOption("db_dbname", "ivle",
261
"""Database name:""",
264
config_options.append(ConfigOption("db_user", "postgres",
265
"""Username for DB server login:""",
267
# Database username"""))
268
config_options.append(ConfigOption("db_password", "",
269
"""Password for DB server login:
270
(Caution: This password is stored in plaintext in lib/conf/conf.py)""",
272
# Database password"""))
273
config_options.append(ConfigOption("ldap_url", "ldaps://www.example.com",
274
"""URL for LDAP authentication server:""",
276
# URL for LDAP authentication server"""))
277
config_options.append(ConfigOption("ldap_format_string",
278
"uid=%s,ou=users,o=example",
279
"""Format string for LDAP auth request:
280
(Must contain a single "%s" for the user's login name)""",
282
# Format string for LDAP auth request
283
# (Must contain a single "%s" for the user's login name)"""))
284
config_options.append(ConfigOption("svn_addr", "http://svn.localhost/",
285
"""The base url for accessing subversion repositories:""",
287
# The base url for accessing subversion repositories."""))
288
config_options.append(ConfigOption("svn_conf", "/opt/ivle/svn/svn.conf",
289
"""The location of the subversion configuration file used by apache
290
to host the user repositories:""",
292
# The location of the subversion configuration file used by
293
# apache to host the user repositories."""))
294
config_options.append(ConfigOption("svn_repo_path", "/home/informatics/repositories",
295
"""The root directory for the subversion repositories:""",
297
# The root directory for the subversion repositories."""))
298
config_options.append(ConfigOption("svn_auth_ivle", "/opt/ivle/svn/ivle.auth",
299
"""The location of the password file used to authenticate users
300
of the subversion repository from the ivle server:""",
302
# The location of the password file used to authenticate users
303
# of the subversion repository from the ivle server."""))
304
config_options.append(ConfigOption("svn_auth_local", "/opt/ivle/svn/local.auth",
305
"""The location of the password file used to authenticate local users
306
of the subversion repository:""",
308
# The location of the password file used to authenticate local users
309
# of the subversion repository."""))
310
config_options.append(ConfigOption("usrmgt_host", "localhost",
311
"""The hostname where the usrmgt-server runs:""",
313
# The hostname where the usrmgt-server runs."""))
314
config_options.append(ConfigOption("usrmgt_port", "2178",
315
"""The port where the usrmgt-server runs:""",
317
# The port where the usrmgt-server runs."""))
318
config_options.append(ConfigOption("usrmgt_magic", "",
319
"""The password for the usrmgt-server:""",
321
# The password for the usrmgt-server."""))
323
# Try importing existing conf, but if we can't just set up defaults
324
# The reason for this is that these settings are used by other phases
325
# of setup besides conf, so we need to know them.
326
# Also this allows you to hit Return to accept the existing value.
328
confmodule = __import__("lib/conf/conf")
329
for opt in config_options:
331
globals()[opt.option_name] = confmodule.__dict__[opt.option_name]
333
globals()[opt.option_name] = opt.default
335
# Just set reasonable defaults
336
for opt in config_options:
337
globals()[opt.option_name] = opt.default
339
# Try importing install_list, but don't fail if we can't, because listmake can
340
# function without it.
346
# Mime types which will automatically be placed in the list by listmake.
347
# Note that listmake is not intended to be run by the final user (the system
348
# administrator who installs this), so the developers can customize the list
349
# as necessary, and include it in the distribution.
350
listmake_mimetypes = ['text/x-python', 'text/html',
351
'application/x-javascript', 'application/javascript',
352
'text/css', 'image/png', 'image/gif', 'image/jpeg', 'application/xml',
353
'application/x-httpd-php']
355
# Main function skeleton from Guido van Rossum
356
# http://www.artima.com/weblogs/viewpost.jsp?thread=4829
33
358
def main(argv=None):
56
oper_func = call_operator(operation)
57
return oper_func(argv[2:])
61
print """Usage: python setup.py operation [options]
67
For help and options for a specific operation use 'help [operation]'."""
70
oper_func = call_operator(operator)
71
oper_func(['operator','--help'])
73
def call_operator(operation):
381
# Disallow run as root unless installing
382
if (operation != 'install' and operation != 'updatejails'
383
and os.geteuid() == 0):
384
print >>sys.stderr, "I do not want to run this stage as root."
385
print >>sys.stderr, "Please run as a normal user."
74
387
# Call the requested operation's function
78
'build' : setup.build.build,
79
'install' : setup.install.install,
393
'listmake' : listmake,
395
'updatejails' : updatejails,
82
398
print >>sys.stderr, (
83
399
"""Invalid operation '%s'. Try python setup.py help."""
402
return oper_func(argv[2:])
404
# Operation functions
408
print """Usage: python setup.py operation [args]
409
Operation (and args) can be:
411
listmake (developer use only)
414
install [--nojail] [--nosubjects] [-n|--dry]
418
print """Usage: python setup.py help [operation]"""
423
if operation == 'help':
424
print """python setup.py help [operation]
425
Prints the usage message or detailed help on an operation, then exits."""
426
elif operation == 'listmake':
427
print """python setup.py listmake
428
(For developer use only)
429
Recurses through the source tree and builds a list of all files which should
430
be copied upon installation. This should be run by the developer before
431
cutting a distribution, and the listfile it generates should be included in
432
the distribution, avoiding the administrator having to run it."""
433
elif operation == 'config':
434
print """python setup.py config [args]
435
Configures IVLE with machine-specific details, most notably, various paths.
436
Either prompts the administrator for these details or accepts them as
437
command-line args. Will be interactive only if there are no arguments given.
438
Takes defaults from existing conf file if it exists.
440
To run IVLE out of the source directory (allowing development without having
441
to rebuild/install), just provide ivle_install_dir as the IVLE trunk
442
directory, and run build/install one time.
444
Creates lib/conf/conf.py and trampoline/conf.h.
447
for opt in config_options:
448
print " --" + opt.option_name
449
print """As explained in the interactive prompt or conf.py.
451
elif operation == 'build':
452
print """python -O setup.py build [--dry|-n]
453
Compiles all files and sets up a jail template in the source directory.
454
-O is recommended to cause compilation to be optimised.
456
Compiles (GCC) trampoline/trampoline.c to trampoline/trampoline.
458
Creates standard subdirs inside the jail, eg bin, opt, home, tmp.
459
Copies console/ to a location within the jail.
460
Copies OS programs and files to corresponding locations within the jail
461
(eg. python and Python libs, ld.so, etc).
462
Generates .pyc or .pyo files for all the IVLE .py files.
464
--dry | -n Print out the actions but don't do anything."""
465
elif operation == 'install':
466
print """sudo python setup.py install [--nojail] [--nosubjects][--dry|-n]
468
Create target install directory ($target).
470
Copy trampoline/trampoline to $target/bin.
471
chown and chmod the installed trampoline.
472
Copy www/ to $target.
473
Copy jail/ to jails template directory (unless --nojail specified).
474
Copy subjects/ to subjects directory (unless --nosubjects specified).
476
--nojail Do not copy the jail.
477
--nosubjects Do not copy the subjects and exercises directories.
478
--dry | -n Print out the actions but don't do anything."""
479
elif operation == 'updatejails':
480
print """sudo python setup.py updatejails [--dry|-n]
482
Copy jail/ to each subdirectory in jails directory.
484
--dry | -n Print out the actions but don't do anything."""
486
print >>sys.stderr, (
487
"""Invalid operation '%s'. Try python setup.py help."""
492
# We build two separate lists, by walking www and console
493
list_www = build_list_py_files('www')
494
list_lib = build_list_py_files('lib')
495
list_subjects = build_list_py_files('subjects', no_top_level=True)
496
list_exercises = build_list_py_files('exercises', no_top_level=True)
498
"scripts/python-console",
499
"scripts/fileservice",
500
"scripts/usrmgt-server",
502
# Make sure that the files generated by conf are in the list
503
# (since listmake is typically run before conf)
504
if "lib/conf/conf.py" not in list_lib:
505
list_lib.append("lib/conf/conf.py")
506
# Write these out to a file
508
# the files that will be created/overwritten
509
listfile = os.path.join(cwd, "install_list.py")
512
file = open(listfile, "w")
514
file.write("""# IVLE Configuration File
516
# Provides lists of all files to be installed by `setup.py install' from
517
# certain directories.
518
# Note that any files with the given filename plus 'c' or 'o' (that is,
519
# compiled .pyc or .pyo files) will be copied as well.
521
# List of all installable files in www directory.
523
writelist_pretty(file, list_www)
525
# List of all installable files in lib directory.
527
writelist_pretty(file, list_lib)
529
# List of all installable files in scripts directory.
531
writelist_pretty(file, list_scripts)
533
# List of all installable files in subjects directory.
534
# This is to install sample subjects and material.
536
writelist_pretty(file, list_subjects)
538
# List of all installable files in exercises directory.
539
# This is to install sample exercise material.
540
list_exercises = """)
541
writelist_pretty(file, list_exercises)
544
except IOError, (errno, strerror):
545
print "IO error(%s): %s" % (errno, strerror)
548
print "Successfully wrote install_list.py"
551
print ("You may modify the set of installable files before cutting the "
558
def build_list_py_files(dir, no_top_level=False):
559
"""Builds a list of all py files found in a directory and its
560
subdirectories. Returns this as a list of strings.
561
no_top_level=True means the file paths will not include the top-level
565
for (dirpath, dirnames, filenames) in os.walk(dir):
566
# Exclude directories beginning with a '.' (such as '.svn')
567
filter_mutate(lambda x: x[0] != '.', dirnames)
568
# All *.py files are added to the list
569
pylist += [os.path.join(dirpath, item) for item in filenames
570
if mimetypes.guess_type(item)[0] in listmake_mimetypes]
572
for i in range(0, len(pylist)):
573
_, pylist[i] = pylist[i].split(os.sep, 1)
576
def writelist_pretty(file, list):
577
"""Writes a list one element per line, to a file."""
583
file.write(' %s,\n' % repr(elem))
587
global db_port, usrmgt_port
588
# Set up some variables
591
# the files that will be created/overwritten
592
conffile = os.path.join(cwd, "lib/conf/conf.py")
593
jailconffile = os.path.join(cwd, "lib/conf/jailconf.py")
594
conf_hfile = os.path.join(cwd, "trampoline/conf.h")
596
# Get command-line arguments to avoid asking questions.
599
for opt in config_options:
600
optnames.append(opt.option_name + "=")
601
(opts, args) = getopt.gnu_getopt(args, "", optnames)
604
print >>sys.stderr, "Invalid arguments:", string.join(args, ' ')
608
# Interactive mode. Prompt the user for all the values.
610
print """This tool will create the following files:
614
prompting you for details about your configuration. The file will be
615
overwritten if it already exists. It will *not* install or deploy IVLE.
617
Please hit Ctrl+C now if you do not wish to do this.
618
""" % (conffile, jailconffile, conf_hfile)
620
# Get information from the administrator
621
# If EOF is encountered at any time during the questioning, just exit
624
for opt in config_options:
625
globals()[opt.option_name] = \
626
query_user(globals()[opt.option_name], opt.prompt)
629
# Non-interactive mode. Parse the options.
630
for opt in config_options:
631
if '--' + opt.option_name in opts:
632
globals()[opt.option_name] = opts['--' + opt.option_name]
634
# Error handling on input values
636
allowed_uids_list = map(int, allowed_uids.split(','))
638
print >>sys.stderr, (
639
"Invalid UID list (%s).\n"
640
"Must be a comma-separated list of integers." % allowed_uids)
643
db_port = int(db_port)
644
if db_port < 0 or db_port >= 65536: raise ValueError()
646
print >>sys.stderr, (
647
"Invalid DB port (%s).\n"
648
"Must be an integer between 0 and 65535." % repr(db_port))
651
usrmgt_port = int(usrmgt_port)
652
if usrmgt_port < 0 or usrmgt_port >= 65536: raise ValueError()
654
print >>sys.stderr, (
655
"Invalid user management port (%s).\n"
656
"Must be an integer between 0 and 65535." % repr(usrmgt_port))
659
# Write lib/conf/conf.py
662
conf = open(conffile, "w")
664
conf.write("""# IVLE Configuration File
666
# Miscellaneous application settings
669
for opt in config_options:
670
conf.write('%s\n%s = %s\n' % (opt.comment, opt.option_name,
671
repr(globals()[opt.option_name])))
674
except IOError, (errno, strerror):
675
print "IO error(%s): %s" % (errno, strerror)
678
print "Successfully wrote lib/conf/conf.py"
680
# Write conf/jailconf.py
683
conf = open(jailconffile, "w")
685
# In the "in-jail" version of conf, we don't need MOST of the details
686
# (it would be a security risk to have them here).
687
# So we just write root_dir, and jail_base is "/".
688
# (jail_base being "/" means "jail-relative" paths are relative to "/"
689
# when inside the jail.)
690
conf.write("""# IVLE Configuration File
692
# Miscellaneous application settings
693
# (User jail version)
696
# In URL space, where in the site is IVLE located. (All URLs will be prefixed
698
# eg. "/" or "/ivle".
701
# In the local file system, where are the student/user file spaces located.
702
# The user jails are expected to be located immediately in subdirectories of
706
# The hostname for serving publicly accessible pages
708
""" % (repr(root_dir),repr(public_host)))
711
except IOError, (errno, strerror):
712
print "IO error(%s): %s" % (errno, strerror)
715
print "Successfully wrote lib/conf/jailconf.py"
717
# Write trampoline/conf.h
720
conf = open(conf_hfile, "w")
722
conf.write("""/* IVLE Configuration File
724
* Administrator settings required by trampoline.
725
* Note: trampoline will have to be rebuilt in order for changes to this file
729
/* In the local file system, where are the jails located.
730
* The trampoline does not allow the creation of a jail anywhere besides
731
* jail_base or a subdirectory of jail_base.
733
static const char* jail_base = "%s";
735
/* Which user IDs are allowed to run the trampoline.
736
* This list should be limited to the web server user.
737
* (Note that root is an implicit member of this list).
739
static const int allowed_uids[] = { %s };
740
""" % (repr(jail_base)[1:-1], repr(allowed_uids_list)[1:-1]))
741
# Note: The above uses PYTHON reprs, not C reprs
742
# However they should be the same with the exception of the outer
743
# characters, which are stripped off and replaced
746
except IOError, (errno, strerror):
747
print "IO error(%s): %s" % (errno, strerror)
750
print "Successfully wrote trampoline/conf.h"
753
print "You may modify the configuration at any time by editing"
761
# Get "dry" variable from command line
762
(opts, args) = getopt.gnu_getopt(args, "n", ['dry'])
764
dry = '-n' in opts or '--dry' in opts
767
print "Dry run (no actions will be executed\n"
769
# Compile the trampoline
771
os.chdir('trampoline')
772
action_runprog('make', [], dry)
775
# Create the jail and its subdirectories
776
# Note: Other subdirs will be made by copying files
777
action_mkdir('jail', dry)
778
action_mkdir('jail/home', dry)
779
action_mkdir('jail/tmp', dry)
781
# Copy all console and operating system files into the jail
782
action_copylist(install_list.list_scripts, 'jail/opt/ivle', dry)
783
copy_os_files_jail(dry)
784
# Chmod the python console
785
action_chmod_x('jail/opt/ivle/scripts/python-console', dry)
786
action_chmod_x('jail/opt/ivle/scripts/fileservice', dry)
788
# Also copy the IVLE lib directory into the jail
789
# This is necessary for running certain scripts
790
action_copylist(install_list.list_lib, 'jail/opt/ivle', dry)
791
# IMPORTANT: The file jail/opt/ivle/lib/conf/conf.py contains details
792
# which could compromise security if left in the jail (such as the DB
794
# The "safe" version is in jailconf.py. Delete conf.py and replace it with
796
action_copyfile('lib/conf/jailconf.py',
797
'jail/opt/ivle/lib/conf/conf.py', dry)
799
# Compile .py files into .pyc or .pyo files
800
compileall.compile_dir('www', quiet=True)
801
compileall.compile_dir('lib', quiet=True)
802
compileall.compile_dir('scripts', quiet=True)
803
compileall.compile_dir('jail/opt/ivle/lib', quiet=True)
805
# Set up ivle.pth inside the jail
806
# Need to set /opt/ivle/lib to be on the import path
808
"jail/usr/lib/python%s/site-packages/ivle.pth" % PYTHON_VERSION
809
f = open(ivle_pth, 'w')
810
f.write('/opt/ivle/lib\n')
815
def copy_os_files_jail(dry):
816
"""Copies necessary Operating System files from their usual locations
817
into the jail/ directory of the cwd."""
818
# Currently source paths are configured for Ubuntu.
819
for filename in JAIL_FILES:
820
copy_file_to_jail(filename, dry)
821
for src, dst in JAIL_LINKS.items():
822
action_symlink(src, dst, dry)
823
for src, dst in JAIL_COPYTREES.items():
824
action_copytree(src, dst, dry)
826
def copy_file_to_jail(src, dry):
827
"""Copies a single file from an absolute location into the same location
828
within the jail. src must begin with a '/'. The jail will be located
829
in a 'jail' subdirectory of the current path."""
830
action_copyfile(src, 'jail' + src, dry)
833
# Get "dry" and "nojail" variables from command line
834
(opts, args) = getopt.gnu_getopt(args, "n",
835
['dry', 'nojail', 'nosubjects'])
837
dry = '-n' in opts or '--dry' in opts
838
nojail = '--nojail' in opts
839
nosubjects = '--nosubjects' in opts
842
print "Dry run (no actions will be executed\n"
844
if not dry and os.geteuid() != 0:
845
print >>sys.stderr, "Must be root to run install"
846
print >>sys.stderr, "(I need to chown some files)."
849
# Create the target (install) directory
850
action_mkdir(ivle_install_dir, dry)
852
# Create bin and copy the compiled files there
853
action_mkdir(os.path.join(ivle_install_dir, 'bin'), dry)
854
tramppath = os.path.join(ivle_install_dir, 'bin/trampoline')
855
action_copyfile('trampoline/trampoline', tramppath, dry)
856
# chown trampoline to root and set setuid bit
857
action_chown_setuid(tramppath, dry)
859
# Copy the www and lib directories using the list
860
action_copylist(install_list.list_www, ivle_install_dir, dry)
861
action_copylist(install_list.list_lib, ivle_install_dir, dry)
864
# Copy the local jail directory built by the build action
865
# to the jails template directory (it will be used as a template
866
# for all the students' jails).
867
action_copytree('jail', os.path.join(jail_base, 'template'), dry)
869
# Copy the subjects and exercises directories across
870
action_copylist(install_list.list_subjects, subjects_base, dry,
872
action_copylist(install_list.list_exercises, exercises_base, dry,
873
srcdir="./exercises")
875
# Append IVLE path to ivle.pth in python site packages
876
# (Unless it's already there)
877
ivle_pth = os.path.join(sys.prefix,
878
"lib/python%s/site-packages/ivle.pth" % PYTHON_VERSION)
879
ivle_www = os.path.join(ivle_install_dir, "www")
880
ivle_lib = os.path.join(ivle_install_dir, "lib")
881
write_ivle_pth = True
882
write_ivle_lib_pth = True
884
file = open(ivle_pth, 'r')
886
if line.strip() == ivle_www:
887
write_ivle_pth = False
888
elif line.strip() == ivle_lib:
889
write_ivle_lib_pth = False
891
except (IOError, OSError):
894
action_append(ivle_pth, ivle_www)
895
if write_ivle_lib_pth:
896
action_append(ivle_pth, ivle_lib)
900
def updatejails(args):
901
# Get "dry" variable from command line
902
(opts, args) = getopt.gnu_getopt(args, "n", ['dry'])
904
dry = '-n' in opts or '--dry' in opts
907
print "Dry run (no actions will be executed\n"
909
if not dry and os.geteuid() != 0:
910
print >>sys.stderr, "Must be root to run install"
911
print >>sys.stderr, "(I need to chown some files)."
914
# Update the template jail directory in case it hasn't been installed
916
action_copytree('jail', os.path.join(jail_base, 'template'), dry)
918
# Re-link all the files in all students jails.
919
for dir in os.listdir(jail_base):
920
if dir == 'template': continue
921
# First back up the student's home directory
922
temp_home = os.tmpnam()
923
action_rename(os.path.join(jail_base, dir, 'home'), temp_home, dry)
924
# Delete the student's jail and relink the jail files
925
action_linktree(os.path.join(jail_base, 'template'),
926
os.path.join(jail_base, dir), dry)
927
# Restore the student's home directory
928
action_rename(temp_home, os.path.join(jail_base, dir, 'home'), dry)
929
# Set up the user's home directory just in case they don't have a
930
# directory for this yet
931
action_mkdir(os.path.join(jail_base, dir, 'home', dir), dry)
935
# The actions call Python os functions but print actions and handle dryness.
936
# May still throw os exceptions if errors occur.
939
"""Represents an error when running a program (nonzero return)."""
940
def __init__(self, prog, retcode):
942
self.retcode = retcode
944
return str(self.prog) + " returned " + repr(self.retcode)
946
def action_runprog(prog, args, dry):
947
"""Runs a unix program. Searches in $PATH. Synchronous (waits for the
948
program to return). Runs in the current environment. First prints the
949
action as a "bash" line.
951
Throws a RunError with a retcode of the return value of the program,
952
if the program did not return 0.
954
prog: String. Name of the program. (No path required, if in $PATH).
955
args: [String]. Arguments to the program.
956
dry: Bool. If True, prints but does not execute.
958
print prog, string.join(args, ' ')
960
ret = os.spawnvp(os.P_WAIT, prog, args)
962
raise RunError(prog, ret)
964
def action_remove(path, dry):
965
"""Calls rmtree, deleting the target file if it exists."""
969
shutil.rmtree(path, True)
970
except OSError, (err, msg):
971
if err != errno.EEXIST:
973
# Otherwise, didn't exist, so we don't care
975
def action_rename(src, dst, dry):
976
"""Calls rename. Deletes the target if it already exists."""
977
action_remove(dst, dry)
978
print "mv ", src, dst
982
except OSError, (err, msg):
983
if err != errno.EEXIST:
986
def action_mkdir(path, dry):
987
"""Calls mkdir. Silently ignored if the directory already exists.
988
Creates all parent directories as necessary."""
989
print "mkdir -p", path
993
except OSError, (err, msg):
994
if err != errno.EEXIST:
997
def action_copytree(src, dst, dry):
998
"""Copies an entire directory tree. Symlinks are seen as normal files and
999
copies of the entire file (not the link) are made. Creates all parent
1000
directories as necessary.
1002
See shutil.copytree."""
1003
action_remove(dst, dry)
1004
print "cp -r", src, dst
1006
shutil.copytree(src, dst, True)
1008
def action_linktree(src, dst, dry):
1009
"""Hard-links an entire directory tree. Same as copytree but the created
1010
files are hard-links not actual copies. Removes the existing destination.
1012
action_remove(dst, dry)
1013
print "<cp with hardlinks> -r", src, dst
1015
common.makeuser.linktree(src, dst)
1017
def action_copylist(srclist, dst, dry, srcdir="."):
1018
"""Copies all files in a list to a new location. The files in the list
1019
are read relative to the current directory, and their destinations are the
1020
same paths relative to dst. Creates all parent directories as necessary.
1021
srcdir is "." by default, can be overridden.
1023
for srcfile in srclist:
1024
dstfile = os.path.join(dst, srcfile)
1025
srcfile = os.path.join(srcdir, srcfile)
1026
dstdir = os.path.split(dstfile)[0]
1027
if not os.path.isdir(dstdir):
1028
action_mkdir(dstdir, dry)
1029
print "cp -f", srcfile, dstfile
1032
shutil.copyfile(srcfile, dstfile)
1033
shutil.copymode(srcfile, dstfile)
1034
except shutil.Error:
1037
def action_copyfile(src, dst, dry):
1038
"""Copies one file to a new location. Creates all parent directories
1040
Warn if file not found.
1042
dstdir = os.path.split(dst)[0]
1043
if not os.path.isdir(dstdir):
1044
action_mkdir(dstdir, dry)
1045
print "cp -f", src, dst
1048
shutil.copyfile(src, dst)
1049
shutil.copymode(src, dst)
1050
except (shutil.Error, IOError), e:
1051
print "Warning: " + str(e)
1053
def action_symlink(src, dst, dry):
1054
"""Creates a symlink in a given location. Creates all parent directories
1057
dstdir = os.path.split(dst)[0]
1058
if not os.path.isdir(dstdir):
1059
action_mkdir(dstdir, dry)
1060
# Delete existing file
1061
if os.path.exists(dst):
1063
print "ln -fs", src, dst
1065
os.symlink(src, dst)
1067
def action_append(ivle_pth, ivle_www):
1068
file = open(ivle_pth, 'a+')
1069
file.write(ivle_www + '\n')
1072
def action_chown_setuid(file, dry):
1073
"""Chowns a file to root, and sets the setuid bit on the file.
1074
Calling this function requires the euid to be root.
1075
The actual mode of path is set to: rws--s--s
1077
print "chown root:root", file
1079
os.chown(file, 0, 0)
1080
print "chmod a+xs", file
1081
print "chmod u+rw", file
1083
os.chmod(file, stat.S_IXUSR | stat.S_IXGRP | stat.S_IXOTH
1084
| stat.S_ISUID | stat.S_IRUSR | stat.S_IWUSR)
1086
def action_chmod_x(file, dry):
1087
"""Chmod 755 a file (sets permissions to rwxr-xr-x)."""
1088
print "chmod 755", file
1090
os.chmod(file, stat.S_IXUSR | stat.S_IRUSR | stat.S_IWUSR
1091
| stat.S_IXGRP | stat.S_IRGRP | stat.S_IXOTH | stat.S_IROTH)
1093
def query_user(default, prompt):
1094
"""Prompts the user for a string, which is read from a line of stdin.
1095
Exits silently if EOF is encountered. Returns the string, with spaces
1096
removed from the beginning and end.
1098
Returns default if a 0-length line (after spaces removed) was read.
1100
sys.stdout.write('%s\n (default: "%s")\n>' % (prompt, default))
1102
val = sys.stdin.readline()
1103
except KeyboardInterrupt:
1105
sys.stdout.write("\n")
1107
sys.stdout.write("\n")
1109
if val == '': sys.exit(1)
1110
# If empty line, return default
1112
if val == '': return default
1115
def filter_mutate(function, list):
1116
"""Like built-in filter, but mutates the given list instead of returning a
1117
new one. Returns None."""
1120
# Delete elements which do not match
1121
if not function(list[i]):
88
1125
if __name__ == "__main__":