diff options
Diffstat (limited to 'internal/netlink/rtnl.go')
| -rw-r--r-- | internal/netlink/rtnl.go | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/internal/netlink/rtnl.go b/internal/netlink/rtnl.go index 2f51fe58..243bc27e 100644 --- a/internal/netlink/rtnl.go +++ b/internal/netlink/rtnl.go @@ -19,8 +19,12 @@ func DialRoute() (*RouteConn, error) { } // rtnlConsume consumes a message from rtnetlink. -func rtnlConsume(resp []syscall.NetlinkMessage) error { +func (c *conn) rtnlConsume(resp []syscall.NetlinkMessage) error { for i := range resp { + if err := c.checkReply(&resp[i].Header); err != nil { + return err + } + switch resp[i].Header.Type { case syscall.NLMSG_DONE: return Complete{} @@ -81,7 +85,7 @@ func (c *RouteConn) SendIfAddrmsg( if !c.writeIfAddrmsg(typ, flags, msg, attrs...) { return syscall.ENOMEM } - return c.Roundtrip(ctx, rtnlConsume) + return c.Roundtrip(ctx, c.conn.rtnlConsume) } // writeNewaddrLo writes a RTM_NEWADDR message for the loopback address. @@ -110,7 +114,7 @@ func (c *RouteConn) SendNewaddrLo(ctx context.Context, lo uint32) error { if !c.writeNewaddrLo(lo) { return syscall.ENOMEM } - return c.Roundtrip(ctx, rtnlConsume) + return c.Roundtrip(ctx, c.conn.rtnlConsume) } // writeIfInfomsg writes an ifinfomsg structure to conn. @@ -131,5 +135,5 @@ func (c *RouteConn) SendIfInfomsg( if !c.writeIfInfomsg(typ, flags, msg) { return syscall.ENOMEM } - return c.Roundtrip(ctx, rtnlConsume) + return c.Roundtrip(ctx, c.conn.rtnlConsume) } |