Merge pull request #41 from nttcom/feature/PCEP_capability

Add struct of TLVs and interface of Capabilities

pkg/packet/pcep/capability.go

@@ -0,0 +1,42 @@
+// Copyright (c) 2022 NTT Communications Corporation
+//
+// This software is released under the MIT License.
+// see https://github.com/nttcom/pola/blob/main/LICENSE
+
+package pcep
+
+type CapabilityInterface interface {
+	TlvInterface
+}
+
+func PolaCapability() []CapabilityInterface {
+	return []CapabilityInterface{
+		&StatefulPceCapability{
+			LspUpdateCapability:        true,
+			IncludeDBVersion:           false,
+			LspInstantiationCapability: true,
+			TriggeredResync:            false,
+			DeltaLspSyncCapability:     false,
+			TriggeredInitialSync:       false,
+		},
+		&PathSetupTypeCapability{
+			Length:         uint16(10),
+			PathSetupTypes: Psts{PST_RSVP_TE, PST_SR_TE},
+			SubTlvs: []TlvInterface{
+				&SrPceCapability{
+					UnlimitedMSD:    false,
+					SupportNAI:      false,
+					MaximumSidDepth: uint8(16),
+				},
+			},
+		},
+		&SrPceCapability{
+			UnlimitedMSD:    false,
+			SupportNAI:      false,
+			MaximumSidDepth: uint8(16),
+		},
+		&AssocTypeList{
+			AssocTypes: []AssocType{AT_PATH_PROTECTION_ASSOCIATION, AT_SR_POLICY_ASSOCIATION},
+		},
+	}
+}

pkg/packet/pcep/message.go

@@ -107,8 +107,8 @@ func (m *OpenMessage) Serialize() []uint8 {
 	return byteOpenMessage
 }
 
-func NewOpenMessage(sessionID uint8, keepalive uint8) (*OpenMessage, error) {
-	oo, err := NewOpenObject(sessionID, keepalive)
+func NewOpenMessage(sessionID uint8, keepalive uint8, capabilities []CapabilityInterface) (*OpenMessage, error) {
+	oo, err := NewOpenObject(sessionID, keepalive, capabilities)
 	if err != nil {
 		return nil, err
 	}