const: fix constant map and add ADDR constants

This commit is contained in:
Jeroen Simonetti 2017-11-30 00:10:35 +01:00
parent 0ec3c1fdfc
commit 6ffa5173f6
2 changed files with 11 additions and 8 deletions

View File

@ -7,6 +7,7 @@ import (
"github.com/mdlayher/netlink"
"github.com/mdlayher/netlink/nlenc"
"syscall"
)
var (
@ -100,15 +101,15 @@ type AddressService struct {
// Constants used to request information from rtnetlink addresses.
const (
rtmNewAddress = 20
rtmDelAddress = 21
rtmGetAddress = 22
RTM_NEWADDR = syscall.RTM_NEWADDR
RTM_DELADDR = syscall.RTM_DELADDR
RTM_GETADDR = syscall.RTM_GETADDR
)
// New creates a new address using the AddressMessage information.
func (a *AddressService) New(req *AddressMessage) error {
flags := netlink.HeaderFlagsRequest
_, err := a.c.Send(req, rtmNewAddress, flags)
_, err := a.c.Send(req, RTM_NEWADDR, flags)
if err != nil {
return err
}
@ -126,7 +127,7 @@ func (a *AddressService) Delete(address net.IP, index uint32) error {
}
flags := netlink.HeaderFlagsRequest
_, err := a.c.Send(req, rtmDelAddress, flags)
_, err := a.c.Send(req, RTM_DELADDR, flags)
if err != nil {
return err
}
@ -139,7 +140,7 @@ func (a *AddressService) List() ([]AddressMessage, error) {
req := &AddressMessage{}
flags := netlink.HeaderFlagsRequest | netlink.HeaderFlagsDump
msgs, err := a.c.Execute(req, rtmGetAddress, flags)
msgs, err := a.c.Execute(req, RTM_GETADDR, flags)
if err != nil {
return nil, err
}

View File

@ -104,10 +104,12 @@ func messageUnmarshall(msgs []netlink.Message) ([]Message, []netlink.Message, er
var m Message
switch nm.Header.Type {
case RTM_GETLINK: fallthrough
case RTM_NEWLINK: fallthrough
case RTM_DELLINK:
m = &LinkMessage{}
case rtmNewAddress: fallthrough
case rtmDelAddress:
case RTM_GETADDR: fallthrough
case RTM_NEWADDR: fallthrough
case RTM_DELADDR:
m = &AddressMessage{}
default:
continue