99ee94728a
refer to https://man7.org/linux/man-pages/man7/ip.7.html IP_MULTICAST_IF (since Linux 1.2) Set the local device for a multicast socket. The argument for setsockopt(2) is an ip_mreqn or (since Linux 3.5) ip_mreq structure similar to IP_ADD_MEMBERSHIP, or an in_addr structure. (The kernel determines which structure is being passed based on the size passed in optlen.) For getsockopt(2), the argument is an in_addr structure. refer to https://man7.org/linux/man-pages/man7/ipv6.7.html IPV6_MULTICAST_IF Set the device for outgoing multicast packets on the socket. This is allowed only for SOCK_DGRAM and SOCK_RAW socket. The argument is a pointer to an interface index (see netdevice(7)) in an integer. testcase1: TEST_IMPL(udp_multicast_interface) { /* TODO(gengjiawen): Fix test on QEMU. */ RETURN_SKIP("Test does not currently work in QEMU"); int r; uv_udp_send_t req; uv_buf_t buf; struct sockaddr_in addr; struct sockaddr_in baddr; close_cb_called = 0; sv_send_cb_called = 0; ASSERT(0 == uv_ip4_addr("239.255.0.1", TEST_PORT, &addr)); r = uv_udp_init(uv_default_loop(), &server); ASSERT(r == 0); ASSERT(0 == uv_ip4_addr("0.0.0.0", 0, &baddr)); r = uv_udp_bind(&server, (const struct sockaddr*)&baddr, 0); ASSERT(r == 0); r = uv_udp_set_multicast_interface(&server, "0.0.0.0"); ASSERT(r == 0); /* server sends "PING" */ buf = uv_buf_init("PING", 4); r = uv_udp_send(&req, &server, &buf, 1, (const struct sockaddr*)&addr, sv_send_cb); ASSERT(r == 0); ASSERT(close_cb_called == 0); ASSERT(sv_send_cb_called == 0); /* run the loop till all events are processed */ uv_run(uv_default_loop(), UV_RUN_DEFAULT); ASSERT(sv_send_cb_called == 1); ASSERT(close_cb_called == 1); ASSERT(client.send_queue_size == 0); ASSERT(server.send_queue_size == 0); MAKE_VALGRIND_HAPPY(); return 0; } testcase2: TEST_IMPL(udp_multicast_interface6) { /* TODO(gengjiawen): Fix test on QEMU. */ RETURN_SKIP("Test does not currently work in QEMU"); int r; uv_udp_send_t req; uv_buf_t buf; struct sockaddr_in6 addr; struct sockaddr_in6 baddr; if (!can_ipv6()) RETURN_SKIP("IPv6 not supported"); close_cb_called = 0; sv_send_cb_called = 0; ASSERT(0 == uv_ip6_addr("::1", TEST_PORT, &addr)); r = uv_udp_init(uv_default_loop(), &server); ASSERT(r == 0); ASSERT(0 == uv_ip6_addr("::", 0, &baddr)); r = uv_udp_bind(&server, (const struct sockaddr*)&baddr, 0); ASSERT(r == 0); r = uv_udp_set_multicast_interface(&server, "::1%lo0"); r = uv_udp_set_multicast_interface(&server, NULL); ASSERT(r == 0); /* server sends "PING" */ buf = uv_buf_init("PING", 4); r = uv_udp_send(&req, &server, &buf, 1, (const struct sockaddr*)&addr, sv_send_cb); ASSERT(r == 0); ASSERT(close_cb_called == 0); ASSERT(sv_send_cb_called == 0); /* run the loop till all events are processed */ uv_run(uv_default_loop(), UV_RUN_DEFAULT); ASSERT(sv_send_cb_called == 1); ASSERT(close_cb_called == 1); MAKE_VALGRIND_HAPPY(); return 0; } Signed-off-by: wangchen <wangchen41@xiaomi.com>
258 lines
8.0 KiB
C
258 lines
8.0 KiB
C
/****************************************************************************
|
|
* net/udp/udp_finddev.c
|
|
*
|
|
* Licensed to the Apache Software Foundation (ASF) under one or more
|
|
* contributor license agreements. See the NOTICE file distributed with
|
|
* this work for additional information regarding copyright ownership. The
|
|
* ASF licenses this file to you under the Apache License, Version 2.0 (the
|
|
* "License"); you may not use this file except in compliance with the
|
|
* License. You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
* License for the specific language governing permissions and limitations
|
|
* under the License.
|
|
*
|
|
****************************************************************************/
|
|
|
|
/****************************************************************************
|
|
* Included Files
|
|
****************************************************************************/
|
|
|
|
#include <nuttx/config.h>
|
|
#if defined(CONFIG_NET) && defined(CONFIG_NET_UDP)
|
|
|
|
#include <string.h>
|
|
|
|
#include <nuttx/net/netdev.h>
|
|
#include <nuttx/net/ip.h>
|
|
|
|
#include "netdev/netdev.h"
|
|
#include "inet/inet.h"
|
|
#include "udp/udp.h"
|
|
#include "utils/utils.h"
|
|
|
|
/****************************************************************************
|
|
* Public Functions
|
|
****************************************************************************/
|
|
|
|
/****************************************************************************
|
|
* Name: udp_find_laddr_device
|
|
*
|
|
* Description:
|
|
* Select the network driver to use with the UDP transaction using the
|
|
* locally bound IP address.
|
|
*
|
|
* This is currently used in the UDP network poll setup to determine
|
|
* which device is being polled.
|
|
*
|
|
* Input Parameters:
|
|
* conn - UDP connection structure (not currently used).
|
|
*
|
|
* Returned Value:
|
|
* A pointer to the network driver to use. NULL is returned if driver is
|
|
* not bound to any local device.
|
|
*
|
|
****************************************************************************/
|
|
|
|
FAR struct net_driver_s *udp_find_laddr_device(FAR struct udp_conn_s *conn)
|
|
{
|
|
/* There are multiple network devices. We need to select the device that
|
|
* is going to route the UDP packet based on the provided IP address.
|
|
*/
|
|
|
|
#ifdef CONFIG_NET_IPv4
|
|
#ifdef CONFIG_NET_IPv6
|
|
if (conn->domain == PF_INET)
|
|
#endif
|
|
{
|
|
/* Make sure that the socket is bound to some non-zero, local
|
|
* address. Zero is used as an indication that the laddr is
|
|
* uninitialized and that the socket is, hence, not bound.
|
|
*/
|
|
|
|
if (conn->u.ipv4.laddr == 0)
|
|
{
|
|
return NULL;
|
|
}
|
|
else
|
|
{
|
|
return netdev_findby_ripv4addr(conn->u.ipv4.laddr,
|
|
conn->u.ipv4.laddr);
|
|
}
|
|
}
|
|
#endif
|
|
|
|
#ifdef CONFIG_NET_IPv6
|
|
#ifdef CONFIG_NET_IPv4
|
|
else
|
|
#endif
|
|
{
|
|
/* Make sure that the socket is bound to some non-zero, local
|
|
* address. The IPv6 unspecified address is used as an indication
|
|
* that the laddr is uninitialized and that the socket is, hence,
|
|
* not bound.
|
|
*/
|
|
|
|
if (net_ipv6addr_cmp(conn->u.ipv6.laddr, g_ipv6_unspecaddr))
|
|
{
|
|
return NULL;
|
|
}
|
|
else
|
|
{
|
|
return netdev_findby_ripv6addr(conn->u.ipv6.laddr,
|
|
conn->u.ipv6.laddr);
|
|
}
|
|
}
|
|
#endif
|
|
}
|
|
|
|
/****************************************************************************
|
|
* Name: udp_find_raddr_device
|
|
*
|
|
* Description:
|
|
* Select the network driver to use with the UDP transaction using the
|
|
* remote IP address.
|
|
*
|
|
* This function is called for UDP sendto() in order to determine which
|
|
* network device that the UDP pack should be sent on.
|
|
*
|
|
* Input Parameters:
|
|
* conn - UDP connection structure.
|
|
*
|
|
* Returned Value:
|
|
* A pointer to the network driver to use.
|
|
*
|
|
****************************************************************************/
|
|
|
|
FAR struct net_driver_s *
|
|
udp_find_raddr_device(FAR struct udp_conn_s *conn,
|
|
FAR struct sockaddr_storage *remote)
|
|
{
|
|
/* We need to select the device that is going to route the UDP packet
|
|
* based on the provided IP address.
|
|
*/
|
|
|
|
#ifdef CONFIG_NET_IPv4
|
|
#ifdef CONFIG_NET_IPv6
|
|
if (conn->domain == PF_INET)
|
|
#endif
|
|
{
|
|
in_addr_t raddr;
|
|
|
|
if (remote)
|
|
{
|
|
FAR const struct sockaddr_in *inaddr =
|
|
(FAR const struct sockaddr_in *)remote;
|
|
net_ipv4addr_copy(raddr, inaddr->sin_addr.s_addr);
|
|
}
|
|
else
|
|
{
|
|
net_ipv4addr_copy(raddr, conn->u.ipv4.raddr);
|
|
}
|
|
|
|
#if defined(CONFIG_NET_IGMP) && defined(CONFIG_NET_BINDTODEVICE)
|
|
if (IN_MULTICAST(NTOHL(raddr)))
|
|
{
|
|
if ((conn->sconn.s_boundto == 0) &&
|
|
(conn->mreq.imr_ifindex != 0))
|
|
{
|
|
return netdev_findbyindex(conn->mreq.imr_ifindex);
|
|
}
|
|
}
|
|
else
|
|
#endif
|
|
{
|
|
if (conn->u.ipv4.laddr != INADDR_ANY)
|
|
{
|
|
/* If the socket is bound to some non-zero, local address.
|
|
* Normal lookup using the verified local address.
|
|
*/
|
|
|
|
return netdev_findby_lipv4addr(conn->u.ipv4.laddr);
|
|
}
|
|
|
|
#ifdef CONFIG_NET_BINDTODEVICE
|
|
if (conn->sconn.s_boundto != 0)
|
|
{
|
|
/* If the socket is bound to a local network device.
|
|
* Select the network device that has been bound.
|
|
* If the index is invalid, return NULL.
|
|
*/
|
|
|
|
return netdev_findbyindex(conn->sconn.s_boundto);
|
|
}
|
|
#endif
|
|
}
|
|
|
|
/* Normal lookup using the verified remote address */
|
|
|
|
return netdev_findby_ripv4addr(conn->u.ipv4.laddr, raddr);
|
|
}
|
|
#endif
|
|
|
|
#ifdef CONFIG_NET_IPv6
|
|
#ifdef CONFIG_NET_IPv4
|
|
else
|
|
#endif
|
|
{
|
|
struct in6_addr raddr;
|
|
if (remote)
|
|
{
|
|
FAR const struct sockaddr_in6 *inaddr =
|
|
(FAR const struct sockaddr_in6 *)remote;
|
|
net_ipv6addr_copy(raddr.in6_u.u6_addr16,
|
|
inaddr->sin6_addr.s6_addr16);
|
|
}
|
|
else
|
|
{
|
|
net_ipv6addr_copy(raddr.in6_u.u6_addr16, conn->u.ipv6.raddr);
|
|
}
|
|
|
|
#if defined(CONFIG_NET_MLD) && defined(CONFIG_NET_BINDTODEVICE)
|
|
if (IN6_IS_ADDR_MULTICAST(&raddr))
|
|
{
|
|
if ((conn->sconn.s_boundto == 0) &&
|
|
(conn->mreq.imr_ifindex != 0))
|
|
{
|
|
return netdev_findbyindex(conn->mreq.imr_ifindex);
|
|
}
|
|
}
|
|
else
|
|
#endif
|
|
{
|
|
if (!net_ipv6addr_cmp(conn->u.ipv6.laddr, g_ipv6_unspecaddr))
|
|
{
|
|
/* If the socket is bound to some non-zero, local address.
|
|
* Normal lookup using the verified local address.
|
|
*/
|
|
|
|
return netdev_findby_lipv6addr(conn->u.ipv6.laddr);
|
|
}
|
|
|
|
#ifdef CONFIG_NET_BINDTODEVICE
|
|
if (conn->sconn.s_boundto != 0)
|
|
{
|
|
/* If the socket is bound to a local network device.
|
|
* Select the network device that has been bound.
|
|
* If the index is invalid, return NULL.
|
|
*/
|
|
|
|
return netdev_findbyindex(conn->sconn.s_boundto);
|
|
}
|
|
#endif
|
|
}
|
|
|
|
/* Normal lookup using the verified remote address */
|
|
|
|
return netdev_findby_ripv6addr(conn->u.ipv6.laddr,
|
|
raddr.in6_u.u6_addr16);
|
|
}
|
|
#endif
|
|
}
|
|
|
|
#endif /* CONFIG_NET && CONFIG_NET_UDP */
|