Module: RocketAMF::Pure::Envelope

Includes:
ReadIOHelpers, WriteIOHelpers
Included in:
Envelope
Defined in:
lib/rocketamf/pure/remoting.rb

Overview

Included into RocketAMF::Envelope, this module replaces the populate_from_stream and serialize methods with actual working versions

Instance Method Summary collapse

Methods included from WriteIOHelpers

#byte_order, #byte_order_little?, #pack_double, #pack_int16_network, #pack_int8, #pack_integer, #pack_word32_network

Methods included from ReadIOHelpers

#byte_order, #byte_order_little?, #read_double, #read_int16_network, #read_int8, #read_word16_network, #read_word32_network, #read_word8

Instance Method Details

#populate_from_stream(stream) ⇒ Object

Included into RocketAMF::Envelope, this method handles deserializing an AMF request/response into the envelope



10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
 
# File 'lib/rocketamf/pure/remoting.rb', line 10

def populate_from_stream stream
  stream = StringIO.new(stream) unless StringIO === stream

  # Initialize
  @amf_version = 0
  @headers = []
  @messages = []

  # Read AMF version
  @amf_version = read_word16_network stream

  # Read in headers
  header_count = read_word16_network stream
  0.upto(header_count-1) do
    name = stream.read(read_word16_network(stream))
    name.force_encoding("UTF-8") if name.respond_to?(:force_encoding)
    must_understand = read_int8(stream) != 0
    length = read_word32_network stream
    data = RocketAMF.deserialize stream
    @headers << RocketAMF::Header.new(name, must_understand, data)
  end

  # Read in messages
  message_count = read_word16_network stream
  0.upto(message_count-1) do
    target_uri = stream.read(read_word16_network(stream))
    target_uri.force_encoding("UTF-8") if target_uri.respond_to?(:force_encoding)
    response_uri = stream.read(read_word16_network(stream))
    response_uri.force_encoding("UTF-8") if response_uri.respond_to?(:force_encoding)
    length = read_word32_network stream
    data = RocketAMF.deserialize stream
    if data.is_a?(Array) && data.length == 1 && data[0].is_a?(::RocketAMF::Values::AbstractMessage)
      data = data[0]
    end
    @messages << RocketAMF::Message.new(target_uri, response_uri, data)
  end

  self
end

#serialize ⇒ Object

Included into RocketAMF::Envelope, this method handles serializing an AMF request/response into the envelope



52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
 
# File 'lib/rocketamf/pure/remoting.rb', line 52

def serialize
  stream = ""

  # Write version
  stream << pack_int16_network(@amf_version)

  # Write headers
  stream << pack_int16_network(@headers.length) # Header count
  @headers.each do |h|
    name_str = h.name
    name_str.encode!("UTF-8").force_encoding("ASCII-8BIT") if name_str.respond_to?(:encode)
    stream << pack_int16_network(name_str.bytesize)
    stream << name_str
    stream << pack_int8(h.must_understand ? 1 : 0)
    stream << pack_word32_network(-1)
    stream << RocketAMF.serialize(h.data, 0)
  end

  # Write messages
  stream << pack_int16_network(@messages.length) # Message count
  @messages.each do |m|
    uri_str = m.target_uri
    uri_str.encode!("UTF-8").force_encoding("ASCII-8BIT") if uri_str.respond_to?(:encode)
    stream << pack_int16_network(uri_str.bytesize)
    stream << uri_str

    uri_str = m.response_uri
    uri_str.encode!("UTF-8").force_encoding("ASCII-8BIT") if uri_str.respond_to?(:encode)
    stream << pack_int16_network(uri_str.bytesize)
    stream << uri_str

    stream << pack_word32_network(-1)
    stream << AMF0_AMF3_MARKER if @amf_version == 3
    stream << RocketAMF.serialize(m.data, @amf_version)
  end

  stream
end