Source Code for Module VMBuilder.hypervisor

  1  # 
  2  #    Uncomplicated VM Builder 
  3  #    Copyright (C) 2007-2009 Canonical Ltd. 
  4  #     
  5  #    See AUTHORS for list of contributors 
  6  # 
  7  #    This program is free software: you can redistribute it and/or modify 
  8  #    it under the terms of the GNU General Public License version 3, as 
  9  #    published by the Free Software Foundation. 
 10  # 
 11  #    This program is distributed in the hope that it will be useful, 
 12  #    but WITHOUT ANY WARRANTY; without even the implied warranty of 
 13  #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the 
 14  #    GNU General Public License for more details. 
 15  # 
 16  #    You should have received a copy of the GNU General Public License 
 17  #    along with this program.  If not, see <http://www.gnu.org/licenses/>. 
 18  # 
 19  #    Hypervisor super class 
 20   
 21  import logging 
 22  import os 
 23  import VMBuilder.distro 
 24  import VMBuilder.disk 
 25  from   VMBuilder.util    import run_cmd, tmpdir 
 26   
 27  STORAGE_DISK_IMAGE = 0 
 28  STORAGE_FS_IMAGE = 1 
 29   
 30 -class Hypervisor(VMBuilder.distro.Context): 
 31      preferred_storage = STORAGE_DISK_IMAGE 
 32   
 33 -    def __init__(self, distro): 
 34          self.plugin_classes = VMBuilder._hypervisor_plugins 
 35          super(Hypervisor, self).__init__() 
 36          self.plugins += [distro] 
 37          self.distro = distro 
 38          self.filesystems = [] 
 39          self.disks = [] 
 40          self.nics = [] 
 41   
 42 -    def add_filesystem(self, *args, **kwargs): 
 43          """Adds a filesystem to the virtual machine""" 
 44          from VMBuilder.disk import Filesystem 
 45   
 46          fs = Filesystem(self, *args, **kwargs) 
 47          self.filesystems.append(fs) 
 48          return fs 
 49   
 50 -    def add_disk(self, *args, **kwargs): 
 51          """Adds a disk image to the virtual machine""" 
 52          from VMBuilder.disk import Disk 
 53   
 54          disk = Disk(self, *args, **kwargs) 
 55          self.disks.append(disk) 
 56          return disk 
 57   
 58 -    def install_os(self): 
 59          self.nics = [self.NIC()] 
 60          self.call_hooks('preflight_check') 
 61          self.call_hooks('configure_networking', self.nics) 
 62          self.call_hooks('create_partitions') 
 63          self.call_hooks('configure_mounting', self.disks, self.filesystems) 
 64   
 65          self.chroot_dir = tmpdir() 
 66          self.call_hooks('mount_partitions', self.chroot_dir) 
 67          run_cmd('rsync', '-aHA', '%s/' % self.distro.chroot_dir, self.chroot_dir) 
 68          self.distro.set_chroot_dir(self.chroot_dir) 
 69          if self.needs_bootloader: 
 70              self.call_hooks('install_bootloader', self.chroot_dir, self.disks) 
 71          self.call_hooks('install_kernel', self.chroot_dir) 
 72          self.distro.call_hooks('post_install') 
 73          self.call_hooks('unmount_partitions') 
 74          os.rmdir(self.chroot_dir) 
 75   
 76 -    def finalise(self, destdir): 
 77          self.call_hooks('convert',  
 78                          self.preferred_storage == STORAGE_DISK_IMAGE and self.disks or self.filesystems, 
 79                          destdir) 
 80          self.call_hooks('deploy', destdir) 
 81   
 82 -    def create_partitions(self): 
 83          """Creates all the vms partitions and formats them """ 
 84          for fs in self.filesystems: 
 85              fs.create() 
 86              fs.mkfs() 
 87          for disk in self.disks: 
 88              disk.create() 
 89              disk.partition() 
 90              disk.map_partitions() 
 91              disk.mkfs() 
 92   
 93 -    def mount_partitions(self, mntdir): 
 94          """Mounts all the vm's partitions and filesystems below .rootmnt""" 
 95          logging.info('Mounting target filesystems') 
 96          fss = VMBuilder.disk.get_ordered_filesystems(self) 
 97          for fs in fss: 
 98              fs.mount(mntdir) 
 99              self.distro.post_mount(fs) 
100   
101 -    def unmount_partitions(self): 
102          """Unmounts all the vm's partitions and filesystems""" 
103          logging.info('Unmounting target filesystem') 
104          fss = VMBuilder.disk.get_ordered_filesystems(self) 
105          fss.reverse() 
106          for fs in fss: 
107              fs.umount() 
108          for disk in self.disks: 
109              disk.unmap() 
110   
111 -    def convert_disks(self, disks, destdir): 
112          for disk in disks: 
113              disk.convert(destdir, self.filetype) 
114   
115 -    class NIC(object): 
116 -        def __init__(self, type='dhcp', ip=None, network=None, netmask=None, 
117                             broadcast=None, dns=None, gateway=None): 
118              self.type = type 
119              self.ip = ip 
120              self.network = network 
121              self.netmask = netmask 
122              self.broadcast = broadcast 
123              self.dns = dns 
124              self.gateway = gateway 
125