Script to switch edge

Makefile

@@ -186,3 +186,8 @@ ue/ping/%:
 	@# pings from ue1 to ue2
 	@TARGET=$(shell docker exec -it ue$(@F)-debug bash -c "ip --brief address show uesimtun0|awk '{print \$$3; exit}'|cut -d"/" -f 1");\
 	docker exec -it ue$(*D)-debug bash -c "ping $$TARGET"
+.PHONY: ue/switch-edge
+ue/switch-edge/%:
+	@# swich edge for ue
+	@UE_IP=$(shell docker exec -it ue$(@F)-debug bash -c "ip --brief address show uesimtun0|awk '{print \$$3; exit}'|cut -d"/" -f 1");\
+	scripts/switch.py $(BCONFIG) $$UE_IP

doc/commands.md

@@ -35,3 +35,4 @@ When you update the git repository, please ensure to update Docker images as wel
 ## Using UEs
 - `make ue/ip/<ue-number>`: show IP Address of the UE within the Mobile Network
 - `make ue/ping/<ue-source-number>/<ue-target-number>`: ping from `ue-source` to `ue-target`
+- `make ue/switch-edge/<ue-number>`: switch edge enabled for this ue (only for NextMN-SRv6)

doc/how-to-test.md

@@ -7,6 +7,5 @@
 3. `make u`
 4. `make t/ue1` and `ping 10.4.0.1`
 5. `make t/s0`/`make t/s1` and `tshark -i any -f icmp`
-6. `make ctrl` ; In the web browser, look for uplink `srgw` rule `UUID` for your UE
-7. Enable new uplink rule & disable old uplink rule using `./script/switch.sh {UUID_TO_ENABLE} {UUID_TO_DISABLE}` (replace `{UUID_TO_ENABLE}`/`{UUID_TO_DISABLE}` with rule's UUIDs).
+6. `make ue/switch-edge/1` to swich between edges
 8. When done with the test, `make d`

scripts/switch.py

@@ -0,0 +1,34 @@
+#!/usr/bin/env python3
+'''Switch UE edge'''
+# Copyright 2024 Louis Royer. All rights reserved.
+# Use of this source code is governed by a MIT-style license that can be
+# found in the LICENSE file.
+# SPDX-License-Identifier: MIT
+import argparse
+import ipaddress
+import requests
+import yaml
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser(
+            prog='switch',
+            description='Switch UE edge'
+        )
+    parser.add_argument('config')
+    parser.add_argument('ue_addr')
+    args = parser.parse_args()
+    with open(args.config, 'r', encoding='utf-8') as f:
+        c = yaml.safe_load(f)
+        srgw = f'http://[{c["subnets"]["control"]["srgw0"]["ipv6_address"]}]:8080'
+        r = requests.get(f'{srgw}/rules', timeout=1)
+        rules = [None, None]
+        for ruleid, rule in r.json().items():
+            if ipaddress.ip_address(args.ue_addr) in ipaddress.ip_network(
+                    rule['Match']['ue-ip-prefix']):
+                if not rule['Enabled']:
+                    rules[0] = ruleid
+                else:
+                    rules[1] = ruleid
+        if (rules[0] is None) or (rules[1] is None):
+            raise ValueError('Could not found 1 enabled and 1 disabled rule for this IP Address')
+        requests.patch(f'{srgw}/rules/switch/{rules[0]}/{rules[1]}', timeout=1)