metasploit-gs/lib/lab/vm_driver.rb

##
## $Id$
##

#
# 	!!WARNING!! - All drivers are expected to filter input before running
#	anything based on it. This is particularly important in the case
#	of the drivers which wrap a command line program to provide 
# 	functionality.  
#


module Lab
module Drivers
class VmDriver

	def register	# Must be implemented in a child *_driver class
		raise Exception, "Command not Implemented"
	end
		
	def unregister	# Must be implemented in a child *_driver class
		raise Exception, "Command Not Implemented"
	end

	def start	# Must be implemented in a child *_driver class
		raise Exception, "Command Not Implemented"
	end

	def stop	# Must be implemented in a child *_driver class
		raise Exception, "Command Not Implemented"
	end

	def suspend	# Must be implemented in a child *_driver class
		raise Exception, "Command Not Implemented"
	end

	def pause	# Must be implemented in a child *_driver class
		raise Exception, "Command Not Implemented"
	end

	def reset	# Must be implemented in a child *_driver class
		raise Exception, "Command Not Implemented"
	end

	def create_snapshot(snapshot)	# Must be implemented in a child *_driver class
		raise Exception, "Command Not Implemented"
	end

	def revert_snapshot(snapshot)	# Must be implemented in a child *_driver class
		raise Exception, "Command Not Implemented"
	end

	def delete_snapshot(snapshot)	# Must be implemented in a child *_driver class
		raise Exception, "Command Not Implemented"
	end

	def run_command(command)	# Must be implemented in a child *_driver class
		raise Exception, "Command Not Implemented"
	end
	
	def copy_from(from, to)	# Must be implemented in a child *_driver class
		raise Exception, "Command Not Implemented"
	end
	
	def copy_to(from, to)	# Must be implemented in a child *_driver class
		raise Exception, "Command Not Implemented"
	end

	def check_file_exists(file)	# Must be implemented in a child *_driver class
		raise Exception, "Command Not Implemented"
	end

	def create_directory(directory)	# Must be implemented in a child *_driver class
		raise Exception, "Command Not Implemented"
	end

	def cleanup	# Must be implemented in a child *_driver class
		raise Exception, "Command Not Implemented"
	end

private
	def scp_to(from,to)
		gem 'net-ssh'
		require 'net/ssh'
		
		gem 'net-scp'
		require 'net/scp'
		
		begin
			# upload a file to a remote server
			Net::SCP.start(@vmid, @vm_user, :password => @vm_pass) do |scp|
				scp.upload!(from,to)
			end	
		rescue Exception => e
			return false
		end
	end
	
	def scp_from(from,to)
		gem 'net-ssh'
		require 'net/ssh'
		
		gem 'net-scp'
		require 'net/scp'
		
		begin
			# download a file from a remote server
			Net::SCP.start(@vmid, @vm_user, :password => @vm_pass) do |scp|
				scp.download!(from,to)
			end	
		rescue Exception => e
			return false
		end

	end
	
	def ssh_exec(command)
		gem 'net-ssh'
		require 'net/ssh'

		begin		
			Net::SSH.start(@vmid, @vm_user, :password => @vm_pass) do |ssh|
				result = ssh.exec!(command)
			end
		rescue Exception => e
			return false
		end

	end

	def filter_input(string)
		return unless string
				
		if !(string =~ /^[\w\s\[\]\{\}\/\\\.\-\"\(\):]*$/)
			raise "WARNING! Invalid character in: #{string}"
		end

	string
	end
	
	def system_command(command)
		begin
			#puts "DEBUG: running command #{command}"
			system(command)
		rescue	Exception => e
			return false			
		end
	end
end

end
end