add type and constants for AE
This commit is contained in:
parent
7283f52d41
commit
b110db1f5e
|
@ -0,0 +1,27 @@
|
|||
// Code generated by "stringer -type=AEType -trimprefix AE"; DO NOT EDIT.
|
||||
|
||||
package tlv
|
||||
|
||||
import "strconv"
|
||||
|
||||
func _() {
|
||||
// An "invalid array index" compiler error signifies that the constant values have changed.
|
||||
// Re-run the stringer command to generate them again.
|
||||
var x [1]struct{}
|
||||
_ = x[AEWildcard-0]
|
||||
_ = x[AEIPv4-1]
|
||||
_ = x[AEIPv6-2]
|
||||
_ = x[AEIPv6LL-3]
|
||||
_ = x[AEIPv4oIPv6-4]
|
||||
}
|
||||
|
||||
const _AEType_name = "WildcardIPv4IPv6IPv6LLIPv4oIPv6"
|
||||
|
||||
var _AEType_index = [...]uint8{0, 8, 12, 16, 22, 31}
|
||||
|
||||
func (i AEType) String() string {
|
||||
if i >= AEType(len(_AEType_index)-1) {
|
||||
return "AEType(" + strconv.FormatInt(int64(i), 10) + ")"
|
||||
}
|
||||
return _AEType_name[_AEType_index[i]:_AEType_index[i+1]]
|
||||
}
|
75
tlv/tlv.go
75
tlv/tlv.go
|
@ -1,5 +1,6 @@
|
|||
//go:generate stringer -type=Type -trimprefix Type
|
||||
//go:generate stringer -type=SubType -trimprefix SubType
|
||||
//go:generate stringer -type=AEType -trimprefix AE
|
||||
package tlv
|
||||
|
||||
import (
|
||||
|
@ -48,6 +49,33 @@ type SubTLV interface {
|
|||
L() uint8
|
||||
}
|
||||
|
||||
type AEType uint8
|
||||
|
||||
const (
|
||||
AEWildcard = AEType(0)
|
||||
AEIPv4 = AEType(1)
|
||||
AEIPv6 = AEType(2)
|
||||
AEIPv6LL = AEType(3)
|
||||
AEIPv4oIPv6 = AEType(4)
|
||||
)
|
||||
|
||||
func AEFromPrefix(p netaddr.IPPrefix) AEType {
|
||||
return AEFromIP(p.IP())
|
||||
}
|
||||
func AEFromIP(p netaddr.IP) AEType {
|
||||
switch {
|
||||
case p.IsZero():
|
||||
return AEWildcard
|
||||
case p.Is4():
|
||||
return AEIPv4
|
||||
case p.IsLinkLocalUnicast():
|
||||
return AEIPv6LL
|
||||
case p.Is6():
|
||||
return AEIPv6
|
||||
}
|
||||
panic("unknown AE")
|
||||
}
|
||||
|
||||
type Scanner struct {
|
||||
buf []byte
|
||||
err error
|
||||
|
@ -340,11 +368,9 @@ type Ack struct {
|
|||
func (Ack) T() Type {
|
||||
return TypeAck
|
||||
}
|
||||
|
||||
func (Ack) L() uint8 {
|
||||
return 4
|
||||
}
|
||||
|
||||
func AckFromBytes(b []byte) (Ack, []byte, error) {
|
||||
if err := assertLengthGreater(b, TypeAck, 2); err != nil {
|
||||
return Ack{}, b, err
|
||||
|
@ -389,7 +415,7 @@ type IHU struct {
|
|||
func (IHU) T() Type {
|
||||
return TypeIHU
|
||||
}
|
||||
func (i IHU) AE() uint8 {
|
||||
func (i IHU) AE() AEType {
|
||||
return AEFromIP(i.Address)
|
||||
}
|
||||
func (i IHU) L() uint8 {
|
||||
|
@ -413,7 +439,7 @@ func IHUFromBytes(b []byte) (IHU, []byte, error) {
|
|||
var ihu IHU
|
||||
var err error
|
||||
|
||||
ae := b[0]
|
||||
ae := AEType(b[0])
|
||||
ihu.Rxcost = uint16(b[2])<<8 | uint16(b[3])
|
||||
ihu.Interval = uint16(b[4])<<8 | uint16(b[5])
|
||||
ihu.Address, b, err = ipFromBytes(ae, b[6:])
|
||||
|
@ -447,15 +473,12 @@ func RouterIDFromBytes(b []byte) (RouterID, []byte, error) {
|
|||
|
||||
return rid, b, nil
|
||||
}
|
||||
|
||||
func (RouterID) T() Type {
|
||||
return TypeRouterID
|
||||
}
|
||||
|
||||
func (RouterID) L() uint8 {
|
||||
return 8
|
||||
}
|
||||
|
||||
func (r RouterID) String() string {
|
||||
var buf bytes.Buffer
|
||||
buf.Grow(8*3 - 1)
|
||||
|
@ -495,7 +518,7 @@ func NextHopFromBytes(b []byte) (NextHop, []byte, error) {
|
|||
|
||||
var nh NextHop
|
||||
var err error
|
||||
ae := b[0]
|
||||
ae := AEType(b[0])
|
||||
nh.Address, b, err = ipFromBytes(ae, b[2:])
|
||||
return nh, b, err
|
||||
}
|
||||
|
@ -516,17 +539,14 @@ type Update struct {
|
|||
func (Update) T() Type {
|
||||
return TypeUpdate
|
||||
}
|
||||
|
||||
func (u Update) L() uint8 {
|
||||
return 10 + psizeFromPlen(u.Prefix.Bits())
|
||||
}
|
||||
|
||||
func (u Update) FormatHeader() string {
|
||||
return fmt.Sprintf("Flags 0x%02x Omitted %2d Interval %4d Seqno %5d Metric %5d RouterID %s",
|
||||
u.Flags, u.Omitted, u.Interval, u.Seqno, u.Metric, u.RouterID,
|
||||
)
|
||||
}
|
||||
|
||||
func (s *PacketDecoder) UpdateFromBytes(b []byte) (Update, []byte, error) {
|
||||
if err := assertLengthGreater(b, TypeUpdate, 10); err != nil {
|
||||
return Update{}, b, err
|
||||
|
@ -580,32 +600,13 @@ func (s *PacketDecoder) UpdateFromBytes(b []byte) (Update, []byte, error) {
|
|||
return u, b, err
|
||||
}
|
||||
|
||||
func AEFromPrefix(p netaddr.IPPrefix) uint8 {
|
||||
return AEFromIP(p.IP())
|
||||
}
|
||||
func AEFromIP(p netaddr.IP) uint8 {
|
||||
if p.IsZero() {
|
||||
return 0
|
||||
}
|
||||
if p.Is4() {
|
||||
return 1
|
||||
}
|
||||
if p.IsLinkLocalUnicast() {
|
||||
return 3
|
||||
}
|
||||
if p.Is6() {
|
||||
return 2
|
||||
}
|
||||
panic("unknown AE")
|
||||
}
|
||||
|
||||
type RouteRequest struct {
|
||||
// AE uint8
|
||||
// plen uint8
|
||||
Prefix netaddr.IPPrefix
|
||||
}
|
||||
|
||||
func (r RouteRequest) AE() uint8 {
|
||||
func (r RouteRequest) AE() AEType {
|
||||
return AEFromPrefix(r.Prefix)
|
||||
}
|
||||
func (RouteRequest) T() Type {
|
||||
|
@ -734,25 +735,25 @@ func prefixV6LL(b []byte) (netaddr.IPPrefix, []byte, error) {
|
|||
copy(ip6ll[8:], b[:8])
|
||||
return netaddr.IPPrefixFrom(netaddr.IPv6Raw(ip6ll), 8), b[8:], nil
|
||||
}
|
||||
func ipFromBytes(ae uint8, b []byte) (netaddr.IP, []byte, error) {
|
||||
func ipFromBytes(ae AEType, b []byte) (netaddr.IP, []byte, error) {
|
||||
switch ae {
|
||||
case 0:
|
||||
case AEWildcard:
|
||||
return netaddr.IP{}, b, nil
|
||||
case 1:
|
||||
case AEIPv4:
|
||||
if len(b) < 4 {
|
||||
return netaddr.IP{}, b, fmt.Errorf("Not enough bytes for v4 address: %d", len(b))
|
||||
}
|
||||
var ip4 [4]byte
|
||||
copy(ip4[:], b[:4])
|
||||
return netaddr.IPFrom4(ip4), b[4:], nil
|
||||
case 2:
|
||||
case AEIPv6:
|
||||
if len(b) < 16 {
|
||||
return netaddr.IP{}, b, fmt.Errorf("Not enough bytes for v6 address: %d", len(b))
|
||||
}
|
||||
var ip6 [16]byte
|
||||
copy(ip6[:], b[:16])
|
||||
return netaddr.IPv6Raw(ip6), b[16:], nil
|
||||
case 3:
|
||||
case AEIPv6LL:
|
||||
if len(b) < 8 {
|
||||
return netaddr.IP{}, b, fmt.Errorf("Not enough bytes for v6ll address: %d", len(b))
|
||||
}
|
||||
|
@ -761,6 +762,8 @@ func ipFromBytes(ae uint8, b []byte) (netaddr.IP, []byte, error) {
|
|||
ip6ll[1] = 0x80
|
||||
copy(ip6ll[8:], b[:8])
|
||||
return netaddr.IPv6Raw(ip6ll), b[8:], nil
|
||||
case AEIPv4oIPv6:
|
||||
return netaddr.IP{}, b, fmt.Errorf("Not implemented AE Type %s", ae)
|
||||
default:
|
||||
return netaddr.IP{}, b, fmt.Errorf("Invalid AE %d", ae)
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue