← Back to branch summary

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

« back to all changes in this revision

Viewing changes to ivle/jailbuilder/debian.py

  • Committer: William Grant
  • Date: 2012-06-28 01:52:02 UTC
  • Revision ID: me@williamgrant.id.au-20120628015202-f6ru7o367gt6nvgz
Hah

Show diffs side-by-side

added added

removed removed

Lines of Context:
ivle/jailbuilder/debian.py
 
1
# IVLE - Informatics Virtual Learning Environment
 
2
# Copyright (C) 2007-2009 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
'''Debian-specific bits of the jail building process.'''
 
19
 
 
20
import os
 
21
import subprocess
 
22
 
 
23
from ivle.jailbuilder.exceptions import JailBuildError
 
24
 
 
25
def debootstrap_create_jail(release, path, components=['main', 'universe'],
 
26
                            mirror=None):
 
27
    """Create a Debian-based jail using debootstrap."""
 
28
    if mirror is None:
 
29
        mirror = 'http://archive.ubuntu.com/ubuntu'
 
30
    ec = os.spawnvp(os.P_WAIT, 'debootstrap',
 
31
              ['debootstrap', '--components=' + ','.join(components),
 
32
               '--include=ubuntu-keyring,gnupg', '--variant=minbase',
 
33
               release, path, mirror])
 
34
    if ec != 0:
 
35
        raise JailBuildError('debootstrap failed with code %d' % ec)
 
36
 
 
37
    mangle_sources_list(path, clobber=True)
 
38
    mangle_sources_list(path, lines=[
 
39
            'deb %s %s%s %s' % (mirror, release, pocket, ' '.join(components))
 
40
            for pocket in ('', '-updates', '-security')])
 
41
 
 
42
def mangle_sources_list(path, lines=[], clobber=False):
 
43
    """Add lines to a sources.list, optionally clobbering an existing one."""
 
44
    f = open(os.path.join(path, 'etc/apt/sources.list'),
 
45
             'w' if clobber else 'a')
 
46
 
 
47
    for line in lines:
 
48
        f.write(line + '\n')
 
49
    f.close()
 
50
 
 
51
def _execute_in_chroot(path, argv, stdin=None):
 
52
    """Execute a binary in a chroot, with optional stdin."""
 
53
    pid = subprocess.Popen(['chroot', path] + argv, stdin=subprocess.PIPE)
 
54
    pid.communicate(stdin)
 
55
    if pid.returncode != 0:
 
56
        raise JailBuildError('command failed with code %d' % pid.returncode)
 
57
 
 
58
def apt_update_cache(path):
 
59
    """Update the apt cache in a chroot."""
 
60
    _execute_in_chroot(path, ['apt-get', '-y', 'update'])
 
61
 
 
62
def apt_upgrade(path):
 
63
    """Upgrade apt packages in a chroot."""
 
64
    _execute_in_chroot(path, ['apt-get', '-y', 'upgrade'])
 
65
 
 
66
def apt_install(path, packages=None):
 
67
    """Install apt packages in a chroot."""
 
68
    if not packages:
 
69
        return
 
70
    _execute_in_chroot(path, ['apt-get', '-y', 'install'] + packages)
 
71
 
 
72
def apt_clean(path):
 
73
    """Clean the apt package cache in a chroot."""
 
74
    _execute_in_chroot(path, ['apt-get', 'clean'])
 
75
 
 
76
def apt_add_key(path, key_text):
 
77
    """Add a key to authenticate apt repositories in a chroot."""
 
78
    _execute_in_chroot(path, ['apt-key', 'add', '-'], stdin=key_text)