From c538ecf3e23c04bfe093c1d4aad8520ba7dc40be Mon Sep 17 00:00:00 2001
From: Malte Bauch <malte.bauch@stud.h-da.de>
Date: Mon, 3 Jul 2023 19:02:03 +0200
Subject: [PATCH] Change to WriteMsgUDP within AddPeer method
---
main_test.go | 2 +-
quantumlayer/quantumlayer-emu-prng.go | 4 ++--
quantumlayer/quantumlayer-emu-prng_test.go | 5 ++---
3 files changed, 5 insertions(+), 6 deletions(-)
diff --git a/main_test.go b/main_test.go
index d2e0e6d..33cdc52 100644
--- a/main_test.go
+++ b/main_test.go
@@ -74,7 +74,7 @@ func emulatedKMS(myName string, myUDPAddr string, peerUDPAddr string) {
return
}
- myQL.QuantumElementLink.AddPeer(*udpQL2Addr)
+ myQL.QuantumElementLink.AddPeer(udpQL2Addr)
// Start the SDN/management and key retrieval interface
go kms.StartETSI(peerUDPAddr, emuKMS)
diff --git a/quantumlayer/quantumlayer-emu-prng.go b/quantumlayer/quantumlayer-emu-prng.go
index 169b5ec..7b437dc 100644
--- a/quantumlayer/quantumlayer-emu-prng.go
+++ b/quantumlayer/quantumlayer-emu-prng.go
@@ -150,7 +150,7 @@ func (qlemuprng *QuantumlayerEmuPRNG) PowerOff() {
log.Println("QuantumlayerEmuPRNG is powered off...discharging.")
}
-func (qlemuprng *QuantumlayerEmuPRNG) AddPeer(addr net.UDPAddr) {
+func (qlemuprng *QuantumlayerEmuPRNG) AddPeer(addr *net.UDPAddr) {
if qlemuprng.poweron == false {
return
}
@@ -184,7 +184,7 @@ func (qlemuprng *QuantumlayerEmuPRNG) AddPeer(addr net.UDPAddr) {
log.Printf("json.Marshal error %s", err)
}
- _, _, err = qlemuprng.udpSrvConn.WriteMsgUDPAddrPort(jsonPayload, nil, addr.AddrPort())
+ _, _, err = qlemuprng.udpSrvConn.WriteMsgUDP(jsonPayload, nil, addr)
if err != nil {
log.Fatalf("WriteMsgUDPAddrPort failed: %s", err)
}
diff --git a/quantumlayer/quantumlayer-emu-prng_test.go b/quantumlayer/quantumlayer-emu-prng_test.go
index 69d30b5..502377a 100644
--- a/quantumlayer/quantumlayer-emu-prng_test.go
+++ b/quantumlayer/quantumlayer-emu-prng_test.go
@@ -35,8 +35,8 @@ func TestQuantumLayer(t *testing.T) {
return
}
- ql1.AddPeer(*udpQL2Addr)
- ql2.AddPeer(*udpQL1Addr)
+ ql1.AddPeer(udpQL2Addr)
+ ql2.AddPeer(udpQL1Addr)
for n := 0; n < 2; n++ {
resultQl1, err := ql1.GetKeyBatchPeer()
@@ -59,5 +59,4 @@ func TestQuantumLayer(t *testing.T) {
time.Sleep(5 * time.Second)
}
-
}
--
GitLab