# -*- coding: binary -*-

module Msf
  module Java
    module Rmi
      module Client
        module Registry
          module Builder

            # Builds an RMI call to java/rmi/registry/RegistryImpl_Stub#lookup() used to
            # retrieve the remote reference bound to a name.
            #
            # @param opts [Hash]
            # @option opts [String] :name the name to lookup
            # @return [Rex::Proto::Rmi::Model::Call]
            # @see Msf::Java::Rmi::Builder.build_call
            def build_registry_lookup(opts = {})
              object_number = opts[:object_number] || 0
              uid_number = opts[:uid_number] || 0
              uid_time = opts[:uid_time] || 0
              uid_count = opts[:uid_count] || 0
              name = opts[:name] || ''

              call = build_call(
                object_number: object_number,
                uid_number: uid_number,
                uid_time: uid_time,
                uid_count: uid_count,
                operation: 2, # java.rmi.Remote lookup(java.lang.String)
                hash: registry_interface_hash,
                arguments: [Rex::Java::Serialization::Model::Utf.new(nil, name)]
              )

              call
            end

            # Builds an RMI call to java/rmi/registry/RegistryImpl_Stub#list() used to
            # enumerate the names bound in a registry
            #
            # @param opts [Hash]
            # @return [Rex::Proto::Rmi::Model::Call]
            # @see Msf::Java::Rmi::Builder.build_call
            def build_registry_list(opts = {})
              object_number = opts[:object_number] || 0
              uid_number = opts[:uid_number] || 0
              uid_time = opts[:uid_time] || 0
              uid_count = opts[:uid_count] || 0

              call = build_call(
                object_number: object_number,
                uid_number: uid_number,
                uid_time: uid_time,
                uid_count: uid_count,
                operation: 1, # java.lang.String list()[]
                hash: registry_interface_hash,
                arguments: []
              )

              call
            end
          end
        end
      end
    end
  end
end