packet: support pretty PathAttribute JSON printing

some are missing though.

Signed-off-by: FUJITA Tomonori <fujita.tomonori@lab.ntt.co.jp>

1 files changed, 115 insertions, 0 deletions

diff --git a/packet/bgp.go b/packet/bgp.go
index 747957fa..6369269e 100644
--- a/packet/bgp.go
+++ b/packet/bgp.go
@@ -17,6 +17,7 @@ package bgp
 
 import (
 	"encoding/binary"
+	"encoding/json"
 	"fmt"
 	"math"
 	"net"
@@ -1141,6 +1142,16 @@ type PathAttributeOrigin struct {
 	PathAttribute
 }
 
+func (p *PathAttributeOrigin) MarshalJSON() ([]byte, error) {
+	return json.Marshal(struct {
+		Type  string
+		Value uint8
+	}{
+		Type:  p.Type.String(),
+		Value: uint8(p.Value[0]),
+	})
+}
+
 func NewPathAttributeOrigin(value uint8) *PathAttributeOrigin {
 	return &PathAttributeOrigin{
 		PathAttribute: PathAttribute{
@@ -1300,6 +1311,25 @@ func (p *PathAttributeAsPath) Serialize() ([]byte, error) {
 	return p.PathAttribute.Serialize()
 }
 
+func (p *PathAttributeAsPath) MarshalJSON() ([]byte, error) {
+	aslist := make([]uint32, 0)
+	for _, a := range p.Value {
+		path, y := a.(*As4PathParam)
+		if y {
+			aslist = append(aslist, path.AS...)
+		} else {
+			// TODO aspathparam
+		}
+	}
+	return json.Marshal(struct {
+		Type   string
+		AsPath []uint32
+	}{
+		Type:   p.Type.String(),
+		AsPath: aslist,
+	})
+}
+
 func NewPathAttributeAsPath(value []AsPathParamInterface) *PathAttributeAsPath {
 	return &PathAttributeAsPath{
 		PathAttribute: PathAttribute{
@@ -1326,6 +1356,16 @@ func (p *PathAttributeNextHop) Serialize() ([]byte, error) {
 	return p.PathAttribute.Serialize()
 }
 
+func (p *PathAttributeNextHop) MarshalJSON() ([]byte, error) {
+	return json.Marshal(struct {
+		Type    string
+		Nexthop string
+	}{
+		Type:    p.Type.String(),
+		Nexthop: p.Value.String(),
+	})
+}
+
 func NewPathAttributeNextHop(value string) *PathAttributeNextHop {
 	return &PathAttributeNextHop{
 		PathAttribute: PathAttribute{
@@ -1354,6 +1394,16 @@ func (p *PathAttributeMultiExitDisc) Serialize() ([]byte, error) {
 	return p.PathAttribute.Serialize()
 }
 
+func (p *PathAttributeMultiExitDisc) MarshalJSON() ([]byte, error) {
+	return json.Marshal(struct {
+		Type   string
+		Metric uint32
+	}{
+		Type:   p.Type.String(),
+		Metric: p.Value,
+	})
+}
+
 func NewPathAttributeMultiExitDisc(value uint32) *PathAttributeMultiExitDisc {
 	return &PathAttributeMultiExitDisc{
 		PathAttribute: PathAttribute{
@@ -1382,6 +1432,16 @@ func (p *PathAttributeLocalPref) Serialize() ([]byte, error) {
 	return p.PathAttribute.Serialize()
 }
 
+func (p *PathAttributeLocalPref) MarshalJSON() ([]byte, error) {
+	return json.Marshal(struct {
+		Type string
+		Pref uint32
+	}{
+		Type: p.Type.String(),
+		Pref: p.Value,
+	})
+}
+
 func NewPathAttributeLocalPref(value uint32) *PathAttributeLocalPref {
 	return &PathAttributeLocalPref{
 		PathAttribute: PathAttribute{
@@ -1396,6 +1456,14 @@ type PathAttributeAtomicAggregate struct {
 	PathAttribute
 }
 
+func (p *PathAttributeAtomicAggregate) MarshalJSON() ([]byte, error) {
+	return json.Marshal(struct {
+		Type string
+	}{
+		Type: p.Type.String(),
+	})
+}
+
 func NewPathAttributeAtomicAggregate() *PathAttributeAtomicAggregate {
 	return &PathAttributeAtomicAggregate{
 		PathAttribute: PathAttribute{
@@ -1447,6 +1515,18 @@ func (p *PathAttributeAggregator) Serialize() ([]byte, error) {
 	return p.PathAttribute.Serialize()
 }
 
+func (p *PathAttributeAggregator) MarshalJSON() ([]byte, error) {
+	return json.Marshal(struct {
+		Type    string
+		AS      uint32
+		Address net.IP
+	}{
+		Type:    p.Type.String(),
+		AS:      p.Value.AS,
+		Address: p.Value.Address,
+	})
+}
+
 func NewPathAttributeAggregator(as interface{}, address string) *PathAttributeAggregator {
 	v := reflect.ValueOf(as)
 	return &PathAttributeAggregator{
@@ -1486,6 +1566,16 @@ func (p *PathAttributeCommunities) Serialize() ([]byte, error) {
 	return p.PathAttribute.Serialize()
 }
 
+func (p *PathAttributeCommunities) MarshalJSON() ([]byte, error) {
+	return json.Marshal(struct {
+		Type  string
+		Value []uint32
+	}{
+		Type:  p.Type.String(),
+		Value: p.Value,
+	})
+}
+
 func NewPathAttributeCommunities(value []uint32) *PathAttributeCommunities {
 	return &PathAttributeCommunities{
 		PathAttribute{BGP_ATTR_FLAG_OPTIONAL | BGP_ATTR_FLAG_TRANSITIVE, BGP_ATTR_TYPE_COMMUNITIES, 0, nil},
@@ -1511,6 +1601,16 @@ func (p *PathAttributeOriginatorId) Serialize() ([]byte, error) {
 	return p.PathAttribute.Serialize()
 }
 
+func (p *PathAttributeOriginatorId) MarshalJSON() ([]byte, error) {
+	return json.Marshal(struct {
+		Type    string
+		Address string
+	}{
+		Type:    p.Type.String(),
+		Address: p.Value.String(),
+	})
+}
+
 func NewPathAttributeOriginatorId(value string) *PathAttributeOriginatorId {
 	return &PathAttributeOriginatorId{
 		PathAttribute{BGP_ATTR_FLAG_OPTIONAL, BGP_ATTR_TYPE_ORIGINATOR_ID, 0, nil},
@@ -1542,6 +1642,21 @@ func (p *PathAttributeClusterList) Serialize() ([]byte, error) {
 	return p.PathAttribute.Serialize()
 }
 
+func (p *PathAttributeClusterList) MarshalJSON() ([]byte, error) {
+	l := make([]string, 0)
+	for _, addr := range p.Value {
+		l = append(l, addr.String())
+	}
+
+	return json.Marshal(struct {
+		Type    string
+		Address []string
+	}{
+		Type:    p.Type.String(),
+		Address: l,
+	})
+}
+
 func NewPathAttributeClusterList(value []string) *PathAttributeClusterList {
 	l := make([]net.IP, len(value))
 	for i, v := range value {