xref: /linux/net/core/netdev-genl.c (revision 2151003e773c7e7dba4d64bed4bfc483681b5f6a)
1 // SPDX-License-Identifier: GPL-2.0-only
2 
3 #include <linux/netdevice.h>
4 #include <linux/notifier.h>
5 #include <linux/rtnetlink.h>
6 #include <net/busy_poll.h>
7 #include <net/net_namespace.h>
8 #include <net/netdev_queues.h>
9 #include <net/netdev_rx_queue.h>
10 #include <net/sock.h>
11 #include <net/xdp.h>
12 #include <net/xdp_sock.h>
13 #include <net/page_pool/memory_provider.h>
14 
15 #include "dev.h"
16 #include "devmem.h"
17 #include "netdev-genl-gen.h"
18 
19 struct netdev_nl_dump_ctx {
20 	unsigned long	ifindex;
21 	unsigned int	rxq_idx;
22 	unsigned int	txq_idx;
23 	unsigned int	napi_id;
24 };
25 
26 static struct netdev_nl_dump_ctx *netdev_dump_ctx(struct netlink_callback *cb)
27 {
28 	NL_ASSERT_CTX_FITS(struct netdev_nl_dump_ctx);
29 
30 	return (struct netdev_nl_dump_ctx *)cb->ctx;
31 }
32 
33 static int
34 netdev_nl_dev_fill(struct net_device *netdev, struct sk_buff *rsp,
35 		   const struct genl_info *info)
36 {
37 	u64 xsk_features = 0;
38 	u64 xdp_rx_meta = 0;
39 	void *hdr;
40 
41 	hdr = genlmsg_iput(rsp, info);
42 	if (!hdr)
43 		return -EMSGSIZE;
44 
45 #define XDP_METADATA_KFUNC(_, flag, __, xmo) \
46 	if (netdev->xdp_metadata_ops && netdev->xdp_metadata_ops->xmo) \
47 		xdp_rx_meta |= flag;
48 XDP_METADATA_KFUNC_xxx
49 #undef XDP_METADATA_KFUNC
50 
51 	if (netdev->xsk_tx_metadata_ops) {
52 		if (netdev->xsk_tx_metadata_ops->tmo_fill_timestamp)
53 			xsk_features |= NETDEV_XSK_FLAGS_TX_TIMESTAMP;
54 		if (netdev->xsk_tx_metadata_ops->tmo_request_checksum)
55 			xsk_features |= NETDEV_XSK_FLAGS_TX_CHECKSUM;
56 	}
57 
58 	if (nla_put_u32(rsp, NETDEV_A_DEV_IFINDEX, netdev->ifindex) ||
59 	    nla_put_u64_64bit(rsp, NETDEV_A_DEV_XDP_FEATURES,
60 			      netdev->xdp_features, NETDEV_A_DEV_PAD) ||
61 	    nla_put_u64_64bit(rsp, NETDEV_A_DEV_XDP_RX_METADATA_FEATURES,
62 			      xdp_rx_meta, NETDEV_A_DEV_PAD) ||
63 	    nla_put_u64_64bit(rsp, NETDEV_A_DEV_XSK_FEATURES,
64 			      xsk_features, NETDEV_A_DEV_PAD))
65 		goto err_cancel_msg;
66 
67 	if (netdev->xdp_features & NETDEV_XDP_ACT_XSK_ZEROCOPY) {
68 		if (nla_put_u32(rsp, NETDEV_A_DEV_XDP_ZC_MAX_SEGS,
69 				netdev->xdp_zc_max_segs))
70 			goto err_cancel_msg;
71 	}
72 
73 	genlmsg_end(rsp, hdr);
74 
75 	return 0;
76 
77 err_cancel_msg:
78 	genlmsg_cancel(rsp, hdr);
79 	return -EMSGSIZE;
80 }
81 
82 static void
83 netdev_genl_dev_notify(struct net_device *netdev, int cmd)
84 {
85 	struct genl_info info;
86 	struct sk_buff *ntf;
87 
88 	if (!genl_has_listeners(&netdev_nl_family, dev_net(netdev),
89 				NETDEV_NLGRP_MGMT))
90 		return;
91 
92 	genl_info_init_ntf(&info, &netdev_nl_family, cmd);
93 
94 	ntf = genlmsg_new(GENLMSG_DEFAULT_SIZE, GFP_KERNEL);
95 	if (!ntf)
96 		return;
97 
98 	if (netdev_nl_dev_fill(netdev, ntf, &info)) {
99 		nlmsg_free(ntf);
100 		return;
101 	}
102 
103 	genlmsg_multicast_netns(&netdev_nl_family, dev_net(netdev), ntf,
104 				0, NETDEV_NLGRP_MGMT, GFP_KERNEL);
105 }
106 
107 int netdev_nl_dev_get_doit(struct sk_buff *skb, struct genl_info *info)
108 {
109 	struct net_device *netdev;
110 	struct sk_buff *rsp;
111 	u32 ifindex;
112 	int err;
113 
114 	if (GENL_REQ_ATTR_CHECK(info, NETDEV_A_DEV_IFINDEX))
115 		return -EINVAL;
116 
117 	ifindex = nla_get_u32(info->attrs[NETDEV_A_DEV_IFINDEX]);
118 
119 	rsp = genlmsg_new(GENLMSG_DEFAULT_SIZE, GFP_KERNEL);
120 	if (!rsp)
121 		return -ENOMEM;
122 
123 	rtnl_lock();
124 
125 	netdev = __dev_get_by_index(genl_info_net(info), ifindex);
126 	if (netdev)
127 		err = netdev_nl_dev_fill(netdev, rsp, info);
128 	else
129 		err = -ENODEV;
130 
131 	rtnl_unlock();
132 
133 	if (err)
134 		goto err_free_msg;
135 
136 	return genlmsg_reply(rsp, info);
137 
138 err_free_msg:
139 	nlmsg_free(rsp);
140 	return err;
141 }
142 
143 int netdev_nl_dev_get_dumpit(struct sk_buff *skb, struct netlink_callback *cb)
144 {
145 	struct netdev_nl_dump_ctx *ctx = netdev_dump_ctx(cb);
146 	struct net *net = sock_net(skb->sk);
147 	struct net_device *netdev;
148 	int err = 0;
149 
150 	rtnl_lock();
151 	for_each_netdev_dump(net, netdev, ctx->ifindex) {
152 		err = netdev_nl_dev_fill(netdev, skb, genl_info_dump(cb));
153 		if (err < 0)
154 			break;
155 	}
156 	rtnl_unlock();
157 
158 	return err;
159 }
160 
161 static int
162 netdev_nl_napi_fill_one(struct sk_buff *rsp, struct napi_struct *napi,
163 			const struct genl_info *info)
164 {
165 	unsigned long irq_suspend_timeout;
166 	unsigned long gro_flush_timeout;
167 	u32 napi_defer_hard_irqs;
168 	void *hdr;
169 	pid_t pid;
170 
171 	if (!napi->dev->up)
172 		return 0;
173 
174 	hdr = genlmsg_iput(rsp, info);
175 	if (!hdr)
176 		return -EMSGSIZE;
177 
178 	if (nla_put_u32(rsp, NETDEV_A_NAPI_ID, napi->napi_id))
179 		goto nla_put_failure;
180 
181 	if (nla_put_u32(rsp, NETDEV_A_NAPI_IFINDEX, napi->dev->ifindex))
182 		goto nla_put_failure;
183 
184 	if (napi->irq >= 0 && nla_put_u32(rsp, NETDEV_A_NAPI_IRQ, napi->irq))
185 		goto nla_put_failure;
186 
187 	if (napi->thread) {
188 		pid = task_pid_nr(napi->thread);
189 		if (nla_put_u32(rsp, NETDEV_A_NAPI_PID, pid))
190 			goto nla_put_failure;
191 	}
192 
193 	napi_defer_hard_irqs = napi_get_defer_hard_irqs(napi);
194 	if (nla_put_s32(rsp, NETDEV_A_NAPI_DEFER_HARD_IRQS,
195 			napi_defer_hard_irqs))
196 		goto nla_put_failure;
197 
198 	irq_suspend_timeout = napi_get_irq_suspend_timeout(napi);
199 	if (nla_put_uint(rsp, NETDEV_A_NAPI_IRQ_SUSPEND_TIMEOUT,
200 			 irq_suspend_timeout))
201 		goto nla_put_failure;
202 
203 	gro_flush_timeout = napi_get_gro_flush_timeout(napi);
204 	if (nla_put_uint(rsp, NETDEV_A_NAPI_GRO_FLUSH_TIMEOUT,
205 			 gro_flush_timeout))
206 		goto nla_put_failure;
207 
208 	genlmsg_end(rsp, hdr);
209 
210 	return 0;
211 
212 nla_put_failure:
213 	genlmsg_cancel(rsp, hdr);
214 	return -EMSGSIZE;
215 }
216 
217 int netdev_nl_napi_get_doit(struct sk_buff *skb, struct genl_info *info)
218 {
219 	struct napi_struct *napi;
220 	struct sk_buff *rsp;
221 	u32 napi_id;
222 	int err;
223 
224 	if (GENL_REQ_ATTR_CHECK(info, NETDEV_A_NAPI_ID))
225 		return -EINVAL;
226 
227 	napi_id = nla_get_u32(info->attrs[NETDEV_A_NAPI_ID]);
228 
229 	rsp = genlmsg_new(GENLMSG_DEFAULT_SIZE, GFP_KERNEL);
230 	if (!rsp)
231 		return -ENOMEM;
232 
233 	napi = netdev_napi_by_id_lock(genl_info_net(info), napi_id);
234 	if (napi) {
235 		err = netdev_nl_napi_fill_one(rsp, napi, info);
236 		netdev_unlock(napi->dev);
237 	} else {
238 		NL_SET_BAD_ATTR(info->extack, info->attrs[NETDEV_A_NAPI_ID]);
239 		err = -ENOENT;
240 	}
241 
242 	if (err) {
243 		goto err_free_msg;
244 	} else if (!rsp->len) {
245 		err = -ENOENT;
246 		goto err_free_msg;
247 	}
248 
249 	return genlmsg_reply(rsp, info);
250 
251 err_free_msg:
252 	nlmsg_free(rsp);
253 	return err;
254 }
255 
256 static int
257 netdev_nl_napi_dump_one(struct net_device *netdev, struct sk_buff *rsp,
258 			const struct genl_info *info,
259 			struct netdev_nl_dump_ctx *ctx)
260 {
261 	struct napi_struct *napi;
262 	unsigned int prev_id;
263 	int err = 0;
264 
265 	if (!netdev->up)
266 		return err;
267 
268 	prev_id = UINT_MAX;
269 	list_for_each_entry(napi, &netdev->napi_list, dev_list) {
270 		if (napi->napi_id < MIN_NAPI_ID)
271 			continue;
272 
273 		/* Dump continuation below depends on the list being sorted */
274 		WARN_ON_ONCE(napi->napi_id >= prev_id);
275 		prev_id = napi->napi_id;
276 
277 		if (ctx->napi_id && napi->napi_id >= ctx->napi_id)
278 			continue;
279 
280 		err = netdev_nl_napi_fill_one(rsp, napi, info);
281 		if (err)
282 			return err;
283 		ctx->napi_id = napi->napi_id;
284 	}
285 	return err;
286 }
287 
288 int netdev_nl_napi_get_dumpit(struct sk_buff *skb, struct netlink_callback *cb)
289 {
290 	struct netdev_nl_dump_ctx *ctx = netdev_dump_ctx(cb);
291 	const struct genl_info *info = genl_info_dump(cb);
292 	struct net *net = sock_net(skb->sk);
293 	struct net_device *netdev;
294 	u32 ifindex = 0;
295 	int err = 0;
296 
297 	if (info->attrs[NETDEV_A_NAPI_IFINDEX])
298 		ifindex = nla_get_u32(info->attrs[NETDEV_A_NAPI_IFINDEX]);
299 
300 	if (ifindex) {
301 		netdev = netdev_get_by_index_lock(net, ifindex);
302 		if (netdev) {
303 			err = netdev_nl_napi_dump_one(netdev, skb, info, ctx);
304 			netdev_unlock(netdev);
305 		} else {
306 			err = -ENODEV;
307 		}
308 	} else {
309 		for_each_netdev_lock_scoped(net, netdev, ctx->ifindex) {
310 			err = netdev_nl_napi_dump_one(netdev, skb, info, ctx);
311 			if (err < 0)
312 				break;
313 			ctx->napi_id = 0;
314 		}
315 	}
316 
317 	return err;
318 }
319 
320 static int
321 netdev_nl_napi_set_config(struct napi_struct *napi, struct genl_info *info)
322 {
323 	u64 irq_suspend_timeout = 0;
324 	u64 gro_flush_timeout = 0;
325 	u32 defer = 0;
326 
327 	if (info->attrs[NETDEV_A_NAPI_DEFER_HARD_IRQS]) {
328 		defer = nla_get_u32(info->attrs[NETDEV_A_NAPI_DEFER_HARD_IRQS]);
329 		napi_set_defer_hard_irqs(napi, defer);
330 	}
331 
332 	if (info->attrs[NETDEV_A_NAPI_IRQ_SUSPEND_TIMEOUT]) {
333 		irq_suspend_timeout = nla_get_uint(info->attrs[NETDEV_A_NAPI_IRQ_SUSPEND_TIMEOUT]);
334 		napi_set_irq_suspend_timeout(napi, irq_suspend_timeout);
335 	}
336 
337 	if (info->attrs[NETDEV_A_NAPI_GRO_FLUSH_TIMEOUT]) {
338 		gro_flush_timeout = nla_get_uint(info->attrs[NETDEV_A_NAPI_GRO_FLUSH_TIMEOUT]);
339 		napi_set_gro_flush_timeout(napi, gro_flush_timeout);
340 	}
341 
342 	return 0;
343 }
344 
345 int netdev_nl_napi_set_doit(struct sk_buff *skb, struct genl_info *info)
346 {
347 	struct napi_struct *napi;
348 	unsigned int napi_id;
349 	int err;
350 
351 	if (GENL_REQ_ATTR_CHECK(info, NETDEV_A_NAPI_ID))
352 		return -EINVAL;
353 
354 	napi_id = nla_get_u32(info->attrs[NETDEV_A_NAPI_ID]);
355 
356 	napi = netdev_napi_by_id_lock(genl_info_net(info), napi_id);
357 	if (napi) {
358 		err = netdev_nl_napi_set_config(napi, info);
359 		netdev_unlock(napi->dev);
360 	} else {
361 		NL_SET_BAD_ATTR(info->extack, info->attrs[NETDEV_A_NAPI_ID]);
362 		err = -ENOENT;
363 	}
364 
365 	return err;
366 }
367 
368 static int nla_put_napi_id(struct sk_buff *skb, const struct napi_struct *napi)
369 {
370 	if (napi && napi_id_valid(napi->napi_id))
371 		return nla_put_u32(skb, NETDEV_A_QUEUE_NAPI_ID, napi->napi_id);
372 	return 0;
373 }
374 
375 static int
376 netdev_nl_queue_fill_one(struct sk_buff *rsp, struct net_device *netdev,
377 			 u32 q_idx, u32 q_type, const struct genl_info *info)
378 {
379 	struct pp_memory_provider_params *params;
380 	struct netdev_rx_queue *rxq;
381 	struct netdev_queue *txq;
382 	void *hdr;
383 
384 	hdr = genlmsg_iput(rsp, info);
385 	if (!hdr)
386 		return -EMSGSIZE;
387 
388 	if (nla_put_u32(rsp, NETDEV_A_QUEUE_ID, q_idx) ||
389 	    nla_put_u32(rsp, NETDEV_A_QUEUE_TYPE, q_type) ||
390 	    nla_put_u32(rsp, NETDEV_A_QUEUE_IFINDEX, netdev->ifindex))
391 		goto nla_put_failure;
392 
393 	switch (q_type) {
394 	case NETDEV_QUEUE_TYPE_RX:
395 		rxq = __netif_get_rx_queue(netdev, q_idx);
396 		if (nla_put_napi_id(rsp, rxq->napi))
397 			goto nla_put_failure;
398 
399 		params = &rxq->mp_params;
400 		if (params->mp_ops &&
401 		    params->mp_ops->nl_fill(params->mp_priv, rsp, rxq))
402 			goto nla_put_failure;
403 		break;
404 	case NETDEV_QUEUE_TYPE_TX:
405 		txq = netdev_get_tx_queue(netdev, q_idx);
406 		if (nla_put_napi_id(rsp, txq->napi))
407 			goto nla_put_failure;
408 	}
409 
410 	genlmsg_end(rsp, hdr);
411 
412 	return 0;
413 
414 nla_put_failure:
415 	genlmsg_cancel(rsp, hdr);
416 	return -EMSGSIZE;
417 }
418 
419 static int netdev_nl_queue_validate(struct net_device *netdev, u32 q_id,
420 				    u32 q_type)
421 {
422 	switch (q_type) {
423 	case NETDEV_QUEUE_TYPE_RX:
424 		if (q_id >= netdev->real_num_rx_queues)
425 			return -EINVAL;
426 		return 0;
427 	case NETDEV_QUEUE_TYPE_TX:
428 		if (q_id >= netdev->real_num_tx_queues)
429 			return -EINVAL;
430 	}
431 	return 0;
432 }
433 
434 static int
435 netdev_nl_queue_fill(struct sk_buff *rsp, struct net_device *netdev, u32 q_idx,
436 		     u32 q_type, const struct genl_info *info)
437 {
438 	int err;
439 
440 	if (!netdev->up)
441 		return -ENOENT;
442 
443 	err = netdev_nl_queue_validate(netdev, q_idx, q_type);
444 	if (err)
445 		return err;
446 
447 	return netdev_nl_queue_fill_one(rsp, netdev, q_idx, q_type, info);
448 }
449 
450 int netdev_nl_queue_get_doit(struct sk_buff *skb, struct genl_info *info)
451 {
452 	u32 q_id, q_type, ifindex;
453 	struct net_device *netdev;
454 	struct sk_buff *rsp;
455 	int err;
456 
457 	if (GENL_REQ_ATTR_CHECK(info, NETDEV_A_QUEUE_ID) ||
458 	    GENL_REQ_ATTR_CHECK(info, NETDEV_A_QUEUE_TYPE) ||
459 	    GENL_REQ_ATTR_CHECK(info, NETDEV_A_QUEUE_IFINDEX))
460 		return -EINVAL;
461 
462 	q_id = nla_get_u32(info->attrs[NETDEV_A_QUEUE_ID]);
463 	q_type = nla_get_u32(info->attrs[NETDEV_A_QUEUE_TYPE]);
464 	ifindex = nla_get_u32(info->attrs[NETDEV_A_QUEUE_IFINDEX]);
465 
466 	rsp = genlmsg_new(GENLMSG_DEFAULT_SIZE, GFP_KERNEL);
467 	if (!rsp)
468 		return -ENOMEM;
469 
470 	rtnl_lock();
471 
472 	netdev = netdev_get_by_index_lock(genl_info_net(info), ifindex);
473 	if (netdev) {
474 		err = netdev_nl_queue_fill(rsp, netdev, q_id, q_type, info);
475 		netdev_unlock(netdev);
476 	} else {
477 		err = -ENODEV;
478 	}
479 
480 	rtnl_unlock();
481 
482 	if (err)
483 		goto err_free_msg;
484 
485 	return genlmsg_reply(rsp, info);
486 
487 err_free_msg:
488 	nlmsg_free(rsp);
489 	return err;
490 }
491 
492 static int
493 netdev_nl_queue_dump_one(struct net_device *netdev, struct sk_buff *rsp,
494 			 const struct genl_info *info,
495 			 struct netdev_nl_dump_ctx *ctx)
496 {
497 	int err = 0;
498 
499 	if (!netdev->up)
500 		return err;
501 
502 	for (; ctx->rxq_idx < netdev->real_num_rx_queues; ctx->rxq_idx++) {
503 		err = netdev_nl_queue_fill_one(rsp, netdev, ctx->rxq_idx,
504 					       NETDEV_QUEUE_TYPE_RX, info);
505 		if (err)
506 			return err;
507 	}
508 	for (; ctx->txq_idx < netdev->real_num_tx_queues; ctx->txq_idx++) {
509 		err = netdev_nl_queue_fill_one(rsp, netdev, ctx->txq_idx,
510 					       NETDEV_QUEUE_TYPE_TX, info);
511 		if (err)
512 			return err;
513 	}
514 
515 	return err;
516 }
517 
518 int netdev_nl_queue_get_dumpit(struct sk_buff *skb, struct netlink_callback *cb)
519 {
520 	struct netdev_nl_dump_ctx *ctx = netdev_dump_ctx(cb);
521 	const struct genl_info *info = genl_info_dump(cb);
522 	struct net *net = sock_net(skb->sk);
523 	struct net_device *netdev;
524 	u32 ifindex = 0;
525 	int err = 0;
526 
527 	if (info->attrs[NETDEV_A_QUEUE_IFINDEX])
528 		ifindex = nla_get_u32(info->attrs[NETDEV_A_QUEUE_IFINDEX]);
529 
530 	rtnl_lock();
531 	if (ifindex) {
532 		netdev = netdev_get_by_index_lock(net, ifindex);
533 		if (netdev) {
534 			err = netdev_nl_queue_dump_one(netdev, skb, info, ctx);
535 			netdev_unlock(netdev);
536 		} else {
537 			err = -ENODEV;
538 		}
539 	} else {
540 		for_each_netdev_lock_scoped(net, netdev, ctx->ifindex) {
541 			err = netdev_nl_queue_dump_one(netdev, skb, info, ctx);
542 			if (err < 0)
543 				break;
544 			ctx->rxq_idx = 0;
545 			ctx->txq_idx = 0;
546 		}
547 	}
548 	rtnl_unlock();
549 
550 	return err;
551 }
552 
553 #define NETDEV_STAT_NOT_SET		(~0ULL)
554 
555 static void netdev_nl_stats_add(void *_sum, const void *_add, size_t size)
556 {
557 	const u64 *add = _add;
558 	u64 *sum = _sum;
559 
560 	while (size) {
561 		if (*add != NETDEV_STAT_NOT_SET && *sum != NETDEV_STAT_NOT_SET)
562 			*sum += *add;
563 		sum++;
564 		add++;
565 		size -= 8;
566 	}
567 }
568 
569 static int netdev_stat_put(struct sk_buff *rsp, unsigned int attr_id, u64 value)
570 {
571 	if (value == NETDEV_STAT_NOT_SET)
572 		return 0;
573 	return nla_put_uint(rsp, attr_id, value);
574 }
575 
576 static int
577 netdev_nl_stats_write_rx(struct sk_buff *rsp, struct netdev_queue_stats_rx *rx)
578 {
579 	if (netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_PACKETS, rx->packets) ||
580 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_BYTES, rx->bytes) ||
581 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_ALLOC_FAIL, rx->alloc_fail) ||
582 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_HW_DROPS, rx->hw_drops) ||
583 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_HW_DROP_OVERRUNS, rx->hw_drop_overruns) ||
584 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_CSUM_COMPLETE, rx->csum_complete) ||
585 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_CSUM_UNNECESSARY, rx->csum_unnecessary) ||
586 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_CSUM_NONE, rx->csum_none) ||
587 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_CSUM_BAD, rx->csum_bad) ||
588 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_HW_GRO_PACKETS, rx->hw_gro_packets) ||
589 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_HW_GRO_BYTES, rx->hw_gro_bytes) ||
590 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_HW_GRO_WIRE_PACKETS, rx->hw_gro_wire_packets) ||
591 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_HW_GRO_WIRE_BYTES, rx->hw_gro_wire_bytes) ||
592 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_RX_HW_DROP_RATELIMITS, rx->hw_drop_ratelimits))
593 		return -EMSGSIZE;
594 	return 0;
595 }
596 
597 static int
598 netdev_nl_stats_write_tx(struct sk_buff *rsp, struct netdev_queue_stats_tx *tx)
599 {
600 	if (netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_PACKETS, tx->packets) ||
601 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_BYTES, tx->bytes) ||
602 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_HW_DROPS, tx->hw_drops) ||
603 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_HW_DROP_ERRORS, tx->hw_drop_errors) ||
604 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_CSUM_NONE, tx->csum_none) ||
605 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_NEEDS_CSUM, tx->needs_csum) ||
606 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_HW_GSO_PACKETS, tx->hw_gso_packets) ||
607 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_HW_GSO_BYTES, tx->hw_gso_bytes) ||
608 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_HW_GSO_WIRE_PACKETS, tx->hw_gso_wire_packets) ||
609 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_HW_GSO_WIRE_BYTES, tx->hw_gso_wire_bytes) ||
610 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_HW_DROP_RATELIMITS, tx->hw_drop_ratelimits) ||
611 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_STOP, tx->stop) ||
612 	    netdev_stat_put(rsp, NETDEV_A_QSTATS_TX_WAKE, tx->wake))
613 		return -EMSGSIZE;
614 	return 0;
615 }
616 
617 static int
618 netdev_nl_stats_queue(struct net_device *netdev, struct sk_buff *rsp,
619 		      u32 q_type, int i, const struct genl_info *info)
620 {
621 	const struct netdev_stat_ops *ops = netdev->stat_ops;
622 	struct netdev_queue_stats_rx rx;
623 	struct netdev_queue_stats_tx tx;
624 	void *hdr;
625 
626 	hdr = genlmsg_iput(rsp, info);
627 	if (!hdr)
628 		return -EMSGSIZE;
629 	if (nla_put_u32(rsp, NETDEV_A_QSTATS_IFINDEX, netdev->ifindex) ||
630 	    nla_put_u32(rsp, NETDEV_A_QSTATS_QUEUE_TYPE, q_type) ||
631 	    nla_put_u32(rsp, NETDEV_A_QSTATS_QUEUE_ID, i))
632 		goto nla_put_failure;
633 
634 	switch (q_type) {
635 	case NETDEV_QUEUE_TYPE_RX:
636 		memset(&rx, 0xff, sizeof(rx));
637 		ops->get_queue_stats_rx(netdev, i, &rx);
638 		if (!memchr_inv(&rx, 0xff, sizeof(rx)))
639 			goto nla_cancel;
640 		if (netdev_nl_stats_write_rx(rsp, &rx))
641 			goto nla_put_failure;
642 		break;
643 	case NETDEV_QUEUE_TYPE_TX:
644 		memset(&tx, 0xff, sizeof(tx));
645 		ops->get_queue_stats_tx(netdev, i, &tx);
646 		if (!memchr_inv(&tx, 0xff, sizeof(tx)))
647 			goto nla_cancel;
648 		if (netdev_nl_stats_write_tx(rsp, &tx))
649 			goto nla_put_failure;
650 		break;
651 	}
652 
653 	genlmsg_end(rsp, hdr);
654 	return 0;
655 
656 nla_cancel:
657 	genlmsg_cancel(rsp, hdr);
658 	return 0;
659 nla_put_failure:
660 	genlmsg_cancel(rsp, hdr);
661 	return -EMSGSIZE;
662 }
663 
664 static int
665 netdev_nl_stats_by_queue(struct net_device *netdev, struct sk_buff *rsp,
666 			 const struct genl_info *info,
667 			 struct netdev_nl_dump_ctx *ctx)
668 {
669 	const struct netdev_stat_ops *ops = netdev->stat_ops;
670 	int i, err;
671 
672 	if (!(netdev->flags & IFF_UP))
673 		return 0;
674 
675 	i = ctx->rxq_idx;
676 	while (ops->get_queue_stats_rx && i < netdev->real_num_rx_queues) {
677 		err = netdev_nl_stats_queue(netdev, rsp, NETDEV_QUEUE_TYPE_RX,
678 					    i, info);
679 		if (err)
680 			return err;
681 		ctx->rxq_idx = ++i;
682 	}
683 	i = ctx->txq_idx;
684 	while (ops->get_queue_stats_tx && i < netdev->real_num_tx_queues) {
685 		err = netdev_nl_stats_queue(netdev, rsp, NETDEV_QUEUE_TYPE_TX,
686 					    i, info);
687 		if (err)
688 			return err;
689 		ctx->txq_idx = ++i;
690 	}
691 
692 	ctx->rxq_idx = 0;
693 	ctx->txq_idx = 0;
694 	return 0;
695 }
696 
697 static int
698 netdev_nl_stats_by_netdev(struct net_device *netdev, struct sk_buff *rsp,
699 			  const struct genl_info *info)
700 {
701 	struct netdev_queue_stats_rx rx_sum, rx;
702 	struct netdev_queue_stats_tx tx_sum, tx;
703 	const struct netdev_stat_ops *ops;
704 	void *hdr;
705 	int i;
706 
707 	ops = netdev->stat_ops;
708 	/* Netdev can't guarantee any complete counters */
709 	if (!ops->get_base_stats)
710 		return 0;
711 
712 	memset(&rx_sum, 0xff, sizeof(rx_sum));
713 	memset(&tx_sum, 0xff, sizeof(tx_sum));
714 
715 	ops->get_base_stats(netdev, &rx_sum, &tx_sum);
716 
717 	/* The op was there, but nothing reported, don't bother */
718 	if (!memchr_inv(&rx_sum, 0xff, sizeof(rx_sum)) &&
719 	    !memchr_inv(&tx_sum, 0xff, sizeof(tx_sum)))
720 		return 0;
721 
722 	hdr = genlmsg_iput(rsp, info);
723 	if (!hdr)
724 		return -EMSGSIZE;
725 	if (nla_put_u32(rsp, NETDEV_A_QSTATS_IFINDEX, netdev->ifindex))
726 		goto nla_put_failure;
727 
728 	for (i = 0; i < netdev->real_num_rx_queues; i++) {
729 		memset(&rx, 0xff, sizeof(rx));
730 		if (ops->get_queue_stats_rx)
731 			ops->get_queue_stats_rx(netdev, i, &rx);
732 		netdev_nl_stats_add(&rx_sum, &rx, sizeof(rx));
733 	}
734 	for (i = 0; i < netdev->real_num_tx_queues; i++) {
735 		memset(&tx, 0xff, sizeof(tx));
736 		if (ops->get_queue_stats_tx)
737 			ops->get_queue_stats_tx(netdev, i, &tx);
738 		netdev_nl_stats_add(&tx_sum, &tx, sizeof(tx));
739 	}
740 
741 	if (netdev_nl_stats_write_rx(rsp, &rx_sum) ||
742 	    netdev_nl_stats_write_tx(rsp, &tx_sum))
743 		goto nla_put_failure;
744 
745 	genlmsg_end(rsp, hdr);
746 	return 0;
747 
748 nla_put_failure:
749 	genlmsg_cancel(rsp, hdr);
750 	return -EMSGSIZE;
751 }
752 
753 static int
754 netdev_nl_qstats_get_dump_one(struct net_device *netdev, unsigned int scope,
755 			      struct sk_buff *skb, const struct genl_info *info,
756 			      struct netdev_nl_dump_ctx *ctx)
757 {
758 	if (!netdev->stat_ops)
759 		return 0;
760 
761 	switch (scope) {
762 	case 0:
763 		return netdev_nl_stats_by_netdev(netdev, skb, info);
764 	case NETDEV_QSTATS_SCOPE_QUEUE:
765 		return netdev_nl_stats_by_queue(netdev, skb, info, ctx);
766 	}
767 
768 	return -EINVAL;	/* Should not happen, per netlink policy */
769 }
770 
771 int netdev_nl_qstats_get_dumpit(struct sk_buff *skb,
772 				struct netlink_callback *cb)
773 {
774 	struct netdev_nl_dump_ctx *ctx = netdev_dump_ctx(cb);
775 	const struct genl_info *info = genl_info_dump(cb);
776 	struct net *net = sock_net(skb->sk);
777 	struct net_device *netdev;
778 	unsigned int ifindex;
779 	unsigned int scope;
780 	int err = 0;
781 
782 	scope = 0;
783 	if (info->attrs[NETDEV_A_QSTATS_SCOPE])
784 		scope = nla_get_uint(info->attrs[NETDEV_A_QSTATS_SCOPE]);
785 
786 	ifindex = 0;
787 	if (info->attrs[NETDEV_A_QSTATS_IFINDEX])
788 		ifindex = nla_get_u32(info->attrs[NETDEV_A_QSTATS_IFINDEX]);
789 
790 	rtnl_lock();
791 	if (ifindex) {
792 		netdev = __dev_get_by_index(net, ifindex);
793 		if (netdev && netdev->stat_ops) {
794 			err = netdev_nl_qstats_get_dump_one(netdev, scope, skb,
795 							    info, ctx);
796 		} else {
797 			NL_SET_BAD_ATTR(info->extack,
798 					info->attrs[NETDEV_A_QSTATS_IFINDEX]);
799 			err = netdev ? -EOPNOTSUPP : -ENODEV;
800 		}
801 	} else {
802 		for_each_netdev_dump(net, netdev, ctx->ifindex) {
803 			err = netdev_nl_qstats_get_dump_one(netdev, scope, skb,
804 							    info, ctx);
805 			if (err < 0)
806 				break;
807 		}
808 	}
809 	rtnl_unlock();
810 
811 	return err;
812 }
813 
814 int netdev_nl_bind_rx_doit(struct sk_buff *skb, struct genl_info *info)
815 {
816 	struct nlattr *tb[ARRAY_SIZE(netdev_queue_id_nl_policy)];
817 	struct net_devmem_dmabuf_binding *binding;
818 	struct list_head *sock_binding_list;
819 	u32 ifindex, dmabuf_fd, rxq_idx;
820 	struct net_device *netdev;
821 	struct sk_buff *rsp;
822 	struct nlattr *attr;
823 	int rem, err = 0;
824 	void *hdr;
825 
826 	if (GENL_REQ_ATTR_CHECK(info, NETDEV_A_DEV_IFINDEX) ||
827 	    GENL_REQ_ATTR_CHECK(info, NETDEV_A_DMABUF_FD) ||
828 	    GENL_REQ_ATTR_CHECK(info, NETDEV_A_DMABUF_QUEUES))
829 		return -EINVAL;
830 
831 	ifindex = nla_get_u32(info->attrs[NETDEV_A_DEV_IFINDEX]);
832 	dmabuf_fd = nla_get_u32(info->attrs[NETDEV_A_DMABUF_FD]);
833 
834 	sock_binding_list = genl_sk_priv_get(&netdev_nl_family,
835 					     NETLINK_CB(skb).sk);
836 	if (IS_ERR(sock_binding_list))
837 		return PTR_ERR(sock_binding_list);
838 
839 	rsp = genlmsg_new(GENLMSG_DEFAULT_SIZE, GFP_KERNEL);
840 	if (!rsp)
841 		return -ENOMEM;
842 
843 	hdr = genlmsg_iput(rsp, info);
844 	if (!hdr) {
845 		err = -EMSGSIZE;
846 		goto err_genlmsg_free;
847 	}
848 
849 	rtnl_lock();
850 
851 	netdev = __dev_get_by_index(genl_info_net(info), ifindex);
852 	if (!netdev || !netif_device_present(netdev)) {
853 		err = -ENODEV;
854 		goto err_unlock;
855 	}
856 
857 	if (dev_xdp_prog_count(netdev)) {
858 		NL_SET_ERR_MSG(info->extack, "unable to bind dmabuf to device with XDP program attached");
859 		err = -EEXIST;
860 		goto err_unlock;
861 	}
862 
863 	binding = net_devmem_bind_dmabuf(netdev, dmabuf_fd, info->extack);
864 	if (IS_ERR(binding)) {
865 		err = PTR_ERR(binding);
866 		goto err_unlock;
867 	}
868 
869 	nla_for_each_attr_type(attr, NETDEV_A_DMABUF_QUEUES,
870 			       genlmsg_data(info->genlhdr),
871 			       genlmsg_len(info->genlhdr), rem) {
872 		err = nla_parse_nested(
873 			tb, ARRAY_SIZE(netdev_queue_id_nl_policy) - 1, attr,
874 			netdev_queue_id_nl_policy, info->extack);
875 		if (err < 0)
876 			goto err_unbind;
877 
878 		if (NL_REQ_ATTR_CHECK(info->extack, attr, tb, NETDEV_A_QUEUE_ID) ||
879 		    NL_REQ_ATTR_CHECK(info->extack, attr, tb, NETDEV_A_QUEUE_TYPE)) {
880 			err = -EINVAL;
881 			goto err_unbind;
882 		}
883 
884 		if (nla_get_u32(tb[NETDEV_A_QUEUE_TYPE]) != NETDEV_QUEUE_TYPE_RX) {
885 			NL_SET_BAD_ATTR(info->extack, tb[NETDEV_A_QUEUE_TYPE]);
886 			err = -EINVAL;
887 			goto err_unbind;
888 		}
889 
890 		rxq_idx = nla_get_u32(tb[NETDEV_A_QUEUE_ID]);
891 
892 		err = net_devmem_bind_dmabuf_to_queue(netdev, rxq_idx, binding,
893 						      info->extack);
894 		if (err)
895 			goto err_unbind;
896 	}
897 
898 	list_add(&binding->list, sock_binding_list);
899 
900 	nla_put_u32(rsp, NETDEV_A_DMABUF_ID, binding->id);
901 	genlmsg_end(rsp, hdr);
902 
903 	err = genlmsg_reply(rsp, info);
904 	if (err)
905 		goto err_unbind;
906 
907 	rtnl_unlock();
908 
909 	return 0;
910 
911 err_unbind:
912 	net_devmem_unbind_dmabuf(binding);
913 err_unlock:
914 	rtnl_unlock();
915 err_genlmsg_free:
916 	nlmsg_free(rsp);
917 	return err;
918 }
919 
920 void netdev_nl_sock_priv_init(struct list_head *priv)
921 {
922 	INIT_LIST_HEAD(priv);
923 }
924 
925 void netdev_nl_sock_priv_destroy(struct list_head *priv)
926 {
927 	struct net_devmem_dmabuf_binding *binding;
928 	struct net_devmem_dmabuf_binding *temp;
929 
930 	list_for_each_entry_safe(binding, temp, priv, list) {
931 		rtnl_lock();
932 		net_devmem_unbind_dmabuf(binding);
933 		rtnl_unlock();
934 	}
935 }
936 
937 static int netdev_genl_netdevice_event(struct notifier_block *nb,
938 				       unsigned long event, void *ptr)
939 {
940 	struct net_device *netdev = netdev_notifier_info_to_dev(ptr);
941 
942 	switch (event) {
943 	case NETDEV_REGISTER:
944 		netdev_genl_dev_notify(netdev, NETDEV_CMD_DEV_ADD_NTF);
945 		break;
946 	case NETDEV_UNREGISTER:
947 		netdev_genl_dev_notify(netdev, NETDEV_CMD_DEV_DEL_NTF);
948 		break;
949 	case NETDEV_XDP_FEAT_CHANGE:
950 		netdev_genl_dev_notify(netdev, NETDEV_CMD_DEV_CHANGE_NTF);
951 		break;
952 	}
953 
954 	return NOTIFY_OK;
955 }
956 
957 static struct notifier_block netdev_genl_nb = {
958 	.notifier_call	= netdev_genl_netdevice_event,
959 };
960 
961 static int __init netdev_genl_init(void)
962 {
963 	int err;
964 
965 	err = register_netdevice_notifier(&netdev_genl_nb);
966 	if (err)
967 		return err;
968 
969 	err = genl_register_family(&netdev_nl_family);
970 	if (err)
971 		goto err_unreg_ntf;
972 
973 	return 0;
974 
975 err_unreg_ntf:
976 	unregister_netdevice_notifier(&netdev_genl_nb);
977 	return err;
978 }
979 
980 subsys_initcall(netdev_genl_init);
981