144 lines
4.8 KiB
Diff
144 lines
4.8 KiB
Diff
From 50833670dbebaa12337b3ce909ab95c55d5d8eca Mon Sep 17 00:00:00 2001
|
|
From: wanfeng <wanfeng@kylinos.cn>
|
|
Date: Wed, 19 Jun 2024 17:04:35 +0800
|
|
Subject: [PATCH] add IP_ADD_SOURCE_MEMBERSHIP to setsockopt for igmpv3
|
|
|
|
modify ip_mreq_source
|
|
|
|
add IP_ADD_SOURCE_MEMBERSHIP to setsockopt for igmpv3
|
|
|
|
add IP_ADD_SOURCE_MEMBERSHIP to setsockopt for igmpv3
|
|
---
|
|
src/api/sockets.c | 93 ++++++++++++++++++++++++++++++++++++++
|
|
src/include/lwip/sockets.h | 2 +-
|
|
2 files changed, 94 insertions(+), 1 deletion(-)
|
|
|
|
diff --git a/src/api/sockets.c b/src/api/sockets.c
|
|
index 5a72b62..a030f4b 100644
|
|
--- a/src/api/sockets.c
|
|
+++ b/src/api/sockets.c
|
|
@@ -3859,6 +3859,24 @@ lwip_setsockopt_impl(int s, int level, int optname, const void *optval, socklen_
|
|
return ENOPROTOOPT;
|
|
}
|
|
break;
|
|
+ case IP_ADD_SOURCE_MEMBERSHIP:
|
|
+ case IP_DROP_SOURCE_MEMBERSHIP:
|
|
+ LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, optlen, struct ip_mreq_source);
|
|
+#if LWIP_UDP
|
|
+ if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) == NETCONN_UDP) {
|
|
+ err = mcast_sock_add_drop_source_membership(s, &sock->conn->pcb.udp->ipmc, optname, (const struct ip_mreq_source *)optval);
|
|
+ } else
|
|
+#endif /* LWIP_UDP */
|
|
+#if LWIP_RAW
|
|
+ if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) == NETCONN_RAW) {
|
|
+ err = mcast_sock_add_drop_source_membership(s, &sock->conn->pcb.raw->ipmc, optname, (const struct ip_mreq_source *)optval);
|
|
+ } else
|
|
+#endif /* LWIP_RAW */
|
|
+ {
|
|
+ done_socket(sock);
|
|
+ return ENOPROTOOPT;
|
|
+ }
|
|
+ break;
|
|
case MCAST_JOIN_GROUP:
|
|
case MCAST_LEAVE_GROUP: {
|
|
LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, optlen, struct group_req);
|
|
@@ -4907,6 +4925,81 @@ out:
|
|
return (u8_t)err_to_errno(err);
|
|
}
|
|
|
|
+/**
|
|
+ * setsockopt() IP_ADD_SOURCE_MEMBERSHIP / IP_DROP_SOURCE_MEMBERSHIP command
|
|
+ */
|
|
+int
|
|
+mcast_sock_add_drop_source_membership(int s, struct ip_mc *ipmc, int optname, const struct ip_mreq_source *imr)
|
|
+{
|
|
+ struct netif *netif;
|
|
+ ip_addr_t if_addr;
|
|
+ ip_addr_t multi_addr;
|
|
+ ip_addr_t src_addr;
|
|
+ err_t err;
|
|
+ u8_t if_idx;
|
|
+
|
|
+ inet_addr_to_ip4addr(ip_2_ip4(&if_addr), &imr->imr_interface);
|
|
+ inet_addr_to_ip4addr(ip_2_ip4(&multi_addr), &imr->imr_multiaddr);
|
|
+ inet_addr_to_ip4addr(ip_2_ip4(&src_addr), &imr->imr_sourceaddr);
|
|
+ IP_SET_TYPE_VAL(if_addr, IPADDR_TYPE_V4);
|
|
+ IP_SET_TYPE_VAL(multi_addr, IPADDR_TYPE_V4);
|
|
+ IP_SET_TYPE_VAL(src_addr, IPADDR_TYPE_V4);
|
|
+
|
|
+ if (!ip4_addr_ismulticast(ip_2_ip4(&multi_addr)) || ip4_addr_isany(ip_2_ip4(&src_addr))) {
|
|
+ return EADDRNOTAVAIL;
|
|
+ }
|
|
+
|
|
+ if (ip4_addr_isany(ip_2_ip4(&if_addr))) { /* To all network interface */
|
|
+ if (optname == IP_ADD_SOURCE_MEMBERSHIP) {
|
|
+ NETIF_FOREACH(netif) {
|
|
+ if_idx = netif_get_index(netif);
|
|
+ if (!lwip_socket_register_membership(s, if_idx, ip_2_ip4(&multi_addr))) {
|
|
+ /* cannot track membership (out of memory) */
|
|
+ err = ENOMEM;
|
|
+ goto out;
|
|
+ }
|
|
+ }
|
|
+ err = mcast_join_group(ipmc, &if_addr, &multi_addr, &src_addr);
|
|
+ } else {
|
|
+ err = mcast_leave_group(ipmc, &if_addr, &multi_addr, &src_addr);
|
|
+ }
|
|
+
|
|
+ } else { /* To specified network interface */
|
|
+ if (!(ip_2_ip4(&if_addr)->addr & PP_HTONL(IP_CLASSA_NET))) { /* IS a BSD style index ? */
|
|
+ u8_t idx = (u8_t)PP_NTOHL(ip_2_ip4(&if_addr)->addr);
|
|
+ netif = netif_get_by_index(idx);
|
|
+ if (netif == NULL) {
|
|
+ return ENXIO;
|
|
+ }
|
|
+ *ip_2_ip4(&if_addr) = *netif_ip4_addr(netif);
|
|
+
|
|
+ } else {
|
|
+ NETIF_FOREACH(netif) {
|
|
+ if (ip4_addr_cmp(ip_2_ip4(&if_addr), netif_ip4_addr(netif))) {
|
|
+ break;
|
|
+ }
|
|
+ }
|
|
+ if (netif == NULL) {
|
|
+ return ENXIO;
|
|
+ }
|
|
+ }
|
|
+
|
|
+ if (optname == IP_ADD_SOURCE_MEMBERSHIP) {
|
|
+ if_idx = netif_get_index(netif);
|
|
+ if (!lwip_socket_register_membership(s, if_idx, ip_2_ip4(&multi_addr))) {
|
|
+ /* cannot track membership (out of memory) */
|
|
+ err = ENOMEM;
|
|
+ goto out;
|
|
+ }
|
|
+ err = mcast_join_netif(ipmc, netif, &multi_addr, &src_addr);
|
|
+ } else {
|
|
+ err = mcast_leave_netif(ipmc, netif, &multi_addr, &src_addr);
|
|
+ }
|
|
+ }
|
|
+out:
|
|
+ return err_to_errno(err);
|
|
+}
|
|
+
|
|
/**
|
|
* setsockopt() MCAST_JOIN_GROUP / MCAST_LEAVE_GROUP command
|
|
*/
|
|
diff --git a/src/include/lwip/sockets.h b/src/include/lwip/sockets.h
|
|
index f2a1b59..038afba 100644
|
|
--- a/src/include/lwip/sockets.h
|
|
+++ b/src/include/lwip/sockets.h
|
|
@@ -785,8 +785,8 @@ typedef struct ip_mreq {
|
|
*/
|
|
struct ip_mreq_source {
|
|
struct in_addr imr_multiaddr; /* IP multicast address of group */
|
|
- struct in_addr imr_sourceaddr; /* IP address of source */
|
|
struct in_addr imr_interface; /* local IP address of interface */
|
|
+ struct in_addr imr_sourceaddr; /* IP address of source */
|
|
};
|
|
|
|
#define IP_MSFILTER_SIZE(numsrc) \
|
|
--
|
|
2.25.1
|
|
|