60 lines
1.6 KiB
Ruby
60 lines
1.6 KiB
Ruby
# -*- coding: binary -*-
|
|
|
|
module Rex
|
|
module Proto
|
|
module Rmi
|
|
module Model
|
|
# This class provides a representation of an RMI call message
|
|
class Call < Element
|
|
|
|
# @!attribute message_id
|
|
# @return [Fixnum] the message id
|
|
attr_accessor :message_id
|
|
# @!attribute call_data
|
|
# @return [Rex::Proto::Rmi::Model::CallData] the call data
|
|
attr_accessor :call_data
|
|
|
|
private
|
|
|
|
# Reads the message id from the IO
|
|
#
|
|
# @param io [IO] the IO to read from
|
|
# @return [String]
|
|
# @raise [Rex::Proto::Rmi::DecodeError] if fails to decode the message id
|
|
def decode_message_id(io)
|
|
message_id = read_byte(io)
|
|
unless message_id == CALL_MESSAGE
|
|
raise Rex::Proto::Rmi::DecodeError, 'Failed to decode Call message id'
|
|
end
|
|
|
|
message_id
|
|
end
|
|
|
|
# Reads and deserializes the call data from the IO
|
|
#
|
|
# @param io [IO] the IO to read from
|
|
# @return [Rex::Java::Serialization::Model::Stream]
|
|
def decode_call_data(io)
|
|
call_data = Rex::Proto::Rmi::Model::CallData.decode(io)
|
|
|
|
call_data
|
|
end
|
|
|
|
# Encodes the message_id field
|
|
#
|
|
# @return [String]
|
|
def encode_message_id
|
|
[message_id].pack('C')
|
|
end
|
|
|
|
# Encodes the address field
|
|
#
|
|
# @return [String]
|
|
def encode_call_data
|
|
call_data.encode
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end |