Add a performance test to connect multiple UEs in parallel

cmd/app.go

@@ -195,6 +195,47 @@ func main() {
 					return nil
 				},
 			},
+			{
+				Name:    "load-test-parallel",
+				Aliases: []string{"load-test-parallel"},
+				Usage: "\nLoad endurance stress tests.\n" +
+					"Example for testing multiple UEs: load-test-parallel -n 1000 -d 30 -t 30 -a\n",
+				Flags: []cli.Flag{
+					&cli.IntFlag{Name: "number-of-ues", Value: 1, Aliases: []string{"n"}},
+					&cli.IntFlag{Name: "delay-per-ue", Value: 1, Aliases: []string{"d"}},
+					&cli.IntFlag{Name: "startup-delay", Value: 1, Aliases: []string{"t"}},
+					&cli.BoolFlag{Name: "enable-analytics", Aliases: []string{"a"}},
+				},
+				Action: func(c *cli.Context) error {
+					var numUes int
+					var delayUes int
+					var delayStart int
+					var showAnalytics bool
+					name := "Testing registration of multiple UEs in parallel"
+					cfg := config.Data
+
+					if c.IsSet("number-of-ues") && c.IsSet("delay-per-ue") && c.IsSet("startup-delay") {
+						numUes = c.Int("number-of-ues")
+						delayUes = c.Int("delay-per-ue")
+						delayStart = c.Int("startup-delay")
+						showAnalytics = c.Bool("enable-analytics")
+					} else {
+						log.Info(c.Command.Usage)
+						return nil
+					}				
+
+					log.Info("---------------------------------------")
+					log.Info("[TESTER] Starting test function: ", name)
+					log.Info("[TESTER][UE] Number of UEs: ", numUes)
+					log.Info("[TESTER][GNB] gNodeB control interface IP/Port: ", cfg.GNodeB.ControlIF.Ip, "/", cfg.GNodeB.ControlIF.Port)
+					log.Info("[TESTER][GNB] gNodeB data interface IP/Port: ", cfg.GNodeB.DataIF.Ip, "/", cfg.GNodeB.DataIF.Port)
+					log.Info("[TESTER][AMF] AMF IP/Port: ", cfg.AMF.Ip, "/", cfg.AMF.Port)
+					log.Info("---------------------------------------")
+					templates.TestMultiUesInParallel(numUes, delayUes, delayStart, showAnalytics)
+
+					return nil
+				},
+			},
 		},
 	}
 	err := app.Run(os.Args)

internal/analytics/log_time/log_time.go

@@ -0,0 +1,31 @@
+package analytics
+
+import (
+	"fmt"
+	"time"
+	log "github.com/sirupsen/logrus"
+)
+
+var enabled = true
+func ChangeAnalyticsState(state bool){
+	enabled = state
+}
+
+var gnbid = 0
+func SetGnodebId(id int){
+	gnbid = id
+}
+
+func LogUeTime(id string, task string) {
+
+	now := time.Now()
+
+	go ShowUeLog(id, task, now)
+}
+
+func ShowUeLog(id string, task string, now time.Time) {
+	if enabled {
+		nsec_now := now.UnixNano()
+		log.Info(fmt.Sprintf("[ANALYTICS] %d, %s, %s, %d", gnbid, id, task, nsec_now))
+	}
+}

internal/control_test_engine/ue/context/context.go

@@ -15,6 +15,8 @@ import (
 	"net"
 	"reflect"
 	"regexp"
+
+	log_time "my5G-RANTester/internal/analytics/log_time"
 )
 
 // 5GMM main states in the UE.
@@ -31,7 +33,7 @@ const SM5G_PDU_SESSION_ACTIVE_PENDING = 0x07
 const SM5G_PDU_SESSION_ACTIVE = 0x08
 
 type UEContext struct {
-	id         uint8
+	id         int64
 	UeSecurity SECURITY
 	StateMM    int
 	StateSM    int
@@ -47,7 +49,7 @@ type Amf struct {
 }
 
 type PDUSession struct {
-	Id        uint8
+	Id        int64
 	ueIP      string
 	ueGnbIP   net.IP
 	Dnn       string
@@ -79,7 +81,7 @@ type SECURITY struct {
 func (ue *UEContext) NewRanUeContext(msin string,
 	cipheringAlg, integrityAlg uint8,
 	k, opc, op, amf, sqn, mcc, mnc, dnn string,
-	sst int32, sd string, id uint8) {
+	sst int32, sd string, id int64) {
 
 	// added SUPI.
 	ue.UeSecurity.Msin = msin
@@ -146,7 +148,7 @@ func (ue *UEContext) NewRanUeContext(msin string,
 
 }
 
-func (ue *UEContext) GetUeId() uint8 {
+func (ue *UEContext) GetUeId() int64 {
 	return ue.id
 }
 
@@ -164,38 +166,47 @@ func (ue *UEContext) GetSupi() string {
 
 func (ue *UEContext) SetStateSM_PDU_SESSION_INACTIVE() {
 	ue.StateSM = SM5G_PDU_SESSION_INACTIVE
+	log_time.LogUeTime(ue.GetMsin(), "SM5G_PDU_SESSION_INACTIVE")
 }
 
 func (ue *UEContext) SetStateSM_PDU_SESSION_ACTIVE() {
 	ue.StateSM = SM5G_PDU_SESSION_ACTIVE
+	log_time.LogUeTime(ue.GetMsin(), "SM5G_PDU_SESSION_ACTIVE")
 }
 
 func (ue *UEContext) SetStateSM_PDU_SESSION_PENDING() {
 	ue.StateSM = SM5G_PDU_SESSION_ACTIVE_PENDING
+	log_time.LogUeTime(ue.GetMsin(), "SM5G_PDU_SESSION_ACTIVE_PENDING")
 }
 
 func (ue *UEContext) SetStateMM_DEREGISTERED_INITIATED() {
 	ue.StateMM = MM5G_DEREGISTERED_INIT
+	log_time.LogUeTime(ue.GetMsin(), "MM5G_DEREGISTERED_INIT")
 }
 
 func (ue *UEContext) SetStateMM_MM5G_SERVICE_REQ_INIT() {
 	ue.StateMM = MM5G_SERVICE_REQ_INIT
+	log_time.LogUeTime(ue.GetMsin(), "MM5G_SERVICE_REQ_INIT")
 }
 
 func (ue *UEContext) SetStateMM_REGISTERED_INITIATED() {
 	ue.StateMM = MM5G_REGISTERED_INITIATED
+	log_time.LogUeTime(ue.GetMsin(), "MM5G_REGISTERED_INITIATED")
 }
 
 func (ue *UEContext) SetStateMM_REGISTERED() {
 	ue.StateMM = MM5G_REGISTERED
+	log_time.LogUeTime(ue.GetMsin(), "MM5G_REGISTERED")
 }
 
 func (ue *UEContext) SetStateMM_NULL() {
 	ue.StateMM = MM5G_NULL
+	log_time.LogUeTime(ue.GetMsin(), "MM5G_NULL")
 }
 
 func (ue *UEContext) SetStateMM_DEREGISTERED() {
 	ue.StateMM = MM5G_DEREGISTERED
+	log_time.LogUeTime(ue.GetMsin(), "MM5G_DEREGISTERED")
 }
 
 func (ue *UEContext) GetStateSM() int {
@@ -234,7 +245,7 @@ func (ue *UEContext) GetGnbIp() net.IP {
 	return ue.PduSession.ueGnbIP
 }
 
-func (ue *UEContext) GetPduSesssionId() uint8 {
+func (ue *UEContext) GetPduSesssionId() int64 {
 	return ue.PduSession.Id
 }
 

internal/control_test_engine/ue/data/service/service.go

@@ -6,6 +6,8 @@ import (
 	"github.com/vishvananda/netlink"
 	"my5G-RANTester/internal/control_test_engine/ue/context"
 	"net"
+
+	log_time "my5G-RANTester/internal/analytics/log_time"
 )
 
 func InitDataPlane(ue *context.UEContext, message []byte) {
@@ -84,9 +86,10 @@ func InitDataPlane(ue *context.UEContext, message []byte) {
 
 	log.Info("[UE][DATA] UE is ready for using data plane")
 
+	log_time.LogUeTime(ue.GetMsin(), "DataPlaneReady")
+
 	// contex of tun interface
 	ue.SetTunInterface(newInterface)
 	ue.SetTunRoute(ueRoute)
 	ue.SetTunRule(ueRule)
-
 }

internal/control_test_engine/ue/nas/message/nas_control/mm_5gs/registration-request.go

@@ -23,7 +23,7 @@ func GetRegistrationRequest(registrationType uint8, requestedNSSAI *nasType.Requ
 	registrationRequest.SpareHalfOctetAndSecurityHeaderType.SetSpareHalfOctet(0x00)
 	registrationRequest.RegistrationRequestMessageIdentity.SetMessageType(nas.MsgTypeRegistrationRequest)
 	registrationRequest.NgksiAndRegistrationType5GS.SetTSC(nasMessage.TypeOfSecurityContextFlagNative)
-	registrationRequest.NgksiAndRegistrationType5GS.SetNasKeySetIdentifiler(ue.GetUeId())
+	registrationRequest.NgksiAndRegistrationType5GS.SetNasKeySetIdentifiler(uint8(ue.GetUeId()))
 	registrationRequest.NgksiAndRegistrationType5GS.SetRegistrationType5GS(registrationType)
 	registrationRequest.MobileIdentity5GS = ue.GetSuci()
 	if capability {

internal/control_test_engine/ue/nas/message/nas_control/mm_5gs/ul-nas-transport.go

@@ -27,7 +27,7 @@ func UlNasTransport(ue *context.UEContext, requestType uint8) ([]byte, error) {
 	return pdu, nil
 }
 
-func getUlNasTransport_PduSessionEstablishmentRequest(pduSessionId uint8, requestType uint8, dnnString string, sNssai *models.Snssai) (nasPdu []byte) {
+func getUlNasTransport_PduSessionEstablishmentRequest(pduSessionId int64, requestType uint8, dnnString string, sNssai *models.Snssai) (nasPdu []byte) {
 
 	pduSessionEstablishmentRequest := sm_5gs.GetPduSessionEstablishmentRequest(pduSessionId)
 
@@ -41,7 +41,7 @@ func getUlNasTransport_PduSessionEstablishmentRequest(pduSessionId uint8, reques
 	ulNasTransport.ExtendedProtocolDiscriminator.SetExtendedProtocolDiscriminator(nasMessage.Epd5GSMobilityManagementMessage)
 	ulNasTransport.PduSessionID2Value = new(nasType.PduSessionID2Value)
 	ulNasTransport.PduSessionID2Value.SetIei(nasMessage.ULNASTransportPduSessionID2ValueType)
-	ulNasTransport.PduSessionID2Value.SetPduSessionID2Value(pduSessionId)
+	ulNasTransport.PduSessionID2Value.SetPduSessionID2Value(uint8(pduSessionId))
 	ulNasTransport.RequestType = new(nasType.RequestType)
 	ulNasTransport.RequestType.SetIei(nasMessage.ULNASTransportRequestTypeType)
 	ulNasTransport.RequestType.SetRequestTypeValue(requestType)

internal/control_test_engine/ue/nas/message/nas_control/sm_5gs/pdu-session-establishment-request.go

@@ -9,7 +9,7 @@ import (
 	"my5G-RANTester/lib/nas/nasType"
 )
 
-func GetPduSessionEstablishmentRequest(pduSessionId uint8) (nasPdu []byte) {
+func GetPduSessionEstablishmentRequest(pduSessionId int64) (nasPdu []byte) {
 
 	m := nas.NewMessage()
 	m.GsmMessage = nas.NewGsmMessage()
@@ -18,7 +18,7 @@ func GetPduSessionEstablishmentRequest(pduSessionId uint8) (nasPdu []byte) {
 	pduSessionEstablishmentRequest := nasMessage.NewPDUSessionEstablishmentRequest(0)
 	pduSessionEstablishmentRequest.ExtendedProtocolDiscriminator.SetExtendedProtocolDiscriminator(nasMessage.Epd5GSSessionManagementMessage)
 	pduSessionEstablishmentRequest.SetMessageType(nas.MsgTypePDUSessionEstablishmentRequest)
-	pduSessionEstablishmentRequest.PDUSessionID.SetPDUSessionID(pduSessionId)
+	pduSessionEstablishmentRequest.PDUSessionID.SetPDUSessionID(uint8(pduSessionId))
 	pduSessionEstablishmentRequest.PTI.SetPTI(0x01)
 	pduSessionEstablishmentRequest.IntegrityProtectionMaximumDataRate.SetMaximumDataRatePerUEForUserPlaneIntegrityProtectionForDownLink(0xff)
 	pduSessionEstablishmentRequest.IntegrityProtectionMaximumDataRate.SetMaximumDataRatePerUEForUserPlaneIntegrityProtectionForUpLink(0xff)

internal/control_test_engine/ue/nas/message/sender/send.go

@@ -6,8 +6,8 @@ import (
 )
 
 func SendToGnb(ue *context.UEContext, message []byte) {
-
 	conn := ue.GetUnixConn()
+
 	_, err := conn.Write(message)
 	if err != nil {
 		fmt.Println("Tratar o erro")

internal/control_test_engine/ue/nas/service/service.go

@@ -29,7 +29,6 @@ func InitConn(ue *context.UEContext) error {
 	return nil
 }
 
-// ue listen unix sockets.
 func UeListen(ue *context.UEContext) {
 
 	buf := make([]byte, 65535)

internal/control_test_engine/ue/nas/trigger/trigger.go

@@ -8,7 +8,6 @@ import (
 )
 
 func InitRegistration(ue *context.UEContext) {
-
 	// registration procedure started.
 	registrationRequest := mm_5gs.GetRegistrationRequest(
 		nasMessage.RegistrationType5GSInitialRegistration,

internal/control_test_engine/ue/state/dispatch.go

@@ -9,7 +9,9 @@ import (
 func DispatchState(ue *context.UEContext, message []byte) {
 
 	// if state is PDU session inactive send to analyze NAS
-	switch ue.GetStateSM() {
+	var state = ue.GetStateSM()
+
+	switch state {
 
 	case context.SM5G_PDU_SESSION_INACTIVE:
 		nas.DispatchNas(ue, message)

internal/control_test_engine/ue/ue.go

@@ -14,9 +14,7 @@ import (
 	"time"
 )
 
-func RegistrationUe(conf config.Config, id uint8, wg *sync.WaitGroup) {
-
-	// wg := sync.WaitGroup{}
+func RegistrationUe(conf config.Config, id int64, wg *sync.WaitGroup) {
 
 	// new UE instance.
 	ue := &context.UEContext{}
@@ -50,8 +48,6 @@ func RegistrationUe(conf config.Config, id uint8, wg *sync.WaitGroup) {
 	// registration procedure started.
 	trigger.InitRegistration(ue)
 
-	// wg.Wait()
-
 	// control the signals
 	sigUe := make(chan os.Signal, 1)
 	signal.Notify(sigUe, os.Interrupt)
@@ -61,11 +57,10 @@ func RegistrationUe(conf config.Config, id uint8, wg *sync.WaitGroup) {
 	ue.Terminate()
 	wg.Done()
 	// os.Exit(0)
-
 }
 
 func RegistrationUeMonitor(conf config.Config,
-	id uint8, monitor *monitoring.Monitor, wg *sync.WaitGroup, start time.Time) {
+	id int64, monitor *monitoring.Monitor, wg *sync.WaitGroup, start time.Time) {
 
 	// new UE instance.
 	ue := &context.UEContext{}

internal/templates/test-multi-ues-in-parallel.go

@@ -0,0 +1,51 @@
+package templates
+
+import (
+	log "github.com/sirupsen/logrus"
+	"my5G-RANTester/config"
+	"my5G-RANTester/internal/control_test_engine/gnb"
+	"my5G-RANTester/internal/control_test_engine/ue"
+	"sync"
+	"time"
+
+	log_time "my5G-RANTester/internal/analytics/log_time"
+	"strconv"
+)
+
+func TestMultiUesInParallel(numUes int, delayUes int, delayStart int, showAnalytics bool) {
+
+	wg := sync.WaitGroup{}
+
+	cfg, err := config.GetConfig()
+	if err != nil {
+		//return nil
+		log.Fatal("Error in get configuration")
+	}
+
+	gnbid, err := strconv.Atoi(cfg.GNodeB.PlmnList.GnbId) // Parse gNB ID
+	log_time.SetGnodebId(gnbid) // Set gNB ID
+	log_time.ChangeAnalyticsState(showAnalytics) // Enable/Disable analytics
+
+	go gnb.InitGnb(cfg, &wg)
+
+	wg.Add(1)
+
+	time.Sleep(time.Duration(delayStart) * time.Second)
+    msin :=  cfg.Ue.Msin
+
+	for i := 1; i <= numUes; i++ {
+		go registerSingleUe(cfg, wg, msin, i)
+		time.Sleep(time.Duration(delayUes) * time.Millisecond)
+	}
+
+	wg.Wait()
+}
+
+func registerSingleUe(cfg config.Config, wg sync.WaitGroup, msin string, i int) {
+	imsi := imsiGenerator(i, msin)
+	log.Info("[TESTER] TESTING REGISTRATION USING IMSI ", imsi, " UE")
+	cfg.Ue.Msin = imsi
+	log_time.LogUeTime(imsi, "StartRegistration")
+	go ue.RegistrationUe(cfg, int64(i), &wg)
+	//wg.Add(1)
+}

internal/templates/test-multi-ues-in-queue.go

@@ -32,7 +32,7 @@ func TestMultiUesInQueue(numUes int) {
 		imsi := imsiGenerator(i, msin)
 		log.Info("[TESTER] TESTING REGISTRATION USING IMSI ", imsi, " UE")
 		cfg.Ue.Msin = imsi
-		go ue.RegistrationUe(cfg, uint8(i), &wg)
+		go ue.RegistrationUe(cfg, int64(i), &wg)
 		wg.Add(1)
 
 		time.Sleep(10 * time.Second)

internal/templates/test-ue-latency-in-interval.go

@@ -45,7 +45,7 @@ func TestUesLatencyInInterval(interval int) int64 {
 
 			time.Sleep(400 * time.Millisecond)
 
-			go ue.RegistrationUeMonitor(cfg, uint8(i), &monitor, &wg, start)
+			go ue.RegistrationUeMonitor(cfg, int64(i), &monitor, &wg, start)
 
 			wg.Add(1)