lab 0.2.0 → 0.2.2

data/lab.gemspec

@@ -29,7 +29,7 @@ Gem::Specification.new do |s|
     s.add_runtime_dependency "net-ssh"
     s.add_runtime_dependency "net-scp"
 
-    # Vmware ESX driver
+    # Vmware vsphere driver
     s.add_runtime_dependency "rbvmomi"
     
 end

data/lib/lab/controller/remote_esxi_controller.rb

@@ -9,7 +9,7 @@ module RemoteEsxiController
   VIM_CMD = 'vim-cmd'.freeze
 
   def self.dir_list(basepath=nil)
-    # Does this method really even make sense for esx?
+    # Does this method really even make sense for esxi?
     return "Unsupported :("
   end
 

data/lib/lab/controller/{remote_esx_controller.rb → vsphere_controller.rb}

@@ -9,7 +9,7 @@ module RemoteEsxiController
   VIM_CMD = 'vim-cmd'.freeze
 
   def self.dir_list(basepath=nil)
-    # Does this method really even make sense for esx?
+    # Does this method really even make sense for esxi?
     return "Unsupported :("
   end
 
@@ -47,7 +47,7 @@ private
       id_and_name = line.split('[datastore').first
       id = id_and_name.split(' ').first
   
-      ## TODO - there's surely a better way to do this.
+      # TODO - there's surely a better way to do this.
       name_array = id_and_name.split(' ')
       name_array.shift
       name = name_array.join(' ')

data/lib/lab/controllers.rb

@@ -4,5 +4,6 @@ require 'controller/fog_controller'
 require 'controller/dynagen_controller'
 require 'controller/remote_workstation_controller'
 require 'controller/remote_esxi_controller'
+require 'controller/vsphere_controller'
 #require 'controller/qemu_controller'
 #require 'controller/qemudo_controller'

data/lib/lab/driver/fog_driver.rb

@@ -17,7 +17,7 @@ class FogDriver < VmDriver
     begin
       require 'fog'
     rescue LoadError
-      raise "WARNING: Library fog not found. Could Not Create Driver"
+      raise "WARNING: Library fog not found. Could not create driver"
     end
 
     if @fog_config['fog_type'] == "ec2"
@@ -44,18 +44,17 @@ class FogDriver < VmDriver
         :aws_access_key_id => @aws_access_key_file,
         :aws_secret_access_key => @aws_secret_access_key_file )
     else
-      raise "Unsupported Fog Type"
+      raise "Unsupported fog type"
     end
   end
 
   def start
     ec2_settings = {
-      :image_id => @ec2_base_ami, 
+      :image_id => @ec2_base_ami,
       :flavor_id =>  @ec2_flavor,
       :public_key_path => @ec2_instance_public_key_file,
       :private_key_path => @ec2_instance_private_key_file,
       :username => @ec2_user}
-      
     begin
       @fog_server = @compute.servers.bootstrap(ec2_settings)
     rescue Fog::Compute::AWS::Error => e
@@ -92,65 +91,6 @@ class FogDriver < VmDriver
     raise "unimplemented"
   end
 
-=begin
-
-  def run_command(command)
-    ## vm_driver will need a little patching for this to work, as 
-    ## amis use keys for auth. i think it's just a matter of not passing the 
-    ## password to ssh_exec. So maybe the thing to do is have a ssh_key_exec
-    ## function in vm_driver.rb that does the right thing. 
-
-    script_rand_name = rand(10000)
-
-    if @os == "windows"
-      local_tempfile_path = "/tmp/lab_script_#{script_rand_name}.bat"
-      remote_tempfile_path = "C:\\\\lab_script_#{script_rand_name}.bat"
-      remote_run_command = remote_tempfile_path
-    else
-      local_tempfile_path = "/tmp/lab_script_#{script_rand_name}.sh"
-      remote_tempfile_path = "/tmp/lab_script_#{script_rand_name}.sh"
-      remote_run_command = "/bin/sh #{remote_tempfile_path}"
-    end
-
-    # write out our script locally
-    File.open(local_tempfile_path, 'w') {|f| f.write(command) }
-
-    # since we can't copy easily w/o tools, let's just run it directly :/
-    if @os == "linux"
-      output_file = "/tmp/lab_command_output_#{rand(1000000)}"
-      
-      scp_to(local_tempfile_path, remote_tempfile_path)
-      ssh_exec(remote_run_command + "> #{output_file}")
-      scp_from(output_file, output_file)
-      ssh_exec("rm #{output_file}")
-      ssh_exec("rm #{remote_tempfile_path}")
-      
-      # Ghettohack!
-      string = File.open(output_file,"r").read
-      `rm #{output_file}`
-      
-    else
-      raise "zomgwtfbbqnotools"
-    end
-  end
-  
-  def copy_from(from, to)
-    raise "unimplemented"
-  end
-
-  def copy_to(from, to)
-    raise "unimplemented"
-  end
-
-  def check_file_exists(file)
-    raise "unimplemented"
-  end
-
-  def create_directory(directory)
-    raise "unimplemented"
-  end
-=end
-
   def cleanup
     @fog_server.destroy
   end

data/lib/lab/driver/vsphere_driver.rb

@@ -0,0 +1,120 @@
+require 'vm_driver'
+
+##
+## $Id$
+##
+
+# This driver was built against: 
+# VMware Vsphere 4.1
+
+module Lab
+module Drivers
+
+class VsphereDriver < VmDriver
+
+  def initialize(config)
+    unless config['user'] then raise ArgumentError, "Must provide a username" end
+    unless config['host'] then raise ArgumentError, "Must provide a hostname" end
+    unless config[''] then raise ArgumentError, "Must provide a password" end
+    super(config)
+
+    @user = filter_command(config['user'])
+    @host = filter_command(config['host'])
+
+    # Soft dependency
+    begin
+      require 'rbvmomi'
+    rescue LoadError
+      raise "WARNING: Library rbvmomi not found. Could not create driver!"
+    end
+
+    vim = RbVmomi::VIM.connect host: @host, user: @user, password: @pass
+    dc = vim.serviceInstance.find_datacenter("datacenter1") or fail "datacenter not found"
+    @vm = dc.find_vm("test") or fail "VM not found"
+  end
+
+  def start
+    @vm.PowerOnVM_Task.wait_for_completion
+  end
+
+  def stop
+    @vm.PowerOffVM_Task.wait_for_completion
+  end
+
+  def suspend
+    @vm.SuspendVM_Task.wait_for_completion
+  end
+
+  def pause
+    raise "Unimplemented"
+  end
+
+  def resume
+    raise "Unimplemented"
+  end
+
+  def reset
+    @vm.ResetVM_Task.wait_for_completion
+  end
+
+  def create_snapshot(snapshot)
+    snapshot = filter_input(snapshot)
+    raise "Unimplemented"
+  end
+
+  def revert_snapshot(snapshot)
+    raise "Unimplemented"
+    # If we got here, the snapshot didn't exist
+    raise "Invalid Snapshot Name"
+  end
+
+  def delete_snapshot(snapshot, remove_children=false)
+    raise "Unimplemented"
+    # If we got here, the snapshot didn't exist
+    raise "Invalid Snapshot Name"
+  end
+  
+  def delete_all_snapshots
+    raise "Unimplemented"
+  end
+    
+  def run_command(command)
+    raise "Unimplemented"
+  end
+  
+  def copy_from(from, to)
+    if @os == "linux"
+      scp_from(from, to)
+    else
+      raise "Unimplemented"
+    end
+  end
+
+  def copy_to(from, to)
+    if @os == "linux"
+      scp_to(from, to)
+    else
+      raise "Unimplemented"
+    end
+  end
+
+  def check_file_exists(file)
+    raise "Unimplemented"
+  end
+
+  def create_directory(directory)
+    raise "Unimplemented"
+  end
+
+  def cleanup
+    raise "Unimplemented"
+  end
+
+  def running?
+    raise "Unimplemented"
+  end 
+
+end
+
+end 
+end

data/lib/lab/drivers.rb

@@ -4,5 +4,6 @@ require 'driver/fog_driver'
 require 'driver/dynagen_driver'
 require 'driver/remote_workstation_driver'
 require 'driver/remote_esxi_driver'
+require 'driver/vsphere_driver'
 #require 'driver/qemu_driver'
 #require 'driver/qemudo_driver'

data/lib/lab/modifier/meterpreter_modifier.rb

@@ -0,0 +1,167 @@
+$:.unshift(File.join(File.dirname(__FILE__), '..', '..'))
+
+module Lab
+module Modifier
+module Meterpreter
+
+end
+end
+end
+
+
+# This allows us to override the default way of running commands
+# Currently useful for the esx controller
+
+module Lab
+class Vm
+	
+	attr_accessor :framework
+	attr_accessor :session
+	attr_accessor :session_input
+	attr_accessor :session_output
+
+	def create_framework
+		return if @framework
+		@framework    = Msf::Simple::Framework.create
+	end
+	
+	# perform the setup only once
+	def setup_session
+		return if @session
+
+		# require the framework (assumes this sits in lib/lab/modifiers)
+		require 'msf/base'
+
+		create_framework 	## TODO - this should use a single framework 
+					## for all hosts, not one-per-host
+
+		@session 		= nil
+		@session_input        	= Rex::Ui::Text::Input::Buffer.new
+		@session_output       	= Rex::Ui::Text::Output::Buffer.new
+	
+		if @os == "windows"
+			exploit_name = 'windows/smb/psexec'
+
+			# TODO - check for x86, choose the appropriate payload
+
+			payload_name = 'windows/meterpreter/bind_tcp'
+			options = {	"RHOST"		  => @hostname, 
+					        "SMBUser" 	=> @vm_user, 
+					        "SMBPass" 	=> @vm_pass}
+
+      			puts "DEBUG: using options #{options}"
+
+			# Initialize the exploit instance
+			exploit = @framework.exploits.create(exploit_name)
+
+			begin
+				# Fire it off.
+				@session = exploit.exploit_simple(
+					'Payload'     	=> payload_name,
+					'Options'     	=> options,
+					'LocalInput'  	=> @session_input,
+					'LocalOutput' 	=> @session_output)
+				@session.load_stdapi
+				
+				puts "DEBUG: Generated session: #{@session}"
+				
+			rescue  Exception => e 
+  			  puts "DEBUG: Unable to exploit"
+  			  puts e.to_s
+			end
+							
+		else
+			module_name = 'scanner/ssh/ssh_login'
+			
+			# TODO - check for x86, choose the appropriate payload
+			
+			payload_name = 'linux/x86/shell_bind_tcp'
+			options = {	"RHOSTS"		=> @hostname, 
+					"USERNAME" 		=> @vm_user, 
+					"PASSWORD" 		=> @vm_pass, 
+					"BLANK_PASSWORDS" 	=> false, 
+					"USER_AS_PASS" 		=> false, 
+					"VERBOSE" 		=> false}
+
+      			puts "DEBUG: using options #{options}"
+
+			# Initialize the module instance
+			aux = @framework.auxiliary.create(module_name)
+			
+			puts "DEBUG: created module: #{aux}"
+			
+			begin 
+				# Fire it off.
+				aux.run_simple(
+					'Payload'     => payload_name,
+					'Options'     => options,
+					'LocalInput'  => @session_input,
+					'LocalOutput' => @session_output)
+				
+				@session = @framework.sessions.first.last
+				puts "DEBUG: Generated session: #{@session}"
+			rescue Exception => e 
+			  puts "DEBUG: Unable to exploit"
+			  puts e.to_s
+			end
+		end
+		
+
+		
+	end
+
+	def run_command(command, timeout=60)
+		
+		setup_session
+		puts "Using session #{@session}"
+		
+		# TODO: pass the timeout down
+	
+		if @session
+			if @session.type == "shell"
+				puts "Running command via shell: #{command}"		
+				@session.shell_command_token(command, timeout)
+			elsif @session.type == "meterpreter" 
+				puts "Running command via meterpreter: #{command}"		
+				@session.shell_command(command) #, timeout)
+			end
+		else
+			raise "No session"
+		end
+	end
+	
+	
+	# This isn't part of the normal API, but too good to pass up. 
+	def run_script(script, options)
+		if @session.type == "meterpreter"
+			@session.execute_script(script, options)
+		else
+			raise "Unsupported on #{@session.type}"
+		end
+	end
+	
+	# For meterpreter API compatibility
+	#def execute_file(script,options)
+	#	run_script(script,options)
+	#end
+	
+	def copy_to(local,remote)
+		setup_session
+		if @session.type == "meterpreter"
+			@session.run_cmd("upload #{local} #{remote}")
+		else
+			@driver.copy_to(local,remote)
+		end
+	end
+	
+	def copy_from(local, remote)
+		setup_session
+		if @session.type == "meterpreter"
+			@session.run_cmd("download #{local} #{remote}")
+		else
+			@driver.copy_from(local,remote)
+		end
+	end
+	
+end
+end

data/lib/lab/version.rb

@@ -1,3 +1,3 @@
 module Lab
-  VERSION = "0.2.0"
+  VERSION = "0.2.2"
 end

data/lib/lab/vm.rb

@@ -7,28 +7,34 @@ class Vm
   
   attr_accessor :vmid
   attr_accessor :hostname
-  attr_accessor :name
   attr_accessor :description
+  attr_accessor :user
+  attr_accessor :host
   attr_accessor :location
   attr_accessor :driver
   attr_accessor :credentials
   attr_accessor :tools
   attr_accessor :type
-  attr_accessor :user
-  attr_accessor :host
   attr_accessor :os
   attr_accessor :arch
+  attr_accessor :tags
 
   ## Initialize takes a vm configuration hash of the form
-  ##  - vmid (unique identifier)
-  ##    driver (vm technology)
+  ##  - vmid (unique id)
+  ##    hostname (unique name)
+  ##    description (describes the vm's contents)
+  ##    driver (vm driver)
+  ##    location (if applicable - local system)
   ##    user (if applicable - remote system)
   ##    host (if applicable - remote system)
-  ##    pass (if applicable - remote system)
-  ##    location (file / uri)
+  ##    pass (if applicable - remote system) # DISCOURAGED - USE KEYS!
+  ##    tools (true if tools are installed)
+  ##    type ( qa / vulnerable / etc - freeform option)
   ##    credentials (of the form [ {'user'=>"user",'pass'=>"pass", 'admin' => false}, ... ])
-  ##    os (currently only linux / windows)
-  ##    arch (currently only 32 / 64
+  ##    os (currently linux / windows / solaris / aix) - may be used in modifiers
+  ##    arch (currently 32 / 64)
+  ##    modifiers - can be anything in the modifiers directory
+  ##    tags - list of strings associated with this vm
   
   def initialize(config = {})  
 
@@ -48,7 +54,6 @@ class Vm
     @driver_type.downcase!
 
     @location = filter_input(config['location'])
-    #@name = config['name'] || ""
     @description = config['description']
     @tools = config['tools']
     @os = config['os']
@@ -67,19 +72,19 @@ class Vm
     @user = filter_input(config['user']) || nil
     @host = filter_input(config['host']) || nil
     @port = filter_input(config['port']) || nil
-    @pass = filter_input(config['pass']) || nil
+    @pass = filter_input(config['pass']) || nil # DISCORAGED, use keys!
 
-    #Only dynagen systems need this
+    # Only dynagen systems need this
     @platform = config['platform']
 
-    #Only fog systems need this
+    # Only fog systems need this
     @fog_config = config['fog_config']
 
-    #puts "Passing driver config: #{config}"
-
     # Process the correct driver
     if @driver_type == "workstation"
       @driver = Lab::Drivers::WorkstationDriver.new(config)
+    elsif @driver_type == "remote_workstation"
+      @driver = Lab::Drivers::RemoteWorkstationDriver.new(config)
     elsif @driver_type == "virtualbox"
       @driver = Lab::Drivers::VirtualBoxDriver.new(config)
     elsif @driver_type == "fog"
@@ -88,8 +93,8 @@ class Vm
       @driver = Lab::Drivers::DynagenDriver.new(config, config['dynagen_config'])  
     elsif @driver_type == "remote_esxi"
       @driver = Lab::Drivers::RemoteEsxiDriver.new(config)
-    elsif @driver_type == "remote_workstation"
-      @driver = Lab::Drivers::RemoteWorkstationDriver.new(config)
+    elsif @driver_type == "vsphere"
+      @driver = Lab::Drivers::VsphereDriver.new(config)
     #elsif @driver_type == "qemu"
     #  @driver = Lab::Drivers::QemuDriver.new
     #elsif @driver_type == "qemudo"
@@ -110,8 +115,14 @@ class Vm
          @modifiers.each { |modifier|  self.class.send(:include, eval("Lab::Modifier::#{modifier}"))}
       rescue Exception => e
         # modifier likely didn't exist
-      end     
+      end
     end
+    
+    #
+    # Grab a tags array
+    #
+    @tags = config['tags']
+    
   end
   
   def running?
@@ -205,7 +216,9 @@ class Vm
     
     # Standard configuration options
     out =  " - vmid: #{@vmid}\n"
+    out += "   hostname: #{@hostname}\n"
     out += "   driver: #{@driver_type}\n"
+    out += "   description: |\n #{@description}\n"
     out += "   location: #{@location}\n"
     out += "   type: #{@type}\n"
     out += "   tools: #{@tools}\n"
@@ -215,6 +228,8 @@ class Vm
     if @user or @host # Remote vm/drivers only
       out += "   user: #{@user}\n"
       out += "   host: #{@host}\n"
+      out += "   port: #{@port}\n"
+      out += "   pass: #{@pass}\n"
     end
 
     if @platform

data/lib/lab/vm_controller.rb

@@ -1,10 +1,10 @@
-##
-## $Id$
-##
-## This is the main lab controller. Require this controller to get all 
-## lab functionality. 
-##
-##
+#
+# $Id$
+#
+# This is the main lab controller. Require this controller to get all 
+# lab functionality. 
+#
+#
 
 $:.unshift(File.expand_path(File.dirname(__FILE__)))
 $:.unshift(File.expand_path(File.join(File.dirname(__FILE__), 'driver')))
@@ -29,16 +29,17 @@ module Controllers
   class VmController 
 
     include Enumerable
-    include Lab::Controllers::WorkstationController   
+    include Lab::Controllers::WorkstationController
+    include Lab::Controllers::RemoteWorkstationController
     include Lab::Controllers::VirtualBoxController 
     include Lab::Controllers::FogController
     include Lab::Controllers::DynagenController 
     include Lab::Controllers::RemoteEsxiController
-    include Lab::Controllers::RemoteWorkstationController   
+    include Lab::Controllers::VsphereDriver
     #include Lab::Controllers::QemuController 
     #include Lab::Controllers::QemudoController 
     def initialize (labdef=nil)
-      
+
       # Start with an empty array of vm objects
       @vms = [] 
 
@@ -59,23 +60,31 @@ module Controllers
       end
     end
 
-    def find_by_vmid(vmid)
+    def find_by_vmid(search)
+      @vms.each do |vm|
+        return vm if vm.hostname.to_s.downcase == search.to_s.downcase
+      end
+      return nil
+    end
+
+    def find_by_tag(search)
       @vms.each do |vm|
-        if (vm.hostname.to_s.downcase == vmid.to_s.downcase)
-          return vm
+        vm.tags.each do |tag| 
+          return vm if tag.downcase == search.to_s.downcase
         end
       end
       return nil
     end
 
-    def add_vm(vmid, location=nil, os=nil, tools=nil, credentials=nil, user=nil, host=nil)      
-      @vms << Vm.new( {  
+    def add_vm(vmid, location=nil, os=nil, tools=nil, credentials=nil, user=nil, host=nil)
+      @vms << Vm.new( {
         'vmid' => vmid, 
         'driver' => type, 
         'location' => location, 
         'credentials' => credentials,
         'user' => user,
-        'host' => host} )
+        'host' => host
+        })
     end
 
     def remove_by_vmid(vmid)
@@ -112,24 +121,30 @@ module Controllers
     false
     end
 
+    #
+    # Build a vm lab from a directory of files. Really only useful for file-based
+    # vm hosts. (vmware workstation)
+    #
     def build_from_dir(driver_type, dir, clear=false)
-    
+
       if clear
         @vms = []
       end
 
       if driver_type.downcase == "workstation"
         vm_list = ::Lab::Controllers::WorkstationController::dir_list(dir)
+      elsif driver_type.downcase == "remote_workstation"  
+        vm_list = ::Lab::Controllers::RemoteWorkstationController::dir_list(dir)
       elsif driver_type.downcase == "virtualbox"  
         vm_list = ::Lab::Controllers::VirtualBoxController::dir_list(dir)
       elsif driver_type.downcase == "fog"
         vm_list = ::Lab::Controllers::FogController::dir_list(dir)
       elsif driver_type.downcase == "Dynagen"  
         vm_list = ::Lab::Controllers::DynagenController::dir_list(dir)
-      elsif driver_type.downcase == "remote_workstation"  
-        vm_list = ::Lab::Controllers::RemoteWorkstationController::dir_list(dir)
       elsif driver_type.downcase == "remote_esxi"
         vm_list =::Lab::Controllers::RemoteEsxiController::dir_list(dir)
+      elsif driver_type.downcase == "vsphere"
+        vm_list =::Lab::Controllers::VsphereController::dir_list(dir)
       else
         raise TypeError, "Unsupported VM Type"
       end
@@ -139,7 +154,12 @@ module Controllers
       end
     end
 
-    def build_from_running(driver_type=nil, user=nil, host=nil, clear=false)
+
+    #
+    # Builds a vm lab from all running vms. Handy for connecting and saving out 
+    # a config or just managing the currently running vms
+    #
+    def build_from_running(driver_type=nil, user=nil, host=nil, clear=false, pass=nil)
     
       if clear
         @vms = []
@@ -148,69 +168,77 @@ module Controllers
       case driver_type.intern
         when :workstation
           vm_list = ::Lab::Controllers::WorkstationController::running_list
-          
           vm_list.each do |item|
-      
-            ## Name the VM
+            # Name the VM
             index = @vms.count + 1
-  
-            ## Add it to the vm list
-            @vms << Vm.new( {  
+            # Add it to the vm list
+            @vms << Vm.new({
+              'vmid' => "vm_#{index}",
+              'driver' => driver_type, 
+              'location' => item
+              })
+          end
+        when :remote_workstation
+          vm_list = ::Lab::Controllers::RemoteWorkstationController::running_list(user, host)
+          vm_list.each do |item|
+            # Name the VM
+            index = @vms.count + 1
+            # Add it to the VM list
+            @vms << Vm.new({
               'vmid' => "vm_#{index}",
               'driver' => driver_type, 
               'location' => item, 
               'user' => user,
-              'host' => host } )
+              'host' => host 
+              })
           end
-          
-          
         when :virtualbox
           vm_list = ::Lab::Controllers::VirtualBoxController::running_list
           vm_list.each do |item|
-            ## Add it to the vm list
+            # Add it to the vm list
             @vms << Vm.new( {  
               'vmid' => "#{item}",
               'driver' => driver_type,
-              'location' => nil })
+              'location' => nil 
+              })
           end
         when :fog
-          raise "Unsupported" # TODO - figure out a way to allow this
+          raise "Unsupported"
         when :dynagen
           raise "Unsupported"
-        when :remote_workstation
-          vm_list = ::Lab::Controllers::RemoteWorkstationController::running_list(user, host)
-          
+        when :remote_esxi
+          vm_list = ::Lab::Controllers::RemoteEsxiController::running_list(user,host)
           vm_list.each do |item|
-      
-            ## Name the VM
-            index = @vms.count + 1
-  
-            ## Add it to the vm list
             @vms << Vm.new( {  
-              'vmid' => "vm_#{index}",
+              'vmid' => "#{item[:id]}",
+              'name' => "#{item[:name]}",
               'driver' => driver_type, 
-              'location' => item, 
               'user' => user,
-              'host' => host } )
+              'host' => host 
+              })
           end
-        when :remote_esxi
-          vm_list = ::Lab::Controllers::RemoteEsxiController::running_list(user,host)
-          
+        when :vsphere
+          vm_list = ::Lab::Controllers::VsphereController::running_list(user,host,pass)
           vm_list.each do |item|
             @vms << Vm.new( {  
               'vmid' => "#{item[:id]}",
               'name' => "#{item[:name]}",
               'driver' => driver_type, 
               'user' => user,
-              'host' => host } )
+              'host' => host,
+              'pass' => pass
+              })
           end
-            
         else
           raise TypeError, "Unsupported VM Type"
         end
 
     end  
 
+    #
+    # Applicable only to virtualbox. Reads the config file & parses / creates
+    # VM objects for each vm.
+    #
     def build_from_config(driver_type=nil, user=nil, host=nil, clear=false)
       if clear
         @vms = []
@@ -221,7 +249,7 @@ module Controllers
           vm_list = ::Lab::Controllers::VirtualBoxController::config_list
           
           vm_list.each do |item|
-            ## Add it to the vm list
+            # Add it to the vm list
             @vms << Vm.new( {  
               'vmid' => "#{item}",
               'driver' => driver_type, 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lab
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-04-02 00:00:00.000000000 Z
+date: 2012-04-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: nokogiri
-  requirement: &14248700 !ruby/object:Gem::Requirement
+  requirement: &21345740 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *14248700
+  version_requirements: *21345740
 - !ruby/object:Gem::Dependency
   name: net-ssh
-  requirement: &14247980 !ruby/object:Gem::Requirement
+  requirement: &21329820 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,10 +32,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *14247980
+  version_requirements: *21329820
 - !ruby/object:Gem::Dependency
   name: net-scp
-  requirement: &14229460 !ruby/object:Gem::Requirement
+  requirement: &21328760 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -43,10 +43,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *14229460
+  version_requirements: *21328760
 - !ruby/object:Gem::Dependency
   name: rbvmomi
-  requirement: &14228480 !ruby/object:Gem::Requirement
+  requirement: &21328000 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -54,7 +54,7 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *14228480
+  version_requirements: *21328000
 description: ! 'Start/Stop/Revert and do other cool stuff w/ Vmware, Virtualbox, and
   ESXi vms. This gem wraps common CLI utilities and other gems to create a common
   inteface for vms. '
@@ -76,10 +76,10 @@ files:
 - lib/lab.rb
 - lib/lab/controller/dynagen_controller.rb
 - lib/lab/controller/fog_controller.rb
-- lib/lab/controller/remote_esx_controller.rb
 - lib/lab/controller/remote_esxi_controller.rb
 - lib/lab/controller/remote_workstation_controller.rb
 - lib/lab/controller/virtualbox_controller.rb
+- lib/lab/controller/vsphere_controller.rb
 - lib/lab/controller/workstation_controller.rb
 - lib/lab/controller/workstation_vixr_controller.rb
 - lib/lab/controllers.rb
@@ -89,18 +89,19 @@ files:
 - lib/lab/driver/remote_workstation_driver.rb
 - lib/lab/driver/virtualbox_driver.rb
 - lib/lab/driver/vm_driver.rb
+- lib/lab/driver/vsphere_driver.rb
 - lib/lab/driver/workstation_driver.rb
 - lib/lab/driver/workstation_vixr_driver.rb
 - lib/lab/drivers.rb
 - lib/lab/modifier/backtrack5_modifier.rb
 - lib/lab/modifier/dos_modifier.rb
+- lib/lab/modifier/meterpreter_modifier.rb
 - lib/lab/modifier/test_modifier.rb
 - lib/lab/modifiers.rb
 - lib/lab/version.rb
 - lib/lab/vm.rb
 - lib/lab/vm_controller.rb
 - src/Gemfile
-- src/README.md
 - src/Rakefile
 - src/TODO
 - src/config/test_lab.yml

data/src/README.md

@@ -1,80 +0,0 @@
-This folder contains the libraries necessary to run the lab plugin, and can also be used in a standalone way to automate virtual machines.
-
-CONCEPTS:
-=========
-
-The lab provides a clean interface to common vm functions such as start / stop / snapshot / revert and even running system commands or higher-level functions like opening a browser to a specified URL. It's designed so the different VM technologies have a similiar interface, and you can ignore the specifics of the VM tech. The majority of the functionality is implemented in the form of drivers and controllers. Drivers implement the underlying command for each vm software (such as start/stop/revert), and controllers implement the commands which apply to all vms (such as listing all running vms, or cloning a vm). 
-
-If you're interested in porting a vm software (see below), please take a look at the workstation_driver.rb and the workstation_controller.rb -- This is a simple driver / controller in the lab, and you can simply copy / modify this to implement a new driver & controller for the software. 
-
-SUPPORTED VM TECHNOLOGIES:
-==========================
-NOTE: The lab libraries have only been tested with linux as a host, porting to windows is not planned at this time.
-
-Implemented:
- - workstation (Tested against 7.x)
- - remote_workstation (Tested against 7.x)
- - virtualbox (Tested against 4.x)
- - remote_esx (VMware ESX Host Agent 4.1.0 build-348481)
-
-Partially Implemented:
- - amazon_ec2 (via fog gem)
- - dynagen
-
-Need Implementation:
- - qemu
- - qemudo
- - others?
-
-PLATFORM SUPPORT:
-=================
-You will need to have this code running on a linux box, Currently this has only been run / tested on Ubuntu 9.04 -> 10.04, though it should run on any linux with an ssh client and the dependencies below. Remote VM Hosts will need to be linux as well, though other platforms may work (untested). If you're interested in porting it to windows, please contact me (jcran). 
-
-Platform Dependencies:
-  - whatever vm software is necessary for the driver you're using (see SUPPORTED VM TECHNOLOGIES above)
-  - net/scp - the gem (net-scp). Required to copy files to/from the devices in the case that tools are not installed. Not necessary if tools are installed. 
-  - fog - require to use the amazon_ec2 driver
-
-STANDALONE API:
-===============
-BACKGROUND:
-
-The lab libraries add tons of useful functionality that isn't exposed through the lab plugin, such as the ability to run commands on hosts. This library can serve as an excellent base for more complex operations on a remote host as well. 
-
-USAGE:
-
-You must first create a yaml file which describes your vm. See data/lab/test_targets.yml for an example.  
-<pre>
-  require 'vm_controller'
-  vm_controller = ::Lab::Controllers::VmController.new(YAML.load_file(lab_def)) 
-  vm_controller['vm1'].start
-  vm_controller['vm1'].snapshot("clean") 
-  vm_controller['vm1'].run_command("rm /etc/resolv.conf")
-  vm_controller['vm1'].open_uri("http://autopwn:8080")
-  vm_controller['vm1'].revert("clean")
-  vm_controller['vm1'].revert("clean")
-</pre>
-METASPLOIT MSFCONSOLE LAB PLUGIN:
-=================================
-
-BACKGROUND:
-
-The lab plugin for msfconsole adds a number of commands which may be useful if you're interested in automating remote hosts with rc scripts, or if you need to control targets / support systems while utilizing the metasploit console. A potential use case is testing an IPS / IDS, and resetting the target after running each exploit. 
-
-USAGE:
-
-Here's some example usage for the lab plugin. 
-<pre>
-  msf> load lab                           // Loads the lab plugin
-  msf> lab_load <path_to_lab_file>        // Loads from a lab configuration file. See data/lab/test_targets.yml for an example
-  msf> lab_load_dir workstation /path/to/vmx/files  // Loads from a local directory.
-  msf> lab_load_running remote_esx root esx_server  // Loads all running vms. 
-  msf> lab_start vm1                      // Start a vm which was loaded above
-  msf> lab_snapshot vm1 snapshot_1        // Snapshot a vm as 'snapshot_1'
-  msf> lab_run_command ("rm -rf /")       // oops!
-  msf> lab_show                           // Show all vms that we're aware of
-  msf> lab_show_running                   // Show only running vms
-  msf> lab_start vm2                      // Start another vm
-  msf> lab_suspend  vm1                   // Suspend a vm
-  msf> lab_revert all snapshot_1          // Revert all vms back to 'snapshot_1'
-</pre>