rbvmomi 1.0.1

data/lib/rbvmomi/profile.rb

@@ -0,0 +1,22 @@
+# Copyright (c) 2010 VMware, Inc.  All Rights Reserved.
+
+module Kernel
+  PROFILE_TIMES = Hash.new 0.0
+
+  def profile sym, &b
+    start_time = Time.now
+    ret = b.call
+    elapsed = Time.now - start_time
+    PROFILE_TIMES[sym] += elapsed
+    puts "#{sym} #{elapsed}s" if $profile
+    ret
+  end
+
+  def dump_profile
+    PROFILE_TIMES.sort_by { |k,v| -v }.each do |sym,total|
+      puts "#{sym} #{total}"
+    end
+  end
+end
+
+at_exit { dump_profile if $profile }

data/lib/rbvmomi/trollop.rb

@@ -0,0 +1,29 @@
+require 'trollop'
+
+class Trollop::Parser
+  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 :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'
+  end
+
+  def rbvmomi_datacenter_opt
+    opt :datacenter, "datacenter", type: :string, short: "D", default: (ENV['RBVMOMI_DATACENTER'])
+  end
+
+  def rbvmomi_folder_opt
+    opt :folder, "VM folder", type: :string, short: "F", default: (ENV['RBVMOMI_FOLDER'] || '')
+  end
+
+  def rbvmomi_computer_opt
+    opt :computer, "Compute resource", type: :string, short: "R", default: ENV['RBVMOMI_COMPUTER']
+  end
+
+  def rbvmomi_datastore_opt
+    opt :datastore, "Datastore", short: 's', default: (ENV['RBVMOMI_DATASTORE'] || 'datstore1')
+  end
+end

data/lib/rbvmomi/types.rb

@@ -0,0 +1,373 @@
+# Copyright (c) 2010 VMware, Inc.  All Rights Reserved.
+require 'tokyocabinet'
+require 'pp'
+require 'set'
+
+include TokyoCabinet
+
+class Class
+  def wsdl_name
+    self.class.name
+  end
+end
+
+module RbVmomi
+
+module VIM
+
+BUILTIN_TYPES = %w(ManagedObject TypeName PropertyPath ManagedObjectReference MethodName MethodFault LocalizedMethodFault)
+
+def self.load fn
+  @db = HDB.new
+  if !@db.open(fn, HDB::OREADER | HDB::ONOLCK)
+    raise "VMODL db open error: #{@db.errmsg(@db.ecode)}"
+  end
+  @vmodl = Hash.new { |h,k| if e = @db[k] then h[k] = Marshal.load(e) end }
+  @typenames = Marshal.load(@db['_typenames']) + BUILTIN_TYPES
+  Object.constants.select { |x| @typenames.member? x.to_s }.each { |x| load_type x }
+end
+
+def self.has_type? name
+  @typenames.member? name.to_s
+end
+
+def self.type name
+  if has_type? name
+    const_get(name.to_sym)
+  else
+    fail "no such type #{name.inspect}"
+  end
+end
+
+def self.const_missing sym
+  if @typenames.member? sym.to_s
+    load_type sym
+  else
+    super
+  end
+end
+
+def self.method_missing sym, *a
+  if @typenames.member? sym.to_s
+    const_get(sym).new *a
+  else
+    super
+  end
+end
+
+def self.load_type sym
+  const_set sym, make_type(sym)
+end
+
+def self.make_type name
+  name = name.to_s
+  desc = @vmodl[name] or fail "unknown VIM 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 self.make_data_type name, desc
+  superclass = const_get(desc['wsdl_base'].to_sym)
+  Class.new(superclass).tap do |klass|
+    klass.initialize name, desc['props']
+  end
+end
+
+def self.make_managed_type name, desc
+  superclass = const_get(desc['wsdl_base'].to_sym)
+  Class.new(superclass).tap do |klass|
+    klass.initialize name, desc['props'], desc['methods']
+  end
+end
+
+def self.make_enum_type name, desc
+  Class.new(Enum).tap do |klass|
+    klass.initialize name, desc['values']
+  end
+end
+
+class Base
+  class << self
+    attr_reader :wsdl_name
+
+    def initialize wsdl_name=self.name
+      @wsdl_name = wsdl_name
+    end
+
+    def to_s
+      @wsdl_name
+    end
+  end
+
+  initialize
+end
+
+class ObjectWithProperties < Base
+  class << self
+    attr_accessor :props_desc
+
+    def initialize name=self.name, props=[]
+      super name
+      @props_desc = props
+      @props_desc.each do |d|
+        sym = d['name'].to_sym
+        define_method(sym) { _get_property sym }
+        define_method(:"#{sym}=") { |x| _set_property sym, x }
+      end
+    end
+
+    # XXX cache
+    def full_props_desc
+      (self == ObjectWithProperties ? [] : superclass.full_props_desc) + props_desc
+    end
+
+    def find_prop_desc name
+      full_props_desc.find { |x| x['name'] == name.to_s }
+    end
+  end
+
+  def _get_property sym
+    fail 'unimplemented'
+  end
+
+  def _set_property sym, val
+    fail 'unimplemented'
+  end
+
+  initialize
+end
+
+class ObjectWithMethods < ObjectWithProperties
+  class << self
+    attr_accessor :methods_desc
+
+    def initialize name=self.name, props=[], methods={}
+      super name, props
+      @methods_desc = methods
+
+      @methods_desc.each do |k,d|
+        sym = k.to_sym
+        define_method(sym) { |*args| _call sym, *args }
+        define_method(:"#{sym}!") { |*args| _call sym, *args }
+      end
+    end
+
+    # XXX cache
+    def full_methods_desc
+      (self == ObjectWithMethods ? {} : superclass.full_methods_desc).merge methods_desc
+    end
+  end
+
+  initialize
+end
+
+class DataObject < ObjectWithProperties
+  attr_reader :props
+
+  def initialize props={}
+    @props = Hash[props.map { |k,v| [k.to_sym, v] }]
+    self.class.full_props_desc.each do |desc|
+      #fail "missing required property #{desc['name'].inspect} of #{self.class.wsdl_name}" if @props[desc['name'].to_sym].nil? and not desc['is-optional']
+    end
+    @props.each do |k,v|
+      fail "unexpected property name #{k}" unless self.class.find_prop_desc(k)
+    end
+  end
+
+  def _get_property sym
+    @props[sym]
+  end
+
+  def [] sym
+    _get_property sym
+  end
+
+  def _set_property sym, val
+    @props[sym] = val
+  end
+
+  def == o
+    return false unless o.class == self.class
+    keys = (props.keys + o.props.keys).uniq
+    keys.all? { |k| props[k] == o.props[k] }
+  end
+
+  def hash
+    props.hash
+  end
+
+  def pretty_print q
+    q.text self.class.name
+    q.group 2 do
+      q.text '('
+      q.breakable
+      props = @props.sort_by { |k,v| k.to_s }
+      q.seplist props, nil, :each do |k, v|
+        q.group do
+          q.text k.to_s
+          q.text ': '
+          q.pp v
+        end
+      end
+    end
+    q.breakable
+    q.text ')'
+  end
+
+  initialize
+end
+
+class ManagedObject < ObjectWithMethods
+  def initialize soap, ref
+    super()
+    @soap = soap
+    @ref = ref
+  end
+
+  def _ref
+    @ref
+  end
+
+  def _get_property sym
+    ret = @soap.propertyCollector.RetrieveProperties(:specSet => [{
+      :propSet => [{ :type => self.class.wsdl_name, :pathSet => [sym.to_s] }],
+      :objectSet => [{ :obj => self }],
+    }])[0]
+
+    if ret.propSet.empty?
+      fail if ret.missingSet.empty?
+      raise ret.missingSet[0].fault
+    else
+      ret.propSet[0].val
+    end
+  end
+
+  def _set_property sym, val
+    fail 'unimplemented'
+  end
+
+  def _call method, o={}
+    fail "parameters must be passed as a hash" unless o.is_a? Hash
+    desc = self.class.full_methods_desc[method.to_s] or fail "unknown method"
+    @soap.call method, desc, self, o
+  end
+
+  def to_s
+    "#{self.class.wsdl_name}(#{@ref.inspect})"
+  end
+
+  def pretty_print pp
+    pp.text to_s
+  end
+
+  def [] k
+    _get_property k
+  end
+
+  def == x
+    x.class == self.class and x._ref == @ref
+  end
+
+  alias eql? ==
+
+  def hash
+    [self.class, @ref].hash
+  end
+
+  initialize 'ManagedObject'
+end
+
+class Enum < Base
+  class << self
+    attr_accessor :values
+
+    def initialize name=self.name, values=[]
+      super name
+      @values = values
+    end
+  end
+
+  attr_reader :value
+
+  def initialize value
+    @value = value
+  end
+
+  initialize
+end
+
+class MethodFault < DataObject
+  initialize 'MethodFault', [
+    {
+      'name' => 'faultCause',
+      'wsdl_type' => 'LocalizedMethodFault',
+      'is-array' => false,
+      'is-optional' => true,
+    }, {
+      'name' => 'faultMessage',
+      'wsdl_type' => 'LocalizableMessage',
+      'is-array' => true,
+      'is-optional' => true,
+    },
+  ]
+end
+
+class LocalizedMethodFault < DataObject
+  initialize 'LocalizedMethodFault', [
+    {
+      'name' => 'fault',
+      'wsdl_type' => 'MethodFault',
+      'is-array' => false,
+      'is-optional' => false,
+    }, {
+      'name' => 'localizedMessage',
+      'wsdl_type' => 'xsd:string',
+      'is-array' => false,
+      'is-optional' => true,
+    },
+  ]
+
+  def exception
+    RbVmomi.fault self.localizedMessage, self.fault
+  end
+end
+
+class RuntimeFault < MethodFault
+  initialize
+end
+
+class MethodName < String
+  def self.wsdl_name; 'MethodName' end
+end
+
+class PropertyPath < String
+  def self.wsdl_name; 'PropertyPath' end
+end
+
+class TypeName < String
+  def self.wsdl_name; 'TypeName' end
+end
+
+class ManagedObjectReference
+  def self.wsdl_name; 'ManagedObjectReference' end
+end
+
+class ::String
+  def self.wsdl_name; 'xsd:string' end
+end
+
+class ::Integer
+  def self.wsdl_name; 'xsd:int' end
+end
+
+class ::Float
+  def self.wsdl_name; 'xsd:float' end
+end
+
+end
+
+end

data/lib/trivial_soap.rb

@@ -0,0 +1,83 @@
+# Copyright (c) 2010 VMware, Inc.  All Rights Reserved.
+require 'rubygems'
+require 'builder'
+require 'nokogiri'
+require 'net/http'
+require 'pp'
+require 'rbvmomi/profile'
+
+class Net::HTTPGenericRequest
+  alias old_exec exec
+
+  def exec sock, ver, path
+    old_exec sock, ver, path
+    sock.io.flush
+  end
+end
+
+class 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
+    @http = Net::HTTP.new(@opts[:host], @opts[:port], @opts[:proxyHost], @opts[:proxyPort])
+    if @opts[:ssl]
+      require 'net/https'
+      @http.use_ssl = true
+      @http.verify_mode = OpenSSL::SSL::VERIFY_NONE # XXX
+      @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 = 60
+    @http.open_timeout = 5
+    @http.start
+    @debug = @opts[:debug]
+    @cookie = nil
+    @lock = Mutex.new
+  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
+      xml.tag!('env:Body') do
+        yield xml if block_given?
+      end
+    end
+    xml
+  end
+
+  def request action, &b
+    headers = { 'content-type' => 'text/xml; charset=utf-8', 'SOAPAction' => action }
+    headers['cookie'] = @cookie if @cookie
+    body = soap_envelope(&b).target!
+    
+    if @debug
+      $stderr.puts "Request:"
+      $stderr.puts body
+      $stderr.puts
+    end
+
+    start_time = Time.now
+    response = @lock.synchronize { profile(:post) { @http.request_post(@opts[:path], body, headers) } }
+    end_time = Time.now
+
+    @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
+  end
+end