mirror of
				https://github.com/torvalds/linux.git
				synced 2025-11-04 10:40:15 +02:00 
			
		
		
		
	-EINVAL will interrupt the dump. The correct error to return
if we have more data to dump is -EMSGSIZE.
Discovered by doing:
  for i in `seq 80`; do ip link add type veth; done
  ./cli.py --dbg-small-recv 5300 --spec netdev.yaml --dump dev-get >> /dev/null
  [...]
     nl_len = 64 (48) nl_flags = 0x0 nl_type = 19
     nl_len = 20 (4) nl_flags = 0x2 nl_type = 3
  	error: -22
Fixes: d3d854fd6a ("netdev-genl: create a simple family for netdev stuff")
Reviewed-by: Amritha Nambiar <amritha.nambiar@intel.com>
Reviewed-by: Eric Dumazet <edumazet@google.com>
Link: https://lore.kernel.org/r/20240613213044.3675745-1-kuba@kernel.org
Signed-off-by: Jakub Kicinski <kuba@kernel.org>
		
	
			
		
			
				
	
	
		
			767 lines
		
	
	
	
		
			19 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			767 lines
		
	
	
	
		
			19 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
// SPDX-License-Identifier: GPL-2.0-only
 | 
						|
 | 
						|
#include <linux/netdevice.h>
 | 
						|
#include <linux/notifier.h>
 | 
						|
#include <linux/rtnetlink.h>
 | 
						|
#include <net/net_namespace.h>
 | 
						|
#include <net/sock.h>
 | 
						|
#include <net/xdp.h>
 | 
						|
#include <net/xdp_sock.h>
 | 
						|
#include <net/netdev_rx_queue.h>
 | 
						|
#include <net/netdev_queues.h>
 | 
						|
#include <net/busy_poll.h>
 | 
						|
 | 
						|
#include "netdev-genl-gen.h"
 | 
						|
#include "dev.h"
 | 
						|
 | 
						|
struct netdev_nl_dump_ctx {
 | 
						|
	unsigned long	ifindex;
 | 
						|
	unsigned int	rxq_idx;
 | 
						|
	unsigned int	txq_idx;
 | 
						|
	unsigned int	napi_id;
 | 
						|
};
 | 
						|
 | 
						|
static struct netdev_nl_dump_ctx *netdev_dump_ctx(struct netlink_callback *cb)
 | 
						|
{
 | 
						|
	NL_ASSERT_DUMP_CTX_FITS(struct netdev_nl_dump_ctx);
 | 
						|
 | 
						|
	return (struct netdev_nl_dump_ctx *)cb->ctx;
 | 
						|
}
 | 
						|
 | 
						|
static int
 | 
						|
netdev_nl_dev_fill(struct net_device *netdev, struct sk_buff *rsp,
 | 
						|
		   const struct genl_info *info)
 | 
						|
{
 | 
						|
	u64 xsk_features = 0;
 | 
						|
	u64 xdp_rx_meta = 0;
 | 
						|
	void *hdr;
 | 
						|
 | 
						|
	hdr = genlmsg_iput(rsp, info);
 | 
						|
	if (!hdr)
 | 
						|
		return -EMSGSIZE;
 | 
						|
 | 
						|
#define XDP_METADATA_KFUNC(_, flag, __, xmo) \
 | 
						|
	if (netdev->xdp_metadata_ops && netdev->xdp_metadata_ops->xmo) \
 | 
						|
		xdp_rx_meta |= flag;
 | 
						|
XDP_METADATA_KFUNC_xxx
 | 
						|
#undef XDP_METADATA_KFUNC
 | 
						|
 | 
						|
	if (netdev->xsk_tx_metadata_ops) {
 | 
						|
		if (netdev->xsk_tx_metadata_ops->tmo_fill_timestamp)
 | 
						|
			xsk_features |= NETDEV_XSK_FLAGS_TX_TIMESTAMP;
 | 
						|
		if (netdev->xsk_tx_metadata_ops->tmo_request_checksum)
 | 
						|
			xsk_features |= NETDEV_XSK_FLAGS_TX_CHECKSUM;
 | 
						|
	}
 | 
						|
 | 
						|
	if (nla_put_u32(rsp, NETDEV_A_DEV_IFINDEX, netdev->ifindex) ||
 | 
						|
	    nla_put_u64_64bit(rsp, NETDEV_A_DEV_XDP_FEATURES,
 | 
						|
			      netdev->xdp_features, NETDEV_A_DEV_PAD) ||
 | 
						|
	    nla_put_u64_64bit(rsp, NETDEV_A_DEV_XDP_RX_METADATA_FEATURES,
 | 
						|
			      xdp_rx_meta, NETDEV_A_DEV_PAD) ||
 | 
						|
	    nla_put_u64_64bit(rsp, NETDEV_A_DEV_XSK_FEATURES,
 | 
						|
			      xsk_features, NETDEV_A_DEV_PAD))
 | 
						|
		goto err_cancel_msg;
 | 
						|
 | 
						|
	if (netdev->xdp_features & NETDEV_XDP_ACT_XSK_ZEROCOPY) {
 | 
						|
		if (nla_put_u32(rsp, NETDEV_A_DEV_XDP_ZC_MAX_SEGS,
 | 
						|
				netdev->xdp_zc_max_segs))
 | 
						|
			goto err_cancel_msg;
 | 
						|
	}
 | 
						|
 | 
						|
	genlmsg_end(rsp, hdr);
 | 
						|
 | 
						|
	return 0;
 | 
						|
 | 
						|
err_cancel_msg:
 | 
						|
	genlmsg_cancel(rsp, hdr);
 | 
						|
	return -EMSGSIZE;
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
netdev_genl_dev_notify(struct net_device *netdev, int cmd)
 | 
						|
{
 | 
						|
	struct genl_info info;
 | 
						|
	struct sk_buff *ntf;
 | 
						|
 | 
						|
	if (!genl_has_listeners(&netdev_nl_family, dev_net(netdev),
 | 
						|
				NETDEV_NLGRP_MGMT))
 | 
						|
		return;
 | 
						|
 | 
						|
	genl_info_init_ntf(&info, &netdev_nl_family, cmd);
 | 
						|
 | 
						|
	ntf = genlmsg_new(GENLMSG_DEFAULT_SIZE, GFP_KERNEL);
 | 
						|
	if (!ntf)
 | 
						|
		return;
 | 
						|
 | 
						|
	if (netdev_nl_dev_fill(netdev, ntf, &info)) {
 | 
						|
		nlmsg_free(ntf);
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	genlmsg_multicast_netns(&netdev_nl_family, dev_net(netdev), ntf,
 | 
						|
				0, NETDEV_NLGRP_MGMT, GFP_KERNEL);
 | 
						|
}
 | 
						|
 | 
						|
int netdev_nl_dev_get_doit(struct sk_buff *skb, struct genl_info *info)
 | 
						|
{
 | 
						|
	struct net_device *netdev;
 | 
						|
	struct sk_buff *rsp;
 | 
						|
	u32 ifindex;
 | 
						|
	int err;
 | 
						|
 | 
						|
	if (GENL_REQ_ATTR_CHECK(info, NETDEV_A_DEV_IFINDEX))
 | 
						|
		return -EINVAL;
 | 
						|
 | 
						|
	ifindex = nla_get_u32(info->attrs[NETDEV_A_DEV_IFINDEX]);
 | 
						|
 | 
						|
	rsp = genlmsg_new(GENLMSG_DEFAULT_SIZE, GFP_KERNEL);
 | 
						|
	if (!rsp)
 | 
						|
		return -ENOMEM;
 | 
						|
 | 
						|
	rtnl_lock();
 | 
						|
 | 
						|
	netdev = __dev_get_by_index(genl_info_net(info), ifindex);
 | 
						|
	if (netdev)
 | 
						|
		err = netdev_nl_dev_fill(netdev, rsp, info);
 | 
						|
	else
 | 
						|
		err = -ENODEV;
 | 
						|
 | 
						|
	rtnl_unlock();
 | 
						|
 | 
						|
	if (err)
 | 
						|
		goto err_free_msg;
 | 
						|
 | 
						|
	return genlmsg_reply(rsp, info);
 | 
						|
 | 
						|
err_free_msg:
 | 
						|
	nlmsg_free(rsp);
 | 
						|
	return err;
 | 
						|
}
 | 
						|
 | 
						|
int netdev_nl_dev_get_dumpit(struct sk_buff *skb, struct netlink_callback *cb)
 | 
						|
{
 | 
						|
	struct netdev_nl_dump_ctx *ctx = netdev_dump_ctx(cb);
 | 
						|
	struct net *net = sock_net(skb->sk);
 | 
						|
	struct net_device *netdev;
 | 
						|
	int err = 0;
 | 
						|
 | 
						|
	rtnl_lock();
 | 
						|
	for_each_netdev_dump(net, netdev, ctx->ifindex) {
 | 
						|
		err = netdev_nl_dev_fill(netdev, skb, genl_info_dump(cb));
 | 
						|
		if (err < 0)
 | 
						|
			break;
 | 
						|
	}
 | 
						|
	rtnl_unlock();
 | 
						|
 | 
						|
	return err;
 | 
						|
}
 | 
						|
 | 
						|
static int
 | 
						|
netdev_nl_napi_fill_one(struct sk_buff *rsp, struct napi_struct *napi,
 | 
						|
			const struct genl_info *info)
 | 
						|
{
 | 
						|
	void *hdr;
 | 
						|
	pid_t pid;
 | 
						|
 | 
						|
	if (WARN_ON_ONCE(!napi->dev))
 | 
						|
		return -EINVAL;
 | 
						|
	if (!(napi->dev->flags & IFF_UP))
 | 
						|
		return 0;
 | 
						|
 | 
						|
	hdr = genlmsg_iput(rsp, info);
 | 
						|
	if (!hdr)
 | 
						|
		return -EMSGSIZE;
 | 
						|
 | 
						|
	if (napi->napi_id >= MIN_NAPI_ID &&
 | 
						|
	    nla_put_u32(rsp, NETDEV_A_NAPI_ID, napi->napi_id))
 | 
						|
		goto nla_put_failure;
 | 
						|
 | 
						|
	if (nla_put_u32(rsp, NETDEV_A_NAPI_IFINDEX, napi->dev->ifindex))
 | 
						|
		goto nla_put_failure;
 | 
						|
 | 
						|
	if (napi->irq >= 0 && nla_put_u32(rsp, NETDEV_A_NAPI_IRQ, napi->irq))
 | 
						|
		goto nla_put_failure;
 | 
						|
 | 
						|
	if (napi->thread) {
 | 
						|
		pid = task_pid_nr(napi->thread);
 | 
						|
		if (nla_put_u32(rsp, NETDEV_A_NAPI_PID, pid))
 | 
						|
			goto nla_put_failure;
 | 
						|
	}
 | 
						|
 | 
						|
	genlmsg_end(rsp, hdr);
 | 
						|
 | 
						|
	return 0;
 | 
						|
 | 
						|
nla_put_failure:
 | 
						|
	genlmsg_cancel(rsp, hdr);
 | 
						|
	return -EMSGSIZE;
 | 
						|
}
 | 
						|
 | 
						|
int netdev_nl_napi_get_doit(struct sk_buff *skb, struct genl_info *info)
 | 
						|
{
 | 
						|
	struct napi_struct *napi;
 | 
						|
	struct sk_buff *rsp;
 | 
						|
	u32 napi_id;
 | 
						|
	int err;
 | 
						|
 | 
						|
	if (GENL_REQ_ATTR_CHECK(info, NETDEV_A_NAPI_ID))
 | 
						|
		return -EINVAL;
 | 
						|
 | 
						|
	napi_id = nla_get_u32(info->attrs[NETDEV_A_NAPI_ID]);
 | 
						|
 | 
						|
	rsp = genlmsg_new(GENLMSG_DEFAULT_SIZE, GFP_KERNEL);
 | 
						|
	if (!rsp)
 | 
						|
		return -ENOMEM;
 | 
						|
 | 
						|
	rtnl_lock();
 | 
						|
 | 
						|
	napi = napi_by_id(napi_id);
 | 
						|
	if (napi)
 | 
						|
		err = netdev_nl_napi_fill_one(rsp, napi, info);
 | 
						|
	else
 | 
						|
		err = -EINVAL;
 | 
						|
 | 
						|
	rtnl_unlock();
 | 
						|
 | 
						|
	if (err)
 | 
						|
		goto err_free_msg;
 | 
						|
 | 
						|
	return genlmsg_reply(rsp, info);
 | 
						|
 | 
						|
err_free_msg:
 | 
						|
	nlmsg_free(rsp);
 | 
						|
	return err;
 | 
						|
}
 | 
						|
 | 
						|
static int
 | 
						|
netdev_nl_napi_dump_one(struct net_device *netdev, struct sk_buff *rsp,
 | 
						|
			const struct genl_info *info,
 | 
						|
			struct netdev_nl_dump_ctx *ctx)
 | 
						|
{
 | 
						|
	struct napi_struct *napi;
 | 
						|
	int err = 0;
 | 
						|
 | 
						|
	if (!(netdev->flags & IFF_UP))
 | 
						|
		return err;
 | 
						|
 | 
						|
	list_for_each_entry(napi, &netdev->napi_list, dev_list) {
 | 
						|
		if (ctx->napi_id && napi->napi_id >= ctx->napi_id)
 | 
						|
			continue;
 | 
						|
 | 
						|
		err = netdev_nl_napi_fill_one(rsp, napi, info);
 | 
						|
		if (err)
 | 
						|
			return err;
 | 
						|
		ctx->napi_id = napi->napi_id;
 | 
						|
	}
 | 
						|
	return err;
 | 
						|
}
 | 
						|
 | 
						|
int netdev_nl_napi_get_dumpit(struct sk_buff *skb, struct netlink_callback *cb)
 | 
						|
{
 | 
						|
	struct netdev_nl_dump_ctx *ctx = netdev_dump_ctx(cb);
 | 
						|
	const struct genl_info *info = genl_info_dump(cb);
 | 
						|
	struct net *net = sock_net(skb->sk);
 | 
						|
	struct net_device *netdev;
 | 
						|
	u32 ifindex = 0;
 | 
						|
	int err = 0;
 | 
						|
 | 
						|
	if (info->attrs[NETDEV_A_NAPI_IFINDEX])
 | 
						|
		ifindex = nla_get_u32(info->attrs[NETDEV_A_NAPI_IFINDEX]);
 | 
						|
 | 
						|
	rtnl_lock();
 | 
						|
	if (ifindex) {
 | 
						|
		netdev = __dev_get_by_index(net, ifindex);
 | 
						|
		if (netdev)
 | 
						|
			err = netdev_nl_napi_dump_one(netdev, skb, info, ctx);
 | 
						|
		else
 | 
						|
			err = -ENODEV;
 | 
						|
	} else {
 | 
						|
		for_each_netdev_dump(net, netdev, ctx->ifindex) {
 | 
						|
			err = netdev_nl_napi_dump_one(netdev, skb, info, ctx);
 | 
						|
			if (err < 0)
 | 
						|
				break;
 | 
						|
			ctx->napi_id = 0;
 | 
						|
		}
 | 
						|
	}
 | 
						|
	rtnl_unlock();
 | 
						|
 | 
						|
	return err;
 | 
						|
}
 | 
						|
 | 
						|
static int
 | 
						|
netdev_nl_queue_fill_one(struct sk_buff *rsp, struct net_device *netdev,
 | 
						|
			 u32 q_idx, u32 q_type, const struct genl_info *info)
 | 
						|
{
 | 
						|
	struct netdev_rx_queue *rxq;
 | 
						|
	struct netdev_queue *txq;
 | 
						|
	void *hdr;
 | 
						|
 | 
						|
	hdr = genlmsg_iput(rsp, info);
 | 
						|
	if (!hdr)
 | 
						|
		return -EMSGSIZE;
 | 
						|
 | 
						|
	if (nla_put_u32(rsp, NETDEV_A_QUEUE_ID, q_idx) ||
 | 
						|
	    nla_put_u32(rsp, NETDEV_A_QUEUE_TYPE, q_type) ||
 | 
						|
	    nla_put_u32(rsp, NETDEV_A_QUEUE_IFINDEX, netdev->ifindex))
 | 
						|
		goto nla_put_failure;
 | 
						|
 | 
						|
	switch (q_type) {
 | 
						|
	case NETDEV_QUEUE_TYPE_RX:
 | 
						|
		rxq = __netif_get_rx_queue(netdev, q_idx);
 | 
						|
		if (rxq->napi && nla_put_u32(rsp, NETDEV_A_QUEUE_NAPI_ID,
 | 
						|
					     rxq->napi->napi_id))
 | 
						|
			goto nla_put_failure;
 | 
						|
		break;
 | 
						|
	case NETDEV_QUEUE_TYPE_TX:
 | 
						|
		txq = netdev_get_tx_queue(netdev, q_idx);
 | 
						|
		if (txq->napi && nla_put_u32(rsp, NETDEV_A_QUEUE_NAPI_ID,
 | 
						|
					     txq->napi->napi_id))
 | 
						|
			goto nla_put_failure;
 | 
						|
	}
 | 
						|
 | 
						|
	genlmsg_end(rsp, hdr);
 | 
						|
 | 
						|
	return 0;
 | 
						|
 | 
						|
nla_put_failure:
 | 
						|
	genlmsg_cancel(rsp, hdr);
 | 
						|
	return -EMSGSIZE;
 | 
						|
}
 | 
						|
 | 
						|
static int netdev_nl_queue_validate(struct net_device *netdev, u32 q_id,
 | 
						|
				    u32 q_type)
 | 
						|
{
 | 
						|
	switch (q_type) {
 | 
						|
	case NETDEV_QUEUE_TYPE_RX:
 | 
						|
		if (q_id >= netdev->real_num_rx_queues)
 | 
						|
			return -EINVAL;
 | 
						|
		return 0;
 | 
						|
	case NETDEV_QUEUE_TYPE_TX:
 | 
						|
		if (q_id >= netdev->real_num_tx_queues)
 | 
						|
			return -EINVAL;
 | 
						|
	}
 | 
						|
	return 0;
 | 
						|
}
 | 
						|
 | 
						|
static int
 | 
						|
netdev_nl_queue_fill(struct sk_buff *rsp, struct net_device *netdev, u32 q_idx,
 | 
						|
		     u32 q_type, const struct genl_info *info)
 | 
						|
{
 | 
						|
	int err = 0;
 | 
						|
 | 
						|
	if (!(netdev->flags & IFF_UP))
 | 
						|
		return err;
 | 
						|
 | 
						|
	err = netdev_nl_queue_validate(netdev, q_idx, q_type);
 | 
						|
	if (err)
 | 
						|
		return err;
 | 
						|
 | 
						|
	return netdev_nl_queue_fill_one(rsp, netdev, q_idx, q_type, info);
 | 
						|
}
 | 
						|
 | 
						|
int netdev_nl_queue_get_doit(struct sk_buff *skb, struct genl_info *info)
 | 
						|
{
 | 
						|
	u32 q_id, q_type, ifindex;
 | 
						|
	struct net_device *netdev;
 | 
						|
	struct sk_buff *rsp;
 | 
						|
	int err;
 | 
						|
 | 
						|
	if (GENL_REQ_ATTR_CHECK(info, NETDEV_A_QUEUE_ID) ||
 | 
						|
	    GENL_REQ_ATTR_CHECK(info, NETDEV_A_QUEUE_TYPE) ||
 | 
						|
	    GENL_REQ_ATTR_CHECK(info, NETDEV_A_QUEUE_IFINDEX))
 | 
						|
		return -EINVAL;
 | 
						|
 | 
						|
	q_id = nla_get_u32(info->attrs[NETDEV_A_QUEUE_ID]);
 | 
						|
	q_type = nla_get_u32(info->attrs[NETDEV_A_QUEUE_TYPE]);
 | 
						|
	ifindex = nla_get_u32(info->attrs[NETDEV_A_QUEUE_IFINDEX]);
 | 
						|
 | 
						|
	rsp = genlmsg_new(GENLMSG_DEFAULT_SIZE, GFP_KERNEL);
 | 
						|
	if (!rsp)
 | 
						|
		return -ENOMEM;
 | 
						|
 | 
						|
	rtnl_lock();
 | 
						|
 | 
						|
	netdev = __dev_get_by_index(genl_info_net(info), ifindex);
 | 
						|
	if (netdev)
 | 
						|
		err = netdev_nl_queue_fill(rsp, netdev, q_id, q_type, info);
 | 
						|
	else
 | 
						|
		err = -ENODEV;
 | 
						|
 | 
						|
	rtnl_unlock();
 | 
						|
 | 
						|
	if (err)
 | 
						|
		goto err_free_msg;
 | 
						|
 | 
						|
	return genlmsg_reply(rsp, info);
 | 
						|
 | 
						|
err_free_msg:
 | 
						|
	nlmsg_free(rsp);
 | 
						|
	return err;
 | 
						|
}
 | 
						|
 | 
						|
static int
 | 
						|
netdev_nl_queue_dump_one(struct net_device *netdev, struct sk_buff *rsp,
 | 
						|
			 const struct genl_info *info,
 | 
						|
			 struct netdev_nl_dump_ctx *ctx)
 | 
						|
{
 | 
						|
	int err = 0;
 | 
						|
	int i;
 | 
						|
 | 
						|
	if (!(netdev->flags & IFF_UP))
 | 
						|
		return err;
 | 
						|
 | 
						|
	for (i = ctx->rxq_idx; i < netdev->real_num_rx_queues;) {
 | 
						|
		err = netdev_nl_queue_fill_one(rsp, netdev, i,
 | 
						|
					       NETDEV_QUEUE_TYPE_RX, info);
 | 
						|
		if (err)
 | 
						|
			return err;
 | 
						|
		ctx->rxq_idx = i++;
 | 
						|
	}
 | 
						|
	for (i = ctx->txq_idx; i < netdev->real_num_tx_queues;) {
 | 
						|
		err = netdev_nl_queue_fill_one(rsp, netdev, i,
 | 
						|
					       NETDEV_QUEUE_TYPE_TX, info);
 | 
						|
		if (err)
 | 
						|
			return err;
 | 
						|
		ctx->txq_idx = i++;
 | 
						|
	}
 | 
						|
 | 
						|
	return err;
 | 
						|
}
 | 
						|
 | 
						|
int netdev_nl_queue_get_dumpit(struct sk_buff *skb, struct netlink_callback *cb)
 | 
						|
{
 | 
						|
	struct netdev_nl_dump_ctx *ctx = netdev_dump_ctx(cb);
 | 
						|
	const struct genl_info *info = genl_info_dump(cb);
 | 
						|
	struct net *net = sock_net(skb->sk);
 | 
						|
	struct net_device *netdev;
 | 
						|
	u32 ifindex = 0;
 | 
						|
	int err = 0;
 | 
						|
 | 
						|
	if (info->attrs[NETDEV_A_QUEUE_IFINDEX])
 | 
						|
		ifindex = nla_get_u32(info->attrs[NETDEV_A_QUEUE_IFINDEX]);
 | 
						|
 | 
						|
	rtnl_lock();
 | 
						|
	if (ifindex) {
 | 
						|
		netdev = __dev_get_by_index(net, ifindex);
 | 
						|
		if (netdev)
 | 
						|
			err = netdev_nl_queue_dump_one(netdev, skb, info, ctx);
 | 
						|
		else
 | 
						|
			err = -ENODEV;
 | 
						|
	} else {
 | 
						|
		for_each_netdev_dump(net, netdev, ctx->ifindex) {
 | 
						|
			err = netdev_nl_queue_dump_one(netdev, skb, info, ctx);
 | 
						|
			if (err < 0)
 | 
						|
				break;
 | 
						|
			ctx->rxq_idx = 0;
 | 
						|
			ctx->txq_idx = 0;
 | 
						|
		}
 | 
						|
	}
 | 
						|
	rtnl_unlock();
 | 
						|
 | 
						|
	return err;
 | 
						|
}
 | 
						|
 | 
						|
#define NETDEV_STAT_NOT_SET		(~0ULL)
 | 
						|
 | 
						|
static void netdev_nl_stats_add(void *_sum, const void *_add, size_t size)
 | 
						|
{
 | 
						|
	const u64 *add = _add;
 | 
						|
	u64 *sum = _sum;
 | 
						|
 | 
						|
	while (size) {
 | 
						|
		if (*add != NETDEV_STAT_NOT_SET && *sum != NETDEV_STAT_NOT_SET)
 | 
						|
			*sum += *add;
 | 
						|
		sum++;
 | 
						|
		add++;
 | 
						|
		size -= 8;
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static int netdev_stat_put(struct sk_buff *rsp, unsigned int attr_id, u64 value)
 | 
						|
{
 | 
						|
	if (value == NETDEV_STAT_NOT_SET)
 | 
						|
		return 0;
 | 
						|
	return nla_put_uint(rsp, attr_id, value);
 | 
						|
}
 | 
						|
 | 
						|
static int
 | 
						|
netdev_nl_stats_write_rx(struct sk_buff *rsp, struct netdev_queue_stats_rx *rx)
 | 
						|
{
 | 
						|
	if (netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_PACKETS, rx->packets) ||
 | 
						|
	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_BYTES, rx->bytes) ||
 | 
						|
	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_ALLOC_FAIL, rx->alloc_fail) ||
 | 
						|
	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_HW_DROPS, rx->hw_drops) ||
 | 
						|
	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_HW_DROP_OVERRUNS, rx->hw_drop_overruns) ||
 | 
						|
	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_CSUM_UNNECESSARY, rx->csum_unnecessary) ||
 | 
						|
	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_CSUM_NONE, rx->csum_none) ||
 | 
						|
	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_CSUM_BAD, rx->csum_bad) ||
 | 
						|
	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_HW_GRO_PACKETS, rx->hw_gro_packets) ||
 | 
						|
	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_HW_GRO_BYTES, rx->hw_gro_bytes) ||
 | 
						|
	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_HW_GRO_WIRE_PACKETS, rx->hw_gro_wire_packets) ||
 | 
						|
	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_HW_GRO_WIRE_BYTES, rx->hw_gro_wire_bytes) ||
 | 
						|
	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_HW_DROP_RATELIMITS, rx->hw_drop_ratelimits))
 | 
						|
		return -EMSGSIZE;
 | 
						|
	return 0;
 | 
						|
}
 | 
						|
 | 
						|
static int
 | 
						|
netdev_nl_stats_write_tx(struct sk_buff *rsp, struct netdev_queue_stats_tx *tx)
 | 
						|
{
 | 
						|
	if (netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_PACKETS, tx->packets) ||
 | 
						|
	    netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_BYTES, tx->bytes) ||
 | 
						|
	    netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_HW_DROPS, tx->hw_drops) ||
 | 
						|
	    netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_HW_DROP_ERRORS, tx->hw_drop_errors) ||
 | 
						|
	    netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_CSUM_NONE, tx->csum_none) ||
 | 
						|
	    netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_NEEDS_CSUM, tx->needs_csum) ||
 | 
						|
	    netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_HW_GSO_PACKETS, tx->hw_gso_packets) ||
 | 
						|
	    netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_HW_GSO_BYTES, tx->hw_gso_bytes) ||
 | 
						|
	    netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_HW_GSO_WIRE_PACKETS, tx->hw_gso_wire_packets) ||
 | 
						|
	    netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_HW_GSO_WIRE_BYTES, tx->hw_gso_wire_bytes) ||
 | 
						|
	    netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_HW_DROP_RATELIMITS, tx->hw_drop_ratelimits) ||
 | 
						|
	    netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_STOP, tx->stop) ||
 | 
						|
	    netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_WAKE, tx->wake))
 | 
						|
		return -EMSGSIZE;
 | 
						|
	return 0;
 | 
						|
}
 | 
						|
 | 
						|
static int
 | 
						|
netdev_nl_stats_queue(struct net_device *netdev, struct sk_buff *rsp,
 | 
						|
		      u32 q_type, int i, const struct genl_info *info)
 | 
						|
{
 | 
						|
	const struct netdev_stat_ops *ops = netdev->stat_ops;
 | 
						|
	struct netdev_queue_stats_rx rx;
 | 
						|
	struct netdev_queue_stats_tx tx;
 | 
						|
	void *hdr;
 | 
						|
 | 
						|
	hdr = genlmsg_iput(rsp, info);
 | 
						|
	if (!hdr)
 | 
						|
		return -EMSGSIZE;
 | 
						|
	if (nla_put_u32(rsp, NETDEV_A_QSTATS_IFINDEX, netdev->ifindex) ||
 | 
						|
	    nla_put_u32(rsp, NETDEV_A_QSTATS_QUEUE_TYPE, q_type) ||
 | 
						|
	    nla_put_u32(rsp, NETDEV_A_QSTATS_QUEUE_ID, i))
 | 
						|
		goto nla_put_failure;
 | 
						|
 | 
						|
	switch (q_type) {
 | 
						|
	case NETDEV_QUEUE_TYPE_RX:
 | 
						|
		memset(&rx, 0xff, sizeof(rx));
 | 
						|
		ops->get_queue_stats_rx(netdev, i, &rx);
 | 
						|
		if (!memchr_inv(&rx, 0xff, sizeof(rx)))
 | 
						|
			goto nla_cancel;
 | 
						|
		if (netdev_nl_stats_write_rx(rsp, &rx))
 | 
						|
			goto nla_put_failure;
 | 
						|
		break;
 | 
						|
	case NETDEV_QUEUE_TYPE_TX:
 | 
						|
		memset(&tx, 0xff, sizeof(tx));
 | 
						|
		ops->get_queue_stats_tx(netdev, i, &tx);
 | 
						|
		if (!memchr_inv(&tx, 0xff, sizeof(tx)))
 | 
						|
			goto nla_cancel;
 | 
						|
		if (netdev_nl_stats_write_tx(rsp, &tx))
 | 
						|
			goto nla_put_failure;
 | 
						|
		break;
 | 
						|
	}
 | 
						|
 | 
						|
	genlmsg_end(rsp, hdr);
 | 
						|
	return 0;
 | 
						|
 | 
						|
nla_cancel:
 | 
						|
	genlmsg_cancel(rsp, hdr);
 | 
						|
	return 0;
 | 
						|
nla_put_failure:
 | 
						|
	genlmsg_cancel(rsp, hdr);
 | 
						|
	return -EMSGSIZE;
 | 
						|
}
 | 
						|
 | 
						|
static int
 | 
						|
netdev_nl_stats_by_queue(struct net_device *netdev, struct sk_buff *rsp,
 | 
						|
			 const struct genl_info *info,
 | 
						|
			 struct netdev_nl_dump_ctx *ctx)
 | 
						|
{
 | 
						|
	const struct netdev_stat_ops *ops = netdev->stat_ops;
 | 
						|
	int i, err;
 | 
						|
 | 
						|
	if (!(netdev->flags & IFF_UP))
 | 
						|
		return 0;
 | 
						|
 | 
						|
	i = ctx->rxq_idx;
 | 
						|
	while (ops->get_queue_stats_rx && i < netdev->real_num_rx_queues) {
 | 
						|
		err = netdev_nl_stats_queue(netdev, rsp, NETDEV_QUEUE_TYPE_RX,
 | 
						|
					    i, info);
 | 
						|
		if (err)
 | 
						|
			return err;
 | 
						|
		ctx->rxq_idx = i++;
 | 
						|
	}
 | 
						|
	i = ctx->txq_idx;
 | 
						|
	while (ops->get_queue_stats_tx && i < netdev->real_num_tx_queues) {
 | 
						|
		err = netdev_nl_stats_queue(netdev, rsp, NETDEV_QUEUE_TYPE_TX,
 | 
						|
					    i, info);
 | 
						|
		if (err)
 | 
						|
			return err;
 | 
						|
		ctx->txq_idx = i++;
 | 
						|
	}
 | 
						|
 | 
						|
	ctx->rxq_idx = 0;
 | 
						|
	ctx->txq_idx = 0;
 | 
						|
	return 0;
 | 
						|
}
 | 
						|
 | 
						|
static int
 | 
						|
netdev_nl_stats_by_netdev(struct net_device *netdev, struct sk_buff *rsp,
 | 
						|
			  const struct genl_info *info)
 | 
						|
{
 | 
						|
	struct netdev_queue_stats_rx rx_sum, rx;
 | 
						|
	struct netdev_queue_stats_tx tx_sum, tx;
 | 
						|
	const struct netdev_stat_ops *ops;
 | 
						|
	void *hdr;
 | 
						|
	int i;
 | 
						|
 | 
						|
	ops = netdev->stat_ops;
 | 
						|
	/* Netdev can't guarantee any complete counters */
 | 
						|
	if (!ops->get_base_stats)
 | 
						|
		return 0;
 | 
						|
 | 
						|
	memset(&rx_sum, 0xff, sizeof(rx_sum));
 | 
						|
	memset(&tx_sum, 0xff, sizeof(tx_sum));
 | 
						|
 | 
						|
	ops->get_base_stats(netdev, &rx_sum, &tx_sum);
 | 
						|
 | 
						|
	/* The op was there, but nothing reported, don't bother */
 | 
						|
	if (!memchr_inv(&rx_sum, 0xff, sizeof(rx_sum)) &&
 | 
						|
	    !memchr_inv(&tx_sum, 0xff, sizeof(tx_sum)))
 | 
						|
		return 0;
 | 
						|
 | 
						|
	hdr = genlmsg_iput(rsp, info);
 | 
						|
	if (!hdr)
 | 
						|
		return -EMSGSIZE;
 | 
						|
	if (nla_put_u32(rsp, NETDEV_A_QSTATS_IFINDEX, netdev->ifindex))
 | 
						|
		goto nla_put_failure;
 | 
						|
 | 
						|
	for (i = 0; i < netdev->real_num_rx_queues; i++) {
 | 
						|
		memset(&rx, 0xff, sizeof(rx));
 | 
						|
		if (ops->get_queue_stats_rx)
 | 
						|
			ops->get_queue_stats_rx(netdev, i, &rx);
 | 
						|
		netdev_nl_stats_add(&rx_sum, &rx, sizeof(rx));
 | 
						|
	}
 | 
						|
	for (i = 0; i < netdev->real_num_tx_queues; i++) {
 | 
						|
		memset(&tx, 0xff, sizeof(tx));
 | 
						|
		if (ops->get_queue_stats_tx)
 | 
						|
			ops->get_queue_stats_tx(netdev, i, &tx);
 | 
						|
		netdev_nl_stats_add(&tx_sum, &tx, sizeof(tx));
 | 
						|
	}
 | 
						|
 | 
						|
	if (netdev_nl_stats_write_rx(rsp, &rx_sum) ||
 | 
						|
	    netdev_nl_stats_write_tx(rsp, &tx_sum))
 | 
						|
		goto nla_put_failure;
 | 
						|
 | 
						|
	genlmsg_end(rsp, hdr);
 | 
						|
	return 0;
 | 
						|
 | 
						|
nla_put_failure:
 | 
						|
	genlmsg_cancel(rsp, hdr);
 | 
						|
	return -EMSGSIZE;
 | 
						|
}
 | 
						|
 | 
						|
static int
 | 
						|
netdev_nl_qstats_get_dump_one(struct net_device *netdev, unsigned int scope,
 | 
						|
			      struct sk_buff *skb, const struct genl_info *info,
 | 
						|
			      struct netdev_nl_dump_ctx *ctx)
 | 
						|
{
 | 
						|
	if (!netdev->stat_ops)
 | 
						|
		return 0;
 | 
						|
 | 
						|
	switch (scope) {
 | 
						|
	case 0:
 | 
						|
		return netdev_nl_stats_by_netdev(netdev, skb, info);
 | 
						|
	case NETDEV_QSTATS_SCOPE_QUEUE:
 | 
						|
		return netdev_nl_stats_by_queue(netdev, skb, info, ctx);
 | 
						|
	}
 | 
						|
 | 
						|
	return -EINVAL;	/* Should not happen, per netlink policy */
 | 
						|
}
 | 
						|
 | 
						|
int netdev_nl_qstats_get_dumpit(struct sk_buff *skb,
 | 
						|
				struct netlink_callback *cb)
 | 
						|
{
 | 
						|
	struct netdev_nl_dump_ctx *ctx = netdev_dump_ctx(cb);
 | 
						|
	const struct genl_info *info = genl_info_dump(cb);
 | 
						|
	struct net *net = sock_net(skb->sk);
 | 
						|
	struct net_device *netdev;
 | 
						|
	unsigned int ifindex;
 | 
						|
	unsigned int scope;
 | 
						|
	int err = 0;
 | 
						|
 | 
						|
	scope = 0;
 | 
						|
	if (info->attrs[NETDEV_A_QSTATS_SCOPE])
 | 
						|
		scope = nla_get_uint(info->attrs[NETDEV_A_QSTATS_SCOPE]);
 | 
						|
 | 
						|
	ifindex = 0;
 | 
						|
	if (info->attrs[NETDEV_A_QSTATS_IFINDEX])
 | 
						|
		ifindex = nla_get_u32(info->attrs[NETDEV_A_QSTATS_IFINDEX]);
 | 
						|
 | 
						|
	rtnl_lock();
 | 
						|
	if (ifindex) {
 | 
						|
		netdev = __dev_get_by_index(net, ifindex);
 | 
						|
		if (netdev && netdev->stat_ops) {
 | 
						|
			err = netdev_nl_qstats_get_dump_one(netdev, scope, skb,
 | 
						|
							    info, ctx);
 | 
						|
		} else {
 | 
						|
			NL_SET_BAD_ATTR(info->extack,
 | 
						|
					info->attrs[NETDEV_A_QSTATS_IFINDEX]);
 | 
						|
			err = netdev ? -EOPNOTSUPP : -ENODEV;
 | 
						|
		}
 | 
						|
	} else {
 | 
						|
		for_each_netdev_dump(net, netdev, ctx->ifindex) {
 | 
						|
			err = netdev_nl_qstats_get_dump_one(netdev, scope, skb,
 | 
						|
							    info, ctx);
 | 
						|
			if (err < 0)
 | 
						|
				break;
 | 
						|
		}
 | 
						|
	}
 | 
						|
	rtnl_unlock();
 | 
						|
 | 
						|
	return err;
 | 
						|
}
 | 
						|
 | 
						|
static int netdev_genl_netdevice_event(struct notifier_block *nb,
 | 
						|
				       unsigned long event, void *ptr)
 | 
						|
{
 | 
						|
	struct net_device *netdev = netdev_notifier_info_to_dev(ptr);
 | 
						|
 | 
						|
	switch (event) {
 | 
						|
	case NETDEV_REGISTER:
 | 
						|
		netdev_genl_dev_notify(netdev, NETDEV_CMD_DEV_ADD_NTF);
 | 
						|
		break;
 | 
						|
	case NETDEV_UNREGISTER:
 | 
						|
		netdev_genl_dev_notify(netdev, NETDEV_CMD_DEV_DEL_NTF);
 | 
						|
		break;
 | 
						|
	case NETDEV_XDP_FEAT_CHANGE:
 | 
						|
		netdev_genl_dev_notify(netdev, NETDEV_CMD_DEV_CHANGE_NTF);
 | 
						|
		break;
 | 
						|
	}
 | 
						|
 | 
						|
	return NOTIFY_OK;
 | 
						|
}
 | 
						|
 | 
						|
static struct notifier_block netdev_genl_nb = {
 | 
						|
	.notifier_call	= netdev_genl_netdevice_event,
 | 
						|
};
 | 
						|
 | 
						|
static int __init netdev_genl_init(void)
 | 
						|
{
 | 
						|
	int err;
 | 
						|
 | 
						|
	err = register_netdevice_notifier(&netdev_genl_nb);
 | 
						|
	if (err)
 | 
						|
		return err;
 | 
						|
 | 
						|
	err = genl_register_family(&netdev_nl_family);
 | 
						|
	if (err)
 | 
						|
		goto err_unreg_ntf;
 | 
						|
 | 
						|
	return 0;
 | 
						|
 | 
						|
err_unreg_ntf:
 | 
						|
	unregister_netdevice_notifier(&netdev_genl_nb);
 | 
						|
	return err;
 | 
						|
}
 | 
						|
 | 
						|
subsys_initcall(netdev_genl_init);
 |