From 897a9df23086605053b080b67e5f3656d6466c46 Mon Sep 17 00:00:00 2001
From: davidkneipp <david@davidkneipp.com>
Date: Tue, 30 Jan 2024 12:31:11 +1000
Subject: [PATCH] Remove assert on missing IMS Services/AAA

---
 CHANGELOG.md    | 4 ++++
 lib/diameter.py | 4 ++--
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 62f5fed..7675912 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -7,6 +7,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [1.0.1] - 2023-01-23
 
+
+### Removed
+ - Assert on missing "IMS Services" for AAA/Audio Request
+
 ### Changed
 
 - Reduced verbosity of failing subscriber lookups to debug
diff --git a/lib/diameter.py b/lib/diameter.py
index 00fb694..b8af15b 100755
--- a/lib/diameter.py
+++ b/lib/diameter.py
@@ -2903,8 +2903,8 @@ def Answer_16777236_265(self, packet_vars, avps):
                 try:
                     afApplicationIdentifier = self.get_avp_data(avps, 504)[0]
                     mediaType = self.get_avp_data(avps, 520)[0]
-                    assert(bytes.fromhex(afApplicationIdentifier).decode('ascii') == "IMS Services")
-                    assert(int(mediaType, 16) == 0)
+                    # In order to send a Gx RAR, we need to ensure that mediaType is AUDIO(0) or VIDEO(1)
+                    assert(int(mediaType, 16) == 0 or int(mediaType, 16) == 1)
 
                     # At this point, we know the AAR is indicating a call setup, so we'll get the serving pgw information, then send a 
                     # RAR to the PGW over Gx, asking it to setup the dedicated bearer.