omf_sfa 0.1.1

data/lib/omf-sfa/am/am-rpc/am_rpc_service.rb

@@ -0,0 +1,402 @@
+
+require 'nokogiri'
+require 'time'
+require 'zlib'
+require 'base64'
+require 'openssl'
+
+require 'omf-sfa/am/am-rpc/abstract_rpc_service'
+require 'omf-sfa/am/am-rpc/am_authorizer'
+
+require 'omf-sfa/am/am-rpc/am_rpc_api'
+#require 'omf-sfa/am/privilege_credential'
+#require 'omf-sfa/am/user_credential'
+
+module OMF::SFA::AM::RPC
+
+  class NotAuthorizedException < XMLRPC::FaultException; end
+
+  class AMService < AbstractService
+    include OMF::Base::Loggable
+
+    attr_accessor :authorizer
+
+    #implement ServiceAPI
+    implement AMServiceAPI
+
+    def get_version(options = {})
+      debug "GetVersion"
+      @return_struct[:geni_api] = 2
+      @return_struct[:code][:geni_code] = 0
+      @return_struct[:value] = {
+        :geni_api => 2,
+        :geni_api_versions => {
+          2 => 'URL'
+        },
+        :geni_request_rspec_versions => [{
+          :type => "geni",
+          :version => "3",
+          :schema => "http://www.geni.net/resources/rspec/3/request.xsd",
+          :namespace => "http://www.geni.net/resources/rspec/3",
+          :extensions => ["http://nitlab.inf.uth.gr/schema/sfa/rspec/1/request-reservation.xsd"]
+        }],
+        :geni_ad_rspec_versions => [{
+          :type => "geni",
+          :version => "3",
+          :schema => "http://www.geni.net/resources/rspec/3/ad.xsd",
+          :namespace => "http://www.geni.net/resources/rspec/3",
+          :extensions => ["http://nitlab.inf.uth.gr/schema/sfa/rspec/1/ad-reservation.xsd"]
+        }],
+        :omf_am => "0.1"
+      }
+      return @return_struct
+      #{
+      #  :geni_api => 2,
+      #  :code => {
+      #    :geni_code => 0
+      #  },
+      #  :value => {
+      #    :geni_api => 2,
+      #    :geni_api_versions => {
+      #      2 => 'URL'
+      #    },
+      #    :geni_request_rspec_versions => [{
+      #      :type => "GENI",
+      #      :version => "3",
+      #      :schema => "http://www.geni.net/resources/rspec/3/request.xsd",
+      #      :namespace => "http://www.geni.net/resources/rspec/3",
+      #      :extensions => ["http://nitlab.inf.uth.gr/schema/sfa/rspec/1/request-reservation.xsd"]
+      #    }],
+      #    :geni_ad_rspec_versions => [{
+      #      :type => "GENI",
+      #      :version => "3",
+      #      :schema => "http://www.geni.net/resources/rspec/3/ad.xsd",
+      #      :namespace => "http://www.geni.net/resources/rspec/3",
+      #      :extensions => ["http://nitlab.inf.uth.gr/schema/sfa/rspec/1/ad-reservation.xsd"]
+      #    }],
+      #    :omf_am => "0.1"
+      #  },
+      #  :output => {}
+      #}
+    end
+
+    def list_resources(credentials, options)
+      debug 'ListResources: Options: ', options.inspect
+
+      only_available = options["geni_available"]
+      compressed = options["geni_compressed"]
+      slice_urn = options["geni_slice_urn"]
+      rspec_version = options["geni_rspec_version"]
+
+      if rspec_version.nil?
+        @return_struct[:code][:geni_code] = 1 # Bad Arguments
+        @return_struct[:output] = "'geni_rspec_version' argument is missing."
+        @return_struct[:value] = {}
+        return @return_struct
+        #ans = {
+        #  :code => {
+        #    :geni_code => 1 # Bad Arguments
+        #  },
+        #  :output => "'geni_rspec_version' argument is missing."
+        #}
+        #return ans
+      end
+      unless rspec_version["type"].downcase.eql?("geni") && (rspec_version["version"].eql?("3.0") ||
+                                                             rspec_version["version"].eql?("3"))
+        @return_struct[:code][:geni_code] = 4 # Bad Version
+        @return_struct[:output] = "'Version' or 'Type' of RSpecs are not the same with that 'GetVersion' returns."
+        @return_struct[:value] = {}
+        return @return_struct
+        #ans = {
+        #  :code => {
+        #    :geni_code => 4 # Bad Version
+        #  },
+        #  :output => "'Version' or 'Type' of RSpecs are not the same with that 'GetVersion' returns."
+        #}
+        #return ans
+      end
+
+      authorizer = OMF::SFA::AM::RPC::AMAuthorizer.create_for_sfa_request(slice_urn, credentials, @request, @manager)
+      #@authorizer.check_credentials(slice_urn, credentials.first, @manager)
+      if slice_urn
+        # have leases in top of rspec
+        resources = @manager.find_all_leases_for_account(authorizer.account, authorizer)
+        resources.concat(@manager.find_all_components_for_account(authorizer.account, authorizer))
+
+        # have leases inside the components
+        #resources = @manager.find_all_components_for_account(authorizer.account, authorizer)
+
+        res = OMF::SFA::Resource::OComponent.sfa_advertisement_xml(resources, {:type => 'manifest'}).to_xml
+      else
+        # have leases in top of rspec
+        resources = @manager.find_all_leases(authorizer)
+        resources.concat(@manager.find_all_components_for_account(@manager._get_nil_account, authorizer))
+
+        # have leases inside the components
+        #resources = @manager.find_all_components_for_account(@manager._get_nil_account, authorizer)
+
+        res = OMF::SFA::Resource::OComponent.sfa_advertisement_xml(resources).to_xml
+      end
+      # TODO: implement the "available_only" option
+
+      # only list independent resources (TODO: What does this mean??)
+      #resources = resources.select {|r| r.independent_component?}
+      #debug "Resources for '#{slice_urn}' >>> #{resources.inspect}"
+
+      #res = OMF::SFA::Resource::OComponent.sfa_advertisement_xml(resources).to_xml
+      if compressed
+	      res = Base64.encode64(Zlib::Deflate.deflate(res))
+      end
+
+      @return_struct[:code][:geni_code] = 0
+      @return_struct[:value] = res
+      return @return_struct
+
+      #{
+      #  :code => {
+      #    :geni_code => 0
+      #  },
+      #  :value => res
+      #}
+    rescue OMF::SFA::AM::InsufficientPrivilegesException => e
+      @return_struct[:code][:geni_code] = 3
+      @return_struct[:output] = e.to_s
+      @return_struct[:value] = {}
+      return @return_struct
+
+      #{
+      #  :code => {
+      #    :geni_code => 3
+      #  },
+      #  :output => e
+      #}
+    end
+
+    def create_sliver(slice_urn, credentials, rspec_s, users, options)
+      debug 'CreateSliver: SLICE URN: ', slice_urn, ' RSPEC: ', rspec_s, ' USERS: ', users.inspect
+
+      if slice_urn.nil? || credentials.nil? || rspec_s.nil?
+        @return_struct[:code][:geni_code] = 1 # Bad Arguments
+        @return_struct[:output] = "Some of the following arguments are missing: 'slice_urn', 'credentials', 'rspec'"
+        @return_struct[:value] = {}
+        return @return_struct
+      end
+
+      #@authorizer.check_credentials(slice_urn, credentials.first, @manager)
+      authorizer = OMF::SFA::AM::RPC::AMAuthorizer.create_for_sfa_request(slice_urn, credentials, @request, @manager)
+
+      rspec = Nokogiri::XML.parse(rspec_s)
+      resources = @manager.update_resources_from_rspec(rspec.root, true, authorizer)
+
+      # TODO: Still need to implement USER handling
+
+      res = OMF::SFA::Resource::OComponent.sfa_advertisement_xml(resources, {:type => 'manifest'}).to_s
+
+      @return_struct[:code][:geni_code] = 0
+      @return_struct[:value] = res
+      return @return_struct
+
+      #{ :code => {
+      #    :geni_code => 0
+      #  },
+      #  :value => res
+      #}
+    rescue OMF::SFA::AM::InsufficientPrivilegesException => e
+      @return_struct[:code][:geni_code] = 3
+      @return_struct[:output] = e.to_s
+      @return_struct[:value] = {}
+      return @return_struct
+    rescue OMF::SFA::AM::UnknownResourceException => e
+      debug('CreateSliver Exception', e.to_s)
+      @return_struct[:code][:geni_code] = 12 # Search Failed
+      @return_struct[:output] = e.to_s
+      @return_struct[:value] = {}
+      return @return_struct
+    rescue OMF::SFA::AM::FormatException => e
+      debug('CreateSliver Exception', e.to_s)
+      @return_struct[:code][:geni_code] = 4 # Bad Version
+      @return_struct[:output] = e.to_s
+      @return_struct[:value] = {}
+      return @return_struct
+    end
+
+    def sliver_status(slice_urn, credentials, options)
+      debug('SliverStatus for ', slice_urn)
+      #@authorizer.check_credentials(slice_urn, credentials.first, @manager)
+
+
+      if slice_urn.nil? || credentials.nil?
+        @return_struct[:code][:geni_code] = 1 # Bad Arguments
+        @return_struct[:output] = "Some of the following arguments are missing: 'slice_urn', 'credentials'"
+        @return_struct[:value] = false
+        return @return_struct
+      end
+      #@authorizer.check_credentials(slice_urn, credentials.first, @manager)
+      authorizer = OMF::SFA::AM::RPC::AMAuthorizer.create_for_sfa_request(slice_urn, credentials, @request, @manager)
+      #raise OMF::SFA::AM::InsufficientPrivilegesException.new("Account is closed.") if authorizer.account.closed?
+
+      status = {}
+      status['omf_expires_at'] = authorizer.account.valid_until.utc.strftime('%Y%m%d%H%M%SZ')
+
+      resources = @manager.find_all_components_for_account(authorizer.account, authorizer)
+      # only list independent resources
+      resources = resources.select {|r| r.independent_component?}
+
+      unless resources.empty?
+        status['geni_urn'] = slice_urn
+        #status['geni_urn'] = "urn:publicid:IDN+omf:nitos+sliver+accdsw"
+
+        # Any of the following configuring, ready, failed, and unknown
+        status['geni_status'] = 'unknown'
+
+        status['geni_resources'] = resources.collect do |r|
+          {
+            'geni_urn'=> r.urn,
+            'geni_status' => r.status,
+            'geni_error' => '',
+          }
+        end
+        @return_struct[:value] = status
+      else
+        @return_struct[:value] = {}
+      end
+
+
+      @return_struct[:code][:geni_code] = 0
+      return @return_struct
+
+      #{ :code => {
+      #    :geni_code => 0
+      #  },
+      #  :value => status
+      #}
+    rescue OMF::SFA::AM::InsufficientPrivilegesException => e
+      @return_struct[:code][:geni_code] = 3
+      @return_struct[:output] = e.to_s
+      @return_struct[:value] = false
+      return @return_struct
+    end
+
+    def renew_sliver(slice_urn, credentials, expiration_time, options)
+      debug('RenewSliver ', slice_urn, ' until <', expiration_time, '>')
+
+      if slice_urn.nil? || credentials.nil? || expiration_time.nil?
+        @return_struct[:code][:geni_code] = 1 # Bad Arguments
+        @return_struct[:output] = "Some of the following arguments are missing: 'slice_urn', 'credentials', 'expiration_time'"
+        @return_struct[:value] = false
+        return @return_struct
+      end
+
+      if expiration_time.kind_of?(XMLRPC::DateTime)
+        expiration_time = expiration_time.to_time
+      else
+        expiration_time = Time.parse(expiration_time)
+      end
+      debug('RenewSliver ', slice_urn, ' until <', expiration_time, '>')
+      #authorizer.check_credentials(slice_urn, credentials.first, @manager)
+      authorizer = OMF::SFA::AM::RPC::AMAuthorizer.create_for_sfa_request(slice_urn, credentials, @request, @manager)
+
+      @manager.renew_account_until({ :urn => slice_urn }, expiration_time, authorizer)
+
+      @return_struct[:code][:geni_code] = 0
+      @return_struct[:value] = true
+      return @return_struct
+
+      #{ :code => {
+      #    :geni_code => 0
+      #  },
+      #  :value => true
+      #}
+    rescue OMF::SFA::AM::UnavailableResourceException => e
+      @return_struct[:code][:geni_code] = 12 # Search Failed
+      @return_struct[:output] = e.to_s
+      @return_struct[:value] = false
+      return @return_struct
+    rescue OMF::SFA::AM::InsufficientPrivilegesException => e
+      @return_struct[:code][:geni_code] = 3
+      @return_struct[:output] = e.to_s
+      @return_struct[:value] = false
+      return @return_struct
+    end
+
+    # close the account and release the attached resources
+    def delete_sliver(slice_urn, credentials, options)
+      debug('DeleteSliver ', slice_urn)
+
+      if slice_urn.nil? || credentials.nil?
+        @return_struct[:code][:geni_code] = 1 # Bad Arguments
+        @return_struct[:output] = "Some of the following arguments are missing: 'slice_urn', 'credentials'"
+        @return_struct[:value] = {}
+        return @return_struct
+      end
+      #@authorizer.check_credentials(slice_urn, credentials.first, @manager)
+      authorizer = OMF::SFA::AM::RPC::AMAuthorizer.create_for_sfa_request(slice_urn, credentials, @request, @manager)
+
+      # We don't like deleting things
+      account = @manager.close_account({ :urn => slice_urn }, authorizer)
+      # TODO: Should this really be here? Seems to be the job of the AM manager.
+      #account = authorizer.account
+      @manager.release_all_components_for_account(account, authorizer)
+      debug "Slice '#{slice_urn}' associated with account '#{account.id}:#{account.closed_at}'"
+
+      @return_struct[:code][:geni_code] = 0
+      @return_struct[:value] = true
+      return @return_struct
+
+      #{ :code => {
+      #    :geni_code => 0
+      #  },
+      #  :value => true
+      #}
+    rescue OMF::SFA::AM::UnavailableResourceException => e
+      @return_struct[:code][:geni_code] = 12 # Search Failed
+      @return_struct[:output] = e.to_s
+      @return_struct[:value] = {}
+      return @return_struct
+    end
+
+    # close the account but do not release its resources
+    def shutdown_sliver(slice_urn, credentials, options = {})
+      #@authorizer.check_credentials(slice_urn, credentials.first, @manager)
+      authorizer = OMF::SFA::AM::RPC::AMAuthorizer.create_for_sfa_request(slice_urn, credentials, @request, @manager)
+
+      if slice_urn.nil? || credentials.nil?
+        @return_struct[:code][:geni_code] = 1 # Bad Arguments
+        @return_struct[:output] = "Some of the following arguments are missing: 'slice_urn', 'credentials'"
+        return @return_struct
+      end
+      #puts "SLICE URN: #{slice_urn}"
+      account = @manager.close_account({ :urn => slice_urn }, authorizer)
+
+      @return_struct[:code][:geni_code] = 0
+      @return_struct[:value] = true
+      return @return_struct
+
+      #{ :code => {
+      #    :geni_code => 0
+      #  },
+      #  :value => true
+      #}
+    end
+
+    private
+
+    def initialize(opts)
+      super
+      @manager = opts[:manager]
+      @liaison = opts[:liaison]
+      @return_struct = {
+        :code => {
+          :geni_code => ""
+        },
+        :value => {},
+        :output => {}
+      }
+    end
+
+  end # AMService
+
+end # module
+
+
+

data/lib/omf-sfa/am/am_liaison.rb

@@ -0,0 +1,93 @@
+require 'omf_base'
+require 'omf_base/dsl/xmpp'
+
+#require 'omf-sfa/resource'
+require 'omf-sfa/am/am_manager'
+
+
+module OMF::SFA::AM
+
+  extend OMF::SFA::AM
+
+  # This class implements the AM Liaison
+  #
+  class AMLiaison < OMF::Base::LObject
+    
+    include OmfCommon::DSL::Xmpp
+
+    attr_accessor :comm
+    @leases = {}
+
+    def initialize
+      #@comm = OmfCommon::Comm.new(:xmpp)
+      #EM.next_tick { @comm.connect('am_liaison', 'pw', 'localhost') }
+    end    
+
+    #def self.leases
+    #  @@leases ||= {}
+    #end
+
+    # It will send the corresponding create messages to the components contained
+    # in the lease when the lease is about to start. At the end of the
+    # lease the corresponding release messages will be sent to the components.
+    #
+    # @param [OLease] lease Contains the lease information "valid_from" and 
+    #                 "valid_until" along with the reserved components
+    #
+    def enable_lease(lease, component)
+      debug "enable_lease: lease: '#{lease.inspect}' component: '#{component.inspect}'"
+
+      resource_topic = @comm.get_topic(component.name)
+      raise UnknownResourceException.new "Cannot find resource's pubsub topic: '#{component.inspect}'" unless resource_topic
+
+      msg = @comm.create_message do |message|
+        message.property('name', component.name)
+        message.property('type', component.type)
+      end
+
+      msg.on_inform_created do |message|
+        logger.info "Resource '#{message.resource_id}' successfully created at #{Time.now}"
+        created_topic = @comm.get_topic(message.resource_id)
+
+        # TODO: we need to define whether we ask for specific resource_id in the create
+        # message or we let the resource_factory to pick one. We need to keep it synced 
+        # with the resource_id we have in the DB. 
+        resource_id = created_topic.id
+
+        r_msg = @comm.release_message { |m| m.element('resource_id', resource_id) }
+
+        r_msg.on_inform_released do |message|
+          logger.info "Resource (#{message.resource_id}) released at #{Time.now}"
+        end
+
+        timer = EventMachine::Timer.new(lease[:valid_until] - Time.now) do
+          r_msg.publish resource_topic.id
+          @leases[lease].delete(component.id)
+          @leases.delete(lease) if @leases[lease].empty?
+        end
+        @leases[lease][component.id] = {:end => timer}
+      end
+
+      timer = EventMachine::Timer.new(lease[:valid_from] - Time.now) do
+        resource_topic.subscribe do
+          # If subscribed, we publish a 'create' message
+          msg.publish resource_topic.id
+        end
+      end
+
+      @leases[lease] = {} unless @leases[lease]
+      @leases[lease] = { component.id => {:start => timer} }
+    end
+
+
+    #def release_lease(resource)
+
+    #  resource_topic = @comm.get_topic(resource.name)
+
+    #  raise UnknownResourceException.new "Cannot find resource's pubsub topic: '#{resource.inspect}'" unless resource_topic
+
+    #end
+
+  end # AMLiaison
+end # OMF::SFA::AM
+