infosecn1nja
/
metasploit-framework
mirror of https://github.com/infosecn1nja/metasploit-framework.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

adding initial version of the lab controller - note that this will be updated to use vixr soon 

git-svn-id: file:///home/svn/framework3/trunk@9687 4d416f70-5f16-0410-b530-b9f4589650da
unstable
Jonathan Cran 2010-07-06 18:18:06 +00:00
parent 8f967af2f3
commit 2786427113
1 changed files with 401 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

401
lib/lab/vmware_lab_controller.rb Normal file
View File

@ -0,0 +1,401 @@
## Crap class which wraps vmrun
#
# Lower level methods which are ~generic to the vm software
#
class VmwareController
def start(vmx)
if File.exist?(vmx) then
system_command("vmrun -T ws start " + "\"" + vmx + "\"")
else
raise ArgumentError, "Couldn't find: " + vmx, caller
end
end
def reset(vmx)
if File.exist?(vmx) then
system_command("vmrun -T ws reset " + "\"" + vmx + "\"")
else
raise ArgumentError, "Couldn't find: " + vmx, caller
end
end
def get_running
output = `vmrun list`
end
def running?(vmx)
output = get_running
output.each_line do |line|
if line.to_s.chomp.eql? vmx.to_s then return true end
end
return false
end
def run_command(vmx, command, user, pass, displayParameter=false)
if File.exist?(vmx) then
vmrunstr = "vmrun -T ws -gu " + user + " -gp " + pass + " runProgramInGuest \"" + vmx + "\" " + "\"" + command + "\" -interactive -noWait -activeWindow"
if displayParameter then
vmrunstr = vmrunstr + " -display :0"
end
system_command(vmrunstr)
else
raise ArgumentError,"Couldn't find: " + vmx, caller
end
end
def copy_file_from(vmx, user, pass, guestpath, hostpath)
vmrunstr = "vmrun -T ws -gu " + user + " -gp " + pass + " copyFileFromGuestToHost \"" + vmx + "\" \"" + guestpath + "\" \"" + hostpath + "\""
system_command(vmrunstr)
end
def copy_file_to(vmx, user, pass, hostpath, guestpath)
#puts "Copying " + hostpath + " to " + guestpath + " on " + vmx + "\n"
vmrunstr = "vmrun -T ws -gu " + user + " -gp " + pass + " copyFileFromHostToGuest \"" + vmx + "\" \"" + hostpath + "\" \"" + guestpath + "\""
system_command(vmrunstr)
end
def scp_copy_file_to(ip, user, pass, hostpath, guestpath)
#puts "Copying " + hostpath + " to " + guestpath + " on " + vmx + "\n"
vmrunstr = "scp -r \"" + hostpath + "\" \"" + user + "@" + ip + ":" + guestpath + "\""
system_command(vmrunstr)
end
def check_file_exists(vmx, user, pass, file)
vmrunstr = "vmrun -T ws -gu " + user + " -gp " + pass + " fileExistsInGuest \"" + vmx + "\" \"" + file + "\" "
system_command(vmrunstr)
end
def create_directory_in_guest(vmx, user, pass, directory)
vmrunstr = "vmrun -T ws -gu " + user + " -gp " + pass + " createDirectoryInGuest \"" + vmx + "\" \"" + directory + "\" "
system_command(vmrunstr)
end
def create_snapshot(vmx, snapshot)
if File.exist?(vmx) then
system_command("vmrun -T ws snapshot " + "\"" + vmx + "\" " + snapshot)
else
raise ArgumentError,"Couldn't find: " + vmx, caller
end
end
def revert_snapshot(vmx, snapshot)
if File.exist?(vmx) then
system_command("vmrun -T ws revertToSnapshot " + "\"" + vmx + "\" " + snapshot)
else
raise "Couldn't find: " + vmx, caller
end
end
def delete_snapshot(vmx, snapshot)
if File.exist?(vmx) then
system_command("vmrun -T ws deleteSnapshot " + "\"" + vmx + "\" " + snapshot )
else
raise ArgumentError,"Couldn't find: " + vmx, caller
end
end
def stop(vmx)
if File.exist?(vmx) then
system_command("vmrun -T ws stop " + "\"" + vmx + "\"")
else
raise ArgumentError,"Couldn't find: " + vmx, caller
end
end
def suspend(vmx)
if File.exist?(vmx) then
system_command("vmrun -T ws suspend " + "\"" + vmx + "\"")
else
raise ArgumentError,"Couldn't find: " + vmx, caller
end
end
def pause(vmx)
if File.exist?(vmx) then
system_command("vmrun -T ws pause " + "\"" + vmx + "\"")
else
raise ArgumentError, "Couldn't find: " + vmx, caller
end
end
private
def system_command(command)
puts "Running System Command: " + command
system(command)
end
end
#
# Higher level methods which are specifc to the types of things we want to do
#
class VmwareLabController < VmwareController
def initialize (basepath = "." )
@vmbase = basepath
@lab = Hash.new()
end
def run_command_on_lab_vm(vmid,command)
begin
## handle linux
display = false
if (@lab[vmid]["os"] == "linux") then
display=true
end
run_command(get_vmx(vmid), command , @lab[vmid]["user"],@lab[vmid]["pass"], display)
rescue Exception => e
puts "error! " + e.to_s
end
end
def start_lab_vm(vmid)
begin
start(get_vmx(vmid))
rescue Exception => e
puts "error! " + e.to_s
end
end
def reset_lab_vm(vmid)
begin
reset(get_vmx(vmid))
rescue Exception => e
puts "error! " + e.to_s
end
end
def pause_lab_vm(vmid)
begin
pause(get_vmx(vmid))
rescue Exception => e
puts "error! " + e.to_s
end
end
def suspend_lab_vm(vmid)
begin
suspend(get_vmx(vmid))
rescue Exception => e
puts "error! " + e.to_s
end
end
def snapshot_lab_vm(vmid, snapshot)
begin
create_snapshot(get_vmx(vmid),snapshot)
rescue Exception => e
puts "error! " + e.to_s
end
end
def revert_lab_vm(vmid, snapshot)
begin
revert_snapshot(get_vmx(vmid),snapshot)
rescue Exception => e
puts "error! " + e.to_s
end
end
def stop_lab_vm(vmid)
begin
stop(get_vmx(vmid))
rescue Exception => e
puts "error! " + e.to_s
end
end
def start_lab
@lab.each { | key, value |
if value != nil then
start_lab_vm(key)
end
}
end
def reset_lab
@lab.each { | key, value |
if value != nil then
reset_lab_vm(key)
end
}
end
def suspend_lab
@lab.each { | key, value |
if value != nil then
suspend_lab_vm(key)
end
}
end
def snapshot_lab(snapshot)
@lab.each { | key, value |
if value != nil then
snapshot_lab_vm(key,snapshot)
end
}
end
def revert_lab(snapshot)
@lab.each { | key, value |
if value != nil then
revert_lab_vm(key,snapshot)
end
}
end
def run_command_on_lab(command)
@lab.each { | key, value |
if value != nil then
run_command_on_lab_vm(key, command)
end
}
end
def pause_lab
@lab.each { | key, value |
if value != nil then
pause_lab_vm(key)
end
}
end
def stop_lab
@lab.each { | key, value |
if value != nil then
stop_lab_vm(key)
end
}
end
def revert_lab(snapshot)
@lab.each { | key, value |
if value != nil then
revert_lab_vm(key, snapshot)
end
}
end
def copy_to_lab(file)
@lab.each { | key, value |
if value != nil then
copy_to_lab_vm(key,file)
end
}
end
def copy_from_lab(file)
@lab.each { | key, value |
if value != nil then
copy_from_lab_vm(key,file)
end
}
end
def copy_to_lab_vm(vmid,file)
## handle linux
guestpath = "C:\\temp2\\"
if (@lab[vmid]["os"] == "linux") then
guestpath = "/tmp/"
end
name = File.basename(file)
if (@lab[vmid]["tools"] == "true") then
create_directory_in_guest(get_vmx(vmid),@lab[vmid]["user"],@lab[vmid]["pass"], guestpath)
begin
copy_file_to(get_vmx(vmid),@lab[vmid]["user"],@lab[vmid]["pass"],file, guestpath + name)
rescue Exception => e
puts "error! " + e.to_s
end
else
scp_copy_file_to(@lab[vmid]["ip"], @lab[vmid]["user"],@lab[vmid]["pass"], file, guestpath + name)
end
end
def copy_from_lab_vm(vmid,file)
hostpath = "/tmp/"
name = File.basename(file.gsub("\\","/"))
#puts "filenaem: " + name + "\n"
begin
copy_file_from(get_vmx(vmid),@lab[vmid]["user"],@lab[vmid]["pass"],file,hostpath + name)
rescue Exception => e
puts "error! " + e.to_s
end
end
def list_lab
str = ""
@lab.map { |key,val|
if val != nil then
str = str + key.to_s + ": " + val["vmx"].to_s + "\n"
end
}
return str
end
def find_lab_vm(search)
str = ""
@lab.map { |key,val|
if val != nil then
if (val["vmx"].to_s.downcase.index(search.downcase) != nil) then
str = str + key.to_s + ": " + val["vmx"].to_s + "\n"
end
end
}
return str
end
def running?(vmid)
if super(get_vmx(vmid)) then
return true
end
return false
end
private
def get_vmx(vmid)
if @lab[vmid] != nil then
@vmbase.to_s + @lab[vmid]["vmx"].to_s ## handle linux
else
raise "VM " + vmid + " does not exist!"
end
end
end