The R13B03 release.OTP_R13B03

1 files changed, 188 insertions, 0 deletions

diff --git a/lib/megaco/test/megaco_test_deliver.erl b/lib/megaco/test/megaco_test_deliver.erl
new file mode 100644
index 0000000000..2d0f0c1cbe
--- /dev/null
+++ b/lib/megaco/test/megaco_test_deliver.erl
@@ -0,0 +1,188 @@
+%%
+%% %CopyrightBegin%
+%% 
+%% Copyright Ericsson AB 2007-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%
+%%
+
+%%
+%%----------------------------------------------------------------------
+%% Purpose: This module takes the role of the main megaco module for
+%%          the transport module. It is used when  delivering 
+%%          received messages. The purpose is to be able to do
+%%          various forms of filtering before passing the message 
+%%          the the megaco stack (by calling the megaco module).
+%%          It can be controlled with the following flags:
+%%          allow_recv_message - Shall the received message be 
+%%                               delivered.
+%%          extra_transport_info - Provide extra info from the transport
+%%                                 module.
+%%----------------------------------------------------------------------
+-module(megaco_test_deliver).
+
+
+%%----------------------------------------------------------------------
+%% Include files
+%%----------------------------------------------------------------------
+-include_lib("megaco/src/udp/megaco_udp.hrl").
+-include("megaco_test_lib.hrl").
+
+
+%%----------------------------------------------------------------------
+%% External exports
+%%----------------------------------------------------------------------
+-export([
+	 process_received_message/4, process_received_message/5,
+	 receive_message/4, receive_message/5
+	]).
+
+
+
+%%======================================================================
+%% External functions
+%%======================================================================
+
+process_received_message(ReceiveHandle, ControlPid, SendHandle, BinMsg) ->
+    i("process_received_message -> entry with"
+      "~n   ReceiveHandle: ~p"
+      "~n   ControlPid:    ~p"
+      "~n   SendHandle:    ~p", 
+      [ReceiveHandle, ControlPid, SendHandle]),
+    case allow_recv_message() of
+	true ->
+	    i("process_received_message -> allowed recv msg"),
+	    case extra_transport_info() of
+		{value, Extra} ->
+		    i("process_received_message -> extra_transport_info: "
+		      "~n   Extra: ~p", [Extra]),
+		    megaco:process_received_message(ReceiveHandle, ControlPid, 
+						    SendHandle, BinMsg,
+						    Extra);
+		_ ->
+		    i("process_received_message -> no extra_transport_info"),
+		    megaco:process_received_message(ReceiveHandle, ControlPid, 
+						    SendHandle, BinMsg)
+	    end;
+	false ->
+	    i("process_received_message -> recv msg not allowed"),
+	    ok;
+	{false, Reason} ->
+	    i("process_received_message -> recv msg not allowed"
+	      "~n   Reason: ~p", [Reason]),
+	    exit(Reason)
+    end.
+	    
+
+process_received_message(ReceiveHandle, ControlPid, SendHandle, BinMsg, Extra) ->
+    i("process_received_message -> entry with"
+      "~n   ReceiveHandle: ~p"
+      "~n   ControlPid:    ~p"
+      "~n   SendHandle:    ~p"
+      "~n   Extra:         ~p", 
+      [ReceiveHandle, ControlPid, SendHandle, Extra]),
+    case allow_recv_message() of
+	true ->
+	    i("process_received_message -> allowed recv msg"),
+	    megaco:process_received_message(ReceiveHandle, ControlPid, 
+					    SendHandle, BinMsg,
+					    Extra);
+	false ->
+	    i("process_received_message -> recv msg not allowed"),
+	    ok;
+	{false, Reason} ->
+	    i("process_received_message -> recv msg not allowed"
+	      "~n   Reason: ~p", [Reason]),
+	    exit(Reason)
+    end.
+
+receive_message(ReceiveHandle, ControlPid, SendHandle, BinMsg) ->
+    i("receive_message -> entry with"
+      "~n   ReceiveHandle: ~p"
+      "~n   ControlPid:    ~p"
+      "~n   SendHandle:    ~p", 
+      [ReceiveHandle, ControlPid, SendHandle]),
+    case allow_recv_message() of
+	true ->
+	    i("receive_message -> allowed recv msg"),
+	    case extra_transport_info() of
+		{value, Extra} ->
+		    i("receive_message -> extra_transport_info: "
+		      "~n   Extra: ~p", [Extra]),
+		    megaco:receive_message(ReceiveHandle, ControlPid,
+					   SendHandle, BinMsg, Extra);
+		_ ->
+		    i("receive_message -> no extra_transport_info"),
+		    megaco:receive_message(ReceiveHandle, ControlPid, 
+					   SendHandle, BinMsg)
+	    end;
+	false ->
+	    i("receive_message -> recv msg not allowed"),
+	    ok;
+	{false, Reason} ->
+	    i("receive_message -> recv msg not allowed"
+	      "~n   Reason: ~p", [Reason]),
+	    exit(Reason)
+    end.
+
+receive_message(ReceiveHandle, ControlPid, SendHandle, BinMsg, Extra) ->
+    i("receive_message -> entry with"
+      "~n   ReceiveHandle: ~p"
+      "~n   ControlPid:    ~p"
+      "~n   SendHandle:    ~p"
+      "~n   Extra:         ~p", 
+      [ReceiveHandle, ControlPid, SendHandle, Extra]),
+    case allow_recv_message() of
+	true ->
+	    i("receive_message -> allowed recv msg"),
+	    megaco:receive_message(ReceiveHandle, ControlPid,
+				   SendHandle, BinMsg,
+				   Extra);
+	false ->
+	    i("receive_message -> recv msg not allowed"),
+	    ok;
+	{false, Reason} ->
+	    i("receive_message -> recv msg not allowed"
+	      "~n   Reason: ~p", [Reason]),
+	    exit(Reason)
+    end.
+
+
+%%======================================================================
+%% Internal functions
+%%======================================================================
+
+allow_recv_message() ->
+    case megaco_tc_controller:lookup(allow_recv_message) of
+	{error, _} ->
+	    true;
+	{value, Else} ->
+	    Else;
+	false ->
+	    true
+    end.
+
+extra_transport_info() ->
+    case megaco_tc_controller:lookup(extra_transport_info) of
+	{error, _} ->
+	    false;
+	Else ->
+	    Else
+    end.
+    
+i(F) ->
+    i(F, []).
+
+i(F, A) ->
+    io:format("~p ~w:" ++ F ++ "~n", [self(), ?MODULE | A]).