Skip to content

Instantly share code, notes, and snippets.

@archaelus

archaelus/mochiweb_loop.erl

Created March 9, 2009 20:30
Show Gist options
  • Save archaelus/76455 to your computer and use it in GitHub Desktop.
Save archaelus/76455 to your computer and use it in GitHub Desktop.
Download ZIP
mochiweb pre-gzipped content example
Raw
mochiweb_loop.erl
serve_gzipped_files(Req, RelPath, DocRoot) ->
case mochiweb_util:safe_relative_path(RelPath) of
undefined ->
Req:not_found();
SafeRelPath ->
serve_gzipped_files2(Req, SafeRelPath, DocRoot)
end.
serve_gzipped_files2(Req, SafeRelPath, DocRoot) ->
Header = case Req:get_header_value("Accept-Encoding") of
undefined -> [];
L when is_list(L) -> L
end,
Encodings = string:tokens(Header, ","),
case lists:member("gzip", Encodings) of
true ->
GZName = SafeRelPath ++ ".gz",
case filelib:is_regular(filename:join([DocRoot, GZName])) of
true ->
?INFO("Serving ~p instead.", [GZName]),
OrigName = filename:join([DocRoot, SafeRelPath]),
Req:serve_file(GZName, DocRoot,
[{"Content-Encoding", "gzip"},
{"Content-Type",
mochiweb_util:guess_mime(OrigName)}]);
false ->
?INFO("No pre-gzipped file.", []),
Req:serve_file(SafeRelPath, DocRoot)
end;
false ->
?INFO("Doesn't accept gzip.", []),
Req:serve_file(SafeRelPath, DocRoot)
end.
zip_static_content(DocRoot) ->
filelib:fold_files(DocRoot,
".*", true,
fun zip_static_file/2,
undefined).
zip_static_file(File, undefined) ->
GZName = File ++ ".gz",
case {filelib:is_regular(GZName), filename:extension(File)} of
{_, ".gz"} -> skip;
{false, _Other} ->
{ok, Bin} = file:read_file(File),
GZBin = zlib:gzip(Bin),
case byte_size(Bin) - byte_size(GZBin) of
Savings when Savings > 0 ->
?INFO("Zipping ~p to save ~w bytes.", [File, Savings]),
file:write_file(GZName, [GZBin]);
_ -> skip
end;
_ -> skip
end,
undefined.
Raw
mochiweb_request.erl
%% @author Bob Ippolito <[email protected]>
%% @copyright 2007 Mochi Media, Inc.
%% @doc MochiWeb HTTP Request abstraction.
-module(mochiweb_request, [Socket, Method, RawPath, Version, Headers]).
-author('[email protected]').
-include_lib("kernel/include/file.hrl").
-define(QUIP, "Any of you quaids got a smint?").
-define(READ_SIZE, 8192).
-export([get_header_value/1, get_primary_header_value/1, get/1, dump/0]).
-export([send/1, recv/1, recv/2, recv_body/0, recv_body/1]).
-export([start_response/1, start_response_length/1, start_raw_response/1]).
-export([respond/1, ok/1]).
-export([not_found/0, not_found/1]).
-export([redirect/1, redirect/2]).
-export([parse_post/0, parse_qs/0]).
-export([should_close/0, cleanup/0]).
-export([parse_cookie/0, get_cookie_value/1]).
-export([serve_file/2, serve_file/3]).
-export([test/0]).
-define(SAVE_QS, mochiweb_request_qs).
-define(SAVE_PATH, mochiweb_request_path).
-define(SAVE_RECV, mochiweb_request_recv).
-define(SAVE_BODY, mochiweb_request_body).
-define(SAVE_BODY_LENGTH, mochiweb_request_body_length).
-define(SAVE_POST, mochiweb_request_post).
-define(SAVE_COOKIE, mochiweb_request_cookie).
-define(SAVE_FORCE_CLOSE, mochiweb_request_force_close).
%% @type iolist() = [iolist() | binary() | char()].
%% @type iodata() = binary() | iolist().
%% @type key() = atom() | string() | binary()
%% @type value() = atom() | string() | binary() | integer()
%% @type headers(). A mochiweb_headers structure.
%% @type response(). A mochiweb_response parameterized module instance.
%% @type ioheaders() = headers() | [{key(), value()}].
% 10 second default idle timeout
-define(IDLE_TIMEOUT, 10000).
% Maximum recv_body() length of 1MB
-define(MAX_RECV_BODY, (1024*1024)).
%% @spec get_header_value(K) -> undefined | Value
%% @doc Get the value of a given request header.
get_header_value(K) ->
mochiweb_headers:get_value(K, Headers).
get_primary_header_value(K) ->
mochiweb_headers:get_primary_value(K, Headers).
%% @type field() = socket | method | raw_path | version | headers | peer | path | body_length | range
%% @spec get(field()) -> term()
%% @doc Return the internal representation of the given field.
get(socket) ->
Socket;
get(method) ->
Method;
get(raw_path) ->
RawPath;
get(version) ->
Version;
get(headers) ->
Headers;
get(peer) ->
case inet:peername(Socket) of
{ok, {Addr={10, _, _, _}, _Port}} ->
case get_header_value("x-forwarded-for") of
undefined ->
inet_parse:ntoa(Addr);
Hosts ->
string:strip(lists:last(string:tokens(Hosts, ",")))
end;
{ok, {{127, 0, 0, 1}, _Port}} ->
case get_header_value("x-forwarded-for") of
undefined ->
"127.0.0.1";
Hosts ->
string:strip(lists:last(string:tokens(Hosts, ",")))
end;
{ok, {Addr, _Port}} ->
inet_parse:ntoa(Addr)
end;
get(path) ->
case erlang:get(?SAVE_PATH) of
undefined ->
{Path0, _, _} = mochiweb_util:urlsplit_path(RawPath),
Path = mochiweb_util:unquote(Path0),
put(?SAVE_PATH, Path),
Path;
Cached ->
Cached
end;
get(body_length) ->
erlang:get(?SAVE_BODY_LENGTH);
get(range) ->
case get_header_value(range) of
undefined ->
undefined;
RawRange ->
parse_range_request(RawRange)
end.
%% @spec dump() -> {mochiweb_request, [{atom(), term()}]}
%% @doc Dump the internal representation to a "human readable" set of terms
%% for debugging/inspection purposes.
dump() ->
{?MODULE, [{method, Method},
{version, Version},
{raw_path, RawPath},
{headers, mochiweb_headers:to_list(Headers)}]}.
%% @spec send(iodata()) -> ok
%% @doc Send data over the socket.
send(Data) ->
case gen_tcp:send(Socket, Data) of
ok ->
ok;
_ ->
exit(normal)
end.
%% @spec recv(integer()) -> binary()
%% @doc Receive Length bytes from the client as a binary, with the default
%% idle timeout.
recv(Length) ->
recv(Length, ?IDLE_TIMEOUT).
%% @spec recv(integer(), integer()) -> binary()
%% @doc Receive Length bytes from the client as a binary, with the given
%% Timeout in msec.
recv(Length, Timeout) ->
case gen_tcp:recv(Socket, Length, Timeout) of
{ok, Data} ->
put(?SAVE_RECV, true),
Data;
_ ->
exit(normal)
end.
%% @spec body_length() -> undefined | chunked | unknown_transfer_encoding | integer()
%% @doc Infer body length from transfer-encoding and content-length headers.
body_length() ->
case get_header_value("transfer-encoding") of
undefined ->
case get_header_value("content-length") of
undefined ->
undefined;
Length ->
list_to_integer(Length)
end;
"chunked" ->
chunked;
Unknown ->
{unknown_transfer_encoding, Unknown}
end.
%% @spec recv_body() -> binary()
%% @doc Receive the body of the HTTP request (defined by Content-Length).
%% Will only receive up to the default max-body length of 1MB.
recv_body() ->
recv_body(?MAX_RECV_BODY).
%% @spec recv_body(integer()) -> binary()
%% @doc Receive the body of the HTTP request (defined by Content-Length).
%% Will receive up to MaxBody bytes.
recv_body(MaxBody) ->
case get_header_value("expect") of
"100-continue" ->
start_raw_response({100, gb_trees:empty()});
_Else ->
ok
end,
Body = case body_length() of
undefined ->
undefined;
{unknown_transfer_encoding, Unknown} ->
exit({unknown_transfer_encoding, Unknown});
chunked ->
read_chunked_body(MaxBody, []);
0 ->
<<>>;
Length when is_integer(Length), Length =< MaxBody ->
recv(Length);
Length ->
exit({body_too_large, Length})
end,
put(?SAVE_BODY, Body),
Body.
%% @spec start_response({integer(), ioheaders()}) -> response()
%% @doc Start the HTTP response by sending the Code HTTP response and
%% ResponseHeaders. The server will set header defaults such as Server
%% and Date if not present in ResponseHeaders.
start_response({Code, ResponseHeaders}) ->
HResponse = mochiweb_headers:make(ResponseHeaders),
HResponse1 = mochiweb_headers:default_from_list(server_headers(),
HResponse),
start_raw_response({Code, HResponse1}).
%% @spec start_raw_response({integer(), headers()}) -> response()
%% @doc Start the HTTP response by sending the Code HTTP response and
%% ResponseHeaders.
start_raw_response({Code, ResponseHeaders}) ->
F = fun ({K, V}, Acc) ->
[make_io(K), <<": ">>, V, <<"\r\n">> | Acc]
end,
End = lists:foldl(F, [<<"\r\n">>],
mochiweb_headers:to_list(ResponseHeaders)),
send([make_version(Version), make_code(Code), <<"\r\n">> | End]),
mochiweb:new_response({THIS, Code, ResponseHeaders}).
%% @spec start_response_length({integer(), ioheaders(), integer()}) -> response()
%% @doc Start the HTTP response by sending the Code HTTP response and
%% ResponseHeaders including a Content-Length of Length. The server
%% will set header defaults such as Server
%% and Date if not present in ResponseHeaders.
start_response_length({Code, ResponseHeaders, Length}) ->
HResponse = mochiweb_headers:make(ResponseHeaders),
HResponse1 = mochiweb_headers:enter("Content-Length", Length, HResponse),
start_response({Code, HResponse1}).
%% @spec respond({integer(), ioheaders(), iodata() | chunked | {file, IoDevice}}) -> response()
%% @doc Start the HTTP response with start_response, and send Body to the
%% client (if the get(method) /= 'HEAD'). The Content-Length header
%% will be set by the Body length, and the server will insert header
%% defaults.
respond({Code, ResponseHeaders, {file, IoDevice}}) ->
Length = iodevice_size(IoDevice),
Response = start_response_length({Code, ResponseHeaders, Length}),
case Method of
'HEAD' ->
ok;
_ ->
iodevice_stream(IoDevice)
end,
Response;
respond({Code, ResponseHeaders, chunked}) ->
HResponse = mochiweb_headers:make(ResponseHeaders),
HResponse1 = case Method of
'HEAD' ->
%% This is what Google does, http://www.google.com/
%% is chunked but HEAD gets Content-Length: 0.
%% The RFC is ambiguous so emulating Google is smart.
mochiweb_headers:enter("Content-Length", "0",
HResponse);
_ when Version >= {1, 1} ->
%% Only use chunked encoding for HTTP/1.1
mochiweb_headers:enter("Transfer-Encoding", "chunked",
HResponse);
_ ->
%% For pre-1.1 clients we send the data as-is
%% without a Content-Length header and without
%% chunk delimiters. Since the end of the document
%% is now ambiguous we must force a close.
put(?SAVE_FORCE_CLOSE, true),
HResponse
end,
start_response({Code, HResponse1});
respond({Code, ResponseHeaders, Body}) ->
Response = start_response_length({Code, ResponseHeaders, iolist_size(Body)}),
case Method of
'HEAD' ->
ok;
_ ->
send(Body)
end,
Response.
%% @spec not_found() -> response()
%% @doc Alias for <code>not_found([])</code>.
not_found() ->
not_found([]).
%% @spec not_found(ExtraHeaders) -> response()
%% @doc Alias for <code>respond({404, [{"Content-Type", "text/plain"}
%% | ExtraHeaders], &lt;&lt;"Not found."&gt;&gt;})</code>.
not_found(ExtraHeaders) ->
respond({404, [{"Content-Type", "text/plain"} | ExtraHeaders],
<<"Not found.">>}).
%% @spec redirect(Path::string()) -> response()
%% @doc Alias for <code>not_found(Path, [])</code>.
redirect(Path) ->
redirect(Path, []).
%% @spec redirect(Path::string(), ExtraHeaders::list()) -> response()
%% @doc Alias for <code>respond({302, [{"Content-Type", "text/plain"}, {"Location", Path},
%% | ExtraHeaders], &lt;&lt;"Found."&gt;&gt;})</code>.
redirect(Path, ExtraHeaders) ->
respond({302, % using a Found instead of See Other as per compatibility note http://www.w3.org/Protocols/rfc2616/rfc2616-sec10.html#sec10.3.3
[{"Content-Type", "text/plain"},
{"Location", Path} | ExtraHeaders],
<<"Found.">>}).
%% @spec ok({value(), iodata()} | {value(), ioheaders(), iodata() | {file, IoDevice}}) ->
%% response()
%% @doc respond({200, [{"Content-Type", ContentType} | Headers], Body}).
ok({ContentType, Body}) ->
ok({ContentType, [], Body});
ok({ContentType, ResponseHeaders, Body}) ->
HResponse = mochiweb_headers:make(ResponseHeaders),
case THIS:get(range) of
X when X =:= undefined; X =:= fail ->
HResponse1 = mochiweb_headers:enter("Content-Type", ContentType, HResponse),
respond({200, HResponse1, Body});
Ranges ->
{PartList, Size} = range_parts(Body, Ranges),
case PartList of
[] -> %% no valid ranges
HResponse1 = mochiweb_headers:enter("Content-Type",
ContentType,
HResponse),
%% could be 416, for now we'll just return 200
respond({200, HResponse1, Body});
PartList ->
{RangeHeaders, RangeBody} =
parts_to_body(PartList, ContentType, Size),
HResponse1 = mochiweb_headers:enter_from_list(
[{"Accept-Ranges", "bytes"} |
RangeHeaders],
HResponse),
respond({206, HResponse1, RangeBody})
end
end.
%% @spec should_close() -> bool()
%% @doc Return true if the connection must be closed. If false, using
%% Keep-Alive should be safe.
should_close() ->
ForceClose = erlang:get(mochiweb_request_force_close) =/= undefined,
DidNotRecv = erlang:get(mochiweb_request_recv) =:= undefined,
ForceClose orelse Version < {1, 0}
%% Connection: close
orelse get_header_value("connection") =:= "close"
%% HTTP 1.0 requires Connection: Keep-Alive
orelse (Version =:= {1, 0}
andalso get_header_value("connection") =/= "Keep-Alive")
%% unread data left on the socket, can't safely continue
orelse (DidNotRecv
andalso get_header_value("content-length") =/= undefined
andalso list_to_integer(get_header_value("content-length")) > 0)
orelse (DidNotRecv
andalso get_header_value("transfer-encoding") =:= "chunked").
%% @spec cleanup() -> ok
%% @doc Clean up any junk in the process dictionary, required before continuing
%% a Keep-Alive request.
cleanup() ->
[erase(K) || K <- [?SAVE_QS,
?SAVE_PATH,
?SAVE_RECV,
?SAVE_BODY,
?SAVE_POST,
?SAVE_COOKIE,
?SAVE_FORCE_CLOSE]],
ok.
%% @spec parse_qs() -> [{Key::string(), Value::string()}]
%% @doc Parse the query string of the URL.
parse_qs() ->
case erlang:get(?SAVE_QS) of
undefined ->
{_, QueryString, _} = mochiweb_util:urlsplit_path(RawPath),
Parsed = mochiweb_util:parse_qs(QueryString),
put(?SAVE_QS, Parsed),
Parsed;
Cached ->
Cached
end.
%% @spec get_cookie_value(Key::string) -> string() | undefined
%% @doc Get the value of the given cookie.
get_cookie_value(Key) ->
proplists:get_value(Key, parse_cookie()).
%% @spec parse_cookie() -> [{Key::string(), Value::string()}]
%% @doc Parse the cookie header.
parse_cookie() ->
case erlang:get(?SAVE_COOKIE) of
undefined ->
Cookies = case get_header_value("cookie") of
undefined ->
[];
Value ->
mochiweb_cookies:parse_cookie(Value)
end,
put(?SAVE_COOKIE, Cookies),
Cookies;
Cached ->
Cached
end.
%% @spec parse_post() -> [{Key::string(), Value::string()}]
%% @doc Parse an application/x-www-form-urlencoded form POST. This
%% has the side-effect of calling recv_body().
parse_post() ->
case erlang:get(?SAVE_POST) of
undefined ->
Parsed = case recv_body() of
undefined ->
[];
Binary ->
case get_primary_header_value("content-type") of
"application/x-www-form-urlencoded" ++ _ ->
mochiweb_util:parse_qs(Binary);
_ ->
[]
end
end,
put(?SAVE_POST, Parsed),
Parsed;
Cached ->
Cached
end.
read_chunked_body(Max, Acc) ->
case read_chunk_length() of
0 ->
read_chunk(0),
iolist_to_binary(lists:reverse(Acc));
Length when Length > Max ->
exit({body_too_large, chunked});
Length ->
read_chunked_body(Max - Length, [read_chunk(Length) | Acc])
end.
%% @spec read_chunk_length() -> integer()
%% @doc Read the length of the next HTTP chunk.
read_chunk_length() ->
inet:setopts(Socket, [{packet, line}]),
case gen_tcp:recv(Socket, 0, ?IDLE_TIMEOUT) of
{ok, Header} ->
inet:setopts(Socket, [{packet, raw}]),
Splitter = fun (C) ->
C =/= $\r andalso C =/= $\n andalso C =/= $
end,
{Hex, _Rest} = lists:splitwith(Splitter, binary_to_list(Header)),
mochihex:to_int(Hex);
_ ->
exit(normal)
end.
%% @spec read_chunk(integer()) -> Chunk::binary() | [Footer::binary()]
%% @doc Read in a HTTP chunk of the given length. If Length is 0, then read the
%% HTTP footers (as a list of binaries, since they're nominal).
read_chunk(0) ->
inet:setopts(Socket, [{packet, line}]),
F = fun (F1, Acc) ->
case gen_tcp:recv(Socket, 0, ?IDLE_TIMEOUT) of
{ok, <<"\r\n">>} ->
Acc;
{ok, Footer} ->
F1(F1, [Footer | Acc]);
_ ->
exit(normal)
end
end,
Footers = F(F, []),
inet:setopts(Socket, [{packet, raw}]),
Footers;
read_chunk(Length) ->
case gen_tcp:recv(Socket, 2 + Length, ?IDLE_TIMEOUT) of
{ok, <<Chunk:Length/binary, "\r\n">>} ->
Chunk;
_ ->
exit(normal)
end.
%% @spec serve_file(Path, DocRoot) -> Response
%% @doc Serve a file relative to DocRoot.
serve_file(Path, DocRoot) ->
serve_file(Path, DocRoot, []).
%% @spec serve_file(Path, DocRoot, ExtraHeaders) -> Response
%% @doc Serve a file relative to DocRoot.
serve_file(Path, DocRoot, ExtraHeaders) ->
case mochiweb_util:safe_relative_path(Path) of
undefined ->
not_found(ExtraHeaders);
RelPath ->
FullPath = filename:join([DocRoot, RelPath]),
File = case filelib:is_dir(FullPath) of
true ->
filename:join([FullPath, "index.html"]);
false ->
FullPath
end,
case file:read_file_info(File) of
{ok, FileInfo} ->
LastModified = httpd_util:rfc1123_date(FileInfo#file_info.mtime),
case get_header_value("if-modified-since") of
LastModified ->
respond({304, ExtraHeaders, ""});
_ ->
case file:open(File, [raw, binary]) of
{ok, IoDevice} ->
ContentType = proplists:get_value("Content-Type",
ExtraHeaders,
mochiweb_util:guess_mime(File)),
Res = ok({ContentType,
[{"last-modified", LastModified}
| ExtraHeaders],
{file, IoDevice}}),
file:close(IoDevice),
Res;
_ ->
not_found(ExtraHeaders)
end
end;
{error, _} ->
not_found(ExtraHeaders)
end
end.
%% Internal API
server_headers() ->
[{"Server", "MochiWeb/1.0 (" ++ ?QUIP ++ ")"},
{"Date", httpd_util:rfc1123_date()}].
make_io(Atom) when is_atom(Atom) ->
atom_to_list(Atom);
make_io(Integer) when is_integer(Integer) ->
integer_to_list(Integer);
make_io(Io) when is_list(Io); is_binary(Io) ->
Io.
make_code(X) when is_integer(X) ->
[integer_to_list(X), [" " | httpd_util:reason_phrase(X)]];
make_code(Io) when is_list(Io); is_binary(Io) ->
Io.
make_version({1, 0}) ->
<<"HTTP/1.0 ">>;
make_version(_) ->
<<"HTTP/1.1 ">>.
iodevice_stream(IoDevice) ->
case file:read(IoDevice, ?READ_SIZE) of
eof ->
ok;
{ok, Data} ->
ok = send(Data),
iodevice_stream(IoDevice)
end.
parts_to_body([{Start, End, Body}], ContentType, Size) ->
%% return body for a range reponse with a single body
HeaderList = [{"Content-Type", ContentType},
{"Content-Range",
["bytes ",
make_io(Start), "-", make_io(End),
"/", make_io(Size)]}],
{HeaderList, Body};
parts_to_body(BodyList, ContentType, Size) when is_list(BodyList) ->
%% return
%% header Content-Type: multipart/byteranges; boundary=441934886133bdee4
%% and multipart body
Boundary = mochihex:to_hex(crypto:rand_bytes(8)),
HeaderList = [{"Content-Type",
["multipart/byteranges; ",
"boundary=", Boundary]}],
MultiPartBody = multipart_body(BodyList, ContentType, Boundary, Size),
{HeaderList, MultiPartBody}.
multipart_body([], _ContentType, Boundary, _Size) ->
["--", Boundary, "--\r\n"];
multipart_body([{Start, End, Body} | BodyList], ContentType, Boundary, Size) ->
["--", Boundary, "\r\n",
"Content-Type: ", ContentType, "\r\n",
"Content-Range: ",
"bytes ", make_io(Start), "-", make_io(End),
"/", make_io(Size), "\r\n\r\n",
Body, "\r\n"
| multipart_body(BodyList, ContentType, Boundary, Size)].
iodevice_size(IoDevice) ->
{ok, Size} = file:position(IoDevice, eof),
{ok, 0} = file:position(IoDevice, bof),
Size.
range_parts({file, IoDevice}, Ranges) ->
Size = iodevice_size(IoDevice),
F = fun (Spec, Acc) ->
case range_skip_length(Spec, Size) of
invalid_range ->
Acc;
V ->
[V | Acc]
end
end,
LocNums = lists:foldr(F, [], Ranges),
{ok, Data} = file:pread(IoDevice, LocNums),
Bodies = lists:zipwith(fun ({Skip, Length}, PartialBody) ->
{Skip, Skip + Length - 1, PartialBody}
end,
LocNums, Data),
{Bodies, Size};
range_parts(Body0, Ranges) ->
Body = iolist_to_binary(Body0),
Size = size(Body),
F = fun(Spec, Acc) ->
case range_skip_length(Spec, Size) of
invalid_range ->
Acc;
{Skip, Length} ->
<<_:Skip/binary, PartialBody:Length/binary, _/binary>> = Body,
[{Skip, Skip + Length - 1, PartialBody} | Acc]
end
end,
{lists:foldr(F, [], Ranges), Size}.
range_skip_length(Spec, Size) ->
case Spec of
{none, R} when R =< Size, R >= 0 ->
{Size - R, R};
{none, _OutOfRange} ->
{0, Size};
{R, none} when R >= 0, R < Size ->
{R, Size - R};
{_OutOfRange, none} ->
invalid_range;
{Start, End} when 0 =< Start, Start =< End, End < Size ->
{Start, End - Start + 1};
{_OutOfRange, _End} ->
invalid_range
end.
parse_range_request(RawRange) when is_list(RawRange) ->
try
"bytes=" ++ RangeString = RawRange,
Ranges = string:tokens(RangeString, ","),
lists:map(fun ("-" ++ V) ->
{none, list_to_integer(V)};
(R) ->
case string:tokens(R, "-") of
[S1, S2] ->
{list_to_integer(S1), list_to_integer(S2)};
[S] ->
{list_to_integer(S), none}
end
end,
Ranges)
catch
_:_ ->
fail
end.
test() ->
ok = test_range(),
ok.
test_range() ->
%% valid, single ranges
io:format("Testing parse_range_request with valid single ranges~n"),
io:format("1"),
[{20, 30}] = parse_range_request("bytes=20-30"),
io:format("2"),
[{20, none}] = parse_range_request("bytes=20-"),
io:format("3"),
[{none, 20}] = parse_range_request("bytes=-20"),
io:format(".. ok ~n"),
%% invalid, single ranges
io:format("Testing parse_range_request with invalid ranges~n"),
io:format("1"),
fail = parse_range_request(""),
io:format("2"),
fail = parse_range_request("garbage"),
io:format("3"),
fail = parse_range_request("bytes=-20-30"),
io:format(".. ok ~n"),
%% valid, multiple range
io:format("Testing parse_range_request with valid multiple ranges~n"),
io:format("1"),
[{20, 30}, {50, 100}, {110, 200}] =
parse_range_request("bytes=20-30,50-100,110-200"),
io:format("2"),
[{20, none}, {50, 100}, {none, 200}] =
parse_range_request("bytes=20-,50-100,-200"),
io:format(".. ok~n"),
%% no ranges
io:format("Testing out parse_range_request with no ranges~n"),
io:format("1"),
[] = parse_range_request("bytes="),
io:format(".. ok~n"),
Body = <<"012345678901234567890123456789012345678901234567890123456789">>,
BodySize = size(Body), %% 60
BodySize = 60,
%% these values assume BodySize =:= 60
io:format("Testing out range_skip_length on valid ranges~n"),
io:format("1"),
{1,9} = range_skip_length({1,9}, BodySize), %% 1-9
io:format("2"),
{10,10} = range_skip_length({10,19}, BodySize), %% 10-19
io:format("3"),
{40, 20} = range_skip_length({none, 20}, BodySize), %% -20
io:format("4"),
{30, 30} = range_skip_length({30, none}, BodySize), %% 30-
io:format(".. ok ~n"),
%% valid edge cases for range_skip_length
io:format("Testing out range_skip_length on valid edge case ranges~n"),
io:format("1"),
{BodySize, 0} = range_skip_length({none, 0}, BodySize),
io:format("2"),
{0, BodySize} = range_skip_length({none, BodySize}, BodySize),
io:format("3"),
{0, BodySize} = range_skip_length({0, none}, BodySize),
BodySizeLess1 = BodySize - 1,
io:format("4"),
{BodySizeLess1, 1} = range_skip_length({BodySize - 1, none}, BodySize),
%% out of range, return whole thing
io:format("5"),
{0, BodySize} = range_skip_length({none, BodySize + 1}, BodySize),
io:format("6"),
{0, BodySize} = range_skip_length({none, -1}, BodySize),
io:format(".. ok ~n"),
%% invalid ranges
io:format("Testing out range_skip_length on invalid ranges~n"),
io:format("1"),
invalid_range = range_skip_length({-1, 30}, BodySize),
io:format("2"),
invalid_range = range_skip_length({0, BodySize + 1}, BodySize),
io:format("3"),
invalid_range = range_skip_length({-1, BodySize + 1}, BodySize),
io:format("4"),
invalid_range = range_skip_length({BodySize, 40}, BodySize),
io:format("5"),
invalid_range = range_skip_length({-1, none}, BodySize),
io:format("6"),
invalid_range = range_skip_length({BodySize, none}, BodySize),
io:format(".. ok ~n"),
ok.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment