nuttx/net/inet/ipv4_setsockopt.c
梁超众 5012195bde support ipv4 ToS and ipv6 TrafficClass
Signed-off-by: 梁超众 <liangchaozhong@xiaomi.com>
2023-01-29 13:43:44 +08:00

302 lines
9.6 KiB
C

/****************************************************************************
* net/inet/ipv4_setsockopt.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>
#include <sys/types.h>
#include <assert.h>
#include <errno.h>
#include <debug.h>
#include <nuttx/net/net.h>
#include <netinet/in.h>
#include "netdev/netdev.h"
#include "netfilter/iptables.h"
#include "igmp/igmp.h"
#include "inet/inet.h"
#include "udp/udp.h"
#ifdef CONFIG_NET_IPv4
/****************************************************************************
* Public Functions
****************************************************************************/
/****************************************************************************
* Name: ipv4_setsockopt
*
* Description:
* ipv4_setsockopt() sets the IPv4-protocol socket option specified by the
* 'option' argument to the value pointed to by the 'value' argument for
* the socket specified by the 'psock' argument.
*
* See <netinet/in.h> for the a complete list of values of IPv4 protocol
* socket options.
*
* Input Parameters:
* psock Socket structure of socket to operate on
* option identifies the option to set
* value Points to the argument value
* value_len The length of the argument value
*
* Returned Value:
* Returns zero (OK) on success. On failure, it returns a negated errno
* value to indicate the nature of the error. See psock_setcockopt() for
* the list of possible error values.
*
****************************************************************************/
int ipv4_setsockopt(FAR struct socket *psock, int option,
FAR const void *value, socklen_t value_len)
{
int ret;
ninfo("option: %d\n", option);
/* With IPv4, the multicast-related socket options are simply an
* alternative way to access IGMP. That IGMP functionality can also be
* accessed via IOCTL commands (see netdev/netdev_ioctl.c)
*
* REVISIT: Clone the logic from netdev_ioctl.c here.
*/
net_lock();
switch (option)
{
#ifdef CONFIG_NET_IGMP
case IP_MSFILTER: /* Access advanced, full-state filtering API */
{
#if 0 /* REVISIT: This is not a proper implementation of IP_MSGFILTER */
FAR const struct ip_msfilter *imsf;
FAR struct net_driver_s *dev;
imsf = (FAR const struct ip_msfilter *)value;
if (imsf == NULL || value_len < sizeof(struct ip_msfilter))
{
nerr("ERROR: Bad value or value_len\n");
ret = -EINVAL;
}
else
{
/* Get the device associated with the local interface address */
dev = netdev_findby_lipv4addr(imsf->imsf_interface.s_addr);
if (dev == NULL)
{
nwarn("WARNING: Could not find device\n");
ret = -ENODEV;
}
else if (imsf->imsf_fmode == MCAST_INCLUDE)
{
ret = igmp_joingroup(dev, &imsf->imsf_multiaddr);
}
else
{
DEBUGASSERT(imsf->imsf_fmode == MCAST_EXCLUDE);
ret = igmp_leavegroup(dev, &imsf->imsf_multiaddr);
}
}
#else
ret = -ENOSYS;
#endif
}
break;
case IP_ADD_MEMBERSHIP: /* Join a multicast group */
case IP_DROP_MEMBERSHIP: /* Leave a multicast group */
{
FAR const struct ip_mreq *mrec;
FAR struct net_driver_s *dev;
/* REVISIT: This is not a proper implementation of IP_MSGFILTER */
mrec = (FAR const struct ip_mreq *)value;
if (mrec == NULL || value_len < sizeof(struct ip_mreq))
{
nerr("ERROR: Bad value or value_len\n");
ret = -EINVAL;
}
else
{
/* Use the default network device is imr_interface is
* INADDRY_ANY.
*/
if (mrec->imr_interface.s_addr == INADDR_ANY)
{
dev = netdev_default();
}
else
{
/* Get the device associated with the local interface
* address
*/
dev = netdev_findby_lipv4addr(mrec->imr_interface.s_addr);
}
if (dev == NULL)
{
nwarn("WARNING: Could not find device\n");
ret = -ENODEV;
}
else if (option == IP_ADD_MEMBERSHIP)
{
ret = igmp_joingroup(dev, &mrec->imr_multiaddr);
}
else
{
ret = igmp_leavegroup(dev, &mrec->imr_multiaddr);
}
}
}
break;
#if defined(CONFIG_NET_UDP) && !defined(CONFIG_NET_UDP_NO_STACK)
case IP_MULTICAST_TTL: /* Set/read the time-to-live value of
* outgoing multicast packets */
{
FAR struct udp_conn_s *conn;
int ttl;
if (psock->s_type != SOCK_DGRAM ||
value == NULL || value_len == 0)
{
ret = -EINVAL;
break;
}
ttl = (value_len >= sizeof(int)) ?
*(FAR int *)value : (int)*(FAR unsigned char *)value;
if (ttl <= 0 || ttl > 255)
{
ret = -EINVAL;
}
else
{
conn = (FAR struct udp_conn_s *)psock->s_conn;
conn->ttl = ttl;
ret = OK;
}
}
break;
#endif
/* The following IPv4 socket options are defined, but not implemented */
case IP_MULTICAST_IF: /* Set local device for a multicast
* socket */
case IP_MULTICAST_LOOP: /* Set/read boolean that determines
* whether sent multicast packets
* should be looped back to local
* sockets. */
case IP_UNBLOCK_SOURCE: /* Unblock previously blocked multicast
* source */
case IP_BLOCK_SOURCE: /* Stop receiving multicast data from
* source */
case IP_ADD_SOURCE_MEMBERSHIP: /* Join a multicast group; allow receive
* only from source */
case IP_DROP_SOURCE_MEMBERSHIP: /* Leave a source-specific group. Stop
* receiving data from a given multicast
* group that come from a given source */
case IP_MULTICAST_ALL: /* Modify the delivery policy of
* multicast messages bound to
* INADDR_ANY */
#warning Missing logic
nwarn("WARNING: Unimplemented IPv4 option: %d\n", option);
ret = -ENOSYS;
break;
#endif /* CONFIG_NET_IGMP */
#if defined(CONFIG_NET_UDP) && !defined(CONFIG_NET_UDP_NO_STACK)
case IP_PKTINFO:
{
FAR struct udp_conn_s *conn;
int enable;
if (psock->s_type != SOCK_DGRAM ||
value == NULL || value_len == 0)
{
ret = -EINVAL;
break;
}
enable = (value_len >= sizeof(int)) ?
*(FAR int *)value : (int)*(FAR unsigned char *)value;
conn = (FAR struct udp_conn_s *)psock->s_conn;
if (enable)
{
conn->flags |= _UDP_FLAG_PKTINFO;
}
else
{
conn->flags &= ~_UDP_FLAG_PKTINFO;
}
ret = OK;
}
break;
#endif
case IP_TOS:
{
FAR struct socket_conn_s *conn =
(FAR struct socket_conn_s *)psock->s_conn;
int tos;
tos = (value_len >= sizeof(int)) ?
*(FAR int *)value : (int)*(FAR unsigned char *)value;
if (tos < 0 || tos > 0xff)
{
nerr("ERROR: invalid tos:%d\n", tos);
ret = -EINVAL;
}
else
{
conn->s_tos = tos;
ret = OK;
}
}
break;
#ifdef CONFIG_NET_IPTABLES
case IPT_SO_SET_REPLACE:
ret = ipt_setsockopt(psock, option, value, value_len);
break;
#endif
default:
nerr("ERROR: Unrecognized IPv4 option: %d\n", option);
ret = -ENOPROTOOPT;
break;
}
net_unlock();
return ret;
}
#endif /* CONFIG_NET_IPv4 */