RDMA/nes: Update copyright to new legal entity and year
[cascardo/linux.git] / drivers / infiniband / hw / nes / nes_verbs.c
1 /*
2  * Copyright (c) 2006 - 2009 Intel-NE, Inc.  All rights reserved.
3  *
4  * This software is available to you under a choice of one of two
5  * licenses.  You may choose to be licensed under the terms of the GNU
6  * General Public License (GPL) Version 2, available from the file
7  * COPYING in the main directory of this source tree, or the
8  * OpenIB.org BSD license below:
9  *
10  *     Redistribution and use in source and binary forms, with or
11  *     without modification, are permitted provided that the following
12  *     conditions are met:
13  *
14  *      - Redistributions of source code must retain the above
15  *        copyright notice, this list of conditions and the following
16  *        disclaimer.
17  *
18  *      - Redistributions in binary form must reproduce the above
19  *        copyright notice, this list of conditions and the following
20  *        disclaimer in the documentation and/or other materials
21  *        provided with the distribution.
22  *
23  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
24  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
25  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
26  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
27  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
28  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
29  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
30  * SOFTWARE.
31  *
32  */
33
34 #include <linux/module.h>
35 #include <linux/moduleparam.h>
36 #include <linux/random.h>
37 #include <linux/highmem.h>
38 #include <asm/byteorder.h>
39
40 #include <rdma/ib_verbs.h>
41 #include <rdma/iw_cm.h>
42 #include <rdma/ib_user_verbs.h>
43
44 #include "nes.h"
45
46 #include <rdma/ib_umem.h>
47
48 atomic_t mod_qp_timouts;
49 atomic_t qps_created;
50 atomic_t sw_qps_destroyed;
51
52 static void nes_unregister_ofa_device(struct nes_ib_device *nesibdev);
53
54 /**
55  * nes_alloc_mw
56  */
57 static struct ib_mw *nes_alloc_mw(struct ib_pd *ibpd) {
58         struct nes_pd *nespd = to_nespd(ibpd);
59         struct nes_vnic *nesvnic = to_nesvnic(ibpd->device);
60         struct nes_device *nesdev = nesvnic->nesdev;
61         struct nes_adapter *nesadapter = nesdev->nesadapter;
62         struct nes_cqp_request *cqp_request;
63         struct nes_mr *nesmr;
64         struct ib_mw *ibmw;
65         struct nes_hw_cqp_wqe *cqp_wqe;
66         int ret;
67         u32 stag;
68         u32 stag_index = 0;
69         u32 next_stag_index = 0;
70         u32 driver_key = 0;
71         u8 stag_key = 0;
72
73         get_random_bytes(&next_stag_index, sizeof(next_stag_index));
74         stag_key = (u8)next_stag_index;
75
76         driver_key = 0;
77
78         next_stag_index >>= 8;
79         next_stag_index %= nesadapter->max_mr;
80
81         ret = nes_alloc_resource(nesadapter, nesadapter->allocated_mrs,
82                         nesadapter->max_mr, &stag_index, &next_stag_index);
83         if (ret) {
84                 return ERR_PTR(ret);
85         }
86
87         nesmr = kzalloc(sizeof(*nesmr), GFP_KERNEL);
88         if (!nesmr) {
89                 nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);
90                 return ERR_PTR(-ENOMEM);
91         }
92
93         stag = stag_index << 8;
94         stag |= driver_key;
95         stag += (u32)stag_key;
96
97         nes_debug(NES_DBG_MR, "Registering STag 0x%08X, index = 0x%08X\n",
98                         stag, stag_index);
99
100         /* Register the region with the adapter */
101         cqp_request = nes_get_cqp_request(nesdev);
102         if (cqp_request == NULL) {
103                 kfree(nesmr);
104                 nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);
105                 return ERR_PTR(-ENOMEM);
106         }
107
108         cqp_request->waiting = 1;
109         cqp_wqe = &cqp_request->cqp_wqe;
110
111         cqp_wqe->wqe_words[NES_CQP_WQE_OPCODE_IDX] =
112                         cpu_to_le32( NES_CQP_ALLOCATE_STAG | NES_CQP_STAG_RIGHTS_REMOTE_READ |
113                         NES_CQP_STAG_RIGHTS_REMOTE_WRITE | NES_CQP_STAG_VA_TO |
114                         NES_CQP_STAG_REM_ACC_EN);
115
116         nes_fill_init_cqp_wqe(cqp_wqe, nesdev);
117         set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_LEN_HIGH_PD_IDX, (nespd->pd_id & 0x00007fff));
118         set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_STAG_IDX, stag);
119
120         atomic_set(&cqp_request->refcount, 2);
121         nes_post_cqp_request(nesdev, cqp_request);
122
123         /* Wait for CQP */
124         ret = wait_event_timeout(cqp_request->waitq, (cqp_request->request_done != 0),
125                         NES_EVENT_TIMEOUT);
126         nes_debug(NES_DBG_MR, "Register STag 0x%08X completed, wait_event_timeout ret = %u,"
127                         " CQP Major:Minor codes = 0x%04X:0x%04X.\n",
128                         stag, ret, cqp_request->major_code, cqp_request->minor_code);
129         if ((!ret) || (cqp_request->major_code)) {
130                 nes_put_cqp_request(nesdev, cqp_request);
131                 kfree(nesmr);
132                 nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);
133                 if (!ret) {
134                         return ERR_PTR(-ETIME);
135                 } else {
136                         return ERR_PTR(-ENOMEM);
137                 }
138         }
139         nes_put_cqp_request(nesdev, cqp_request);
140
141         nesmr->ibmw.rkey = stag;
142         nesmr->mode = IWNES_MEMREG_TYPE_MW;
143         ibmw = &nesmr->ibmw;
144         nesmr->pbl_4k = 0;
145         nesmr->pbls_used = 0;
146
147         return ibmw;
148 }
149
150
151 /**
152  * nes_dealloc_mw
153  */
154 static int nes_dealloc_mw(struct ib_mw *ibmw)
155 {
156         struct nes_mr *nesmr = to_nesmw(ibmw);
157         struct nes_vnic *nesvnic = to_nesvnic(ibmw->device);
158         struct nes_device *nesdev = nesvnic->nesdev;
159         struct nes_adapter *nesadapter = nesdev->nesadapter;
160         struct nes_hw_cqp_wqe *cqp_wqe;
161         struct nes_cqp_request *cqp_request;
162         int err = 0;
163         int ret;
164
165         /* Deallocate the window with the adapter */
166         cqp_request = nes_get_cqp_request(nesdev);
167         if (cqp_request == NULL) {
168                 nes_debug(NES_DBG_MR, "Failed to get a cqp_request.\n");
169                 return -ENOMEM;
170         }
171         cqp_request->waiting = 1;
172         cqp_wqe = &cqp_request->cqp_wqe;
173         nes_fill_init_cqp_wqe(cqp_wqe, nesdev);
174         set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_WQE_OPCODE_IDX, NES_CQP_DEALLOCATE_STAG);
175         set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_STAG_IDX, ibmw->rkey);
176
177         atomic_set(&cqp_request->refcount, 2);
178         nes_post_cqp_request(nesdev, cqp_request);
179
180         /* Wait for CQP */
181         nes_debug(NES_DBG_MR, "Waiting for deallocate STag 0x%08X to complete.\n",
182                         ibmw->rkey);
183         ret = wait_event_timeout(cqp_request->waitq, (0 != cqp_request->request_done),
184                         NES_EVENT_TIMEOUT);
185         nes_debug(NES_DBG_MR, "Deallocate STag completed, wait_event_timeout ret = %u,"
186                         " CQP Major:Minor codes = 0x%04X:0x%04X.\n",
187                         ret, cqp_request->major_code, cqp_request->minor_code);
188         if (!ret)
189                 err = -ETIME;
190         else if (cqp_request->major_code)
191                 err = -EIO;
192
193         nes_put_cqp_request(nesdev, cqp_request);
194
195         nes_free_resource(nesadapter, nesadapter->allocated_mrs,
196                         (ibmw->rkey & 0x0fffff00) >> 8);
197         kfree(nesmr);
198
199         return err;
200 }
201
202
203 /**
204  * nes_bind_mw
205  */
206 static int nes_bind_mw(struct ib_qp *ibqp, struct ib_mw *ibmw,
207                 struct ib_mw_bind *ibmw_bind)
208 {
209         u64 u64temp;
210         struct nes_vnic *nesvnic = to_nesvnic(ibqp->device);
211         struct nes_device *nesdev = nesvnic->nesdev;
212         /* struct nes_mr *nesmr = to_nesmw(ibmw); */
213         struct nes_qp *nesqp = to_nesqp(ibqp);
214         struct nes_hw_qp_wqe *wqe;
215         unsigned long flags = 0;
216         u32 head;
217         u32 wqe_misc = 0;
218         u32 qsize;
219
220         if (nesqp->ibqp_state > IB_QPS_RTS)
221                 return -EINVAL;
222
223         spin_lock_irqsave(&nesqp->lock, flags);
224
225         head = nesqp->hwqp.sq_head;
226         qsize = nesqp->hwqp.sq_tail;
227
228         /* Check for SQ overflow */
229         if (((head + (2 * qsize) - nesqp->hwqp.sq_tail) % qsize) == (qsize - 1)) {
230                 spin_unlock_irqrestore(&nesqp->lock, flags);
231                 return -EINVAL;
232         }
233
234         wqe = &nesqp->hwqp.sq_vbase[head];
235         /* nes_debug(NES_DBG_MR, "processing sq wqe at %p, head = %u.\n", wqe, head); */
236         nes_fill_init_qp_wqe(wqe, nesqp, head);
237         u64temp = ibmw_bind->wr_id;
238         set_wqe_64bit_value(wqe->wqe_words, NES_IWARP_SQ_WQE_COMP_SCRATCH_LOW_IDX, u64temp);
239         wqe_misc = NES_IWARP_SQ_OP_BIND;
240
241         wqe_misc |= NES_IWARP_SQ_WQE_LOCAL_FENCE;
242
243         if (ibmw_bind->send_flags & IB_SEND_SIGNALED)
244                 wqe_misc |= NES_IWARP_SQ_WQE_SIGNALED_COMPL;
245
246         if (ibmw_bind->mw_access_flags & IB_ACCESS_REMOTE_WRITE) {
247                 wqe_misc |= NES_CQP_STAG_RIGHTS_REMOTE_WRITE;
248         }
249         if (ibmw_bind->mw_access_flags & IB_ACCESS_REMOTE_READ) {
250                 wqe_misc |= NES_CQP_STAG_RIGHTS_REMOTE_READ;
251         }
252
253         set_wqe_32bit_value(wqe->wqe_words, NES_IWARP_SQ_WQE_MISC_IDX, wqe_misc);
254         set_wqe_32bit_value(wqe->wqe_words, NES_IWARP_SQ_BIND_WQE_MR_IDX, ibmw_bind->mr->lkey);
255         set_wqe_32bit_value(wqe->wqe_words, NES_IWARP_SQ_BIND_WQE_MW_IDX, ibmw->rkey);
256         set_wqe_32bit_value(wqe->wqe_words, NES_IWARP_SQ_BIND_WQE_LENGTH_LOW_IDX,
257                         ibmw_bind->length);
258         wqe->wqe_words[NES_IWARP_SQ_BIND_WQE_LENGTH_HIGH_IDX] = 0;
259         u64temp = (u64)ibmw_bind->addr;
260         set_wqe_64bit_value(wqe->wqe_words, NES_IWARP_SQ_BIND_WQE_VA_FBO_LOW_IDX, u64temp);
261
262         head++;
263         if (head >= qsize)
264                 head = 0;
265
266         nesqp->hwqp.sq_head = head;
267         barrier();
268
269         nes_write32(nesdev->regs+NES_WQE_ALLOC,
270                         (1 << 24) | 0x00800000 | nesqp->hwqp.qp_id);
271
272         spin_unlock_irqrestore(&nesqp->lock, flags);
273
274         return 0;
275 }
276
277
278 /**
279  * nes_alloc_fmr
280  */
281 static struct ib_fmr *nes_alloc_fmr(struct ib_pd *ibpd,
282                 int ibmr_access_flags,
283                 struct ib_fmr_attr *ibfmr_attr)
284 {
285         unsigned long flags;
286         struct nes_pd *nespd = to_nespd(ibpd);
287         struct nes_vnic *nesvnic = to_nesvnic(ibpd->device);
288         struct nes_device *nesdev = nesvnic->nesdev;
289         struct nes_adapter *nesadapter = nesdev->nesadapter;
290         struct nes_fmr *nesfmr;
291         struct nes_cqp_request *cqp_request;
292         struct nes_hw_cqp_wqe *cqp_wqe;
293         int ret;
294         u32 stag;
295         u32 stag_index = 0;
296         u32 next_stag_index = 0;
297         u32 driver_key = 0;
298         u32 opcode = 0;
299         u8 stag_key = 0;
300         int i=0;
301         struct nes_vpbl vpbl;
302
303         get_random_bytes(&next_stag_index, sizeof(next_stag_index));
304         stag_key = (u8)next_stag_index;
305
306         driver_key = 0;
307
308         next_stag_index >>= 8;
309         next_stag_index %= nesadapter->max_mr;
310
311         ret = nes_alloc_resource(nesadapter, nesadapter->allocated_mrs,
312                         nesadapter->max_mr, &stag_index, &next_stag_index);
313         if (ret) {
314                 goto failed_resource_alloc;
315         }
316
317         nesfmr = kzalloc(sizeof(*nesfmr), GFP_KERNEL);
318         if (!nesfmr) {
319                 ret = -ENOMEM;
320                 goto failed_fmr_alloc;
321         }
322
323         nesfmr->nesmr.mode = IWNES_MEMREG_TYPE_FMR;
324         if (ibfmr_attr->max_pages == 1) {
325                 /* use zero length PBL */
326                 nesfmr->nesmr.pbl_4k = 0;
327                 nesfmr->nesmr.pbls_used = 0;
328         } else if (ibfmr_attr->max_pages <= 32) {
329                 /* use PBL 256 */
330                 nesfmr->nesmr.pbl_4k = 0;
331                 nesfmr->nesmr.pbls_used = 1;
332         } else if (ibfmr_attr->max_pages <= 512) {
333                 /* use 4K PBLs */
334                 nesfmr->nesmr.pbl_4k = 1;
335                 nesfmr->nesmr.pbls_used = 1;
336         } else {
337                 /* use two level 4K PBLs */
338                 /* add support for two level 256B PBLs */
339                 nesfmr->nesmr.pbl_4k = 1;
340                 nesfmr->nesmr.pbls_used = 1 + (ibfmr_attr->max_pages >> 9) +
341                                 ((ibfmr_attr->max_pages & 511) ? 1 : 0);
342         }
343         /* Register the region with the adapter */
344         spin_lock_irqsave(&nesadapter->pbl_lock, flags);
345
346         /* track PBL resources */
347         if (nesfmr->nesmr.pbls_used != 0) {
348                 if (nesfmr->nesmr.pbl_4k) {
349                         if (nesfmr->nesmr.pbls_used > nesadapter->free_4kpbl) {
350                                 spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
351                                 ret = -ENOMEM;
352                                 goto failed_vpbl_avail;
353                         } else {
354                                 nesadapter->free_4kpbl -= nesfmr->nesmr.pbls_used;
355                         }
356                 } else {
357                         if (nesfmr->nesmr.pbls_used > nesadapter->free_256pbl) {
358                                 spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
359                                 ret = -ENOMEM;
360                                 goto failed_vpbl_avail;
361                         } else {
362                                 nesadapter->free_256pbl -= nesfmr->nesmr.pbls_used;
363                         }
364                 }
365         }
366
367         /* one level pbl */
368         if (nesfmr->nesmr.pbls_used == 0) {
369                 nesfmr->root_vpbl.pbl_vbase = NULL;
370                 nes_debug(NES_DBG_MR,  "zero level pbl \n");
371         } else if (nesfmr->nesmr.pbls_used == 1) {
372                 /* can change it to kmalloc & dma_map_single */
373                 nesfmr->root_vpbl.pbl_vbase = pci_alloc_consistent(nesdev->pcidev, 4096,
374                                 &nesfmr->root_vpbl.pbl_pbase);
375                 if (!nesfmr->root_vpbl.pbl_vbase) {
376                         spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
377                         ret = -ENOMEM;
378                         goto failed_vpbl_alloc;
379                 }
380                 nesfmr->leaf_pbl_cnt = 0;
381                 nes_debug(NES_DBG_MR, "one level pbl, root_vpbl.pbl_vbase=%p \n",
382                                 nesfmr->root_vpbl.pbl_vbase);
383         }
384         /* two level pbl */
385         else {
386                 nesfmr->root_vpbl.pbl_vbase = pci_alloc_consistent(nesdev->pcidev, 8192,
387                                 &nesfmr->root_vpbl.pbl_pbase);
388                 if (!nesfmr->root_vpbl.pbl_vbase) {
389                         spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
390                         ret = -ENOMEM;
391                         goto failed_vpbl_alloc;
392                 }
393
394                 nesfmr->leaf_pbl_cnt = nesfmr->nesmr.pbls_used-1;
395                 nesfmr->root_vpbl.leaf_vpbl = kzalloc(sizeof(*nesfmr->root_vpbl.leaf_vpbl)*1024, GFP_ATOMIC);
396                 if (!nesfmr->root_vpbl.leaf_vpbl) {
397                         spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
398                         ret = -ENOMEM;
399                         goto failed_leaf_vpbl_alloc;
400                 }
401
402                 nes_debug(NES_DBG_MR, "two level pbl, root_vpbl.pbl_vbase=%p"
403                                 " leaf_pbl_cnt=%d root_vpbl.leaf_vpbl=%p\n",
404                                 nesfmr->root_vpbl.pbl_vbase, nesfmr->leaf_pbl_cnt, nesfmr->root_vpbl.leaf_vpbl);
405
406                 for (i=0; i<nesfmr->leaf_pbl_cnt; i++)
407                         nesfmr->root_vpbl.leaf_vpbl[i].pbl_vbase = NULL;
408
409                 for (i=0; i<nesfmr->leaf_pbl_cnt; i++) {
410                         vpbl.pbl_vbase = pci_alloc_consistent(nesdev->pcidev, 4096,
411                                         &vpbl.pbl_pbase);
412
413                         if (!vpbl.pbl_vbase) {
414                                 ret = -ENOMEM;
415                                 spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
416                                 goto failed_leaf_vpbl_pages_alloc;
417                         }
418
419                         nesfmr->root_vpbl.pbl_vbase[i].pa_low = cpu_to_le32((u32)vpbl.pbl_pbase);
420                         nesfmr->root_vpbl.pbl_vbase[i].pa_high = cpu_to_le32((u32)((((u64)vpbl.pbl_pbase)>>32)));
421                         nesfmr->root_vpbl.leaf_vpbl[i] = vpbl;
422
423                         nes_debug(NES_DBG_MR, "pbase_low=0x%x, pbase_high=0x%x, vpbl=%p\n",
424                                         nesfmr->root_vpbl.pbl_vbase[i].pa_low,
425                                         nesfmr->root_vpbl.pbl_vbase[i].pa_high,
426                                         &nesfmr->root_vpbl.leaf_vpbl[i]);
427                 }
428         }
429         nesfmr->ib_qp = NULL;
430         nesfmr->access_rights =0;
431
432         stag = stag_index << 8;
433         stag |= driver_key;
434         stag += (u32)stag_key;
435
436         spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
437         cqp_request = nes_get_cqp_request(nesdev);
438         if (cqp_request == NULL) {
439                 nes_debug(NES_DBG_MR, "Failed to get a cqp_request.\n");
440                 ret = -ENOMEM;
441                 goto failed_leaf_vpbl_pages_alloc;
442         }
443         cqp_request->waiting = 1;
444         cqp_wqe = &cqp_request->cqp_wqe;
445
446         nes_debug(NES_DBG_MR, "Registering STag 0x%08X, index = 0x%08X\n",
447                         stag, stag_index);
448
449         opcode = NES_CQP_ALLOCATE_STAG | NES_CQP_STAG_VA_TO | NES_CQP_STAG_MR;
450
451         if (nesfmr->nesmr.pbl_4k == 1)
452                 opcode |= NES_CQP_STAG_PBL_BLK_SIZE;
453
454         if (ibmr_access_flags & IB_ACCESS_REMOTE_WRITE) {
455                 opcode |= NES_CQP_STAG_RIGHTS_REMOTE_WRITE |
456                                 NES_CQP_STAG_RIGHTS_LOCAL_WRITE | NES_CQP_STAG_REM_ACC_EN;
457                 nesfmr->access_rights |=
458                                 NES_CQP_STAG_RIGHTS_REMOTE_WRITE | NES_CQP_STAG_RIGHTS_LOCAL_WRITE |
459                                 NES_CQP_STAG_REM_ACC_EN;
460         }
461
462         if (ibmr_access_flags & IB_ACCESS_REMOTE_READ) {
463                 opcode |= NES_CQP_STAG_RIGHTS_REMOTE_READ |
464                                 NES_CQP_STAG_RIGHTS_LOCAL_READ | NES_CQP_STAG_REM_ACC_EN;
465                 nesfmr->access_rights |=
466                                 NES_CQP_STAG_RIGHTS_REMOTE_READ | NES_CQP_STAG_RIGHTS_LOCAL_READ |
467                                 NES_CQP_STAG_REM_ACC_EN;
468         }
469
470         nes_fill_init_cqp_wqe(cqp_wqe, nesdev);
471         set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_WQE_OPCODE_IDX, opcode);
472         set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_LEN_HIGH_PD_IDX, (nespd->pd_id & 0x00007fff));
473         set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_STAG_IDX, stag);
474
475         cqp_wqe->wqe_words[NES_CQP_STAG_WQE_PBL_BLK_COUNT_IDX] =
476                         cpu_to_le32((nesfmr->nesmr.pbls_used>1) ?
477                         (nesfmr->nesmr.pbls_used-1) : nesfmr->nesmr.pbls_used);
478
479         atomic_set(&cqp_request->refcount, 2);
480         nes_post_cqp_request(nesdev, cqp_request);
481
482         /* Wait for CQP */
483         ret = wait_event_timeout(cqp_request->waitq, (cqp_request->request_done != 0),
484                         NES_EVENT_TIMEOUT);
485         nes_debug(NES_DBG_MR, "Register STag 0x%08X completed, wait_event_timeout ret = %u,"
486                         " CQP Major:Minor codes = 0x%04X:0x%04X.\n",
487                         stag, ret, cqp_request->major_code, cqp_request->minor_code);
488
489         if ((!ret) || (cqp_request->major_code)) {
490                 nes_put_cqp_request(nesdev, cqp_request);
491                 ret = (!ret) ? -ETIME : -EIO;
492                 goto failed_leaf_vpbl_pages_alloc;
493         }
494         nes_put_cqp_request(nesdev, cqp_request);
495         nesfmr->nesmr.ibfmr.lkey = stag;
496         nesfmr->nesmr.ibfmr.rkey = stag;
497         nesfmr->attr = *ibfmr_attr;
498
499         return &nesfmr->nesmr.ibfmr;
500
501         failed_leaf_vpbl_pages_alloc:
502         /* unroll all allocated pages */
503         for (i=0; i<nesfmr->leaf_pbl_cnt; i++) {
504                 if (nesfmr->root_vpbl.leaf_vpbl[i].pbl_vbase) {
505                         pci_free_consistent(nesdev->pcidev, 4096, nesfmr->root_vpbl.leaf_vpbl[i].pbl_vbase,
506                                         nesfmr->root_vpbl.leaf_vpbl[i].pbl_pbase);
507                 }
508         }
509         if (nesfmr->root_vpbl.leaf_vpbl)
510                 kfree(nesfmr->root_vpbl.leaf_vpbl);
511
512         failed_leaf_vpbl_alloc:
513         if (nesfmr->leaf_pbl_cnt == 0) {
514                 if (nesfmr->root_vpbl.pbl_vbase)
515                         pci_free_consistent(nesdev->pcidev, 4096, nesfmr->root_vpbl.pbl_vbase,
516                                         nesfmr->root_vpbl.pbl_pbase);
517         } else
518                 pci_free_consistent(nesdev->pcidev, 8192, nesfmr->root_vpbl.pbl_vbase,
519                                 nesfmr->root_vpbl.pbl_pbase);
520
521         failed_vpbl_alloc:
522         if (nesfmr->nesmr.pbls_used != 0) {
523                 spin_lock_irqsave(&nesadapter->pbl_lock, flags);
524                 if (nesfmr->nesmr.pbl_4k)
525                         nesadapter->free_4kpbl += nesfmr->nesmr.pbls_used;
526                 else
527                         nesadapter->free_256pbl += nesfmr->nesmr.pbls_used;
528                 spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
529         }
530
531 failed_vpbl_avail:
532         kfree(nesfmr);
533
534         failed_fmr_alloc:
535         nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);
536
537         failed_resource_alloc:
538         return ERR_PTR(ret);
539 }
540
541
542 /**
543  * nes_dealloc_fmr
544  */
545 static int nes_dealloc_fmr(struct ib_fmr *ibfmr)
546 {
547         unsigned long flags;
548         struct nes_mr *nesmr = to_nesmr_from_ibfmr(ibfmr);
549         struct nes_fmr *nesfmr = to_nesfmr(nesmr);
550         struct nes_vnic *nesvnic = to_nesvnic(ibfmr->device);
551         struct nes_device *nesdev = nesvnic->nesdev;
552         struct nes_adapter *nesadapter = nesdev->nesadapter;
553         int i = 0;
554         int rc;
555
556         /* free the resources */
557         if (nesfmr->leaf_pbl_cnt == 0) {
558                 /* single PBL case */
559                 if (nesfmr->root_vpbl.pbl_vbase)
560                         pci_free_consistent(nesdev->pcidev, 4096, nesfmr->root_vpbl.pbl_vbase,
561                                         nesfmr->root_vpbl.pbl_pbase);
562         } else {
563                 for (i = 0; i < nesfmr->leaf_pbl_cnt; i++) {
564                         pci_free_consistent(nesdev->pcidev, 4096, nesfmr->root_vpbl.leaf_vpbl[i].pbl_vbase,
565                                         nesfmr->root_vpbl.leaf_vpbl[i].pbl_pbase);
566                 }
567                 kfree(nesfmr->root_vpbl.leaf_vpbl);
568                 pci_free_consistent(nesdev->pcidev, 8192, nesfmr->root_vpbl.pbl_vbase,
569                                 nesfmr->root_vpbl.pbl_pbase);
570         }
571         nesmr->ibmw.device = ibfmr->device;
572         nesmr->ibmw.pd = ibfmr->pd;
573         nesmr->ibmw.rkey = ibfmr->rkey;
574         nesmr->ibmw.uobject = NULL;
575
576         rc = nes_dealloc_mw(&nesmr->ibmw);
577
578         if ((rc == 0) && (nesfmr->nesmr.pbls_used != 0)) {
579                 spin_lock_irqsave(&nesadapter->pbl_lock, flags);
580                 if (nesfmr->nesmr.pbl_4k) {
581                         nesadapter->free_4kpbl += nesfmr->nesmr.pbls_used;
582                         WARN_ON(nesadapter->free_4kpbl > nesadapter->max_4kpbl);
583                 } else {
584                         nesadapter->free_256pbl += nesfmr->nesmr.pbls_used;
585                         WARN_ON(nesadapter->free_256pbl > nesadapter->max_256pbl);
586                 }
587                 spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
588         }
589
590         return rc;
591 }
592
593
594 /**
595  * nes_map_phys_fmr
596  */
597 static int nes_map_phys_fmr(struct ib_fmr *ibfmr, u64 *page_list,
598                 int list_len, u64 iova)
599 {
600         return 0;
601 }
602
603
604 /**
605  * nes_unmap_frm
606  */
607 static int nes_unmap_fmr(struct list_head *ibfmr_list)
608 {
609         return 0;
610 }
611
612
613
614 /**
615  * nes_query_device
616  */
617 static int nes_query_device(struct ib_device *ibdev, struct ib_device_attr *props)
618 {
619         struct nes_vnic *nesvnic = to_nesvnic(ibdev);
620         struct nes_device *nesdev = nesvnic->nesdev;
621         struct nes_ib_device *nesibdev = nesvnic->nesibdev;
622
623         memset(props, 0, sizeof(*props));
624         memcpy(&props->sys_image_guid, nesvnic->netdev->dev_addr, 6);
625
626         props->fw_ver = nesdev->nesadapter->fw_ver;
627         props->device_cap_flags = nesdev->nesadapter->device_cap_flags;
628         props->vendor_id = nesdev->nesadapter->vendor_id;
629         props->vendor_part_id = nesdev->nesadapter->vendor_part_id;
630         props->hw_ver = nesdev->nesadapter->hw_rev;
631         props->max_mr_size = 0x80000000;
632         props->max_qp = nesibdev->max_qp;
633         props->max_qp_wr = nesdev->nesadapter->max_qp_wr - 2;
634         props->max_sge = nesdev->nesadapter->max_sge;
635         props->max_cq = nesibdev->max_cq;
636         props->max_cqe = nesdev->nesadapter->max_cqe - 1;
637         props->max_mr = nesibdev->max_mr;
638         props->max_mw = nesibdev->max_mr;
639         props->max_pd = nesibdev->max_pd;
640         props->max_sge_rd = 1;
641         switch (nesdev->nesadapter->max_irrq_wr) {
642                 case 0:
643                         props->max_qp_rd_atom = 1;
644                         break;
645                 case 1:
646                         props->max_qp_rd_atom = 4;
647                         break;
648                 case 2:
649                         props->max_qp_rd_atom = 16;
650                         break;
651                 case 3:
652                         props->max_qp_rd_atom = 32;
653                         break;
654                 default:
655                         props->max_qp_rd_atom = 0;
656         }
657         props->max_qp_init_rd_atom = props->max_qp_wr;
658         props->atomic_cap = IB_ATOMIC_NONE;
659         props->max_map_per_fmr = 1;
660
661         return 0;
662 }
663
664
665 /**
666  * nes_query_port
667  */
668 static int nes_query_port(struct ib_device *ibdev, u8 port, struct ib_port_attr *props)
669 {
670         memset(props, 0, sizeof(*props));
671
672         props->max_mtu = IB_MTU_2048;
673         props->active_mtu = IB_MTU_2048;
674         props->lid = 1;
675         props->lmc = 0;
676         props->sm_lid = 0;
677         props->sm_sl = 0;
678         props->state = IB_PORT_ACTIVE;
679         props->phys_state = 0;
680         props->port_cap_flags = IB_PORT_CM_SUP | IB_PORT_REINIT_SUP |
681                         IB_PORT_VENDOR_CLASS_SUP | IB_PORT_BOOT_MGMT_SUP;
682         props->gid_tbl_len = 1;
683         props->pkey_tbl_len = 1;
684         props->qkey_viol_cntr = 0;
685         props->active_width = IB_WIDTH_4X;
686         props->active_speed = 1;
687         props->max_msg_sz = 0x80000000;
688
689         return 0;
690 }
691
692
693 /**
694  * nes_modify_port
695  */
696 static int nes_modify_port(struct ib_device *ibdev, u8 port,
697                 int port_modify_mask, struct ib_port_modify *props)
698 {
699         return 0;
700 }
701
702
703 /**
704  * nes_query_pkey
705  */
706 static int nes_query_pkey(struct ib_device *ibdev, u8 port, u16 index, u16 *pkey)
707 {
708         *pkey = 0;
709         return 0;
710 }
711
712
713 /**
714  * nes_query_gid
715  */
716 static int nes_query_gid(struct ib_device *ibdev, u8 port,
717                 int index, union ib_gid *gid)
718 {
719         struct nes_vnic *nesvnic = to_nesvnic(ibdev);
720
721         memset(&(gid->raw[0]), 0, sizeof(gid->raw));
722         memcpy(&(gid->raw[0]), nesvnic->netdev->dev_addr, 6);
723
724         return 0;
725 }
726
727
728 /**
729  * nes_alloc_ucontext - Allocate the user context data structure. This keeps track
730  * of all objects associated with a particular user-mode client.
731  */
732 static struct ib_ucontext *nes_alloc_ucontext(struct ib_device *ibdev,
733                 struct ib_udata *udata)
734 {
735         struct nes_vnic *nesvnic = to_nesvnic(ibdev);
736         struct nes_device *nesdev = nesvnic->nesdev;
737         struct nes_adapter *nesadapter = nesdev->nesadapter;
738         struct nes_alloc_ucontext_req req;
739         struct nes_alloc_ucontext_resp uresp;
740         struct nes_ucontext *nes_ucontext;
741         struct nes_ib_device *nesibdev = nesvnic->nesibdev;
742
743
744         if (ib_copy_from_udata(&req, udata, sizeof(struct nes_alloc_ucontext_req))) {
745                 printk(KERN_ERR PFX "Invalid structure size on allocate user context.\n");
746                 return ERR_PTR(-EINVAL);
747         }
748
749         if (req.userspace_ver != NES_ABI_USERSPACE_VER) {
750                 printk(KERN_ERR PFX "Invalid userspace driver version detected. Detected version %d, should be %d\n",
751                         req.userspace_ver, NES_ABI_USERSPACE_VER);
752                 return ERR_PTR(-EINVAL);
753         }
754
755
756         memset(&uresp, 0, sizeof uresp);
757
758         uresp.max_qps = nesibdev->max_qp;
759         uresp.max_pds = nesibdev->max_pd;
760         uresp.wq_size = nesdev->nesadapter->max_qp_wr * 2;
761         uresp.virtwq = nesadapter->virtwq;
762         uresp.kernel_ver = NES_ABI_KERNEL_VER;
763
764         nes_ucontext = kzalloc(sizeof *nes_ucontext, GFP_KERNEL);
765         if (!nes_ucontext)
766                 return ERR_PTR(-ENOMEM);
767
768         nes_ucontext->nesdev = nesdev;
769         nes_ucontext->mmap_wq_offset = uresp.max_pds;
770         nes_ucontext->mmap_cq_offset = nes_ucontext->mmap_wq_offset +
771                         ((sizeof(struct nes_hw_qp_wqe) * uresp.max_qps * 2) + PAGE_SIZE-1) /
772                         PAGE_SIZE;
773
774
775         if (ib_copy_to_udata(udata, &uresp, sizeof uresp)) {
776                 kfree(nes_ucontext);
777                 return ERR_PTR(-EFAULT);
778         }
779
780         INIT_LIST_HEAD(&nes_ucontext->cq_reg_mem_list);
781         INIT_LIST_HEAD(&nes_ucontext->qp_reg_mem_list);
782         atomic_set(&nes_ucontext->usecnt, 1);
783         return &nes_ucontext->ibucontext;
784 }
785
786
787 /**
788  * nes_dealloc_ucontext
789  */
790 static int nes_dealloc_ucontext(struct ib_ucontext *context)
791 {
792         /* struct nes_vnic *nesvnic = to_nesvnic(context->device); */
793         /* struct nes_device *nesdev = nesvnic->nesdev; */
794         struct nes_ucontext *nes_ucontext = to_nesucontext(context);
795
796         if (!atomic_dec_and_test(&nes_ucontext->usecnt))
797           return 0;
798         kfree(nes_ucontext);
799         return 0;
800 }
801
802
803 /**
804  * nes_mmap
805  */
806 static int nes_mmap(struct ib_ucontext *context, struct vm_area_struct *vma)
807 {
808         unsigned long index;
809         struct nes_vnic *nesvnic = to_nesvnic(context->device);
810         struct nes_device *nesdev = nesvnic->nesdev;
811         /* struct nes_adapter *nesadapter = nesdev->nesadapter; */
812         struct nes_ucontext *nes_ucontext;
813         struct nes_qp *nesqp;
814
815         nes_ucontext = to_nesucontext(context);
816
817
818         if (vma->vm_pgoff >= nes_ucontext->mmap_wq_offset) {
819                 index = (vma->vm_pgoff - nes_ucontext->mmap_wq_offset) * PAGE_SIZE;
820                 index /= ((sizeof(struct nes_hw_qp_wqe) * nesdev->nesadapter->max_qp_wr * 2) +
821                                 PAGE_SIZE-1) & (~(PAGE_SIZE-1));
822                 if (!test_bit(index, nes_ucontext->allocated_wqs)) {
823                         nes_debug(NES_DBG_MMAP, "wq %lu not allocated\n", index);
824                         return -EFAULT;
825                 }
826                 nesqp = nes_ucontext->mmap_nesqp[index];
827                 if (nesqp == NULL) {
828                         nes_debug(NES_DBG_MMAP, "wq %lu has a NULL QP base.\n", index);
829                         return -EFAULT;
830                 }
831                 if (remap_pfn_range(vma, vma->vm_start,
832                                 virt_to_phys(nesqp->hwqp.sq_vbase) >> PAGE_SHIFT,
833                                 vma->vm_end - vma->vm_start,
834                                 vma->vm_page_prot)) {
835                         nes_debug(NES_DBG_MMAP, "remap_pfn_range failed.\n");
836                         return -EAGAIN;
837                 }
838                 vma->vm_private_data = nesqp;
839                 return 0;
840         } else {
841                 index = vma->vm_pgoff;
842                 if (!test_bit(index, nes_ucontext->allocated_doorbells))
843                         return -EFAULT;
844
845                 vma->vm_page_prot = pgprot_noncached(vma->vm_page_prot);
846                 if (io_remap_pfn_range(vma, vma->vm_start,
847                                 (nesdev->doorbell_start +
848                                 ((nes_ucontext->mmap_db_index[index] - nesdev->base_doorbell_index) * 4096))
849                                 >> PAGE_SHIFT, PAGE_SIZE, vma->vm_page_prot))
850                         return -EAGAIN;
851                 vma->vm_private_data = nes_ucontext;
852                 return 0;
853         }
854
855         return -ENOSYS;
856 }
857
858
859 /**
860  * nes_alloc_pd
861  */
862 static struct ib_pd *nes_alloc_pd(struct ib_device *ibdev,
863                 struct ib_ucontext *context, struct ib_udata *udata)
864 {
865         struct nes_pd *nespd;
866         struct nes_vnic *nesvnic = to_nesvnic(ibdev);
867         struct nes_device *nesdev = nesvnic->nesdev;
868         struct nes_adapter *nesadapter = nesdev->nesadapter;
869         struct nes_ucontext *nesucontext;
870         struct nes_alloc_pd_resp uresp;
871         u32 pd_num = 0;
872         int err;
873
874         nes_debug(NES_DBG_PD, "nesvnic=%p, netdev=%p %s, ibdev=%p, context=%p, netdev refcnt=%u\n",
875                         nesvnic, nesdev->netdev[0], nesdev->netdev[0]->name, ibdev, context,
876                         atomic_read(&nesvnic->netdev->refcnt));
877
878         err = nes_alloc_resource(nesadapter, nesadapter->allocated_pds,
879                         nesadapter->max_pd, &pd_num, &nesadapter->next_pd);
880         if (err) {
881                 return ERR_PTR(err);
882         }
883
884         nespd = kzalloc(sizeof (struct nes_pd), GFP_KERNEL);
885         if (!nespd) {
886                 nes_free_resource(nesadapter, nesadapter->allocated_pds, pd_num);
887                 return ERR_PTR(-ENOMEM);
888         }
889
890         nes_debug(NES_DBG_PD, "Allocating PD (%p) for ib device %s\n",
891                         nespd, nesvnic->nesibdev->ibdev.name);
892
893         nespd->pd_id = (pd_num << (PAGE_SHIFT-12)) + nesadapter->base_pd;
894
895         if (context) {
896                 nesucontext = to_nesucontext(context);
897                 nespd->mmap_db_index = find_next_zero_bit(nesucontext->allocated_doorbells,
898                                 NES_MAX_USER_DB_REGIONS, nesucontext->first_free_db);
899                 nes_debug(NES_DBG_PD, "find_first_zero_biton doorbells returned %u, mapping pd_id %u.\n",
900                                 nespd->mmap_db_index, nespd->pd_id);
901                 if (nespd->mmap_db_index >= NES_MAX_USER_DB_REGIONS) {
902                         nes_debug(NES_DBG_PD, "mmap_db_index > MAX\n");
903                         nes_free_resource(nesadapter, nesadapter->allocated_pds, pd_num);
904                         kfree(nespd);
905                         return ERR_PTR(-ENOMEM);
906                 }
907
908                 uresp.pd_id = nespd->pd_id;
909                 uresp.mmap_db_index = nespd->mmap_db_index;
910                 if (ib_copy_to_udata(udata, &uresp, sizeof (struct nes_alloc_pd_resp))) {
911                         nes_free_resource(nesadapter, nesadapter->allocated_pds, pd_num);
912                         kfree(nespd);
913                         return ERR_PTR(-EFAULT);
914                 }
915
916                 set_bit(nespd->mmap_db_index, nesucontext->allocated_doorbells);
917                 nesucontext->mmap_db_index[nespd->mmap_db_index] = nespd->pd_id;
918                 nesucontext->first_free_db = nespd->mmap_db_index + 1;
919         }
920
921         nes_debug(NES_DBG_PD, "PD%u structure located @%p.\n", nespd->pd_id, nespd);
922         return &nespd->ibpd;
923 }
924
925
926 /**
927  * nes_dealloc_pd
928  */
929 static int nes_dealloc_pd(struct ib_pd *ibpd)
930 {
931         struct nes_ucontext *nesucontext;
932         struct nes_pd *nespd = to_nespd(ibpd);
933         struct nes_vnic *nesvnic = to_nesvnic(ibpd->device);
934         struct nes_device *nesdev = nesvnic->nesdev;
935         struct nes_adapter *nesadapter = nesdev->nesadapter;
936
937         if ((ibpd->uobject) && (ibpd->uobject->context)) {
938                 nesucontext = to_nesucontext(ibpd->uobject->context);
939                 nes_debug(NES_DBG_PD, "Clearing bit %u from allocated doorbells\n",
940                                 nespd->mmap_db_index);
941                 clear_bit(nespd->mmap_db_index, nesucontext->allocated_doorbells);
942                 nesucontext->mmap_db_index[nespd->mmap_db_index] = 0;
943                 if (nesucontext->first_free_db > nespd->mmap_db_index) {
944                         nesucontext->first_free_db = nespd->mmap_db_index;
945                 }
946         }
947
948         nes_debug(NES_DBG_PD, "Deallocating PD%u structure located @%p.\n",
949                         nespd->pd_id, nespd);
950         nes_free_resource(nesadapter, nesadapter->allocated_pds,
951                         (nespd->pd_id-nesadapter->base_pd)>>(PAGE_SHIFT-12));
952         kfree(nespd);
953
954         return 0;
955 }
956
957
958 /**
959  * nes_create_ah
960  */
961 static struct ib_ah *nes_create_ah(struct ib_pd *pd, struct ib_ah_attr *ah_attr)
962 {
963         return ERR_PTR(-ENOSYS);
964 }
965
966
967 /**
968  * nes_destroy_ah
969  */
970 static int nes_destroy_ah(struct ib_ah *ah)
971 {
972         return -ENOSYS;
973 }
974
975
976 /**
977  * nes_get_encoded_size
978  */
979 static inline u8 nes_get_encoded_size(int *size)
980 {
981         u8 encoded_size = 0;
982         if (*size <= 32) {
983                 *size = 32;
984                 encoded_size = 1;
985         } else if (*size <= 128) {
986                 *size = 128;
987                 encoded_size = 2;
988         } else if (*size <= 512) {
989                 *size = 512;
990                 encoded_size = 3;
991         }
992         return (encoded_size);
993 }
994
995
996
997 /**
998  * nes_setup_virt_qp
999  */
1000 static int nes_setup_virt_qp(struct nes_qp *nesqp, struct nes_pbl *nespbl,
1001                 struct nes_vnic *nesvnic, int sq_size, int rq_size)
1002 {
1003         unsigned long flags;
1004         void *mem;
1005         __le64 *pbl = NULL;
1006         __le64 *tpbl;
1007         __le64 *pblbuffer;
1008         struct nes_device *nesdev = nesvnic->nesdev;
1009         struct nes_adapter *nesadapter = nesdev->nesadapter;
1010         u32 pbl_entries;
1011         u8 rq_pbl_entries;
1012         u8 sq_pbl_entries;
1013
1014         pbl_entries = nespbl->pbl_size >> 3;
1015         nes_debug(NES_DBG_QP, "Userspace PBL, pbl_size=%u, pbl_entries = %d pbl_vbase=%p, pbl_pbase=%lx\n",
1016                         nespbl->pbl_size, pbl_entries,
1017                         (void *)nespbl->pbl_vbase,
1018                         (unsigned long) nespbl->pbl_pbase);
1019         pbl = (__le64 *) nespbl->pbl_vbase; /* points to first pbl entry */
1020         /* now lets set the sq_vbase as well as rq_vbase addrs we will assign */
1021         /* the first pbl to be fro the rq_vbase... */
1022         rq_pbl_entries = (rq_size * sizeof(struct nes_hw_qp_wqe)) >> 12;
1023         sq_pbl_entries = (sq_size * sizeof(struct nes_hw_qp_wqe)) >> 12;
1024         nesqp->hwqp.sq_pbase = (le32_to_cpu(((__le32 *)pbl)[0])) | ((u64)((le32_to_cpu(((__le32 *)pbl)[1]))) << 32);
1025         if (!nespbl->page) {
1026                 nes_debug(NES_DBG_QP, "QP nespbl->page is NULL \n");
1027                 kfree(nespbl);
1028                 return -ENOMEM;
1029         }
1030
1031         nesqp->hwqp.sq_vbase = kmap(nespbl->page);
1032         nesqp->page = nespbl->page;
1033         if (!nesqp->hwqp.sq_vbase) {
1034                 nes_debug(NES_DBG_QP, "QP sq_vbase kmap failed\n");
1035                 kfree(nespbl);
1036                 return -ENOMEM;
1037         }
1038
1039         /* Now to get to sq.. we need to calculate how many */
1040         /* PBL entries were used by the rq.. */
1041         pbl += sq_pbl_entries;
1042         nesqp->hwqp.rq_pbase = (le32_to_cpu(((__le32 *)pbl)[0])) | ((u64)((le32_to_cpu(((__le32 *)pbl)[1]))) << 32);
1043         /* nesqp->hwqp.rq_vbase = bus_to_virt(*pbl); */
1044         /*nesqp->hwqp.rq_vbase = phys_to_virt(*pbl); */
1045
1046         nes_debug(NES_DBG_QP, "QP sq_vbase= %p sq_pbase=%lx rq_vbase=%p rq_pbase=%lx\n",
1047                   nesqp->hwqp.sq_vbase, (unsigned long) nesqp->hwqp.sq_pbase,
1048                   nesqp->hwqp.rq_vbase, (unsigned long) nesqp->hwqp.rq_pbase);
1049         spin_lock_irqsave(&nesadapter->pbl_lock, flags);
1050         if (!nesadapter->free_256pbl) {
1051                 pci_free_consistent(nesdev->pcidev, nespbl->pbl_size, nespbl->pbl_vbase,
1052                                 nespbl->pbl_pbase);
1053                 spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
1054                 kunmap(nesqp->page);
1055                 kfree(nespbl);
1056                 return -ENOMEM;
1057         }
1058         nesadapter->free_256pbl--;
1059         spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
1060
1061         nesqp->pbl_vbase = pci_alloc_consistent(nesdev->pcidev, 256, &nesqp->pbl_pbase);
1062         pblbuffer = nesqp->pbl_vbase;
1063         if (!nesqp->pbl_vbase) {
1064                 /* memory allocated during nes_reg_user_mr() */
1065                 pci_free_consistent(nesdev->pcidev, nespbl->pbl_size, nespbl->pbl_vbase,
1066                                     nespbl->pbl_pbase);
1067                 kfree(nespbl);
1068                 spin_lock_irqsave(&nesadapter->pbl_lock, flags);
1069                 nesadapter->free_256pbl++;
1070                 spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
1071                 kunmap(nesqp->page);
1072                 return -ENOMEM;
1073         }
1074         memset(nesqp->pbl_vbase, 0, 256);
1075         /* fill in the page address in the pbl buffer.. */
1076         tpbl = pblbuffer + 16;
1077         pbl = (__le64 *)nespbl->pbl_vbase;
1078         while (sq_pbl_entries--)
1079                 *tpbl++ = *pbl++;
1080         tpbl = pblbuffer;
1081         while (rq_pbl_entries--)
1082                 *tpbl++ = *pbl++;
1083
1084         /* done with memory allocated during nes_reg_user_mr() */
1085         pci_free_consistent(nesdev->pcidev, nespbl->pbl_size, nespbl->pbl_vbase,
1086                             nespbl->pbl_pbase);
1087         kfree(nespbl);
1088
1089         nesqp->qp_mem_size =
1090                         max((u32)sizeof(struct nes_qp_context), ((u32)256)) + 256;     /* this is Q2 */
1091         /* Round up to a multiple of a page */
1092         nesqp->qp_mem_size += PAGE_SIZE - 1;
1093         nesqp->qp_mem_size &= ~(PAGE_SIZE - 1);
1094
1095         mem = pci_alloc_consistent(nesdev->pcidev, nesqp->qp_mem_size,
1096                         &nesqp->hwqp.q2_pbase);
1097
1098         if (!mem) {
1099                 pci_free_consistent(nesdev->pcidev, 256, nesqp->pbl_vbase, nesqp->pbl_pbase);
1100                 nesqp->pbl_vbase = NULL;
1101                 spin_lock_irqsave(&nesadapter->pbl_lock, flags);
1102                 nesadapter->free_256pbl++;
1103                 spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
1104                 kunmap(nesqp->page);
1105                 return -ENOMEM;
1106         }
1107         nesqp->hwqp.q2_vbase = mem;
1108         mem += 256;
1109         memset(nesqp->hwqp.q2_vbase, 0, 256);
1110         nesqp->nesqp_context = mem;
1111         memset(nesqp->nesqp_context, 0, sizeof(*nesqp->nesqp_context));
1112         nesqp->nesqp_context_pbase = nesqp->hwqp.q2_pbase + 256;
1113
1114         return 0;
1115 }
1116
1117
1118 /**
1119  * nes_setup_mmap_qp
1120  */
1121 static int nes_setup_mmap_qp(struct nes_qp *nesqp, struct nes_vnic *nesvnic,
1122                 int sq_size, int rq_size)
1123 {
1124         void *mem;
1125         struct nes_device *nesdev = nesvnic->nesdev;
1126
1127         nesqp->qp_mem_size = (sizeof(struct nes_hw_qp_wqe) * sq_size) +
1128                         (sizeof(struct nes_hw_qp_wqe) * rq_size) +
1129                         max((u32)sizeof(struct nes_qp_context), ((u32)256)) +
1130                         256; /* this is Q2 */
1131         /* Round up to a multiple of a page */
1132         nesqp->qp_mem_size += PAGE_SIZE - 1;
1133         nesqp->qp_mem_size &= ~(PAGE_SIZE - 1);
1134
1135         mem = pci_alloc_consistent(nesdev->pcidev, nesqp->qp_mem_size,
1136                         &nesqp->hwqp.sq_pbase);
1137         if (!mem)
1138                 return -ENOMEM;
1139         nes_debug(NES_DBG_QP, "PCI consistent memory for "
1140                         "host descriptor rings located @ %p (pa = 0x%08lX.) size = %u.\n",
1141                         mem, (unsigned long)nesqp->hwqp.sq_pbase, nesqp->qp_mem_size);
1142
1143         memset(mem, 0, nesqp->qp_mem_size);
1144
1145         nesqp->hwqp.sq_vbase = mem;
1146         mem += sizeof(struct nes_hw_qp_wqe) * sq_size;
1147
1148         nesqp->hwqp.rq_vbase = mem;
1149         nesqp->hwqp.rq_pbase = nesqp->hwqp.sq_pbase +
1150                         sizeof(struct nes_hw_qp_wqe) * sq_size;
1151         mem += sizeof(struct nes_hw_qp_wqe) * rq_size;
1152
1153         nesqp->hwqp.q2_vbase = mem;
1154         nesqp->hwqp.q2_pbase = nesqp->hwqp.rq_pbase +
1155                         sizeof(struct nes_hw_qp_wqe) * rq_size;
1156         mem += 256;
1157         memset(nesqp->hwqp.q2_vbase, 0, 256);
1158
1159         nesqp->nesqp_context = mem;
1160         nesqp->nesqp_context_pbase = nesqp->hwqp.q2_pbase + 256;
1161         memset(nesqp->nesqp_context, 0, sizeof(*nesqp->nesqp_context));
1162         return 0;
1163 }
1164
1165
1166 /**
1167  * nes_free_qp_mem() is to free up the qp's pci_alloc_consistent() memory.
1168  */
1169 static inline void nes_free_qp_mem(struct nes_device *nesdev,
1170                 struct nes_qp *nesqp, int virt_wqs)
1171 {
1172         unsigned long flags;
1173         struct nes_adapter *nesadapter = nesdev->nesadapter;
1174         if (!virt_wqs) {
1175                 pci_free_consistent(nesdev->pcidev, nesqp->qp_mem_size,
1176                                 nesqp->hwqp.sq_vbase, nesqp->hwqp.sq_pbase);
1177         }else {
1178                 spin_lock_irqsave(&nesadapter->pbl_lock, flags);
1179                 nesadapter->free_256pbl++;
1180                 spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
1181                 pci_free_consistent(nesdev->pcidev, nesqp->qp_mem_size, nesqp->hwqp.q2_vbase, nesqp->hwqp.q2_pbase);
1182                 pci_free_consistent(nesdev->pcidev, 256, nesqp->pbl_vbase, nesqp->pbl_pbase );
1183                 nesqp->pbl_vbase = NULL;
1184                 kunmap(nesqp->page);
1185         }
1186 }
1187
1188
1189 /**
1190  * nes_create_qp
1191  */
1192 static struct ib_qp *nes_create_qp(struct ib_pd *ibpd,
1193                 struct ib_qp_init_attr *init_attr, struct ib_udata *udata)
1194 {
1195         u64 u64temp= 0;
1196         u64 u64nesqp = 0;
1197         struct nes_pd *nespd = to_nespd(ibpd);
1198         struct nes_vnic *nesvnic = to_nesvnic(ibpd->device);
1199         struct nes_device *nesdev = nesvnic->nesdev;
1200         struct nes_adapter *nesadapter = nesdev->nesadapter;
1201         struct nes_qp *nesqp;
1202         struct nes_cq *nescq;
1203         struct nes_ucontext *nes_ucontext;
1204         struct nes_hw_cqp_wqe *cqp_wqe;
1205         struct nes_cqp_request *cqp_request;
1206         struct nes_create_qp_req req;
1207         struct nes_create_qp_resp uresp;
1208         struct nes_pbl  *nespbl = NULL;
1209         u32 qp_num = 0;
1210         u32 opcode = 0;
1211         /* u32 counter = 0; */
1212         void *mem;
1213         unsigned long flags;
1214         int ret;
1215         int err;
1216         int virt_wqs = 0;
1217         int sq_size;
1218         int rq_size;
1219         u8 sq_encoded_size;
1220         u8 rq_encoded_size;
1221         /* int counter; */
1222
1223         if (init_attr->create_flags)
1224                 return ERR_PTR(-EINVAL);
1225
1226         atomic_inc(&qps_created);
1227         switch (init_attr->qp_type) {
1228                 case IB_QPT_RC:
1229                         if (nes_drv_opt & NES_DRV_OPT_NO_INLINE_DATA) {
1230                                 init_attr->cap.max_inline_data = 0;
1231                         } else {
1232                                 init_attr->cap.max_inline_data = 64;
1233                         }
1234                         sq_size = init_attr->cap.max_send_wr;
1235                         rq_size = init_attr->cap.max_recv_wr;
1236
1237                         /* check if the encoded sizes are OK or not... */
1238                         sq_encoded_size = nes_get_encoded_size(&sq_size);
1239                         rq_encoded_size = nes_get_encoded_size(&rq_size);
1240
1241                         if ((!sq_encoded_size) || (!rq_encoded_size)) {
1242                                 nes_debug(NES_DBG_QP, "ERROR bad rq (%u) or sq (%u) size\n",
1243                                                 rq_size, sq_size);
1244                                 return ERR_PTR(-EINVAL);
1245                         }
1246
1247                         init_attr->cap.max_send_wr = sq_size -2;
1248                         init_attr->cap.max_recv_wr = rq_size -1;
1249                         nes_debug(NES_DBG_QP, "RQ size=%u, SQ Size=%u\n", rq_size, sq_size);
1250
1251                         ret = nes_alloc_resource(nesadapter, nesadapter->allocated_qps,
1252                                         nesadapter->max_qp, &qp_num, &nesadapter->next_qp);
1253                         if (ret) {
1254                                 return ERR_PTR(ret);
1255                         }
1256
1257                         /* Need 512 (actually now 1024) byte alignment on this structure */
1258                         mem = kzalloc(sizeof(*nesqp)+NES_SW_CONTEXT_ALIGN-1, GFP_KERNEL);
1259                         if (!mem) {
1260                                 nes_free_resource(nesadapter, nesadapter->allocated_qps, qp_num);
1261                                 nes_debug(NES_DBG_QP, "Unable to allocate QP\n");
1262                                 return ERR_PTR(-ENOMEM);
1263                         }
1264                         u64nesqp = (unsigned long)mem;
1265                         u64nesqp += ((u64)NES_SW_CONTEXT_ALIGN) - 1;
1266                         u64temp = ((u64)NES_SW_CONTEXT_ALIGN) - 1;
1267                         u64nesqp &= ~u64temp;
1268                         nesqp = (struct nes_qp *)(unsigned long)u64nesqp;
1269                         /* nes_debug(NES_DBG_QP, "nesqp=%p, allocated buffer=%p.  Rounded to closest %u\n",
1270                                         nesqp, mem, NES_SW_CONTEXT_ALIGN); */
1271                         nesqp->allocated_buffer = mem;
1272
1273                         if (udata) {
1274                                 if (ib_copy_from_udata(&req, udata, sizeof(struct nes_create_qp_req))) {
1275                                         nes_free_resource(nesadapter, nesadapter->allocated_qps, qp_num);
1276                                         kfree(nesqp->allocated_buffer);
1277                                         nes_debug(NES_DBG_QP, "ib_copy_from_udata() Failed \n");
1278                                         return NULL;
1279                                 }
1280                                 if (req.user_wqe_buffers) {
1281                                         virt_wqs = 1;
1282                                 }
1283                                 if ((ibpd->uobject) && (ibpd->uobject->context)) {
1284                                         nesqp->user_mode = 1;
1285                                         nes_ucontext = to_nesucontext(ibpd->uobject->context);
1286                                         if (virt_wqs) {
1287                                                 err = 1;
1288                                                 list_for_each_entry(nespbl, &nes_ucontext->qp_reg_mem_list, list) {
1289                                                         if (nespbl->user_base == (unsigned long )req.user_wqe_buffers) {
1290                                                                 list_del(&nespbl->list);
1291                                                                 err = 0;
1292                                                                 nes_debug(NES_DBG_QP, "Found PBL for virtual QP. nespbl=%p. user_base=0x%lx\n",
1293                                                                           nespbl, nespbl->user_base);
1294                                                                 break;
1295                                                         }
1296                                                 }
1297                                                 if (err) {
1298                                                         nes_debug(NES_DBG_QP, "Didn't Find PBL for virtual QP. address = %llx.\n",
1299                                                                   (long long unsigned int)req.user_wqe_buffers);
1300                                                         nes_free_resource(nesadapter, nesadapter->allocated_qps, qp_num);
1301                                                         kfree(nesqp->allocated_buffer);
1302                                                         return ERR_PTR(-EFAULT);
1303                                                 }
1304                                         }
1305
1306                                         nes_ucontext = to_nesucontext(ibpd->uobject->context);
1307                                         nesqp->mmap_sq_db_index =
1308                                                 find_next_zero_bit(nes_ucontext->allocated_wqs,
1309                                                                    NES_MAX_USER_WQ_REGIONS, nes_ucontext->first_free_wq);
1310                                         /* nes_debug(NES_DBG_QP, "find_first_zero_biton wqs returned %u\n",
1311                                                         nespd->mmap_db_index); */
1312                                         if (nesqp->mmap_sq_db_index >= NES_MAX_USER_WQ_REGIONS) {
1313                                                 nes_debug(NES_DBG_QP,
1314                                                           "db index > max user regions, failing create QP\n");
1315                                                 nes_free_resource(nesadapter, nesadapter->allocated_qps, qp_num);
1316                                                 if (virt_wqs) {
1317                                                         pci_free_consistent(nesdev->pcidev, nespbl->pbl_size, nespbl->pbl_vbase,
1318                                                                             nespbl->pbl_pbase);
1319                                                         kfree(nespbl);
1320                                                 }
1321                                                 kfree(nesqp->allocated_buffer);
1322                                                 return ERR_PTR(-ENOMEM);
1323                                         }
1324                                         set_bit(nesqp->mmap_sq_db_index, nes_ucontext->allocated_wqs);
1325                                         nes_ucontext->mmap_nesqp[nesqp->mmap_sq_db_index] = nesqp;
1326                                         nes_ucontext->first_free_wq = nesqp->mmap_sq_db_index + 1;
1327                                 } else {
1328                                         nes_free_resource(nesadapter, nesadapter->allocated_qps, qp_num);
1329                                         kfree(nesqp->allocated_buffer);
1330                                         return ERR_PTR(-EFAULT);
1331                                 }
1332                         }
1333                         err = (!virt_wqs) ? nes_setup_mmap_qp(nesqp, nesvnic, sq_size, rq_size) :
1334                                         nes_setup_virt_qp(nesqp, nespbl, nesvnic, sq_size, rq_size);
1335                         if (err) {
1336                                 nes_debug(NES_DBG_QP,
1337                                           "error geting qp mem code = %d\n", err);
1338                                 nes_free_resource(nesadapter, nesadapter->allocated_qps, qp_num);
1339                                 kfree(nesqp->allocated_buffer);
1340                                 return ERR_PTR(-ENOMEM);
1341                         }
1342
1343                         nesqp->hwqp.sq_size = sq_size;
1344                         nesqp->hwqp.sq_encoded_size = sq_encoded_size;
1345                         nesqp->hwqp.sq_head = 1;
1346                         nesqp->hwqp.rq_size = rq_size;
1347                         nesqp->hwqp.rq_encoded_size = rq_encoded_size;
1348                         /* nes_debug(NES_DBG_QP, "nesqp->nesqp_context_pbase = %p\n",
1349                                         (void *)nesqp->nesqp_context_pbase);
1350                         */
1351                         nesqp->hwqp.qp_id = qp_num;
1352                         nesqp->ibqp.qp_num = nesqp->hwqp.qp_id;
1353                         nesqp->nespd = nespd;
1354
1355                         nescq = to_nescq(init_attr->send_cq);
1356                         nesqp->nesscq = nescq;
1357                         nescq = to_nescq(init_attr->recv_cq);
1358                         nesqp->nesrcq = nescq;
1359
1360                         nesqp->nesqp_context->misc |= cpu_to_le32((u32)PCI_FUNC(nesdev->pcidev->devfn) <<
1361                                         NES_QPCONTEXT_MISC_PCI_FCN_SHIFT);
1362                         nesqp->nesqp_context->misc |= cpu_to_le32((u32)nesqp->hwqp.rq_encoded_size <<
1363                                         NES_QPCONTEXT_MISC_RQ_SIZE_SHIFT);
1364                         nesqp->nesqp_context->misc |= cpu_to_le32((u32)nesqp->hwqp.sq_encoded_size <<
1365                                         NES_QPCONTEXT_MISC_SQ_SIZE_SHIFT);
1366                                 nesqp->nesqp_context->misc |= cpu_to_le32(NES_QPCONTEXT_MISC_PRIV_EN);
1367                                 nesqp->nesqp_context->misc |= cpu_to_le32(NES_QPCONTEXT_MISC_FAST_REGISTER_EN);
1368                         nesqp->nesqp_context->cqs = cpu_to_le32(nesqp->nesscq->hw_cq.cq_number +
1369                                         ((u32)nesqp->nesrcq->hw_cq.cq_number << 16));
1370                         u64temp = (u64)nesqp->hwqp.sq_pbase;
1371                         nesqp->nesqp_context->sq_addr_low = cpu_to_le32((u32)u64temp);
1372                         nesqp->nesqp_context->sq_addr_high = cpu_to_le32((u32)(u64temp >> 32));
1373
1374
1375                         if (!virt_wqs) {
1376                                 u64temp = (u64)nesqp->hwqp.sq_pbase;
1377                                 nesqp->nesqp_context->sq_addr_low = cpu_to_le32((u32)u64temp);
1378                                 nesqp->nesqp_context->sq_addr_high = cpu_to_le32((u32)(u64temp >> 32));
1379                                 u64temp = (u64)nesqp->hwqp.rq_pbase;
1380                                 nesqp->nesqp_context->rq_addr_low = cpu_to_le32((u32)u64temp);
1381                                 nesqp->nesqp_context->rq_addr_high = cpu_to_le32((u32)(u64temp >> 32));
1382                         } else {
1383                                 u64temp = (u64)nesqp->pbl_pbase;
1384                                 nesqp->nesqp_context->rq_addr_low = cpu_to_le32((u32)u64temp);
1385                                 nesqp->nesqp_context->rq_addr_high = cpu_to_le32((u32)(u64temp >> 32));
1386                         }
1387
1388                         /* nes_debug(NES_DBG_QP, "next_qp_nic_index=%u, using nic_index=%d\n",
1389                                         nesvnic->next_qp_nic_index,
1390                                         nesvnic->qp_nic_index[nesvnic->next_qp_nic_index]); */
1391                         spin_lock_irqsave(&nesdev->cqp.lock, flags);
1392                         nesqp->nesqp_context->misc2 |= cpu_to_le32(
1393                                         (u32)nesvnic->qp_nic_index[nesvnic->next_qp_nic_index] <<
1394                                         NES_QPCONTEXT_MISC2_NIC_INDEX_SHIFT);
1395                         nesvnic->next_qp_nic_index++;
1396                         if ((nesvnic->next_qp_nic_index > 3) ||
1397                                         (nesvnic->qp_nic_index[nesvnic->next_qp_nic_index] == 0xf)) {
1398                                 nesvnic->next_qp_nic_index = 0;
1399                         }
1400                         spin_unlock_irqrestore(&nesdev->cqp.lock, flags);
1401
1402                         nesqp->nesqp_context->pd_index_wscale |= cpu_to_le32((u32)nesqp->nespd->pd_id << 16);
1403                         u64temp = (u64)nesqp->hwqp.q2_pbase;
1404                         nesqp->nesqp_context->q2_addr_low = cpu_to_le32((u32)u64temp);
1405                         nesqp->nesqp_context->q2_addr_high = cpu_to_le32((u32)(u64temp >> 32));
1406                         nesqp->nesqp_context->aeq_token_low =  cpu_to_le32((u32)((unsigned long)(nesqp)));
1407                         nesqp->nesqp_context->aeq_token_high =  cpu_to_le32((u32)(upper_32_bits((unsigned long)(nesqp))));
1408                         nesqp->nesqp_context->ird_ord_sizes = cpu_to_le32(NES_QPCONTEXT_ORDIRD_ALSMM |
1409                                         ((((u32)nesadapter->max_irrq_wr) <<
1410                                         NES_QPCONTEXT_ORDIRD_IRDSIZE_SHIFT) & NES_QPCONTEXT_ORDIRD_IRDSIZE_MASK));
1411                         if (disable_mpa_crc) {
1412                                 nes_debug(NES_DBG_QP, "Disabling MPA crc checking due to module option.\n");
1413                                 nesqp->nesqp_context->ird_ord_sizes |= cpu_to_le32(NES_QPCONTEXT_ORDIRD_RNMC);
1414                         }
1415
1416
1417                         /* Create the QP */
1418                         cqp_request = nes_get_cqp_request(nesdev);
1419                         if (cqp_request == NULL) {
1420                                 nes_debug(NES_DBG_QP, "Failed to get a cqp_request\n");
1421                                 nes_free_resource(nesadapter, nesadapter->allocated_qps, qp_num);
1422                                 nes_free_qp_mem(nesdev, nesqp,virt_wqs);
1423                                 kfree(nesqp->allocated_buffer);
1424                                 return ERR_PTR(-ENOMEM);
1425                         }
1426                         cqp_request->waiting = 1;
1427                         cqp_wqe = &cqp_request->cqp_wqe;
1428
1429                         if (!virt_wqs) {
1430                                 opcode = NES_CQP_CREATE_QP | NES_CQP_QP_TYPE_IWARP |
1431                                         NES_CQP_QP_IWARP_STATE_IDLE;
1432                         } else {
1433                                 opcode = NES_CQP_CREATE_QP | NES_CQP_QP_TYPE_IWARP | NES_CQP_QP_VIRT_WQS |
1434                                         NES_CQP_QP_IWARP_STATE_IDLE;
1435                         }
1436                         opcode |= NES_CQP_QP_CQS_VALID;
1437                         nes_fill_init_cqp_wqe(cqp_wqe, nesdev);
1438                         set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_WQE_OPCODE_IDX, opcode);
1439                         set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_WQE_ID_IDX, nesqp->hwqp.qp_id);
1440
1441                         u64temp = (u64)nesqp->nesqp_context_pbase;
1442                         set_wqe_64bit_value(cqp_wqe->wqe_words, NES_CQP_QP_WQE_CONTEXT_LOW_IDX, u64temp);
1443
1444                         atomic_set(&cqp_request->refcount, 2);
1445                         nes_post_cqp_request(nesdev, cqp_request);
1446
1447                         /* Wait for CQP */
1448                         nes_debug(NES_DBG_QP, "Waiting for create iWARP QP%u to complete.\n",
1449                                         nesqp->hwqp.qp_id);
1450                         ret = wait_event_timeout(cqp_request->waitq,
1451                                         (cqp_request->request_done != 0), NES_EVENT_TIMEOUT);
1452                         nes_debug(NES_DBG_QP, "Create iwarp QP%u completed, wait_event_timeout ret=%u,"
1453                                         " nesdev->cqp_head = %u, nesdev->cqp.sq_tail = %u,"
1454                                         " CQP Major:Minor codes = 0x%04X:0x%04X.\n",
1455                                         nesqp->hwqp.qp_id, ret, nesdev->cqp.sq_head, nesdev->cqp.sq_tail,
1456                                         cqp_request->major_code, cqp_request->minor_code);
1457                         if ((!ret) || (cqp_request->major_code)) {
1458                                 nes_put_cqp_request(nesdev, cqp_request);
1459                                 nes_free_resource(nesadapter, nesadapter->allocated_qps, qp_num);
1460                                 nes_free_qp_mem(nesdev, nesqp,virt_wqs);
1461                                 kfree(nesqp->allocated_buffer);
1462                                 if (!ret) {
1463                                         return ERR_PTR(-ETIME);
1464                                 } else {
1465                                         return ERR_PTR(-EIO);
1466                                 }
1467                         }
1468
1469                         nes_put_cqp_request(nesdev, cqp_request);
1470
1471                         if (ibpd->uobject) {
1472                                 uresp.mmap_sq_db_index = nesqp->mmap_sq_db_index;
1473                                 uresp.actual_sq_size = sq_size;
1474                                 uresp.actual_rq_size = rq_size;
1475                                 uresp.qp_id = nesqp->hwqp.qp_id;
1476                                 uresp.nes_drv_opt = nes_drv_opt;
1477                                 if (ib_copy_to_udata(udata, &uresp, sizeof uresp)) {
1478                                         nes_free_resource(nesadapter, nesadapter->allocated_qps, qp_num);
1479                                         nes_free_qp_mem(nesdev, nesqp,virt_wqs);
1480                                         kfree(nesqp->allocated_buffer);
1481                                         return ERR_PTR(-EFAULT);
1482                                 }
1483                         }
1484
1485                         nes_debug(NES_DBG_QP, "QP%u structure located @%p.Size = %u.\n",
1486                                         nesqp->hwqp.qp_id, nesqp, (u32)sizeof(*nesqp));
1487                         spin_lock_init(&nesqp->lock);
1488                         init_waitqueue_head(&nesqp->state_waitq);
1489                         init_waitqueue_head(&nesqp->kick_waitq);
1490                         nes_add_ref(&nesqp->ibqp);
1491                         break;
1492                 default:
1493                         nes_debug(NES_DBG_QP, "Invalid QP type: %d\n", init_attr->qp_type);
1494                         return ERR_PTR(-EINVAL);
1495         }
1496
1497         /* update the QP table */
1498         nesdev->nesadapter->qp_table[nesqp->hwqp.qp_id-NES_FIRST_QPN] = nesqp;
1499         nes_debug(NES_DBG_QP, "netdev refcnt=%u\n",
1500                         atomic_read(&nesvnic->netdev->refcnt));
1501
1502         return &nesqp->ibqp;
1503 }
1504
1505
1506 /**
1507  * nes_destroy_qp
1508  */
1509 static int nes_destroy_qp(struct ib_qp *ibqp)
1510 {
1511         struct nes_qp *nesqp = to_nesqp(ibqp);
1512         /* struct nes_vnic *nesvnic = to_nesvnic(ibqp->device); */
1513         struct nes_ucontext *nes_ucontext;
1514         struct ib_qp_attr attr;
1515         struct iw_cm_id *cm_id;
1516         struct iw_cm_event cm_event;
1517         int ret;
1518
1519         atomic_inc(&sw_qps_destroyed);
1520         nesqp->destroyed = 1;
1521
1522         /* Blow away the connection if it exists. */
1523         if (nesqp->ibqp_state >= IB_QPS_INIT && nesqp->ibqp_state <= IB_QPS_RTS) {
1524                 /* if (nesqp->ibqp_state == IB_QPS_RTS) { */
1525                 attr.qp_state = IB_QPS_ERR;
1526                 nes_modify_qp(&nesqp->ibqp, &attr, IB_QP_STATE, NULL);
1527         }
1528
1529         if (((nesqp->ibqp_state == IB_QPS_INIT) ||
1530                         (nesqp->ibqp_state == IB_QPS_RTR)) && (nesqp->cm_id)) {
1531                 cm_id = nesqp->cm_id;
1532                 cm_event.event = IW_CM_EVENT_CONNECT_REPLY;
1533                 cm_event.status = IW_CM_EVENT_STATUS_TIMEOUT;
1534                 cm_event.local_addr = cm_id->local_addr;
1535                 cm_event.remote_addr = cm_id->remote_addr;
1536                 cm_event.private_data = NULL;
1537                 cm_event.private_data_len = 0;
1538
1539                 nes_debug(NES_DBG_QP, "Generating a CM Timeout Event for "
1540                                 "QP%u. cm_id = %p, refcount = %u. \n",
1541                                 nesqp->hwqp.qp_id, cm_id, atomic_read(&nesqp->refcount));
1542
1543                 cm_id->rem_ref(cm_id);
1544                 ret = cm_id->event_handler(cm_id, &cm_event);
1545                 if (ret)
1546                         nes_debug(NES_DBG_QP, "OFA CM event_handler returned, ret=%d\n", ret);
1547         }
1548
1549
1550         if (nesqp->user_mode) {
1551                 if ((ibqp->uobject)&&(ibqp->uobject->context)) {
1552                         nes_ucontext = to_nesucontext(ibqp->uobject->context);
1553                         clear_bit(nesqp->mmap_sq_db_index, nes_ucontext->allocated_wqs);
1554                         nes_ucontext->mmap_nesqp[nesqp->mmap_sq_db_index] = NULL;
1555                         if (nes_ucontext->first_free_wq > nesqp->mmap_sq_db_index) {
1556                                 nes_ucontext->first_free_wq = nesqp->mmap_sq_db_index;
1557                         }
1558                 }
1559                 if (nesqp->pbl_pbase)
1560                         kunmap(nesqp->page);
1561         }
1562
1563         nes_rem_ref(&nesqp->ibqp);
1564         return 0;
1565 }
1566
1567
1568 /**
1569  * nes_create_cq
1570  */
1571 static struct ib_cq *nes_create_cq(struct ib_device *ibdev, int entries,
1572                 int comp_vector,
1573                 struct ib_ucontext *context, struct ib_udata *udata)
1574 {
1575         u64 u64temp;
1576         struct nes_vnic *nesvnic = to_nesvnic(ibdev);
1577         struct nes_device *nesdev = nesvnic->nesdev;
1578         struct nes_adapter *nesadapter = nesdev->nesadapter;
1579         struct nes_cq *nescq;
1580         struct nes_ucontext *nes_ucontext = NULL;
1581         struct nes_cqp_request *cqp_request;
1582         void *mem = NULL;
1583         struct nes_hw_cqp_wqe *cqp_wqe;
1584         struct nes_pbl *nespbl = NULL;
1585         struct nes_create_cq_req req;
1586         struct nes_create_cq_resp resp;
1587         u32 cq_num = 0;
1588         u32 opcode = 0;
1589         u32 pbl_entries = 1;
1590         int err;
1591         unsigned long flags;
1592         int ret;
1593
1594         err = nes_alloc_resource(nesadapter, nesadapter->allocated_cqs,
1595                         nesadapter->max_cq, &cq_num, &nesadapter->next_cq);
1596         if (err) {
1597                 return ERR_PTR(err);
1598         }
1599
1600         nescq = kzalloc(sizeof(struct nes_cq), GFP_KERNEL);
1601         if (!nescq) {
1602                 nes_free_resource(nesadapter, nesadapter->allocated_cqs, cq_num);
1603                 nes_debug(NES_DBG_CQ, "Unable to allocate nes_cq struct\n");
1604                 return ERR_PTR(-ENOMEM);
1605         }
1606
1607         nescq->hw_cq.cq_size = max(entries + 1, 5);
1608         nescq->hw_cq.cq_number = cq_num;
1609         nescq->ibcq.cqe = nescq->hw_cq.cq_size - 1;
1610
1611
1612         if (context) {
1613                 nes_ucontext = to_nesucontext(context);
1614                 if (ib_copy_from_udata(&req, udata, sizeof (struct nes_create_cq_req))) {
1615                         nes_free_resource(nesadapter, nesadapter->allocated_cqs, cq_num);
1616                         kfree(nescq);
1617                         return ERR_PTR(-EFAULT);
1618                 }
1619                 nesvnic->mcrq_ucontext = nes_ucontext;
1620                 nes_ucontext->mcrqf = req.mcrqf;
1621                 if (nes_ucontext->mcrqf) {
1622                         if (nes_ucontext->mcrqf & 0x80000000)
1623                                 nescq->hw_cq.cq_number = nesvnic->nic.qp_id + 28 + 2 * ((nes_ucontext->mcrqf & 0xf) - 1);
1624                         else if (nes_ucontext->mcrqf & 0x40000000)
1625                                 nescq->hw_cq.cq_number = nes_ucontext->mcrqf & 0xffff;
1626                         else
1627                                 nescq->hw_cq.cq_number = nesvnic->mcrq_qp_id + nes_ucontext->mcrqf-1;
1628                         nes_free_resource(nesadapter, nesadapter->allocated_cqs, cq_num);
1629                 }
1630                 nes_debug(NES_DBG_CQ, "CQ Virtual Address = %08lX, size = %u.\n",
1631                                 (unsigned long)req.user_cq_buffer, entries);
1632                 err = 1;
1633                 list_for_each_entry(nespbl, &nes_ucontext->cq_reg_mem_list, list) {
1634                         if (nespbl->user_base == (unsigned long )req.user_cq_buffer) {
1635                                 list_del(&nespbl->list);
1636                                 err = 0;
1637                                 nes_debug(NES_DBG_CQ, "Found PBL for virtual CQ. nespbl=%p.\n",
1638                                                 nespbl);
1639                                 break;
1640                         }
1641                 }
1642                 if (err) {
1643                         nes_free_resource(nesadapter, nesadapter->allocated_cqs, cq_num);
1644                         kfree(nescq);
1645                         return ERR_PTR(-EFAULT);
1646                 }
1647
1648                 pbl_entries = nespbl->pbl_size >> 3;
1649                 nescq->cq_mem_size = 0;
1650         } else {
1651                 nescq->cq_mem_size = nescq->hw_cq.cq_size * sizeof(struct nes_hw_cqe);
1652                 nes_debug(NES_DBG_CQ, "Attempting to allocate pci memory (%u entries, %u bytes) for CQ%u.\n",
1653                                 entries, nescq->cq_mem_size, nescq->hw_cq.cq_number);
1654
1655                 /* allocate the physical buffer space */
1656                 mem = pci_alloc_consistent(nesdev->pcidev, nescq->cq_mem_size,
1657                                 &nescq->hw_cq.cq_pbase);
1658                 if (!mem) {
1659                         printk(KERN_ERR PFX "Unable to allocate pci memory for cq\n");
1660                         nes_free_resource(nesadapter, nesadapter->allocated_cqs, cq_num);
1661                         kfree(nescq);
1662                         return ERR_PTR(-ENOMEM);
1663                 }
1664
1665                 memset(mem, 0, nescq->cq_mem_size);
1666                 nescq->hw_cq.cq_vbase = mem;
1667                 nescq->hw_cq.cq_head = 0;
1668                 nes_debug(NES_DBG_CQ, "CQ%u virtual address @ %p, phys = 0x%08X\n",
1669                                 nescq->hw_cq.cq_number, nescq->hw_cq.cq_vbase,
1670                                 (u32)nescq->hw_cq.cq_pbase);
1671         }
1672
1673         nescq->hw_cq.ce_handler = nes_iwarp_ce_handler;
1674         spin_lock_init(&nescq->lock);
1675
1676         /* send CreateCQ request to CQP */
1677         cqp_request = nes_get_cqp_request(nesdev);
1678         if (cqp_request == NULL) {
1679                 nes_debug(NES_DBG_CQ, "Failed to get a cqp_request.\n");
1680                 if (!context)
1681                         pci_free_consistent(nesdev->pcidev, nescq->cq_mem_size, mem,
1682                                         nescq->hw_cq.cq_pbase);
1683                 nes_free_resource(nesadapter, nesadapter->allocated_cqs, cq_num);
1684                 kfree(nescq);
1685                 return ERR_PTR(-ENOMEM);
1686         }
1687         cqp_request->waiting = 1;
1688         cqp_wqe = &cqp_request->cqp_wqe;
1689
1690         opcode = NES_CQP_CREATE_CQ | NES_CQP_CQ_CEQ_VALID |
1691                         NES_CQP_CQ_CHK_OVERFLOW |
1692                         NES_CQP_CQ_CEQE_MASK | ((u32)nescq->hw_cq.cq_size << 16);
1693
1694         spin_lock_irqsave(&nesadapter->pbl_lock, flags);
1695
1696         if (pbl_entries != 1) {
1697                 if (pbl_entries > 32) {
1698                         /* use 4k pbl */
1699                         nes_debug(NES_DBG_CQ, "pbl_entries=%u, use a 4k PBL\n", pbl_entries);
1700                         if (nesadapter->free_4kpbl == 0) {
1701                                 spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
1702                                 nes_free_cqp_request(nesdev, cqp_request);
1703                                 if (!context)
1704                                         pci_free_consistent(nesdev->pcidev, nescq->cq_mem_size, mem,
1705                                                         nescq->hw_cq.cq_pbase);
1706                                 nes_free_resource(nesadapter, nesadapter->allocated_cqs, cq_num);
1707                                 kfree(nescq);
1708                                 return ERR_PTR(-ENOMEM);
1709                         } else {
1710                                 opcode |= (NES_CQP_CQ_VIRT | NES_CQP_CQ_4KB_CHUNK);
1711                                 nescq->virtual_cq = 2;
1712                                 nesadapter->free_4kpbl--;
1713                         }
1714                 } else {
1715                         /* use 256 byte pbl */
1716                         nes_debug(NES_DBG_CQ, "pbl_entries=%u, use a 256 byte PBL\n", pbl_entries);
1717                         if (nesadapter->free_256pbl == 0) {
1718                                 spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
1719                                 nes_free_cqp_request(nesdev, cqp_request);
1720                                 if (!context)
1721                                         pci_free_consistent(nesdev->pcidev, nescq->cq_mem_size, mem,
1722                                                         nescq->hw_cq.cq_pbase);
1723                                 nes_free_resource(nesadapter, nesadapter->allocated_cqs, cq_num);
1724                                 kfree(nescq);
1725                                 return ERR_PTR(-ENOMEM);
1726                         } else {
1727                                 opcode |= NES_CQP_CQ_VIRT;
1728                                 nescq->virtual_cq = 1;
1729                                 nesadapter->free_256pbl--;
1730                         }
1731                 }
1732         }
1733
1734         spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
1735
1736         nes_fill_init_cqp_wqe(cqp_wqe, nesdev);
1737         set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_WQE_OPCODE_IDX, opcode);
1738         set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_WQE_ID_IDX,
1739                         (nescq->hw_cq.cq_number | ((u32)nesdev->ceq_index << 16)));
1740
1741         if (context) {
1742                 if (pbl_entries != 1)
1743                         u64temp = (u64)nespbl->pbl_pbase;
1744                 else
1745                         u64temp = le64_to_cpu(nespbl->pbl_vbase[0]);
1746                 set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_CQ_WQE_DOORBELL_INDEX_HIGH_IDX,
1747                                 nes_ucontext->mmap_db_index[0]);
1748         } else {
1749                 u64temp = (u64)nescq->hw_cq.cq_pbase;
1750                 cqp_wqe->wqe_words[NES_CQP_CQ_WQE_DOORBELL_INDEX_HIGH_IDX] = 0;
1751         }
1752         set_wqe_64bit_value(cqp_wqe->wqe_words, NES_CQP_CQ_WQE_PBL_LOW_IDX, u64temp);
1753         cqp_wqe->wqe_words[NES_CQP_CQ_WQE_CQ_CONTEXT_HIGH_IDX] = 0;
1754         u64temp = (u64)(unsigned long)&nescq->hw_cq;
1755         cqp_wqe->wqe_words[NES_CQP_CQ_WQE_CQ_CONTEXT_LOW_IDX] =
1756                         cpu_to_le32((u32)(u64temp >> 1));
1757         cqp_wqe->wqe_words[NES_CQP_CQ_WQE_CQ_CONTEXT_HIGH_IDX] =
1758                         cpu_to_le32(((u32)((u64temp) >> 33)) & 0x7FFFFFFF);
1759
1760         atomic_set(&cqp_request->refcount, 2);
1761         nes_post_cqp_request(nesdev, cqp_request);
1762
1763         /* Wait for CQP */
1764         nes_debug(NES_DBG_CQ, "Waiting for create iWARP CQ%u to complete.\n",
1765                         nescq->hw_cq.cq_number);
1766         ret = wait_event_timeout(cqp_request->waitq, (0 != cqp_request->request_done),
1767                         NES_EVENT_TIMEOUT * 2);
1768         nes_debug(NES_DBG_CQ, "Create iWARP CQ%u completed, wait_event_timeout ret = %d.\n",
1769                         nescq->hw_cq.cq_number, ret);
1770         if ((!ret) || (cqp_request->major_code)) {
1771                 nes_put_cqp_request(nesdev, cqp_request);
1772                 if (!context)
1773                         pci_free_consistent(nesdev->pcidev, nescq->cq_mem_size, mem,
1774                                         nescq->hw_cq.cq_pbase);
1775                 nes_free_resource(nesadapter, nesadapter->allocated_cqs, cq_num);
1776                 kfree(nescq);
1777                 return ERR_PTR(-EIO);
1778         }
1779         nes_put_cqp_request(nesdev, cqp_request);
1780
1781         if (context) {
1782                 /* free the nespbl */
1783                 pci_free_consistent(nesdev->pcidev, nespbl->pbl_size, nespbl->pbl_vbase,
1784                                 nespbl->pbl_pbase);
1785                 kfree(nespbl);
1786                 resp.cq_id = nescq->hw_cq.cq_number;
1787                 resp.cq_size = nescq->hw_cq.cq_size;
1788                 resp.mmap_db_index = 0;
1789                 if (ib_copy_to_udata(udata, &resp, sizeof resp)) {
1790                         nes_free_resource(nesadapter, nesadapter->allocated_cqs, cq_num);
1791                         kfree(nescq);
1792                         return ERR_PTR(-EFAULT);
1793                 }
1794         }
1795
1796         return &nescq->ibcq;
1797 }
1798
1799
1800 /**
1801  * nes_destroy_cq
1802  */
1803 static int nes_destroy_cq(struct ib_cq *ib_cq)
1804 {
1805         struct nes_cq *nescq;
1806         struct nes_device *nesdev;
1807         struct nes_vnic *nesvnic;
1808         struct nes_adapter *nesadapter;
1809         struct nes_hw_cqp_wqe *cqp_wqe;
1810         struct nes_cqp_request *cqp_request;
1811         unsigned long flags;
1812         u32 opcode = 0;
1813         int ret;
1814
1815         if (ib_cq == NULL)
1816                 return 0;
1817
1818         nescq = to_nescq(ib_cq);
1819         nesvnic = to_nesvnic(ib_cq->device);
1820         nesdev = nesvnic->nesdev;
1821         nesadapter = nesdev->nesadapter;
1822
1823         nes_debug(NES_DBG_CQ, "Destroy CQ%u\n", nescq->hw_cq.cq_number);
1824
1825         /* Send DestroyCQ request to CQP */
1826         cqp_request = nes_get_cqp_request(nesdev);
1827         if (cqp_request == NULL) {
1828                 nes_debug(NES_DBG_CQ, "Failed to get a cqp_request.\n");
1829                 return -ENOMEM;
1830         }
1831         cqp_request->waiting = 1;
1832         cqp_wqe = &cqp_request->cqp_wqe;
1833         opcode = NES_CQP_DESTROY_CQ | (nescq->hw_cq.cq_size << 16);
1834         spin_lock_irqsave(&nesadapter->pbl_lock, flags);
1835         if (nescq->virtual_cq == 1) {
1836                 nesadapter->free_256pbl++;
1837                 if (nesadapter->free_256pbl > nesadapter->max_256pbl) {
1838                         printk(KERN_ERR PFX "%s: free 256B PBLs(%u) has exceeded the max(%u)\n",
1839                                         __func__, nesadapter->free_256pbl, nesadapter->max_256pbl);
1840                 }
1841         } else if (nescq->virtual_cq == 2) {
1842                 nesadapter->free_4kpbl++;
1843                 if (nesadapter->free_4kpbl > nesadapter->max_4kpbl) {
1844                         printk(KERN_ERR PFX "%s: free 4K PBLs(%u) has exceeded the max(%u)\n",
1845                                         __func__, nesadapter->free_4kpbl, nesadapter->max_4kpbl);
1846                 }
1847                 opcode |= NES_CQP_CQ_4KB_CHUNK;
1848         }
1849
1850         spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
1851
1852         nes_fill_init_cqp_wqe(cqp_wqe, nesdev);
1853         set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_WQE_OPCODE_IDX, opcode);
1854         set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_WQE_ID_IDX,
1855                 (nescq->hw_cq.cq_number | ((u32)PCI_FUNC(nesdev->pcidev->devfn) << 16)));
1856         nes_free_resource(nesadapter, nesadapter->allocated_cqs, nescq->hw_cq.cq_number);
1857         atomic_set(&cqp_request->refcount, 2);
1858         nes_post_cqp_request(nesdev, cqp_request);
1859
1860         /* Wait for CQP */
1861         nes_debug(NES_DBG_CQ, "Waiting for destroy iWARP CQ%u to complete.\n",
1862                         nescq->hw_cq.cq_number);
1863         ret = wait_event_timeout(cqp_request->waitq, (0 != cqp_request->request_done),
1864                         NES_EVENT_TIMEOUT);
1865         nes_debug(NES_DBG_CQ, "Destroy iWARP CQ%u completed, wait_event_timeout ret = %u,"
1866                         " CQP Major:Minor codes = 0x%04X:0x%04X.\n",
1867                         nescq->hw_cq.cq_number, ret, cqp_request->major_code,
1868                         cqp_request->minor_code);
1869         if (!ret) {
1870                 nes_debug(NES_DBG_CQ, "iWARP CQ%u destroy timeout expired\n",
1871                                         nescq->hw_cq.cq_number);
1872                 ret = -ETIME;
1873         } else if (cqp_request->major_code) {
1874                 nes_debug(NES_DBG_CQ, "iWARP CQ%u destroy failed\n",
1875                                         nescq->hw_cq.cq_number);
1876                 ret = -EIO;
1877         } else {
1878                 ret = 0;
1879         }
1880         nes_put_cqp_request(nesdev, cqp_request);
1881
1882         if (nescq->cq_mem_size)
1883                 pci_free_consistent(nesdev->pcidev, nescq->cq_mem_size,
1884                                     nescq->hw_cq.cq_vbase, nescq->hw_cq.cq_pbase);
1885         kfree(nescq);
1886
1887         return ret;
1888 }
1889
1890
1891 /**
1892  * nes_reg_mr
1893  */
1894 static int nes_reg_mr(struct nes_device *nesdev, struct nes_pd *nespd,
1895                 u32 stag, u64 region_length, struct nes_root_vpbl *root_vpbl,
1896                 dma_addr_t single_buffer, u16 pbl_count, u16 residual_page_count,
1897                 int acc, u64 *iova_start)
1898 {
1899         struct nes_hw_cqp_wqe *cqp_wqe;
1900         struct nes_cqp_request *cqp_request;
1901         unsigned long flags;
1902         int ret;
1903         struct nes_adapter *nesadapter = nesdev->nesadapter;
1904         /* int count; */
1905         u32 opcode = 0;
1906         u16 major_code;
1907
1908         /* Register the region with the adapter */
1909         cqp_request = nes_get_cqp_request(nesdev);
1910         if (cqp_request == NULL) {
1911                 nes_debug(NES_DBG_MR, "Failed to get a cqp_request.\n");
1912                 return -ENOMEM;
1913         }
1914         cqp_request->waiting = 1;
1915         cqp_wqe = &cqp_request->cqp_wqe;
1916
1917         spin_lock_irqsave(&nesadapter->pbl_lock, flags);
1918         /* track PBL resources */
1919         if (pbl_count != 0) {
1920                 if (pbl_count > 1) {
1921                         /* Two level PBL */
1922                         if ((pbl_count+1) > nesadapter->free_4kpbl) {
1923                                 nes_debug(NES_DBG_MR, "Out of 4KB Pbls for two level request.\n");
1924                                 spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
1925                                 nes_free_cqp_request(nesdev, cqp_request);
1926                                 return -ENOMEM;
1927                         } else {
1928                                 nesadapter->free_4kpbl -= pbl_count+1;
1929                         }
1930                 } else if (residual_page_count > 32) {
1931                         if (pbl_count > nesadapter->free_4kpbl) {
1932                                 nes_debug(NES_DBG_MR, "Out of 4KB Pbls.\n");
1933                                 spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
1934                                 nes_free_cqp_request(nesdev, cqp_request);
1935                                 return -ENOMEM;
1936                         } else {
1937                                 nesadapter->free_4kpbl -= pbl_count;
1938                         }
1939                 } else {
1940                         if (pbl_count > nesadapter->free_256pbl) {
1941                                 nes_debug(NES_DBG_MR, "Out of 256B Pbls.\n");
1942                                 spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
1943                                 nes_free_cqp_request(nesdev, cqp_request);
1944                                 return -ENOMEM;
1945                         } else {
1946                                 nesadapter->free_256pbl -= pbl_count;
1947                         }
1948                 }
1949         }
1950
1951         spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
1952
1953         opcode = NES_CQP_REGISTER_STAG | NES_CQP_STAG_RIGHTS_LOCAL_READ |
1954                                         NES_CQP_STAG_VA_TO | NES_CQP_STAG_MR;
1955         if (acc & IB_ACCESS_LOCAL_WRITE)
1956                 opcode |= NES_CQP_STAG_RIGHTS_LOCAL_WRITE;
1957         if (acc & IB_ACCESS_REMOTE_WRITE)
1958                 opcode |= NES_CQP_STAG_RIGHTS_REMOTE_WRITE | NES_CQP_STAG_REM_ACC_EN;
1959         if (acc & IB_ACCESS_REMOTE_READ)
1960                 opcode |= NES_CQP_STAG_RIGHTS_REMOTE_READ | NES_CQP_STAG_REM_ACC_EN;
1961         if (acc & IB_ACCESS_MW_BIND)
1962                 opcode |= NES_CQP_STAG_RIGHTS_WINDOW_BIND | NES_CQP_STAG_REM_ACC_EN;
1963
1964         nes_fill_init_cqp_wqe(cqp_wqe, nesdev);
1965         set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_WQE_OPCODE_IDX, opcode);
1966         set_wqe_64bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_VA_LOW_IDX, *iova_start);
1967         set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_LEN_LOW_IDX, region_length);
1968
1969         cqp_wqe->wqe_words[NES_CQP_STAG_WQE_LEN_HIGH_PD_IDX] =
1970                         cpu_to_le32((u32)(region_length >> 8) & 0xff000000);
1971         cqp_wqe->wqe_words[NES_CQP_STAG_WQE_LEN_HIGH_PD_IDX] |=
1972                         cpu_to_le32(nespd->pd_id & 0x00007fff);
1973         set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_STAG_IDX, stag);
1974
1975         if (pbl_count == 0) {
1976                 set_wqe_64bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_PA_LOW_IDX, single_buffer);
1977         } else {
1978                 set_wqe_64bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_PA_LOW_IDX, root_vpbl->pbl_pbase);
1979                 set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_PBL_BLK_COUNT_IDX, pbl_count);
1980                 set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_PBL_LEN_IDX,
1981                                 (((pbl_count - 1) * 4096) + (residual_page_count*8)));
1982
1983                 if ((pbl_count > 1) || (residual_page_count > 32))
1984                         cqp_wqe->wqe_words[NES_CQP_WQE_OPCODE_IDX] |= cpu_to_le32(NES_CQP_STAG_PBL_BLK_SIZE);
1985         }
1986         barrier();
1987
1988         atomic_set(&cqp_request->refcount, 2);
1989         nes_post_cqp_request(nesdev, cqp_request);
1990
1991         /* Wait for CQP */
1992         ret = wait_event_timeout(cqp_request->waitq, (0 != cqp_request->request_done),
1993                         NES_EVENT_TIMEOUT);
1994         nes_debug(NES_DBG_MR, "Register STag 0x%08X completed, wait_event_timeout ret = %u,"
1995                         " CQP Major:Minor codes = 0x%04X:0x%04X.\n",
1996                         stag, ret, cqp_request->major_code, cqp_request->minor_code);
1997         major_code = cqp_request->major_code;
1998         nes_put_cqp_request(nesdev, cqp_request);
1999         if ((!ret || major_code) && pbl_count != 0) {
2000                 spin_lock_irqsave(&nesadapter->pbl_lock, flags);
2001                 if (pbl_count > 1)
2002                         nesadapter->free_4kpbl += pbl_count+1;
2003                 else if (residual_page_count > 32)
2004                         nesadapter->free_4kpbl += pbl_count;
2005                 else
2006                         nesadapter->free_256pbl += pbl_count;
2007                 spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
2008         }
2009         if (!ret)
2010                 return -ETIME;
2011         else if (major_code)
2012                 return -EIO;
2013         else
2014                 return 0;
2015
2016         return 0;
2017 }
2018
2019
2020 /**
2021  * nes_reg_phys_mr
2022  */
2023 static struct ib_mr *nes_reg_phys_mr(struct ib_pd *ib_pd,
2024                 struct ib_phys_buf *buffer_list, int num_phys_buf, int acc,
2025                 u64 * iova_start)
2026 {
2027         u64 region_length;
2028         struct nes_pd *nespd = to_nespd(ib_pd);
2029         struct nes_vnic *nesvnic = to_nesvnic(ib_pd->device);
2030         struct nes_device *nesdev = nesvnic->nesdev;
2031         struct nes_adapter *nesadapter = nesdev->nesadapter;
2032         struct nes_mr *nesmr;
2033         struct ib_mr *ibmr;
2034         struct nes_vpbl vpbl;
2035         struct nes_root_vpbl root_vpbl;
2036         u32 stag;
2037         u32 i;
2038         u32 stag_index = 0;
2039         u32 next_stag_index = 0;
2040         u32 driver_key = 0;
2041         u32 root_pbl_index = 0;
2042         u32 cur_pbl_index = 0;
2043         int err = 0, pbl_depth = 0;
2044         int ret = 0;
2045         u16 pbl_count = 0;
2046         u8 single_page = 1;
2047         u8 stag_key = 0;
2048
2049         pbl_depth = 0;
2050         region_length = 0;
2051         vpbl.pbl_vbase = NULL;
2052         root_vpbl.pbl_vbase = NULL;
2053         root_vpbl.pbl_pbase = 0;
2054
2055         get_random_bytes(&next_stag_index, sizeof(next_stag_index));
2056         stag_key = (u8)next_stag_index;
2057
2058         driver_key = 0;
2059
2060         next_stag_index >>= 8;
2061         next_stag_index %= nesadapter->max_mr;
2062         if (num_phys_buf > (1024*512)) {
2063                 return ERR_PTR(-E2BIG);
2064         }
2065
2066         err = nes_alloc_resource(nesadapter, nesadapter->allocated_mrs, nesadapter->max_mr,
2067                         &stag_index, &next_stag_index);
2068         if (err) {
2069                 return ERR_PTR(err);
2070         }
2071
2072         nesmr = kzalloc(sizeof(*nesmr), GFP_KERNEL);
2073         if (!nesmr) {
2074                 nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);
2075                 return ERR_PTR(-ENOMEM);
2076         }
2077
2078         for (i = 0; i < num_phys_buf; i++) {
2079
2080                 if ((i & 0x01FF) == 0) {
2081                         if (root_pbl_index == 1) {
2082                                 /* Allocate the root PBL */
2083                                 root_vpbl.pbl_vbase = pci_alloc_consistent(nesdev->pcidev, 8192,
2084                                                 &root_vpbl.pbl_pbase);
2085                                 nes_debug(NES_DBG_MR, "Allocating root PBL, va = %p, pa = 0x%08X\n",
2086                                                 root_vpbl.pbl_vbase, (unsigned int)root_vpbl.pbl_pbase);
2087                                 if (!root_vpbl.pbl_vbase) {
2088                                         pci_free_consistent(nesdev->pcidev, 4096, vpbl.pbl_vbase,
2089                                                         vpbl.pbl_pbase);
2090                                         nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);
2091                                         kfree(nesmr);
2092                                         return ERR_PTR(-ENOMEM);
2093                                 }
2094                                 root_vpbl.leaf_vpbl = kzalloc(sizeof(*root_vpbl.leaf_vpbl)*1024, GFP_KERNEL);
2095                                 if (!root_vpbl.leaf_vpbl) {
2096                                         pci_free_consistent(nesdev->pcidev, 8192, root_vpbl.pbl_vbase,
2097                                                         root_vpbl.pbl_pbase);
2098                                         pci_free_consistent(nesdev->pcidev, 4096, vpbl.pbl_vbase,
2099                                                         vpbl.pbl_pbase);
2100                                         nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);
2101                                         kfree(nesmr);
2102                                         return ERR_PTR(-ENOMEM);
2103                                 }
2104                                 root_vpbl.pbl_vbase[0].pa_low = cpu_to_le32((u32)vpbl.pbl_pbase);
2105                                 root_vpbl.pbl_vbase[0].pa_high =
2106                                                 cpu_to_le32((u32)((((u64)vpbl.pbl_pbase) >> 32)));
2107                                 root_vpbl.leaf_vpbl[0] = vpbl;
2108                         }
2109                         /* Allocate a 4K buffer for the PBL */
2110                         vpbl.pbl_vbase = pci_alloc_consistent(nesdev->pcidev, 4096,
2111                                         &vpbl.pbl_pbase);
2112                         nes_debug(NES_DBG_MR, "Allocating leaf PBL, va = %p, pa = 0x%016lX\n",
2113                                         vpbl.pbl_vbase, (unsigned long)vpbl.pbl_pbase);
2114                         if (!vpbl.pbl_vbase) {
2115                                 nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);
2116                                 ibmr = ERR_PTR(-ENOMEM);
2117                                 kfree(nesmr);
2118                                 goto reg_phys_err;
2119                         }
2120                         /* Fill in the root table */
2121                         if (1 <= root_pbl_index) {
2122                                 root_vpbl.pbl_vbase[root_pbl_index].pa_low =
2123                                                 cpu_to_le32((u32)vpbl.pbl_pbase);
2124                                 root_vpbl.pbl_vbase[root_pbl_index].pa_high =
2125                                                 cpu_to_le32((u32)((((u64)vpbl.pbl_pbase) >> 32)));
2126                                 root_vpbl.leaf_vpbl[root_pbl_index] = vpbl;
2127                         }
2128                         root_pbl_index++;
2129                         cur_pbl_index = 0;
2130                 }
2131                 if (buffer_list[i].addr & ~PAGE_MASK) {
2132                         /* TODO: Unwind allocated buffers */
2133                         nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);
2134                         nes_debug(NES_DBG_MR, "Unaligned Memory Buffer: 0x%x\n",
2135                                         (unsigned int) buffer_list[i].addr);
2136                         ibmr = ERR_PTR(-EINVAL);
2137                         kfree(nesmr);
2138                         goto reg_phys_err;
2139                 }
2140
2141                 if (!buffer_list[i].size) {
2142                         nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);
2143                         nes_debug(NES_DBG_MR, "Invalid Buffer Size\n");
2144                         ibmr = ERR_PTR(-EINVAL);
2145                         kfree(nesmr);
2146                         goto reg_phys_err;
2147                 }
2148
2149                 region_length += buffer_list[i].size;
2150                 if ((i != 0) && (single_page)) {
2151                         if ((buffer_list[i-1].addr+PAGE_SIZE) != buffer_list[i].addr)
2152                                 single_page = 0;
2153                 }
2154                 vpbl.pbl_vbase[cur_pbl_index].pa_low = cpu_to_le32((u32)buffer_list[i].addr);
2155                 vpbl.pbl_vbase[cur_pbl_index++].pa_high =
2156                                 cpu_to_le32((u32)((((u64)buffer_list[i].addr) >> 32)));
2157         }
2158
2159         stag = stag_index << 8;
2160         stag |= driver_key;
2161         stag += (u32)stag_key;
2162
2163         nes_debug(NES_DBG_MR, "Registering STag 0x%08X, VA = 0x%016lX,"
2164                         " length = 0x%016lX, index = 0x%08X\n",
2165                         stag, (unsigned long)*iova_start, (unsigned long)region_length, stag_index);
2166
2167         region_length -= (*iova_start)&PAGE_MASK;
2168
2169         /* Make the leaf PBL the root if only one PBL */
2170         if (root_pbl_index == 1) {
2171                 root_vpbl.pbl_pbase = vpbl.pbl_pbase;
2172         }
2173
2174         if (single_page) {
2175                 pbl_count = 0;
2176         } else {
2177                 pbl_count = root_pbl_index;
2178         }
2179         ret = nes_reg_mr(nesdev, nespd, stag, region_length, &root_vpbl,
2180                         buffer_list[0].addr, pbl_count, (u16)cur_pbl_index, acc, iova_start);
2181
2182         if (ret == 0) {
2183                 nesmr->ibmr.rkey = stag;
2184                 nesmr->ibmr.lkey = stag;
2185                 nesmr->mode = IWNES_MEMREG_TYPE_MEM;
2186                 ibmr = &nesmr->ibmr;
2187                 nesmr->pbl_4k = ((pbl_count > 1) || (cur_pbl_index > 32)) ? 1 : 0;
2188                 nesmr->pbls_used = pbl_count;
2189                 if (pbl_count > 1) {
2190                         nesmr->pbls_used++;
2191                 }
2192         } else {
2193                 kfree(nesmr);
2194                 ibmr = ERR_PTR(-ENOMEM);
2195         }
2196
2197         reg_phys_err:
2198         /* free the resources */
2199         if (root_pbl_index == 1) {
2200                 /* single PBL case */
2201                 pci_free_consistent(nesdev->pcidev, 4096, vpbl.pbl_vbase, vpbl.pbl_pbase);
2202         } else {
2203                 for (i=0; i<root_pbl_index; i++) {
2204                         pci_free_consistent(nesdev->pcidev, 4096, root_vpbl.leaf_vpbl[i].pbl_vbase,
2205                                         root_vpbl.leaf_vpbl[i].pbl_pbase);
2206                 }
2207                 kfree(root_vpbl.leaf_vpbl);
2208                 pci_free_consistent(nesdev->pcidev, 8192, root_vpbl.pbl_vbase,
2209                                 root_vpbl.pbl_pbase);
2210         }
2211
2212         return ibmr;
2213 }
2214
2215
2216 /**
2217  * nes_get_dma_mr
2218  */
2219 static struct ib_mr *nes_get_dma_mr(struct ib_pd *pd, int acc)
2220 {
2221         struct ib_phys_buf bl;
2222         u64 kva = 0;
2223
2224         nes_debug(NES_DBG_MR, "\n");
2225
2226         bl.size = (u64)0xffffffffffULL;
2227         bl.addr = 0;
2228         return nes_reg_phys_mr(pd, &bl, 1, acc, &kva);
2229 }
2230
2231
2232 /**
2233  * nes_reg_user_mr
2234  */
2235 static struct ib_mr *nes_reg_user_mr(struct ib_pd *pd, u64 start, u64 length,
2236                 u64 virt, int acc, struct ib_udata *udata)
2237 {
2238         u64 iova_start;
2239         __le64 *pbl;
2240         u64 region_length;
2241         dma_addr_t last_dma_addr = 0;
2242         dma_addr_t first_dma_addr = 0;
2243         struct nes_pd *nespd = to_nespd(pd);
2244         struct nes_vnic *nesvnic = to_nesvnic(pd->device);
2245         struct nes_device *nesdev = nesvnic->nesdev;
2246         struct nes_adapter *nesadapter = nesdev->nesadapter;
2247         struct ib_mr *ibmr = ERR_PTR(-EINVAL);
2248         struct ib_umem_chunk *chunk;
2249         struct nes_ucontext *nes_ucontext;
2250         struct nes_pbl *nespbl;
2251         struct nes_mr *nesmr;
2252         struct ib_umem *region;
2253         struct nes_mem_reg_req req;
2254         struct nes_vpbl vpbl;
2255         struct nes_root_vpbl root_vpbl;
2256         int nmap_index, page_index;
2257         int page_count = 0;
2258         int err, pbl_depth = 0;
2259         int chunk_pages;
2260         int ret;
2261         u32 stag;
2262         u32 stag_index = 0;
2263         u32 next_stag_index;
2264         u32 driver_key;
2265         u32 root_pbl_index = 0;
2266         u32 cur_pbl_index = 0;
2267         u32 skip_pages;
2268         u16 pbl_count;
2269         u8 single_page = 1;
2270         u8 stag_key;
2271
2272         region = ib_umem_get(pd->uobject->context, start, length, acc, 0);
2273         if (IS_ERR(region)) {
2274                 return (struct ib_mr *)region;
2275         }
2276
2277         nes_debug(NES_DBG_MR, "User base = 0x%lX, Virt base = 0x%lX, length = %u,"
2278                         " offset = %u, page size = %u.\n",
2279                         (unsigned long int)start, (unsigned long int)virt, (u32)length,
2280                         region->offset, region->page_size);
2281
2282         skip_pages = ((u32)region->offset) >> 12;
2283
2284         if (ib_copy_from_udata(&req, udata, sizeof(req)))
2285                 return ERR_PTR(-EFAULT);
2286         nes_debug(NES_DBG_MR, "Memory Registration type = %08X.\n", req.reg_type);
2287
2288         switch (req.reg_type) {
2289                 case IWNES_MEMREG_TYPE_MEM:
2290                         pbl_depth = 0;
2291                         region_length = 0;
2292                         vpbl.pbl_vbase = NULL;
2293                         root_vpbl.pbl_vbase = NULL;
2294                         root_vpbl.pbl_pbase = 0;
2295
2296                         get_random_bytes(&next_stag_index, sizeof(next_stag_index));
2297                         stag_key = (u8)next_stag_index;
2298
2299                         driver_key = next_stag_index & 0x70000000;
2300
2301                         next_stag_index >>= 8;
2302                         next_stag_index %= nesadapter->max_mr;
2303
2304                         err = nes_alloc_resource(nesadapter, nesadapter->allocated_mrs,
2305                                         nesadapter->max_mr, &stag_index, &next_stag_index);
2306                         if (err) {
2307                                 ib_umem_release(region);
2308                                 return ERR_PTR(err);
2309                         }
2310
2311                         nesmr = kzalloc(sizeof(*nesmr), GFP_KERNEL);
2312                         if (!nesmr) {
2313                                 ib_umem_release(region);
2314                                 nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);
2315                                 return ERR_PTR(-ENOMEM);
2316                         }
2317                         nesmr->region = region;
2318
2319                         list_for_each_entry(chunk, &region->chunk_list, list) {
2320                                 nes_debug(NES_DBG_MR, "Chunk: nents = %u, nmap = %u .\n",
2321                                                 chunk->nents, chunk->nmap);
2322                                 for (nmap_index = 0; nmap_index < chunk->nmap; ++nmap_index) {
2323                                         if (sg_dma_address(&chunk->page_list[nmap_index]) & ~PAGE_MASK) {
2324                                                 ib_umem_release(region);
2325                                                 nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);
2326                                                 nes_debug(NES_DBG_MR, "Unaligned Memory Buffer: 0x%x\n",
2327                                                                 (unsigned int) sg_dma_address(&chunk->page_list[nmap_index]));
2328                                                 ibmr = ERR_PTR(-EINVAL);
2329                                                 kfree(nesmr);
2330                                                 goto reg_user_mr_err;
2331                                         }
2332
2333                                         if (!sg_dma_len(&chunk->page_list[nmap_index])) {
2334                                                 ib_umem_release(region);
2335                                                 nes_free_resource(nesadapter, nesadapter->allocated_mrs,
2336                                                                 stag_index);
2337                                                 nes_debug(NES_DBG_MR, "Invalid Buffer Size\n");
2338                                                 ibmr = ERR_PTR(-EINVAL);
2339                                                 kfree(nesmr);
2340                                                 goto reg_user_mr_err;
2341                                         }
2342
2343                                         region_length += sg_dma_len(&chunk->page_list[nmap_index]);
2344                                         chunk_pages = sg_dma_len(&chunk->page_list[nmap_index]) >> 12;
2345                                         region_length -= skip_pages << 12;
2346                                         for (page_index=skip_pages; page_index < chunk_pages; page_index++) {
2347                                                 skip_pages = 0;
2348                                                 if ((page_count!=0)&&(page_count<<12)-(region->offset&(4096-1))>=region->length)
2349                                                         goto enough_pages;
2350                                                 if ((page_count&0x01FF) == 0) {
2351                                                         if (page_count >= 1024 * 512) {
2352                                                                 ib_umem_release(region);
2353                                                                 nes_free_resource(nesadapter,
2354                                                                                 nesadapter->allocated_mrs, stag_index);
2355                                                                 kfree(nesmr);
2356                                                                 ibmr = ERR_PTR(-E2BIG);
2357                                                                 goto reg_user_mr_err;
2358                                                         }
2359                                                         if (root_pbl_index == 1) {
2360                                                                 root_vpbl.pbl_vbase = pci_alloc_consistent(nesdev->pcidev,
2361                                                                                 8192, &root_vpbl.pbl_pbase);
2362                                                                 nes_debug(NES_DBG_MR, "Allocating root PBL, va = %p, pa = 0x%08X\n",
2363                                                                                 root_vpbl.pbl_vbase, (unsigned int)root_vpbl.pbl_pbase);
2364                                                                 if (!root_vpbl.pbl_vbase) {
2365                                                                         ib_umem_release(region);
2366                                                                         pci_free_consistent(nesdev->pcidev, 4096, vpbl.pbl_vbase,
2367                                                                                         vpbl.pbl_pbase);
2368                                                                         nes_free_resource(nesadapter, nesadapter->allocated_mrs,
2369                                                                                         stag_index);
2370                                                                         kfree(nesmr);
2371                                                                         ibmr = ERR_PTR(-ENOMEM);
2372                                                                         goto reg_user_mr_err;
2373                                                                 }
2374                                                                 root_vpbl.leaf_vpbl = kzalloc(sizeof(*root_vpbl.leaf_vpbl)*1024,
2375                                                                                 GFP_KERNEL);
2376                                                                 if (!root_vpbl.leaf_vpbl) {
2377                                                                         ib_umem_release(region);
2378                                                                         pci_free_consistent(nesdev->pcidev, 8192, root_vpbl.pbl_vbase,
2379                                                                                         root_vpbl.pbl_pbase);
2380                                                                         pci_free_consistent(nesdev->pcidev, 4096, vpbl.pbl_vbase,
2381                                                                                         vpbl.pbl_pbase);
2382                                                                         nes_free_resource(nesadapter, nesadapter->allocated_mrs,
2383                                                                                         stag_index);
2384                                                                         kfree(nesmr);
2385                                                                         ibmr = ERR_PTR(-ENOMEM);
2386                                                                         goto reg_user_mr_err;
2387                                                                 }
2388                                                                 root_vpbl.pbl_vbase[0].pa_low =
2389                                                                                 cpu_to_le32((u32)vpbl.pbl_pbase);
2390                                                                 root_vpbl.pbl_vbase[0].pa_high =
2391                                                                                 cpu_to_le32((u32)((((u64)vpbl.pbl_pbase) >> 32)));
2392                                                                 root_vpbl.leaf_vpbl[0] = vpbl;
2393                                                         }
2394                                                         vpbl.pbl_vbase = pci_alloc_consistent(nesdev->pcidev, 4096,
2395                                                                         &vpbl.pbl_pbase);
2396                                                         nes_debug(NES_DBG_MR, "Allocating leaf PBL, va = %p, pa = 0x%08X\n",
2397                                                                         vpbl.pbl_vbase, (unsigned int)vpbl.pbl_pbase);
2398                                                         if (!vpbl.pbl_vbase) {
2399                                                                 ib_umem_release(region);
2400                                                                 nes_free_resource(nesadapter, nesadapter->allocated_mrs, stag_index);
2401                                                                 ibmr = ERR_PTR(-ENOMEM);
2402                                                                 kfree(nesmr);
2403                                                                 goto reg_user_mr_err;
2404                                                         }
2405                                                         if (1 <= root_pbl_index) {
2406                                                                 root_vpbl.pbl_vbase[root_pbl_index].pa_low =
2407                                                                                 cpu_to_le32((u32)vpbl.pbl_pbase);
2408                                                                 root_vpbl.pbl_vbase[root_pbl_index].pa_high =
2409                                                                                 cpu_to_le32((u32)((((u64)vpbl.pbl_pbase)>>32)));
2410                                                                 root_vpbl.leaf_vpbl[root_pbl_index] = vpbl;
2411                                                         }
2412                                                         root_pbl_index++;
2413                                                         cur_pbl_index = 0;
2414                                                 }
2415                                                 if (single_page) {
2416                                                         if (page_count != 0) {
2417                                                                 if ((last_dma_addr+4096) !=
2418                                                                                 (sg_dma_address(&chunk->page_list[nmap_index])+
2419                                                                                 (page_index*4096)))
2420                                                                         single_page = 0;
2421                                                                 last_dma_addr = sg_dma_address(&chunk->page_list[nmap_index])+
2422                                                                                 (page_index*4096);
2423                                                         } else {
2424                                                                 first_dma_addr = sg_dma_address(&chunk->page_list[nmap_index])+
2425                                                                                 (page_index*4096);
2426                                                                 last_dma_addr = first_dma_addr;
2427                                                         }
2428                                                 }
2429
2430                                                 vpbl.pbl_vbase[cur_pbl_index].pa_low =
2431                                                                 cpu_to_le32((u32)(sg_dma_address(&chunk->page_list[nmap_index])+
2432                                                                 (page_index*4096)));
2433                                                 vpbl.pbl_vbase[cur_pbl_index].pa_high =
2434                                                                 cpu_to_le32((u32)((((u64)(sg_dma_address(&chunk->page_list[nmap_index])+
2435                                                                 (page_index*4096))) >> 32)));
2436                                                 cur_pbl_index++;
2437                                                 page_count++;
2438                                         }
2439                                 }
2440                         }
2441                         enough_pages:
2442                         nes_debug(NES_DBG_MR, "calculating stag, stag_index=0x%08x, driver_key=0x%08x,"
2443                                         " stag_key=0x%08x\n",
2444                                         stag_index, driver_key, stag_key);
2445                         stag = stag_index << 8;
2446                         stag |= driver_key;
2447                         stag += (u32)stag_key;
2448                         if (stag == 0) {
2449                                 stag = 1;
2450                         }
2451
2452                         iova_start = virt;
2453                         /* Make the leaf PBL the root if only one PBL */
2454                         if (root_pbl_index == 1) {
2455                                 root_vpbl.pbl_pbase = vpbl.pbl_pbase;
2456                         }
2457
2458                         if (single_page) {
2459                                 pbl_count = 0;
2460                         } else {
2461                                 pbl_count = root_pbl_index;
2462                                 first_dma_addr = 0;
2463                         }
2464                         nes_debug(NES_DBG_MR, "Registering STag 0x%08X, VA = 0x%08X, length = 0x%08X,"
2465                                         " index = 0x%08X, region->length=0x%08llx, pbl_count = %u\n",
2466                                         stag, (unsigned int)iova_start,
2467                                         (unsigned int)region_length, stag_index,
2468                                         (unsigned long long)region->length, pbl_count);
2469                         ret = nes_reg_mr( nesdev, nespd, stag, region->length, &root_vpbl,
2470                                         first_dma_addr, pbl_count, (u16)cur_pbl_index, acc, &iova_start);
2471
2472                         nes_debug(NES_DBG_MR, "ret=%d\n", ret);
2473
2474                         if (ret == 0) {
2475                                 nesmr->ibmr.rkey = stag;
2476                                 nesmr->ibmr.lkey = stag;
2477                                 nesmr->mode = IWNES_MEMREG_TYPE_MEM;
2478                                 ibmr = &nesmr->ibmr;
2479                                 nesmr->pbl_4k = ((pbl_count > 1) || (cur_pbl_index > 32)) ? 1 : 0;
2480                                 nesmr->pbls_used = pbl_count;
2481                                 if (pbl_count > 1) {
2482                                         nesmr->pbls_used++;
2483                                 }
2484                         } else {
2485                                 ib_umem_release(region);
2486                                 kfree(nesmr);
2487                                 ibmr = ERR_PTR(-ENOMEM);
2488                         }
2489
2490                         reg_user_mr_err:
2491                         /* free the resources */
2492                         if (root_pbl_index == 1) {
2493                                 pci_free_consistent(nesdev->pcidev, 4096, vpbl.pbl_vbase,
2494                                                 vpbl.pbl_pbase);
2495                         } else {
2496                                 for (page_index=0; page_index<root_pbl_index; page_index++) {
2497                                         pci_free_consistent(nesdev->pcidev, 4096,
2498                                                         root_vpbl.leaf_vpbl[page_index].pbl_vbase,
2499                                                         root_vpbl.leaf_vpbl[page_index].pbl_pbase);
2500                                 }
2501                                 kfree(root_vpbl.leaf_vpbl);
2502                                 pci_free_consistent(nesdev->pcidev, 8192, root_vpbl.pbl_vbase,
2503                                                 root_vpbl.pbl_pbase);
2504                         }
2505
2506                         nes_debug(NES_DBG_MR, "Leaving, ibmr=%p", ibmr);
2507
2508                         return ibmr;
2509                 case IWNES_MEMREG_TYPE_QP:
2510                 case IWNES_MEMREG_TYPE_CQ:
2511                         nespbl = kzalloc(sizeof(*nespbl), GFP_KERNEL);
2512                         if (!nespbl) {
2513                                 nes_debug(NES_DBG_MR, "Unable to allocate PBL\n");
2514                                 ib_umem_release(region);
2515                                 return ERR_PTR(-ENOMEM);
2516                         }
2517                         nesmr = kzalloc(sizeof(*nesmr), GFP_KERNEL);
2518                         if (!nesmr) {
2519                                 ib_umem_release(region);
2520                                 kfree(nespbl);
2521                                 nes_debug(NES_DBG_MR, "Unable to allocate nesmr\n");
2522                                 return ERR_PTR(-ENOMEM);
2523                         }
2524                         nesmr->region = region;
2525                         nes_ucontext = to_nesucontext(pd->uobject->context);
2526                         pbl_depth = region->length >> 12;
2527                         pbl_depth += (region->length & (4096-1)) ? 1 : 0;
2528                         nespbl->pbl_size = pbl_depth*sizeof(u64);
2529                         if (req.reg_type == IWNES_MEMREG_TYPE_QP) {
2530                                 nes_debug(NES_DBG_MR, "Attempting to allocate QP PBL memory");
2531                         } else {
2532                                 nes_debug(NES_DBG_MR, "Attempting to allocate CP PBL memory");
2533                         }
2534
2535                         nes_debug(NES_DBG_MR, " %u bytes, %u entries.\n",
2536                                         nespbl->pbl_size, pbl_depth);
2537                         pbl = pci_alloc_consistent(nesdev->pcidev, nespbl->pbl_size,
2538                                         &nespbl->pbl_pbase);
2539                         if (!pbl) {
2540                                 ib_umem_release(region);
2541                                 kfree(nesmr);
2542                                 kfree(nespbl);
2543                                 nes_debug(NES_DBG_MR, "Unable to allocate PBL memory\n");
2544                                 return ERR_PTR(-ENOMEM);
2545                         }
2546
2547                         nespbl->pbl_vbase = (u64 *)pbl;
2548                         nespbl->user_base = start;
2549                         nes_debug(NES_DBG_MR, "Allocated PBL memory, %u bytes, pbl_pbase=%lx,"
2550                                         " pbl_vbase=%p user_base=0x%lx\n",
2551                                   nespbl->pbl_size, (unsigned long) nespbl->pbl_pbase,
2552                                   (void *) nespbl->pbl_vbase, nespbl->user_base);
2553
2554                         list_for_each_entry(chunk, &region->chunk_list, list) {
2555                                 for (nmap_index = 0; nmap_index < chunk->nmap; ++nmap_index) {
2556                                         chunk_pages = sg_dma_len(&chunk->page_list[nmap_index]) >> 12;
2557                                         chunk_pages += (sg_dma_len(&chunk->page_list[nmap_index]) & (4096-1)) ? 1 : 0;
2558                                         nespbl->page = sg_page(&chunk->page_list[0]);
2559                                         for (page_index=0; page_index<chunk_pages; page_index++) {
2560                                                 ((__le32 *)pbl)[0] = cpu_to_le32((u32)
2561                                                                 (sg_dma_address(&chunk->page_list[nmap_index])+
2562                                                                 (page_index*4096)));
2563                                                 ((__le32 *)pbl)[1] = cpu_to_le32(((u64)
2564                                                                 (sg_dma_address(&chunk->page_list[nmap_index])+
2565                                                                 (page_index*4096)))>>32);
2566                                                 nes_debug(NES_DBG_MR, "pbl=%p, *pbl=0x%016llx, 0x%08x%08x\n", pbl,
2567                                                                 (unsigned long long)*pbl,
2568                                                                 le32_to_cpu(((__le32 *)pbl)[1]), le32_to_cpu(((__le32 *)pbl)[0]));
2569                                                 pbl++;
2570                                         }
2571                                 }
2572                         }
2573                         if (req.reg_type == IWNES_MEMREG_TYPE_QP) {
2574                                 list_add_tail(&nespbl->list, &nes_ucontext->qp_reg_mem_list);
2575                         } else {
2576                                 list_add_tail(&nespbl->list, &nes_ucontext->cq_reg_mem_list);
2577                         }
2578                         nesmr->ibmr.rkey = -1;
2579                         nesmr->ibmr.lkey = -1;
2580                         nesmr->mode = req.reg_type;
2581                         return &nesmr->ibmr;
2582         }
2583
2584         return ERR_PTR(-ENOSYS);
2585 }
2586
2587
2588 /**
2589  * nes_dereg_mr
2590  */
2591 static int nes_dereg_mr(struct ib_mr *ib_mr)
2592 {
2593         struct nes_mr *nesmr = to_nesmr(ib_mr);
2594         struct nes_vnic *nesvnic = to_nesvnic(ib_mr->device);
2595         struct nes_device *nesdev = nesvnic->nesdev;
2596         struct nes_adapter *nesadapter = nesdev->nesadapter;
2597         struct nes_hw_cqp_wqe *cqp_wqe;
2598         struct nes_cqp_request *cqp_request;
2599         unsigned long flags;
2600         int ret;
2601         u16 major_code;
2602         u16 minor_code;
2603
2604         if (nesmr->region) {
2605                 ib_umem_release(nesmr->region);
2606         }
2607         if (nesmr->mode != IWNES_MEMREG_TYPE_MEM) {
2608                 kfree(nesmr);
2609                 return 0;
2610         }
2611
2612         /* Deallocate the region with the adapter */
2613
2614         cqp_request = nes_get_cqp_request(nesdev);
2615         if (cqp_request == NULL) {
2616                 nes_debug(NES_DBG_MR, "Failed to get a cqp_request.\n");
2617                 return -ENOMEM;
2618         }
2619         cqp_request->waiting = 1;
2620         cqp_wqe = &cqp_request->cqp_wqe;
2621
2622         nes_fill_init_cqp_wqe(cqp_wqe, nesdev);
2623         set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_WQE_OPCODE_IDX,
2624                         NES_CQP_DEALLOCATE_STAG | NES_CQP_STAG_VA_TO |
2625                         NES_CQP_STAG_DEALLOC_PBLS | NES_CQP_STAG_MR);
2626         set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_STAG_WQE_STAG_IDX, ib_mr->rkey);
2627
2628         atomic_set(&cqp_request->refcount, 2);
2629         nes_post_cqp_request(nesdev, cqp_request);
2630
2631         /* Wait for CQP */
2632         nes_debug(NES_DBG_MR, "Waiting for deallocate STag 0x%08X completed\n", ib_mr->rkey);
2633         ret = wait_event_timeout(cqp_request->waitq, (cqp_request->request_done != 0),
2634                         NES_EVENT_TIMEOUT);
2635         nes_debug(NES_DBG_MR, "Deallocate STag 0x%08X completed, wait_event_timeout ret = %u,"
2636                         " CQP Major:Minor codes = 0x%04X:0x%04X\n",
2637                         ib_mr->rkey, ret, cqp_request->major_code, cqp_request->minor_code);
2638
2639         major_code = cqp_request->major_code;
2640         minor_code = cqp_request->minor_code;
2641
2642         nes_put_cqp_request(nesdev, cqp_request);
2643
2644         if (!ret) {
2645                 nes_debug(NES_DBG_MR, "Timeout waiting to destroy STag,"
2646                                 " ib_mr=%p, rkey = 0x%08X\n",
2647                                 ib_mr, ib_mr->rkey);
2648                 return -ETIME;
2649         } else if (major_code) {
2650                 nes_debug(NES_DBG_MR, "Error (0x%04X:0x%04X) while attempting"
2651                                 " to destroy STag, ib_mr=%p, rkey = 0x%08X\n",
2652                                 major_code, minor_code, ib_mr, ib_mr->rkey);
2653                 return -EIO;
2654         }
2655
2656         if (nesmr->pbls_used != 0) {
2657                 spin_lock_irqsave(&nesadapter->pbl_lock, flags);
2658                 if (nesmr->pbl_4k) {
2659                         nesadapter->free_4kpbl += nesmr->pbls_used;
2660                         if (nesadapter->free_4kpbl > nesadapter->max_4kpbl)
2661                                 printk(KERN_ERR PFX "free 4KB PBLs(%u) has "
2662                                         "exceeded the max(%u)\n",
2663                                         nesadapter->free_4kpbl,
2664                                         nesadapter->max_4kpbl);
2665                 } else {
2666                         nesadapter->free_256pbl += nesmr->pbls_used;
2667                         if (nesadapter->free_256pbl > nesadapter->max_256pbl)
2668                                 printk(KERN_ERR PFX "free 256B PBLs(%u) has "
2669                                         "exceeded the max(%u)\n",
2670                                         nesadapter->free_256pbl,
2671                                         nesadapter->max_256pbl);
2672                 }
2673                 spin_unlock_irqrestore(&nesadapter->pbl_lock, flags);
2674         }
2675         nes_free_resource(nesadapter, nesadapter->allocated_mrs,
2676                         (ib_mr->rkey & 0x0fffff00) >> 8);
2677
2678         kfree(nesmr);
2679
2680         return 0;
2681 }
2682
2683
2684 /**
2685  * show_rev
2686  */
2687 static ssize_t show_rev(struct device *dev, struct device_attribute *attr,
2688                         char *buf)
2689 {
2690         struct nes_ib_device *nesibdev =
2691                         container_of(dev, struct nes_ib_device, ibdev.dev);
2692         struct nes_vnic *nesvnic = nesibdev->nesvnic;
2693
2694         nes_debug(NES_DBG_INIT, "\n");
2695         return sprintf(buf, "%x\n", nesvnic->nesdev->nesadapter->hw_rev);
2696 }
2697
2698
2699 /**
2700  * show_fw_ver
2701  */
2702 static ssize_t show_fw_ver(struct device *dev, struct device_attribute *attr,
2703                            char *buf)
2704 {
2705         struct nes_ib_device *nesibdev =
2706                         container_of(dev, struct nes_ib_device, ibdev.dev);
2707         struct nes_vnic *nesvnic = nesibdev->nesvnic;
2708
2709         nes_debug(NES_DBG_INIT, "\n");
2710         return sprintf(buf, "%x.%x.%x\n",
2711                         (int)(nesvnic->nesdev->nesadapter->fw_ver >> 32),
2712                         (int)(nesvnic->nesdev->nesadapter->fw_ver >> 16) & 0xffff,
2713                         (int)(nesvnic->nesdev->nesadapter->fw_ver & 0xffff));
2714 }
2715
2716
2717 /**
2718  * show_hca
2719  */
2720 static ssize_t show_hca(struct device *dev, struct device_attribute *attr,
2721                         char *buf)
2722 {
2723         nes_debug(NES_DBG_INIT, "\n");
2724         return sprintf(buf, "NES020\n");
2725 }
2726
2727
2728 /**
2729  * show_board
2730  */
2731 static ssize_t show_board(struct device *dev, struct device_attribute *attr,
2732                           char *buf)
2733 {
2734         nes_debug(NES_DBG_INIT, "\n");
2735         return sprintf(buf, "%.*s\n", 32, "NES020 Board ID");
2736 }
2737
2738
2739 static DEVICE_ATTR(hw_rev, S_IRUGO, show_rev, NULL);
2740 static DEVICE_ATTR(fw_ver, S_IRUGO, show_fw_ver, NULL);
2741 static DEVICE_ATTR(hca_type, S_IRUGO, show_hca, NULL);
2742 static DEVICE_ATTR(board_id, S_IRUGO, show_board, NULL);
2743
2744 static struct device_attribute *nes_dev_attributes[] = {
2745         &dev_attr_hw_rev,
2746         &dev_attr_fw_ver,
2747         &dev_attr_hca_type,
2748         &dev_attr_board_id
2749 };
2750
2751
2752 /**
2753  * nes_query_qp
2754  */
2755 static int nes_query_qp(struct ib_qp *ibqp, struct ib_qp_attr *attr,
2756                 int attr_mask, struct ib_qp_init_attr *init_attr)
2757 {
2758         struct nes_qp *nesqp = to_nesqp(ibqp);
2759
2760         nes_debug(NES_DBG_QP, "\n");
2761
2762         attr->qp_access_flags = 0;
2763         attr->cap.max_send_wr = nesqp->hwqp.sq_size;
2764         attr->cap.max_recv_wr = nesqp->hwqp.rq_size;
2765         attr->cap.max_recv_sge = 1;
2766         if (nes_drv_opt & NES_DRV_OPT_NO_INLINE_DATA) {
2767                 init_attr->cap.max_inline_data = 0;
2768         } else {
2769                 init_attr->cap.max_inline_data = 64;
2770         }
2771
2772         init_attr->event_handler = nesqp->ibqp.event_handler;
2773         init_attr->qp_context = nesqp->ibqp.qp_context;
2774         init_attr->send_cq = nesqp->ibqp.send_cq;
2775         init_attr->recv_cq = nesqp->ibqp.recv_cq;
2776         init_attr->srq = nesqp->ibqp.srq = nesqp->ibqp.srq;
2777         init_attr->cap = attr->cap;
2778
2779         return 0;
2780 }
2781
2782
2783 /**
2784  * nes_hw_modify_qp
2785  */
2786 int nes_hw_modify_qp(struct nes_device *nesdev, struct nes_qp *nesqp,
2787                 u32 next_iwarp_state, u32 wait_completion)
2788 {
2789         struct nes_hw_cqp_wqe *cqp_wqe;
2790         /* struct iw_cm_id *cm_id = nesqp->cm_id; */
2791         /* struct iw_cm_event cm_event; */
2792         struct nes_cqp_request *cqp_request;
2793         int ret;
2794         u16 major_code;
2795
2796         nes_debug(NES_DBG_MOD_QP, "QP%u, refcount=%d\n",
2797                         nesqp->hwqp.qp_id, atomic_read(&nesqp->refcount));
2798
2799         cqp_request = nes_get_cqp_request(nesdev);
2800         if (cqp_request == NULL) {
2801                 nes_debug(NES_DBG_MOD_QP, "Failed to get a cqp_request.\n");
2802                 return -ENOMEM;
2803         }
2804         if (wait_completion) {
2805                 cqp_request->waiting = 1;
2806         } else {
2807                 cqp_request->waiting = 0;
2808         }
2809         cqp_wqe = &cqp_request->cqp_wqe;
2810
2811         set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_WQE_OPCODE_IDX,
2812                         NES_CQP_MODIFY_QP | NES_CQP_QP_TYPE_IWARP | next_iwarp_state);
2813         nes_debug(NES_DBG_MOD_QP, "using next_iwarp_state=%08x, wqe_words=%08x\n",
2814                         next_iwarp_state, le32_to_cpu(cqp_wqe->wqe_words[NES_CQP_WQE_OPCODE_IDX]));
2815         nes_fill_init_cqp_wqe(cqp_wqe, nesdev);
2816         set_wqe_32bit_value(cqp_wqe->wqe_words, NES_CQP_WQE_ID_IDX, nesqp->hwqp.qp_id);
2817         set_wqe_64bit_value(cqp_wqe->wqe_words, NES_CQP_QP_WQE_CONTEXT_LOW_IDX, (u64)nesqp->nesqp_context_pbase);
2818
2819         atomic_set(&cqp_request->refcount, 2);
2820         nes_post_cqp_request(nesdev, cqp_request);
2821
2822         /* Wait for CQP */
2823         if (wait_completion) {
2824                 /* nes_debug(NES_DBG_MOD_QP, "Waiting for modify iWARP QP%u to complete.\n",
2825                                 nesqp->hwqp.qp_id); */
2826                 ret = wait_event_timeout(cqp_request->waitq, (cqp_request->request_done != 0),
2827                                 NES_EVENT_TIMEOUT);
2828                 nes_debug(NES_DBG_MOD_QP, "Modify iwarp QP%u completed, wait_event_timeout ret=%u, "
2829                                 "CQP Major:Minor codes = 0x%04X:0x%04X.\n",
2830                                 nesqp->hwqp.qp_id, ret, cqp_request->major_code, cqp_request->minor_code);
2831                 major_code = cqp_request->major_code;
2832                 if (major_code) {
2833                         nes_debug(NES_DBG_MOD_QP, "Modify iwarp QP%u failed"
2834                                         "CQP Major:Minor codes = 0x%04X:0x%04X, intended next state = 0x%08X.\n",
2835                                         nesqp->hwqp.qp_id, cqp_request->major_code,
2836                                         cqp_request->minor_code, next_iwarp_state);
2837                 }
2838
2839                 nes_put_cqp_request(nesdev, cqp_request);
2840
2841                 if (!ret)
2842                         return -ETIME;
2843                 else if (major_code)
2844                         return -EIO;
2845                 else
2846                         return 0;
2847         } else {
2848                 return 0;
2849         }
2850 }
2851
2852
2853 /**
2854  * nes_modify_qp
2855  */
2856 int nes_modify_qp(struct ib_qp *ibqp, struct ib_qp_attr *attr,
2857                 int attr_mask, struct ib_udata *udata)
2858 {
2859         struct nes_qp *nesqp = to_nesqp(ibqp);
2860         struct nes_vnic *nesvnic = to_nesvnic(ibqp->device);
2861         struct nes_device *nesdev = nesvnic->nesdev;
2862         /* u32 cqp_head; */
2863         /* u32 counter; */
2864         u32 next_iwarp_state = 0;
2865         int err;
2866         unsigned long qplockflags;
2867         int ret;
2868         u16 original_last_aeq;
2869         u8 issue_modify_qp = 0;
2870         u8 issue_disconnect = 0;
2871         u8 dont_wait = 0;
2872
2873         nes_debug(NES_DBG_MOD_QP, "QP%u: QP State=%u, cur QP State=%u,"
2874                         " iwarp_state=0x%X, refcount=%d\n",
2875                         nesqp->hwqp.qp_id, attr->qp_state, nesqp->ibqp_state,
2876                         nesqp->iwarp_state, atomic_read(&nesqp->refcount));
2877
2878         spin_lock_irqsave(&nesqp->lock, qplockflags);
2879
2880         nes_debug(NES_DBG_MOD_QP, "QP%u: hw_iwarp_state=0x%X, hw_tcp_state=0x%X,"
2881                         " QP Access Flags=0x%X, attr_mask = 0x%0x\n",
2882                         nesqp->hwqp.qp_id, nesqp->hw_iwarp_state,
2883                         nesqp->hw_tcp_state, attr->qp_access_flags, attr_mask);
2884
2885         if (attr_mask & IB_QP_STATE) {
2886                 switch (attr->qp_state) {
2887                         case IB_QPS_INIT:
2888                                 nes_debug(NES_DBG_MOD_QP, "QP%u: new state = init\n",
2889                                                 nesqp->hwqp.qp_id);
2890                                 if (nesqp->iwarp_state > (u32)NES_CQP_QP_IWARP_STATE_IDLE) {
2891                                         spin_unlock_irqrestore(&nesqp->lock, qplockflags);
2892                                         return -EINVAL;
2893                                 }
2894                                 next_iwarp_state = NES_CQP_QP_IWARP_STATE_IDLE;
2895                                 issue_modify_qp = 1;
2896                                 break;
2897                         case IB_QPS_RTR:
2898                                 nes_debug(NES_DBG_MOD_QP, "QP%u: new state = rtr\n",
2899                                                 nesqp->hwqp.qp_id);
2900                                 if (nesqp->iwarp_state>(u32)NES_CQP_QP_IWARP_STATE_IDLE) {
2901                                         spin_unlock_irqrestore(&nesqp->lock, qplockflags);
2902                                         return -EINVAL;
2903                                 }
2904                                 next_iwarp_state = NES_CQP_QP_IWARP_STATE_IDLE;
2905                                 issue_modify_qp = 1;
2906                                 break;
2907                         case IB_QPS_RTS:
2908                                 nes_debug(NES_DBG_MOD_QP, "QP%u: new state = rts\n",
2909                                                 nesqp->hwqp.qp_id);
2910                                 if (nesqp->iwarp_state>(u32)NES_CQP_QP_IWARP_STATE_RTS) {
2911                                         spin_unlock_irqrestore(&nesqp->lock, qplockflags);
2912                                         return -EINVAL;
2913                                 }
2914                                 if (nesqp->cm_id == NULL) {
2915                                         nes_debug(NES_DBG_MOD_QP, "QP%u: Failing attempt to move QP to RTS without a CM_ID. \n",
2916                                                         nesqp->hwqp.qp_id );
2917                                         spin_unlock_irqrestore(&nesqp->lock, qplockflags);
2918                                         return -EINVAL;
2919                                 }
2920                                 next_iwarp_state = NES_CQP_QP_IWARP_STATE_RTS;
2921                                 if (nesqp->iwarp_state != NES_CQP_QP_IWARP_STATE_RTS)
2922                                         next_iwarp_state |= NES_CQP_QP_CONTEXT_VALID |
2923                                                         NES_CQP_QP_ARP_VALID | NES_CQP_QP_ORD_VALID;
2924                                 issue_modify_qp = 1;
2925                                 nesqp->hw_tcp_state = NES_AEQE_TCP_STATE_ESTABLISHED;
2926                                 nesqp->hw_iwarp_state = NES_AEQE_IWARP_STATE_RTS;
2927                                 nesqp->hte_added = 1;
2928                                 break;
2929                         case IB_QPS_SQD:
2930                                 issue_modify_qp = 1;
2931                                 nes_debug(NES_DBG_MOD_QP, "QP%u: new state=closing. SQ head=%u, SQ tail=%u\n",
2932                                                 nesqp->hwqp.qp_id, nesqp->hwqp.sq_head, nesqp->hwqp.sq_tail);
2933                                 if (nesqp->iwarp_state == (u32)NES_CQP_QP_IWARP_STATE_CLOSING) {
2934                                         spin_unlock_irqrestore(&nesqp->lock, qplockflags);
2935                                         return 0;
2936                                 } else {
2937                                         if (nesqp->iwarp_state > (u32)NES_CQP_QP_IWARP_STATE_CLOSING) {
2938                                                 nes_debug(NES_DBG_MOD_QP, "QP%u: State change to closing"
2939                                                                 " ignored due to current iWARP state\n",
2940                                                                 nesqp->hwqp.qp_id);
2941                                                 spin_unlock_irqrestore(&nesqp->lock, qplockflags);
2942                                                 return -EINVAL;
2943                                         }
2944                                         if (nesqp->hw_iwarp_state != NES_AEQE_IWARP_STATE_RTS) {
2945                                                 nes_debug(NES_DBG_MOD_QP, "QP%u: State change to closing"
2946                                                                 " already done based on hw state.\n",
2947                                                                 nesqp->hwqp.qp_id);
2948                                                 issue_modify_qp = 0;
2949                                                 nesqp->in_disconnect = 0;
2950                                         }
2951                                         switch (nesqp->hw_iwarp_state) {
2952                                                 case NES_AEQE_IWARP_STATE_CLOSING:
2953                                                         next_iwarp_state = NES_CQP_QP_IWARP_STATE_CLOSING;
2954                                                 case NES_AEQE_IWARP_STATE_TERMINATE:
2955                                                         next_iwarp_state = NES_CQP_QP_IWARP_STATE_TERMINATE;
2956                                                         break;
2957                                                 case NES_AEQE_IWARP_STATE_ERROR:
2958                                                         next_iwarp_state = NES_CQP_QP_IWARP_STATE_ERROR;
2959                                                         break;
2960                                                 default:
2961                                                         next_iwarp_state = NES_CQP_QP_IWARP_STATE_CLOSING;
2962                                                         nesqp->in_disconnect = 1;
2963                                                         nesqp->hw_iwarp_state = NES_AEQE_IWARP_STATE_CLOSING;
2964                                                         break;
2965                                         }
2966                                 }
2967                                 break;
2968                         case IB_QPS_SQE:
2969                                 nes_debug(NES_DBG_MOD_QP, "QP%u: new state = terminate\n",
2970                                                 nesqp->hwqp.qp_id);
2971                                 if (nesqp->iwarp_state>=(u32)NES_CQP_QP_IWARP_STATE_TERMINATE) {
2972                                         spin_unlock_irqrestore(&nesqp->lock, qplockflags);
2973                                         return -EINVAL;
2974                                 }
2975                                 /* next_iwarp_state = (NES_CQP_QP_IWARP_STATE_TERMINATE | 0x02000000); */
2976                                 next_iwarp_state = NES_CQP_QP_IWARP_STATE_TERMINATE;
2977                                 nesqp->hw_iwarp_state = NES_AEQE_IWARP_STATE_TERMINATE;
2978                                 issue_modify_qp = 1;
2979                                 nesqp->in_disconnect = 1;
2980                                 break;
2981                         case IB_QPS_ERR:
2982                         case IB_QPS_RESET:
2983                                 if (nesqp->iwarp_state == (u32)NES_CQP_QP_IWARP_STATE_ERROR) {
2984                                         spin_unlock_irqrestore(&nesqp->lock, qplockflags);
2985                                         return -EINVAL;
2986                                 }
2987                                 nes_debug(NES_DBG_MOD_QP, "QP%u: new state = error\n",
2988                                                 nesqp->hwqp.qp_id);
2989                                 next_iwarp_state = NES_CQP_QP_IWARP_STATE_ERROR;
2990                                 /* next_iwarp_state = (NES_CQP_QP_IWARP_STATE_TERMINATE | 0x02000000); */
2991                                         if (nesqp->hte_added) {
2992                                                 nes_debug(NES_DBG_MOD_QP, "set CQP_QP_DEL_HTE\n");
2993                                                 next_iwarp_state |= NES_CQP_QP_DEL_HTE;
2994                                                 nesqp->hte_added = 0;
2995                                         }
2996                                 if ((nesqp->hw_tcp_state > NES_AEQE_TCP_STATE_CLOSED) &&
2997                                                 (nesqp->hw_tcp_state != NES_AEQE_TCP_STATE_TIME_WAIT)) {
2998                                         next_iwarp_state |= NES_CQP_QP_RESET;
2999                                         nesqp->in_disconnect = 1;
3000                                 } else {
3001                                         nes_debug(NES_DBG_MOD_QP, "QP%u NOT setting NES_CQP_QP_RESET since TCP state = %u\n",
3002                                                         nesqp->hwqp.qp_id, nesqp->hw_tcp_state);
3003                                         dont_wait = 1;
3004                                 }
3005                                 issue_modify_qp = 1;
3006                                 nesqp->hw_iwarp_state = NES_AEQE_IWARP_STATE_ERROR;
3007                                 break;
3008                         default:
3009                                 spin_unlock_irqrestore(&nesqp->lock, qplockflags);
3010                                 return -EINVAL;
3011                                 break;
3012                 }
3013
3014                 nesqp->ibqp_state = attr->qp_state;
3015                 if (((nesqp->iwarp_state & NES_CQP_QP_IWARP_STATE_MASK) ==
3016                                 (u32)NES_CQP_QP_IWARP_STATE_RTS) &&
3017                                 ((next_iwarp_state & NES_CQP_QP_IWARP_STATE_MASK) >
3018                                 (u32)NES_CQP_QP_IWARP_STATE_RTS)) {
3019                         nesqp->iwarp_state = next_iwarp_state & NES_CQP_QP_IWARP_STATE_MASK;
3020                         nes_debug(NES_DBG_MOD_QP, "Change nesqp->iwarp_state=%08x\n",
3021                                         nesqp->iwarp_state);
3022                         issue_disconnect = 1;
3023                 } else {
3024                         nesqp->iwarp_state = next_iwarp_state & NES_CQP_QP_IWARP_STATE_MASK;
3025                         nes_debug(NES_DBG_MOD_QP, "Change nesqp->iwarp_state=%08x\n",
3026                                         nesqp->iwarp_state);
3027                 }
3028         }
3029
3030         if (attr_mask & IB_QP_ACCESS_FLAGS) {
3031                 if (attr->qp_access_flags & IB_ACCESS_LOCAL_WRITE) {
3032                         nesqp->nesqp_context->misc |= cpu_to_le32(NES_QPCONTEXT_MISC_RDMA_WRITE_EN |
3033                                         NES_QPCONTEXT_MISC_RDMA_READ_EN);
3034                         issue_modify_qp = 1;
3035                 }
3036                 if (attr->qp_access_flags & IB_ACCESS_REMOTE_WRITE) {
3037                         nesqp->nesqp_context->misc |= cpu_to_le32(NES_QPCONTEXT_MISC_RDMA_WRITE_EN);
3038                         issue_modify_qp = 1;
3039                 }
3040                 if (attr->qp_access_flags & IB_ACCESS_REMOTE_READ) {
3041                         nesqp->nesqp_context->misc |= cpu_to_le32(NES_QPCONTEXT_MISC_RDMA_READ_EN);
3042                         issue_modify_qp = 1;
3043                 }
3044                 if (attr->qp_access_flags & IB_ACCESS_MW_BIND) {
3045                         nesqp->nesqp_context->misc |= cpu_to_le32(NES_QPCONTEXT_MISC_WBIND_EN);
3046                         issue_modify_qp = 1;
3047                 }
3048
3049                 if (nesqp->user_mode) {
3050                         nesqp->nesqp_context->misc |= cpu_to_le32(NES_QPCONTEXT_MISC_RDMA_WRITE_EN |
3051                                         NES_QPCONTEXT_MISC_RDMA_READ_EN);
3052                         issue_modify_qp = 1;
3053                 }
3054         }
3055
3056         original_last_aeq = nesqp->last_aeq;
3057         spin_unlock_irqrestore(&nesqp->lock, qplockflags);
3058
3059         nes_debug(NES_DBG_MOD_QP, "issue_modify_qp=%u\n", issue_modify_qp);
3060
3061         ret = 0;
3062
3063
3064         if (issue_modify_qp) {
3065                 nes_debug(NES_DBG_MOD_QP, "call nes_hw_modify_qp\n");
3066                 ret = nes_hw_modify_qp(nesdev, nesqp, next_iwarp_state, 1);
3067                 if (ret)
3068                         nes_debug(NES_DBG_MOD_QP, "nes_hw_modify_qp (next_iwarp_state = 0x%08X)"
3069                                         " failed for QP%u.\n",
3070                                         next_iwarp_state, nesqp->hwqp.qp_id);
3071
3072         }
3073
3074         if ((issue_modify_qp) && (nesqp->ibqp_state > IB_QPS_RTS)) {
3075                 nes_debug(NES_DBG_MOD_QP, "QP%u Issued ModifyQP refcount (%d),"
3076                                 " original_last_aeq = 0x%04X. last_aeq = 0x%04X.\n",
3077                                 nesqp->hwqp.qp_id, atomic_read(&nesqp->refcount),
3078                                 original_last_aeq, nesqp->last_aeq);
3079                 if ((!ret) ||
3080                                 ((original_last_aeq != NES_AEQE_AEID_RDMAP_ROE_BAD_LLP_CLOSE) &&
3081                                 (ret))) {
3082                         if (dont_wait) {
3083                                 if (nesqp->cm_id && nesqp->hw_tcp_state != 0) {
3084                                         nes_debug(NES_DBG_MOD_QP, "QP%u Queuing fake disconnect for QP refcount (%d),"
3085                                                         " original_last_aeq = 0x%04X. last_aeq = 0x%04X.\n",
3086                                                         nesqp->hwqp.qp_id, atomic_read(&nesqp->refcount),
3087                                                         original_last_aeq, nesqp->last_aeq);
3088                                         /* this one is for the cm_disconnect thread */
3089                                         spin_lock_irqsave(&nesqp->lock, qplockflags);
3090                                         nesqp->hw_tcp_state = NES_AEQE_TCP_STATE_CLOSED;
3091                                         nesqp->last_aeq = NES_AEQE_AEID_RESET_SENT;
3092                                         spin_unlock_irqrestore(&nesqp->lock, qplockflags);
3093                                         nes_cm_disconn(nesqp);
3094                                 } else {
3095                                         nes_debug(NES_DBG_MOD_QP, "QP%u No fake disconnect, QP refcount=%d\n",
3096                                                         nesqp->hwqp.qp_id, atomic_read(&nesqp->refcount));
3097                                 }
3098                         } else {
3099                                 spin_lock_irqsave(&nesqp->lock, qplockflags);
3100                                 if (nesqp->cm_id) {
3101                                         /* These two are for the timer thread */
3102                                         if (atomic_inc_return(&nesqp->close_timer_started) == 1) {
3103                                                 nesqp->cm_id->add_ref(nesqp->cm_id);
3104                                                 nes_debug(NES_DBG_MOD_QP, "QP%u Not decrementing QP refcount (%d),"
3105                                                                 " need ae to finish up, original_last_aeq = 0x%04X."
3106                                                                 " last_aeq = 0x%04X, scheduling timer.\n",
3107                                                                 nesqp->hwqp.qp_id, atomic_read(&nesqp->refcount),
3108                                                                 original_last_aeq, nesqp->last_aeq);
3109                                                 schedule_nes_timer(nesqp->cm_node, (struct sk_buff *) nesqp, NES_TIMER_TYPE_CLOSE, 1, 0);
3110                                         }
3111                                         spin_unlock_irqrestore(&nesqp->lock, qplockflags);
3112                                 } else {
3113                                         spin_unlock_irqrestore(&nesqp->lock, qplockflags);
3114                                         nes_debug(NES_DBG_MOD_QP, "QP%u Not decrementing QP refcount (%d),"
3115                                                         " need ae to finish up, original_last_aeq = 0x%04X."
3116                                                         " last_aeq = 0x%04X.\n",
3117                                                         nesqp->hwqp.qp_id, atomic_read(&nesqp->refcount),
3118                                                         original_last_aeq, nesqp->last_aeq);
3119                                 }
3120                         }
3121                 } else {
3122                         nes_debug(NES_DBG_MOD_QP, "QP%u Decrementing QP refcount (%d), No ae to finish up,"
3123                                         " original_last_aeq = 0x%04X. last_aeq = 0x%04X.\n",
3124                                         nesqp->hwqp.qp_id, atomic_read(&nesqp->refcount),
3125                                         original_last_aeq, nesqp->last_aeq);
3126                 }
3127         } else {
3128                 nes_debug(NES_DBG_MOD_QP, "QP%u Decrementing QP refcount (%d), No ae to finish up,"
3129                                 " original_last_aeq = 0x%04X. last_aeq = 0x%04X.\n",
3130                                 nesqp->hwqp.qp_id, atomic_read(&nesqp->refcount),
3131                                 original_last_aeq, nesqp->last_aeq);
3132         }
3133
3134         err = 0;
3135
3136         nes_debug(NES_DBG_MOD_QP, "QP%u Leaving, refcount=%d\n",
3137                         nesqp->hwqp.qp_id, atomic_read(&nesqp->refcount));
3138
3139         return err;
3140 }
3141
3142
3143 /**
3144  * nes_muticast_attach
3145  */
3146 static int nes_multicast_attach(struct ib_qp *ibqp, union ib_gid *gid, u16 lid)
3147 {
3148         nes_debug(NES_DBG_INIT, "\n");
3149         return -ENOSYS;
3150 }
3151
3152
3153 /**
3154  * nes_multicast_detach
3155  */
3156 static int nes_multicast_detach(struct ib_qp *ibqp, union ib_gid *gid, u16 lid)
3157 {
3158         nes_debug(NES_DBG_INIT, "\n");
3159         return -ENOSYS;
3160 }
3161
3162
3163 /**
3164  * nes_process_mad
3165  */
3166 static int nes_process_mad(struct ib_device *ibdev, int mad_flags,
3167                 u8 port_num, struct ib_wc *in_wc, struct ib_grh *in_grh,
3168                 struct ib_mad *in_mad, struct ib_mad *out_mad)
3169 {
3170         nes_debug(NES_DBG_INIT, "\n");
3171         return -ENOSYS;
3172 }
3173
3174 static inline void
3175 fill_wqe_sg_send(struct nes_hw_qp_wqe *wqe, struct ib_send_wr *ib_wr, u32 uselkey)
3176 {
3177         int sge_index;
3178         int total_payload_length = 0;
3179         for (sge_index = 0; sge_index < ib_wr->num_sge; sge_index++) {
3180                 set_wqe_64bit_value(wqe->wqe_words, NES_IWARP_SQ_WQE_FRAG0_LOW_IDX+(sge_index*4),
3181                         ib_wr->sg_list[sge_index].addr);
3182                 set_wqe_32bit_value(wqe->wqe_words, NES_IWARP_SQ_WQE_LENGTH0_IDX + (sge_index*4),
3183                         ib_wr->sg_list[sge_index].length);
3184                 if (uselkey)
3185                         set_wqe_32bit_value(wqe->wqe_words, NES_IWARP_SQ_WQE_STAG0_IDX + (sge_index*4),
3186                                                 (ib_wr->sg_list[sge_index].lkey));
3187                 else
3188                         set_wqe_32bit_value(wqe->wqe_words, NES_IWARP_SQ_WQE_STAG0_IDX + (sge_index*4), 0);
3189
3190                 total_payload_length += ib_wr->sg_list[sge_index].length;
3191         }
3192         nes_debug(NES_DBG_IW_TX, "UC UC UC, sending total_payload_length=%u \n",
3193                         total_payload_length);
3194         set_wqe_32bit_value(wqe->wqe_words, NES_IWARP_SQ_WQE_TOTAL_PAYLOAD_IDX,
3195                                 total_payload_length);
3196 }
3197
3198 /**
3199  * nes_post_send
3200  */
3201 static int nes_post_send(struct ib_qp *ibqp, struct ib_send_wr *ib_wr,
3202                 struct ib_send_wr **bad_wr)
3203 {
3204         u64 u64temp;
3205         unsigned long flags = 0;
3206         struct nes_vnic *nesvnic = to_nesvnic(ibqp->device);
3207         struct nes_device *nesdev = nesvnic->nesdev;
3208         struct nes_qp *nesqp = to_nesqp(ibqp);
3209         struct nes_hw_qp_wqe *wqe;
3210         int err;
3211         u32 qsize = nesqp->hwqp.sq_size;
3212         u32 head;
3213         u32 wqe_misc;
3214         u32 wqe_count;
3215         u32 counter;
3216         u32 total_payload_length;
3217
3218         err = 0;
3219         wqe_misc = 0;
3220         wqe_count = 0;
3221         total_payload_length = 0;
3222
3223         if (nesqp->ibqp_state > IB_QPS_RTS)
3224                 return -EINVAL;
3225
3226         spin_lock_irqsave(&nesqp->lock, flags);
3227
3228         head = nesqp->hwqp.sq_head;
3229
3230         while (ib_wr) {
3231                 /* Check for SQ overflow */
3232                 if (((head + (2 * qsize) - nesqp->hwqp.sq_tail) % qsize) == (qsize - 1)) {
3233                         err = -EINVAL;
3234                         break;
3235                 }
3236
3237                 wqe = &nesqp->hwqp.sq_vbase[head];
3238                 /* nes_debug(NES_DBG_IW_TX, "processing sq wqe for QP%u at %p, head = %u.\n",
3239                                 nesqp->hwqp.qp_id, wqe, head); */
3240                 nes_fill_init_qp_wqe(wqe, nesqp, head);
3241                 u64temp = (u64)(ib_wr->wr_id);
3242                 set_wqe_64bit_value(wqe->wqe_words, NES_IWARP_SQ_WQE_COMP_SCRATCH_LOW_IDX,
3243                                         u64temp);
3244                         switch (ib_wr->opcode) {
3245                                 case IB_WR_SEND:
3246                                         if (ib_wr->send_flags & IB_SEND_SOLICITED) {
3247                                                 wqe_misc = NES_IWARP_SQ_OP_SENDSE;
3248                                         } else {
3249                                                 wqe_misc = NES_IWARP_SQ_OP_SEND;
3250                                         }
3251                                         if (ib_wr->num_sge > nesdev->nesadapter->max_sge) {
3252                                                 err = -EINVAL;
3253                                                 break;
3254                                         }
3255                                         if (ib_wr->send_flags & IB_SEND_FENCE) {
3256                                                 wqe_misc |= NES_IWARP_SQ_WQE_LOCAL_FENCE;
3257                                         }
3258                                         if ((ib_wr->send_flags & IB_SEND_INLINE) &&
3259                                                         ((nes_drv_opt & NES_DRV_OPT_NO_INLINE_DATA) == 0) &&
3260                                                         (ib_wr->sg_list[0].length <= 64)) {
3261                                                 memcpy(&wqe->wqe_words[NES_IWARP_SQ_WQE_IMM_DATA_START_IDX],
3262                                                                (void *)(unsigned long)ib_wr->sg_list[0].addr, ib_wr->sg_list[0].length);
3263                                                 set_wqe_32bit_value(wqe->wqe_words, NES_IWARP_SQ_WQE_TOTAL_PAYLOAD_IDX,
3264                                                                 ib_wr->sg_list[0].length);
3265                                                 wqe_misc |= NES_IWARP_SQ_WQE_IMM_DATA;
3266                                         } else {
3267                                                 fill_wqe_sg_send(wqe, ib_wr, 1);
3268                                         }
3269
3270                                         break;
3271                                 case IB_WR_RDMA_WRITE:
3272                                         wqe_misc = NES_IWARP_SQ_OP_RDMAW;
3273                                         if (ib_wr->num_sge > nesdev->nesadapter->max_sge) {
3274                                                 nes_debug(NES_DBG_IW_TX, "Exceeded max sge, ib_wr=%u, max=%u\n",
3275                                                                 ib_wr->num_sge,
3276                                                                 nesdev->nesadapter->max_sge);
3277                                                 err = -EINVAL;
3278                                                 break;
3279                                         }
3280                                         if (ib_wr->send_flags & IB_SEND_FENCE) {
3281                                                 wqe_misc |= NES_IWARP_SQ_WQE_LOCAL_FENCE;
3282                                         }
3283
3284                                         set_wqe_32bit_value(wqe->wqe_words, NES_IWARP_SQ_WQE_RDMA_STAG_IDX,
3285                                                         ib_wr->wr.rdma.rkey);
3286                                         set_wqe_64bit_value(wqe->wqe_words, NES_IWARP_SQ_WQE_RDMA_TO_LOW_IDX,
3287                                                         ib_wr->wr.rdma.remote_addr);
3288
3289                                         if ((ib_wr->send_flags & IB_SEND_INLINE) &&
3290                                                         ((nes_drv_opt & NES_DRV_OPT_NO_INLINE_DATA) == 0) &&
3291                                                         (ib_wr->sg_list[0].length <= 64)) {
3292                                                 memcpy(&wqe->wqe_words[NES_IWARP_SQ_WQE_IMM_DATA_START_IDX],
3293                                                                (void *)(unsigned long)ib_wr->sg_list[0].addr, ib_wr->sg_list[0].length);
3294                                                 set_wqe_32bit_value(wqe->wqe_words, NES_IWARP_SQ_WQE_TOTAL_PAYLOAD_IDX,
3295                                                                 ib_wr->sg_list[0].length);
3296                                                 wqe_misc |= NES_IWARP_SQ_WQE_IMM_DATA;
3297                                         } else {
3298                                                 fill_wqe_sg_send(wqe, ib_wr, 1);
3299                                         }
3300                                         wqe->wqe_words[NES_IWARP_SQ_WQE_RDMA_LENGTH_IDX] =
3301                                                         wqe->wqe_words[NES_IWARP_SQ_WQE_TOTAL_PAYLOAD_IDX];
3302                                         break;
3303                                 case IB_WR_RDMA_READ:
3304                                         /* iWARP only supports 1 sge for RDMA reads */
3305                                         if (ib_wr->num_sge > 1) {
3306                                                 nes_debug(NES_DBG_IW_TX, "Exceeded max sge, ib_wr=%u, max=1\n",
3307                                                                 ib_wr->num_sge);
3308                                                 err = -EINVAL;
3309                                                 break;
3310                                         }
3311                                         wqe_misc = NES_IWARP_SQ_OP_RDMAR;
3312                                         set_wqe_64bit_value(wqe->wqe_words, NES_IWARP_SQ_WQE_RDMA_TO_LOW_IDX,
3313                                                         ib_wr->wr.rdma.remote_addr);
3314                                         set_wqe_32bit_value(wqe->wqe_words, NES_IWARP_SQ_WQE_RDMA_STAG_IDX,
3315                                                         ib_wr->wr.rdma.rkey);
3316                                         set_wqe_32bit_value(wqe->wqe_words, NES_IWARP_SQ_WQE_RDMA_LENGTH_IDX,
3317                                                         ib_wr->sg_list->length);
3318                                         set_wqe_64bit_value(wqe->wqe_words, NES_IWARP_SQ_WQE_FRAG0_LOW_IDX,
3319                                                         ib_wr->sg_list->addr);
3320                                         set_wqe_32bit_value(wqe->wqe_words, NES_IWARP_SQ_WQE_STAG0_IDX,
3321                                                         ib_wr->sg_list->lkey);
3322                                         break;
3323                                 default:
3324                                         /* error */
3325                                         err = -EINVAL;
3326                                         break;
3327                         }
3328
3329                 if (ib_wr->send_flags & IB_SEND_SIGNALED) {
3330                         wqe_misc |= NES_IWARP_SQ_WQE_SIGNALED_COMPL;
3331                 }
3332                 wqe->wqe_words[NES_IWARP_SQ_WQE_MISC_IDX] = cpu_to_le32(wqe_misc);
3333
3334                 ib_wr = ib_wr->next;
3335                 head++;
3336                 wqe_count++;
3337                 if (head >= qsize)
3338                         head = 0;
3339
3340         }
3341
3342         nesqp->hwqp.sq_head = head;
3343         barrier();
3344         while (wqe_count) {
3345                 counter = min(wqe_count, ((u32)255));
3346                 wqe_count -= counter;
3347                 nes_write32(nesdev->regs + NES_WQE_ALLOC,
3348                                 (counter << 24) | 0x00800000 | nesqp->hwqp.qp_id);
3349         }
3350
3351         spin_unlock_irqrestore(&nesqp->lock, flags);
3352
3353         if (err)
3354                 *bad_wr = ib_wr;
3355         return err;
3356 }
3357
3358
3359 /**
3360  * nes_post_recv
3361  */
3362 static int nes_post_recv(struct ib_qp *ibqp, struct ib_recv_wr *ib_wr,
3363                 struct ib_recv_wr **bad_wr)
3364 {
3365         u64 u64temp;
3366         unsigned long flags = 0;
3367         struct nes_vnic *nesvnic = to_nesvnic(ibqp->device);
3368         struct nes_device *nesdev = nesvnic->nesdev;
3369         struct nes_qp *nesqp = to_nesqp(ibqp);
3370         struct nes_hw_qp_wqe *wqe;
3371         int err = 0;
3372         int sge_index;
3373         u32 qsize = nesqp->hwqp.rq_size;
3374         u32 head;
3375         u32 wqe_count = 0;
3376         u32 counter;
3377         u32 total_payload_length;
3378
3379         if (nesqp->ibqp_state > IB_QPS_RTS)
3380                 return -EINVAL;
3381
3382         spin_lock_irqsave(&nesqp->lock, flags);
3383
3384         head = nesqp->hwqp.rq_head;
3385
3386         while (ib_wr) {
3387                 if (ib_wr->num_sge > nesdev->nesadapter->max_sge) {
3388                         err = -EINVAL;
3389                         break;
3390                 }
3391                 /* Check for RQ overflow */
3392                 if (((head + (2 * qsize) - nesqp->hwqp.rq_tail) % qsize) == (qsize - 1)) {
3393                         err = -EINVAL;
3394                         break;
3395                 }
3396
3397                 nes_debug(NES_DBG_IW_RX, "ibwr sge count = %u.\n", ib_wr->num_sge);
3398                 wqe = &nesqp->hwqp.rq_vbase[head];
3399
3400                 /* nes_debug(NES_DBG_IW_RX, "QP%u:processing rq wqe at %p, head = %u.\n",
3401                                 nesqp->hwqp.qp_id, wqe, head); */
3402                 nes_fill_init_qp_wqe(wqe, nesqp, head);
3403                 u64temp = (u64)(ib_wr->wr_id);
3404                 set_wqe_64bit_value(wqe->wqe_words, NES_IWARP_SQ_WQE_COMP_SCRATCH_LOW_IDX,
3405                                         u64temp);
3406                 total_payload_length = 0;
3407                 for (sge_index=0; sge_index < ib_wr->num_sge; sge_index++) {
3408                         set_wqe_64bit_value(wqe->wqe_words, NES_IWARP_RQ_WQE_FRAG0_LOW_IDX+(sge_index*4),
3409                                         ib_wr->sg_list[sge_index].addr);
3410                         set_wqe_32bit_value(wqe->wqe_words, NES_IWARP_RQ_WQE_LENGTH0_IDX+(sge_index*4),
3411                                         ib_wr->sg_list[sge_index].length);
3412                         set_wqe_32bit_value(wqe->wqe_words,NES_IWARP_RQ_WQE_STAG0_IDX+(sge_index*4),
3413                                         ib_wr->sg_list[sge_index].lkey);
3414
3415                         total_payload_length += ib_wr->sg_list[sge_index].length;
3416                 }
3417                 set_wqe_32bit_value(wqe->wqe_words, NES_IWARP_RQ_WQE_TOTAL_PAYLOAD_IDX,
3418                                         total_payload_length);
3419
3420                 ib_wr = ib_wr->next;
3421                 head++;
3422                 wqe_count++;
3423                 if (head >= qsize)
3424                         head = 0;
3425         }
3426
3427         nesqp->hwqp.rq_head = head;
3428         barrier();
3429         while (wqe_count) {
3430                 counter = min(wqe_count, ((u32)255));
3431                 wqe_count -= counter;
3432                 nes_write32(nesdev->regs+NES_WQE_ALLOC, (counter<<24) | nesqp->hwqp.qp_id);
3433         }
3434
3435         spin_unlock_irqrestore(&nesqp->lock, flags);
3436
3437         if (err)
3438                 *bad_wr = ib_wr;
3439         return err;
3440 }
3441
3442
3443 /**
3444  * nes_poll_cq
3445  */
3446 static int nes_poll_cq(struct ib_cq *ibcq, int num_entries, struct ib_wc *entry)
3447 {
3448         u64 u64temp;
3449         u64 wrid;
3450         /* u64 u64temp; */
3451         unsigned long flags = 0;
3452         struct nes_vnic *nesvnic = to_nesvnic(ibcq->device);
3453         struct nes_device *nesdev = nesvnic->nesdev;
3454         struct nes_cq *nescq = to_nescq(ibcq);
3455         struct nes_qp *nesqp;
3456         struct nes_hw_cqe cqe;
3457         u32 head;
3458         u32 wq_tail;
3459         u32 cq_size;
3460         u32 cqe_count = 0;
3461         u32 wqe_index;
3462         u32 u32temp;
3463         /* u32 counter; */
3464
3465         nes_debug(NES_DBG_CQ, "\n");
3466
3467         spin_lock_irqsave(&nescq->lock, flags);
3468
3469         head = nescq->hw_cq.cq_head;
3470         cq_size = nescq->hw_cq.cq_size;
3471
3472         while (cqe_count < num_entries) {
3473                 if (le32_to_cpu(nescq->hw_cq.cq_vbase[head].cqe_words[NES_CQE_OPCODE_IDX]) &
3474                                 NES_CQE_VALID) {
3475                         /*
3476                          * Make sure we read CQ entry contents *after*
3477                          * we've checked the valid bit.
3478                          */
3479                         rmb();
3480
3481                         cqe = nescq->hw_cq.cq_vbase[head];
3482                         nescq->hw_cq.cq_vbase[head].cqe_words[NES_CQE_OPCODE_IDX] = 0;
3483                         u32temp = le32_to_cpu(cqe.cqe_words[NES_CQE_COMP_COMP_CTX_LOW_IDX]);
3484                         wqe_index = u32temp &
3485                                         (nesdev->nesadapter->max_qp_wr - 1);
3486                         u32temp &= ~(NES_SW_CONTEXT_ALIGN-1);
3487                         /* parse CQE, get completion context from WQE (either rq or sq */
3488                         u64temp = (((u64)(le32_to_cpu(cqe.cqe_words[NES_CQE_COMP_COMP_CTX_HIGH_IDX])))<<32) |
3489                                         ((u64)u32temp);
3490                         nesqp = *((struct nes_qp **)&u64temp);
3491                         memset(entry, 0, sizeof *entry);
3492                         if (cqe.cqe_words[NES_CQE_ERROR_CODE_IDX] == 0) {
3493                                 entry->status = IB_WC_SUCCESS;
3494                         } else {
3495                                 entry->status = IB_WC_WR_FLUSH_ERR;
3496                         }
3497
3498                         entry->qp = &nesqp->ibqp;
3499                         entry->src_qp = nesqp->hwqp.qp_id;
3500
3501                         if (le32_to_cpu(cqe.cqe_words[NES_CQE_OPCODE_IDX]) & NES_CQE_SQ) {
3502                                 if (nesqp->skip_lsmm) {
3503                                         nesqp->skip_lsmm = 0;
3504                                         wq_tail = nesqp->hwqp.sq_tail++;
3505                                 }
3506
3507                                 /* Working on a SQ Completion*/
3508                                 wq_tail = wqe_index;
3509                                 nesqp->hwqp.sq_tail = (wqe_index+1)&(nesqp->hwqp.sq_size - 1);
3510                                 wrid = (((u64)(cpu_to_le32((u32)nesqp->hwqp.sq_vbase[wq_tail].
3511                                                 wqe_words[NES_IWARP_SQ_WQE_COMP_SCRATCH_HIGH_IDX]))) << 32) |
3512                                                 ((u64)(cpu_to_le32((u32)nesqp->hwqp.sq_vbase[wq_tail].
3513                                                 wqe_words[NES_IWARP_SQ_WQE_COMP_SCRATCH_LOW_IDX])));
3514                                 entry->byte_len = le32_to_cpu(nesqp->hwqp.sq_vbase[wq_tail].
3515                                                 wqe_words[NES_IWARP_SQ_WQE_TOTAL_PAYLOAD_IDX]);
3516
3517                                 switch (le32_to_cpu(nesqp->hwqp.sq_vbase[wq_tail].
3518                                                 wqe_words[NES_IWARP_SQ_WQE_MISC_IDX]) & 0x3f) {
3519                                         case NES_IWARP_SQ_OP_RDMAW:
3520                                                 nes_debug(NES_DBG_CQ, "Operation = RDMA WRITE.\n");
3521                                                 entry->opcode = IB_WC_RDMA_WRITE;
3522                                                 break;
3523                                         case NES_IWARP_SQ_OP_RDMAR:
3524                                                 nes_debug(NES_DBG_CQ, "Operation = RDMA READ.\n");
3525                                                 entry->opcode = IB_WC_RDMA_READ;
3526                                                 entry->byte_len = le32_to_cpu(nesqp->hwqp.sq_vbase[wq_tail].
3527                                                                 wqe_words[NES_IWARP_SQ_WQE_RDMA_LENGTH_IDX]);
3528                                                 break;
3529                                         case NES_IWARP_SQ_OP_SENDINV:
3530                                         case NES_IWARP_SQ_OP_SENDSEINV:
3531                                         case NES_IWARP_SQ_OP_SEND:
3532                                         case NES_IWARP_SQ_OP_SENDSE:
3533                                                 nes_debug(NES_DBG_CQ, "Operation = Send.\n");
3534                                                 entry->opcode = IB_WC_SEND;
3535                                                 break;
3536                                 }
3537                         } else {
3538                                 /* Working on a RQ Completion*/
3539                                 wq_tail = wqe_index;
3540                                         nesqp->hwqp.rq_tail = (wqe_index+1)&(nesqp->hwqp.rq_size - 1);
3541                                 entry->byte_len = le32_to_cpu(cqe.cqe_words[NES_CQE_PAYLOAD_LENGTH_IDX]);
3542                                 wrid = ((u64)(le32_to_cpu(nesqp->hwqp.rq_vbase[wq_tail].wqe_words[NES_IWARP_RQ_WQE_COMP_SCRATCH_LOW_IDX]))) |
3543                                         ((u64)(le32_to_cpu(nesqp->hwqp.rq_vbase[wq_tail].wqe_words[NES_IWARP_RQ_WQE_COMP_SCRATCH_HIGH_IDX]))<<32);
3544                                         entry->opcode = IB_WC_RECV;
3545                         }
3546                         entry->wr_id = wrid;
3547
3548                         if (++head >= cq_size)
3549                                 head = 0;
3550                         cqe_count++;
3551                         nescq->polled_completions++;
3552                         if ((nescq->polled_completions > (cq_size / 2)) ||
3553                                         (nescq->polled_completions == 255)) {
3554                                 nes_debug(NES_DBG_CQ, "CQ%u Issuing CQE Allocate since more than half of cqes"
3555                                                 " are pending %u of %u.\n",
3556                                                 nescq->hw_cq.cq_number, nescq->polled_completions, cq_size);
3557                                 nes_write32(nesdev->regs+NES_CQE_ALLOC,
3558                                                 nescq->hw_cq.cq_number | (nescq->polled_completions << 16));
3559                                 nescq->polled_completions = 0;
3560                         }
3561                         entry++;
3562                 } else
3563                         break;
3564         }
3565
3566         if (nescq->polled_completions) {
3567                 nes_write32(nesdev->regs+NES_CQE_ALLOC,
3568                                 nescq->hw_cq.cq_number | (nescq->polled_completions << 16));
3569                 nescq->polled_completions = 0;
3570         }
3571
3572         nescq->hw_cq.cq_head = head;
3573         nes_debug(NES_DBG_CQ, "Reporting %u completions for CQ%u.\n",
3574                         cqe_count, nescq->hw_cq.cq_number);
3575
3576         spin_unlock_irqrestore(&nescq->lock, flags);
3577
3578         return cqe_count;
3579 }
3580
3581
3582 /**
3583  * nes_req_notify_cq
3584  */
3585 static int nes_req_notify_cq(struct ib_cq *ibcq, enum ib_cq_notify_flags notify_flags)
3586                 {
3587         struct nes_vnic *nesvnic = to_nesvnic(ibcq->device);
3588         struct nes_device *nesdev = nesvnic->nesdev;
3589         struct nes_cq *nescq = to_nescq(ibcq);
3590         u32 cq_arm;
3591
3592         nes_debug(NES_DBG_CQ, "Requesting notification for CQ%u.\n",
3593                         nescq->hw_cq.cq_number);
3594
3595         cq_arm = nescq->hw_cq.cq_number;
3596         if ((notify_flags & IB_CQ_SOLICITED_MASK) == IB_CQ_NEXT_COMP)
3597                 cq_arm |= NES_CQE_ALLOC_NOTIFY_NEXT;
3598         else if ((notify_flags & IB_CQ_SOLICITED_MASK) == IB_CQ_SOLICITED)
3599                 cq_arm |= NES_CQE_ALLOC_NOTIFY_SE;
3600         else
3601                 return -EINVAL;
3602
3603         nes_write32(nesdev->regs+NES_CQE_ALLOC, cq_arm);
3604         nes_read32(nesdev->regs+NES_CQE_ALLOC);
3605
3606         return 0;
3607 }
3608
3609
3610 /**
3611  * nes_init_ofa_device
3612  */
3613 struct nes_ib_device *nes_init_ofa_device(struct net_device *netdev)
3614 {
3615         struct nes_ib_device *nesibdev;
3616         struct nes_vnic *nesvnic = netdev_priv(netdev);
3617         struct nes_device *nesdev = nesvnic->nesdev;
3618
3619         nesibdev = (struct nes_ib_device *)ib_alloc_device(sizeof(struct nes_ib_device));
3620         if (nesibdev == NULL) {
3621                 return NULL;
3622         }
3623         strlcpy(nesibdev->ibdev.name, "nes%d", IB_DEVICE_NAME_MAX);
3624         nesibdev->ibdev.owner = THIS_MODULE;
3625
3626         nesibdev->ibdev.node_type = RDMA_NODE_RNIC;
3627         memset(&nesibdev->ibdev.node_guid, 0, sizeof(nesibdev->ibdev.node_guid));
3628         memcpy(&nesibdev->ibdev.node_guid, netdev->dev_addr, 6);
3629
3630         nesibdev->ibdev.uverbs_cmd_mask =
3631                         (1ull << IB_USER_VERBS_CMD_GET_CONTEXT) |
3632                         (1ull << IB_USER_VERBS_CMD_QUERY_DEVICE) |
3633                         (1ull << IB_USER_VERBS_CMD_QUERY_PORT) |
3634                         (1ull << IB_USER_VERBS_CMD_ALLOC_PD) |
3635                         (1ull << IB_USER_VERBS_CMD_DEALLOC_PD) |
3636                         (1ull << IB_USER_VERBS_CMD_REG_MR) |
3637                         (1ull << IB_USER_VERBS_CMD_DEREG_MR) |
3638                         (1ull << IB_USER_VERBS_CMD_CREATE_COMP_CHANNEL) |
3639                         (1ull << IB_USER_VERBS_CMD_CREATE_CQ) |
3640                         (1ull << IB_USER_VERBS_CMD_DESTROY_CQ) |
3641                         (1ull << IB_USER_VERBS_CMD_CREATE_AH) |
3642                         (1ull << IB_USER_VERBS_CMD_DESTROY_AH) |
3643                         (1ull << IB_USER_VERBS_CMD_REQ_NOTIFY_CQ) |
3644                         (1ull << IB_USER_VERBS_CMD_CREATE_QP) |
3645                         (1ull << IB_USER_VERBS_CMD_MODIFY_QP) |
3646                         (1ull << IB_USER_VERBS_CMD_POLL_CQ) |
3647                         (1ull << IB_USER_VERBS_CMD_DESTROY_QP) |
3648                         (1ull << IB_USER_VERBS_CMD_ALLOC_MW) |
3649                         (1ull << IB_USER_VERBS_CMD_BIND_MW) |
3650                         (1ull << IB_USER_VERBS_CMD_DEALLOC_MW) |
3651                         (1ull << IB_USER_VERBS_CMD_POST_RECV) |
3652                         (1ull << IB_USER_VERBS_CMD_POST_SEND);
3653
3654         nesibdev->ibdev.phys_port_cnt = 1;
3655         nesibdev->ibdev.num_comp_vectors = 1;
3656         nesibdev->ibdev.dma_device = &nesdev->pcidev->dev;
3657         nesibdev->ibdev.dev.parent = &nesdev->pcidev->dev;
3658         nesibdev->ibdev.query_device = nes_query_device;
3659         nesibdev->ibdev.query_port = nes_query_port;
3660         nesibdev->ibdev.modify_port = nes_modify_port;
3661         nesibdev->ibdev.query_pkey = nes_query_pkey;
3662         nesibdev->ibdev.query_gid = nes_query_gid;
3663         nesibdev->ibdev.alloc_ucontext = nes_alloc_ucontext;
3664         nesibdev->ibdev.dealloc_ucontext = nes_dealloc_ucontext;
3665         nesibdev->ibdev.mmap = nes_mmap;
3666         nesibdev->ibdev.alloc_pd = nes_alloc_pd;
3667         nesibdev->ibdev.dealloc_pd = nes_dealloc_pd;
3668         nesibdev->ibdev.create_ah = nes_create_ah;
3669         nesibdev->ibdev.destroy_ah = nes_destroy_ah;
3670         nesibdev->ibdev.create_qp = nes_create_qp;
3671         nesibdev->ibdev.modify_qp = nes_modify_qp;
3672         nesibdev->ibdev.query_qp = nes_query_qp;
3673         nesibdev->ibdev.destroy_qp = nes_destroy_qp;
3674         nesibdev->ibdev.create_cq = nes_create_cq;
3675         nesibdev->ibdev.destroy_cq = nes_destroy_cq;
3676         nesibdev->ibdev.poll_cq = nes_poll_cq;
3677         nesibdev->ibdev.get_dma_mr = nes_get_dma_mr;
3678         nesibdev->ibdev.reg_phys_mr = nes_reg_phys_mr;
3679         nesibdev->ibdev.reg_user_mr = nes_reg_user_mr;
3680         nesibdev->ibdev.dereg_mr = nes_dereg_mr;
3681         nesibdev->ibdev.alloc_mw = nes_alloc_mw;
3682         nesibdev->ibdev.dealloc_mw = nes_dealloc_mw;
3683         nesibdev->ibdev.bind_mw = nes_bind_mw;
3684
3685         nesibdev->ibdev.alloc_fmr = nes_alloc_fmr;
3686         nesibdev->ibdev.unmap_fmr = nes_unmap_fmr;
3687         nesibdev->ibdev.dealloc_fmr = nes_dealloc_fmr;
3688         nesibdev->ibdev.map_phys_fmr = nes_map_phys_fmr;
3689
3690         nesibdev->ibdev.attach_mcast = nes_multicast_attach;
3691         nesibdev->ibdev.detach_mcast = nes_multicast_detach;
3692         nesibdev->ibdev.process_mad = nes_process_mad;
3693
3694         nesibdev->ibdev.req_notify_cq = nes_req_notify_cq;
3695         nesibdev->ibdev.post_send = nes_post_send;
3696         nesibdev->ibdev.post_recv = nes_post_recv;
3697
3698         nesibdev->ibdev.iwcm = kzalloc(sizeof(*nesibdev->ibdev.iwcm), GFP_KERNEL);
3699         if (nesibdev->ibdev.iwcm == NULL) {
3700                 ib_dealloc_device(&nesibdev->ibdev);
3701                 return NULL;
3702         }
3703         nesibdev->ibdev.iwcm->add_ref = nes_add_ref;
3704         nesibdev->ibdev.iwcm->rem_ref = nes_rem_ref;
3705         nesibdev->ibdev.iwcm->get_qp = nes_get_qp;
3706         nesibdev->ibdev.iwcm->connect = nes_connect;
3707         nesibdev->ibdev.iwcm->accept = nes_accept;
3708         nesibdev->ibdev.iwcm->reject = nes_reject;
3709         nesibdev->ibdev.iwcm->create_listen = nes_create_listen;
3710         nesibdev->ibdev.iwcm->destroy_listen = nes_destroy_listen;
3711
3712         return nesibdev;
3713 }
3714
3715
3716 /**
3717  * nes_destroy_ofa_device
3718  */
3719 void nes_destroy_ofa_device(struct nes_ib_device *nesibdev)
3720 {
3721         if (nesibdev == NULL)
3722                 return;
3723
3724         nes_unregister_ofa_device(nesibdev);
3725
3726         kfree(nesibdev->ibdev.iwcm);
3727         ib_dealloc_device(&nesibdev->ibdev);
3728 }
3729
3730
3731 /**
3732  * nes_register_ofa_device
3733  */
3734 int nes_register_ofa_device(struct nes_ib_device *nesibdev)
3735 {
3736         struct nes_vnic *nesvnic = nesibdev->nesvnic;
3737         struct nes_device *nesdev = nesvnic->nesdev;
3738         struct nes_adapter *nesadapter = nesdev->nesadapter;
3739         int i, ret;
3740
3741         ret = ib_register_device(&nesvnic->nesibdev->ibdev);
3742         if (ret) {
3743                 return ret;
3744         }
3745
3746         /* Get the resources allocated to this device */
3747         nesibdev->max_cq = (nesadapter->max_cq-NES_FIRST_QPN) / nesadapter->port_count;
3748         nesibdev->max_mr = nesadapter->max_mr / nesadapter->port_count;
3749         nesibdev->max_qp = (nesadapter->max_qp-NES_FIRST_QPN) / nesadapter->port_count;
3750         nesibdev->max_pd = nesadapter->max_pd / nesadapter->port_count;
3751
3752         for (i = 0; i < ARRAY_SIZE(nes_dev_attributes); ++i) {
3753                 ret = device_create_file(&nesibdev->ibdev.dev, nes_dev_attributes[i]);
3754                 if (ret) {
3755                         while (i > 0) {
3756                                 i--;
3757                                 device_remove_file(&nesibdev->ibdev.dev,
3758                                                    nes_dev_attributes[i]);
3759                         }
3760                         ib_unregister_device(&nesibdev->ibdev);
3761                         return ret;
3762                 }
3763         }
3764
3765         nesvnic->of_device_registered = 1;
3766
3767         return 0;
3768 }
3769
3770
3771 /**
3772  * nes_unregister_ofa_device
3773  */
3774 static void nes_unregister_ofa_device(struct nes_ib_device *nesibdev)
3775 {
3776         struct nes_vnic *nesvnic = nesibdev->nesvnic;
3777         int i;
3778
3779         for (i = 0; i < ARRAY_SIZE(nes_dev_attributes); ++i) {
3780                 device_remove_file(&nesibdev->ibdev.dev, nes_dev_attributes[i]);
3781         }
3782
3783         if (nesvnic->of_device_registered) {
3784                 ib_unregister_device(&nesibdev->ibdev);
3785         }
3786
3787         nesvnic->of_device_registered = 0;
3788 }