kvm

   1 #!/usr/bin/python
   2
   3 import sys, os, time, re
   4 import optparse, commands
   5 import ConfigParser, StringIO
   6
   7 class ShellConfigParser(ConfigParser.ConfigParser):
   8     def read(self, filename):
   9         try:
  10             text = open(filename).read()
  11         except IOError:
  12             pass
  13         else:
  14             file = StringIO.StringIO("[shell]\n" + text)
  15             self.readfp(file, filename)
  16
  17 config = ShellConfigParser()
  18 config.read('config.mak')
  19
  20 external_module = config.get('shell', 'want_module')
  21 privileged = os.getuid() == 0
  22
  23 optparser = optparse.OptionParser()
  24
  25 optparser.add_option('--no-reload-module',
  26                      help = 'do not reload kvm module',
  27                      action = 'store_false',
  28                      dest = 'reload',
  29                      default = privileged,
  30                      )
  31
  32 optparser.add_option('--install',
  33                      help = 'start up guest in installer boot cd',
  34                      action = 'store_true',
  35                      default = False,
  36                      )
  37
  38 optparser.add_option('-m', '--memory',
  39                      help = 'guest memory in MB',
  40                      type = 'int',
  41                      default = 384,
  42                      dest = 'memory',
  43                      )
  44
  45 optparser.add_option('--debugger',
  46                      help = 'wait for gdb',
  47                      action = 'store_true',
  48                      default = False,
  49                      )
  50
  51 optparser.add_option('--no-tap',
  52                      help = 'run the guest without tap netif',
  53                      action = 'store_true',
  54                      dest = 'notap',
  55                      default = not privileged,
  56                      )
  57
  58 optparser.add_option('--nictype',
  59                      help = 'use this specific nic type (vendor)',
  60                      dest = 'nictype',
  61                      default = 'rtl8139',
  62                      )
  63
  64 optparser.add_option('--mac',
  65                      help = 'use this specific mac addr',
  66                      dest = 'mac',
  67                      default = None,
  68                      )
  69
  70 optparser.add_option('--vnc',
  71                      help = 'use VNC rather than SDL',
  72                      dest = 'vnc',
  73                      default = None,
  74                      )
  75
  76 optparser.add_option('--no-kvm',
  77                      help = 'use standard qemu, without kvm',
  78                      action = 'store_false',
  79                      dest = 'kvm',
  80                      default = True,
  81                      )
  82 optparser.add_option('--image',
  83                      help = 'select disk image',
  84                      dest = 'image',
  85                      default = '/tmp/disk',
  86                      )
  87 optparser.add_option('--cdrom',
  88                      help = 'select cdrom image',
  89                      dest = 'cdrom',
  90                      default = None,
  91                      )
  92
  93 optparser.add_option('--hdb',
  94                      help = 'secondary hard disk image',
  95                      dest = 'hdb',
  96                      default = None,
  97                      )
  98
  99 optparser.add_option('--loadvm',
 100                      help = 'select saved vm-image',
 101                      dest = 'saved_image',
 102                      default = None,
 103                      )
 104
 105 optparser.add_option('--monitor',
 106                      help = 'redirect monitor (currently only stdio or tcp)',
 107                      dest = 'monitor',
 108                      default = None,
 109                      )
 110
 111 optparser.add_option('--stopped',
 112                      help = 'start image in stopped mode',
 113                      action = 'store_true',
 114                      default = False,
 115                      )
 116
 117 optparser.add_option('-s', '--smp',
 118                      type = 'int',
 119                      default = 1,
 120                      dest = 'vcpus',
 121                      help = 'define number of vcpus',
 122                      )
 123
 124 optparser.add_option('--no-kvm-irqchip',
 125                      action = 'store_false',
 126                      default = True,
 127                      dest = 'irqchip',
 128                      help = 'avoid using in-kernel irqchip',
 129                      )
 130
 131 optparser.add_option('-n', '--dry-run',
 132                      help = "just print the qemu command line; don't run it",
 133                      action = 'store_true',
 134                      dest = 'dry_run',
 135                      default = False,
 136                      )
 137
 138
 139 (options, args) = optparser.parse_args()
 140
 141 if len(args) > 0:
 142     options.image = args[0]
 143
 144 if len(args) > 1:
 145     options.cdrom = args[1]
 146
 147 def remove_module(module):
 148     module = module.replace('-', '_')
 149     lines = commands.getoutput('/sbin/lsmod').split('\n')
 150     for x in lines:
 151         if x.startswith(module + ' '):
 152             if os.spawnl(os.P_WAIT, '/sbin/rmmod', 'rmmod', module) != 0:
 153                 raise Exception('failed to remove %s module' % (module,))
 154
 155 def insert_module(module):
 156     if os.spawnl(os.P_WAIT, '/sbin/insmod', 'insmod',
 157                  'kernel/%s.ko' % (module,)) != 0:
 158         raise Exception('failed to load kvm module')
 159
 160 def probe_module(module):
 161     if os.spawnl(os.P_WAIT, '/sbin/modprobe', 'modprobe', module) != 0:
 162         raise Exception('failed to load kvm module')
 163
 164 def vendor():
 165     for x in file('/proc/cpuinfo').readlines():
 166         m = re.match(r'vendor_id[ \t]*: *([a-zA-Z]+),*', x)
 167         if m:
 168             return m.group(1)
 169     return unknown
 170
 171 vendor_module = {
 172     'GenuineIntel': 'kvm-intel',
 173     'AuthenticAMD': 'kvm-amd',
 174     }[vendor()]
 175
 176 if options.kvm and options.reload:
 177     for module in [vendor_module, 'kvm']:
 178         remove_module(module)
 179     if external_module:
 180         insmod = insert_module
 181     else:
 182         insmod = probe_module
 183     for module in ['kvm', vendor_module]:
 184         insmod(module)
 185     commands.getstatusoutput('/sbin/udevsettle')
 186     if not os.access('/dev/kvm', os.F_OK):
 187         print '/dev/kvm not present'
 188
 189 disk = options.image
 190 if options.install:
 191     (status, output) = commands.getstatusoutput(
 192         'qemu/qemu-img create -f qcow2 "%s" 30G' % disk)
 193     if status:
 194         raise Exception, output
 195
 196 bootdisk = 'c'
 197 if options.install:
 198     bootdisk = 'd'
 199
 200 arch = 'x86_64'
 201
 202 if arch == 'x86_64':
 203     cmd = 'qemu-system-' + arch
 204 else:
 205     cmd = 'qemu'
 206
 207 local_cmd = 'qemu/' + arch + '-softmmu/' + cmd
 208 if os.access(local_cmd, os.F_OK):
 209     cmd = local_cmd
 210 else:
 211     cmd = '/usr/bin/kvm'
 212
 213 qemu_args = (cmd, '-boot', bootdisk,
 214              '-hda', disk, '-m', str(options.memory),
 215              '-serial', 'file:/tmp/serial.log',
 216              '-smp', str(options.vcpus),
 217              #'-usbdevice', 'tablet',
 218           )
 219
 220 if options.cdrom:
 221     qemu_args += ('-cdrom', options.cdrom,)
 222
 223 if options.hdb:
 224     qemu_args += ('-hdb', options.hdb,)
 225
 226 if not options.kvm:
 227     qemu_args += ('-no-kvm',)
 228
 229 if options.debugger:
 230     qemu_args += ('-s',)
 231
 232 if not options.irqchip:
 233     qemu_args += ('-no-kvm-irqchip',)
 234
 235 if not options.notap:
 236     mac = options.mac
 237     if not mac:
 238         for line in commands.getoutput('/sbin/ip link show eth0').splitlines():
 239             m = re.match(r'.*link/ether (..:..:..:..:..:..).*', line)
 240             if m:
 241                 mac = m.group(1)
 242         if not mac:
 243             raise Exception, 'Unable to determine eth0 mac address'
 244         mac_components = mac.split(':')
 245         mac_components[0] = 'a0'
 246         mac = ':'.join(mac_components)
 247
 248     qemu_args += ('-net', 'nic,macaddr=%s,model=%s' % (mac,options.nictype,),
 249                   '-net', 'tap,script=/etc/kvm/qemu-ifup',)
 250
 251 if options.vnc:
 252     qemu_args += ('-vnc', str(options.vnc))
 253
 254 if options.saved_image:
 255     qemu_args += ('-loadvm' , options.saved_image, )
 256
 257 if options.monitor:
 258     if options.monitor == 'stdio':
 259         qemu_args += ('-monitor' , 'stdio', )
 260     elif options.monitor == 'tcp':
 261         qemu_args += ('-monitor' , 'tcp:0:5555,server,nowait', )
 262     else:
 263         raise Exception('illegal monitor option %s' % option.monitor)
 264
 265 if options.stopped:
 266     qemu_args += ('-S',)
 267
 268 if options.dry_run:
 269     def concat_func(x,y): return x + ' ' + y
 270     print reduce(concat_func, qemu_args)
 271     sys.exit(0)
 272
 273 os.execvp(cmd, qemu_args)