From 6deda594a6250d29f472e39bf1b0b1021b0f90c3 Mon Sep 17 00:00:00 2001
From: Harald Welte <laforge@gnumonks.org>
Date: Mon, 28 May 2012 23:58:42 +0200
Subject: add more tcap related helper functions

---
 src/tcap_helper.erl | 45 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)

(limited to 'src/tcap_helper.erl')

diff --git a/src/tcap_helper.erl b/src/tcap_helper.erl
index 7de068b..8052895 100644
--- a/src/tcap_helper.erl
+++ b/src/tcap_helper.erl
@@ -43,6 +43,9 @@
 	 build_inv_begin/5, build_retresl_end/4
  	]).
 
+-export([get_tcap_components/1, get_tcap_operation/1, get_tcap_operations/1,
+	 check_for_tcap_op/3]).
+
 format_id(undefined) ->
 	asn1_NOVALUE;
 format_id(Int) when is_integer(Int) ->
@@ -133,3 +136,45 @@ build_inv_begin(DlgId, InvId, ACname, Opcode, Argument) ->
 build_retresl_end(Dtid, InvId, ACname, Argument) ->
 	C = build_retreslast_comp(InvId, Argument),
 	build_end(Dtid, ACname, [C]).
+
+% get a list of components from the decoded TCAP+MAP nested record
+get_tcap_components({'begin', Beg}) ->
+	get_tcap_components(Beg);
+get_tcap_components({'end', Beg}) ->
+	get_tcap_components(Beg);
+get_tcap_components({'continue', Beg}) ->
+	get_tcap_components(Beg);
+% tcap_asn.erl
+get_tcap_components(#'Begin'{components=Comps}) ->
+	Comps;
+get_tcap_components(#'Continue'{components=Comps}) ->
+	Comps;
+get_tcap_components(#'End'{components=Comps}) ->
+	Comps;
+get_tcap_components(_) ->
+	[].
+
+% get the MAP operation of a specific component
+get_tcap_operation({basicROS, Rec}) ->
+	get_tcap_operation(Rec);
+get_tcap_operation({invoke, Rec}) ->
+	get_tcap_operation(Rec);
+get_tcap_operation({returnResult, Rec}) ->
+	get_tcap_operation(Rec);
+get_tcap_operation({returnResultNotLast, Rec}) ->
+	get_tcap_operation(Rec);
+get_tcap_operation(#'Invoke'{opcode=Op}) ->
+	{invoke, Op};
+get_tcap_operation(#'ReturnResult'{result=Res}) ->
+	{returnResult, Res#'ReturnResult_result'.opcode}.
+
+% get a list of the MAP operations inside the components of a MAP message
+get_tcap_operations(MapDec) ->
+	Comps = get_tcap_components(MapDec),
+	[get_tcap_operation(X) || X <- Comps].
+
+
+check_for_tcap_op(Comp, Op, MapDec) ->
+	MapOps = get_tcap_operations(MapDec),
+	% check for invoke of SRI-for-SM:
+	lists:member({Comp, Op}, MapOps).
-- 
cgit v1.2.3