Class: Dorothy::Doro_VSM::ESX

Inherits:
Object
  • Object
show all
Defined in:
lib/dorothy2/VSM.rb

Overview

ESX vSphere5 interface

Instance Method Summary collapse

Constructor Details

#initialize(server, user, pass, vmname, guestuser, guestpass) ⇒ ESX

Creates a new instance for communicating with ESX through the vSpere5’s API



14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
 
# File 'lib/dorothy2/VSM.rb', line 14

def initialize(server,user,pass,vmname,guestuser,guestpass)

  begin
    vim = RbVmomi::VIM.connect(:host => server , :user => user, :password=> pass, :insecure => true)
  rescue Timeout::Error
    raise "Fail to connect to the ESXi server #{server} - TimeOut (Are you sure that is the right address?)"
  end

  @server = server
  dc = vim.serviceInstance.find_datacenter
  @vm = dc.find_vm(vmname)

  raise "Virtual Machine #{vmname} not present within ESX!!" if @vm.nil?

  om = vim.serviceContent.guestOperationsManager
  am = om.authManager
  @pm = om.processManager
  @fm = om.fileManager

  #AUTHENTICATION
  guestauth = {:interactiveSession => false, :username => guestuser, :password => guestpass}
  r = 0
  begin
    @auth=RbVmomi::VIM::NamePasswordAuthentication(guestauth)
    abort if am.ValidateCredentialsInGuest(:vm => @vm, :auth => @auth) != nil
  rescue RbVmomi::Fault => e
    if e.inspect =~ /InvalidPowerState/
      if r <= 5
        r = r+1
        LOGGER.debug "VSM", "VM busy (maybe still revertig, retrying.."
        sleep 2
        retry
      end
      LOGGER.error "VSM", "Error, can't connect to VM #{@vm[:name]}"
      LOGGER.debug "VSM", e
      raise "VSM Error"
    end
  end

end

Instance Method Details

#check_internet ⇒ Object 



89
90
91
 
# File 'lib/dorothy2/VSM.rb', line 89

def check_internet
  exec_file_raw("windows\\system32\\ping.exe", "-n 1 www.google.com")  #make www.google.com customizable, move to doroconf
end

#copy_file(filename, file) ⇒ Object 



59
60
61
62
63
64
65
66
67
68
69
70
71
72
 
# File 'lib/dorothy2/VSM.rb', line 59

def copy_file(filename,file)
  filepath = "C:\\#{filename}" #put md5 hash

  begin
    url = @fm.InitiateFileTransferToGuest(:vm => @vm, :auth=> @auth, :guestFilePath=> filepath, :fileSize => file.size, :fileAttributes => '', :overwrite => true).sub('*:443', @server)

    RestClient.put(url, file)

  rescue RbVmomi::Fault
    LOGGER.error "VSM", "Fail to copy the file #{file} to #{@vm}: #{$!}"
    abort
  end

end

#exec_file(filename, program) ⇒ Object 



74
75
76
77
78
79
80
 
# File 'lib/dorothy2/VSM.rb', line 74

def exec_file(filename, program)
  program["prog_args"].nil? ? args = "" : args = program["prog_args"]
  args << " #{filename}"
  cmd = { :programPath => program["prog_path"], :arguments => args }
  pid = @pm.StartProgramInGuest(:vm => @vm , :auth => @auth, :spec => cmd )
  pid.to_i
end

#exec_file_raw(filename, arguments = "") ⇒ Object 



82
83
84
85
86
87
 
# File 'lib/dorothy2/VSM.rb', line 82

def exec_file_raw(filename, arguments="")
  filepath = "C:\\#{filename}"
  cmd = { :programPath => filepath, :arguments => arguments }
  pid = @pm.StartProgramInGuest(:vm => @vm , :auth => @auth, :spec => cmd )
  pid.to_i
end

#get_files(path) ⇒ Object 



118
119
120
121
122
123
124
125
 
# File 'lib/dorothy2/VSM.rb', line 118

def get_files(path)
  fm_files = @fm.ListFilesInGuest(:vm => @vm, :auth=> @auth, :filePath=> path).files
  @files = Hash.new
  fm_files.each {|file|
    @files.merge!(Hash[file.path, Hash[:size, file.size, :type, file.type, :attrs, file.attributes]])
  }
  @files
end

#get_new_procs(current_procs, original_procs = BASELINE_PROCS) ⇒ Object 



110
111
112
113
114
115
116
 
# File 'lib/dorothy2/VSM.rb', line 110

def get_new_procs(current_procs, original_procs=BASELINE_PROCS)
  @new_procs = Hash.new
  current_procs.each_key {|pid|
    @new_procs.merge!(Hash[pid, current_procs[pid]]) unless original_procs.has_key?(pid)
  }
  @new_procs
end

#get_running_procs(pid = nil, save_tofile = false, filename = "#{DoroSettings.env[:home]}/etc/baseline_processes.yml") ⇒ Object 



98
99
100
101
102
103
104
105
106
107
108
 
# File 'lib/dorothy2/VSM.rb', line 98

def get_running_procs(pid=nil, save_tofile=false, filename="#{DoroSettings.env[:home]}/etc/baseline_processes.yml")
  pid = Array(pid) unless pid.nil?
  @pp2 = Hash.new
  procs = @pm.ListProcessesInGuest(:vm => @vm , :auth => @auth, :pids => pid )
  procs.each {|pp2| @pp2.merge! Hash[pp2.pid, Hash["pname", pp2.name, "owner", pp2.owner, "cmdLine", pp2.cmdLine, "startTime", pp2.startTime, "endTime", pp2.endTime, "exitCode", pp2.exitCode]]}
    if save_tofile
    Util.write(filename, @pp2.to_yaml)
    LOGGER.info "VSM", "Current running processes saved to #{filename}"
  end
  @pp2
end

#get_status(pid) ⇒ Object 



93
94
95
96
 
# File 'lib/dorothy2/VSM.rb', line 93

def get_status(pid)
  p = get_running_procs(pid)
  p["exitCode"]
end

#revert_vm ⇒ Object 



55
56
57
 
# File 'lib/dorothy2/VSM.rb', line 55

def revert_vm
  @vm.RevertToCurrentSnapshot_Task
end

#screenshot ⇒ Object 



127
128
129
130
131
 
# File 'lib/dorothy2/VSM.rb', line 127

def screenshot
  a = @vm.CreateScreenshot_Task.wait_for_completion.split(" ")
  screenpath = "/vmfs/volumes/" + a[0].delete("[]") + "/" + a[1]
  return screenpath
end