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

#buf ⇒ Object 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

#length ⇒ Object 



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_bool ⇒ Object 



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

def read_bool
  read_int32 != 0
end

#read_double ⇒ Object 



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

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

#read_fixed32 ⇒ Object 



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

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

#read_fixed64 ⇒ Object 



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_float ⇒ Object 



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

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

#read_info ⇒ Object 



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_int64 ⇒ Object 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_sfixed32 ⇒ Object 



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

def read_sfixed32
  decode_zigzag(read_fixed32)
end

#read_sfixed64 ⇒ Object 



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

def read_sfixed64
  decode_zigzag(read_fixed64)
end

#read_sint64 ⇒ Object 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_string ⇒ Object Also known as: read_bytes 



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

def read_string
  read(read_uint64)
end

#read_uint64 ⇒ Object 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