diff options
| author | Andreas Schneider <asn@cryptomilk.org> | 2014-05-13 16:01:25 +0200 |
|---|---|---|
| committer | Michael Adam <obnox@samba.org> | 2014-05-22 12:05:37 +0200 |
| commit | 37dfa9ba9484488995b4338b6e955b54ea1912aa (patch) | |
| tree | 002c25793a4ea395d52f02f6a9bb10460cbbd19e | |
| parent | c609a8a5547d0d1edc17069bcaf22bbb63802c3b (diff) | |
| download | socket_wrapper-37dfa9ba9484488995b4338b6e955b54ea1912aa.tar.gz socket_wrapper-37dfa9ba9484488995b4338b6e955b54ea1912aa.tar.xz socket_wrapper-37dfa9ba9484488995b4338b6e955b54ea1912aa.zip | |
swrap: Add swrap_msghdr_add_pktinfo().
Pair-Programmed-With: Stefan Metzmacher <metze@samba.org>
Pair-Programmed-With: Michael Adam <obnox@samba.org>
Signed-off-by: Andreas Schneider <asn@samba.org>
Signed-off-by: Stefan Metzmacher <metze@samba.org>
Signed-off-by: Michael Adam <obnox@samba.org>
| -rw-r--r-- | src/socket_wrapper.c | 65 |
1 files changed, 65 insertions, 0 deletions
diff --git a/src/socket_wrapper.c b/src/socket_wrapper.c index d382459..288ca13 100644 --- a/src/socket_wrapper.c +++ b/src/socket_wrapper.c @@ -3033,6 +3033,71 @@ static void swrap_msghdr_add_cmsghdr(struct msghdr *msg, return; } + +static int swrap_msghdr_add_pktinfo(struct socket_info *si, + struct msghdr *msg) +{ + /* Add packet info */ + switch (si->pktinfo) { +#ifdef IP_PKTINFO + case AF_INET: { + struct sockaddr_in *sin; + struct in_pktinfo pkt; + + if (si->bindname_len == sizeof(struct sockaddr_in)) { + sin = (struct sockaddr_in*)si->bindname; + } else { + if (si->myname_len != sizeof(struct sockaddr_in)) { + return 0; + } + sin = (struct sockaddr_in*)si->myname; + } + + ZERO_STRUCT(pkt); + + pkt.ipi_ifindex = socket_wrapper_default_iface(); + pkt.ipi_addr.s_addr = sin->sin_addr.s_addr; + + swrap_msghdr_add_cmsghdr(msg, IPPROTO_IP, IP_PKTINFO, + &pkt, sizeof(pkt)); + + break; + } +#endif /* IP_PKTINFO */ +#if defined(HAVE_IPV6) + case AF_INET6: { +#if defined(IPV6_PKTINFO) && defined(HAVE_STRUCT_IN6_PKTINFO) + struct sockaddr_in6 *sin6; + struct in6_pktinfo pkt6; + + if (si->bindname_len == sizeof(struct sockaddr_in6)) { + sin6 = (struct sockaddr_in6*)si->bindname; + } else { + if (si->myname_len != sizeof(struct sockaddr_in6)) { + return 0; + } + sin6 = (struct sockaddr_in6*)si->myname; + } + + ZERO_STRUCT(pkt6); + + pkt6.ipi6_ifindex = socket_wrapper_default_iface(); + pkt6.ipi6_addr = sin6->sin6_addr; + + swrap_msghdr_add_cmsghdr(msg, IPPROTO_IPV6, IPV6_PKTINFO, + &pkt6, sizeof(pkt6)); +#endif /* HAVE_STRUCT_IN6_PKTINFO */ + + break; + } +#endif /* IPV6_PKTINFO */ + default: + return -1; + } + + return 0; +} + #endif /* HAVE_STRUCT_MSGHDR_MSG_CONTROL */ static ssize_t swrap_sendmsg_before(int fd, |