Add IMEISV to NASREQ

travelping · May 16, 2024 · aeb5b34 · aeb5b34
1 parent 0ede810
commit aeb5b34
Show file tree

Hide file tree

Showing 4 changed files with 35 additions and 23 deletions.
diff --git a/dia/diameter_3gpp_ts29_061_sgi.dia b/dia/diameter_3gpp_ts29_061_sgi.dia
@@ -103,6 +103,8 @@
               [ 3GPP-CAMEL-Charging-Info ]
               [ 3GPP-Negotiated-DSCP ]
               [ 3GPP-Allocate-IP-Type ]
+              ;; Added for consistency with RADIUS
+              [ 3GPP-IMEISV ]
               [ TP-NAT-Pool-Id ]
             * [ AVP ]
 

diff --git a/test/diameter_nasreq_SUITE.erl b/test/diameter_nasreq_SUITE.erl
@@ -409,7 +409,7 @@ simple(Config, TermOpts) ->
 
 simple_tdf_userid(Config, TermOpts) ->
     Stats0 = get_stats(?SERVICE),
-    Msisdn = <<"543148000012345">>,
+    Msisdn = ?MSISDN_FOR_IMEI_SV, % <<"543148000012345">>,
     Imsi = <<"250071234567890">>,
     Imei = <<82,21,50,96,32,80,30,0>>,
     {ok, Session} = ergw_aaa_session_sup:new_session(

diff --git a/test/diameter_test_server.erl b/test/diameter_test_server.erl
@@ -18,6 +18,7 @@
 -include("../include/diameter_3gpp_ts29_212.hrl").
 -include("../include/diameter_3gpp_ts32_299.hrl").
 -include("../include/diameter_3gpp_ts32_299_ro.hrl").
+-include("ergw_aaa_test_lib.hrl").
 
 -export([start/0, start/2,
 	 start_nasreq/0, start_nasreq/2,
@@ -249,7 +250,6 @@ handle_request(#diameter_packet{msg = ['ACR' | Msg]}, _SvcName, {_, Caps}, _Extr
 	    #{'Session-Id' := Id,
 	      'Accounting-Record-Type' := Type,
 	      'Accounting-Record-Number' := Number} = Msg,
-	    ct:pal("Received msg: ~p", [Msg]),
 	    ACA =  #{'Session-Id' => Id,
 		     'Result-Code' => 2001,
 		     'Origin-Host' => OH,
@@ -276,30 +276,38 @@ handle_request(#diameter_packet{msg = ['AAR' | Msg]}, _SvcName, {_, Caps}, _Extr
     #{'Session-Id' := Id,
       'Auth-Request-Type' := Type,
       'Auth-Application-Id' := AppId} = Msg,
+    ct:pal("AAR received: ~p", [Msg]),
     FramedIP = case maps:get('Framed-IP-Address', Msg, [<<0,0,0,0>>]) of
 		   [<<0,0,0,0>>] -> [<<10,106,14,227>>];
 		   FramedIPReq -> FramedIPReq
 	       end,
-    AAA0 =  #{'Session-Id' => Id,
-	      'Result-Code' => 2001,
-	      'Origin-Host' => OH,
-	      'Origin-Realm' => OR,
-  	      'SN-Primary-DNS-Server' => [{1,2,3,4}],
-	      'SN-Secondary-DNS-Server' => [{5,6,7,8}],
- 	      'Acct-Interim-Interval' => [InterimAccounting],
-	      'Authorization-Lifetime' => [AuthLifeTime],
-	      'Framed-IP-Address' => FramedIP,
-	      'Framed-MTU' => 1500,
-	      'Auth-Request-Type' => Type,
-	      'Auth-Application-Id' => AppId},
-    AAA = if is_map_key('Framed-IPv6-Prefix', Msg) ->
-		  AAA0#{'3GPP-IPv6-DNS-Servers' =>
-			    ergw_aaa_3gpp_dict:encode('3GPP-IPv6-DNS-Servers',
-						      [{16#fe80,0,0,0,0,0,0,1},
-						       {16#fe80,0,0,0,0,0,0,2}])};
-	     true -> AAA0
-	  end,
-    {reply, ['AAA' | AAA]};
+    OkResponse = fun() ->
+	AAA0 =  #{'Session-Id' => Id,
+	          'Result-Code' => 2001,
+	          'Origin-Host' => OH,
+	          'Origin-Realm' => OR,
+  	          'SN-Primary-DNS-Server' => [{1,2,3,4}],
+	          'SN-Secondary-DNS-Server' => [{5,6,7,8}],
+ 	          'Acct-Interim-Interval' => [InterimAccounting],
+	          'Authorization-Lifetime' => [AuthLifeTime],
+	          'Framed-IP-Address' => FramedIP,
+	          'Framed-MTU' => 1500,
+	          'Auth-Request-Type' => Type,
+	          'Auth-Application-Id' => AppId},
+	AAA = if is_map_key('Framed-IPv6-Prefix', Msg) ->
+	    AAA0#{'3GPP-IPv6-DNS-Servers' =>
+		    ergw_aaa_3gpp_dict:encode('3GPP-IPv6-DNS-Servers',
+					      [{16#fe80,0,0,0,0,0,0,1},
+					       {16#fe80,0,0,0,0,0,0,2}])};
+		    true -> AAA0
+	end,
+	{reply, ['AAA' | AAA]}
+    end,
+    case Msg of
+	#{'Calling-Station-Id' := ?MSISDN_FOR_IMEI_SV, '3GPP-IMEISV' := _} -> OkResponse();
+	#{'Calling-Station-Id' := ?MSISDN_FOR_IMEI_SV} -> {answer_message, 5005};
+	_ -> OkResponse()
+    end;
 
 handle_request(#diameter_packet{msg = ['STR' | Msg]}, _SvcName, {_, Caps}, _Extra)
   when is_map(Msg) ->
@@ -593,7 +601,7 @@ discard_msg(#{'Service-Information' :=
 		[#{'Subscription-Id' := Ids}]}) ->
     lists:member(#{'Subscription-Id-Type' => 1,'Subscription-Id-Data' => <<"999999999999999">>}, Ids);
 discard_msg(_) ->
-    false.	    
+    false.
 
 fail_answer(<<"FAIL-BROKEN-ANSWER">>, Pkt) ->
     broken_answer(Pkt);

diff --git a/test/ergw_aaa_test_lib.hrl b/test/ergw_aaa_test_lib.hrl
@@ -30,3 +30,5 @@
 			    outstanding_reqs/0, get_session_stats/0, get_session_stats/2, wait_for_session/4,
 			    reset_session_stats/0]).
 -endif.
+
+-define(MSISDN_FOR_IMEI_SV, "AAAAAAAAAAAAAAA").