xref: /openbmc/linux/drivers/net/ethernet/microsoft/mana/mana_en.c (revision f74f1ec22dc232be0296739148d126e9158eadf9)
1 // SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
2 /* Copyright (c) 2021, Microsoft Corporation. */
3 
4 #include <uapi/linux/bpf.h>
5 
6 #include <linux/inetdevice.h>
7 #include <linux/etherdevice.h>
8 #include <linux/ethtool.h>
9 #include <linux/filter.h>
10 #include <linux/mm.h>
11 
12 #include <net/checksum.h>
13 #include <net/ip6_checksum.h>
14 
15 #include <net/mana/mana.h>
16 #include <net/mana/mana_auxiliary.h>
17 
18 static DEFINE_IDA(mana_adev_ida);
19 
20 static int mana_adev_idx_alloc(void)
21 {
22 	return ida_alloc(&mana_adev_ida, GFP_KERNEL);
23 }
24 
25 static void mana_adev_idx_free(int idx)
26 {
27 	ida_free(&mana_adev_ida, idx);
28 }
29 
30 /* Microsoft Azure Network Adapter (MANA) functions */
31 
32 static int mana_open(struct net_device *ndev)
33 {
34 	struct mana_port_context *apc = netdev_priv(ndev);
35 	int err;
36 
37 	err = mana_alloc_queues(ndev);
38 	if (err)
39 		return err;
40 
41 	apc->port_is_up = true;
42 
43 	/* Ensure port state updated before txq state */
44 	smp_wmb();
45 
46 	netif_carrier_on(ndev);
47 	netif_tx_wake_all_queues(ndev);
48 
49 	return 0;
50 }
51 
52 static int mana_close(struct net_device *ndev)
53 {
54 	struct mana_port_context *apc = netdev_priv(ndev);
55 
56 	if (!apc->port_is_up)
57 		return 0;
58 
59 	return mana_detach(ndev, true);
60 }
61 
62 static bool mana_can_tx(struct gdma_queue *wq)
63 {
64 	return mana_gd_wq_avail_space(wq) >= MAX_TX_WQE_SIZE;
65 }
66 
67 static unsigned int mana_checksum_info(struct sk_buff *skb)
68 {
69 	if (skb->protocol == htons(ETH_P_IP)) {
70 		struct iphdr *ip = ip_hdr(skb);
71 
72 		if (ip->protocol == IPPROTO_TCP)
73 			return IPPROTO_TCP;
74 
75 		if (ip->protocol == IPPROTO_UDP)
76 			return IPPROTO_UDP;
77 	} else if (skb->protocol == htons(ETH_P_IPV6)) {
78 		struct ipv6hdr *ip6 = ipv6_hdr(skb);
79 
80 		if (ip6->nexthdr == IPPROTO_TCP)
81 			return IPPROTO_TCP;
82 
83 		if (ip6->nexthdr == IPPROTO_UDP)
84 			return IPPROTO_UDP;
85 	}
86 
87 	/* No csum offloading */
88 	return 0;
89 }
90 
91 static int mana_map_skb(struct sk_buff *skb, struct mana_port_context *apc,
92 			struct mana_tx_package *tp)
93 {
94 	struct mana_skb_head *ash = (struct mana_skb_head *)skb->head;
95 	struct gdma_dev *gd = apc->ac->gdma_dev;
96 	struct gdma_context *gc;
97 	struct device *dev;
98 	skb_frag_t *frag;
99 	dma_addr_t da;
100 	int i;
101 
102 	gc = gd->gdma_context;
103 	dev = gc->dev;
104 	da = dma_map_single(dev, skb->data, skb_headlen(skb), DMA_TO_DEVICE);
105 
106 	if (dma_mapping_error(dev, da))
107 		return -ENOMEM;
108 
109 	ash->dma_handle[0] = da;
110 	ash->size[0] = skb_headlen(skb);
111 
112 	tp->wqe_req.sgl[0].address = ash->dma_handle[0];
113 	tp->wqe_req.sgl[0].mem_key = gd->gpa_mkey;
114 	tp->wqe_req.sgl[0].size = ash->size[0];
115 
116 	for (i = 0; i < skb_shinfo(skb)->nr_frags; i++) {
117 		frag = &skb_shinfo(skb)->frags[i];
118 		da = skb_frag_dma_map(dev, frag, 0, skb_frag_size(frag),
119 				      DMA_TO_DEVICE);
120 
121 		if (dma_mapping_error(dev, da))
122 			goto frag_err;
123 
124 		ash->dma_handle[i + 1] = da;
125 		ash->size[i + 1] = skb_frag_size(frag);
126 
127 		tp->wqe_req.sgl[i + 1].address = ash->dma_handle[i + 1];
128 		tp->wqe_req.sgl[i + 1].mem_key = gd->gpa_mkey;
129 		tp->wqe_req.sgl[i + 1].size = ash->size[i + 1];
130 	}
131 
132 	return 0;
133 
134 frag_err:
135 	for (i = i - 1; i >= 0; i--)
136 		dma_unmap_page(dev, ash->dma_handle[i + 1], ash->size[i + 1],
137 			       DMA_TO_DEVICE);
138 
139 	dma_unmap_single(dev, ash->dma_handle[0], ash->size[0], DMA_TO_DEVICE);
140 
141 	return -ENOMEM;
142 }
143 
144 netdev_tx_t mana_start_xmit(struct sk_buff *skb, struct net_device *ndev)
145 {
146 	enum mana_tx_pkt_format pkt_fmt = MANA_SHORT_PKT_FMT;
147 	struct mana_port_context *apc = netdev_priv(ndev);
148 	u16 txq_idx = skb_get_queue_mapping(skb);
149 	struct gdma_dev *gd = apc->ac->gdma_dev;
150 	bool ipv4 = false, ipv6 = false;
151 	struct mana_tx_package pkg = {};
152 	struct netdev_queue *net_txq;
153 	struct mana_stats_tx *tx_stats;
154 	struct gdma_queue *gdma_sq;
155 	unsigned int csum_type;
156 	struct mana_txq *txq;
157 	struct mana_cq *cq;
158 	int err, len;
159 
160 	if (unlikely(!apc->port_is_up))
161 		goto tx_drop;
162 
163 	if (skb_cow_head(skb, MANA_HEADROOM))
164 		goto tx_drop_count;
165 
166 	txq = &apc->tx_qp[txq_idx].txq;
167 	gdma_sq = txq->gdma_sq;
168 	cq = &apc->tx_qp[txq_idx].tx_cq;
169 
170 	pkg.tx_oob.s_oob.vcq_num = cq->gdma_id;
171 	pkg.tx_oob.s_oob.vsq_frame = txq->vsq_frame;
172 
173 	if (txq->vp_offset > MANA_SHORT_VPORT_OFFSET_MAX) {
174 		pkg.tx_oob.l_oob.long_vp_offset = txq->vp_offset;
175 		pkt_fmt = MANA_LONG_PKT_FMT;
176 	} else {
177 		pkg.tx_oob.s_oob.short_vp_offset = txq->vp_offset;
178 	}
179 
180 	pkg.tx_oob.s_oob.pkt_fmt = pkt_fmt;
181 
182 	if (pkt_fmt == MANA_SHORT_PKT_FMT)
183 		pkg.wqe_req.inline_oob_size = sizeof(struct mana_tx_short_oob);
184 	else
185 		pkg.wqe_req.inline_oob_size = sizeof(struct mana_tx_oob);
186 
187 	pkg.wqe_req.inline_oob_data = &pkg.tx_oob;
188 	pkg.wqe_req.flags = 0;
189 	pkg.wqe_req.client_data_unit = 0;
190 
191 	pkg.wqe_req.num_sge = 1 + skb_shinfo(skb)->nr_frags;
192 	WARN_ON_ONCE(pkg.wqe_req.num_sge > MAX_TX_WQE_SGL_ENTRIES);
193 
194 	if (pkg.wqe_req.num_sge <= ARRAY_SIZE(pkg.sgl_array)) {
195 		pkg.wqe_req.sgl = pkg.sgl_array;
196 	} else {
197 		pkg.sgl_ptr = kmalloc_array(pkg.wqe_req.num_sge,
198 					    sizeof(struct gdma_sge),
199 					    GFP_ATOMIC);
200 		if (!pkg.sgl_ptr)
201 			goto tx_drop_count;
202 
203 		pkg.wqe_req.sgl = pkg.sgl_ptr;
204 	}
205 
206 	if (skb->protocol == htons(ETH_P_IP))
207 		ipv4 = true;
208 	else if (skb->protocol == htons(ETH_P_IPV6))
209 		ipv6 = true;
210 
211 	if (skb_is_gso(skb)) {
212 		pkg.tx_oob.s_oob.is_outer_ipv4 = ipv4;
213 		pkg.tx_oob.s_oob.is_outer_ipv6 = ipv6;
214 
215 		pkg.tx_oob.s_oob.comp_iphdr_csum = 1;
216 		pkg.tx_oob.s_oob.comp_tcp_csum = 1;
217 		pkg.tx_oob.s_oob.trans_off = skb_transport_offset(skb);
218 
219 		pkg.wqe_req.client_data_unit = skb_shinfo(skb)->gso_size;
220 		pkg.wqe_req.flags = GDMA_WR_OOB_IN_SGL | GDMA_WR_PAD_BY_SGE0;
221 		if (ipv4) {
222 			ip_hdr(skb)->tot_len = 0;
223 			ip_hdr(skb)->check = 0;
224 			tcp_hdr(skb)->check =
225 				~csum_tcpudp_magic(ip_hdr(skb)->saddr,
226 						   ip_hdr(skb)->daddr, 0,
227 						   IPPROTO_TCP, 0);
228 		} else {
229 			ipv6_hdr(skb)->payload_len = 0;
230 			tcp_hdr(skb)->check =
231 				~csum_ipv6_magic(&ipv6_hdr(skb)->saddr,
232 						 &ipv6_hdr(skb)->daddr, 0,
233 						 IPPROTO_TCP, 0);
234 		}
235 	} else if (skb->ip_summed == CHECKSUM_PARTIAL) {
236 		csum_type = mana_checksum_info(skb);
237 
238 		if (csum_type == IPPROTO_TCP) {
239 			pkg.tx_oob.s_oob.is_outer_ipv4 = ipv4;
240 			pkg.tx_oob.s_oob.is_outer_ipv6 = ipv6;
241 
242 			pkg.tx_oob.s_oob.comp_tcp_csum = 1;
243 			pkg.tx_oob.s_oob.trans_off = skb_transport_offset(skb);
244 
245 		} else if (csum_type == IPPROTO_UDP) {
246 			pkg.tx_oob.s_oob.is_outer_ipv4 = ipv4;
247 			pkg.tx_oob.s_oob.is_outer_ipv6 = ipv6;
248 
249 			pkg.tx_oob.s_oob.comp_udp_csum = 1;
250 		} else {
251 			/* Can't do offload of this type of checksum */
252 			if (skb_checksum_help(skb))
253 				goto free_sgl_ptr;
254 		}
255 	}
256 
257 	if (mana_map_skb(skb, apc, &pkg))
258 		goto free_sgl_ptr;
259 
260 	skb_queue_tail(&txq->pending_skbs, skb);
261 
262 	len = skb->len;
263 	net_txq = netdev_get_tx_queue(ndev, txq_idx);
264 
265 	err = mana_gd_post_work_request(gdma_sq, &pkg.wqe_req,
266 					(struct gdma_posted_wqe_info *)skb->cb);
267 	if (!mana_can_tx(gdma_sq)) {
268 		netif_tx_stop_queue(net_txq);
269 		apc->eth_stats.stop_queue++;
270 	}
271 
272 	if (err) {
273 		(void)skb_dequeue_tail(&txq->pending_skbs);
274 		netdev_warn(ndev, "Failed to post TX OOB: %d\n", err);
275 		err = NETDEV_TX_BUSY;
276 		goto tx_busy;
277 	}
278 
279 	err = NETDEV_TX_OK;
280 	atomic_inc(&txq->pending_sends);
281 
282 	mana_gd_wq_ring_doorbell(gd->gdma_context, gdma_sq);
283 
284 	/* skb may be freed after mana_gd_post_work_request. Do not use it. */
285 	skb = NULL;
286 
287 	tx_stats = &txq->stats;
288 	u64_stats_update_begin(&tx_stats->syncp);
289 	tx_stats->packets++;
290 	tx_stats->bytes += len;
291 	u64_stats_update_end(&tx_stats->syncp);
292 
293 tx_busy:
294 	if (netif_tx_queue_stopped(net_txq) && mana_can_tx(gdma_sq)) {
295 		netif_tx_wake_queue(net_txq);
296 		apc->eth_stats.wake_queue++;
297 	}
298 
299 	kfree(pkg.sgl_ptr);
300 	return err;
301 
302 free_sgl_ptr:
303 	kfree(pkg.sgl_ptr);
304 tx_drop_count:
305 	ndev->stats.tx_dropped++;
306 tx_drop:
307 	dev_kfree_skb_any(skb);
308 	return NETDEV_TX_OK;
309 }
310 
311 static void mana_get_stats64(struct net_device *ndev,
312 			     struct rtnl_link_stats64 *st)
313 {
314 	struct mana_port_context *apc = netdev_priv(ndev);
315 	unsigned int num_queues = apc->num_queues;
316 	struct mana_stats_rx *rx_stats;
317 	struct mana_stats_tx *tx_stats;
318 	unsigned int start;
319 	u64 packets, bytes;
320 	int q;
321 
322 	if (!apc->port_is_up)
323 		return;
324 
325 	netdev_stats_to_stats64(st, &ndev->stats);
326 
327 	for (q = 0; q < num_queues; q++) {
328 		rx_stats = &apc->rxqs[q]->stats;
329 
330 		do {
331 			start = u64_stats_fetch_begin(&rx_stats->syncp);
332 			packets = rx_stats->packets;
333 			bytes = rx_stats->bytes;
334 		} while (u64_stats_fetch_retry(&rx_stats->syncp, start));
335 
336 		st->rx_packets += packets;
337 		st->rx_bytes += bytes;
338 	}
339 
340 	for (q = 0; q < num_queues; q++) {
341 		tx_stats = &apc->tx_qp[q].txq.stats;
342 
343 		do {
344 			start = u64_stats_fetch_begin(&tx_stats->syncp);
345 			packets = tx_stats->packets;
346 			bytes = tx_stats->bytes;
347 		} while (u64_stats_fetch_retry(&tx_stats->syncp, start));
348 
349 		st->tx_packets += packets;
350 		st->tx_bytes += bytes;
351 	}
352 }
353 
354 static int mana_get_tx_queue(struct net_device *ndev, struct sk_buff *skb,
355 			     int old_q)
356 {
357 	struct mana_port_context *apc = netdev_priv(ndev);
358 	u32 hash = skb_get_hash(skb);
359 	struct sock *sk = skb->sk;
360 	int txq;
361 
362 	txq = apc->indir_table[hash & MANA_INDIRECT_TABLE_MASK];
363 
364 	if (txq != old_q && sk && sk_fullsock(sk) &&
365 	    rcu_access_pointer(sk->sk_dst_cache))
366 		sk_tx_queue_set(sk, txq);
367 
368 	return txq;
369 }
370 
371 static u16 mana_select_queue(struct net_device *ndev, struct sk_buff *skb,
372 			     struct net_device *sb_dev)
373 {
374 	int txq;
375 
376 	if (ndev->real_num_tx_queues == 1)
377 		return 0;
378 
379 	txq = sk_tx_queue_get(skb->sk);
380 
381 	if (txq < 0 || skb->ooo_okay || txq >= ndev->real_num_tx_queues) {
382 		if (skb_rx_queue_recorded(skb))
383 			txq = skb_get_rx_queue(skb);
384 		else
385 			txq = mana_get_tx_queue(ndev, skb, txq);
386 	}
387 
388 	return txq;
389 }
390 
391 static const struct net_device_ops mana_devops = {
392 	.ndo_open		= mana_open,
393 	.ndo_stop		= mana_close,
394 	.ndo_select_queue	= mana_select_queue,
395 	.ndo_start_xmit		= mana_start_xmit,
396 	.ndo_validate_addr	= eth_validate_addr,
397 	.ndo_get_stats64	= mana_get_stats64,
398 	.ndo_bpf		= mana_bpf,
399 	.ndo_xdp_xmit		= mana_xdp_xmit,
400 };
401 
402 static void mana_cleanup_port_context(struct mana_port_context *apc)
403 {
404 	kfree(apc->rxqs);
405 	apc->rxqs = NULL;
406 }
407 
408 static int mana_init_port_context(struct mana_port_context *apc)
409 {
410 	apc->rxqs = kcalloc(apc->num_queues, sizeof(struct mana_rxq *),
411 			    GFP_KERNEL);
412 
413 	return !apc->rxqs ? -ENOMEM : 0;
414 }
415 
416 static int mana_send_request(struct mana_context *ac, void *in_buf,
417 			     u32 in_len, void *out_buf, u32 out_len)
418 {
419 	struct gdma_context *gc = ac->gdma_dev->gdma_context;
420 	struct gdma_resp_hdr *resp = out_buf;
421 	struct gdma_req_hdr *req = in_buf;
422 	struct device *dev = gc->dev;
423 	static atomic_t activity_id;
424 	int err;
425 
426 	req->dev_id = gc->mana.dev_id;
427 	req->activity_id = atomic_inc_return(&activity_id);
428 
429 	err = mana_gd_send_request(gc, in_len, in_buf, out_len,
430 				   out_buf);
431 	if (err || resp->status) {
432 		dev_err(dev, "Failed to send mana message: %d, 0x%x\n",
433 			err, resp->status);
434 		return err ? err : -EPROTO;
435 	}
436 
437 	if (req->dev_id.as_uint32 != resp->dev_id.as_uint32 ||
438 	    req->activity_id != resp->activity_id) {
439 		dev_err(dev, "Unexpected mana message response: %x,%x,%x,%x\n",
440 			req->dev_id.as_uint32, resp->dev_id.as_uint32,
441 			req->activity_id, resp->activity_id);
442 		return -EPROTO;
443 	}
444 
445 	return 0;
446 }
447 
448 static int mana_verify_resp_hdr(const struct gdma_resp_hdr *resp_hdr,
449 				const enum mana_command_code expected_code,
450 				const u32 min_size)
451 {
452 	if (resp_hdr->response.msg_type != expected_code)
453 		return -EPROTO;
454 
455 	if (resp_hdr->response.msg_version < GDMA_MESSAGE_V1)
456 		return -EPROTO;
457 
458 	if (resp_hdr->response.msg_size < min_size)
459 		return -EPROTO;
460 
461 	return 0;
462 }
463 
464 static int mana_pf_register_hw_vport(struct mana_port_context *apc)
465 {
466 	struct mana_register_hw_vport_resp resp = {};
467 	struct mana_register_hw_vport_req req = {};
468 	int err;
469 
470 	mana_gd_init_req_hdr(&req.hdr, MANA_REGISTER_HW_PORT,
471 			     sizeof(req), sizeof(resp));
472 	req.attached_gfid = 1;
473 	req.is_pf_default_vport = 1;
474 	req.allow_all_ether_types = 1;
475 
476 	err = mana_send_request(apc->ac, &req, sizeof(req), &resp,
477 				sizeof(resp));
478 	if (err) {
479 		netdev_err(apc->ndev, "Failed to register hw vPort: %d\n", err);
480 		return err;
481 	}
482 
483 	err = mana_verify_resp_hdr(&resp.hdr, MANA_REGISTER_HW_PORT,
484 				   sizeof(resp));
485 	if (err || resp.hdr.status) {
486 		netdev_err(apc->ndev, "Failed to register hw vPort: %d, 0x%x\n",
487 			   err, resp.hdr.status);
488 		return err ? err : -EPROTO;
489 	}
490 
491 	apc->port_handle = resp.hw_vport_handle;
492 	return 0;
493 }
494 
495 static void mana_pf_deregister_hw_vport(struct mana_port_context *apc)
496 {
497 	struct mana_deregister_hw_vport_resp resp = {};
498 	struct mana_deregister_hw_vport_req req = {};
499 	int err;
500 
501 	mana_gd_init_req_hdr(&req.hdr, MANA_DEREGISTER_HW_PORT,
502 			     sizeof(req), sizeof(resp));
503 	req.hw_vport_handle = apc->port_handle;
504 
505 	err = mana_send_request(apc->ac, &req, sizeof(req), &resp,
506 				sizeof(resp));
507 	if (err) {
508 		netdev_err(apc->ndev, "Failed to unregister hw vPort: %d\n",
509 			   err);
510 		return;
511 	}
512 
513 	err = mana_verify_resp_hdr(&resp.hdr, MANA_DEREGISTER_HW_PORT,
514 				   sizeof(resp));
515 	if (err || resp.hdr.status)
516 		netdev_err(apc->ndev,
517 			   "Failed to deregister hw vPort: %d, 0x%x\n",
518 			   err, resp.hdr.status);
519 }
520 
521 static int mana_pf_register_filter(struct mana_port_context *apc)
522 {
523 	struct mana_register_filter_resp resp = {};
524 	struct mana_register_filter_req req = {};
525 	int err;
526 
527 	mana_gd_init_req_hdr(&req.hdr, MANA_REGISTER_FILTER,
528 			     sizeof(req), sizeof(resp));
529 	req.vport = apc->port_handle;
530 	memcpy(req.mac_addr, apc->mac_addr, ETH_ALEN);
531 
532 	err = mana_send_request(apc->ac, &req, sizeof(req), &resp,
533 				sizeof(resp));
534 	if (err) {
535 		netdev_err(apc->ndev, "Failed to register filter: %d\n", err);
536 		return err;
537 	}
538 
539 	err = mana_verify_resp_hdr(&resp.hdr, MANA_REGISTER_FILTER,
540 				   sizeof(resp));
541 	if (err || resp.hdr.status) {
542 		netdev_err(apc->ndev, "Failed to register filter: %d, 0x%x\n",
543 			   err, resp.hdr.status);
544 		return err ? err : -EPROTO;
545 	}
546 
547 	apc->pf_filter_handle = resp.filter_handle;
548 	return 0;
549 }
550 
551 static void mana_pf_deregister_filter(struct mana_port_context *apc)
552 {
553 	struct mana_deregister_filter_resp resp = {};
554 	struct mana_deregister_filter_req req = {};
555 	int err;
556 
557 	mana_gd_init_req_hdr(&req.hdr, MANA_DEREGISTER_FILTER,
558 			     sizeof(req), sizeof(resp));
559 	req.filter_handle = apc->pf_filter_handle;
560 
561 	err = mana_send_request(apc->ac, &req, sizeof(req), &resp,
562 				sizeof(resp));
563 	if (err) {
564 		netdev_err(apc->ndev, "Failed to unregister filter: %d\n",
565 			   err);
566 		return;
567 	}
568 
569 	err = mana_verify_resp_hdr(&resp.hdr, MANA_DEREGISTER_FILTER,
570 				   sizeof(resp));
571 	if (err || resp.hdr.status)
572 		netdev_err(apc->ndev,
573 			   "Failed to deregister filter: %d, 0x%x\n",
574 			   err, resp.hdr.status);
575 }
576 
577 static int mana_query_device_cfg(struct mana_context *ac, u32 proto_major_ver,
578 				 u32 proto_minor_ver, u32 proto_micro_ver,
579 				 u16 *max_num_vports)
580 {
581 	struct gdma_context *gc = ac->gdma_dev->gdma_context;
582 	struct mana_query_device_cfg_resp resp = {};
583 	struct mana_query_device_cfg_req req = {};
584 	struct device *dev = gc->dev;
585 	int err = 0;
586 
587 	mana_gd_init_req_hdr(&req.hdr, MANA_QUERY_DEV_CONFIG,
588 			     sizeof(req), sizeof(resp));
589 	req.proto_major_ver = proto_major_ver;
590 	req.proto_minor_ver = proto_minor_ver;
591 	req.proto_micro_ver = proto_micro_ver;
592 
593 	err = mana_send_request(ac, &req, sizeof(req), &resp, sizeof(resp));
594 	if (err) {
595 		dev_err(dev, "Failed to query config: %d", err);
596 		return err;
597 	}
598 
599 	err = mana_verify_resp_hdr(&resp.hdr, MANA_QUERY_DEV_CONFIG,
600 				   sizeof(resp));
601 	if (err || resp.hdr.status) {
602 		dev_err(dev, "Invalid query result: %d, 0x%x\n", err,
603 			resp.hdr.status);
604 		if (!err)
605 			err = -EPROTO;
606 		return err;
607 	}
608 
609 	*max_num_vports = resp.max_num_vports;
610 
611 	return 0;
612 }
613 
614 static int mana_query_vport_cfg(struct mana_port_context *apc, u32 vport_index,
615 				u32 *max_sq, u32 *max_rq, u32 *num_indir_entry)
616 {
617 	struct mana_query_vport_cfg_resp resp = {};
618 	struct mana_query_vport_cfg_req req = {};
619 	int err;
620 
621 	mana_gd_init_req_hdr(&req.hdr, MANA_QUERY_VPORT_CONFIG,
622 			     sizeof(req), sizeof(resp));
623 
624 	req.vport_index = vport_index;
625 
626 	err = mana_send_request(apc->ac, &req, sizeof(req), &resp,
627 				sizeof(resp));
628 	if (err)
629 		return err;
630 
631 	err = mana_verify_resp_hdr(&resp.hdr, MANA_QUERY_VPORT_CONFIG,
632 				   sizeof(resp));
633 	if (err)
634 		return err;
635 
636 	if (resp.hdr.status)
637 		return -EPROTO;
638 
639 	*max_sq = resp.max_num_sq;
640 	*max_rq = resp.max_num_rq;
641 	*num_indir_entry = resp.num_indirection_ent;
642 
643 	apc->port_handle = resp.vport;
644 	ether_addr_copy(apc->mac_addr, resp.mac_addr);
645 
646 	return 0;
647 }
648 
649 void mana_uncfg_vport(struct mana_port_context *apc)
650 {
651 	mutex_lock(&apc->vport_mutex);
652 	apc->vport_use_count--;
653 	WARN_ON(apc->vport_use_count < 0);
654 	mutex_unlock(&apc->vport_mutex);
655 }
656 EXPORT_SYMBOL_NS(mana_uncfg_vport, NET_MANA);
657 
658 int mana_cfg_vport(struct mana_port_context *apc, u32 protection_dom_id,
659 		   u32 doorbell_pg_id)
660 {
661 	struct mana_config_vport_resp resp = {};
662 	struct mana_config_vport_req req = {};
663 	int err;
664 
665 	/* This function is used to program the Ethernet port in the hardware
666 	 * table. It can be called from the Ethernet driver or the RDMA driver.
667 	 *
668 	 * For Ethernet usage, the hardware supports only one active user on a
669 	 * physical port. The driver checks on the port usage before programming
670 	 * the hardware when creating the RAW QP (RDMA driver) or exposing the
671 	 * device to kernel NET layer (Ethernet driver).
672 	 *
673 	 * Because the RDMA driver doesn't know in advance which QP type the
674 	 * user will create, it exposes the device with all its ports. The user
675 	 * may not be able to create RAW QP on a port if this port is already
676 	 * in used by the Ethernet driver from the kernel.
677 	 *
678 	 * This physical port limitation only applies to the RAW QP. For RC QP,
679 	 * the hardware doesn't have this limitation. The user can create RC
680 	 * QPs on a physical port up to the hardware limits independent of the
681 	 * Ethernet usage on the same port.
682 	 */
683 	mutex_lock(&apc->vport_mutex);
684 	if (apc->vport_use_count > 0) {
685 		mutex_unlock(&apc->vport_mutex);
686 		return -EBUSY;
687 	}
688 	apc->vport_use_count++;
689 	mutex_unlock(&apc->vport_mutex);
690 
691 	mana_gd_init_req_hdr(&req.hdr, MANA_CONFIG_VPORT_TX,
692 			     sizeof(req), sizeof(resp));
693 	req.vport = apc->port_handle;
694 	req.pdid = protection_dom_id;
695 	req.doorbell_pageid = doorbell_pg_id;
696 
697 	err = mana_send_request(apc->ac, &req, sizeof(req), &resp,
698 				sizeof(resp));
699 	if (err) {
700 		netdev_err(apc->ndev, "Failed to configure vPort: %d\n", err);
701 		goto out;
702 	}
703 
704 	err = mana_verify_resp_hdr(&resp.hdr, MANA_CONFIG_VPORT_TX,
705 				   sizeof(resp));
706 	if (err || resp.hdr.status) {
707 		netdev_err(apc->ndev, "Failed to configure vPort: %d, 0x%x\n",
708 			   err, resp.hdr.status);
709 		if (!err)
710 			err = -EPROTO;
711 
712 		goto out;
713 	}
714 
715 	apc->tx_shortform_allowed = resp.short_form_allowed;
716 	apc->tx_vp_offset = resp.tx_vport_offset;
717 
718 	netdev_info(apc->ndev, "Configured vPort %llu PD %u DB %u\n",
719 		    apc->port_handle, protection_dom_id, doorbell_pg_id);
720 out:
721 	if (err)
722 		mana_uncfg_vport(apc);
723 
724 	return err;
725 }
726 EXPORT_SYMBOL_NS(mana_cfg_vport, NET_MANA);
727 
728 static int mana_cfg_vport_steering(struct mana_port_context *apc,
729 				   enum TRI_STATE rx,
730 				   bool update_default_rxobj, bool update_key,
731 				   bool update_tab)
732 {
733 	u16 num_entries = MANA_INDIRECT_TABLE_SIZE;
734 	struct mana_cfg_rx_steer_req *req = NULL;
735 	struct mana_cfg_rx_steer_resp resp = {};
736 	struct net_device *ndev = apc->ndev;
737 	mana_handle_t *req_indir_tab;
738 	u32 req_buf_size;
739 	int err;
740 
741 	req_buf_size = sizeof(*req) + sizeof(mana_handle_t) * num_entries;
742 	req = kzalloc(req_buf_size, GFP_KERNEL);
743 	if (!req)
744 		return -ENOMEM;
745 
746 	mana_gd_init_req_hdr(&req->hdr, MANA_CONFIG_VPORT_RX, req_buf_size,
747 			     sizeof(resp));
748 
749 	req->vport = apc->port_handle;
750 	req->num_indir_entries = num_entries;
751 	req->indir_tab_offset = sizeof(*req);
752 	req->rx_enable = rx;
753 	req->rss_enable = apc->rss_state;
754 	req->update_default_rxobj = update_default_rxobj;
755 	req->update_hashkey = update_key;
756 	req->update_indir_tab = update_tab;
757 	req->default_rxobj = apc->default_rxobj;
758 
759 	if (update_key)
760 		memcpy(&req->hashkey, apc->hashkey, MANA_HASH_KEY_SIZE);
761 
762 	if (update_tab) {
763 		req_indir_tab = (mana_handle_t *)(req + 1);
764 		memcpy(req_indir_tab, apc->rxobj_table,
765 		       req->num_indir_entries * sizeof(mana_handle_t));
766 	}
767 
768 	err = mana_send_request(apc->ac, req, req_buf_size, &resp,
769 				sizeof(resp));
770 	if (err) {
771 		netdev_err(ndev, "Failed to configure vPort RX: %d\n", err);
772 		goto out;
773 	}
774 
775 	err = mana_verify_resp_hdr(&resp.hdr, MANA_CONFIG_VPORT_RX,
776 				   sizeof(resp));
777 	if (err) {
778 		netdev_err(ndev, "vPort RX configuration failed: %d\n", err);
779 		goto out;
780 	}
781 
782 	if (resp.hdr.status) {
783 		netdev_err(ndev, "vPort RX configuration failed: 0x%x\n",
784 			   resp.hdr.status);
785 		err = -EPROTO;
786 	}
787 
788 	netdev_info(ndev, "Configured steering vPort %llu entries %u\n",
789 		    apc->port_handle, num_entries);
790 out:
791 	kfree(req);
792 	return err;
793 }
794 
795 int mana_create_wq_obj(struct mana_port_context *apc,
796 		       mana_handle_t vport,
797 		       u32 wq_type, struct mana_obj_spec *wq_spec,
798 		       struct mana_obj_spec *cq_spec,
799 		       mana_handle_t *wq_obj)
800 {
801 	struct mana_create_wqobj_resp resp = {};
802 	struct mana_create_wqobj_req req = {};
803 	struct net_device *ndev = apc->ndev;
804 	int err;
805 
806 	mana_gd_init_req_hdr(&req.hdr, MANA_CREATE_WQ_OBJ,
807 			     sizeof(req), sizeof(resp));
808 	req.vport = vport;
809 	req.wq_type = wq_type;
810 	req.wq_gdma_region = wq_spec->gdma_region;
811 	req.cq_gdma_region = cq_spec->gdma_region;
812 	req.wq_size = wq_spec->queue_size;
813 	req.cq_size = cq_spec->queue_size;
814 	req.cq_moderation_ctx_id = cq_spec->modr_ctx_id;
815 	req.cq_parent_qid = cq_spec->attached_eq;
816 
817 	err = mana_send_request(apc->ac, &req, sizeof(req), &resp,
818 				sizeof(resp));
819 	if (err) {
820 		netdev_err(ndev, "Failed to create WQ object: %d\n", err);
821 		goto out;
822 	}
823 
824 	err = mana_verify_resp_hdr(&resp.hdr, MANA_CREATE_WQ_OBJ,
825 				   sizeof(resp));
826 	if (err || resp.hdr.status) {
827 		netdev_err(ndev, "Failed to create WQ object: %d, 0x%x\n", err,
828 			   resp.hdr.status);
829 		if (!err)
830 			err = -EPROTO;
831 		goto out;
832 	}
833 
834 	if (resp.wq_obj == INVALID_MANA_HANDLE) {
835 		netdev_err(ndev, "Got an invalid WQ object handle\n");
836 		err = -EPROTO;
837 		goto out;
838 	}
839 
840 	*wq_obj = resp.wq_obj;
841 	wq_spec->queue_index = resp.wq_id;
842 	cq_spec->queue_index = resp.cq_id;
843 
844 	return 0;
845 out:
846 	return err;
847 }
848 EXPORT_SYMBOL_NS(mana_create_wq_obj, NET_MANA);
849 
850 void mana_destroy_wq_obj(struct mana_port_context *apc, u32 wq_type,
851 			 mana_handle_t wq_obj)
852 {
853 	struct mana_destroy_wqobj_resp resp = {};
854 	struct mana_destroy_wqobj_req req = {};
855 	struct net_device *ndev = apc->ndev;
856 	int err;
857 
858 	mana_gd_init_req_hdr(&req.hdr, MANA_DESTROY_WQ_OBJ,
859 			     sizeof(req), sizeof(resp));
860 	req.wq_type = wq_type;
861 	req.wq_obj_handle = wq_obj;
862 
863 	err = mana_send_request(apc->ac, &req, sizeof(req), &resp,
864 				sizeof(resp));
865 	if (err) {
866 		netdev_err(ndev, "Failed to destroy WQ object: %d\n", err);
867 		return;
868 	}
869 
870 	err = mana_verify_resp_hdr(&resp.hdr, MANA_DESTROY_WQ_OBJ,
871 				   sizeof(resp));
872 	if (err || resp.hdr.status)
873 		netdev_err(ndev, "Failed to destroy WQ object: %d, 0x%x\n", err,
874 			   resp.hdr.status);
875 }
876 EXPORT_SYMBOL_NS(mana_destroy_wq_obj, NET_MANA);
877 
878 static void mana_destroy_eq(struct mana_context *ac)
879 {
880 	struct gdma_context *gc = ac->gdma_dev->gdma_context;
881 	struct gdma_queue *eq;
882 	int i;
883 
884 	if (!ac->eqs)
885 		return;
886 
887 	for (i = 0; i < gc->max_num_queues; i++) {
888 		eq = ac->eqs[i].eq;
889 		if (!eq)
890 			continue;
891 
892 		mana_gd_destroy_queue(gc, eq);
893 	}
894 
895 	kfree(ac->eqs);
896 	ac->eqs = NULL;
897 }
898 
899 static int mana_create_eq(struct mana_context *ac)
900 {
901 	struct gdma_dev *gd = ac->gdma_dev;
902 	struct gdma_context *gc = gd->gdma_context;
903 	struct gdma_queue_spec spec = {};
904 	int err;
905 	int i;
906 
907 	ac->eqs = kcalloc(gc->max_num_queues, sizeof(struct mana_eq),
908 			  GFP_KERNEL);
909 	if (!ac->eqs)
910 		return -ENOMEM;
911 
912 	spec.type = GDMA_EQ;
913 	spec.monitor_avl_buf = false;
914 	spec.queue_size = EQ_SIZE;
915 	spec.eq.callback = NULL;
916 	spec.eq.context = ac->eqs;
917 	spec.eq.log2_throttle_limit = LOG2_EQ_THROTTLE;
918 
919 	for (i = 0; i < gc->max_num_queues; i++) {
920 		err = mana_gd_create_mana_eq(gd, &spec, &ac->eqs[i].eq);
921 		if (err)
922 			goto out;
923 	}
924 
925 	return 0;
926 out:
927 	mana_destroy_eq(ac);
928 	return err;
929 }
930 
931 static int mana_fence_rq(struct mana_port_context *apc, struct mana_rxq *rxq)
932 {
933 	struct mana_fence_rq_resp resp = {};
934 	struct mana_fence_rq_req req = {};
935 	int err;
936 
937 	init_completion(&rxq->fence_event);
938 
939 	mana_gd_init_req_hdr(&req.hdr, MANA_FENCE_RQ,
940 			     sizeof(req), sizeof(resp));
941 	req.wq_obj_handle =  rxq->rxobj;
942 
943 	err = mana_send_request(apc->ac, &req, sizeof(req), &resp,
944 				sizeof(resp));
945 	if (err) {
946 		netdev_err(apc->ndev, "Failed to fence RQ %u: %d\n",
947 			   rxq->rxq_idx, err);
948 		return err;
949 	}
950 
951 	err = mana_verify_resp_hdr(&resp.hdr, MANA_FENCE_RQ, sizeof(resp));
952 	if (err || resp.hdr.status) {
953 		netdev_err(apc->ndev, "Failed to fence RQ %u: %d, 0x%x\n",
954 			   rxq->rxq_idx, err, resp.hdr.status);
955 		if (!err)
956 			err = -EPROTO;
957 
958 		return err;
959 	}
960 
961 	if (wait_for_completion_timeout(&rxq->fence_event, 10 * HZ) == 0) {
962 		netdev_err(apc->ndev, "Failed to fence RQ %u: timed out\n",
963 			   rxq->rxq_idx);
964 		return -ETIMEDOUT;
965 	}
966 
967 	return 0;
968 }
969 
970 static void mana_fence_rqs(struct mana_port_context *apc)
971 {
972 	unsigned int rxq_idx;
973 	struct mana_rxq *rxq;
974 	int err;
975 
976 	for (rxq_idx = 0; rxq_idx < apc->num_queues; rxq_idx++) {
977 		rxq = apc->rxqs[rxq_idx];
978 		err = mana_fence_rq(apc, rxq);
979 
980 		/* In case of any error, use sleep instead. */
981 		if (err)
982 			msleep(100);
983 	}
984 }
985 
986 static int mana_move_wq_tail(struct gdma_queue *wq, u32 num_units)
987 {
988 	u32 used_space_old;
989 	u32 used_space_new;
990 
991 	used_space_old = wq->head - wq->tail;
992 	used_space_new = wq->head - (wq->tail + num_units);
993 
994 	if (WARN_ON_ONCE(used_space_new > used_space_old))
995 		return -ERANGE;
996 
997 	wq->tail += num_units;
998 	return 0;
999 }
1000 
1001 static void mana_unmap_skb(struct sk_buff *skb, struct mana_port_context *apc)
1002 {
1003 	struct mana_skb_head *ash = (struct mana_skb_head *)skb->head;
1004 	struct gdma_context *gc = apc->ac->gdma_dev->gdma_context;
1005 	struct device *dev = gc->dev;
1006 	int i;
1007 
1008 	dma_unmap_single(dev, ash->dma_handle[0], ash->size[0], DMA_TO_DEVICE);
1009 
1010 	for (i = 1; i < skb_shinfo(skb)->nr_frags + 1; i++)
1011 		dma_unmap_page(dev, ash->dma_handle[i], ash->size[i],
1012 			       DMA_TO_DEVICE);
1013 }
1014 
1015 static void mana_poll_tx_cq(struct mana_cq *cq)
1016 {
1017 	struct gdma_comp *completions = cq->gdma_comp_buf;
1018 	struct gdma_posted_wqe_info *wqe_info;
1019 	unsigned int pkt_transmitted = 0;
1020 	unsigned int wqe_unit_cnt = 0;
1021 	struct mana_txq *txq = cq->txq;
1022 	struct mana_port_context *apc;
1023 	struct netdev_queue *net_txq;
1024 	struct gdma_queue *gdma_wq;
1025 	unsigned int avail_space;
1026 	struct net_device *ndev;
1027 	struct sk_buff *skb;
1028 	bool txq_stopped;
1029 	int comp_read;
1030 	int i;
1031 
1032 	ndev = txq->ndev;
1033 	apc = netdev_priv(ndev);
1034 
1035 	comp_read = mana_gd_poll_cq(cq->gdma_cq, completions,
1036 				    CQE_POLLING_BUFFER);
1037 
1038 	if (comp_read < 1)
1039 		return;
1040 
1041 	for (i = 0; i < comp_read; i++) {
1042 		struct mana_tx_comp_oob *cqe_oob;
1043 
1044 		if (WARN_ON_ONCE(!completions[i].is_sq))
1045 			return;
1046 
1047 		cqe_oob = (struct mana_tx_comp_oob *)completions[i].cqe_data;
1048 		if (WARN_ON_ONCE(cqe_oob->cqe_hdr.client_type !=
1049 				 MANA_CQE_COMPLETION))
1050 			return;
1051 
1052 		switch (cqe_oob->cqe_hdr.cqe_type) {
1053 		case CQE_TX_OKAY:
1054 			break;
1055 
1056 		case CQE_TX_SA_DROP:
1057 		case CQE_TX_MTU_DROP:
1058 		case CQE_TX_INVALID_OOB:
1059 		case CQE_TX_INVALID_ETH_TYPE:
1060 		case CQE_TX_HDR_PROCESSING_ERROR:
1061 		case CQE_TX_VF_DISABLED:
1062 		case CQE_TX_VPORT_IDX_OUT_OF_RANGE:
1063 		case CQE_TX_VPORT_DISABLED:
1064 		case CQE_TX_VLAN_TAGGING_VIOLATION:
1065 			WARN_ONCE(1, "TX: CQE error %d: ignored.\n",
1066 				  cqe_oob->cqe_hdr.cqe_type);
1067 			break;
1068 
1069 		default:
1070 			/* If the CQE type is unexpected, log an error, assert,
1071 			 * and go through the error path.
1072 			 */
1073 			WARN_ONCE(1, "TX: Unexpected CQE type %d: HW BUG?\n",
1074 				  cqe_oob->cqe_hdr.cqe_type);
1075 			return;
1076 		}
1077 
1078 		if (WARN_ON_ONCE(txq->gdma_txq_id != completions[i].wq_num))
1079 			return;
1080 
1081 		skb = skb_dequeue(&txq->pending_skbs);
1082 		if (WARN_ON_ONCE(!skb))
1083 			return;
1084 
1085 		wqe_info = (struct gdma_posted_wqe_info *)skb->cb;
1086 		wqe_unit_cnt += wqe_info->wqe_size_in_bu;
1087 
1088 		mana_unmap_skb(skb, apc);
1089 
1090 		napi_consume_skb(skb, cq->budget);
1091 
1092 		pkt_transmitted++;
1093 	}
1094 
1095 	if (WARN_ON_ONCE(wqe_unit_cnt == 0))
1096 		return;
1097 
1098 	mana_move_wq_tail(txq->gdma_sq, wqe_unit_cnt);
1099 
1100 	gdma_wq = txq->gdma_sq;
1101 	avail_space = mana_gd_wq_avail_space(gdma_wq);
1102 
1103 	/* Ensure tail updated before checking q stop */
1104 	smp_mb();
1105 
1106 	net_txq = txq->net_txq;
1107 	txq_stopped = netif_tx_queue_stopped(net_txq);
1108 
1109 	/* Ensure checking txq_stopped before apc->port_is_up. */
1110 	smp_rmb();
1111 
1112 	if (txq_stopped && apc->port_is_up && avail_space >= MAX_TX_WQE_SIZE) {
1113 		netif_tx_wake_queue(net_txq);
1114 		apc->eth_stats.wake_queue++;
1115 	}
1116 
1117 	if (atomic_sub_return(pkt_transmitted, &txq->pending_sends) < 0)
1118 		WARN_ON_ONCE(1);
1119 
1120 	cq->work_done = pkt_transmitted;
1121 }
1122 
1123 static void mana_post_pkt_rxq(struct mana_rxq *rxq)
1124 {
1125 	struct mana_recv_buf_oob *recv_buf_oob;
1126 	u32 curr_index;
1127 	int err;
1128 
1129 	curr_index = rxq->buf_index++;
1130 	if (rxq->buf_index == rxq->num_rx_buf)
1131 		rxq->buf_index = 0;
1132 
1133 	recv_buf_oob = &rxq->rx_oobs[curr_index];
1134 
1135 	err = mana_gd_post_and_ring(rxq->gdma_rq, &recv_buf_oob->wqe_req,
1136 				    &recv_buf_oob->wqe_inf);
1137 	if (WARN_ON_ONCE(err))
1138 		return;
1139 
1140 	WARN_ON_ONCE(recv_buf_oob->wqe_inf.wqe_size_in_bu != 1);
1141 }
1142 
1143 static struct sk_buff *mana_build_skb(void *buf_va, uint pkt_len,
1144 				      struct xdp_buff *xdp)
1145 {
1146 	struct sk_buff *skb = build_skb(buf_va, PAGE_SIZE);
1147 
1148 	if (!skb)
1149 		return NULL;
1150 
1151 	if (xdp->data_hard_start) {
1152 		skb_reserve(skb, xdp->data - xdp->data_hard_start);
1153 		skb_put(skb, xdp->data_end - xdp->data);
1154 	} else {
1155 		skb_reserve(skb, XDP_PACKET_HEADROOM);
1156 		skb_put(skb, pkt_len);
1157 	}
1158 
1159 	return skb;
1160 }
1161 
1162 static void mana_rx_skb(void *buf_va, struct mana_rxcomp_oob *cqe,
1163 			struct mana_rxq *rxq)
1164 {
1165 	struct mana_stats_rx *rx_stats = &rxq->stats;
1166 	struct net_device *ndev = rxq->ndev;
1167 	uint pkt_len = cqe->ppi[0].pkt_len;
1168 	u16 rxq_idx = rxq->rxq_idx;
1169 	struct napi_struct *napi;
1170 	struct xdp_buff xdp = {};
1171 	struct sk_buff *skb;
1172 	u32 hash_value;
1173 	u32 act;
1174 
1175 	rxq->rx_cq.work_done++;
1176 	napi = &rxq->rx_cq.napi;
1177 
1178 	if (!buf_va) {
1179 		++ndev->stats.rx_dropped;
1180 		return;
1181 	}
1182 
1183 	act = mana_run_xdp(ndev, rxq, &xdp, buf_va, pkt_len);
1184 
1185 	if (act == XDP_REDIRECT && !rxq->xdp_rc)
1186 		return;
1187 
1188 	if (act != XDP_PASS && act != XDP_TX)
1189 		goto drop_xdp;
1190 
1191 	skb = mana_build_skb(buf_va, pkt_len, &xdp);
1192 
1193 	if (!skb)
1194 		goto drop;
1195 
1196 	skb->dev = napi->dev;
1197 
1198 	skb->protocol = eth_type_trans(skb, ndev);
1199 	skb_checksum_none_assert(skb);
1200 	skb_record_rx_queue(skb, rxq_idx);
1201 
1202 	if ((ndev->features & NETIF_F_RXCSUM) && cqe->rx_iphdr_csum_succeed) {
1203 		if (cqe->rx_tcp_csum_succeed || cqe->rx_udp_csum_succeed)
1204 			skb->ip_summed = CHECKSUM_UNNECESSARY;
1205 	}
1206 
1207 	if (cqe->rx_hashtype != 0 && (ndev->features & NETIF_F_RXHASH)) {
1208 		hash_value = cqe->ppi[0].pkt_hash;
1209 
1210 		if (cqe->rx_hashtype & MANA_HASH_L4)
1211 			skb_set_hash(skb, hash_value, PKT_HASH_TYPE_L4);
1212 		else
1213 			skb_set_hash(skb, hash_value, PKT_HASH_TYPE_L3);
1214 	}
1215 
1216 	u64_stats_update_begin(&rx_stats->syncp);
1217 	rx_stats->packets++;
1218 	rx_stats->bytes += pkt_len;
1219 
1220 	if (act == XDP_TX)
1221 		rx_stats->xdp_tx++;
1222 	u64_stats_update_end(&rx_stats->syncp);
1223 
1224 	if (act == XDP_TX) {
1225 		skb_set_queue_mapping(skb, rxq_idx);
1226 		mana_xdp_tx(skb, ndev);
1227 		return;
1228 	}
1229 
1230 	napi_gro_receive(napi, skb);
1231 
1232 	return;
1233 
1234 drop_xdp:
1235 	u64_stats_update_begin(&rx_stats->syncp);
1236 	rx_stats->xdp_drop++;
1237 	u64_stats_update_end(&rx_stats->syncp);
1238 
1239 drop:
1240 	WARN_ON_ONCE(rxq->xdp_save_page);
1241 	rxq->xdp_save_page = virt_to_page(buf_va);
1242 
1243 	++ndev->stats.rx_dropped;
1244 
1245 	return;
1246 }
1247 
1248 static void mana_process_rx_cqe(struct mana_rxq *rxq, struct mana_cq *cq,
1249 				struct gdma_comp *cqe)
1250 {
1251 	struct mana_rxcomp_oob *oob = (struct mana_rxcomp_oob *)cqe->cqe_data;
1252 	struct gdma_context *gc = rxq->gdma_rq->gdma_dev->gdma_context;
1253 	struct net_device *ndev = rxq->ndev;
1254 	struct mana_recv_buf_oob *rxbuf_oob;
1255 	struct device *dev = gc->dev;
1256 	void *new_buf, *old_buf;
1257 	struct page *new_page;
1258 	u32 curr, pktlen;
1259 	dma_addr_t da;
1260 
1261 	switch (oob->cqe_hdr.cqe_type) {
1262 	case CQE_RX_OKAY:
1263 		break;
1264 
1265 	case CQE_RX_TRUNCATED:
1266 		++ndev->stats.rx_dropped;
1267 		rxbuf_oob = &rxq->rx_oobs[rxq->buf_index];
1268 		netdev_warn_once(ndev, "Dropped a truncated packet\n");
1269 		goto drop;
1270 
1271 	case CQE_RX_COALESCED_4:
1272 		netdev_err(ndev, "RX coalescing is unsupported\n");
1273 		return;
1274 
1275 	case CQE_RX_OBJECT_FENCE:
1276 		complete(&rxq->fence_event);
1277 		return;
1278 
1279 	default:
1280 		netdev_err(ndev, "Unknown RX CQE type = %d\n",
1281 			   oob->cqe_hdr.cqe_type);
1282 		return;
1283 	}
1284 
1285 	pktlen = oob->ppi[0].pkt_len;
1286 
1287 	if (pktlen == 0) {
1288 		/* data packets should never have packetlength of zero */
1289 		netdev_err(ndev, "RX pkt len=0, rq=%u, cq=%u, rxobj=0x%llx\n",
1290 			   rxq->gdma_id, cq->gdma_id, rxq->rxobj);
1291 		return;
1292 	}
1293 
1294 	curr = rxq->buf_index;
1295 	rxbuf_oob = &rxq->rx_oobs[curr];
1296 	WARN_ON_ONCE(rxbuf_oob->wqe_inf.wqe_size_in_bu != 1);
1297 
1298 	/* Reuse XDP dropped page if available */
1299 	if (rxq->xdp_save_page) {
1300 		new_page = rxq->xdp_save_page;
1301 		rxq->xdp_save_page = NULL;
1302 	} else {
1303 		new_page = alloc_page(GFP_ATOMIC);
1304 	}
1305 
1306 	if (new_page) {
1307 		da = dma_map_page(dev, new_page, XDP_PACKET_HEADROOM, rxq->datasize,
1308 				  DMA_FROM_DEVICE);
1309 
1310 		if (dma_mapping_error(dev, da)) {
1311 			__free_page(new_page);
1312 			new_page = NULL;
1313 		}
1314 	}
1315 
1316 	new_buf = new_page ? page_to_virt(new_page) : NULL;
1317 
1318 	if (new_buf) {
1319 		dma_unmap_page(dev, rxbuf_oob->buf_dma_addr, rxq->datasize,
1320 			       DMA_FROM_DEVICE);
1321 
1322 		old_buf = rxbuf_oob->buf_va;
1323 
1324 		/* refresh the rxbuf_oob with the new page */
1325 		rxbuf_oob->buf_va = new_buf;
1326 		rxbuf_oob->buf_dma_addr = da;
1327 		rxbuf_oob->sgl[0].address = rxbuf_oob->buf_dma_addr;
1328 	} else {
1329 		old_buf = NULL; /* drop the packet if no memory */
1330 	}
1331 
1332 	mana_rx_skb(old_buf, oob, rxq);
1333 
1334 drop:
1335 	mana_move_wq_tail(rxq->gdma_rq, rxbuf_oob->wqe_inf.wqe_size_in_bu);
1336 
1337 	mana_post_pkt_rxq(rxq);
1338 }
1339 
1340 static void mana_poll_rx_cq(struct mana_cq *cq)
1341 {
1342 	struct gdma_comp *comp = cq->gdma_comp_buf;
1343 	struct mana_rxq *rxq = cq->rxq;
1344 	int comp_read, i;
1345 
1346 	comp_read = mana_gd_poll_cq(cq->gdma_cq, comp, CQE_POLLING_BUFFER);
1347 	WARN_ON_ONCE(comp_read > CQE_POLLING_BUFFER);
1348 
1349 	rxq->xdp_flush = false;
1350 
1351 	for (i = 0; i < comp_read; i++) {
1352 		if (WARN_ON_ONCE(comp[i].is_sq))
1353 			return;
1354 
1355 		/* verify recv cqe references the right rxq */
1356 		if (WARN_ON_ONCE(comp[i].wq_num != cq->rxq->gdma_id))
1357 			return;
1358 
1359 		mana_process_rx_cqe(rxq, cq, &comp[i]);
1360 	}
1361 
1362 	if (rxq->xdp_flush)
1363 		xdp_do_flush();
1364 }
1365 
1366 static void mana_cq_handler(void *context, struct gdma_queue *gdma_queue)
1367 {
1368 	struct mana_cq *cq = context;
1369 	u8 arm_bit;
1370 
1371 	WARN_ON_ONCE(cq->gdma_cq != gdma_queue);
1372 
1373 	if (cq->type == MANA_CQ_TYPE_RX)
1374 		mana_poll_rx_cq(cq);
1375 	else
1376 		mana_poll_tx_cq(cq);
1377 
1378 	if (cq->work_done < cq->budget &&
1379 	    napi_complete_done(&cq->napi, cq->work_done)) {
1380 		arm_bit = SET_ARM_BIT;
1381 	} else {
1382 		arm_bit = 0;
1383 	}
1384 
1385 	mana_gd_ring_cq(gdma_queue, arm_bit);
1386 }
1387 
1388 static int mana_poll(struct napi_struct *napi, int budget)
1389 {
1390 	struct mana_cq *cq = container_of(napi, struct mana_cq, napi);
1391 
1392 	cq->work_done = 0;
1393 	cq->budget = budget;
1394 
1395 	mana_cq_handler(cq, cq->gdma_cq);
1396 
1397 	return min(cq->work_done, budget);
1398 }
1399 
1400 static void mana_schedule_napi(void *context, struct gdma_queue *gdma_queue)
1401 {
1402 	struct mana_cq *cq = context;
1403 
1404 	napi_schedule_irqoff(&cq->napi);
1405 }
1406 
1407 static void mana_deinit_cq(struct mana_port_context *apc, struct mana_cq *cq)
1408 {
1409 	struct gdma_dev *gd = apc->ac->gdma_dev;
1410 
1411 	if (!cq->gdma_cq)
1412 		return;
1413 
1414 	mana_gd_destroy_queue(gd->gdma_context, cq->gdma_cq);
1415 }
1416 
1417 static void mana_deinit_txq(struct mana_port_context *apc, struct mana_txq *txq)
1418 {
1419 	struct gdma_dev *gd = apc->ac->gdma_dev;
1420 
1421 	if (!txq->gdma_sq)
1422 		return;
1423 
1424 	mana_gd_destroy_queue(gd->gdma_context, txq->gdma_sq);
1425 }
1426 
1427 static void mana_destroy_txq(struct mana_port_context *apc)
1428 {
1429 	struct napi_struct *napi;
1430 	int i;
1431 
1432 	if (!apc->tx_qp)
1433 		return;
1434 
1435 	for (i = 0; i < apc->num_queues; i++) {
1436 		napi = &apc->tx_qp[i].tx_cq.napi;
1437 		napi_synchronize(napi);
1438 		napi_disable(napi);
1439 		netif_napi_del(napi);
1440 
1441 		mana_destroy_wq_obj(apc, GDMA_SQ, apc->tx_qp[i].tx_object);
1442 
1443 		mana_deinit_cq(apc, &apc->tx_qp[i].tx_cq);
1444 
1445 		mana_deinit_txq(apc, &apc->tx_qp[i].txq);
1446 	}
1447 
1448 	kfree(apc->tx_qp);
1449 	apc->tx_qp = NULL;
1450 }
1451 
1452 static int mana_create_txq(struct mana_port_context *apc,
1453 			   struct net_device *net)
1454 {
1455 	struct mana_context *ac = apc->ac;
1456 	struct gdma_dev *gd = ac->gdma_dev;
1457 	struct mana_obj_spec wq_spec;
1458 	struct mana_obj_spec cq_spec;
1459 	struct gdma_queue_spec spec;
1460 	struct gdma_context *gc;
1461 	struct mana_txq *txq;
1462 	struct mana_cq *cq;
1463 	u32 txq_size;
1464 	u32 cq_size;
1465 	int err;
1466 	int i;
1467 
1468 	apc->tx_qp = kcalloc(apc->num_queues, sizeof(struct mana_tx_qp),
1469 			     GFP_KERNEL);
1470 	if (!apc->tx_qp)
1471 		return -ENOMEM;
1472 
1473 	/*  The minimum size of the WQE is 32 bytes, hence
1474 	 *  MAX_SEND_BUFFERS_PER_QUEUE represents the maximum number of WQEs
1475 	 *  the SQ can store. This value is then used to size other queues
1476 	 *  to prevent overflow.
1477 	 */
1478 	txq_size = MAX_SEND_BUFFERS_PER_QUEUE * 32;
1479 	BUILD_BUG_ON(!PAGE_ALIGNED(txq_size));
1480 
1481 	cq_size = MAX_SEND_BUFFERS_PER_QUEUE * COMP_ENTRY_SIZE;
1482 	cq_size = PAGE_ALIGN(cq_size);
1483 
1484 	gc = gd->gdma_context;
1485 
1486 	for (i = 0; i < apc->num_queues; i++) {
1487 		apc->tx_qp[i].tx_object = INVALID_MANA_HANDLE;
1488 
1489 		/* Create SQ */
1490 		txq = &apc->tx_qp[i].txq;
1491 
1492 		u64_stats_init(&txq->stats.syncp);
1493 		txq->ndev = net;
1494 		txq->net_txq = netdev_get_tx_queue(net, i);
1495 		txq->vp_offset = apc->tx_vp_offset;
1496 		skb_queue_head_init(&txq->pending_skbs);
1497 
1498 		memset(&spec, 0, sizeof(spec));
1499 		spec.type = GDMA_SQ;
1500 		spec.monitor_avl_buf = true;
1501 		spec.queue_size = txq_size;
1502 		err = mana_gd_create_mana_wq_cq(gd, &spec, &txq->gdma_sq);
1503 		if (err)
1504 			goto out;
1505 
1506 		/* Create SQ's CQ */
1507 		cq = &apc->tx_qp[i].tx_cq;
1508 		cq->type = MANA_CQ_TYPE_TX;
1509 
1510 		cq->txq = txq;
1511 
1512 		memset(&spec, 0, sizeof(spec));
1513 		spec.type = GDMA_CQ;
1514 		spec.monitor_avl_buf = false;
1515 		spec.queue_size = cq_size;
1516 		spec.cq.callback = mana_schedule_napi;
1517 		spec.cq.parent_eq = ac->eqs[i].eq;
1518 		spec.cq.context = cq;
1519 		err = mana_gd_create_mana_wq_cq(gd, &spec, &cq->gdma_cq);
1520 		if (err)
1521 			goto out;
1522 
1523 		memset(&wq_spec, 0, sizeof(wq_spec));
1524 		memset(&cq_spec, 0, sizeof(cq_spec));
1525 
1526 		wq_spec.gdma_region = txq->gdma_sq->mem_info.dma_region_handle;
1527 		wq_spec.queue_size = txq->gdma_sq->queue_size;
1528 
1529 		cq_spec.gdma_region = cq->gdma_cq->mem_info.dma_region_handle;
1530 		cq_spec.queue_size = cq->gdma_cq->queue_size;
1531 		cq_spec.modr_ctx_id = 0;
1532 		cq_spec.attached_eq = cq->gdma_cq->cq.parent->id;
1533 
1534 		err = mana_create_wq_obj(apc, apc->port_handle, GDMA_SQ,
1535 					 &wq_spec, &cq_spec,
1536 					 &apc->tx_qp[i].tx_object);
1537 
1538 		if (err)
1539 			goto out;
1540 
1541 		txq->gdma_sq->id = wq_spec.queue_index;
1542 		cq->gdma_cq->id = cq_spec.queue_index;
1543 
1544 		txq->gdma_sq->mem_info.dma_region_handle =
1545 			GDMA_INVALID_DMA_REGION;
1546 		cq->gdma_cq->mem_info.dma_region_handle =
1547 			GDMA_INVALID_DMA_REGION;
1548 
1549 		txq->gdma_txq_id = txq->gdma_sq->id;
1550 
1551 		cq->gdma_id = cq->gdma_cq->id;
1552 
1553 		if (WARN_ON(cq->gdma_id >= gc->max_num_cqs)) {
1554 			err = -EINVAL;
1555 			goto out;
1556 		}
1557 
1558 		gc->cq_table[cq->gdma_id] = cq->gdma_cq;
1559 
1560 		netif_napi_add_tx(net, &cq->napi, mana_poll);
1561 		napi_enable(&cq->napi);
1562 
1563 		mana_gd_ring_cq(cq->gdma_cq, SET_ARM_BIT);
1564 	}
1565 
1566 	return 0;
1567 out:
1568 	mana_destroy_txq(apc);
1569 	return err;
1570 }
1571 
1572 static void mana_destroy_rxq(struct mana_port_context *apc,
1573 			     struct mana_rxq *rxq, bool validate_state)
1574 
1575 {
1576 	struct gdma_context *gc = apc->ac->gdma_dev->gdma_context;
1577 	struct mana_recv_buf_oob *rx_oob;
1578 	struct device *dev = gc->dev;
1579 	struct napi_struct *napi;
1580 	int i;
1581 
1582 	if (!rxq)
1583 		return;
1584 
1585 	napi = &rxq->rx_cq.napi;
1586 
1587 	if (validate_state)
1588 		napi_synchronize(napi);
1589 
1590 	napi_disable(napi);
1591 
1592 	xdp_rxq_info_unreg(&rxq->xdp_rxq);
1593 
1594 	netif_napi_del(napi);
1595 
1596 	mana_destroy_wq_obj(apc, GDMA_RQ, rxq->rxobj);
1597 
1598 	mana_deinit_cq(apc, &rxq->rx_cq);
1599 
1600 	if (rxq->xdp_save_page)
1601 		__free_page(rxq->xdp_save_page);
1602 
1603 	for (i = 0; i < rxq->num_rx_buf; i++) {
1604 		rx_oob = &rxq->rx_oobs[i];
1605 
1606 		if (!rx_oob->buf_va)
1607 			continue;
1608 
1609 		dma_unmap_page(dev, rx_oob->buf_dma_addr, rxq->datasize,
1610 			       DMA_FROM_DEVICE);
1611 
1612 		free_page((unsigned long)rx_oob->buf_va);
1613 		rx_oob->buf_va = NULL;
1614 	}
1615 
1616 	if (rxq->gdma_rq)
1617 		mana_gd_destroy_queue(gc, rxq->gdma_rq);
1618 
1619 	kfree(rxq);
1620 }
1621 
1622 #define MANA_WQE_HEADER_SIZE 16
1623 #define MANA_WQE_SGE_SIZE 16
1624 
1625 static int mana_alloc_rx_wqe(struct mana_port_context *apc,
1626 			     struct mana_rxq *rxq, u32 *rxq_size, u32 *cq_size)
1627 {
1628 	struct gdma_context *gc = apc->ac->gdma_dev->gdma_context;
1629 	struct mana_recv_buf_oob *rx_oob;
1630 	struct device *dev = gc->dev;
1631 	struct page *page;
1632 	dma_addr_t da;
1633 	u32 buf_idx;
1634 
1635 	WARN_ON(rxq->datasize == 0 || rxq->datasize > PAGE_SIZE);
1636 
1637 	*rxq_size = 0;
1638 	*cq_size = 0;
1639 
1640 	for (buf_idx = 0; buf_idx < rxq->num_rx_buf; buf_idx++) {
1641 		rx_oob = &rxq->rx_oobs[buf_idx];
1642 		memset(rx_oob, 0, sizeof(*rx_oob));
1643 
1644 		page = alloc_page(GFP_KERNEL);
1645 		if (!page)
1646 			return -ENOMEM;
1647 
1648 		da = dma_map_page(dev, page, XDP_PACKET_HEADROOM, rxq->datasize,
1649 				  DMA_FROM_DEVICE);
1650 
1651 		if (dma_mapping_error(dev, da)) {
1652 			__free_page(page);
1653 			return -ENOMEM;
1654 		}
1655 
1656 		rx_oob->buf_va = page_to_virt(page);
1657 		rx_oob->buf_dma_addr = da;
1658 
1659 		rx_oob->num_sge = 1;
1660 		rx_oob->sgl[0].address = rx_oob->buf_dma_addr;
1661 		rx_oob->sgl[0].size = rxq->datasize;
1662 		rx_oob->sgl[0].mem_key = apc->ac->gdma_dev->gpa_mkey;
1663 
1664 		rx_oob->wqe_req.sgl = rx_oob->sgl;
1665 		rx_oob->wqe_req.num_sge = rx_oob->num_sge;
1666 		rx_oob->wqe_req.inline_oob_size = 0;
1667 		rx_oob->wqe_req.inline_oob_data = NULL;
1668 		rx_oob->wqe_req.flags = 0;
1669 		rx_oob->wqe_req.client_data_unit = 0;
1670 
1671 		*rxq_size += ALIGN(MANA_WQE_HEADER_SIZE +
1672 				   MANA_WQE_SGE_SIZE * rx_oob->num_sge, 32);
1673 		*cq_size += COMP_ENTRY_SIZE;
1674 	}
1675 
1676 	return 0;
1677 }
1678 
1679 static int mana_push_wqe(struct mana_rxq *rxq)
1680 {
1681 	struct mana_recv_buf_oob *rx_oob;
1682 	u32 buf_idx;
1683 	int err;
1684 
1685 	for (buf_idx = 0; buf_idx < rxq->num_rx_buf; buf_idx++) {
1686 		rx_oob = &rxq->rx_oobs[buf_idx];
1687 
1688 		err = mana_gd_post_and_ring(rxq->gdma_rq, &rx_oob->wqe_req,
1689 					    &rx_oob->wqe_inf);
1690 		if (err)
1691 			return -ENOSPC;
1692 	}
1693 
1694 	return 0;
1695 }
1696 
1697 static struct mana_rxq *mana_create_rxq(struct mana_port_context *apc,
1698 					u32 rxq_idx, struct mana_eq *eq,
1699 					struct net_device *ndev)
1700 {
1701 	struct gdma_dev *gd = apc->ac->gdma_dev;
1702 	struct mana_obj_spec wq_spec;
1703 	struct mana_obj_spec cq_spec;
1704 	struct gdma_queue_spec spec;
1705 	struct mana_cq *cq = NULL;
1706 	struct gdma_context *gc;
1707 	u32 cq_size, rq_size;
1708 	struct mana_rxq *rxq;
1709 	int err;
1710 
1711 	gc = gd->gdma_context;
1712 
1713 	rxq = kzalloc(struct_size(rxq, rx_oobs, RX_BUFFERS_PER_QUEUE),
1714 		      GFP_KERNEL);
1715 	if (!rxq)
1716 		return NULL;
1717 
1718 	rxq->ndev = ndev;
1719 	rxq->num_rx_buf = RX_BUFFERS_PER_QUEUE;
1720 	rxq->rxq_idx = rxq_idx;
1721 	rxq->datasize = ALIGN(MAX_FRAME_SIZE, 64);
1722 	rxq->rxobj = INVALID_MANA_HANDLE;
1723 
1724 	err = mana_alloc_rx_wqe(apc, rxq, &rq_size, &cq_size);
1725 	if (err)
1726 		goto out;
1727 
1728 	rq_size = PAGE_ALIGN(rq_size);
1729 	cq_size = PAGE_ALIGN(cq_size);
1730 
1731 	/* Create RQ */
1732 	memset(&spec, 0, sizeof(spec));
1733 	spec.type = GDMA_RQ;
1734 	spec.monitor_avl_buf = true;
1735 	spec.queue_size = rq_size;
1736 	err = mana_gd_create_mana_wq_cq(gd, &spec, &rxq->gdma_rq);
1737 	if (err)
1738 		goto out;
1739 
1740 	/* Create RQ's CQ */
1741 	cq = &rxq->rx_cq;
1742 	cq->type = MANA_CQ_TYPE_RX;
1743 	cq->rxq = rxq;
1744 
1745 	memset(&spec, 0, sizeof(spec));
1746 	spec.type = GDMA_CQ;
1747 	spec.monitor_avl_buf = false;
1748 	spec.queue_size = cq_size;
1749 	spec.cq.callback = mana_schedule_napi;
1750 	spec.cq.parent_eq = eq->eq;
1751 	spec.cq.context = cq;
1752 	err = mana_gd_create_mana_wq_cq(gd, &spec, &cq->gdma_cq);
1753 	if (err)
1754 		goto out;
1755 
1756 	memset(&wq_spec, 0, sizeof(wq_spec));
1757 	memset(&cq_spec, 0, sizeof(cq_spec));
1758 	wq_spec.gdma_region = rxq->gdma_rq->mem_info.dma_region_handle;
1759 	wq_spec.queue_size = rxq->gdma_rq->queue_size;
1760 
1761 	cq_spec.gdma_region = cq->gdma_cq->mem_info.dma_region_handle;
1762 	cq_spec.queue_size = cq->gdma_cq->queue_size;
1763 	cq_spec.modr_ctx_id = 0;
1764 	cq_spec.attached_eq = cq->gdma_cq->cq.parent->id;
1765 
1766 	err = mana_create_wq_obj(apc, apc->port_handle, GDMA_RQ,
1767 				 &wq_spec, &cq_spec, &rxq->rxobj);
1768 	if (err)
1769 		goto out;
1770 
1771 	rxq->gdma_rq->id = wq_spec.queue_index;
1772 	cq->gdma_cq->id = cq_spec.queue_index;
1773 
1774 	rxq->gdma_rq->mem_info.dma_region_handle = GDMA_INVALID_DMA_REGION;
1775 	cq->gdma_cq->mem_info.dma_region_handle = GDMA_INVALID_DMA_REGION;
1776 
1777 	rxq->gdma_id = rxq->gdma_rq->id;
1778 	cq->gdma_id = cq->gdma_cq->id;
1779 
1780 	err = mana_push_wqe(rxq);
1781 	if (err)
1782 		goto out;
1783 
1784 	if (WARN_ON(cq->gdma_id >= gc->max_num_cqs)) {
1785 		err = -EINVAL;
1786 		goto out;
1787 	}
1788 
1789 	gc->cq_table[cq->gdma_id] = cq->gdma_cq;
1790 
1791 	netif_napi_add_weight(ndev, &cq->napi, mana_poll, 1);
1792 
1793 	WARN_ON(xdp_rxq_info_reg(&rxq->xdp_rxq, ndev, rxq_idx,
1794 				 cq->napi.napi_id));
1795 	WARN_ON(xdp_rxq_info_reg_mem_model(&rxq->xdp_rxq,
1796 					   MEM_TYPE_PAGE_SHARED, NULL));
1797 
1798 	napi_enable(&cq->napi);
1799 
1800 	mana_gd_ring_cq(cq->gdma_cq, SET_ARM_BIT);
1801 out:
1802 	if (!err)
1803 		return rxq;
1804 
1805 	netdev_err(ndev, "Failed to create RXQ: err = %d\n", err);
1806 
1807 	mana_destroy_rxq(apc, rxq, false);
1808 
1809 	if (cq)
1810 		mana_deinit_cq(apc, cq);
1811 
1812 	return NULL;
1813 }
1814 
1815 static int mana_add_rx_queues(struct mana_port_context *apc,
1816 			      struct net_device *ndev)
1817 {
1818 	struct mana_context *ac = apc->ac;
1819 	struct mana_rxq *rxq;
1820 	int err = 0;
1821 	int i;
1822 
1823 	for (i = 0; i < apc->num_queues; i++) {
1824 		rxq = mana_create_rxq(apc, i, &ac->eqs[i], ndev);
1825 		if (!rxq) {
1826 			err = -ENOMEM;
1827 			goto out;
1828 		}
1829 
1830 		u64_stats_init(&rxq->stats.syncp);
1831 
1832 		apc->rxqs[i] = rxq;
1833 	}
1834 
1835 	apc->default_rxobj = apc->rxqs[0]->rxobj;
1836 out:
1837 	return err;
1838 }
1839 
1840 static void mana_destroy_vport(struct mana_port_context *apc)
1841 {
1842 	struct gdma_dev *gd = apc->ac->gdma_dev;
1843 	struct mana_rxq *rxq;
1844 	u32 rxq_idx;
1845 
1846 	for (rxq_idx = 0; rxq_idx < apc->num_queues; rxq_idx++) {
1847 		rxq = apc->rxqs[rxq_idx];
1848 		if (!rxq)
1849 			continue;
1850 
1851 		mana_destroy_rxq(apc, rxq, true);
1852 		apc->rxqs[rxq_idx] = NULL;
1853 	}
1854 
1855 	mana_destroy_txq(apc);
1856 	mana_uncfg_vport(apc);
1857 
1858 	if (gd->gdma_context->is_pf)
1859 		mana_pf_deregister_hw_vport(apc);
1860 }
1861 
1862 static int mana_create_vport(struct mana_port_context *apc,
1863 			     struct net_device *net)
1864 {
1865 	struct gdma_dev *gd = apc->ac->gdma_dev;
1866 	int err;
1867 
1868 	apc->default_rxobj = INVALID_MANA_HANDLE;
1869 
1870 	if (gd->gdma_context->is_pf) {
1871 		err = mana_pf_register_hw_vport(apc);
1872 		if (err)
1873 			return err;
1874 	}
1875 
1876 	err = mana_cfg_vport(apc, gd->pdid, gd->doorbell);
1877 	if (err)
1878 		return err;
1879 
1880 	return mana_create_txq(apc, net);
1881 }
1882 
1883 static void mana_rss_table_init(struct mana_port_context *apc)
1884 {
1885 	int i;
1886 
1887 	for (i = 0; i < MANA_INDIRECT_TABLE_SIZE; i++)
1888 		apc->indir_table[i] =
1889 			ethtool_rxfh_indir_default(i, apc->num_queues);
1890 }
1891 
1892 int mana_config_rss(struct mana_port_context *apc, enum TRI_STATE rx,
1893 		    bool update_hash, bool update_tab)
1894 {
1895 	u32 queue_idx;
1896 	int err;
1897 	int i;
1898 
1899 	if (update_tab) {
1900 		for (i = 0; i < MANA_INDIRECT_TABLE_SIZE; i++) {
1901 			queue_idx = apc->indir_table[i];
1902 			apc->rxobj_table[i] = apc->rxqs[queue_idx]->rxobj;
1903 		}
1904 	}
1905 
1906 	err = mana_cfg_vport_steering(apc, rx, true, update_hash, update_tab);
1907 	if (err)
1908 		return err;
1909 
1910 	mana_fence_rqs(apc);
1911 
1912 	return 0;
1913 }
1914 
1915 static int mana_init_port(struct net_device *ndev)
1916 {
1917 	struct mana_port_context *apc = netdev_priv(ndev);
1918 	u32 max_txq, max_rxq, max_queues;
1919 	int port_idx = apc->port_idx;
1920 	u32 num_indirect_entries;
1921 	int err;
1922 
1923 	err = mana_init_port_context(apc);
1924 	if (err)
1925 		return err;
1926 
1927 	err = mana_query_vport_cfg(apc, port_idx, &max_txq, &max_rxq,
1928 				   &num_indirect_entries);
1929 	if (err) {
1930 		netdev_err(ndev, "Failed to query info for vPort %d\n",
1931 			   port_idx);
1932 		goto reset_apc;
1933 	}
1934 
1935 	max_queues = min_t(u32, max_txq, max_rxq);
1936 	if (apc->max_queues > max_queues)
1937 		apc->max_queues = max_queues;
1938 
1939 	if (apc->num_queues > apc->max_queues)
1940 		apc->num_queues = apc->max_queues;
1941 
1942 	eth_hw_addr_set(ndev, apc->mac_addr);
1943 
1944 	return 0;
1945 
1946 reset_apc:
1947 	kfree(apc->rxqs);
1948 	apc->rxqs = NULL;
1949 	return err;
1950 }
1951 
1952 int mana_alloc_queues(struct net_device *ndev)
1953 {
1954 	struct mana_port_context *apc = netdev_priv(ndev);
1955 	struct gdma_dev *gd = apc->ac->gdma_dev;
1956 	int err;
1957 
1958 	err = mana_create_vport(apc, ndev);
1959 	if (err)
1960 		return err;
1961 
1962 	err = netif_set_real_num_tx_queues(ndev, apc->num_queues);
1963 	if (err)
1964 		goto destroy_vport;
1965 
1966 	err = mana_add_rx_queues(apc, ndev);
1967 	if (err)
1968 		goto destroy_vport;
1969 
1970 	apc->rss_state = apc->num_queues > 1 ? TRI_STATE_TRUE : TRI_STATE_FALSE;
1971 
1972 	err = netif_set_real_num_rx_queues(ndev, apc->num_queues);
1973 	if (err)
1974 		goto destroy_vport;
1975 
1976 	mana_rss_table_init(apc);
1977 
1978 	err = mana_config_rss(apc, TRI_STATE_TRUE, true, true);
1979 	if (err)
1980 		goto destroy_vport;
1981 
1982 	if (gd->gdma_context->is_pf) {
1983 		err = mana_pf_register_filter(apc);
1984 		if (err)
1985 			goto destroy_vport;
1986 	}
1987 
1988 	mana_chn_setxdp(apc, mana_xdp_get(apc));
1989 
1990 	return 0;
1991 
1992 destroy_vport:
1993 	mana_destroy_vport(apc);
1994 	return err;
1995 }
1996 
1997 int mana_attach(struct net_device *ndev)
1998 {
1999 	struct mana_port_context *apc = netdev_priv(ndev);
2000 	int err;
2001 
2002 	ASSERT_RTNL();
2003 
2004 	err = mana_init_port(ndev);
2005 	if (err)
2006 		return err;
2007 
2008 	if (apc->port_st_save) {
2009 		err = mana_alloc_queues(ndev);
2010 		if (err) {
2011 			mana_cleanup_port_context(apc);
2012 			return err;
2013 		}
2014 	}
2015 
2016 	apc->port_is_up = apc->port_st_save;
2017 
2018 	/* Ensure port state updated before txq state */
2019 	smp_wmb();
2020 
2021 	if (apc->port_is_up)
2022 		netif_carrier_on(ndev);
2023 
2024 	netif_device_attach(ndev);
2025 
2026 	return 0;
2027 }
2028 
2029 static int mana_dealloc_queues(struct net_device *ndev)
2030 {
2031 	struct mana_port_context *apc = netdev_priv(ndev);
2032 	struct gdma_dev *gd = apc->ac->gdma_dev;
2033 	struct mana_txq *txq;
2034 	int i, err;
2035 
2036 	if (apc->port_is_up)
2037 		return -EINVAL;
2038 
2039 	mana_chn_setxdp(apc, NULL);
2040 
2041 	if (gd->gdma_context->is_pf)
2042 		mana_pf_deregister_filter(apc);
2043 
2044 	/* No packet can be transmitted now since apc->port_is_up is false.
2045 	 * There is still a tiny chance that mana_poll_tx_cq() can re-enable
2046 	 * a txq because it may not timely see apc->port_is_up being cleared
2047 	 * to false, but it doesn't matter since mana_start_xmit() drops any
2048 	 * new packets due to apc->port_is_up being false.
2049 	 *
2050 	 * Drain all the in-flight TX packets
2051 	 */
2052 	for (i = 0; i < apc->num_queues; i++) {
2053 		txq = &apc->tx_qp[i].txq;
2054 
2055 		while (atomic_read(&txq->pending_sends) > 0)
2056 			usleep_range(1000, 2000);
2057 	}
2058 
2059 	/* We're 100% sure the queues can no longer be woken up, because
2060 	 * we're sure now mana_poll_tx_cq() can't be running.
2061 	 */
2062 
2063 	apc->rss_state = TRI_STATE_FALSE;
2064 	err = mana_config_rss(apc, TRI_STATE_FALSE, false, false);
2065 	if (err) {
2066 		netdev_err(ndev, "Failed to disable vPort: %d\n", err);
2067 		return err;
2068 	}
2069 
2070 	mana_destroy_vport(apc);
2071 
2072 	return 0;
2073 }
2074 
2075 int mana_detach(struct net_device *ndev, bool from_close)
2076 {
2077 	struct mana_port_context *apc = netdev_priv(ndev);
2078 	int err;
2079 
2080 	ASSERT_RTNL();
2081 
2082 	apc->port_st_save = apc->port_is_up;
2083 	apc->port_is_up = false;
2084 
2085 	/* Ensure port state updated before txq state */
2086 	smp_wmb();
2087 
2088 	netif_tx_disable(ndev);
2089 	netif_carrier_off(ndev);
2090 
2091 	if (apc->port_st_save) {
2092 		err = mana_dealloc_queues(ndev);
2093 		if (err)
2094 			return err;
2095 	}
2096 
2097 	if (!from_close) {
2098 		netif_device_detach(ndev);
2099 		mana_cleanup_port_context(apc);
2100 	}
2101 
2102 	return 0;
2103 }
2104 
2105 static int mana_probe_port(struct mana_context *ac, int port_idx,
2106 			   struct net_device **ndev_storage)
2107 {
2108 	struct gdma_context *gc = ac->gdma_dev->gdma_context;
2109 	struct mana_port_context *apc;
2110 	struct net_device *ndev;
2111 	int err;
2112 
2113 	ndev = alloc_etherdev_mq(sizeof(struct mana_port_context),
2114 				 gc->max_num_queues);
2115 	if (!ndev)
2116 		return -ENOMEM;
2117 
2118 	*ndev_storage = ndev;
2119 
2120 	apc = netdev_priv(ndev);
2121 	apc->ac = ac;
2122 	apc->ndev = ndev;
2123 	apc->max_queues = gc->max_num_queues;
2124 	apc->num_queues = gc->max_num_queues;
2125 	apc->port_handle = INVALID_MANA_HANDLE;
2126 	apc->pf_filter_handle = INVALID_MANA_HANDLE;
2127 	apc->port_idx = port_idx;
2128 
2129 	mutex_init(&apc->vport_mutex);
2130 	apc->vport_use_count = 0;
2131 
2132 	ndev->netdev_ops = &mana_devops;
2133 	ndev->ethtool_ops = &mana_ethtool_ops;
2134 	ndev->mtu = ETH_DATA_LEN;
2135 	ndev->max_mtu = ndev->mtu;
2136 	ndev->min_mtu = ndev->mtu;
2137 	ndev->needed_headroom = MANA_HEADROOM;
2138 	ndev->dev_port = port_idx;
2139 	SET_NETDEV_DEV(ndev, gc->dev);
2140 
2141 	netif_carrier_off(ndev);
2142 
2143 	netdev_rss_key_fill(apc->hashkey, MANA_HASH_KEY_SIZE);
2144 
2145 	err = mana_init_port(ndev);
2146 	if (err)
2147 		goto free_net;
2148 
2149 	netdev_lockdep_set_classes(ndev);
2150 
2151 	ndev->hw_features = NETIF_F_SG | NETIF_F_IP_CSUM | NETIF_F_IPV6_CSUM;
2152 	ndev->hw_features |= NETIF_F_RXCSUM;
2153 	ndev->hw_features |= NETIF_F_TSO | NETIF_F_TSO6;
2154 	ndev->hw_features |= NETIF_F_RXHASH;
2155 	ndev->features = ndev->hw_features;
2156 	ndev->vlan_features = 0;
2157 
2158 	err = register_netdev(ndev);
2159 	if (err) {
2160 		netdev_err(ndev, "Unable to register netdev.\n");
2161 		goto reset_apc;
2162 	}
2163 
2164 	return 0;
2165 
2166 reset_apc:
2167 	kfree(apc->rxqs);
2168 	apc->rxqs = NULL;
2169 free_net:
2170 	*ndev_storage = NULL;
2171 	netdev_err(ndev, "Failed to probe vPort %d: %d\n", port_idx, err);
2172 	free_netdev(ndev);
2173 	return err;
2174 }
2175 
2176 static void adev_release(struct device *dev)
2177 {
2178 	struct mana_adev *madev = container_of(dev, struct mana_adev, adev.dev);
2179 
2180 	kfree(madev);
2181 }
2182 
2183 static void remove_adev(struct gdma_dev *gd)
2184 {
2185 	struct auxiliary_device *adev = gd->adev;
2186 	int id = adev->id;
2187 
2188 	auxiliary_device_delete(adev);
2189 	auxiliary_device_uninit(adev);
2190 
2191 	mana_adev_idx_free(id);
2192 	gd->adev = NULL;
2193 }
2194 
2195 static int add_adev(struct gdma_dev *gd)
2196 {
2197 	struct auxiliary_device *adev;
2198 	struct mana_adev *madev;
2199 	int ret;
2200 
2201 	madev = kzalloc(sizeof(*madev), GFP_KERNEL);
2202 	if (!madev)
2203 		return -ENOMEM;
2204 
2205 	adev = &madev->adev;
2206 	ret = mana_adev_idx_alloc();
2207 	if (ret < 0)
2208 		goto idx_fail;
2209 	adev->id = ret;
2210 
2211 	adev->name = "rdma";
2212 	adev->dev.parent = gd->gdma_context->dev;
2213 	adev->dev.release = adev_release;
2214 	madev->mdev = gd;
2215 
2216 	ret = auxiliary_device_init(adev);
2217 	if (ret)
2218 		goto init_fail;
2219 
2220 	ret = auxiliary_device_add(adev);
2221 	if (ret)
2222 		goto add_fail;
2223 
2224 	gd->adev = adev;
2225 	return 0;
2226 
2227 add_fail:
2228 	auxiliary_device_uninit(adev);
2229 
2230 init_fail:
2231 	mana_adev_idx_free(adev->id);
2232 
2233 idx_fail:
2234 	kfree(madev);
2235 
2236 	return ret;
2237 }
2238 
2239 int mana_probe(struct gdma_dev *gd, bool resuming)
2240 {
2241 	struct gdma_context *gc = gd->gdma_context;
2242 	struct mana_context *ac = gd->driver_data;
2243 	struct device *dev = gc->dev;
2244 	u16 num_ports = 0;
2245 	int err;
2246 	int i;
2247 
2248 	dev_info(dev,
2249 		 "Microsoft Azure Network Adapter protocol version: %d.%d.%d\n",
2250 		 MANA_MAJOR_VERSION, MANA_MINOR_VERSION, MANA_MICRO_VERSION);
2251 
2252 	err = mana_gd_register_device(gd);
2253 	if (err)
2254 		return err;
2255 
2256 	if (!resuming) {
2257 		ac = kzalloc(sizeof(*ac), GFP_KERNEL);
2258 		if (!ac)
2259 			return -ENOMEM;
2260 
2261 		ac->gdma_dev = gd;
2262 		gd->driver_data = ac;
2263 	}
2264 
2265 	err = mana_create_eq(ac);
2266 	if (err)
2267 		goto out;
2268 
2269 	err = mana_query_device_cfg(ac, MANA_MAJOR_VERSION, MANA_MINOR_VERSION,
2270 				    MANA_MICRO_VERSION, &num_ports);
2271 	if (err)
2272 		goto out;
2273 
2274 	if (!resuming) {
2275 		ac->num_ports = num_ports;
2276 	} else {
2277 		if (ac->num_ports != num_ports) {
2278 			dev_err(dev, "The number of vPorts changed: %d->%d\n",
2279 				ac->num_ports, num_ports);
2280 			err = -EPROTO;
2281 			goto out;
2282 		}
2283 	}
2284 
2285 	if (ac->num_ports == 0)
2286 		dev_err(dev, "Failed to detect any vPort\n");
2287 
2288 	if (ac->num_ports > MAX_PORTS_IN_MANA_DEV)
2289 		ac->num_ports = MAX_PORTS_IN_MANA_DEV;
2290 
2291 	if (!resuming) {
2292 		for (i = 0; i < ac->num_ports; i++) {
2293 			err = mana_probe_port(ac, i, &ac->ports[i]);
2294 			if (err)
2295 				break;
2296 		}
2297 	} else {
2298 		for (i = 0; i < ac->num_ports; i++) {
2299 			rtnl_lock();
2300 			err = mana_attach(ac->ports[i]);
2301 			rtnl_unlock();
2302 			if (err)
2303 				break;
2304 		}
2305 	}
2306 
2307 	err = add_adev(gd);
2308 out:
2309 	if (err)
2310 		mana_remove(gd, false);
2311 
2312 	return err;
2313 }
2314 
2315 void mana_remove(struct gdma_dev *gd, bool suspending)
2316 {
2317 	struct gdma_context *gc = gd->gdma_context;
2318 	struct mana_context *ac = gd->driver_data;
2319 	struct device *dev = gc->dev;
2320 	struct net_device *ndev;
2321 	int err;
2322 	int i;
2323 
2324 	/* adev currently doesn't support suspending, always remove it */
2325 	if (gd->adev)
2326 		remove_adev(gd);
2327 
2328 	for (i = 0; i < ac->num_ports; i++) {
2329 		ndev = ac->ports[i];
2330 		if (!ndev) {
2331 			if (i == 0)
2332 				dev_err(dev, "No net device to remove\n");
2333 			goto out;
2334 		}
2335 
2336 		/* All cleanup actions should stay after rtnl_lock(), otherwise
2337 		 * other functions may access partially cleaned up data.
2338 		 */
2339 		rtnl_lock();
2340 
2341 		err = mana_detach(ndev, false);
2342 		if (err)
2343 			netdev_err(ndev, "Failed to detach vPort %d: %d\n",
2344 				   i, err);
2345 
2346 		if (suspending) {
2347 			/* No need to unregister the ndev. */
2348 			rtnl_unlock();
2349 			continue;
2350 		}
2351 
2352 		unregister_netdevice(ndev);
2353 
2354 		rtnl_unlock();
2355 
2356 		free_netdev(ndev);
2357 	}
2358 
2359 	mana_destroy_eq(ac);
2360 out:
2361 	mana_gd_deregister_device(gd);
2362 
2363 	if (suspending)
2364 		return;
2365 
2366 	gd->driver_data = NULL;
2367 	gd->gdma_context = NULL;
2368 	kfree(ac);
2369 }
2370