xref: /linux/drivers/infiniband/sw/rxe/rxe_resp.c (revision 908fc4c2)
1 // SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB
2 /*
3  * Copyright (c) 2016 Mellanox Technologies Ltd. All rights reserved.
4  * Copyright (c) 2015 System Fabric Works, Inc. All rights reserved.
5  */
6 
7 #include <linux/skbuff.h>
8 
9 #include "rxe.h"
10 #include "rxe_loc.h"
11 #include "rxe_queue.h"
12 
13 enum resp_states {
14 	RESPST_NONE,
15 	RESPST_GET_REQ,
16 	RESPST_CHK_PSN,
17 	RESPST_CHK_OP_SEQ,
18 	RESPST_CHK_OP_VALID,
19 	RESPST_CHK_RESOURCE,
20 	RESPST_CHK_LENGTH,
21 	RESPST_CHK_RKEY,
22 	RESPST_EXECUTE,
23 	RESPST_READ_REPLY,
24 	RESPST_COMPLETE,
25 	RESPST_ACKNOWLEDGE,
26 	RESPST_CLEANUP,
27 	RESPST_DUPLICATE_REQUEST,
28 	RESPST_ERR_MALFORMED_WQE,
29 	RESPST_ERR_UNSUPPORTED_OPCODE,
30 	RESPST_ERR_MISALIGNED_ATOMIC,
31 	RESPST_ERR_PSN_OUT_OF_SEQ,
32 	RESPST_ERR_MISSING_OPCODE_FIRST,
33 	RESPST_ERR_MISSING_OPCODE_LAST_C,
34 	RESPST_ERR_MISSING_OPCODE_LAST_D1E,
35 	RESPST_ERR_TOO_MANY_RDMA_ATM_REQ,
36 	RESPST_ERR_RNR,
37 	RESPST_ERR_RKEY_VIOLATION,
38 	RESPST_ERR_INVALIDATE_RKEY,
39 	RESPST_ERR_LENGTH,
40 	RESPST_ERR_CQ_OVERFLOW,
41 	RESPST_ERROR,
42 	RESPST_RESET,
43 	RESPST_DONE,
44 	RESPST_EXIT,
45 };
46 
47 static char *resp_state_name[] = {
48 	[RESPST_NONE]				= "NONE",
49 	[RESPST_GET_REQ]			= "GET_REQ",
50 	[RESPST_CHK_PSN]			= "CHK_PSN",
51 	[RESPST_CHK_OP_SEQ]			= "CHK_OP_SEQ",
52 	[RESPST_CHK_OP_VALID]			= "CHK_OP_VALID",
53 	[RESPST_CHK_RESOURCE]			= "CHK_RESOURCE",
54 	[RESPST_CHK_LENGTH]			= "CHK_LENGTH",
55 	[RESPST_CHK_RKEY]			= "CHK_RKEY",
56 	[RESPST_EXECUTE]			= "EXECUTE",
57 	[RESPST_READ_REPLY]			= "READ_REPLY",
58 	[RESPST_COMPLETE]			= "COMPLETE",
59 	[RESPST_ACKNOWLEDGE]			= "ACKNOWLEDGE",
60 	[RESPST_CLEANUP]			= "CLEANUP",
61 	[RESPST_DUPLICATE_REQUEST]		= "DUPLICATE_REQUEST",
62 	[RESPST_ERR_MALFORMED_WQE]		= "ERR_MALFORMED_WQE",
63 	[RESPST_ERR_UNSUPPORTED_OPCODE]		= "ERR_UNSUPPORTED_OPCODE",
64 	[RESPST_ERR_MISALIGNED_ATOMIC]		= "ERR_MISALIGNED_ATOMIC",
65 	[RESPST_ERR_PSN_OUT_OF_SEQ]		= "ERR_PSN_OUT_OF_SEQ",
66 	[RESPST_ERR_MISSING_OPCODE_FIRST]	= "ERR_MISSING_OPCODE_FIRST",
67 	[RESPST_ERR_MISSING_OPCODE_LAST_C]	= "ERR_MISSING_OPCODE_LAST_C",
68 	[RESPST_ERR_MISSING_OPCODE_LAST_D1E]	= "ERR_MISSING_OPCODE_LAST_D1E",
69 	[RESPST_ERR_TOO_MANY_RDMA_ATM_REQ]	= "ERR_TOO_MANY_RDMA_ATM_REQ",
70 	[RESPST_ERR_RNR]			= "ERR_RNR",
71 	[RESPST_ERR_RKEY_VIOLATION]		= "ERR_RKEY_VIOLATION",
72 	[RESPST_ERR_INVALIDATE_RKEY]		= "ERR_INVALIDATE_RKEY_VIOLATION",
73 	[RESPST_ERR_LENGTH]			= "ERR_LENGTH",
74 	[RESPST_ERR_CQ_OVERFLOW]		= "ERR_CQ_OVERFLOW",
75 	[RESPST_ERROR]				= "ERROR",
76 	[RESPST_RESET]				= "RESET",
77 	[RESPST_DONE]				= "DONE",
78 	[RESPST_EXIT]				= "EXIT",
79 };
80 
81 /* rxe_recv calls here to add a request packet to the input queue */
82 void rxe_resp_queue_pkt(struct rxe_qp *qp, struct sk_buff *skb)
83 {
84 	int must_sched;
85 	struct rxe_pkt_info *pkt = SKB_TO_PKT(skb);
86 
87 	skb_queue_tail(&qp->req_pkts, skb);
88 
89 	must_sched = (pkt->opcode == IB_OPCODE_RC_RDMA_READ_REQUEST) ||
90 			(skb_queue_len(&qp->req_pkts) > 1);
91 
92 	rxe_run_task(&qp->resp.task, must_sched);
93 }
94 
95 static inline enum resp_states get_req(struct rxe_qp *qp,
96 				       struct rxe_pkt_info **pkt_p)
97 {
98 	struct sk_buff *skb;
99 
100 	if (qp->resp.state == QP_STATE_ERROR) {
101 		while ((skb = skb_dequeue(&qp->req_pkts))) {
102 			rxe_put(qp);
103 			kfree_skb(skb);
104 			ib_device_put(qp->ibqp.device);
105 		}
106 
107 		/* go drain recv wr queue */
108 		return RESPST_CHK_RESOURCE;
109 	}
110 
111 	skb = skb_peek(&qp->req_pkts);
112 	if (!skb)
113 		return RESPST_EXIT;
114 
115 	*pkt_p = SKB_TO_PKT(skb);
116 
117 	return (qp->resp.res) ? RESPST_READ_REPLY : RESPST_CHK_PSN;
118 }
119 
120 static enum resp_states check_psn(struct rxe_qp *qp,
121 				  struct rxe_pkt_info *pkt)
122 {
123 	int diff = psn_compare(pkt->psn, qp->resp.psn);
124 	struct rxe_dev *rxe = to_rdev(qp->ibqp.device);
125 
126 	switch (qp_type(qp)) {
127 	case IB_QPT_RC:
128 		if (diff > 0) {
129 			if (qp->resp.sent_psn_nak)
130 				return RESPST_CLEANUP;
131 
132 			qp->resp.sent_psn_nak = 1;
133 			rxe_counter_inc(rxe, RXE_CNT_OUT_OF_SEQ_REQ);
134 			return RESPST_ERR_PSN_OUT_OF_SEQ;
135 
136 		} else if (diff < 0) {
137 			rxe_counter_inc(rxe, RXE_CNT_DUP_REQ);
138 			return RESPST_DUPLICATE_REQUEST;
139 		}
140 
141 		if (qp->resp.sent_psn_nak)
142 			qp->resp.sent_psn_nak = 0;
143 
144 		break;
145 
146 	case IB_QPT_UC:
147 		if (qp->resp.drop_msg || diff != 0) {
148 			if (pkt->mask & RXE_START_MASK) {
149 				qp->resp.drop_msg = 0;
150 				return RESPST_CHK_OP_SEQ;
151 			}
152 
153 			qp->resp.drop_msg = 1;
154 			return RESPST_CLEANUP;
155 		}
156 		break;
157 	default:
158 		break;
159 	}
160 
161 	return RESPST_CHK_OP_SEQ;
162 }
163 
164 static enum resp_states check_op_seq(struct rxe_qp *qp,
165 				     struct rxe_pkt_info *pkt)
166 {
167 	switch (qp_type(qp)) {
168 	case IB_QPT_RC:
169 		switch (qp->resp.opcode) {
170 		case IB_OPCODE_RC_SEND_FIRST:
171 		case IB_OPCODE_RC_SEND_MIDDLE:
172 			switch (pkt->opcode) {
173 			case IB_OPCODE_RC_SEND_MIDDLE:
174 			case IB_OPCODE_RC_SEND_LAST:
175 			case IB_OPCODE_RC_SEND_LAST_WITH_IMMEDIATE:
176 			case IB_OPCODE_RC_SEND_LAST_WITH_INVALIDATE:
177 				return RESPST_CHK_OP_VALID;
178 			default:
179 				return RESPST_ERR_MISSING_OPCODE_LAST_C;
180 			}
181 
182 		case IB_OPCODE_RC_RDMA_WRITE_FIRST:
183 		case IB_OPCODE_RC_RDMA_WRITE_MIDDLE:
184 			switch (pkt->opcode) {
185 			case IB_OPCODE_RC_RDMA_WRITE_MIDDLE:
186 			case IB_OPCODE_RC_RDMA_WRITE_LAST:
187 			case IB_OPCODE_RC_RDMA_WRITE_LAST_WITH_IMMEDIATE:
188 				return RESPST_CHK_OP_VALID;
189 			default:
190 				return RESPST_ERR_MISSING_OPCODE_LAST_C;
191 			}
192 
193 		default:
194 			switch (pkt->opcode) {
195 			case IB_OPCODE_RC_SEND_MIDDLE:
196 			case IB_OPCODE_RC_SEND_LAST:
197 			case IB_OPCODE_RC_SEND_LAST_WITH_IMMEDIATE:
198 			case IB_OPCODE_RC_SEND_LAST_WITH_INVALIDATE:
199 			case IB_OPCODE_RC_RDMA_WRITE_MIDDLE:
200 			case IB_OPCODE_RC_RDMA_WRITE_LAST:
201 			case IB_OPCODE_RC_RDMA_WRITE_LAST_WITH_IMMEDIATE:
202 				return RESPST_ERR_MISSING_OPCODE_FIRST;
203 			default:
204 				return RESPST_CHK_OP_VALID;
205 			}
206 		}
207 		break;
208 
209 	case IB_QPT_UC:
210 		switch (qp->resp.opcode) {
211 		case IB_OPCODE_UC_SEND_FIRST:
212 		case IB_OPCODE_UC_SEND_MIDDLE:
213 			switch (pkt->opcode) {
214 			case IB_OPCODE_UC_SEND_MIDDLE:
215 			case IB_OPCODE_UC_SEND_LAST:
216 			case IB_OPCODE_UC_SEND_LAST_WITH_IMMEDIATE:
217 				return RESPST_CHK_OP_VALID;
218 			default:
219 				return RESPST_ERR_MISSING_OPCODE_LAST_D1E;
220 			}
221 
222 		case IB_OPCODE_UC_RDMA_WRITE_FIRST:
223 		case IB_OPCODE_UC_RDMA_WRITE_MIDDLE:
224 			switch (pkt->opcode) {
225 			case IB_OPCODE_UC_RDMA_WRITE_MIDDLE:
226 			case IB_OPCODE_UC_RDMA_WRITE_LAST:
227 			case IB_OPCODE_UC_RDMA_WRITE_LAST_WITH_IMMEDIATE:
228 				return RESPST_CHK_OP_VALID;
229 			default:
230 				return RESPST_ERR_MISSING_OPCODE_LAST_D1E;
231 			}
232 
233 		default:
234 			switch (pkt->opcode) {
235 			case IB_OPCODE_UC_SEND_MIDDLE:
236 			case IB_OPCODE_UC_SEND_LAST:
237 			case IB_OPCODE_UC_SEND_LAST_WITH_IMMEDIATE:
238 			case IB_OPCODE_UC_RDMA_WRITE_MIDDLE:
239 			case IB_OPCODE_UC_RDMA_WRITE_LAST:
240 			case IB_OPCODE_UC_RDMA_WRITE_LAST_WITH_IMMEDIATE:
241 				qp->resp.drop_msg = 1;
242 				return RESPST_CLEANUP;
243 			default:
244 				return RESPST_CHK_OP_VALID;
245 			}
246 		}
247 		break;
248 
249 	default:
250 		return RESPST_CHK_OP_VALID;
251 	}
252 }
253 
254 static enum resp_states check_op_valid(struct rxe_qp *qp,
255 				       struct rxe_pkt_info *pkt)
256 {
257 	switch (qp_type(qp)) {
258 	case IB_QPT_RC:
259 		if (((pkt->mask & RXE_READ_MASK) &&
260 		     !(qp->attr.qp_access_flags & IB_ACCESS_REMOTE_READ)) ||
261 		    ((pkt->mask & RXE_WRITE_MASK) &&
262 		     !(qp->attr.qp_access_flags & IB_ACCESS_REMOTE_WRITE)) ||
263 		    ((pkt->mask & RXE_ATOMIC_MASK) &&
264 		     !(qp->attr.qp_access_flags & IB_ACCESS_REMOTE_ATOMIC))) {
265 			return RESPST_ERR_UNSUPPORTED_OPCODE;
266 		}
267 
268 		break;
269 
270 	case IB_QPT_UC:
271 		if ((pkt->mask & RXE_WRITE_MASK) &&
272 		    !(qp->attr.qp_access_flags & IB_ACCESS_REMOTE_WRITE)) {
273 			qp->resp.drop_msg = 1;
274 			return RESPST_CLEANUP;
275 		}
276 
277 		break;
278 
279 	case IB_QPT_UD:
280 	case IB_QPT_GSI:
281 		break;
282 
283 	default:
284 		WARN_ON_ONCE(1);
285 		break;
286 	}
287 
288 	return RESPST_CHK_RESOURCE;
289 }
290 
291 static enum resp_states get_srq_wqe(struct rxe_qp *qp)
292 {
293 	struct rxe_srq *srq = qp->srq;
294 	struct rxe_queue *q = srq->rq.queue;
295 	struct rxe_recv_wqe *wqe;
296 	struct ib_event ev;
297 	unsigned int count;
298 	size_t size;
299 	unsigned long flags;
300 
301 	if (srq->error)
302 		return RESPST_ERR_RNR;
303 
304 	spin_lock_irqsave(&srq->rq.consumer_lock, flags);
305 
306 	wqe = queue_head(q, QUEUE_TYPE_FROM_CLIENT);
307 	if (!wqe) {
308 		spin_unlock_irqrestore(&srq->rq.consumer_lock, flags);
309 		return RESPST_ERR_RNR;
310 	}
311 
312 	/* don't trust user space data */
313 	if (unlikely(wqe->dma.num_sge > srq->rq.max_sge)) {
314 		spin_unlock_irqrestore(&srq->rq.consumer_lock, flags);
315 		pr_warn("%s: invalid num_sge in SRQ entry\n", __func__);
316 		return RESPST_ERR_MALFORMED_WQE;
317 	}
318 	size = sizeof(*wqe) + wqe->dma.num_sge*sizeof(struct rxe_sge);
319 	memcpy(&qp->resp.srq_wqe, wqe, size);
320 
321 	qp->resp.wqe = &qp->resp.srq_wqe.wqe;
322 	queue_advance_consumer(q, QUEUE_TYPE_FROM_CLIENT);
323 	count = queue_count(q, QUEUE_TYPE_FROM_CLIENT);
324 
325 	if (srq->limit && srq->ibsrq.event_handler && (count < srq->limit)) {
326 		srq->limit = 0;
327 		goto event;
328 	}
329 
330 	spin_unlock_irqrestore(&srq->rq.consumer_lock, flags);
331 	return RESPST_CHK_LENGTH;
332 
333 event:
334 	spin_unlock_irqrestore(&srq->rq.consumer_lock, flags);
335 	ev.device = qp->ibqp.device;
336 	ev.element.srq = qp->ibqp.srq;
337 	ev.event = IB_EVENT_SRQ_LIMIT_REACHED;
338 	srq->ibsrq.event_handler(&ev, srq->ibsrq.srq_context);
339 	return RESPST_CHK_LENGTH;
340 }
341 
342 static enum resp_states check_resource(struct rxe_qp *qp,
343 				       struct rxe_pkt_info *pkt)
344 {
345 	struct rxe_srq *srq = qp->srq;
346 
347 	if (qp->resp.state == QP_STATE_ERROR) {
348 		if (qp->resp.wqe) {
349 			qp->resp.status = IB_WC_WR_FLUSH_ERR;
350 			return RESPST_COMPLETE;
351 		} else if (!srq) {
352 			qp->resp.wqe = queue_head(qp->rq.queue,
353 					QUEUE_TYPE_FROM_CLIENT);
354 			if (qp->resp.wqe) {
355 				qp->resp.status = IB_WC_WR_FLUSH_ERR;
356 				return RESPST_COMPLETE;
357 			} else {
358 				return RESPST_EXIT;
359 			}
360 		} else {
361 			return RESPST_EXIT;
362 		}
363 	}
364 
365 	if (pkt->mask & RXE_READ_OR_ATOMIC_MASK) {
366 		/* it is the requesters job to not send
367 		 * too many read/atomic ops, we just
368 		 * recycle the responder resource queue
369 		 */
370 		if (likely(qp->attr.max_dest_rd_atomic > 0))
371 			return RESPST_CHK_LENGTH;
372 		else
373 			return RESPST_ERR_TOO_MANY_RDMA_ATM_REQ;
374 	}
375 
376 	if (pkt->mask & RXE_RWR_MASK) {
377 		if (srq)
378 			return get_srq_wqe(qp);
379 
380 		qp->resp.wqe = queue_head(qp->rq.queue,
381 				QUEUE_TYPE_FROM_CLIENT);
382 		return (qp->resp.wqe) ? RESPST_CHK_LENGTH : RESPST_ERR_RNR;
383 	}
384 
385 	return RESPST_CHK_LENGTH;
386 }
387 
388 static enum resp_states check_length(struct rxe_qp *qp,
389 				     struct rxe_pkt_info *pkt)
390 {
391 	switch (qp_type(qp)) {
392 	case IB_QPT_RC:
393 		return RESPST_CHK_RKEY;
394 
395 	case IB_QPT_UC:
396 		return RESPST_CHK_RKEY;
397 
398 	default:
399 		return RESPST_CHK_RKEY;
400 	}
401 }
402 
403 static enum resp_states check_rkey(struct rxe_qp *qp,
404 				   struct rxe_pkt_info *pkt)
405 {
406 	struct rxe_mr *mr = NULL;
407 	struct rxe_mw *mw = NULL;
408 	u64 va;
409 	u32 rkey;
410 	u32 resid;
411 	u32 pktlen;
412 	int mtu = qp->mtu;
413 	enum resp_states state;
414 	int access;
415 
416 	if (pkt->mask & RXE_READ_OR_WRITE_MASK) {
417 		if (pkt->mask & RXE_RETH_MASK) {
418 			qp->resp.va = reth_va(pkt);
419 			qp->resp.offset = 0;
420 			qp->resp.rkey = reth_rkey(pkt);
421 			qp->resp.resid = reth_len(pkt);
422 			qp->resp.length = reth_len(pkt);
423 		}
424 		access = (pkt->mask & RXE_READ_MASK) ? IB_ACCESS_REMOTE_READ
425 						     : IB_ACCESS_REMOTE_WRITE;
426 	} else if (pkt->mask & RXE_ATOMIC_MASK) {
427 		qp->resp.va = atmeth_va(pkt);
428 		qp->resp.offset = 0;
429 		qp->resp.rkey = atmeth_rkey(pkt);
430 		qp->resp.resid = sizeof(u64);
431 		access = IB_ACCESS_REMOTE_ATOMIC;
432 	} else {
433 		return RESPST_EXECUTE;
434 	}
435 
436 	/* A zero-byte op is not required to set an addr or rkey. */
437 	if ((pkt->mask & RXE_READ_OR_WRITE_MASK) &&
438 	    (pkt->mask & RXE_RETH_MASK) &&
439 	    reth_len(pkt) == 0) {
440 		return RESPST_EXECUTE;
441 	}
442 
443 	va	= qp->resp.va;
444 	rkey	= qp->resp.rkey;
445 	resid	= qp->resp.resid;
446 	pktlen	= payload_size(pkt);
447 
448 	if (rkey_is_mw(rkey)) {
449 		mw = rxe_lookup_mw(qp, access, rkey);
450 		if (!mw) {
451 			pr_err("%s: no MW matches rkey %#x\n", __func__, rkey);
452 			state = RESPST_ERR_RKEY_VIOLATION;
453 			goto err;
454 		}
455 
456 		mr = mw->mr;
457 		if (!mr) {
458 			pr_err("%s: MW doesn't have an MR\n", __func__);
459 			state = RESPST_ERR_RKEY_VIOLATION;
460 			goto err;
461 		}
462 
463 		if (mw->access & IB_ZERO_BASED)
464 			qp->resp.offset = mw->addr;
465 
466 		rxe_put(mw);
467 		rxe_get(mr);
468 	} else {
469 		mr = lookup_mr(qp->pd, access, rkey, RXE_LOOKUP_REMOTE);
470 		if (!mr) {
471 			pr_err("%s: no MR matches rkey %#x\n", __func__, rkey);
472 			state = RESPST_ERR_RKEY_VIOLATION;
473 			goto err;
474 		}
475 	}
476 
477 	if (mr_check_range(mr, va + qp->resp.offset, resid)) {
478 		state = RESPST_ERR_RKEY_VIOLATION;
479 		goto err;
480 	}
481 
482 	if (pkt->mask & RXE_WRITE_MASK)	 {
483 		if (resid > mtu) {
484 			if (pktlen != mtu || bth_pad(pkt)) {
485 				state = RESPST_ERR_LENGTH;
486 				goto err;
487 			}
488 		} else {
489 			if (pktlen != resid) {
490 				state = RESPST_ERR_LENGTH;
491 				goto err;
492 			}
493 			if ((bth_pad(pkt) != (0x3 & (-resid)))) {
494 				/* This case may not be exactly that
495 				 * but nothing else fits.
496 				 */
497 				state = RESPST_ERR_LENGTH;
498 				goto err;
499 			}
500 		}
501 	}
502 
503 	WARN_ON_ONCE(qp->resp.mr);
504 
505 	qp->resp.mr = mr;
506 	return RESPST_EXECUTE;
507 
508 err:
509 	if (mr)
510 		rxe_put(mr);
511 	if (mw)
512 		rxe_put(mw);
513 
514 	return state;
515 }
516 
517 static enum resp_states send_data_in(struct rxe_qp *qp, void *data_addr,
518 				     int data_len)
519 {
520 	int err;
521 
522 	err = copy_data(qp->pd, IB_ACCESS_LOCAL_WRITE, &qp->resp.wqe->dma,
523 			data_addr, data_len, RXE_TO_MR_OBJ);
524 	if (unlikely(err))
525 		return (err == -ENOSPC) ? RESPST_ERR_LENGTH
526 					: RESPST_ERR_MALFORMED_WQE;
527 
528 	return RESPST_NONE;
529 }
530 
531 static enum resp_states write_data_in(struct rxe_qp *qp,
532 				      struct rxe_pkt_info *pkt)
533 {
534 	enum resp_states rc = RESPST_NONE;
535 	int	err;
536 	int data_len = payload_size(pkt);
537 
538 	err = rxe_mr_copy(qp->resp.mr, qp->resp.va + qp->resp.offset,
539 			  payload_addr(pkt), data_len, RXE_TO_MR_OBJ);
540 	if (err) {
541 		rc = RESPST_ERR_RKEY_VIOLATION;
542 		goto out;
543 	}
544 
545 	qp->resp.va += data_len;
546 	qp->resp.resid -= data_len;
547 
548 out:
549 	return rc;
550 }
551 
552 /* Guarantee atomicity of atomic operations at the machine level. */
553 static DEFINE_SPINLOCK(atomic_ops_lock);
554 
555 static enum resp_states process_atomic(struct rxe_qp *qp,
556 				       struct rxe_pkt_info *pkt)
557 {
558 	u64 *vaddr;
559 	enum resp_states ret;
560 	struct rxe_mr *mr = qp->resp.mr;
561 
562 	if (mr->state != RXE_MR_STATE_VALID) {
563 		ret = RESPST_ERR_RKEY_VIOLATION;
564 		goto out;
565 	}
566 
567 	vaddr = iova_to_vaddr(mr, qp->resp.va + qp->resp.offset, sizeof(u64));
568 
569 	/* check vaddr is 8 bytes aligned. */
570 	if (!vaddr || (uintptr_t)vaddr & 7) {
571 		ret = RESPST_ERR_MISALIGNED_ATOMIC;
572 		goto out;
573 	}
574 
575 	spin_lock_bh(&atomic_ops_lock);
576 
577 	qp->resp.atomic_orig = *vaddr;
578 
579 	if (pkt->opcode == IB_OPCODE_RC_COMPARE_SWAP) {
580 		if (*vaddr == atmeth_comp(pkt))
581 			*vaddr = atmeth_swap_add(pkt);
582 	} else {
583 		*vaddr += atmeth_swap_add(pkt);
584 	}
585 
586 	spin_unlock_bh(&atomic_ops_lock);
587 
588 	ret = RESPST_NONE;
589 out:
590 	return ret;
591 }
592 
593 static struct sk_buff *prepare_ack_packet(struct rxe_qp *qp,
594 					  struct rxe_pkt_info *pkt,
595 					  struct rxe_pkt_info *ack,
596 					  int opcode,
597 					  int payload,
598 					  u32 psn,
599 					  u8 syndrome)
600 {
601 	struct rxe_dev *rxe = to_rdev(qp->ibqp.device);
602 	struct sk_buff *skb;
603 	int paylen;
604 	int pad;
605 	int err;
606 
607 	/*
608 	 * allocate packet
609 	 */
610 	pad = (-payload) & 0x3;
611 	paylen = rxe_opcode[opcode].length + payload + pad + RXE_ICRC_SIZE;
612 
613 	skb = rxe_init_packet(rxe, &qp->pri_av, paylen, ack);
614 	if (!skb)
615 		return NULL;
616 
617 	ack->qp = qp;
618 	ack->opcode = opcode;
619 	ack->mask = rxe_opcode[opcode].mask;
620 	ack->paylen = paylen;
621 	ack->psn = psn;
622 
623 	bth_init(ack, opcode, 0, 0, pad, IB_DEFAULT_PKEY_FULL,
624 		 qp->attr.dest_qp_num, 0, psn);
625 
626 	if (ack->mask & RXE_AETH_MASK) {
627 		aeth_set_syn(ack, syndrome);
628 		aeth_set_msn(ack, qp->resp.msn);
629 	}
630 
631 	if (ack->mask & RXE_ATMACK_MASK)
632 		atmack_set_orig(ack, qp->resp.atomic_orig);
633 
634 	err = rxe_prepare(&qp->pri_av, ack, skb);
635 	if (err) {
636 		kfree_skb(skb);
637 		return NULL;
638 	}
639 
640 	return skb;
641 }
642 
643 static struct resp_res *rxe_prepare_read_res(struct rxe_qp *qp,
644 					struct rxe_pkt_info *pkt)
645 {
646 	struct resp_res *res;
647 	u32 pkts;
648 
649 	res = &qp->resp.resources[qp->resp.res_head];
650 	rxe_advance_resp_resource(qp);
651 	free_rd_atomic_resource(qp, res);
652 
653 	res->type = RXE_READ_MASK;
654 	res->replay = 0;
655 	res->read.va = qp->resp.va + qp->resp.offset;
656 	res->read.va_org = qp->resp.va + qp->resp.offset;
657 	res->read.resid = qp->resp.resid;
658 	res->read.length = qp->resp.resid;
659 	res->read.rkey = qp->resp.rkey;
660 
661 	pkts = max_t(u32, (reth_len(pkt) + qp->mtu - 1)/qp->mtu, 1);
662 	res->first_psn = pkt->psn;
663 	res->cur_psn = pkt->psn;
664 	res->last_psn = (pkt->psn + pkts - 1) & BTH_PSN_MASK;
665 
666 	res->state = rdatm_res_state_new;
667 
668 	return res;
669 }
670 
671 /**
672  * rxe_recheck_mr - revalidate MR from rkey and get a reference
673  * @qp: the qp
674  * @rkey: the rkey
675  *
676  * This code allows the MR to be invalidated or deregistered or
677  * the MW if one was used to be invalidated or deallocated.
678  * It is assumed that the access permissions if originally good
679  * are OK and the mappings to be unchanged.
680  *
681  * TODO: If someone reregisters an MR to change its size or
682  * access permissions during the processing of an RDMA read
683  * we should kill the responder resource and complete the
684  * operation with an error.
685  *
686  * Return: mr on success else NULL
687  */
688 static struct rxe_mr *rxe_recheck_mr(struct rxe_qp *qp, u32 rkey)
689 {
690 	struct rxe_dev *rxe = to_rdev(qp->ibqp.device);
691 	struct rxe_mr *mr;
692 	struct rxe_mw *mw;
693 
694 	if (rkey_is_mw(rkey)) {
695 		mw = rxe_pool_get_index(&rxe->mw_pool, rkey >> 8);
696 		if (!mw)
697 			return NULL;
698 
699 		mr = mw->mr;
700 		if (mw->rkey != rkey || mw->state != RXE_MW_STATE_VALID ||
701 		    !mr || mr->state != RXE_MR_STATE_VALID) {
702 			rxe_put(mw);
703 			return NULL;
704 		}
705 
706 		rxe_get(mr);
707 		rxe_put(mw);
708 
709 		return mr;
710 	}
711 
712 	mr = rxe_pool_get_index(&rxe->mr_pool, rkey >> 8);
713 	if (!mr)
714 		return NULL;
715 
716 	if (mr->rkey != rkey || mr->state != RXE_MR_STATE_VALID) {
717 		rxe_put(mr);
718 		return NULL;
719 	}
720 
721 	return mr;
722 }
723 
724 /* RDMA read response. If res is not NULL, then we have a current RDMA request
725  * being processed or replayed.
726  */
727 static enum resp_states read_reply(struct rxe_qp *qp,
728 				   struct rxe_pkt_info *req_pkt)
729 {
730 	struct rxe_pkt_info ack_pkt;
731 	struct sk_buff *skb;
732 	int mtu = qp->mtu;
733 	enum resp_states state;
734 	int payload;
735 	int opcode;
736 	int err;
737 	struct resp_res *res = qp->resp.res;
738 	struct rxe_mr *mr;
739 
740 	if (!res) {
741 		res = rxe_prepare_read_res(qp, req_pkt);
742 		qp->resp.res = res;
743 	}
744 
745 	if (res->state == rdatm_res_state_new) {
746 		if (!res->replay) {
747 			mr = qp->resp.mr;
748 			qp->resp.mr = NULL;
749 		} else {
750 			mr = rxe_recheck_mr(qp, res->read.rkey);
751 			if (!mr)
752 				return RESPST_ERR_RKEY_VIOLATION;
753 		}
754 
755 		if (res->read.resid <= mtu)
756 			opcode = IB_OPCODE_RC_RDMA_READ_RESPONSE_ONLY;
757 		else
758 			opcode = IB_OPCODE_RC_RDMA_READ_RESPONSE_FIRST;
759 	} else {
760 		mr = rxe_recheck_mr(qp, res->read.rkey);
761 		if (!mr)
762 			return RESPST_ERR_RKEY_VIOLATION;
763 
764 		if (res->read.resid > mtu)
765 			opcode = IB_OPCODE_RC_RDMA_READ_RESPONSE_MIDDLE;
766 		else
767 			opcode = IB_OPCODE_RC_RDMA_READ_RESPONSE_LAST;
768 	}
769 
770 	res->state = rdatm_res_state_next;
771 
772 	payload = min_t(int, res->read.resid, mtu);
773 
774 	skb = prepare_ack_packet(qp, req_pkt, &ack_pkt, opcode, payload,
775 				 res->cur_psn, AETH_ACK_UNLIMITED);
776 	if (!skb)
777 		return RESPST_ERR_RNR;
778 
779 	err = rxe_mr_copy(mr, res->read.va, payload_addr(&ack_pkt),
780 			  payload, RXE_FROM_MR_OBJ);
781 	if (err)
782 		pr_err("Failed copying memory\n");
783 	if (mr)
784 		rxe_put(mr);
785 
786 	if (bth_pad(&ack_pkt)) {
787 		u8 *pad = payload_addr(&ack_pkt) + payload;
788 
789 		memset(pad, 0, bth_pad(&ack_pkt));
790 	}
791 
792 	err = rxe_xmit_packet(qp, &ack_pkt, skb);
793 	if (err) {
794 		pr_err("Failed sending RDMA reply.\n");
795 		return RESPST_ERR_RNR;
796 	}
797 
798 	res->read.va += payload;
799 	res->read.resid -= payload;
800 	res->cur_psn = (res->cur_psn + 1) & BTH_PSN_MASK;
801 
802 	if (res->read.resid > 0) {
803 		state = RESPST_DONE;
804 	} else {
805 		qp->resp.res = NULL;
806 		if (!res->replay)
807 			qp->resp.opcode = -1;
808 		if (psn_compare(res->cur_psn, qp->resp.psn) >= 0)
809 			qp->resp.psn = res->cur_psn;
810 		state = RESPST_CLEANUP;
811 	}
812 
813 	return state;
814 }
815 
816 static int invalidate_rkey(struct rxe_qp *qp, u32 rkey)
817 {
818 	if (rkey_is_mw(rkey))
819 		return rxe_invalidate_mw(qp, rkey);
820 	else
821 		return rxe_invalidate_mr(qp, rkey);
822 }
823 
824 /* Executes a new request. A retried request never reach that function (send
825  * and writes are discarded, and reads and atomics are retried elsewhere.
826  */
827 static enum resp_states execute(struct rxe_qp *qp, struct rxe_pkt_info *pkt)
828 {
829 	enum resp_states err;
830 	struct sk_buff *skb = PKT_TO_SKB(pkt);
831 	union rdma_network_hdr hdr;
832 
833 	if (pkt->mask & RXE_SEND_MASK) {
834 		if (qp_type(qp) == IB_QPT_UD ||
835 		    qp_type(qp) == IB_QPT_GSI) {
836 			if (skb->protocol == htons(ETH_P_IP)) {
837 				memset(&hdr.reserved, 0,
838 						sizeof(hdr.reserved));
839 				memcpy(&hdr.roce4grh, ip_hdr(skb),
840 						sizeof(hdr.roce4grh));
841 				err = send_data_in(qp, &hdr, sizeof(hdr));
842 			} else {
843 				err = send_data_in(qp, ipv6_hdr(skb),
844 						sizeof(hdr));
845 			}
846 			if (err)
847 				return err;
848 		}
849 		err = send_data_in(qp, payload_addr(pkt), payload_size(pkt));
850 		if (err)
851 			return err;
852 	} else if (pkt->mask & RXE_WRITE_MASK) {
853 		err = write_data_in(qp, pkt);
854 		if (err)
855 			return err;
856 	} else if (pkt->mask & RXE_READ_MASK) {
857 		/* For RDMA Read we can increment the msn now. See C9-148. */
858 		qp->resp.msn++;
859 		return RESPST_READ_REPLY;
860 	} else if (pkt->mask & RXE_ATOMIC_MASK) {
861 		err = process_atomic(qp, pkt);
862 		if (err)
863 			return err;
864 	} else {
865 		/* Unreachable */
866 		WARN_ON_ONCE(1);
867 	}
868 
869 	if (pkt->mask & RXE_IETH_MASK) {
870 		u32 rkey = ieth_rkey(pkt);
871 
872 		err = invalidate_rkey(qp, rkey);
873 		if (err)
874 			return RESPST_ERR_INVALIDATE_RKEY;
875 	}
876 
877 	if (pkt->mask & RXE_END_MASK)
878 		/* We successfully processed this new request. */
879 		qp->resp.msn++;
880 
881 	/* next expected psn, read handles this separately */
882 	qp->resp.psn = (pkt->psn + 1) & BTH_PSN_MASK;
883 	qp->resp.ack_psn = qp->resp.psn;
884 
885 	qp->resp.opcode = pkt->opcode;
886 	qp->resp.status = IB_WC_SUCCESS;
887 
888 	if (pkt->mask & RXE_COMP_MASK)
889 		return RESPST_COMPLETE;
890 	else if (qp_type(qp) == IB_QPT_RC)
891 		return RESPST_ACKNOWLEDGE;
892 	else
893 		return RESPST_CLEANUP;
894 }
895 
896 static enum resp_states do_complete(struct rxe_qp *qp,
897 				    struct rxe_pkt_info *pkt)
898 {
899 	struct rxe_cqe cqe;
900 	struct ib_wc *wc = &cqe.ibwc;
901 	struct ib_uverbs_wc *uwc = &cqe.uibwc;
902 	struct rxe_recv_wqe *wqe = qp->resp.wqe;
903 	struct rxe_dev *rxe = to_rdev(qp->ibqp.device);
904 
905 	if (!wqe)
906 		goto finish;
907 
908 	memset(&cqe, 0, sizeof(cqe));
909 
910 	if (qp->rcq->is_user) {
911 		uwc->status		= qp->resp.status;
912 		uwc->qp_num		= qp->ibqp.qp_num;
913 		uwc->wr_id		= wqe->wr_id;
914 	} else {
915 		wc->status		= qp->resp.status;
916 		wc->qp			= &qp->ibqp;
917 		wc->wr_id		= wqe->wr_id;
918 	}
919 
920 	if (wc->status == IB_WC_SUCCESS) {
921 		rxe_counter_inc(rxe, RXE_CNT_RDMA_RECV);
922 		wc->opcode = (pkt->mask & RXE_IMMDT_MASK &&
923 				pkt->mask & RXE_WRITE_MASK) ?
924 					IB_WC_RECV_RDMA_WITH_IMM : IB_WC_RECV;
925 		wc->byte_len = (pkt->mask & RXE_IMMDT_MASK &&
926 				pkt->mask & RXE_WRITE_MASK) ?
927 					qp->resp.length : wqe->dma.length - wqe->dma.resid;
928 
929 		/* fields after byte_len are different between kernel and user
930 		 * space
931 		 */
932 		if (qp->rcq->is_user) {
933 			uwc->wc_flags = IB_WC_GRH;
934 
935 			if (pkt->mask & RXE_IMMDT_MASK) {
936 				uwc->wc_flags |= IB_WC_WITH_IMM;
937 				uwc->ex.imm_data = immdt_imm(pkt);
938 			}
939 
940 			if (pkt->mask & RXE_IETH_MASK) {
941 				uwc->wc_flags |= IB_WC_WITH_INVALIDATE;
942 				uwc->ex.invalidate_rkey = ieth_rkey(pkt);
943 			}
944 
945 			if (pkt->mask & RXE_DETH_MASK)
946 				uwc->src_qp = deth_sqp(pkt);
947 
948 			uwc->port_num		= qp->attr.port_num;
949 		} else {
950 			struct sk_buff *skb = PKT_TO_SKB(pkt);
951 
952 			wc->wc_flags = IB_WC_GRH | IB_WC_WITH_NETWORK_HDR_TYPE;
953 			if (skb->protocol == htons(ETH_P_IP))
954 				wc->network_hdr_type = RDMA_NETWORK_IPV4;
955 			else
956 				wc->network_hdr_type = RDMA_NETWORK_IPV6;
957 
958 			if (is_vlan_dev(skb->dev)) {
959 				wc->wc_flags |= IB_WC_WITH_VLAN;
960 				wc->vlan_id = vlan_dev_vlan_id(skb->dev);
961 			}
962 
963 			if (pkt->mask & RXE_IMMDT_MASK) {
964 				wc->wc_flags |= IB_WC_WITH_IMM;
965 				wc->ex.imm_data = immdt_imm(pkt);
966 			}
967 
968 			if (pkt->mask & RXE_IETH_MASK) {
969 				wc->wc_flags |= IB_WC_WITH_INVALIDATE;
970 				wc->ex.invalidate_rkey = ieth_rkey(pkt);
971 			}
972 
973 			if (pkt->mask & RXE_DETH_MASK)
974 				wc->src_qp = deth_sqp(pkt);
975 
976 			wc->port_num		= qp->attr.port_num;
977 		}
978 	}
979 
980 	/* have copy for srq and reference for !srq */
981 	if (!qp->srq)
982 		queue_advance_consumer(qp->rq.queue, QUEUE_TYPE_FROM_CLIENT);
983 
984 	qp->resp.wqe = NULL;
985 
986 	if (rxe_cq_post(qp->rcq, &cqe, pkt ? bth_se(pkt) : 1))
987 		return RESPST_ERR_CQ_OVERFLOW;
988 
989 finish:
990 	if (unlikely(qp->resp.state == QP_STATE_ERROR))
991 		return RESPST_CHK_RESOURCE;
992 	if (unlikely(!pkt))
993 		return RESPST_DONE;
994 	if (qp_type(qp) == IB_QPT_RC)
995 		return RESPST_ACKNOWLEDGE;
996 	else
997 		return RESPST_CLEANUP;
998 }
999 
1000 static int send_ack(struct rxe_qp *qp, struct rxe_pkt_info *pkt,
1001 		    u8 syndrome, u32 psn)
1002 {
1003 	int err = 0;
1004 	struct rxe_pkt_info ack_pkt;
1005 	struct sk_buff *skb;
1006 
1007 	skb = prepare_ack_packet(qp, pkt, &ack_pkt, IB_OPCODE_RC_ACKNOWLEDGE,
1008 				 0, psn, syndrome);
1009 	if (!skb) {
1010 		err = -ENOMEM;
1011 		goto err1;
1012 	}
1013 
1014 	err = rxe_xmit_packet(qp, &ack_pkt, skb);
1015 	if (err)
1016 		pr_err_ratelimited("Failed sending ack\n");
1017 
1018 err1:
1019 	return err;
1020 }
1021 
1022 static int send_atomic_ack(struct rxe_qp *qp, struct rxe_pkt_info *pkt,
1023 			   u8 syndrome)
1024 {
1025 	int rc = 0;
1026 	struct rxe_pkt_info ack_pkt;
1027 	struct sk_buff *skb;
1028 	struct resp_res *res;
1029 
1030 	skb = prepare_ack_packet(qp, pkt, &ack_pkt,
1031 				 IB_OPCODE_RC_ATOMIC_ACKNOWLEDGE, 0, pkt->psn,
1032 				 syndrome);
1033 	if (!skb) {
1034 		rc = -ENOMEM;
1035 		goto out;
1036 	}
1037 
1038 	res = &qp->resp.resources[qp->resp.res_head];
1039 	free_rd_atomic_resource(qp, res);
1040 	rxe_advance_resp_resource(qp);
1041 
1042 	skb_get(skb);
1043 	res->type = RXE_ATOMIC_MASK;
1044 	res->atomic.skb = skb;
1045 	res->first_psn = ack_pkt.psn;
1046 	res->last_psn  = ack_pkt.psn;
1047 	res->cur_psn   = ack_pkt.psn;
1048 
1049 	rc = rxe_xmit_packet(qp, &ack_pkt, skb);
1050 	if (rc) {
1051 		pr_err_ratelimited("Failed sending ack\n");
1052 		rxe_put(qp);
1053 	}
1054 out:
1055 	return rc;
1056 }
1057 
1058 static enum resp_states acknowledge(struct rxe_qp *qp,
1059 				    struct rxe_pkt_info *pkt)
1060 {
1061 	if (qp_type(qp) != IB_QPT_RC)
1062 		return RESPST_CLEANUP;
1063 
1064 	if (qp->resp.aeth_syndrome != AETH_ACK_UNLIMITED)
1065 		send_ack(qp, pkt, qp->resp.aeth_syndrome, pkt->psn);
1066 	else if (pkt->mask & RXE_ATOMIC_MASK)
1067 		send_atomic_ack(qp, pkt, AETH_ACK_UNLIMITED);
1068 	else if (bth_ack(pkt))
1069 		send_ack(qp, pkt, AETH_ACK_UNLIMITED, pkt->psn);
1070 
1071 	return RESPST_CLEANUP;
1072 }
1073 
1074 static enum resp_states cleanup(struct rxe_qp *qp,
1075 				struct rxe_pkt_info *pkt)
1076 {
1077 	struct sk_buff *skb;
1078 
1079 	if (pkt) {
1080 		skb = skb_dequeue(&qp->req_pkts);
1081 		rxe_put(qp);
1082 		kfree_skb(skb);
1083 		ib_device_put(qp->ibqp.device);
1084 	}
1085 
1086 	if (qp->resp.mr) {
1087 		rxe_put(qp->resp.mr);
1088 		qp->resp.mr = NULL;
1089 	}
1090 
1091 	return RESPST_DONE;
1092 }
1093 
1094 static struct resp_res *find_resource(struct rxe_qp *qp, u32 psn)
1095 {
1096 	int i;
1097 
1098 	for (i = 0; i < qp->attr.max_dest_rd_atomic; i++) {
1099 		struct resp_res *res = &qp->resp.resources[i];
1100 
1101 		if (res->type == 0)
1102 			continue;
1103 
1104 		if (psn_compare(psn, res->first_psn) >= 0 &&
1105 		    psn_compare(psn, res->last_psn) <= 0) {
1106 			return res;
1107 		}
1108 	}
1109 
1110 	return NULL;
1111 }
1112 
1113 static enum resp_states duplicate_request(struct rxe_qp *qp,
1114 					  struct rxe_pkt_info *pkt)
1115 {
1116 	enum resp_states rc;
1117 	u32 prev_psn = (qp->resp.ack_psn - 1) & BTH_PSN_MASK;
1118 
1119 	if (pkt->mask & RXE_SEND_MASK ||
1120 	    pkt->mask & RXE_WRITE_MASK) {
1121 		/* SEND. Ack again and cleanup. C9-105. */
1122 		send_ack(qp, pkt, AETH_ACK_UNLIMITED, prev_psn);
1123 		return RESPST_CLEANUP;
1124 	} else if (pkt->mask & RXE_READ_MASK) {
1125 		struct resp_res *res;
1126 
1127 		res = find_resource(qp, pkt->psn);
1128 		if (!res) {
1129 			/* Resource not found. Class D error.  Drop the
1130 			 * request.
1131 			 */
1132 			rc = RESPST_CLEANUP;
1133 			goto out;
1134 		} else {
1135 			/* Ensure this new request is the same as the previous
1136 			 * one or a subset of it.
1137 			 */
1138 			u64 iova = reth_va(pkt);
1139 			u32 resid = reth_len(pkt);
1140 
1141 			if (iova < res->read.va_org ||
1142 			    resid > res->read.length ||
1143 			    (iova + resid) > (res->read.va_org +
1144 					      res->read.length)) {
1145 				rc = RESPST_CLEANUP;
1146 				goto out;
1147 			}
1148 
1149 			if (reth_rkey(pkt) != res->read.rkey) {
1150 				rc = RESPST_CLEANUP;
1151 				goto out;
1152 			}
1153 
1154 			res->cur_psn = pkt->psn;
1155 			res->state = (pkt->psn == res->first_psn) ?
1156 					rdatm_res_state_new :
1157 					rdatm_res_state_replay;
1158 			res->replay = 1;
1159 
1160 			/* Reset the resource, except length. */
1161 			res->read.va_org = iova;
1162 			res->read.va = iova;
1163 			res->read.resid = resid;
1164 
1165 			/* Replay the RDMA read reply. */
1166 			qp->resp.res = res;
1167 			rc = RESPST_READ_REPLY;
1168 			goto out;
1169 		}
1170 	} else {
1171 		struct resp_res *res;
1172 
1173 		/* Find the operation in our list of responder resources. */
1174 		res = find_resource(qp, pkt->psn);
1175 		if (res) {
1176 			skb_get(res->atomic.skb);
1177 			/* Resend the result. */
1178 			rc = rxe_xmit_packet(qp, pkt, res->atomic.skb);
1179 			if (rc) {
1180 				pr_err("Failed resending result. This flow is not handled - skb ignored\n");
1181 				rc = RESPST_CLEANUP;
1182 				goto out;
1183 			}
1184 		}
1185 
1186 		/* Resource not found. Class D error. Drop the request. */
1187 		rc = RESPST_CLEANUP;
1188 		goto out;
1189 	}
1190 out:
1191 	return rc;
1192 }
1193 
1194 /* Process a class A or C. Both are treated the same in this implementation. */
1195 static void do_class_ac_error(struct rxe_qp *qp, u8 syndrome,
1196 			      enum ib_wc_status status)
1197 {
1198 	qp->resp.aeth_syndrome	= syndrome;
1199 	qp->resp.status		= status;
1200 
1201 	/* indicate that we should go through the ERROR state */
1202 	qp->resp.goto_error	= 1;
1203 }
1204 
1205 static enum resp_states do_class_d1e_error(struct rxe_qp *qp)
1206 {
1207 	/* UC */
1208 	if (qp->srq) {
1209 		/* Class E */
1210 		qp->resp.drop_msg = 1;
1211 		if (qp->resp.wqe) {
1212 			qp->resp.status = IB_WC_REM_INV_REQ_ERR;
1213 			return RESPST_COMPLETE;
1214 		} else {
1215 			return RESPST_CLEANUP;
1216 		}
1217 	} else {
1218 		/* Class D1. This packet may be the start of a
1219 		 * new message and could be valid. The previous
1220 		 * message is invalid and ignored. reset the
1221 		 * recv wr to its original state
1222 		 */
1223 		if (qp->resp.wqe) {
1224 			qp->resp.wqe->dma.resid = qp->resp.wqe->dma.length;
1225 			qp->resp.wqe->dma.cur_sge = 0;
1226 			qp->resp.wqe->dma.sge_offset = 0;
1227 			qp->resp.opcode = -1;
1228 		}
1229 
1230 		if (qp->resp.mr) {
1231 			rxe_put(qp->resp.mr);
1232 			qp->resp.mr = NULL;
1233 		}
1234 
1235 		return RESPST_CLEANUP;
1236 	}
1237 }
1238 
1239 static void rxe_drain_req_pkts(struct rxe_qp *qp, bool notify)
1240 {
1241 	struct sk_buff *skb;
1242 	struct rxe_queue *q = qp->rq.queue;
1243 
1244 	while ((skb = skb_dequeue(&qp->req_pkts))) {
1245 		rxe_put(qp);
1246 		kfree_skb(skb);
1247 		ib_device_put(qp->ibqp.device);
1248 	}
1249 
1250 	if (notify)
1251 		return;
1252 
1253 	while (!qp->srq && q && queue_head(q, q->type))
1254 		queue_advance_consumer(q, q->type);
1255 }
1256 
1257 int rxe_responder(void *arg)
1258 {
1259 	struct rxe_qp *qp = (struct rxe_qp *)arg;
1260 	struct rxe_dev *rxe = to_rdev(qp->ibqp.device);
1261 	enum resp_states state;
1262 	struct rxe_pkt_info *pkt = NULL;
1263 	int ret = 0;
1264 
1265 	if (!rxe_get(qp))
1266 		return -EAGAIN;
1267 
1268 	qp->resp.aeth_syndrome = AETH_ACK_UNLIMITED;
1269 
1270 	if (!qp->valid) {
1271 		ret = -EINVAL;
1272 		goto done;
1273 	}
1274 
1275 	switch (qp->resp.state) {
1276 	case QP_STATE_RESET:
1277 		state = RESPST_RESET;
1278 		break;
1279 
1280 	default:
1281 		state = RESPST_GET_REQ;
1282 		break;
1283 	}
1284 
1285 	while (1) {
1286 		pr_debug("qp#%d state = %s\n", qp_num(qp),
1287 			 resp_state_name[state]);
1288 		switch (state) {
1289 		case RESPST_GET_REQ:
1290 			state = get_req(qp, &pkt);
1291 			break;
1292 		case RESPST_CHK_PSN:
1293 			state = check_psn(qp, pkt);
1294 			break;
1295 		case RESPST_CHK_OP_SEQ:
1296 			state = check_op_seq(qp, pkt);
1297 			break;
1298 		case RESPST_CHK_OP_VALID:
1299 			state = check_op_valid(qp, pkt);
1300 			break;
1301 		case RESPST_CHK_RESOURCE:
1302 			state = check_resource(qp, pkt);
1303 			break;
1304 		case RESPST_CHK_LENGTH:
1305 			state = check_length(qp, pkt);
1306 			break;
1307 		case RESPST_CHK_RKEY:
1308 			state = check_rkey(qp, pkt);
1309 			break;
1310 		case RESPST_EXECUTE:
1311 			state = execute(qp, pkt);
1312 			break;
1313 		case RESPST_COMPLETE:
1314 			state = do_complete(qp, pkt);
1315 			break;
1316 		case RESPST_READ_REPLY:
1317 			state = read_reply(qp, pkt);
1318 			break;
1319 		case RESPST_ACKNOWLEDGE:
1320 			state = acknowledge(qp, pkt);
1321 			break;
1322 		case RESPST_CLEANUP:
1323 			state = cleanup(qp, pkt);
1324 			break;
1325 		case RESPST_DUPLICATE_REQUEST:
1326 			state = duplicate_request(qp, pkt);
1327 			break;
1328 		case RESPST_ERR_PSN_OUT_OF_SEQ:
1329 			/* RC only - Class B. Drop packet. */
1330 			send_ack(qp, pkt, AETH_NAK_PSN_SEQ_ERROR, qp->resp.psn);
1331 			state = RESPST_CLEANUP;
1332 			break;
1333 
1334 		case RESPST_ERR_TOO_MANY_RDMA_ATM_REQ:
1335 		case RESPST_ERR_MISSING_OPCODE_FIRST:
1336 		case RESPST_ERR_MISSING_OPCODE_LAST_C:
1337 		case RESPST_ERR_UNSUPPORTED_OPCODE:
1338 		case RESPST_ERR_MISALIGNED_ATOMIC:
1339 			/* RC Only - Class C. */
1340 			do_class_ac_error(qp, AETH_NAK_INVALID_REQ,
1341 					  IB_WC_REM_INV_REQ_ERR);
1342 			state = RESPST_COMPLETE;
1343 			break;
1344 
1345 		case RESPST_ERR_MISSING_OPCODE_LAST_D1E:
1346 			state = do_class_d1e_error(qp);
1347 			break;
1348 		case RESPST_ERR_RNR:
1349 			if (qp_type(qp) == IB_QPT_RC) {
1350 				rxe_counter_inc(rxe, RXE_CNT_SND_RNR);
1351 				/* RC - class B */
1352 				send_ack(qp, pkt, AETH_RNR_NAK |
1353 					 (~AETH_TYPE_MASK &
1354 					 qp->attr.min_rnr_timer),
1355 					 pkt->psn);
1356 			} else {
1357 				/* UD/UC - class D */
1358 				qp->resp.drop_msg = 1;
1359 			}
1360 			state = RESPST_CLEANUP;
1361 			break;
1362 
1363 		case RESPST_ERR_RKEY_VIOLATION:
1364 			if (qp_type(qp) == IB_QPT_RC) {
1365 				/* Class C */
1366 				do_class_ac_error(qp, AETH_NAK_REM_ACC_ERR,
1367 						  IB_WC_REM_ACCESS_ERR);
1368 				state = RESPST_COMPLETE;
1369 			} else {
1370 				qp->resp.drop_msg = 1;
1371 				if (qp->srq) {
1372 					/* UC/SRQ Class D */
1373 					qp->resp.status = IB_WC_REM_ACCESS_ERR;
1374 					state = RESPST_COMPLETE;
1375 				} else {
1376 					/* UC/non-SRQ Class E. */
1377 					state = RESPST_CLEANUP;
1378 				}
1379 			}
1380 			break;
1381 
1382 		case RESPST_ERR_INVALIDATE_RKEY:
1383 			/* RC - Class J. */
1384 			qp->resp.goto_error = 1;
1385 			qp->resp.status = IB_WC_REM_INV_REQ_ERR;
1386 			state = RESPST_COMPLETE;
1387 			break;
1388 
1389 		case RESPST_ERR_LENGTH:
1390 			if (qp_type(qp) == IB_QPT_RC) {
1391 				/* Class C */
1392 				do_class_ac_error(qp, AETH_NAK_INVALID_REQ,
1393 						  IB_WC_REM_INV_REQ_ERR);
1394 				state = RESPST_COMPLETE;
1395 			} else if (qp->srq) {
1396 				/* UC/UD - class E */
1397 				qp->resp.status = IB_WC_REM_INV_REQ_ERR;
1398 				state = RESPST_COMPLETE;
1399 			} else {
1400 				/* UC/UD - class D */
1401 				qp->resp.drop_msg = 1;
1402 				state = RESPST_CLEANUP;
1403 			}
1404 			break;
1405 
1406 		case RESPST_ERR_MALFORMED_WQE:
1407 			/* All, Class A. */
1408 			do_class_ac_error(qp, AETH_NAK_REM_OP_ERR,
1409 					  IB_WC_LOC_QP_OP_ERR);
1410 			state = RESPST_COMPLETE;
1411 			break;
1412 
1413 		case RESPST_ERR_CQ_OVERFLOW:
1414 			/* All - Class G */
1415 			state = RESPST_ERROR;
1416 			break;
1417 
1418 		case RESPST_DONE:
1419 			if (qp->resp.goto_error) {
1420 				state = RESPST_ERROR;
1421 				break;
1422 			}
1423 
1424 			goto done;
1425 
1426 		case RESPST_EXIT:
1427 			if (qp->resp.goto_error) {
1428 				state = RESPST_ERROR;
1429 				break;
1430 			}
1431 
1432 			goto exit;
1433 
1434 		case RESPST_RESET:
1435 			rxe_drain_req_pkts(qp, false);
1436 			qp->resp.wqe = NULL;
1437 			goto exit;
1438 
1439 		case RESPST_ERROR:
1440 			qp->resp.goto_error = 0;
1441 			pr_warn("qp#%d moved to error state\n", qp_num(qp));
1442 			rxe_qp_error(qp);
1443 			goto exit;
1444 
1445 		default:
1446 			WARN_ON_ONCE(1);
1447 		}
1448 	}
1449 
1450 exit:
1451 	ret = -EAGAIN;
1452 done:
1453 	rxe_put(qp);
1454 	return ret;
1455 }
1456