mirror of
https://github.com/netbirdio/netbird.git
synced 2026-09-15 11:19:08 +02:00
38 lines
1.5 KiB
Go
38 lines
1.5 KiB
Go
package internal
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/netbirdio/netbird/client/internal/pqkem"
|
|
)
|
|
|
|
type pqNoopHandler struct{}
|
|
|
|
func (pqNoopHandler) OnNewPSKReady(pqkem.RemoteID, pqkem.PSK) error { return nil }
|
|
func (pqNoopHandler) OnRekeyFailed(pqkem.RemoteID) error { return nil }
|
|
|
|
// TestPQAdapter_CapabilityRoleAware locks the role-aware capability signal: the KEM
|
|
// payload only flows initiator-offer -> responder-answer, so an empty message in the
|
|
// other direction comes from a perfectly capable peer and must NOT flag it. Only the
|
|
// message that should carry material (the answer we receive as initiator) marks a peer
|
|
// non-capable when empty.
|
|
func TestPQAdapter_CapabilityRoleAware(t *testing.T) {
|
|
// localID "zzzz" > "aaaa" => this manager is the KEM initiator for peer "aaaa".
|
|
mgr := pqkem.NewManager("zzzz", pqNoopHandler{}, nil)
|
|
defer mgr.Stop()
|
|
h := pqHandshaker{mgr: mgr}
|
|
|
|
// An empty OFFER from our peer is normal here: as the initiator's responder it puts
|
|
// its material in the answer, not the offer. It must not disable our offering.
|
|
h.AnswerPayload("aaaa", nil)
|
|
payload, _ := h.OfferPayload("aaaa")
|
|
require.NotNil(t, payload, "an empty offer from a responder-role peer must not mark it non-capable")
|
|
|
|
// An empty ANSWER to our offer means the peer does not run the KEM -> stop offering.
|
|
h.OnAnswer("aaaa", nil)
|
|
payload2, _ := h.OfferPayload("aaaa")
|
|
require.Nil(t, payload2, "an empty answer to our offer marks the peer non-capable, so we stop offering")
|
|
}
|