vmit 0.0.3 → 0.0.3.99

data/bin/vmit

@@ -41,14 +41,14 @@ module Vmit
 
     def execute
       options = {}
-      vm = Vmit::VirtualMachine.new(File.expand_path(Dir.pwd))
+      vm = Vmit::Workspace.new(File.expand_path(Dir.pwd))
 
       OPTION_KEYS.each do |k|
         val = self.send(k)
         options[k] = val if val
       end
 
-      vm.options.merge!(options)
+      vm.config.configure(options)
       vm.save_config!
 
       vm.disk_image_init!(options)
@@ -57,7 +57,7 @@ module Vmit
 
   end
 
-  class RunCommand < BaseVmCommand
+  class UpCommand < BaseVmCommand
 
     OPTION_KEYS = [:memory, :cdrom]
 
@@ -70,23 +70,89 @@ module Vmit
 
     def execute
       options = {}
-      vm = Vmit::VirtualMachine.new(File.expand_path(Dir.pwd))
+      workspace = Vmit::Workspace.new(File.expand_path(Dir.pwd))
 
       OPTION_KEYS.each do |k|
         val = self.send(k)
         options[k] = val if val
       end
 
-      vm.run(options)
+      vm = Vmit::LibvirtVM.new(workspace, options)
+      vm.up
     end
 
   end
 
+  class SpiceCommand < Clamp::Command
+    def execute
+      vm = LibvirtVM.from_pwd
+      vm.spice
+    end
+  end
+
+  class VncCommand < Clamp::Command
+    def execute
+      vm = LibvirtVM.from_pwd
+      vm.vnc
+    end
+  end
+
+  class SshCommand < Clamp::Command
+    def execute
+      vm = LibvirtVM.from_pwd
+      vm.assert_up
+
+      while not vm.ip_address
+        sleep(1)
+        Vmit.logger.info("Waiting for ip address...")
+      end
+      Vmit.logger.info("ip address is #{vm.ip_address}")
+      Vmit::Utils.wait_for_port(vm.ip_address, 22)
+      system("ssh root@#{vm.ip_address}")
+    end
+  end
+
+  class StatusCommand < Clamp::Command
+    def execute
+      vm = LibvirtVM.from_pwd
+      st, reason = vm.state
+
+      puts "VM is #{st}...(#{reason})"
+      if vm.ip_address
+        puts "  #{vm.ip_address}"
+      end
+
+    end
+  end
+
+  class RebootCommand < Clamp::Command
+    def execute
+      vm = LibvirtVM.from_pwd
+      vm.reboot
+    end
+  end
+
+  class ShutdownCommand < Clamp::Command
+    def execute
+      vm = LibvirtVM.from_pwd
+      vm.shutdown
+    end
+  end
+
+  class DestroyCommand < Clamp::Command
+    def execute
+      vm = LibvirtVM.from_pwd
+      vm.destroy
+    end
+  end
+
   class DiskSnapshotCommand < Clamp::Command
 
     def execute
-      vm = Vmit::VirtualMachine.new(File.expand_path(Dir.pwd))
-      vm.disk_snapshot!
+      vm = LibvirtVM.from_pwd
+      vm.assert_down
+
+      vm.workspace.disk_snapshot!
     end
 
   end
@@ -94,8 +160,8 @@ module Vmit
   class DiskImagesCommand < Clamp::Command
 
     def execute
-      vm = Vmit::VirtualMachine.new(File.expand_path(Dir.pwd))
-      puts vm.disk_images.last
+      vm = LibvirtVM.from_pwd
+      puts vm.workspace.disk_images.last
     end
 
   end
@@ -103,19 +169,30 @@ module Vmit
   class DiskRollbackCommand < Clamp::Command
 
     def execute
-      vm = Vmit::VirtualMachine.new(File.expand_path(Dir.pwd))
-      vm.disk_rollback!
+      vm = LibvirtVM.from_pwd
+      vm.assert_down
+
+      vm.workspace.disk_rollback!
     end
 
   end
 
+  Vmit.load_plugins!
+
   class MainCommand < Clamp::Command
 
     subcommand "init", "Initialize the vm", InitCommand
-    subcommand "run", "Run the vm", RunCommand
+    subcommand "up", "Run the vm", UpCommand
+    subcommand "status", "Show VM status", StatusCommand
+    subcommand "shutdown", "Shutdown the VM", ShutdownCommand
+    subcommand "reboot", "Reboot the VM", RebootCommand
+    subcommand "destroy", "Destroy the VM", DestroyCommand
     subcommand "disk-images", "List disk images", DiskImagesCommand
     subcommand "disk-snapshot", "Creates a new disk-snapshot", DiskSnapshotCommand
     subcommand "disk-rollback", "Rollbacks to previous disk snapshot", DiskRollbackCommand
+    subcommand "spice", "SPICE into the machine", SpiceCommand
+    subcommand "ssh", "SSH into the machine", SshCommand
+    subcommand "vnc", "VNC into the machine", VncCommand
 
     # Add commands offered via plugins
     Vmit.plugins.each do |plugin|
@@ -127,7 +204,6 @@ module Vmit
     end
   end
 
-
 end
 
 begin

data/lib/vmit.rb

@@ -19,13 +19,18 @@
 # CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 #
 require 'vmit/version'
-require 'vmit/bootstrap'
+require 'vmit/unattended_install'
 require 'vmit/logger'
-require 'vmit/refcounted_resource'
 require 'vmit/network'
 require 'vmit/vfs'
-require 'vmit/virtual_machine'
+require 'vmit/workspace'
+require 'vmit/libvirt_vm'
+require 'vmit/autoyast'
+require 'vmit/kickstart'
+require 'vmit/debian_preseed'
+require 'vmit/install_media'
 require 'vmit/ext'
+require 'vmit/utils'
 require 'pidfile'
 
 module Vmit
@@ -47,21 +52,23 @@ module Vmit
     @plugins ||= []
   end
 
-end
+  def self.load_plugins!
+    # Scan plugins
+    plugin_glob = File.join(File.dirname(__FILE__), 'vmit', 'plugins', '*.rb')
+    Dir.glob(plugin_glob).each do |plugin|
+      Vmit.logger.debug("Loading file: #{plugin}")
+      #puts "Loading file: #{plugin}"
+      load plugin
+    end
 
-# Scan plugins
-plugin_glob = File.join(File.dirname(__FILE__), 'vmit', 'plugins', '*.rb')
-Dir.glob(plugin_glob).each do |plugin|
-  Vmit.logger.debug("Loading file: #{plugin}")
-  #puts "Loading file: #{plugin}"
-  load plugin
-end
+    # instantiate plugins
+    ::Vmit::Plugins.constants.each do |cnt|
+      pl_class = ::Vmit::Plugins.const_get(cnt)
+      #pl_instance = pl_class.new
+      Vmit.add_plugin(pl_class)
+      Vmit.logger.debug("Loaded: #{pl_class}")
+      #puts "Loaded: #{pl_class}"
+    end
+  end
 
-# instantiate plugins
-::Vmit::Plugins.constants.each do |cnt|
-  pl_class = ::Vmit::Plugins.const_get(cnt)
-  #pl_instance = pl_class.new
-  Vmit.add_plugin(pl_class)
-  Vmit.logger.debug("Loaded: #{pl_class}")
-  #puts "Loaded: #{pl_class}"
 end

data/lib/vmit/autoyast.rb

@@ -19,38 +19,92 @@
 # CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 #
 require 'nokogiri'
+require 'vmit/unattended_install'
+require 'vmit/vfs'
 
 module Vmit
 
-  class AutoYaST
+  class AutoYaST < UnattendedInstall
 
-    attr_accessor :patterns
-    attr_accessor :packages
-
-    def initialize
-      @net_udev = {}
-      @patterns = []
-      @packages = []
+    def initialize(location)
+      super(location)
+      base!
     end
 
-    def minimal_opensuse!
-      @patterns << 'base'
+    def base!
+      config.add_patterns! 'base'
+      config.add_packages! 'zypper'
+      config.add_packages! 'openssh'
     end
 
-    def minimal_sle!
-      @patterns << 'Minimal'
+    # minimal installation
+    # does not work with openSUSE
+    def minimal!
+      config.add_patterns! 'Minimal'
+      config.add_packages! 'zypper'
+      config.add_packages! 'openssh'
     end
 
-    # Map a network device name and make
-    # it persistent
-    #
-    # @param [String] MAC mac address
-    # @param [String] device name
-    def name_network_device(mac, name)
-      if @net_udev.has_key?(mac) or @net_udev.has_value?(mac)
-        raise "Device with MAC #{mac} is already assigned to #{@net_udev[name]}"
+    # @param [Hash] args Arguments for 1st stage
+    def execute_autoinstall(vm, args)
+      vm.config.push!
+      begin
+        vm.config.configure(args)
+        media = Vmit::VFS.from(location)
+        kernel_append_arg = case media
+          when Vmit::VFS::URI then "install=#{location}"
+          when Vmit::VFS::ISO then 'install=cdrom'
+          else raise ArgumentError.new("Unsupported autoinstallation: #{location}")
+        end
+        vm.config.add_kernel_cmdline!(kernel_append_arg)
+
+        if media.is_a?(Vmit::VFS::ISO)
+          vm.config.cdrom = location.to_s
+        end
+
+        Dir.mktmpdir do |floppy_dir|
+          FileUtils.chmod_R 0775, floppy_dir
+          vm.config.floppy = floppy_dir
+          #vm.config.add_kernel_cmdline!('autoyast=device://fd0/autoinst.xml')
+          vm.config.add_kernel_cmdline!('autoyast=floppy')
+          vm.config.reboot = false
+
+          # WTF SLE and openSUSE have different
+          # base pattern names
+          #media.open('/content') do |content_file|
+          #  content_file.each_line do |line|
+          #    case line
+          #      when /^DISTRIBUTION (.+)$/
+          #        case $1
+          #          when /SUSE_SLE/ then autoyast.minimal_sle!
+          #          when /openSUSE/ then autoyast.minimal_opensuse!
+          #        end
+          #    end
+          #  end
+          #end
+
+          File.write(File.join(floppy_dir, 'autoinst.xml'), to_xml)
+          Vmit.logger.info "AutoYaST: 1st stage."
+          puts vm.config.inspect
+          vm.up
+          vm.wait_until_shutdown! do
+            vm.vnc
+          end
+          vm.config.pop!
+
+          Vmit.logger.info "AutoYaST: 2st stage."
+          # 2nd stage
+          vm.config.push!
+          vm.config.configure(:reboot => false)
+          vm.up
+          vm.wait_until_shutdown! do
+            vm.vnc
+          end
+
+        end
+      ensure
+        vm.config.pop!
       end
-      @net_udev[mac] = name
     end
 
     def to_xml
@@ -87,10 +141,15 @@ module Vmit
           }
           xml.software {
             xml.patterns('config:type' => 'list') {
-              @patterns.each do |pat|
+              config.patterns.each do |pat|
                 xml.pattern pat
               end
             }
+            xml.packages('config:type' => 'list') {
+              config.packages.each do |pkg|
+                xml.package pkg
+              end
+            }
           }
           # SLE 11 can do without this basic partitioning but
           # SLE 10 is not that smart.

data/lib/vmit/debian_preseed.rb

@@ -19,23 +19,54 @@
 # CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 #
 require 'erb'
+require 'vmit/unattended_install'
 
 module Vmit
 
   # Some good references here:
   # http://www.hps.com/~tpg/notebook/autoinstall.php
-  class DebianPreseed
+  class DebianPreseed < UnattendedInstall
+
+    def execute_autoinstall(vm, args)
+      vm.config.push!
+      begin
+        vm.config.configure(args)
+        Dir.mktmpdir do |floppy_dir|
+          FileUtils.chmod_R 0755, floppy_dir
+          vm.config.floppy = floppy_dir
+          vm.config.add_kernel_cmdline!('preseed/file=/floppy/preseed.cfg')
+          vm.config.add_kernel_cmdline!('auto=true')
+          vm.config.add_kernel_cmdline!('priority=critical')
+          vm.config.reboot = false
+
+          File.write(File.join(floppy_dir, 'preseed.cfg'), to_txt)
+          Vmit.logger.info "Preseed: 1st stage."
+          vm.up
+          vm.wait_until_shutdown! do
+            vm.vnc
+          end
+        end
+      ensure
+        vm.config.pop!
+      end
+    end
 
     def to_txt
       template = ERB.new <<-EOF
+d-i debconf/priority select critical
+d-i auto-install/enabled boolean true
 d-i debian-installer/locale string en_US
 d-i console-tools/archs select at
 d-i console-keymaps-at/keymap select American English
+
 d-i debian-installer/keymap string us
+
+d-i netcfg/choose_interface            select auto
 d-i netcfg/get_hostname string unassigned-hostname
 d-i netcfg/get_hostname seen true
 d-i netcfg/get_domain string unassigned-domain
 d-i netcfg/get_domain seen true
+
 d-i mirror/protocol string ftp
 d-i mirror/ftp/hostname string ftp.de.debian.org
 d-i mirror/ftp/directory string /debian/
@@ -51,7 +82,10 @@ d-i clock-setup/utc boolean true
 d-i time/zone string US/Eastern
 d-i clock-setup/ntp boolean true
 popularity-contest popularity-contest/participate boolean false
+
 d-i pkgsel/include string ssh rsync initrd-tools cramfsprogs lzop
+
+d-i passwd/root-login boolean true
 d-i passwd/root-password password linux
 d-i passwd/root-password-again password linux
 d-i passwd/make-user boolean false

data/lib/vmit/install_media.rb

@@ -0,0 +1,224 @@
+#
+# Copyright (C) 2013 Duncan Mac-Vicar P. <dmacvicar@suse.de>
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy of
+# this software and associated documentation files (the "Software"), to deal in
+# the Software without restriction, including without limitation the rights to
+# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+# the Software, and to permit persons to whom the Software is furnished to do so,
+# subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+# FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+#
+require 'vmit/utils'
+require 'vmit/vfs'
+require 'confstruct'
+
+module Vmit
+
+  class InstallMedia
+    abstract_method :unattended_install_class
+    abstract_method :kernel_path
+    abstract_method :initrd_path
+
+    attr_reader :location
+
+    def initialize(location)
+      @location = location
+    end
+
+    def to_s
+      "#{super.to_s}:#{location}"
+    end
+
+    def unattended_install
+      unless @unattended_install
+        @unattended_install = unattended_install_class.new(location)
+      end
+      @unattended_install
+    end
+
+    def self.class_for_media_type(type)
+      case type.to_s.downcase
+        when /fedora|redhat|centos/ then FedoraInstallMedia
+        when /suse/ then SUSEInstallMedia
+        when /debian/ then DebianInstallMedia
+        else
+          raise "Don't know how to bootstrap media #{location}"
+      end
+    end
+
+    # @return [InstallMedia] an install media for an arbitrary
+    #   string like 'sles 11 sp2' or 'SLED-11_SP3'
+    #
+    # @raise [ArgumentError] if can't figure out an install media
+    #   from the string
+    def self.alias_to_install_media(key)
+      case key.to_s.downcase.gsub(/[\s_\-]+/, '')
+        when /^opensuse(\d+\.\d+)$/
+          SUSEInstallMedia.new(
+            'http://download.opensuse.org/distribution/$version/repo/oss/'
+              .gsub('$version', $1))
+        when /^(opensuse)?factory$/
+          SUSEInstallMedia.new(
+            'http://download.opensuse.org/factory/repo/oss/')
+        when /^debian(.+)$/
+          DebianInstallMedia.new(
+            'http://cdn.debian.net/debian/dists/$version'
+              .gsub('$version', $1))
+        when /^ubuntu(.+)$/
+          UbuntuInstallMedia.new(
+            'http://archive.ubuntu.com/ubuntu/dists/$version'
+              .gsub('$version', $1))
+        when /^fedora(\d+)/
+          FedoraInstallMedia.new(
+            'http://mirrors.n-ix.net/fedora/linux/releases/$release/Fedora/$arch/os/'
+              .gsub('$arch', Vmit::Utils.arch)
+              .gsub('$release', $1))
+        when /^sle(s|d)?(\d+)(sp(\d+))?$/
+          edition = case $1
+            when 's' then 'sle-server'
+            when 'd' then 'sle-desktop'
+            else
+              Vmit.logger.warn "SLE given. Assuming server."
+              'sle-server'
+          end
+          release = $2
+          sp = $4 || '0'
+          klass = if release.to_i > 9
+            SUSEInstallMedia
+          else
+            SUSE9InstallMedia
+          end
+          suffix = if (release.to_i > 9)
+            '/DVD1'
+          else
+            if (sp.to_i > 0)
+              '/CD1'
+            else
+              ''
+            end
+          end
+          klass.new(
+            "http://schnell.suse.de/BY_PRODUCT/$edition-$release-sp$sp-$arch$topdir"
+              .gsub('$edition', edition)
+              .gsub('$arch', Vmit::Utils.arch)
+              .gsub('$release', release)
+              .gsub('$sp', sp)
+              .gsub('$topdir', suffix))
+        else raise ArgumentError.new("Unknown install media '#{key}'")
+      end
+    end
+
+    # @return [InstallMedia] an install media for a url.
+    #   it accesses the network in order to detect the
+    #   url type.
+    #
+    # @raise [ArgumentError] if can't figure out an install media
+    #   from the string
+    def self.url_to_install_media(url)
+      begin
+        media = Vmit::VFS.from(url)
+        media.open('/content')
+        return SUSEInstallMedia.new(url)
+      rescue
+        raise ArgumentError.new("Don't know the install media '#{url}'")
+      end
+    end
+
+    # @return [InstallMedia] scans the install media
+    #   and returns a specific type (suse, debian, etc...)
+    def self.scan(key)
+      case key
+        when /^http:\/|ftp:\// then url_to_install_media(key)
+        else alias_to_install_media(key)
+      end
+    end
+
+    def autoinstall(vm)
+      Vmit.logger.debug("Autoinstall from #{location}")
+      media = Vmit::VFS.from(location)
+      kernel = media.open(kernel_path)
+      initrd = media.open(initrd_path)
+      opts = {:kernel => kernel.path, :initrd => initrd.path}
+      unattended_install.execute_autoinstall(vm, opts)
+    end
+
+  end
+
+  class SUSEInstallMedia < InstallMedia
+
+    def unattended_install_class
+      Vmit::AutoYaST
+    end
+
+    def initrd_path
+      "/boot/#{Vmit::Utils.arch}/loader/initrd"
+    end
+
+    def kernel_path
+      "/boot/#{Vmit::Utils.arch}/loader/linux"
+    end
+  end
+
+  class SUSE9InstallMedia < SUSEInstallMedia
+    def initrd_path
+      "/boot/loader/initrd"
+    end
+
+    def kernel_path
+      "/boot/loader/linux"
+    end
+  end
+
+  class FedoraInstallMedia < InstallMedia
+
+    def unattended_install_class
+      Vmit::Kickstart
+    end
+
+    def initrd_path
+      '/images/pxeboot/initrd.img'
+    end
+
+    def kernel_path
+      '/images/pxeboot/vmlinuz'
+    end
+  end
+
+  class DebianInstallMedia < InstallMedia
+
+    def name
+      'debian'
+    end
+
+    def unattended_install_class
+      Vmit::DebianPreseed
+    end
+
+    def initrd_path
+      arch = Vmit::Utils.arch.gsub(/x86_64/, 'amd64')
+      "/main/installer-#{arch}/current/images/netboot/#{name}-installer/#{arch}/initrd.gz"
+    end
+
+    def kernel_path
+      arch = Vmit::Utils.arch.gsub(/x86_64/, 'amd64')
+      "/main/installer-#{arch}/current/images/netboot/#{name}-installer/#{arch}/linux"
+    end
+  end
+
+  class UbuntuInstallMedia < DebianInstallMedia
+    def name
+      'ubuntu'
+    end
+  end
+
+end