diff options
Diffstat (limited to 'proto/bfd/packets.c')
| -rw-r--r-- | proto/bfd/packets.c | 27 |
1 files changed, 20 insertions, 7 deletions
diff --git a/proto/bfd/packets.c b/proto/bfd/packets.c index cb40bcda..b7a057f1 100644 --- a/proto/bfd/packets.c +++ b/proto/bfd/packets.c @@ -186,7 +186,7 @@ bfd_err_hook(sock *sk, int err) } sock * -bfd_open_rx_sk(struct bfd_proto *p, int multihop) +bfd_open_rx_sk(struct bfd_proto *p, int multihop, int inet_version) { sock *sk = sk_new(p->tpool); sk->type = SK_UDP; @@ -202,9 +202,18 @@ bfd_open_rx_sk(struct bfd_proto *p, int multihop) sk->priority = sk_priority_control; sk->flags = SKF_THREAD | SKF_LADDR_RX | (!multihop ? SKF_TTL_RX : 0); -#ifdef IPV6 - sk->flags |= SKF_V6ONLY; -#endif + switch (inet_version) { + case 4: + sk->fam = SK_FAM_IPV4; + sk->flags |= SKF_V4ONLY; + break; + case 6: + sk->fam = SK_FAM_IPV6; + sk->flags |= SKF_V6ONLY; + break; + default: + ASSERT(0); + } if (sk_open(sk) < 0) goto err; @@ -237,9 +246,13 @@ bfd_open_tx_sk(struct bfd_proto *p, ip_addr local, struct iface *ifa) sk->ttl = ifa ? 255 : -1; sk->flags = SKF_THREAD | SKF_BIND | SKF_HIGH_PORT; -#ifdef IPV6 - sk->flags |= SKF_V6ONLY; -#endif + if (ipa_is_ip4(local)) { + sk->fam = SK_FAM_IPV4; + sk->flags |= SKF_V4ONLY; + } else { + sk->fam = SK_FAM_IPV6; + sk->flags |= SKF_V6ONLY; + } if (sk_open(sk) < 0) goto err; |