echo: pass in the protocol family to gen_*

msantos · May 24, 2011 · c4fb02e · c4fb02e
1 parent 5407f2b
commit c4fb02e
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/examples/echo.erl b/examples/echo.erl
@@ -44,15 +44,16 @@ start(udp) ->
     start(?PORT, [{protocol, udp}, {family, inet}, {type, dgram}]).
 start(Port, Options) ->
     Protocol = proplists:get_value(protocol, Options, tcp),
+    Family = proplists:get_value(family, Options, inet),
     {ok, Fd} = procket:open(Port, Options),
     io:format("Listening on: ~p/~p~n", [Port, Protocol]),
-    listen(Protocol, Fd, Port).
+    listen(Protocol, Family, Fd, Port).
 
-listen(tcp, Fd, Port) ->
+listen(tcp, Family, Fd, Port) ->
-    {ok, S} = gen_tcp:listen(Port, [binary, {fd, Fd}]),
+    {ok, S} = gen_tcp:listen(Port, [binary, Family, {fd, Fd}]),
     accept(S);
-listen(udp, Fd, Port) ->
+listen(udp, Family, Fd, Port) ->
-    {ok, S} = gen_udp:open(Port, [binary, {fd, Fd}]),
+    {ok, S} = gen_udp:open(Port, [binary, Family, {fd, Fd}]),
     recv(S).
 
 accept(LS) ->