Updated Branches:
  refs/heads/import-master [created] 1167b0e3c

http://git-wip-us.apache.org/repos/asf/couchdb-ibrowse/blob/4d1d8294/ibrowse_lib.erl
----------------------------------------------------------------------
diff --git a/ibrowse_lib.erl b/ibrowse_lib.erl
new file mode 100644
index 0000000..67c5eee
--- /dev/null
+++ b/ibrowse_lib.erl
@@ -0,0 +1,399 @@
+%%% File    : ibrowse_lib.erl
+%%% Author  : Chandrashekhar Mullaparthi 
<chandrashekhar.mullapar...@t-mobile.co.uk>
+%%% Description : 
+%%% Created : 27 Feb 2004 by Chandrashekhar Mullaparthi 
<chandrashekhar.mullapar...@t-mobile.co.uk>
+%% @doc Module with a few useful functions
+
+-module(ibrowse_lib).
+-vsn('$Id: ibrowse_lib.erl,v 1.6 2008/03/27 01:35:50 chandrusf Exp $ ').
+-author('chandru').
+-ifdef(debug).
+-compile(export_all).
+-endif.
+
+-include("ibrowse.hrl").
+
+-export([
+        get_trace_status/2,
+        do_trace/2,
+        do_trace/3,
+        url_encode/1,
+        decode_rfc822_date/1,
+        status_code/1,
+        dec2hex/2,
+        drv_ue/1,
+        drv_ue/2,
+        encode_base64/1,
+        decode_base64/1,
+        get_value/2,
+        get_value/3,
+        parse_url/1,
+        printable_date/0
+       ]).
+
+get_trace_status(Host, Port) ->
+    ibrowse:get_config_value({trace, Host, Port}, false).
+
+drv_ue(Str) ->
+    [{port, Port}| _] = ets:lookup(ibrowse_table, port),
+    drv_ue(Str, Port).
+drv_ue(Str, Port) ->
+    case erlang:port_control(Port, 1, Str) of
+       [] ->
+           Str;
+       Res ->
+           Res
+    end.
+
+%% @doc URL-encodes a string based on RFC 1738. Returns a flat list.
+%% @spec url_encode(Str) -> UrlEncodedStr
+%% Str = string()
+%% UrlEncodedStr = string()
+url_encode(Str) when list(Str) ->
+    url_encode_char(lists:reverse(Str), []).
+
+url_encode_char([X | T], Acc) when X >= $0, X =< $9 ->
+    url_encode_char(T, [X | Acc]);
+url_encode_char([X | T], Acc) when X >= $a, X =< $z ->
+    url_encode_char(T, [X | Acc]);
+url_encode_char([X | T], Acc) when X >= $A, X =< $Z ->
+    url_encode_char(T, [X | Acc]);
+url_encode_char([X | T], Acc) when X == $-; X == $_; X == $. ->
+    url_encode_char(T, [X | Acc]);
+url_encode_char([32 | T], Acc) ->
+    url_encode_char(T, [$+ | Acc]);
+url_encode_char([X | T], Acc) ->
+    url_encode_char(T, [$%, d2h(X bsr 4), d2h(X band 16#0f) | Acc]);
+url_encode_char([], Acc) ->
+    Acc.
+
+d2h(N) when N<10 -> N+$0;
+d2h(N) -> N+$a-10.
+
+decode_rfc822_date(String) when list(String) ->
+    case catch decode_rfc822_date_1(string:tokens(String, ", \t\r\n")) of
+       {'EXIT', _} ->
+           {error, invalid_date};
+       Res ->
+           Res
+    end.
+
+% TODO: Have to handle the Zone
+decode_rfc822_date_1([_,DayInt,Month,Year, Time,Zone]) ->
+    decode_rfc822_date_1([DayInt,Month,Year, Time,Zone]);
+decode_rfc822_date_1([Day,Month,Year, Time,_Zone]) ->
+    DayI = list_to_integer(Day),
+    MonthI = month_int(Month),
+    YearI = list_to_integer(Year),
+    TimeTup = case string:tokens(Time, ":") of
+                 [H,M] ->
+                     {list_to_integer(H),
+                      list_to_integer(M),
+                      0};
+                 [H,M,S] ->
+                     {list_to_integer(H),
+                      list_to_integer(M),
+                      list_to_integer(S)}
+             end,
+    {{YearI,MonthI,DayI}, TimeTup}.
+
+month_int("Jan") -> 1;
+month_int("Feb") -> 2;
+month_int("Mar") -> 3;
+month_int("Apr") -> 4;
+month_int("May") -> 5;
+month_int("Jun") -> 6;
+month_int("Jul") -> 7;
+month_int("Aug") -> 8;
+month_int("Sep") -> 9;
+month_int("Oct") -> 10;
+month_int("Nov") -> 11;
+month_int("Dec") -> 12.
+
+%% @doc Given a status code, returns an atom describing the status code. 
+%% @spec status_code(StatusCode::status_code()) -> StatusDescription
+%% status_code() = string() | integer()
+%% StatusDescription = atom()
+status_code(100) -> continue;
+status_code(101) -> switching_protocols;
+status_code(102) -> processing;
+status_code(200) -> ok;
+status_code(201) -> created;
+status_code(202) -> accepted;
+status_code(203) -> non_authoritative_information;
+status_code(204) -> no_content;
+status_code(205) -> reset_content;
+status_code(206) -> partial_content;
+status_code(207) -> multi_status;
+status_code(300) -> multiple_choices;
+status_code(301) -> moved_permanently;
+status_code(302) -> found;
+status_code(303) -> see_other;
+status_code(304) -> not_modified;
+status_code(305) -> use_proxy;
+status_code(306) -> unused;
+status_code(307) -> temporary_redirect;
+status_code(400) -> bad_request;
+status_code(401) -> unauthorized;
+status_code(402) -> payment_required;
+status_code(403) -> forbidden;
+status_code(404) -> not_found;
+status_code(405) -> method_not_allowed;
+status_code(406) -> not_acceptable;
+status_code(407) -> proxy_authentication_required;
+status_code(408) -> request_timeout;
+status_code(409) -> conflict;
+status_code(410) -> gone;
+status_code(411) -> length_required;
+status_code(412) -> precondition_failed;
+status_code(413) -> request_entity_too_large;
+status_code(414) -> request_uri_too_long;
+status_code(415) -> unsupported_media_type;
+status_code(416) -> requested_range_not_satisfiable;
+status_code(417) -> expectation_failed;
+status_code(422) -> unprocessable_entity;
+status_code(423) -> locked;
+status_code(424) -> failed_dependency;
+status_code(500) -> internal_server_error;
+status_code(501) -> not_implemented;
+status_code(502) -> bad_gateway;
+status_code(503) -> service_unavailable;
+status_code(504) -> gateway_timeout;
+status_code(505) -> http_version_not_supported;
+status_code(507) -> insufficient_storage;
+status_code(X) when is_list(X) -> status_code(list_to_integer(X));
+status_code(_)   -> unknown_status_code.
+
+%% @doc dec2hex taken from gtk.erl in std dist
+%% M = integer() -- number of hex digits required
+%% N = integer() -- the number to represent as hex
+%% @spec dec2hex(M::integer(), N::integer()) -> string()
+dec2hex(M,N) -> dec2hex(M,N,[]).
+
+dec2hex(0,_N,Ack) -> Ack;
+dec2hex(M,N,Ack) -> dec2hex(M-1,N bsr 4,[d2h(N band 15)|Ack]).
+
+%% @doc Implements the base64 encoding algorithm. The output data type matches 
in the input data type.
+%% @spec encode_base64(In) -> Out
+%% In = string() | binary()
+%% Out = string() | binary()
+encode_base64(List) when list(List) ->
+    encode_base64_1(list_to_binary(List));
+encode_base64(Bin) when binary(Bin) ->
+    List = encode_base64_1(Bin),
+    list_to_binary(List).
+
+encode_base64_1(<<A:6, B:6, C:6, D:6, Rest/binary>>) ->
+    [int_to_b64(A), int_to_b64(B),
+     int_to_b64(C), int_to_b64(D) | encode_base64_1(Rest)];
+encode_base64_1(<<A:6, B:6, C:4>>) ->
+    [int_to_b64(A), int_to_b64(B), int_to_b64(C bsl 2), $=];
+encode_base64_1(<<A:6, B:2>>) ->
+    [int_to_b64(A), int_to_b64(B bsl 4), $=, $=];
+encode_base64_1(<<>>) ->
+    [].
+
+%% @doc Implements the base64 decoding algorithm. The output data type matches 
in the input data type.
+%% @spec decode_base64(In) -> Out | exit({error, invalid_input})
+%% In = string() | binary()
+%% Out = string() | binary()
+decode_base64(List) when list(List) ->
+    decode_base64_1(List, []);
+decode_base64(Bin) when binary(Bin) ->
+    List = decode_base64_1(binary_to_list(Bin), []),
+    list_to_binary(List).
+
+decode_base64_1([H | T], Acc) when ((H == $\t) or
+                                   (H == 32) or
+                                   (H == $\r) or
+                                   (H == $\n)) ->
+    decode_base64_1(T, Acc);
+
+decode_base64_1([$=, $=], Acc) ->
+    lists:reverse(Acc);
+decode_base64_1([$=, _ | _], _Acc) ->
+    exit({error, invalid_input});
+
+decode_base64_1([A1, B1, $=, $=], Acc) ->
+    A = b64_to_int(A1),
+    B = b64_to_int(B1),
+    Oct1 = (A bsl 2) bor (B bsr 4),
+    decode_base64_1([], [Oct1 | Acc]);
+decode_base64_1([A1, B1, C1, $=], Acc) ->
+    A = b64_to_int(A1),
+    B = b64_to_int(B1),
+    C = b64_to_int(C1),
+    Oct1 = (A bsl 2) bor (B bsr 4),
+    Oct2 = ((B band 16#f) bsl 6) bor (C bsr 2),
+    decode_base64_1([], [Oct2, Oct1 | Acc]);
+decode_base64_1([A1, B1, C1, D1 | T], Acc) ->
+    A = b64_to_int(A1),
+    B = b64_to_int(B1),
+    C = b64_to_int(C1),
+    D = b64_to_int(D1),
+    Oct1 = (A bsl 2) bor (B bsr 4),
+    Oct2 = ((B band 16#f) bsl 4) bor (C bsr 2),
+    Oct3 = ((C band 2#11) bsl 6) bor D,
+    decode_base64_1(T, [Oct3, Oct2, Oct1 | Acc]);
+decode_base64_1([], Acc) ->
+    lists:reverse(Acc).
+
+%% Taken from httpd_util.erl
+int_to_b64(X) when X >= 0, X =< 25 -> X + $A;
+int_to_b64(X) when X >= 26, X =< 51 -> X - 26 + $a;
+int_to_b64(X) when X >= 52, X =< 61 -> X - 52 + $0;
+int_to_b64(62) -> $+;
+int_to_b64(63) -> $/.
+
+%% Taken from httpd_util.erl
+b64_to_int(X) when X >= $A, X =< $Z -> X - $A;
+b64_to_int(X) when X >= $a, X =< $z -> X - $a + 26;
+b64_to_int(X) when X >= $0, X =< $9 -> X - $0 + 52;
+b64_to_int($+) -> 62;
+b64_to_int($/) -> 63.
+
+get_value(Tag, TVL, DefVal) ->
+    case lists:keysearch(Tag, 1, TVL) of
+       false ->
+           DefVal;
+       {value, {_, Val}} ->
+           Val
+    end.
+
+get_value(Tag, TVL) ->
+    {value, {_, V}} = lists:keysearch(Tag,1,TVL),
+    V.
+
+parse_url(Url) ->
+    parse_url(Url, get_protocol, #url{abspath=Url}, []).
+
+parse_url([$:, $/, $/ | _], get_protocol, Url, []) ->
+    {invalid_uri_1, Url};
+parse_url([$:, $/, $/ | T], get_protocol, Url, TmpAcc) ->
+    Prot = list_to_atom(lists:reverse(TmpAcc)),
+    parse_url(T, get_username, 
+             Url#url{protocol = Prot},
+             []);
+parse_url([$/ | T], get_username, Url, TmpAcc) ->
+    %% No username/password. No  port number
+    Url#url{host = lists:reverse(TmpAcc),
+           port = default_port(Url#url.protocol),
+           path = [$/ | T]};
+parse_url([$: | T], get_username, Url, TmpAcc) ->
+    %% It is possible that no username/password has been
+    %% specified. But we'll continue with the assumption that there is
+    %% a username/password. If we encounter a '@' later on, there is a
+    %% username/password indeed. If we encounter a '/', it was
+    %% actually the hostname
+    parse_url(T, get_password, 
+             Url#url{username = lists:reverse(TmpAcc)},
+             []);
+parse_url([$@ | T], get_username, Url, TmpAcc) ->
+    parse_url(T, get_host, 
+             Url#url{username = lists:reverse(TmpAcc),
+                     password = ""},
+             []);
+parse_url([$@ | T], get_password, Url, TmpAcc) ->
+    parse_url(T, get_host, 
+             Url#url{password = lists:reverse(TmpAcc)},
+             []);
+parse_url([$/ | T], get_password, Url, TmpAcc) ->
+    %% Ok, what we thought was the username/password was the hostname
+    %% and portnumber
+    #url{username=User} = Url,
+    Port = list_to_integer(lists:reverse(TmpAcc)),
+    Url#url{host = User,
+           port = Port,
+           username = undefined,
+           password = undefined,
+           path = [$/ | T]};
+parse_url([$: | T], get_host, #url{} = Url, TmpAcc) ->
+    parse_url(T, get_port, 
+             Url#url{host = lists:reverse(TmpAcc)},
+             []);
+parse_url([$/ | T], get_host, #url{protocol=Prot} = Url, TmpAcc) ->
+    Url#url{host = lists:reverse(TmpAcc),
+           port = default_port(Prot),
+           path = [$/ | T]};
+parse_url([$/ | T], get_port, #url{protocol=Prot} = Url, TmpAcc) ->
+    Port = case TmpAcc of
+              [] ->
+                  default_port(Prot);
+              _ ->
+                  list_to_integer(lists:reverse(TmpAcc))
+          end,
+    Url#url{port = Port, path = [$/ | T]};
+parse_url([H | T], State, Url, TmpAcc) ->
+    parse_url(T, State, Url, [H | TmpAcc]);
+parse_url([], get_host, Url, TmpAcc) when TmpAcc /= [] ->
+    Url#url{host = lists:reverse(TmpAcc),
+           port = default_port(Url#url.protocol),
+           path = "/"};
+parse_url([], get_username, Url, TmpAcc) when TmpAcc /= [] ->
+    Url#url{host = lists:reverse(TmpAcc),
+           port = default_port(Url#url.protocol),
+           path = "/"};
+parse_url([], get_port, #url{protocol=Prot} = Url, TmpAcc) ->
+    Port = case TmpAcc of
+              [] ->
+                  default_port(Prot);
+              _ ->
+                  list_to_integer(lists:reverse(TmpAcc))
+          end,
+    Url#url{port = Port, 
+           path = "/"};
+parse_url([], get_password, Url, TmpAcc) ->
+    %% Ok, what we thought was the username/password was the hostname
+    %% and portnumber
+    #url{username=User} = Url,
+    Port = case TmpAcc of
+              [] ->
+                  default_port(Url#url.protocol);
+              _ ->
+                  list_to_integer(lists:reverse(TmpAcc))
+          end,
+    Url#url{host = User,
+           port = Port,
+           username = undefined,
+           password = undefined,
+           path = "/"};
+parse_url([], State, Url, TmpAcc) ->
+    {invalid_uri_2, State, Url, TmpAcc}.
+
+default_port(http)  -> 80;
+default_port(https) -> 443;
+default_port(ftp)   -> 21.
+
+printable_date() ->
+    {{Y,Mo,D},{H, M, S}} = calendar:local_time(),
+    {_,_,MicroSecs} = now(),
+    [integer_to_list(Y),
+     $-,
+     integer_to_list(Mo),
+     $-,
+     integer_to_list(D),
+     $_,
+     integer_to_list(H),
+     $:,
+     integer_to_list(M),
+     $:,
+     integer_to_list(S),
+     $:,
+     integer_to_list(MicroSecs div 1000)].
+
+do_trace(Fmt, Args) ->
+    do_trace(get(my_trace_flag), Fmt, Args).
+
+-ifdef(DEBUG).
+do_trace(_, Fmt, Args) ->
+    io:format("~s -- (~s) - "++Fmt,
+             [printable_date(), 
+              get(ibrowse_trace_token) | Args]).
+-else.
+do_trace(true, Fmt, Args) ->
+    io:format("~s -- (~s) - "++Fmt,
+             [printable_date(), 
+              get(ibrowse_trace_token) | Args]);
+do_trace(_, _, _) ->
+    ok.
+-endif.

http://git-wip-us.apache.org/repos/asf/couchdb-ibrowse/blob/4d1d8294/ibrowse_sup.erl
----------------------------------------------------------------------
diff --git a/ibrowse_sup.erl b/ibrowse_sup.erl
new file mode 100644
index 0000000..300435d
--- /dev/null
+++ b/ibrowse_sup.erl
@@ -0,0 +1,65 @@
+%%%-------------------------------------------------------------------
+%%% File    : ibrowse_sup.erl
+%%% Author  : Chandrashekhar Mullaparthi 
<chandrashekhar.mullapar...@t-mobile.co.uk>
+%%% Description : 
+%%%
+%%% Created : 15 Oct 2003 by Chandrashekhar Mullaparthi 
<chandrashekhar.mullapar...@t-mobile.co.uk>
+%%%-------------------------------------------------------------------
+-module(ibrowse_sup).
+-vsn('$Id: ibrowse_sup.erl,v 1.1 2005/05/05 22:28:28 chandrusf Exp $ ').
+
+-behaviour(supervisor).
+%%--------------------------------------------------------------------
+%% Include files
+%%--------------------------------------------------------------------
+
+%%--------------------------------------------------------------------
+%% External exports
+%%--------------------------------------------------------------------
+-export([
+        start_link/0
+        ]).
+
+%%--------------------------------------------------------------------
+%% Internal exports
+%%--------------------------------------------------------------------
+-export([
+        init/1
+        ]).
+
+%%--------------------------------------------------------------------
+%% Macros
+%%--------------------------------------------------------------------
+-define(SERVER, ?MODULE).
+
+%%--------------------------------------------------------------------
+%% Records
+%%--------------------------------------------------------------------
+
+%%====================================================================
+%% External functions
+%%====================================================================
+%%--------------------------------------------------------------------
+%% Function: start_link/0
+%% Description: Starts the supervisor
+%%--------------------------------------------------------------------
+start_link() ->
+    supervisor:start_link({local, ?SERVER}, ?MODULE, []).
+
+%%====================================================================
+%% Server functions
+%%====================================================================
+%%--------------------------------------------------------------------
+%% Func: init/1
+%% Returns: {ok,  {SupFlags,  [ChildSpec]}} |
+%%          ignore                          |
+%%          {error, Reason}   
+%%--------------------------------------------------------------------
+init([]) ->
+    AChild = {ibrowse,{ibrowse,start_link,[]},
+             permanent,2000,worker,[ibrowse, ibrowse_http_client]},
+    {ok,{{one_for_all,10,1}, [AChild]}}.
+
+%%====================================================================
+%% Internal functions
+%%====================================================================

http://git-wip-us.apache.org/repos/asf/couchdb-ibrowse/blob/4d1d8294/ibrowse_test.erl
----------------------------------------------------------------------
diff --git a/ibrowse_test.erl b/ibrowse_test.erl
new file mode 100644
index 0000000..b4429c9
--- /dev/null
+++ b/ibrowse_test.erl
@@ -0,0 +1,226 @@
+%%% File    : ibrowse_test.erl
+%%% Author  : Chandrashekhar Mullaparthi 
<chandrashekhar.mullapar...@t-mobile.co.uk>
+%%% Description : Test ibrowse
+%%% Created : 14 Oct 2003 by Chandrashekhar Mullaparthi 
<chandrashekhar.mullapar...@t-mobile.co.uk>
+
+-module(ibrowse_test).
+-vsn('$Id: ibrowse_test.erl,v 1.3 2008/05/21 15:28:11 chandrusf Exp $ ').
+-export([
+        load_test/3,
+        send_reqs_1/3,
+        do_send_req/2,
+        unit_tests/0,
+        unit_tests/1,
+        drv_ue_test/0,
+        drv_ue_test/1,
+        ue_test/0,
+        ue_test/1
+       ]).
+
+-import(ibrowse_lib, [printable_date/0]).
+
+%% Use ibrowse:set_max_sessions/3 and ibrowse:set_max_pipeline_size/3 to
+%% tweak settings before running the load test. The defaults are 10 and 10.
+load_test(Url, NumWorkers, NumReqsPerWorker) when is_list(Url),
+                                                  is_integer(NumWorkers),
+                                                  is_integer(NumReqsPerWorker),
+                                                  NumWorkers > 0,
+                                                  NumReqsPerWorker > 0 ->
+    proc_lib:spawn(?MODULE, send_reqs_1, [Url, NumWorkers, NumReqsPerWorker]).
+
+send_reqs_1(Url, NumWorkers, NumReqsPerWorker) ->
+    Start_time = now(),
+    ets:new(pid_table, [named_table, public]),
+    ets:new(ibrowse_test_results, [named_table, public]),
+    ets:new(ibrowse_errors, [named_table, public, ordered_set]),
+    init_results(),
+    process_flag(trap_exit, true),
+    log_msg("Starting spawning of workers...~n", []),
+    spawn_workers(Url, NumWorkers, NumReqsPerWorker),
+    log_msg("Finished spawning workers...~n", []),
+    do_wait(),
+    End_time = now(),
+    log_msg("All workers are done...~n", []),
+    log_msg("ibrowse_test_results table: ~n~p~n", 
[ets:tab2list(ibrowse_test_results)]),
+    log_msg("Start time: ~1000.p~n", [calendar:now_to_local_time(Start_time)]),
+    log_msg("End time  : ~1000.p~n", [calendar:now_to_local_time(End_time)]),
+    Elapsed_time_secs = trunc(timer:now_diff(End_time, Start_time) / 1000000),
+    log_msg("Elapsed   : ~p~n", [Elapsed_time_secs]),
+    log_msg("Reqs/sec  : ~p~n", [(NumWorkers*NumReqsPerWorker) / 
Elapsed_time_secs]),
+    dump_errors().
+
+init_results() ->
+    ets:insert(ibrowse_test_results, {crash, 0}),
+    ets:insert(ibrowse_test_results, {send_failed, 0}),
+    ets:insert(ibrowse_test_results, {other_error, 0}),
+    ets:insert(ibrowse_test_results, {success, 0}),
+    ets:insert(ibrowse_test_results, {retry_later, 0}),
+    ets:insert(ibrowse_test_results, {trid_mismatch, 0}),
+    ets:insert(ibrowse_test_results, {success_no_trid, 0}),
+    ets:insert(ibrowse_test_results, {failed, 0}),
+    ets:insert(ibrowse_test_results, {timeout, 0}),
+    ets:insert(ibrowse_test_results, {req_id, 0}).
+
+spawn_workers(_Url, 0, _) ->
+    ok;
+spawn_workers(Url, NumWorkers, NumReqsPerWorker) ->
+    Pid = proc_lib:spawn_link(?MODULE, do_send_req, [Url, NumReqsPerWorker]),
+    ets:insert(pid_table, {Pid, []}),
+    spawn_workers(Url, NumWorkers - 1, NumReqsPerWorker).
+
+do_wait() ->
+    receive
+       {'EXIT', _, normal} ->
+           do_wait();
+       {'EXIT', Pid, Reason} ->
+           ets:delete(pid_table, Pid),
+           ets:insert(ibrowse_errors, {Pid, Reason}),
+           ets:update_counter(ibrowse_test_results, crash, 1),
+           do_wait();
+       Msg ->
+           io:format("Recvd unknown message...~p~n", [Msg]),
+           do_wait()
+    after 1000 ->
+           case ets:info(pid_table, size) of
+               0 ->
+                   done;
+               _ ->
+                   do_wait()
+           end
+    end.
+                    
+do_send_req(Url, NumReqs) ->
+    do_send_req_1(Url, NumReqs).
+
+do_send_req_1(_Url, 0) ->
+    ets:delete(pid_table, self());
+do_send_req_1(Url, NumReqs) ->
+    Counter = integer_to_list(ets:update_counter(ibrowse_test_results, req_id, 
1)),
+    case ibrowse:send_req(Url, [{"ib_req_id", Counter}], get, [], [], 10000) of
+       {ok, _Status, Headers, _Body} ->
+           case lists:keysearch("ib_req_id", 1, Headers) of
+               {value, {_, Counter}} ->
+                   ets:update_counter(ibrowse_test_results, success, 1);
+               {value, _} ->
+                   ets:update_counter(ibrowse_test_results, trid_mismatch, 1);
+               false ->
+                   ets:update_counter(ibrowse_test_results, success_no_trid, 1)
+           end;
+       {error, req_timedout} ->
+           ets:update_counter(ibrowse_test_results, timeout, 1);
+       {error, send_failed} ->
+           ets:update_counter(ibrowse_test_results, send_failed, 1);
+       {error, retry_later} ->
+           ets:update_counter(ibrowse_test_results, retry_later, 1);
+       Err ->
+           ets:insert(ibrowse_errors, {now(), Err}),
+           ets:update_counter(ibrowse_test_results, other_error, 1),
+           ok
+    end,
+    do_send_req_1(Url, NumReqs-1).
+
+dump_errors() ->
+    case ets:info(ibrowse_errors, size) of
+       0 ->
+           ok;
+       _ ->
+           {A, B, C} = now(),
+           Filename = lists:flatten(
+                        io_lib:format("ibrowse_errors_~p_~p_~p.txt" , [A, B, 
C])),
+           case file:open(Filename, [write, delayed_write, raw]) of
+               {ok, Iod} ->
+                   dump_errors(ets:first(ibrowse_errors), Iod);
+               Err ->
+                   io:format("failed to create file ~s. Reason: ~p~n", 
[Filename, Err]),
+                   ok
+           end
+    end.
+
+dump_errors('$end_of_table', Iod) ->
+    file:close(Iod);
+dump_errors(Key, Iod) ->
+    [{_, Term}] = ets:lookup(ibrowse_errors, Key),
+    file:write(Iod, io_lib:format("~p~n", [Term])),
+    dump_errors(ets:next(ibrowse_errors, Key), Iod).
+
+%%------------------------------------------------------------------------------
+%% Unit Tests
+%%------------------------------------------------------------------------------
+-define(TEST_LIST, [{"http://intranet/messenger";, get},
+                   {"http://www.google.co.uk";, get},
+                   {"http://www.google.com";, get},
+                   {"http://www.google.com";, options}, 
+                   {"http://www.sun.com";, get},
+                   {"http://www.oracle.com";, get},
+                   {"http://www.bbc.co.uk";, get},
+                   {"http://www.bbc.co.uk";, trace},
+                   {"http://www.bbc.co.uk";, options},
+                   {"http://yaws.hyber.org";, get},
+                   {"http://jigsaw.w3.org/HTTP/ChunkedScript";, get},
+                   {"http://jigsaw.w3.org/HTTP/TE/foo.txt";, get},
+                   {"http://jigsaw.w3.org/HTTP/TE/bar.txt";, get},
+                   {"http://jigsaw.w3.org/HTTP/connection.html";, get},
+                   {"http://jigsaw.w3.org/HTTP/cc.html";, get},
+                   {"http://jigsaw.w3.org/HTTP/cc-private.html";, get},
+                   {"http://jigsaw.w3.org/HTTP/cc-proxy-revalidate.html";, get},
+                   {"http://jigsaw.w3.org/HTTP/cc-nocache.html";, get},
+                   {"http://jigsaw.w3.org/HTTP/h-content-md5.html";, get},
+                   {"http://jigsaw.w3.org/HTTP/h-retry-after.html";, get},
+                   {"http://jigsaw.w3.org/HTTP/h-retry-after-date.html";, get},
+                   {"http://jigsaw.w3.org/HTTP/neg";, get},
+                   {"http://jigsaw.w3.org/HTTP/negbad";, get},
+                   {"http://jigsaw.w3.org/HTTP/400/toolong/";, get},
+                   {"http://jigsaw.w3.org/HTTP/300/";, get},
+                   {"http://jigsaw.w3.org/HTTP/Basic/";, get, [{basic_auth, 
{"guest", "guest"}}]},
+                   {"http://jigsaw.w3.org/HTTP/CL/";, get}
+                  ]).
+
+unit_tests() ->
+    unit_tests([]).
+
+unit_tests(Options) ->
+    lists:foreach(fun({Url, Method}) ->
+                         execute_req(Url, Method, Options);
+                    ({Url, Method, X_Opts}) ->
+                         execute_req(Url, Method, X_Opts ++ Options)
+                 end, ?TEST_LIST).
+
+execute_req(Url, Method) ->
+    execute_req(Url, Method, []).
+
+execute_req(Url, Method, Options) ->
+    io:format("~s, ~p: ", [Url, Method]),
+    Result = (catch ibrowse:send_req(Url, [], Method, [], Options)),
+    case Result of 
+       {ok, SCode, _H, _B} ->
+           io:format("Status code: ~p~n", [SCode]);
+       Err ->
+           io:format("Err -> ~p~n", [Err])
+    end.
+
+drv_ue_test() ->
+    drv_ue_test(lists:duplicate(1024, 127)).
+drv_ue_test(Data) ->
+    [{port, Port}| _] = ets:lookup(ibrowse_table, port),
+%     erl_ddll:unload_driver("ibrowse_drv"),
+%     timer:sleep(1000),
+%     erl_ddll:load_driver("../priv", "ibrowse_drv"),
+%     Port = open_port({spawn, "ibrowse_drv"}, []),
+    {Time, Res} = timer:tc(ibrowse_lib, drv_ue, [Data, Port]),
+    io:format("Time -> ~p~n", [Time]),
+    io:format("Data Length -> ~p~n", [length(Data)]),
+    io:format("Res Length -> ~p~n", [length(Res)]).
+%    io:format("Result -> ~s~n", [Res]).
+
+ue_test() ->
+    ue_test(lists:duplicate(1024, $?)).
+ue_test(Data) ->
+    {Time, Res} = timer:tc(ibrowse_lib, url_encode, [Data]),
+    io:format("Time -> ~p~n", [Time]),
+    io:format("Data Length -> ~p~n", [length(Data)]),
+    io:format("Res Length -> ~p~n", [length(Res)]).
+%    io:format("Result -> ~s~n", [Res]).
+
+log_msg(Fmt, Args) ->
+    io:format("~s -- " ++ Fmt,
+             [ibrowse_lib:printable_date() | Args]).

Reply via email to