veewee-atlassian 0.3.11

data/lib/veewee/provider/core/provider/tunnel.rb

@@ -0,0 +1,46 @@
+#require 'net/ssh/multi'
+
+module Veewee
+  module Provider
+    module Core
+      module ProviderCommand
+
+
+      def ssh_tunnel_start(forwardings)
+        #ssh_options={ :keys => [ vm.private_key ], :paranoid => false, :keys_only => true}
+
+        ssh_options={ :paranoid => false}
+        host=@connection.uri.host
+        user=@connection.uri.user
+
+        ui.info "Enabling tunneling"
+        @forward_threads=Array.new
+        @forward_threads<< Thread.new {
+          Net::SSH.start(host, user, ssh_options) do |ssh_session|
+          forwardings.each do |forwarding|
+            begin
+              ui.info "Forwarding remote port #{forwarding[:remote_port]} from #{box_name} to local port #{forwarding[:local_port]}"
+              ui.info host
+              ui.info user
+              ui.info ""
+              ssh_session.forward.local(forwarding[:local_port], "127.0.0.1",forwarding[:remote_port])
+            rescue Errno::EACCES
+              ui.info "  Error - Access denied to forward remote port #{forwarding[:remote_port]} from #{name} to local port #{forwarding[:local_port]}"
+            end
+          end
+          ssh_session.loop {true}
+          end
+        }
+        ui.info @forward_threads.first.status
+        @forward_threads.first.run
+        ui.info @forward_threads.first.status
+
+      end
+
+      def ssh_tunnel_stop
+        Thread.kill(@forward_threads.first)
+      end
+    end
+    end
+  end
+end

data/lib/veewee/provider/kvm/box.rb

@@ -0,0 +1,47 @@
+require 'veewee/provider/core/box'
+require 'veewee/provider/core/box/vnc'
+require 'veewee/provider/kvm/box/validate_kvm'
+
+require 'veewee/provider/kvm/box/build'
+require 'veewee/provider/kvm/box/create'
+require 'veewee/provider/kvm/box/up'
+require 'veewee/provider/kvm/box/halt'
+require 'veewee/provider/kvm/box/poweroff'
+require 'veewee/provider/kvm/box/destroy'
+
+require 'veewee/provider/kvm/box/helper/ip'
+require 'veewee/provider/kvm/box/helper/ssh_options'
+require 'veewee/provider/kvm/box/helper/status'
+require 'veewee/provider/kvm/box/helper/console_type'
+
+module Veewee
+  module Provider
+    module Kvm
+      class Box < Veewee::Provider::Core::Box
+
+        include ::Veewee::Provider::Core
+        include ::Veewee::Provider::Kvm
+
+        include ::Veewee::Provider::Core::BoxCommand
+        include ::Veewee::Provider::Kvm::BoxCommand
+
+        attr_accessor :connection
+
+        def initialize(name,env)
+
+          super(name,env)
+
+          @connection=::Fog::Compute[:libvirt]
+
+          # Many of the existing templates have disk_format set to "VDI"
+          # Use "raw" instead as a Libvirt-compatible default
+          definition.disk_format.downcase!
+          definition.disk_format = "raw" if definition.disk_format == "vdi"
+          @volume_name = "#{name}.#{definition.disk_format}"
+
+        end
+
+      end # End Class
+    end # End Module
+  end # End Module
+end # End Module

data/lib/veewee/provider/kvm/box/build.rb

@@ -0,0 +1,11 @@
+module Veewee
+  module Provider
+    module Kvm
+      module BoxCommand
+        def build(options)
+          super(options)
+        end
+      end
+    end # End Module
+  end # End Module
+end # End Module

data/lib/veewee/provider/kvm/box/create.rb

@@ -0,0 +1,184 @@
+require 'nokogiri'
+require 'fileutils'
+
+module Veewee
+  module Provider
+    module Kvm
+      module BoxCommand
+        # Create a new vm
+        def create(options={})
+          # Assemble the Virtualmachine and set all the memory and other stuff"
+          create_server(options)
+          create_volume(options)
+          add_virtio_drivers if File.exists?(File.join(definition.path, 'Autounattend.xml'))
+          self.create_floppy("virtualfloppy.img")
+          FileUtils.move(File.join(definition.path, 'Autounattend.xml.virtio'), File.join(definition.path, 'Autounattend.xml')) if File.exists?(File.join(definition.path, 'Autounattend.xml.virtio'))
+          add_floppy unless definition.floppy_files.nil?
+        end
+
+        def create_server(options)
+
+          # set volume pool name to user specified volume pool and fall back to first available volume pool
+          if options["pool_name"]
+            raise Veewee::Error, "Specified storage pool #{options["pool_name"]} does not exist" if @connection.pools.select { |pool| pool.name == options["pool_name"] }.empty?
+            volume_pool_name = options["pool_name"]
+          end
+          volume_pool_name ||= @connection.pools.first.name
+          env.logger.info "Using storage pool #{volume_pool_name}"
+
+          # set network name to user specified network and fall back to default network or first available network
+          if options["network_name"]
+            raise Veewee::Error, "Specified network #{options["network_name"]} does not exist" if @connection.networks.select { |net| net.name == options["network_name"] }.empty?
+            network_name = options["network_name"]
+          end
+          network_name ||= "default" unless @connection.networks.select { |net| net.name == 'default' }.empty?
+          network_name ||= @connection.networks.first.name
+          env.logger.info "Using network #{network_name}"
+
+          # Create the "server"
+          attributes = {
+              :name => name,
+              :memory_size => definition.memory_size.to_i*1024,
+              :cpus => definition.cpu_count.to_i,
+              :volume_capacity => "#{definition.disk_size}M",
+              :domain_type => options['use_emulation'] ? 'qemu' : 'kvm',
+              :iso_file => definition.iso_file,
+              :arch => definition.os_type_id.end_with?("_64") ? "x86_64" : "i686",
+              :iso_dir => env.config.veewee.iso_dir,
+              :volume_pool_name => volume_pool_name,
+              :volume_format_type => definition.disk_format,
+              :network_nat_network => network_name
+          }
+
+          @connection.servers.create(attributes)
+        end
+
+        # Create the volume of a new vm
+        def create_volume(options)
+          # Creating the volume is part of the server creation
+        end
+
+        def add_floppy
+          env.logger.info 'Adding floppy disk'
+          # Get a raw libvirt connection
+          conn = @connection.client
+
+          # Retrieve the domain
+          domain=conn.lookup_domain_by_name(name)
+
+          # Retrieve the existing XML from the domain
+          domain_xml=domain.xml_desc
+
+          # Convert the xml nokogiri doc
+          domain_doc=Nokogiri::XML(domain_xml)
+
+          # Find the device section
+          devices=domain_doc.xpath('/domain/devices').first
+          # The floppy xml representation
+          floppy_xml="<disk type='file' device='floppy'><driver name='qemu' type='raw'/><source file='"+
+              File.join(definition.path, "virtualfloppy.img") +
+              "'/><target dev='fda' bus='fdc'/><address type='drive' controller='0' bus='0' unit='0'/></disk>
+          <controller type='fdc' index='0'>"
+
+          # Convert the floppy xml to nokogiri
+          floppy_doc=Nokogiri::XML(floppy_xml)
+
+          # Add the floppy to the devices section
+          devices.add_child(floppy_doc.root)
+
+          # Get the raw xml of the changed document
+          new_xml=domain_doc.to_xml
+
+          # Undefine the existing domain
+          domain.undefine
+
+          # Re-define the domain
+          conn.define_domain_xml(new_xml)
+        end
+
+        def add_virtio_drivers
+          env.logger.info 'Adding virtio drivers for windows system to the virtual machine'
+          # Get a raw libvirt connection
+          conn = @connection.client
+
+          # Retrieve the domain
+          domain=conn.lookup_domain_by_name(name)
+
+          # Retrieve the existing XML from the domain
+          domain_xml=domain.xml_desc
+
+          # Convert the xml nokogiri doc
+          domain_doc=Nokogiri::XML(domain_xml)
+
+          # Find the device section
+          devices=domain_doc.xpath('/domain/devices').first
+
+          # get latest version of virtio drivers
+          url ='http://alt.fedoraproject.org/pub/alt/virtio-win/latest/images/bin/'
+          filename = open(url).read.scan(/\"(virtio-win-.*.iso)\"/).first.first
+          download_iso(url + filename, filename)
+          path = File.join(env.config.veewee.iso_dir, filename)
+
+          # The disk xml representation
+          disk_xml="<disk type='file' device='cdrom'><driver name='qemu' type='raw'/><source file='" +
+              path + "'/><target dev='hdd' bus='ide'/></disk>"
+
+          # Convert the disk xml to nokogiri
+          disk_doc=Nokogiri::XML(disk_xml)
+
+          # Add the floppy to the devices section
+          devices.add_child(disk_doc.root)
+
+          # Get the raw xml of the changed document
+          new_xml=domain_doc.to_xml
+
+          # Undefine the existing domain
+          domain.undefine
+
+          # Re-define the domain
+          conn.define_domain_xml(new_xml)
+
+          env.logger.info 'Add search path for virtio drivers to Autounattend.xml'
+          # parse Autoattend.xml to document
+          FileUtils.copy(File.join(definition.path, 'Autounattend.xml'), File.join(definition.path, 'Autounattend.xml.virtio'))
+          doc = Nokogiri::XML.parse(File.read(File.join(definition.path, 'Autounattend.xml')))
+          # determine platform and windows version
+          platform = definition.os_type_id.end_with?("_64") ? "amd64" : "x86"
+          version = case definition.os_type_id.downcase
+                      when /windows-?7/
+                        'win7'
+                      when /windows-?2008/
+                        'win7'
+                      when /windows-?8/
+                        'win8'
+                      when /xp/
+                        'xp'
+                      when /vista/
+                        'vista'
+                      else
+                        raise 'could not determine windows version'
+                    end
+          # create new element
+          component=Nokogiri::XML(%Q|<component name="Microsoft-Windows-PnpCustomizationsWinPE"
+processorArchitecture="#{platform}" publicKeyToken="31bf3856ad364e35"
+language="neutral" versionScope="nonSxS"
+xmlns:wcm="http://schemas.microsoft.com/WMIConfig/2002/State"
+xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
+<DriverPaths>
+<PathAndCredentials wcm:keyValue="1" wcm:action="add">
+<Path>e:\\#{version}\\#{platform}</Path>
+</PathAndCredentials>
+</DriverPaths>
+</component>|)
+          doc.xpath('//unattend:settings[@pass="windowsPE"]', 'unattend' => 'urn:schemas-microsoft-com:unattend').first.add_child component.root
+
+          file = File.open(File.join(definition.path, 'Autounattend.xml'), 'w')
+          file.write(doc)
+          file.close
+
+        end
+
+      end #End Module
+    end # End Module
+  end # End Module
+end # End Module

data/lib/veewee/provider/kvm/box/destroy.rb

@@ -0,0 +1,32 @@
+module Veewee
+  module Provider
+    module Kvm
+      module BoxCommand
+        # Destroy a vm
+        def destroy(options={})
+          unless exists_vm? or exists_volume?
+            env.ui.error "Error:: You tried to destroy a non-existing box '#{name}'"
+            raise Veewee::Error,"Error:: You tried to destroy a non-existing box '#{name}'"
+          end
+
+          self.poweroff if running?
+          destroy_vm if exists_vm?
+
+          vol_exists = exists_volume?
+          env.logger.info "Volume exists?: #{vol_exists}"
+          destroy_volume if vol_exists
+        end
+
+        def destroy_vm
+          matched_servers=@connection.servers.all(:name => name)
+          matched_servers.first.destroy() unless matched_servers.nil?
+        end
+
+        def destroy_volume
+          @connection.volumes.all(:name => @volume_name).first.destroy
+        end
+      end # End Module
+
+    end # End Module
+  end # End Module
+end # End Module

data/lib/veewee/provider/kvm/box/halt.rb

@@ -0,0 +1,19 @@
+module Veewee
+  module Provider
+    module Kvm
+      module BoxCommand
+
+        def halt(options={})
+          matched_servers=@connection.servers.all(:name => name)
+          matched_servers.first.halt unless matched_servers.nil?
+        end
+
+        def stop(options={})
+          matched_servers=@connection.servers.all(:name => name)
+          matched_servers.first.stop unless matched_servers.nil?
+        end
+
+      end # End Module
+    end # End Module
+  end # End Module
+end # End Module

data/lib/veewee/provider/kvm/box/helper/console_type.rb

@@ -0,0 +1,20 @@
+require 'veewee/provider/core/box'
+require 'veewee/provider/core/box/vnc'
+require 'veewee/provider/kvm/box/validate_kvm'
+
+module Veewee
+  module Provider
+    module Kvm
+      module BoxCommand
+        # Type on the console
+        def console_type(sequence,type_options={})
+          tcp_port=@connection.servers.all(:name => name).first.display[:port]
+          display_port=tcp_port.to_i - 5900
+          ui.success "Sending keystrokes to VNC port :#{display_port} - TCP port: #{tcp_port}"
+          vnc_type(sequence,"127.0.0.1",display_port)
+        end
+
+      end # End Module
+    end # End Module
+  end # End Module
+end # End Module

data/lib/veewee/provider/kvm/box/helper/ip.rb

@@ -0,0 +1,14 @@
+module Veewee
+  module Provider
+    module Kvm
+      module BoxCommand
+        def ip_address
+          ip=@connection.servers.all(:name => "#{name}").first.public_ip_address
+          return [*ip].first unless ip.nil?
+          return ip
+        end
+
+      end # End Module
+    end # End Module
+  end # End Module
+end # End Module

data/lib/veewee/provider/kvm/box/helper/ssh_options.rb

@@ -0,0 +1,19 @@
+module Veewee
+  module Provider
+    module Kvm
+      module BoxCommand
+
+        def ssh_options
+          ssh_options={
+            :user => definition.ssh_user,
+            :port => 22,
+            :password => definition.ssh_password,
+            :timeout => definition.ssh_login_timeout.to_i
+          }
+          return ssh_options
+        end
+
+      end # End Module
+    end # End Module
+  end # End Module
+end # End Module

data/lib/veewee/provider/kvm/box/helper/status.rb

@@ -0,0 +1,32 @@
+module Veewee
+  module Provider
+    module Kvm
+      module BoxCommand
+        def running?
+          if exists_vm?
+            @connection.servers.all(:name => name).first.ready?
+          else
+            false
+          end
+        end
+
+        def exists?
+          exists_volume? || exists_vm?
+        end
+
+        def exists_volume?
+          !@connection.list_volumes(:name => @volume_name).first.empty?
+        end
+
+        def exists_vm?
+          begin
+            @connection.list_domains(:name => name)
+          rescue Libvirt::RetrieveError
+            false
+          end
+        end
+
+      end # End Module
+    end # End Module
+  end # End Module
+end # End Module

data/lib/veewee/provider/kvm/box/poweroff.rb

@@ -0,0 +1,14 @@
+module Veewee
+  module Provider
+    module Kvm
+      module BoxCommand
+
+        def poweroff(options={})
+          matched_servers=@connection.servers.all(:name => name)
+          matched_servers.first.stop unless matched_servers.nil?
+        end
+
+      end # End Module
+    end # End Module
+  end # End Module
+end # End Module