lab 0.1.5 → 0.2.0

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/config/test_lab.yml

@@ -0,0 +1,11 @@
+ - vmid: backtrack
+   driver: workstation
+   location: /opt/vm/backtrack5/Backtrack5x64.vmx
+   modifiers:
+     - Test
+   credentials:
+     - user: root
+       pass: toor
+   os: linux
+   flavor: ubuntu
+   arch: 64

data/config/test_targets.yml

@@ -0,0 +1,21 @@
+ - vmid: metasploitable
+   driver: workstation
+   location: /opt/vm/lab/user/Metasploitable/Metasploitable.vmx
+   tools: false
+   credentials: 
+     - user: msfadmin
+       pass: msfadmin
+ - vmid: windows2000_target
+   driver: workstation
+   location: /opt/vm/lab/vuln/msf_Win2000SP4/Windows 2000 AS.vmx
+   tools: true
+   credentials: 
+ - vmid: windowsxp_target
+   driver: remote_workstation
+   host: vmhost
+   user: root
+   location: /opt/vm/lab/vuln/msf_WinXPSP1/Windows XP Professional.vmx
+   tools: true
+   credentials: 
+     - user: administrator
+       pass: administrator

data/lib/lab/controller/dynagen_controller.rb

@@ -2,13 +2,13 @@ module Lab
 module Controllers
 module DynagenController
 
-	def self.running_list
-		raise "Unsupported"
-	end
+  def self.running_list
+    raise "Unsupported"
+  end
 
-	def self.dir_list(basepath=nil)
-		raise "Unsupported"
-	end
+  def self.dir_list(basepath=nil)
+    raise "Unsupported"
+  end
 end
 end
 end

data/lib/lab/controller/remote_esx_controller.rb

@@ -3,60 +3,60 @@
 
 module Lab
 module Controllers
-module RemoteEsxController
-		
-	# Note that 3.5 was different (vmware-vim-cmd)
-	VIM_CMD = 'vim-cmd'.freeze
-
-	def self.dir_list(basepath=nil)
-		# Does this method really even make sense for esx?
-		return "Unsupported :("
-	end
-
-	def self.running_list(user, host)
-		user.gsub!(/(\W)*/, '')
-		host.gsub!(/(\W)*/, '')
-
-		# first get all registered vms
-		registered_vms = self.get_vms(user, host) || []
-		running_vms = []
-
-		# now let's see which ones are running
-		# TODO:  this is ghetto, would be better not to connect repeatedly
-		registered_vms.each do |vm|
-			remote_cmd = "ssh #{user}@#{host} \"#{VIM_CMD} vmsvc/power.getstate #{vm[:id]}\""
-			raw = `#{remote_cmd}`
-			running_vms << vm if raw =~ /Powered on/			
-		end
-
-		return running_vms
-	end
+module RemoteEsxiController
+    
+  # Note that 3.5 was different (vmware-vim-cmd)
+  VIM_CMD = 'vim-cmd'.freeze
+
+  def self.dir_list(basepath=nil)
+    # Does this method really even make sense for esx?
+    return "Unsupported :("
+  end
+
+  def self.running_list(user, host)
+    user.gsub!(/(\W)*/, '')
+    host.gsub!(/(\W)*/, '')
+
+    # first get all registered vms
+    registered_vms = self.get_vms(user, host) || []
+    running_vms = []
+
+    # now let's see which ones are running
+    # TODO:  this is ghetto, would be better not to connect repeatedly
+    registered_vms.each do |vm|
+      remote_cmd = "ssh #{user}@#{host} \"#{VIM_CMD} vmsvc/power.getstate #{vm[:id]}\""
+      raw = `#{remote_cmd}`
+      running_vms << vm if raw =~ /Powered on/
+    end
+
+    return running_vms
+  end
 
 private 
 
-	def self.get_vms(user, host)
-		user.gsub!(/(\W)*/, '')
-		host.gsub!(/(\W)*/, '')
-		
-		vms = [] # array of VM hashes
-		remote_cmd = "ssh #{user}@#{host} \"#{VIM_CMD} vmsvc/getallvms | grep ^[0-9] | sed 's/[[:blank:]]\\{3,\\}/ /g'\""
-		raw = `#{remote_cmd}`.split("\n")
-
-		raw.each do |line|
-			# So effing ghetto
-			id_and_name = line.split('[datastore').first
-			id = id_and_name.split(' ').first
-	
-			## TODO - there's surely a better way to do this.
-			name_array = id_and_name.split(' ')
-			name_array.shift
-			name = name_array.join(' ')
-			vms << {:id => id, :name => name}
-		end
-		
-		return vms
-	end
-	
+  def self.get_vms(user, host)
+    user.gsub!(/(\W)*/, '')
+    host.gsub!(/(\W)*/, '')
+    
+    vms = [] # array of VM hashes
+    remote_cmd = "ssh #{user}@#{host} \"#{VIM_CMD} vmsvc/getallvms | grep ^[0-9] | sed 's/[[:blank:]]\\{3,\\}/ /g'\""
+    raw = `#{remote_cmd}`.split("\n")
+
+    raw.each do |line|
+      # So effing ghetto
+      id_and_name = line.split('[datastore').first
+      id = id_and_name.split(' ').first
+  
+      ## TODO - there's surely a better way to do this.
+      name_array = id_and_name.split(' ')
+      name_array.shift
+      name = name_array.join(' ')
+      vms << {:id => id, :name => name}
+    end
+    
+    return vms
+  end
+  
 end
 end
 end

data/lib/lab/controller/remote_esxi_controller.rb

@@ -0,0 +1,62 @@
+# This controller was built against: 
+# VMware ESX Host Agent 4.1.0 build-348481
+
+module Lab
+module Controllers
+module RemoteEsxiController
+    
+  # Note that 3.5 was different (vmware-vim-cmd)
+  VIM_CMD = 'vim-cmd'.freeze
+
+  def self.dir_list(basepath=nil)
+    # Does this method really even make sense for esx?
+    return "Unsupported :("
+  end
+
+  def self.running_list(user, host)
+    user.gsub!(/(\W)*/, '')
+    host.gsub!(/(\W)*/, '')
+
+    # first get all registered vms
+    registered_vms = self.get_vms(user, host) || []
+    running_vms = []
+
+    # now let's see which ones are running
+    # TODO:  this is ghetto, would be better not to connect repeatedly
+    registered_vms.each do |vm|
+      remote_cmd = "ssh #{user}@#{host} \"#{VIM_CMD} vmsvc/power.getstate #{vm[:id]}\""
+      raw = `#{remote_cmd}`
+      running_vms << vm if raw =~ /Powered on/
+    end
+
+    return running_vms
+  end
+
+private 
+
+  def self.get_vms(user, host)
+    user.gsub!(/(\W)*/, '')
+    host.gsub!(/(\W)*/, '')
+    
+    vms = [] # array of VM hashes
+    remote_cmd = "ssh #{user}@#{host} \"#{VIM_CMD} vmsvc/getallvms | grep ^[0-9] | sed 's/[[:blank:]]\\{3,\\}/ /g'\""
+    raw = `#{remote_cmd}`.split("\n")
+
+    raw.each do |line|
+      # So effing ghetto
+      id_and_name = line.split('[datastore').first
+      id = id_and_name.split(' ').first
+  
+      ## TODO - there's surely a better way to do this.
+      name_array = id_and_name.split(' ')
+      name_array.shift
+      name = name_array.join(' ')
+      vms << {:id => id, :name => name}
+    end
+    
+    return vms
+  end
+  
+end
+end
+end

data/lib/lab/controller/remote_workstation_controller.rb

@@ -2,21 +2,21 @@ module Lab
 module Controllers
 module RemoteWorkstationController
 
-	def self.running_list(user, host)
-		user.gsub!(/(\W)*/, '')
-		host.gsub!(/(\W)*/, '')
+  def self.running_list(user, host)
+    user.gsub!(/(\W)*/, '')
+    host.gsub!(/(\W)*/, '')
 
-		remote_cmd = "ssh #{user}@#{host} \"vmrun list nogui\""
-		vm_list = `#{remote_cmd}`.split("\n")
-		vm_list.shift
+    remote_cmd = "ssh #{user}@#{host} \"vmrun list nogui\""
+    vm_list = `#{remote_cmd}`.split("\n")
+    vm_list.shift
 
-		return vm_list
-	end
+    return vm_list
+  end
 
-	def self.dir_list(basepath=nil)
-		vm_list = Find.find(basepath).select { |f| f =~ /\.vmx$/ }
-		return vm_list
-	end
+  def self.dir_list(basepath=nil)
+    vm_list = Find.find(basepath).select { |f| f =~ /\.vmx$/ }
+    return vm_list
+  end
 end
 end
 end

data/lib/lab/controller/virtualbox_controller.rb

@@ -2,24 +2,24 @@ module Lab
 module Controllers
 module VirtualBoxController
 
-	def self.running_list
-		vm_names_and_uuids = `VBoxManage list runningvms`
-		return vm_names_and_uuids.scan(/\"(.*)\" {.*}/).flatten
-	end
+  def self.running_list
+    vm_names_and_uuids = `VBoxManage list runningvms`
+    return vm_names_and_uuids.scan(/\"(.*)\" {.*}/).flatten
+  end
 
-	def self.config_list
-		vm_names_and_uuids = `VBoxManage list vms`
-		return vm_names_and_uuids.scan(/\"(.*)\" {.*}/).flatten
-	end
+  def self.config_list
+    vm_names_and_uuids = `VBoxManage list vms`
+    return vm_names_and_uuids.scan(/\"(.*)\" {.*}/).flatten
+  end
 
-	def self.config_list_uuid
-		vm_names_and_uuids = `VBoxManage list vms`
-		return vm_names_and_uuids.scan(/\".*\" {(.*)}/).flatten
-	end
-		
-	def self.dir_list(basepath=nil)
-		vm_list = Find.find(basepath).select { |f| f =~ /\.xml$/ }
-	end
+  def self.config_list_uuid
+    vm_names_and_uuids = `VBoxManage list vms`
+    return vm_names_and_uuids.scan(/\".*\" {(.*)}/).flatten
+  end
+    
+  def self.dir_list(basepath=nil)
+    vm_list = Find.find(basepath).select { |f| f =~ /\.xml$/ }
+  end
 end
 end
 end

data/lib/lab/controller/workstation_controller.rb

@@ -2,16 +2,16 @@ module Lab
 module Controllers
 module WorkstationController
 
-	def self.running_list
-		vm_list = `vmrun list`.split("\n")
-		vm_list.shift
-		return vm_list
-	end
+  def self.running_list
+    vm_list = `vmrun list`.split("\n")
+    vm_list.shift
+    return vm_list
+  end
 
-	def self.dir_list(basepath=nil)
-		vm_list = Find.find(basepath).select { |f| f =~ /\.vmx$/ }
-		return vm_list
-	end
+  def self.dir_list(basepath=nil)
+    vm_list = Find.find(basepath).select { |f| f =~ /\.vmx$/ }
+    return vm_list
+  end
 end
 end
 end

data/lib/lab/controller/workstation_vixr_controller.rb

@@ -2,18 +2,18 @@ module Lab
 module Controllers
 module WorkstationVixrController
 
-	def self.running_list
-		vm_list = `vmrun list`.split("\n")
-		vm_list.shift
+  def self.running_list
+    vm_list = `vmrun list`.split("\n")
+    vm_list.shift
 
-		return vm_list
-	end
+    return vm_list
+  end
 
-	def self.dir_list(basepath=nil)
-		vm_list = Find.find(basepath).select { |f| f =~ /\.vmx$/ }
+  def self.dir_list(basepath=nil)
+    vm_list = Find.find(basepath).select { |f| f =~ /\.vmx$/ }
 
-		return vm_list
-	end
+    return vm_list
+  end
 end
 end
 end

data/lib/lab/controllers.rb

@@ -3,8 +3,6 @@ require 'controller/virtualbox_controller'
 require 'controller/fog_controller'
 require 'controller/dynagen_controller'
 require 'controller/remote_workstation_controller'
-require 'controller/remote_esx_controller'
+require 'controller/remote_esxi_controller'
 #require 'controller/qemu_controller'
 #require 'controller/qemudo_controller'
-
-

data/lib/lab/driver/dynagen_driver.rb

@@ -11,37 +11,37 @@ require 'vm_driver'
 
 module Lab
 module Drivers
-	class DynagenDriver < VmDriver
-		def initialize(config,dynagen_config)
-			super(config)
-			@running = false
-			@dynagen_platform = filter_command(dynagen_config['dynagen_platform'])
-		end
-
-		def start
-			# TODO - write the location-file to a temp-file 
-			#        and set the autostart property 
-
-			## start background dynamips process
-			system_command("dynamips -H #{@dynagen_platform} &")
-			system_command("dynagen	#{@location}")
-			@running = true
-		end
-
-		def stop
-			system_command("killall dynagen")
-			@running = false
-		end
-
-		def cleanup
-			`killall dynagen`
-			`killall dynamips`
-			@running = false
-		end
-
-		def running?
-			return @running
-		end
-	end
+  class DynagenDriver < VmDriver
+    def initialize(config,dynagen_config)
+      super(config)
+      @running = false
+      @dynagen_platform = filter_command(dynagen_config['dynagen_platform'])
+    end
+
+    def start
+      # TODO - write the location-file to a temp-file 
+      #        and set the autostart property 
+
+      ## start background dynamips process
+      system_command("dynamips -H #{@dynagen_platform} &")
+      system_command("dynagen  #{@location}")
+      @running = true
+    end
+
+    def stop
+      system_command("killall dynagen")
+      @running = false
+    end
+
+    def cleanup
+      `killall dynagen`
+      `killall dynamips`
+      @running = false
+    end
+
+    def running?
+      return @running
+    end
+  end
 end
 end