Class: VagrantPlugins::ProviderLibvirt::Driver

Inherits:
Object
  • Object
show all
Defined in:
lib/vagrant-libvirt/driver.rb

Constant Summary collapse

@@connection =

store the connection at the process level

possibly this should be a connection pool using the connection settings as a key to allow per machine connection attributes to be used.

nil
@@system_connection = 
nil

Instance Method Summary collapse

Constructor Details

#initialize(machine) ⇒ Driver

Returns a new instance of Driver.



16
17
18
19
 
# File 'lib/vagrant-libvirt/driver.rb', line 16

def initialize(machine)
  @logger = Log4r::Logger.new('vagrant_libvirt::driver')
  @machine = machine
end

Instance Method Details

#connection ⇒ Object 



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
 
# File 'lib/vagrant-libvirt/driver.rb', line 21

def connection
  # If already connected to Libvirt, just use it and don't connect
  # again.
  return @@connection if @@connection

  # Get config options for Libvirt provider.
  config = @machine.provider_config
  uri = config.uri

  conn_attr = {}
  conn_attr[:provider] = 'libvirt'
  conn_attr[:libvirt_uri] = uri
  conn_attr[:libvirt_username] = config.username if config.username
  conn_attr[:libvirt_password] = config.password if config.password

  # Setup command for retrieving IP address for newly created machine
  # with some MAC address. Get it from dnsmasq leases table
  ip_command = %q( awk "/$mac/ {print \$1}" /proc/net/arp )
  conn_attr[:libvirt_ip_command] = ip_command

  @logger.info("Connecting to Libvirt (#{uri}) ...")
  begin
    @@connection = Fog::Compute.new(conn_attr)
  rescue Fog::Errors::Error => e
    raise Errors::FogLibvirtConnectionError,
          error_message: e.message
  end

  @@connection
end

#created?(mid) ⇒ Boolean

Returns:

  • (Boolean) 


78
79
80
81
 
# File 'lib/vagrant-libvirt/driver.rb', line 78

def created?(mid)
  domain = get_domain(mid)
  !domain.nil?
end

#get_domain(mid) ⇒ Object 



63
64
65
66
67
68
69
70
71
72
73
74
75
76
 
# File 'lib/vagrant-libvirt/driver.rb', line 63

def get_domain(mid)
  begin
    domain = connection.servers.get(mid)
  rescue Libvirt::RetrieveError => e
    if e.libvirt_code == ProviderLibvirt::Util::ErrorCodes::VIR_ERR_NO_DOMAIN
      @logger.debug("machine #{mid} not found #{e}.")
      return nil
    else
      raise e
    end
  end

  domain
end

#get_ipaddress(machine) ⇒ Object 



83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
 
# File 'lib/vagrant-libvirt/driver.rb', line 83

def get_ipaddress(machine)
  # Find the machine
  domain = get_domain(machine.id)
  if @machine.provider_config.qemu_use_session
    return get_ipaddress_system domain.mac
  end

  if domain.nil?
    # The machine can't be found
    return nil
  end

  # Get IP address from arp table
  ip_address = nil
  begin
    domain.wait_for(2) do
      addresses.each_pair do |_type, ip|
        # Multiple leases are separated with a newline, return only
        # the most recent address
        ip_address = ip[0].split("\n").first unless ip[0].nil?
      end
      !ip_address.nil?
    end
  rescue Fog::Errors::TimeoutError
    @logger.info('Timeout at waiting for an ip address for machine %s' % machine.name)
  end

  unless ip_address
    @logger.info('No arp table entry found for machine %s' % machine.name)
    return nil
  end

  ip_address
end

#get_ipaddress_system(mac) ⇒ Object 



118
119
120
121
122
123
124
125
126
127
128
129
 
# File 'lib/vagrant-libvirt/driver.rb', line 118

def get_ipaddress_system(mac)
  ip_address = nil

  system_connection.list_all_networks.each do |net|
    leases = net.dhcp_leases(mac, 0)
    # Assume the lease expiring last is the current IP address
    ip_address = leases.sort_by { |lse| lse["expirytime"] }.last["ipaddr"] if !leases.empty?
    break if ip_address
  end

  return ip_address
end

#state(machine) ⇒ Object 



131
132
133
134
135
136
137
138
139
140
141
142
143
144
 
# File 'lib/vagrant-libvirt/driver.rb', line 131

def state(machine)
  # may be other error states with initial retreival we can't handle
  begin
    domain = get_domain(machine.id)
  rescue Libvirt::RetrieveError => e
    @logger.debug("Machine #{machine.id} not found #{e}.")
    return :not_created
  end

  # TODO: terminated no longer appears to be a valid fog state, remove?
  return :not_created if domain.nil? || domain.state.to_sym == :terminated

  domain.state.tr('-', '_').to_sym
end

#system_connection ⇒ Object 



52
53
54
55
56
57
58
59
60
61
 
# File 'lib/vagrant-libvirt/driver.rb', line 52

def system_connection
  # If already connected to Libvirt, just use it and don't connect
  # again.
  return @@system_connection if @@system_connection

  config = @machine.provider_config

  @@system_connection = Libvirt::open_read_only(config.system_uri)
  @@system_connection
end