diff options
Diffstat (limited to 'cmd/vrouter')
| -rw-r--r-- | cmd/vrouter/main.go | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/cmd/vrouter/main.go b/cmd/vrouter/main.go index ba7f285..59e2816 100644 --- a/cmd/vrouter/main.go +++ b/cmd/vrouter/main.go @@ -72,9 +72,15 @@ func main() { messageToSendBytes := []byte(messageToSend) hop, err := ipstack.LongestPrefix(netip.MustParseAddr(ipAddr)) + if err != nil { + fmt.Println(err) + continue + } myAddr := hop.Interface.IpPrefix.Addr() for _, neighbor := range ipstack.GetNeighbors()[hop.Interface.Name] { - if neighbor.VipAddr == netip.MustParseAddr(ipAddr) { + if neighbor.VipAddr == hop.VIP || + neighbor.VipAddr == netip.MustParseAddr(ipAddr) || + hop.Type == "S" { err = ipstack.SendIP(&myAddr, neighbor, ipstack.TEST_PROTOCOL, messageToSendBytes, ipAddr, nil) if err != nil { fmt.Println(err) |