← Back to branch summary

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

« back to all changes in this revision

Viewing changes to setup/install.py

  • Committer: William Grant
  • Date: 2008-07-07 04:55:07 UTC
  • mfrom: (unknown (missing))
  • Revision ID: wgrant@ugrad.unimelb.edu.au-20080707045507-wca484wpjxppy7ln
Merge jails-redux branch. We now use aufs rather than hardlinking tens
of thousands of jail files hundreds of times.

Reconfiguration and reconstruction of all jails will be required.
Probably a good idea to start jails from scratch if possible.

Show diffs side-by-side

added added

removed removed

Lines of Context:
setup/install.py
19
19
# Author: Matt Giuca, Refactored by David Coles
20
20
# Date:   03/07/2008
21
21
 
 
22
# setup/build.py
 
23
# Compiles all files and sets up a jail template in the source directory.
 
24
# Details:
 
25
# Compiles (GCC) trampoline/trampoline.c to trampoline/trampoline.
 
26
# Creates jail/.
 
27
# Creates standard subdirs inside the jail, eg bin, opt, home, tmp.
 
28
# Copies console/ to a location within the jail.
 
29
# Copies OS programs and files to corresponding locations within the jail
 
30
#   (eg. python and Python libs, ld.so, etc).
 
31
# Generates .pyc files for all the IVLE .py files.
 
32
 
22
33
import optparse
23
34
import os
24
 
import sys
25
 
import functools
26
 
import distutils.sysconfig
27
 
 
28
 
from setup import util
 
35
from setuputil import *
29
36
 
30
37
def install(args):
31
38
    usage = """usage: %prog install [options]
32
39
(Requires root)
33
40
Create target install directory ($target).
34
41
Create $target/bin.
35
 
Copy bin/trampoline/trampoline to $target/bin.
36
 
Copy bin/timount/timount to $target/bin.
 
42
Copy trampoline/trampoline to $target/bin.
 
43
Copy timount/timount to $target/bin.
37
44
chown and chmod the installed trampoline.
 
45
Copy www/ to $target.
 
46
Copy jail/ to jail_system directory (unless --nojail specified).
 
47
Copy subjects/ to subjects directory (unless --nosubjects specified).
38
48
"""
39
49
 
40
50
    # Parse arguments
42
52
    parser.add_option("-n", "--dry",
43
53
        action="store_true", dest="dry",
44
54
        help="Print out the actions but don't do anything.")
45
 
    parser.add_option("--root",
46
 
        action="store", dest="rootdir",
47
 
        help="Install into a different root directory.",
48
 
        default='/')
49
 
    parser.add_option("--prefix",
50
 
        action="store", dest="prefix",
51
 
        help="Base prefix to install IVLE into (default: /usr/local).",
52
 
        default='/usr/local')
53
 
    parser.add_option("--python-site-packages",
54
 
        action="store", dest="python_site_packages",
55
 
        help="Path to Python site packages directory.",
56
 
        default=None)
 
55
    parser.add_option("-J", "--nojail",
 
56
        action="store_true", dest="nojail",
 
57
        help="Don't copy jail/ to jail_system directory")
 
58
    parser.add_option("-S", "--nosubjects",
 
59
        action="store_true", dest="nosubjects",
 
60
        help="Don't copy subject/ to subjects directory.")
57
61
    (options, args) = parser.parse_args(args)
58
62
 
59
 
    # Prefix must be absolute (not really necessary, but since a relative
60
 
    # prefix will be taken relative to *root* not working directory, it is
61
 
    # confusing if we allow it).
62
 
    if options.prefix[:1] not in (os.path.sep, os.path.altsep):
63
 
        print >>sys.stderr, """prefix must be an absolute path.
64
 
    (This will be interpreted relative to root, so provide --root=. if you
65
 
    want a path relative to the working directory)."""
66
 
        return 1
67
 
 
68
 
    # Calculate python_site_packages using the supplied prefix
69
 
    if options.python_site_packages is None:
70
 
        options.python_site_packages = distutils.sysconfig.get_python_lib(
71
 
            prefix=options.prefix)
72
 
 
73
63
    # Call the real function
74
 
    return __install(prefix=options.prefix,
75
 
                     python_site_packages=options.python_site_packages,
76
 
                     dry=options.dry, rootdir=options.rootdir)
77
 
 
78
 
def __install(prefix, python_site_packages, dry=False, rootdir=None):
79
 
    install_list = util.InstallList()
80
 
 
81
 
    # We need to apply make_install_path with the rootdir to an awful lot of
82
 
    # config variables, so make it easy:
83
 
    mip = functools.partial(util.make_install_path, rootdir)
84
 
 
85
 
    # Compute the lib_path, share_path and bin_path (copied from
86
 
    # ivle/conf/conf.py).
87
 
    lib_path = mip(os.path.join(prefix, 'lib/ivle'))
88
 
    share_path = mip(os.path.join(prefix, 'share/ivle'))
89
 
    bin_path = mip(os.path.join(prefix, 'bin'))
90
 
    python_site_packages = mip(python_site_packages)
 
64
    __install(options.dry, options.nojail, options.nosubjects)
 
65
 
 
66
def __install(dry=False,nojail=False,nosubjects=False):
 
67
    # Importing configuration is a little tricky
 
68
    sys.path.append(os.path.join(os.pardir, 'lib'))
 
69
    sys.path.append(os.pardir)
 
70
    import conf.conf
 
71
    import install_list
 
72
 
 
73
    # Pull the required varibles out of the config
 
74
    ivle_install_dir = conf.conf.ivle_install_dir
 
75
    jail_base = conf.conf.jail_base
 
76
    jail_system = conf.conf.jail_system
 
77
    subjects_base = conf.conf.subjects_base
 
78
    exercises_base = conf.conf.exercises_base
91
79
 
92
80
    # Must be run as root or a dry run  
93
81
    if dry:
94
82
        print "Dry run (no actions will be executed)\n"
95
83
    
96
84
    if not dry and os.geteuid() != 0:
97
 
        print >>sys.stderr, "Must be root to run install"
 
85
        print >>sys.stderr, "Must be root to run build"
 
86
        print >>sys.stderr, "(I need to chown)."
98
87
        return 1
99
 
 
100
 
    # Create the config directory.
101
 
    util.action_mkdir(mip('/etc/ivle/plugins.d'), dry)
102
 
 
103
 
    # Create lib and copy the compiled files there
104
 
    util.action_mkdir(lib_path, dry)
105
 
 
106
 
    tramppath = os.path.join(lib_path, 'trampoline')
107
 
    util.action_copyfile('bin/trampoline/trampoline', tramppath, dry)
 
88
    
 
89
    # Create the target (install) directory
 
90
    action_mkdir(ivle_install_dir, dry)
 
91
 
 
92
    # Create bin and copy the compiled files there
 
93
    action_mkdir(os.path.join(ivle_install_dir, 'bin'), dry)
 
94
    tramppath = os.path.join(ivle_install_dir, 'bin/trampoline')
 
95
    action_copyfile('trampoline/trampoline', tramppath, dry)
108
96
    # chown trampoline to root and set setuid bit
109
 
    util.action_chown_setuid(tramppath, dry)
110
 
 
111
 
    timountpath = os.path.join(lib_path, 'timount')
112
 
    util.action_copyfile('bin/timount/timount', timountpath, dry)
113
 
 
114
 
    # Copy in the services (only usrmgt-server is needed on the host, but
115
 
    # the jail build requires the rest).
116
 
    util.action_copylist(install_list.list_services, share_path, dry)
117
 
    usrmgtpath = os.path.join(share_path, 'services/usrmgt-server')
118
 
    util.action_chmod_x(usrmgtpath, dry)
119
 
 
120
 
    # Copy the user-executable binaries using the list.
121
 
    util.action_copylist(install_list.list_user_binaries, bin_path, dry,
122
 
                         onlybasename=True)
123
 
 
124
 
    # Copy the lib directory (using the list)
125
 
    util.action_copylist(install_list.list_ivle_lib, python_site_packages, dry)
 
97
    action_chown_setuid(tramppath, dry)
 
98
 
 
99
    timountpath = os.path.join(ivle_install_dir, 'bin/timount')
 
100
    action_copyfile('timount/timount', timountpath, dry)
 
101
 
 
102
    # Create a scripts directory to put the usrmgt-server in.
 
103
    action_mkdir(os.path.join(ivle_install_dir, 'scripts'), dry)
 
104
    usrmgtpath = os.path.join(ivle_install_dir, 'scripts/usrmgt-server')
 
105
    action_copyfile('scripts/usrmgt-server', usrmgtpath, dry)
 
106
    action_chmod_x(usrmgtpath, dry)
 
107
 
 
108
    # Copy the www and lib directories using the list
 
109
    action_copylist(install_list.list_www, ivle_install_dir, dry)
 
110
    action_copylist(install_list.list_lib, ivle_install_dir, dry)
 
111
    
 
112
    # Copy the php directory
 
113
    forum_dir = "www/php/phpBB3"
 
114
    forum_path = os.path.join(ivle_install_dir, forum_dir)
 
115
    action_copytree(forum_dir, forum_path, dry)
 
116
    print "chown -R www-data:www-data %s" % forum_path
 
117
    if not dry:
 
118
        os.system("chown -R www-data:www-data %s" % forum_path)
 
119
 
 
120
    if not nojail:
 
121
        # Copy the local jail directory built by the build action
 
122
        # to the jail_system directory (it will be used to help build
 
123
        # all the students' jails).
 
124
        action_copytree('jail', jail_system, dry)
 
125
    if not nosubjects:
 
126
        # Copy the subjects and exercises directories across
 
127
        action_copylist(install_list.list_subjects, subjects_base, dry,
 
128
            srcdir="./subjects")
 
129
        action_copylist(install_list.list_exercises, exercises_base, dry,
 
130
            srcdir="./exercises")
 
131
 
 
132
    # Append IVLE path to ivle.pth in python site packages
 
133
    # (Unless it's already there)
 
134
    ivle_pth = os.path.join(sys.prefix,
 
135
        "lib/python%s/site-packages/ivle.pth" % PYTHON_VERSION)
 
136
    ivle_www = os.path.join(ivle_install_dir, "www")
 
137
    ivle_lib = os.path.join(ivle_install_dir, "lib")
 
138
    write_ivle_pth = True
 
139
    write_ivle_lib_pth = True
 
140
    try:
 
141
        file = open(ivle_pth, 'r')
 
142
        for line in file:
 
143
            if line.strip() == ivle_www:
 
144
                write_ivle_pth = False
 
145
            elif line.strip() == ivle_lib:
 
146
                write_ivle_lib_pth = False
 
147
        file.close()
 
148
    except (IOError, OSError):
 
149
        pass
 
150
    if write_ivle_pth:
 
151
        action_append(ivle_pth, ivle_www)
 
152
    if write_ivle_lib_pth:
 
153
        action_append(ivle_pth, ivle_lib)
 
154
 
 
155
 
 
156
    # Create the ivle working revision record file
 
157
    action_mkdir(os.path.join(ivle_install_dir, 'version'), dry)
 
158
    ivle_revision_record_file = os.path.join(ivle_install_dir, 'version/ivle-revision.txt')
 
159
    if not dry:
 
160
        try:
 
161
            conf = open(ivle_revision_record_file, "w")
 
162
 
 
163
            conf.write( "# IVLE code revision listing generated by running 'svn status -v ..' from " + os.getcwd() + "\n#\n\n")
 
164
 
 
165
            conf.close()
 
166
        except IOError, (errno, strerror):
 
167
            print "IO error(%s): %s" % (errno, strerror)
 
168
            sys.exit(1)
 
169
 
 
170
        os.system("svn status -v . >> %s" % ivle_revision_record_file)
 
171
 
 
172
    print "Wrote IVLE code revision status to %s" % ivle_revision_record_file
126
173
 
127
174
    return 0
128
175