From 056e05be581575f1fae2b5fdb9093dd2d4bb9e8c Mon Sep 17 00:00:00 2001 From: Harald Welte Date: Sat, 7 Aug 2010 13:28:45 +0200 Subject: Initial (untested) version of the full UDP<->patching<->IPA chain --- src/osmo_map_masq.erl | 62 +++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 62 insertions(+) create mode 100644 src/osmo_map_masq.erl diff --git a/src/osmo_map_masq.erl b/src/osmo_map_masq.erl new file mode 100644 index 0000000..7034888 --- /dev/null +++ b/src/osmo_map_masq.erl @@ -0,0 +1,62 @@ +% MAP masquerading application + +% (C) 2010 by Harald Welte +% (C) 2010 by On-Waves +% +% All Rights Reserved +% +% This program is free software; you can redistribute it and/or modify +% it under the terms of the GNU General Public License as published by +% the Free Software Foundation; either version 2 of the License, or +% (at your option) any later version. +% +% This program is distributed in the hope that it will be useful, +% but WITHOUT ANY WARRANTY; without even the implied warranty of +% MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +% GNU General Public License for more details. +% +% You should have received a copy of the GNU General Public License along +% with this program; if not, write to the Free Software Foundation, Inc., +% 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + +-module(osmo_map_masq). +-author('Harald Welte '). +-export([init/4]). + +-record(loop_data, + {udpSocket, udpSrvPort, udpRxAddr, udpRxPort, + ipaSocket, ipaStreamID, ipaDstAddr, ipaDstPort}). + + +init(UdpServerPort, IpaDstAddr, IpaDstPort, IpaStreamID) -> + {ok, UdpSock} = gen_udp:open(UdpServerPort, [binary, inet]), + {ok, TcpSock} = ipa_proto:connect(IpaDstAddr, IpaDstPort, []), + % FIXME: result evaluation + ipa_proto:register_stream(TcpSock, IpaStreamID, self()), + loop(#loop_data{udpSocket=UdpSock, udpSrvPort=UdpServerPort, + ipaSocket=TcpSock, ipaStreamID=IpaStreamID, ipaDstAddr=IpaDstAddr, ipaDstPort=IpaDstPort}). + + +loop(#loop_data{}=LoopData) -> + io:format("osmo_masq_loop~n"), + receive + {udp, S, ClientIP, ClientPort, Packet} when S =:= LoopData#loop_data.udpSocket -> + io:format("UDP received from ~p:~p ==> ~p~n", [ClientIP, ClientPort, Packet]), + {ok, SccpHdr, SccpPayload} = sccp_proto:split_udt_sync(Packet), + {ok, NewSccpPayload} = tcap_map_patch:handle_tcap_msg(binary_to_list(SccpPayload)), + {ok, NewSccpBinary} = sccp_proto:wrap_udt_sync(SccpHdr, list_to_binary(NewSccpPayload)), + ipa_proto:send(LoopData#loop_data.ipaSocket, LoopData#loop_data.ipaStreamID, NewSccpBinary), + % loop using the (new?) UDP socket Rx IP and Port number + loop(LoopData#loop_data{udpRxAddr=ClientIP, udpRxPort=ClientPort}); + {ipa, S, StreamID, Message} when S =:= LoopData#loop_data.ipaSocket, + StreamID =:= LoopData#loop_data.ipaStreamID -> + io:format("IPA received (Stream ~p)~n", [LoopData#loop_data.ipaStreamID]), + {ok, SccpHdr, SccpPayload} = sccp_proto:split_udt_sync(Message), + {ok, NewSccpPayload} = tcap_map_patch:handle_tcap_msg(binary_to_list(SccpPayload)), + {ok, NewSccpBinary} = sccp_proto:wrap_udt_sync(SccpHdr, list_to_binary(NewSccpPayload)), + gen_udp:send(LoopData#loop_data.udpSocket, LoopData#loop_data.udpRxAddr, + LoopData#loop_data.udpRxPort, NewSccpBinary), + loop(LoopData); + {ipa_closed, S} when S =:= LoopData#loop_data.ipaSocket -> + io:format("IPA socket was closed, stopping process~n") + end. -- cgit v1.2.3