summaryrefslogtreecommitdiffhomepage
path: root/packet/bgp/bgp.go
diff options
context:
space:
mode:
Diffstat (limited to 'packet/bgp/bgp.go')
-rw-r--r--packet/bgp/bgp.go18
1 files changed, 12 insertions, 6 deletions
diff --git a/packet/bgp/bgp.go b/packet/bgp/bgp.go
index f309f8dd..b74aa838 100644
--- a/packet/bgp/bgp.go
+++ b/packet/bgp/bgp.go
@@ -362,12 +362,18 @@ func (c *CapGracefulRestart) DecodeFromBytes(data []byte) error {
c.Flags = uint8(restart >> 12)
c.Time = restart & 0xfff
data = data[2:]
- c.Tuples = make([]*CapGracefulRestartTuple, 0, len(data)/4)
- for len(data) >= 4 {
- t := &CapGracefulRestartTuple{binary.BigEndian.Uint16(data[0:2]),
- data[2], data[3]}
- c.Tuples = append(c.Tuples, t)
- data = data[4:]
+
+ valueLen := int(c.CapLen) - 2
+
+ if valueLen >= 4 && len(data) >= valueLen {
+ c.Tuples = make([]*CapGracefulRestartTuple, 0, valueLen/4)
+
+ for i := valueLen; i >= 4; i -= 4 {
+ t := &CapGracefulRestartTuple{binary.BigEndian.Uint16(data[0:2]),
+ data[2], data[3]}
+ c.Tuples = append(c.Tuples, t)
+ data = data[4:]
+ }
}
return nil
}