umgeher/cowboy
0
Fork 0
mirror of https://github.com/ninenines/cowboy.git synced 2025-07-14 20:30:23 +00:00
Code Issues Wiki Activity

Remove usage of the inflate buffer

Browse source
This commit is contained in:
Ali Sabil 2013-07-02 10:58:12 +02:00
parent 39caf34fe3
commit 4edef3c45c
1 changed files with 9 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

21
src/cowboy_websocket.erl
View file

@ -57,9 +57,8 @@
frag_state = undefined :: frag_state(),
utf8_state = <<>> :: binary(),
deflate_frame = false :: boolean(),
inflate_state :: any(),
inflate_buffer = <<>> :: binary(),
deflate_state :: any()
inflate_state :: undefined | port(),
deflate_state :: undefined | port()
}).
%% @doc Upgrade an HTTP request to the Websocket protocol.
@ -121,7 +120,6 @@ websocket_extensions(State, Req) ->
{ok, State#state{
deflate_frame = true,
inflate_state = Inflate,
inflate_buffer = <<>>,
deflate_state = Deflate
}, Req2};
_ ->
@ -450,14 +448,13 @@ websocket_inflate_frame(Data, << Rsv1:1, _:2 >>, _,
#state{deflate_frame = DeflateFrame} = State)
when DeflateFrame =:= false orelse Rsv1 =:= 0 ->
{Data, State};
websocket_inflate_frame(Data, << 1:1, _:2 >>, false,
#state{inflate_buffer = Buffer} = State) ->
{<<>>, State#state{inflate_buffer = << Buffer/binary, Data/binary >>}};
websocket_inflate_frame(Data, << 1:1, _:2 >>, true,
#state{inflate_state = Inflate, inflate_buffer = Buffer} = State) ->
Deflated = << Buffer/binary, Data/binary, 0:8, 0:8, 255:8, 255:8 >>,
Result = zlib:inflate(Inflate, Deflated),
{iolist_to_binary(Result), State#state{inflate_buffer = <<>>}}.
websocket_inflate_frame(Data, << 1:1, _:2 >>, false, State) ->
Result = zlib:inflate(State#state.inflate_state, Data),
{iolist_to_binary(Result), State};
websocket_inflate_frame(Data, << 1:1, _:2 >>, true, State) ->
Result = zlib:inflate(State#state.inflate_state,
<< Data/binary, 0:8, 0:8, 255:8, 255:8 >>),
{iolist_to_binary(Result), State}.
-spec websocket_unmask(B, mask_key(), B) -> B when B::binary().
websocket_unmask(<<>>, _, Unmasked) ->