Class: Beefcake::Buffer

Inherits:
Object
  • Object
show all
Defined in:
lib/beefcake/buffer/base.rb,
lib/beefcake/buffer/decode.rb,
lib/beefcake/buffer/encode.rb

Defined Under Namespace

Classes: BufferOverflowError, OutOfRangeError, UnknownType

Constant Summary collapse

MinUint32 =
0
MaxUint32 =
(1<<32)-1
MinInt32 =
-(1<<31)
MaxInt32 =
(1<<31)-1
MinUint64 =
0
MaxUint64 =
(1<<64)-1
MinInt64 =
-(1<<63)
MaxInt64 =
(1<<63)-1

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(buf = "") ⇒ Buffer

Returns a new instance of Buffer.



65
66
67
# File 'lib/beefcake/buffer/base.rb', line 65

def initialize(buf="")
  self.buf = buf
end

Instance Attribute Details

#bufObject Also known as: to_s, to_str

Returns the value of attribute buf.



42
43
44
# File 'lib/beefcake/buffer/base.rb', line 42

def buf
  @buf
end

Class Method Details

.encodable?(type) ⇒ Boolean

Returns:

  • (Boolean)


36
37
38
39
40
# File 'lib/beefcake/buffer/base.rb', line 36

def self.encodable?(type)
  return false if ! type.is_a?(Class)
  pims = type.public_instance_methods
  pims.include?(:encode) || pims.include?("encode")
end

.wire_for(type) ⇒ Object



15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
# File 'lib/beefcake/buffer/base.rb', line 15

def self.wire_for(type)
  case type
  when Class
    if encodable?(type)
      2
    else
      raise UnknownType, type
    end
  when :int32, :uint32, :sint32, :int64, :uint64, :sint64, :bool, Module
    0
  when :fixed64, :sfixed64, :double
    1
  when :string, :bytes
    2
  when :fixed32, :sfixed32, :float
    5
  else
    raise UnknownType, type
  end
end

Instance Method Details

#<<(bytes) ⇒ Object



79
80
81
# File 'lib/beefcake/buffer/base.rb', line 79

def <<(bytes)
  buf << bytes
end

#append(type, val, fn) ⇒ Object



7
8
9
10
11
12
13
14
# File 'lib/beefcake/buffer/encode.rb', line 7

def append(type, val, fn)
  if fn != 0
    wire = Buffer.wire_for(type)
    append_info(fn, wire)
  end

  __send__("append_#{type}", val)
end

#append_bool(n) ⇒ Object



104
105
106
# File 'lib/beefcake/buffer/encode.rb', line 104

def append_bool(n)
  append_int64(n ? 1 : 0)
end

#append_double(n) ⇒ Object



100
101
102
# File 'lib/beefcake/buffer/encode.rb', line 100

def append_double(n)
  self << [n].pack("E")
end

#append_fixed32(n, tag = false) ⇒ Object



21
22
23
24
25
26
27
# File 'lib/beefcake/buffer/encode.rb', line 21

def append_fixed32(n, tag=false)
  if n < MinUint32 || n > MaxUint32
    raise OutOfRangeError, n
  end

  self << [n].pack("V")
end

#append_fixed64(n) ⇒ Object



29
30
31
32
33
34
35
# File 'lib/beefcake/buffer/encode.rb', line 29

def append_fixed64(n)
  if n < MinUint64 || n > MaxUint64
    raise OutOfRangeError, n
  end

  self << [n & 0xFFFFFFFF, n >> 32].pack("VV")
end

#append_float(n) ⇒ Object



96
97
98
# File 'lib/beefcake/buffer/encode.rb', line 96

def append_float(n)
  self << [n].pack("e")
end

#append_info(fn, wire) ⇒ Object



16
17
18
19
# File 'lib/beefcake/buffer/encode.rb', line 16

def append_info(fn, wire)
  #self << ((fn << 3) | wire)
  append_uint32((fn << 3) | wire)
end

#append_int32(n) ⇒ Object



37
38
39
40
41
42
43
# File 'lib/beefcake/buffer/encode.rb', line 37

def append_int32(n)
  if n < MinInt32 || n > MaxInt32
    raise OutOfRangeError, n
  end

  append_int64(n)
end

#append_int64(n) ⇒ Object



53
54
55
56
57
58
59
60
61
62
63
# File 'lib/beefcake/buffer/encode.rb', line 53

def append_int64(n)
  if n < MinInt64 || n > MaxInt64
    raise OutOfRangeError, n
  end

  if n < 0
    n += (1 << 64)
  end

  append_uint64(n)
end

#append_sfixed32(n) ⇒ Object



69
70
71
# File 'lib/beefcake/buffer/encode.rb', line 69

def append_sfixed32(n)
  append_fixed32((n << 1) ^ (n >> 31))
end

#append_sfixed64(n) ⇒ Object



77
78
79
# File 'lib/beefcake/buffer/encode.rb', line 77

def append_sfixed64(n)
  append_fixed64((n << 1) ^ (n >> 63))
end

#append_sint32(n) ⇒ Object



65
66
67
# File 'lib/beefcake/buffer/encode.rb', line 65

def append_sint32(n)
  append_uint32((n << 1) ^ (n >> 31))
end

#append_sint64(n) ⇒ Object



73
74
75
# File 'lib/beefcake/buffer/encode.rb', line 73

def append_sint64(n)
  append_uint64((n << 1) ^ (n >> 63))
end

#append_string(s) ⇒ Object Also known as: append_bytes



108
109
110
111
# File 'lib/beefcake/buffer/encode.rb', line 108

def append_string(s)
  append_uint64(s.length)
  self << s
end

#append_uint32(n) ⇒ Object



45
46
47
48
49
50
51
# File 'lib/beefcake/buffer/encode.rb', line 45

def append_uint32(n)
  if n < MinUint32 || n > MaxUint32
    raise OutOfRangeError, n
  end

  append_uint64(n)
end

#append_uint64(n) ⇒ Object



81
82
83
84
85
86
87
88
89
90
91
92
93
94
# File 'lib/beefcake/buffer/encode.rb', line 81

def append_uint64(n)
  if n < MinUint64 || n > MaxUint64
    raise OutOfRangeError, n
  end

  while true
    bits = n & 0x7F
    n >>= 7
    if n == 0
      return self << bits
    end
    self << (bits | 0x80)
  end
end

#lengthObject



75
76
77
# File 'lib/beefcake/buffer/base.rb', line 75

def length
  @buf.respond_to?(:bytesize) ? @buf.bytesize : @buf.length
end

#read(n) ⇒ Object



83
84
85
86
87
88
89
90
91
92
93
94
# File 'lib/beefcake/buffer/base.rb', line 83

def read(n)
  case n
  when Class
    n.decode(read_string)
  when Symbol
    __send__("read_#{n}")
  when Module
    read_uint64
  else
    buf.slice!(0, n)
  end
end

#read_boolObject



85
86
87
# File 'lib/beefcake/buffer/decode.rb', line 85

def read_bool
  read_int32 != 0
end

#read_doubleObject



80
81
82
83
# File 'lib/beefcake/buffer/decode.rb', line 80

def read_double
  bytes = read(8)
  bytes.unpack("E").first
end

#read_fixed32Object



20
21
22
23
# File 'lib/beefcake/buffer/decode.rb', line 20

def read_fixed32
  bytes = read(4)
  bytes.unpack("V").first
end

#read_fixed64Object



25
26
27
28
29
# File 'lib/beefcake/buffer/decode.rb', line 25

def read_fixed64
  bytes = read(8)
  x, y = bytes.unpack("VV")
  x + (y << 32)
end

#read_floatObject



75
76
77
78
# File 'lib/beefcake/buffer/decode.rb', line 75

def read_float
  bytes = read(4)
  bytes.unpack("e").first
end

#read_infoObject



7
8
9
10
11
12
13
# File 'lib/beefcake/buffer/decode.rb', line 7

def read_info
  n    = read_uint64
  fn   = n >> 3
  wire = n & 0x7

  [fn, wire]
end

#read_int64Object Also known as: read_int32



31
32
33
34
35
36
37
# File 'lib/beefcake/buffer/decode.rb', line 31

def read_int64
  n = read_uint64
  if n > MaxInt64
    n -= (1 << 64)
  end
  n
end

#read_sfixed32Object



67
68
69
# File 'lib/beefcake/buffer/decode.rb', line 67

def read_sfixed32
  decode_zigzag(read_fixed32)
end

#read_sfixed64Object



71
72
73
# File 'lib/beefcake/buffer/decode.rb', line 71

def read_sfixed64
  decode_zigzag(read_fixed64)
end

#read_sint64Object Also known as: read_sint32



62
63
64
# File 'lib/beefcake/buffer/decode.rb', line 62

def read_sint64
  decode_zigzag(read_uint64)
end

#read_stringObject Also known as: read_bytes



15
16
17
# File 'lib/beefcake/buffer/decode.rb', line 15

def read_string
  read(read_uint64)
end

#read_uint64Object Also known as: read_uint32



40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
# File 'lib/beefcake/buffer/decode.rb', line 40

def read_uint64
  n = shift = 0
  while true
    if shift >= 64
      raise BufferOverflowError, "varint"
    end
    b = buf.slice!(0)

    ## 1.8.6 to 1.9 Compat
    if b.respond_to?(:ord)
      b = b.ord
    end

    n |= ((b & 0x7F) << shift)
    shift += 7
    if (b & 0x80) == 0
      return n
    end
  end
end

#skip(wire) ⇒ Object



89
90
91
92
93
94
95
96
# File 'lib/beefcake/buffer/decode.rb', line 89

def skip(wire)
  case wire
  when 0 then read_uint64
  when 1 then read_fixed64
  when 2 then read_string
  when 5 then read_fixed32
  end
end