vagrant-rbvmomi 1.8.1

data/lib/rbvmomi/fault.rb

@@ -0,0 +1,17 @@
+# Copyright (c) 2010 VMware, Inc.  All Rights Reserved.
+module RbVmomi
+
+class Fault < StandardError
+  attr_reader :fault
+
+  def initialize msg, fault
+    super "#{fault.class.wsdl_name}: #{msg}"
+    @fault = fault
+  end
+
+  def method_missing *a
+    @fault.send(*a)
+  end
+end
+
+end

data/lib/rbvmomi/pbm.rb

@@ -0,0 +1,66 @@
+# Copyright (c) 2012 VMware, Inc.  All Rights Reserved.
+require 'rbvmomi'
+
+module RbVmomi
+
+# A connection to one vSphere ProfileBasedManagement endpoint.
+# @see #serviceInstance
+class PBM < Connection
+  # Connect to a vSphere ProfileBasedManagement endpoint
+  #
+  # @param [VIM] Connection to main vSphere API endpoint
+  # @param [Hash] opts The options hash.
+  # @option opts [String]  :host Host to connect to.
+  # @option opts [Numeric] :port (443) Port to connect to.
+  # @option opts [Boolean] :ssl (true) Whether to use SSL.
+  # @option opts [Boolean] :insecure (false) If true, ignore SSL certificate errors.
+  # @option opts [String]  :path (/pbm/sdk) SDK endpoint path.
+  # @option opts [Boolean] :debug (false) If true, print SOAP traffic to stderr.
+  def self.connect vim, opts = {}
+    fail unless opts.is_a? Hash
+    opts[:host] = vim.host
+    opts[:ssl] = true unless opts.member? :ssl or opts[:"no-ssl"]
+    opts[:insecure] ||= false
+    opts[:port] ||= (opts[:ssl] ? 443 : 80)
+    opts[:path] ||= '/pbm/sdk'
+    opts[:ns] ||= 'urn:pbm'
+    rev_given = opts[:rev] != nil
+    opts[:rev] = '1.0' unless rev_given
+    opts[:debug] = (!ENV['RBVMOMI_DEBUG'].empty? rescue false) unless opts.member? :debug
+
+    new(opts).tap do |pbm|
+      pbm.vcSessionCookie = vim.cookie.split('"')[1]
+    end
+  end
+  
+  def vcSessionCookie= cookie
+    @vcSessionCookie = cookie
+  end
+
+  def rev= x
+    super
+    @serviceContent = nil
+  end
+
+  # Return the ServiceInstance
+  #
+  # The ServiceInstance is the root of the vSphere inventory.
+  def serviceInstance
+    @serviceInstance ||= VIM::PbmServiceInstance self, 'ServiceInstance'
+  end
+
+  # Alias to serviceInstance.PbmRetrieveServiceContent
+  def serviceContent
+    @serviceContent ||= serviceInstance.PbmRetrieveServiceContent
+  end
+
+  # @private
+  def pretty_print pp
+    pp.text "PBM(#{@opts[:host]})"
+  end
+
+  add_extension_dir File.join(File.dirname(__FILE__), "pbm")
+  load_vmodl(ENV['VMODL'] || File.join(File.dirname(__FILE__), "../../vmodl.db"))
+end
+
+end

data/lib/rbvmomi/sms.rb

@@ -0,0 +1,61 @@
+# Copyright (c) 2013 VMware, Inc.  All Rights Reserved.
+require 'rbvmomi'
+module RbVmomi
+
+# A connection to one vSphere SMS endpoint.
+# @see #serviceInstance
+class SMS < Connection
+  # Connect to a vSphere SMS endpoint
+  #
+  # @param [VIM] Connection to main vSphere API endpoint
+  # @param [Hash] opts The options hash.
+  # @option opts [String]  :host Host to connect to.
+  # @option opts [Numeric] :port (443) Port to connect to.
+  # @option opts [Boolean] :ssl (true) Whether to use SSL.
+  # @option opts [Boolean] :insecure (false) If true, ignore SSL certificate errors.
+  # @option opts [String]  :path (/sms/sdk) SDK endpoint path.
+  # @option opts [Boolean] :debug (false) If true, print SOAP traffic to stderr.
+  def self.connect vim, opts = {}
+    fail unless opts.is_a? Hash
+    opts[:host] = vim.host
+    opts[:ssl] = true unless opts.member? :ssl or opts[:"no-ssl"]
+    opts[:insecure] ||= true
+    opts[:port] ||= (opts[:ssl] ? 443 : 80)
+    opts[:path] ||= '/sms/sdk'
+    opts[:ns] ||= 'urn:sms'
+    rev_given = opts[:rev] != nil
+    opts[:rev] = '4.0' unless rev_given
+    opts[:debug] = (!ENV['RBVMOMI_DEBUG'].empty? rescue false) unless opts.member? :debug
+
+    new(opts).tap do |sms|
+      sms.vcSessionCookie = vim.cookie.split('"')[1]
+    end
+  end
+
+  def vcSessionCookie= cookie
+    @vcSessionCookie = cookie
+  end
+
+  def rev= x
+    super
+    @serviceContent = nil
+  end
+
+  # Return the ServiceInstance
+  #
+  # The ServiceInstance is the root of the vSphere inventory.
+  def serviceInstance
+    @serviceInstance ||= VIM::SmsServiceInstance self, 'ServiceInstance'
+  end
+
+  # @private
+  def pretty_print pp
+    pp.text "SMS(#{@opts[:host]})"
+  end
+
+  add_extension_dir File.join(File.dirname(__FILE__), "sms")
+  load_vmodl(ENV['VMODL'] || File.join(File.dirname(__FILE__), "../../vmodl.db"))
+end
+
+end
+

data/lib/rbvmomi/sms/SmsStorageManager.rb

@@ -0,0 +1,7 @@
+class RbVmomi::SMS::SmsStorageManager
+
+  def RegisterProvider_Task2 providerSpec
+     self.RegisterProvider_Task providerSpec
+  end
+
+end

data/lib/rbvmomi/trivial_soap.rb

@@ -0,0 +1,114 @@
+# Copyright (c) 2010 VMware, Inc.  All Rights Reserved.
+require 'rubygems'
+require 'builder'
+require 'nokogiri'
+require 'net/http'
+require 'pp'
+
+class RbVmomi::TrivialSoap
+  attr_accessor :debug, :cookie
+  attr_reader :http
+
+  def initialize opts
+    fail unless opts.is_a? Hash
+    @opts = opts
+    return unless @opts[:host] # for testcases
+    @debug = @opts[:debug]
+    @cookie = @opts[:cookie]
+    @lock = Mutex.new
+    @http = nil
+    restart_http
+  end
+
+  def host
+    @opts[:host]
+  end
+
+  def close
+    @http.finish rescue IOError
+  end
+
+  def restart_http
+    begin 
+      @http.finish if @http
+    rescue Exception => ex
+      puts "WARNING: Ignoring exception: #{ex.message}"
+      puts ex.backtrace.join("\n")
+    end
+    @http = Net::HTTP.new(@opts[:host], @opts[:port], @opts[:proxyHost], @opts[:proxyPort])
+    if @opts[:ssl]
+      require 'net/https'
+      @http.use_ssl = true
+      if @opts[:insecure]
+        @http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+      else
+        @http.verify_mode = OpenSSL::SSL::VERIFY_PEER
+      end
+      @http.cert = OpenSSL::X509::Certificate.new(@opts[:cert]) if @opts[:cert]
+      @http.key = OpenSSL::PKey::RSA.new(@opts[:key]) if @opts[:key]
+    end
+    @http.set_debug_output(STDERR) if $DEBUG
+    @http.read_timeout = 1000000
+    @http.open_timeout = 60
+    def @http.on_connect
+      @socket.io.setsockopt(Socket::IPPROTO_TCP, Socket::TCP_NODELAY, 1)
+    end
+    @http.start
+  end
+
+  def soap_envelope
+    xsd = 'http://www.w3.org/2001/XMLSchema'
+    env = 'http://schemas.xmlsoap.org/soap/envelope/'
+    xsi = 'http://www.w3.org/2001/XMLSchema-instance'
+    xml = Builder::XmlMarkup.new :indent => 0
+    xml.tag!('env:Envelope', 'xmlns:xsd' => xsd, 'xmlns:env' => env, 'xmlns:xsi' => xsi) do
+      if @vcSessionCookie
+        xml.tag!('env:Header') do
+          xml.tag!('vcSessionCookie', @vcSessionCookie)
+        end
+      end
+      xml.tag!('env:Body') do
+        yield xml if block_given?
+      end
+    end
+    xml
+  end
+
+  def request action, body
+    headers = { 'content-type' => 'text/xml; charset=utf-8', 'SOAPAction' => action }
+    headers['cookie'] = @cookie if @cookie
+
+    if @debug
+      $stderr.puts "Request:"
+      $stderr.puts body
+      $stderr.puts
+    end
+
+    start_time = Time.now
+    response = @lock.synchronize do
+      begin
+        @http.request_post(@opts[:path], body, headers)
+      rescue Exception
+        restart_http
+        raise
+      end
+    end
+    end_time = Time.now
+    
+    if response.is_a? Net::HTTPServiceUnavailable
+      raise "Got HTTP 503: Service unavailable"
+    end
+
+    self.cookie = response['set-cookie'] if response.key? 'set-cookie'
+
+    nk = Nokogiri(response.body)
+
+    if @debug
+      $stderr.puts "Response (in #{'%.3f' % (end_time - start_time)} s)"
+      $stderr.puts nk
+      $stderr.puts
+    end
+
+    [nk.xpath('//soapenv:Body/*').select(&:element?).first, response.body.size]
+  end
+end

data/lib/rbvmomi/trollop.rb

@@ -0,0 +1,70 @@
+# Copyright (c) 2011 VMware, Inc.  All Rights Reserved.
+require 'trollop'
+
+# Convenience methods for Trollop, Ruby's premier option parser.
+# @see http://trollop.rubyforge.org/
+# @see Trollop::Parser
+module Trollop
+
+# Convenience methods for Trollop, Ruby's premier option parser.
+#
+# See the examples directory for sample code.
+# Descriptions are of the form:
+#  <key>: <options> <environment variable> (<default>)
+# @see http://trollop.rubyforge.org/
+class Parser
+  # Options used by VIM.connect
+  #
+  #  !!!plain
+  #  host: -o --host RBVMOMI_HOST
+  #  port: --port RBVMOMI_PORT (443)
+  #  no-ssl: --no-ssl RBVMOMI_SSL (false)
+  #  insecure: -k --insecure RBVMOMI_INSECURE (false)
+  #  user: -u --user RBVMOMI_USER (root)
+  #  password: -p --password RBVMOMI_PASSWORD ()
+  #  path: --path RBVMOMI_PATH (/sdk)
+  #  debug: -d --debug RBVMOMI_DEBUG (false)
+  def rbvmomi_connection_opts
+    opt :host, "host", :type => :string, :short => 'o', :default => ENV['RBVMOMI_HOST']
+    opt :port, "port", :type => :int, :short => :none, :default => (ENV.member?('RBVMOMI_PORT') ? ENV['RBVMOMI_PORT'].to_i : 443)
+    opt :"no-ssl", "don't use ssl", :short => :none, :default => (ENV['RBVMOMI_SSL'] == '0')
+    opt :insecure, "don't verify ssl certificate", :short => 'k', :default => (ENV['RBVMOMI_INSECURE'] == '1')
+    opt :user, "username", :short => 'u', :default => (ENV['RBVMOMI_USER'] || 'root')
+    opt :password, "password", :short => 'p', :default => (ENV['RBVMOMI_PASSWORD'] || '')
+    opt :path, "SOAP endpoint path", :short => :none, :default => (ENV['RBVMOMI_PATH'] || '/sdk')
+    opt :debug, "Log SOAP messages", :short => 'd', :default => (ENV['RBVMOMI_DEBUG'] || false)
+  end
+
+  # Select a datacenter
+  #
+  #  !!!plain
+  #  datacenter: -D --datacenter RBVMOMI_DATACENTER (ha-datacenter)
+  def rbvmomi_datacenter_opt
+    opt :datacenter, "datacenter", :type => :string, :short => "D", :default => (ENV['RBVMOMI_DATACENTER'] || 'ha-datacenter')
+  end
+
+  # Select a folder
+  #
+  #  !!!plain
+  #  folder: -F --folder RBVMOMI_FOLDER ()
+  def rbvmomi_folder_opt
+    opt :folder, "VM folder", :type => :string, :short => "F", :default => (ENV['RBVMOMI_FOLDER'] || '')
+  end
+
+  # Select a compute resource
+  #
+  #  !!!plain
+  #  computer: -R --computer RBVMOMI_COMPUTER
+  def rbvmomi_computer_opt
+    opt :computer, "Compute resource", :type => :string, :short => "R", :default => (ENV['RBVMOMI_COMPUTER']||'ha-compute-res')
+  end
+
+  # Select a datastore
+  #
+  #  !!!plain
+  #  datastore: -s --datastore RBVMOMI_DATASTORE (datastore1)
+  def rbvmomi_datastore_opt
+    opt :datastore, "Datastore", :short => 's', :default => (ENV['RBVMOMI_DATASTORE'] || 'datastore1')
+  end
+end
+end

data/lib/rbvmomi/type_loader.rb

@@ -0,0 +1,136 @@
+# Copyright (c) 2010 VMware, Inc.  All Rights Reserved.
+require 'set'
+require 'monitor'
+
+module RbVmomi
+
+class TypeLoader
+  def initialize fn, extension_dirs, namespace
+    @extension_dirs = extension_dirs
+    @namespace = namespace
+    @lock = Monitor.new
+    @db = {}
+    @id2wsdl = {}
+    @loaded = {}
+    add_types Hash[BasicTypes::BUILTIN.map { |k| [k,nil] }]
+    vmodl_database = File.open(fn, 'r') { |io| Marshal.load io }
+    vmodl_database.reject! { |k,v| k =~ /^_/ }
+    add_types vmodl_database
+    preload
+  end
+
+  def preload
+    names = (@namespace.constants + Object.constants).map(&:to_s).uniq.
+                                                      select { |x| has? x }
+    names.each { |x| get(x) }
+  end
+
+  # Reload all extensions for loaded VMODL types
+  def reload_extensions
+    @extension_dirs.each do |path|
+      reload_extensions_dir path
+    end
+  end
+
+  # Reload all extensions for loaded VMODL types from the given directory
+  def reload_extensions_dir path
+    loaded = Set.new(typenames.select { |x| @namespace.const_defined? x })
+    Dir.open(path) do |dir|
+      dir.each do |file|
+        next unless file =~ /\.rb$/
+        next unless loaded.member? $`
+        file_path = File.join(dir, file)
+        load file_path
+      end
+    end
+  end
+
+  def has? name
+    fail unless name.is_a? String
+
+    @db.member?(name) or BasicTypes::BUILTIN.member?(name)
+  end
+
+  def get name
+    fail "name '#{name}' is #{name.class} expecting String" unless name.is_a? String
+
+    first_char = name[0].chr
+    if first_char.downcase == first_char
+      name = "%s%s" % [first_char.upcase, name[1..-1]]
+    end
+
+    return @loaded[name] if @loaded.member? name
+    @lock.synchronize do
+      return @loaded[name] if @loaded.member? name
+      klass = make_type(name)
+      @namespace.const_set name, klass
+      load_extension name
+      @loaded[name] = klass
+    end
+  end
+
+  def add_types types
+    @lock.synchronize do
+      @db.merge! types
+      @db = Hash[@db.map do |name, value|
+        if value
+          value['wsdl_name'] ||= name
+        end
+        first_char = name[0].chr
+        if first_char.downcase == first_char
+          name = "%s%s" % [first_char.upcase, name[1..-1]]
+        end
+        [name, value]
+      end]
+    end
+  end
+
+  def typenames
+    @db.keys
+  end
+
+  private
+
+  def load_extension name
+    @extension_dirs.map { |x| File.join(x, "#{name}.rb") }.
+                    select { |x| File.exists? x }.
+                    each { |x| load x }
+  end
+
+  def make_type name
+    name = name.to_s
+    return BasicTypes.const_get(name) if BasicTypes::BUILTIN.member? name
+    desc = @db[name] or fail "unknown VMODL type #{name}"
+    case desc['kind']
+    when 'data' then make_data_type name, desc
+    when 'managed' then make_managed_type name, desc
+    when 'enum' then make_enum_type name, desc
+    else fail desc.inspect
+    end
+  end
+
+  def make_data_type name, desc
+    superclass = get desc['wsdl_base']
+    Class.new(superclass).tap do |klass|
+      klass.init name, desc['props']
+      klass.wsdl_name = desc['wsdl_name']
+    end
+  end
+
+  def make_managed_type name, desc
+    superclass = get desc['wsdl_base']
+    Class.new(superclass).tap do |klass|
+      klass.init name, desc['props'], desc['methods']
+      klass.wsdl_name = desc['wsdl_name']
+    end
+  end
+
+  def make_enum_type name, desc
+    Class.new(BasicTypes::Enum).tap do |klass|
+      klass.init name, desc['values']
+      klass.wsdl_name = desc['wsdl_name']
+    end
+  end
+end
+
+end