diff options
author | Erlang/OTP <[email protected]> | 2009-11-20 14:54:40 +0000 |
---|---|---|
committer | Erlang/OTP <[email protected]> | 2009-11-20 14:54:40 +0000 |
commit | 84adefa331c4159d432d22840663c38f155cd4c1 (patch) | |
tree | bff9a9c66adda4df2106dfd0e5c053ab182a12bd /lib/kernel/src/inet_udp.erl | |
download | otp-84adefa331c4159d432d22840663c38f155cd4c1.tar.gz otp-84adefa331c4159d432d22840663c38f155cd4c1.tar.bz2 otp-84adefa331c4159d432d22840663c38f155cd4c1.zip |
The R13B03 release.OTP_R13B03
Diffstat (limited to 'lib/kernel/src/inet_udp.erl')
-rw-r--r-- | lib/kernel/src/inet_udp.erl | 132 |
1 files changed, 132 insertions, 0 deletions
diff --git a/lib/kernel/src/inet_udp.erl b/lib/kernel/src/inet_udp.erl new file mode 100644 index 0000000000..9a4089ab19 --- /dev/null +++ b/lib/kernel/src/inet_udp.erl @@ -0,0 +1,132 @@ +%% +%% %CopyrightBegin% +%% +%% Copyright Ericsson AB 1997-2009. All Rights Reserved. +%% +%% The contents of this file are subject to the Erlang Public License, +%% Version 1.1, (the "License"); you may not use this file except in +%% compliance with the License. You should have received a copy of the +%% Erlang Public License along with this software. If not, it can be +%% retrieved online at http://www.erlang.org/. +%% +%% Software distributed under the License is distributed on an "AS IS" +%% basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See +%% the License for the specific language governing rights and limitations +%% under the License. +%% +%% %CopyrightEnd% +%% +-module(inet_udp). + +-export([open/1, open/2, close/1]). +-export([send/2, send/4, recv/2, recv/3, connect/3]). +-export([controlling_process/2]). +-export([fdopen/2]). + +-export([getserv/1, getaddr/1, getaddr/2]). + +-include("inet_int.hrl"). + +-define(RECBUF, (8*1024)). + + + +%% inet_udp port lookup +getserv(Port) when is_integer(Port) -> {ok, Port}; +getserv(Name) when is_atom(Name) -> inet:getservbyname(Name,udp). + +%% inet_udp address lookup +getaddr(Address) -> inet:getaddr(Address, inet). +getaddr(Address,Timer) -> inet:getaddr_tm(Address, inet, Timer). + +open(Port) -> open(Port, []). + +open(Port, Opts) -> + case inet:udp_options( + [{port,Port}, {recbuf, ?RECBUF} | Opts], + inet) of + {error, Reason} -> exit(Reason); + {ok, #udp_opts{fd=Fd, + ifaddr=BAddr={A,B,C,D}, + port=BPort, + opts=SockOpts}} when ?ip(A,B,C,D), ?port(BPort) -> + inet:open(Fd,BAddr,BPort,SockOpts,udp,inet,?MODULE); + {ok, _} -> exit(badarg) + end. + +send(S,{A,B,C,D},P,Data) when ?ip(A,B,C,D), ?port(P) -> + prim_inet:sendto(S, {A,B,C,D}, P, Data). + +send(S, Data) -> + prim_inet:sendto(S, {0,0,0,0}, 0, Data). + +connect(S, {A,B,C,D}, P) when ?ip(A,B,C,D), ?port(P) -> + prim_inet:connect(S, {A,B,C,D}, P). + +recv(S,Len) -> + prim_inet:recvfrom(S, Len). + +recv(S,Len,Time) -> + prim_inet:recvfrom(S, Len, Time). + +close(S) -> + inet:udp_close(S). + +%% +%% Set controlling process: +%% 1) First sync socket into a known state +%% 2) Move all messages onto the new owners message queue +%% 3) Commit the owner +%% 4) Wait for ack of new Owner (since socket does some link and unlink) +%% + +controlling_process(Socket, NewOwner) -> + inet:udp_controlling_process(Socket, NewOwner). + +%% +%% Create a port/socket from a file descriptor +%% +fdopen(Fd, Opts) -> + inet:fdopen(Fd, + optuniquify([{recbuf, ?RECBUF} | Opts]), + udp, inet, ?MODULE). + + +%% Remove all duplicate options from an option list. +%% The last occurring duplicate is used, and the order is preserved. +%% +%% Here's how: +%% Reverse the list. +%% For each head option go through the tail and remove +%% all occurences of the same option from the tail. +%% Store that head option and iterate using the new tail. +%% Return the list of stored head options. +optuniquify(List) -> + optuniquify(lists:reverse(List), []). + +optuniquify([], Result) -> + Result; +optuniquify([Opt | Tail], Result) -> + %% Remove all occurences of Opt in Tail, + %% prepend Opt to Result, + %% then iterate back here. + optuniquify(Opt, Tail, [], Result). + +%% All duplicates of current option are now removed +optuniquify(Opt, [], Rest, Result) -> + %% Store unique option + optuniquify(lists:reverse(Rest), [Opt | Result]); +%% Duplicate option tuple +optuniquify(Opt0, [Opt1 | Tail], Rest, Result) + when tuple_size(Opt0) =:= tuple_size(Opt1), + element(1, Opt0) =:= element(1, Opt1) -> + %% Waste duplicate + optuniquify(Opt0, Tail, Rest, Result); +%% Duplicate option atom or other term +optuniquify(Opt, [Opt | Tail], Rest, Result) -> + %% Waste duplicate + optuniquify(Opt, Tail, Rest, Result); +%% Non-duplicate option +optuniquify(Opt, [X | Tail], Rest, Result) -> + %% Keep non-duplicate + optuniquify(Opt, Tail, [X | Rest], Result). |