← Back to branch summary

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

« back to all changes in this revision

Viewing changes to bin/ivle-buildjail

  • Committer: David Coles
  • Date: 2009-08-13 21:28:32 UTC
  • mto: This revision was merged to the branch mainline in revision 1318.
  • Revision ID: coles.david@gmail.com-20090813212832-o3dq367fwa0d2y75
Set project.deadline to NOT NULL since it is required for submits.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bin/ivle-buildjail
 
1
#!/usr/bin/python
 
2
# IVLE - Informatics Virtual Learning Environment
 
3
# Copyright (C) 2009 The University of Melbourne
 
4
#
 
5
# This program is free software; you can redistribute it and/or modify
 
6
# it under the terms of the GNU General Public License as published by
 
7
# the Free Software Foundation; either version 2 of the License, or
 
8
# (at your option) any later version.
 
9
#
 
10
# This program is distributed in the hope that it will be useful,
 
11
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
12
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
13
# GNU General Public License for more details.
 
14
#
 
15
# You should have received a copy of the GNU General Public License
 
16
# along with this program; if not, write to the Free Software
 
17
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 
18
 
 
19
import optparse
 
20
import os
 
21
import stat
 
22
import sys
 
23
import shutil
 
24
 
 
25
import ivle.conf
 
26
import ivle.config
 
27
import ivle.jailbuilder.debian
 
28
 
 
29
class UnsafeJail(Exception):
 
30
    pass
 
31
 
 
32
usage = """usage: %prog [options]
 
33
(requires root)
 
34
Builds or updates the base IVLE jail."""
 
35
 
 
36
conf = ivle.config.Config()
 
37
build_path = ivle.conf.jail_system_build
 
38
 
 
39
# Parse arguments
 
40
parser = optparse.OptionParser(usage)
 
41
parser.add_option("-r", "--recreate",
 
42
    action="store_true", dest="recreate",
 
43
    help='''Completely recreate the jail - don't just update its IVLE code.
 
44
Be warned, this may download hundreds of megabytes!''')
 
45
parser.add_option("-u", "--upgrade",
 
46
    action="store_true", dest="upgrade",
 
47
    help='''Apply any package updates in the jail.''')
 
48
parser.add_option("-m", "--mirror",
 
49
    action="store", dest="apt_mirror",
 
50
    help="Sets the apt mirror.", default=conf['jail']['mirror'])
 
51
parser.add_option("--python-site-packages",
 
52
    action="store", dest="python_site_packages",
 
53
    help="Path to Python site packages directory inside the jail.",
 
54
    default=None)
 
55
(options, args) = parser.parse_args(sys.argv)
 
56
 
 
57
if os.geteuid() != 0:
 
58
    print >> sys.stderr, "Must be root to run buildjail."
 
59
    sys.exit(1)
 
60
 
 
61
if not options.recreate and not os.path.exists(build_path):
 
62
    print >> sys.stderr, "No jail exists -- please rerun with -r."
 
63
    sys.exit(1)
 
64
 
 
65
if (options.python_site_packages is not None and
 
66
    options.python_site_packages[:1] not in (os.path.sep, os.path.altsep)):
 
67
    print >> sys.stderr, "python-site-packages must be an absolute path."
 
68
    sys.exit(1)
 
69
 
 
70
if options.recreate:
 
71
    options.upgrade = True
 
72
 
 
73
    # Create the jail and its subdirectories
 
74
    # Note: Other subdirs will be made by copying files
 
75
    if options.apt_mirror is not None:
 
76
        os.environ['MIRROR'] = options.apt_mirror
 
77
 
 
78
    os.system('rm -rf --one-file-system ' + build_path)
 
79
    ivle.jailbuilder.debian.debootstrap_create_jail(conf['jail']['suite'],
 
80
              build_path, mirror=options.apt_mirror)
 
81
 
 
82
    ivle.jailbuilder.debian.apt_update_cache(build_path)
 
83
    ivle.jailbuilder.debian.apt_install(build_path,
 
84
                        ['python2.5', 'python-cjson', 'python-svn'])
 
85
 
 
86
    ivle.jailbuilder.debian.apt_clean(build_path)
 
87
 
 
88
if options.upgrade:
 
89
    # Run apt-get update, apt-get upgrade and apt-get clean.
 
90
    ivle.jailbuilder.debian.mangle_sources_list(build_path, clobber=True)
 
91
    ivle.jailbuilder.debian.mangle_sources_list(build_path, lines=[
 
92
            'deb %s %s%s %s' % (options.apt_mirror, conf['jail']['suite'],
 
93
                                pocket, ' '.join(['main', 'universe']))
 
94
            for pocket in ('', '-updates', '-security')])
 
95
 
 
96
    # Add any extra site apt sources.
 
97
    if conf['jail']['extra_sources']:
 
98
        ivle.jailbuilder.debian.mangle_sources_list(build_path,
 
99
                  conf['jail']['extra_sources'])
 
100
 
 
101
    # Add any extra site apt keys.
 
102
    if conf['jail']['extra_keys']:
 
103
        ivle.jailbuilder.debian.apt_add_key(build_path,
 
104
                                            conf['jail']['extra_keys'])
 
105
 
 
106
    ivle.jailbuilder.debian.apt_update_cache(build_path)
 
107
    ivle.jailbuilder.debian.apt_upgrade(build_path)
 
108
 
 
109
    # Install any extra site packages.
 
110
    if conf['jail']['extra_packages']:
 
111
        ivle.jailbuilder.debian.apt_install(build_path,
 
112
                  conf['jail']['extra_packages'])
 
113
 
 
114
    ivle.jailbuilder.debian.apt_clean(build_path)
 
115
 
 
116
if conf['jail']['devmode']:
 
117
    # Copy all console and operating system files into the jail
 
118
    services_path = os.path.join(ivle.conf.share_path, 'services')
 
119
    jail_services_path = os.path.join(build_path, services_path[1:])
 
120
    if os.path.exists(jail_services_path):
 
121
        shutil.rmtree(jail_services_path)
 
122
    shutil.copytree(services_path, jail_services_path)
 
123
 
 
124
    # Also copy the IVLE lib directory into the jail
 
125
    # This is necessary for running certain services
 
126
 
 
127
    # ivle_site_packages is the IVLE install location outside the jail
 
128
    ivle_site_packages = os.path.dirname(ivle.__file__)
 
129
 
 
130
    if options.python_site_packages is None:
 
131
        # Get the site packages from the IVLE install location *OUTSIDE* the
 
132
        # jail. Warning! This only works if you have the same Python site
 
133
        # packages directory inside and out (ie. same Python version).
 
134
        # If not, you should use --python-site-packages.
 
135
        jail_site_packages = os.path.join(build_path, ivle_site_packages[1:])
 
136
    else:
 
137
        # User-specified site packages
 
138
        jail_site_packages = os.path.join(build_path,
 
139
                                options.python_site_packages[1:], "ivle")
 
140
    if os.path.exists(jail_site_packages):
 
141
        shutil.rmtree(jail_site_packages)
 
142
    shutil.copytree(ivle_site_packages, jail_site_packages)
 
143
 
 
144
# Make /tmp and /var/lock un-world-writable. /tmp will be mounted over,
 
145
# and /var/{lock,tmp} should die.
 
146
for path in ('tmp', 'var/lock', 'var/tmp'):
 
147
    path = os.path.join(build_path, path)
 
148
    os.chmod(path, os.stat(path).st_mode & ~stat.S_IWOTH)
 
149
 
 
150
# Verify that nothing in the jail is world-writable.
 
151
# We don't want students to write into places that others can see.
 
152
for path, dirs, files in os.walk(build_path):
 
153
    for dname in dirs:
 
154
        d = os.path.join(path, dname)
 
155
        if os.path.islink(d):
 
156
            continue
 
157
        if os.stat(d).st_mode & stat.S_IWOTH:
 
158
            raise UnsafeJail(d)
 
159
 
 
160
    for fname in files:
 
161
        f = os.path.join(path, fname)
 
162
        if os.path.islink(f):
 
163
            continue
 
164
        if os.stat(f).st_mode & stat.S_IWOTH:
 
165
            if (os.path.dirname(f) == os.path.join(build_path, 'dev') and
 
166
                os.path.basename(f) in ('ptmx', 'null', 'tty', 'full', 'zero',
 
167
                                        'random', 'urandom')
 
168
                ):
 
169
                continue
 
170
            raise UnsafeJail(f)
 
171
    
 
172
 
 
173
if os.spawnvp(os.P_WAIT, 'rsync', ['rsync', '-a', '--delete',
 
174
              build_path + '/', ivle.conf.jail_system]) != 0:
 
175
    print >> sys.stderr, "Jail copying failed."
 
176
    sys.exit(1)
 
177
 
 
178
# Now mangle things a bit, so we can bind-mount the user bits in.
 
179
# /etc/passwd and /etc/ivle/ivle.conf need to be symlinks to somewhere in /home
 
180
 
 
181
os.rename(os.path.join(ivle.conf.jail_system, 'etc/passwd'),
 
182
          os.path.join(ivle.conf.jail_system, 'home/.passwd')
 
183
          )
 
184
os.symlink('../home/.passwd', os.path.join(ivle.conf.jail_system, 'etc/passwd'))
 
185
 
 
186
os.makedirs(os.path.join(ivle.conf.jail_system, "etc/ivle"))
 
187
os.symlink('../../home/.ivle.conf',
 
188
           os.path.join(ivle.conf.jail_system, "etc/ivle/ivle.conf"))