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
90
91
92
93
94
95
96
|
# File 'lib/httpx/transcoder/chunker.rb', line 53
def each
loop do
case @state
when :length
index = @buffer.index(CRLF)
return unless index && index.positive?
hexlen = @buffer.byteslice(0, index)
@buffer = @buffer.byteslice(index..-1) || "".b
hexlen[/\h/] || raise(Error, "wrong chunk size line: #{hexlen}")
@chunk_length = hexlen.hex
@finished = @chunk_length.zero?
nextstate(:crlf)
when :crlf
crlf_size = @finished && !@trailers ? 4 : 2
return if @buffer.bytesize < crlf_size
raise Error, "wrong chunked encoding format" unless @buffer.start_with?(CRLF * (crlf_size / 2))
@buffer = @buffer.byteslice(crlf_size..-1)
if @chunk_length.nil?
nextstate(:length)
else
return if @finished
nextstate(:data)
end
when :data
chunk = @buffer.byteslice(0, @chunk_length)
@buffer = @buffer.byteslice(@chunk_length..-1) || "".b
@chunk_buffer << chunk
@chunk_length -= chunk.bytesize
if @chunk_length.zero?
yield @chunk_buffer unless @chunk_buffer.empty?
@chunk_buffer.clear
@chunk_length = nil
nextstate(:crlf)
end
end
break if @buffer.empty?
end
end
|