file_ops.c 40.7 KB
Newer Older
M
Mike Marciniszyn 已提交
1
/*
2
 * Copyright(c) 2015-2017 Intel Corporation.
M
Mike Marciniszyn 已提交
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50
 *
 * This file is provided under a dual BSD/GPLv2 license.  When using or
 * redistributing this file, you may do so under either license.
 *
 * GPL LICENSE SUMMARY
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of version 2 of the GNU General Public License as
 * published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * BSD LICENSE
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 *  - Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 *  - Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in
 *    the documentation and/or other materials provided with the
 *    distribution.
 *  - Neither the name of Intel Corporation nor the names of its
 *    contributors may be used to endorse or promote products derived
 *    from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 */
#include <linux/poll.h>
#include <linux/cdev.h>
#include <linux/vmalloc.h>
#include <linux/io.h>
51
#include <linux/sched/mm.h>
52
#include <linux/bitmap.h>
M
Mike Marciniszyn 已提交
53

54 55
#include <rdma/ib.h>

M
Mike Marciniszyn 已提交
56 57 58 59 60 61
#include "hfi.h"
#include "pio.h"
#include "device.h"
#include "common.h"
#include "trace.h"
#include "user_sdma.h"
62
#include "user_exp_rcv.h"
63
#include "aspm.h"
64
#include "mmu_rb.h"
M
Mike Marciniszyn 已提交
65 66 67 68 69 70 71 72 73

#undef pr_fmt
#define pr_fmt(fmt) DRIVER_NAME ": " fmt

#define SEND_CTXT_HALT_TIMEOUT 1000 /* msecs */

/*
 * File operation functions
 */
74 75 76 77 78
static int hfi1_file_open(struct inode *inode, struct file *fp);
static int hfi1_file_close(struct inode *inode, struct file *fp);
static ssize_t hfi1_write_iter(struct kiocb *kiocb, struct iov_iter *from);
static unsigned int hfi1_poll(struct file *fp, struct poll_table_struct *pt);
static int hfi1_file_mmap(struct file *fp, struct vm_area_struct *vma);
M
Mike Marciniszyn 已提交
79

80
static u64 kvirt_to_phys(void *addr);
81
static int assign_ctxt(struct hfi1_filedata *fd, struct hfi1_user_info *uinfo);
82 83
static int init_subctxts(struct hfi1_ctxtdata *uctxt,
			 const struct hfi1_user_info *uinfo);
84
static int init_user_ctxt(struct hfi1_filedata *fd);
85
static void user_init(struct hfi1_ctxtdata *uctxt);
86 87 88 89
static int get_ctxt_info(struct hfi1_filedata *fd, void __user *ubase,
			 __u32 len);
static int get_base_info(struct hfi1_filedata *fd, void __user *ubase,
			 __u32 len);
90
static int setup_base_ctxt(struct hfi1_filedata *fd);
91
static int setup_subctxt(struct hfi1_ctxtdata *uctxt);
92

93 94
static int find_sub_ctxt(struct hfi1_filedata *fd,
			 const struct hfi1_user_info *uinfo);
95
static int allocate_ctxt(struct hfi1_filedata *fd, struct hfi1_devdata *dd,
96 97 98
			 struct hfi1_user_info *uinfo);
static unsigned int poll_urgent(struct file *fp, struct poll_table_struct *pt);
static unsigned int poll_next(struct file *fp, struct poll_table_struct *pt);
99
static int user_event_ack(struct hfi1_ctxtdata *uctxt, u16 subctxt,
100
			  unsigned long events);
101 102
static int set_ctxt_pkey(struct hfi1_ctxtdata *uctxt, u16 subctxt, u16 pkey);
static int manage_rcvq(struct hfi1_ctxtdata *uctxt, u16 subctxt,
103 104
		       int start_stop);
static int vma_fault(struct vm_fault *vmf);
105 106
static long hfi1_file_ioctl(struct file *fp, unsigned int cmd,
			    unsigned long arg);
M
Mike Marciniszyn 已提交
107 108 109 110 111 112

static const struct file_operations hfi1_file_ops = {
	.owner = THIS_MODULE,
	.write_iter = hfi1_write_iter,
	.open = hfi1_file_open,
	.release = hfi1_file_close,
113
	.unlocked_ioctl = hfi1_file_ioctl,
M
Mike Marciniszyn 已提交
114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166
	.poll = hfi1_poll,
	.mmap = hfi1_file_mmap,
	.llseek = noop_llseek,
};

static struct vm_operations_struct vm_ops = {
	.fault = vma_fault,
};

/*
 * Types of memories mapped into user processes' space
 */
enum mmap_types {
	PIO_BUFS = 1,
	PIO_BUFS_SOP,
	PIO_CRED,
	RCV_HDRQ,
	RCV_EGRBUF,
	UREGS,
	EVENTS,
	STATUS,
	RTAIL,
	SUBCTXT_UREGS,
	SUBCTXT_RCV_HDRQ,
	SUBCTXT_EGRBUF,
	SDMA_COMP
};

/*
 * Masks and offsets defining the mmap tokens
 */
#define HFI1_MMAP_OFFSET_MASK   0xfffULL
#define HFI1_MMAP_OFFSET_SHIFT  0
#define HFI1_MMAP_SUBCTXT_MASK  0xfULL
#define HFI1_MMAP_SUBCTXT_SHIFT 12
#define HFI1_MMAP_CTXT_MASK     0xffULL
#define HFI1_MMAP_CTXT_SHIFT    16
#define HFI1_MMAP_TYPE_MASK     0xfULL
#define HFI1_MMAP_TYPE_SHIFT    24
#define HFI1_MMAP_MAGIC_MASK    0xffffffffULL
#define HFI1_MMAP_MAGIC_SHIFT   32

#define HFI1_MMAP_MAGIC         0xdabbad00

#define HFI1_MMAP_TOKEN_SET(field, val)	\
	(((val) & HFI1_MMAP_##field##_MASK) << HFI1_MMAP_##field##_SHIFT)
#define HFI1_MMAP_TOKEN_GET(field, token) \
	(((token) >> HFI1_MMAP_##field##_SHIFT) & HFI1_MMAP_##field##_MASK)
#define HFI1_MMAP_TOKEN(type, ctxt, subctxt, addr)   \
	(HFI1_MMAP_TOKEN_SET(MAGIC, HFI1_MMAP_MAGIC) | \
	HFI1_MMAP_TOKEN_SET(TYPE, type) | \
	HFI1_MMAP_TOKEN_SET(CTXT, ctxt) | \
	HFI1_MMAP_TOKEN_SET(SUBCTXT, subctxt) | \
G
Geliang Tang 已提交
167
	HFI1_MMAP_TOKEN_SET(OFFSET, (offset_in_page(addr))))
M
Mike Marciniszyn 已提交
168 169 170 171 172 173 174 175 176 177 178

#define dbg(fmt, ...)				\
	pr_info(fmt, ##__VA_ARGS__)

static inline int is_valid_mmap(u64 token)
{
	return (HFI1_MMAP_TOKEN_GET(MAGIC, token) == HFI1_MMAP_MAGIC);
}

static int hfi1_file_open(struct inode *inode, struct file *fp)
{
I
Ira Weiny 已提交
179
	struct hfi1_filedata *fd;
180 181 182 183
	struct hfi1_devdata *dd = container_of(inode->i_cdev,
					       struct hfi1_devdata,
					       user_cdev);

184 185 186
	if (!((dd->flags & HFI1_PRESENT) && dd->kregbase))
		return -EINVAL;

187 188 189
	if (!atomic_inc_not_zero(&dd->user_refcount))
		return -ENXIO;

190 191 192
	/* Just take a ref now. Not all opens result in a context assign */
	kobject_get(&dd->kobj);

M
Mike Marciniszyn 已提交
193
	/* The real work is performed later in assign_ctxt() */
I
Ira Weiny 已提交
194 195 196

	fd = kzalloc(sizeof(*fd), GFP_KERNEL);

I
Ira Weiny 已提交
197 198 199
	if (fd) {
		fd->rec_cpu_num = -1; /* no cpu affinity by default */
		fd->mm = current->mm;
V
Vegard Nossum 已提交
200
		mmgrab(fd->mm);
201
		fd->dd = dd;
202 203 204 205 206 207
		fp->private_data = fd;
	} else {
		fp->private_data = NULL;

		if (atomic_dec_and_test(&dd->user_refcount))
			complete(&dd->user_comp);
I
Ira Weiny 已提交
208

209 210
		return -ENOMEM;
	}
I
Ira Weiny 已提交
211

212
	return 0;
M
Mike Marciniszyn 已提交
213 214
}

215 216 217 218 219 220 221 222 223 224 225 226 227
static long hfi1_file_ioctl(struct file *fp, unsigned int cmd,
			    unsigned long arg)
{
	struct hfi1_filedata *fd = fp->private_data;
	struct hfi1_ctxtdata *uctxt = fd->uctxt;
	struct hfi1_user_info uinfo;
	struct hfi1_tid_info tinfo;
	int ret = 0;
	unsigned long addr;
	int uval = 0;
	unsigned long ul_uval = 0;
	u16 uval16 = 0;

228
	hfi1_cdbg(IOCTL, "IOCTL recv: 0x%x", cmd);
229 230 231 232 233 234 235
	if (cmd != HFI1_IOCTL_ASSIGN_CTXT &&
	    cmd != HFI1_IOCTL_GET_VERS &&
	    !uctxt)
		return -EINVAL;

	switch (cmd) {
	case HFI1_IOCTL_ASSIGN_CTXT:
I
Ira Weiny 已提交
236 237 238
		if (uctxt)
			return -EINVAL;

239 240 241 242 243
		if (copy_from_user(&uinfo,
				   (struct hfi1_user_info __user *)arg,
				   sizeof(uinfo)))
			return -EFAULT;

244
		ret = assign_ctxt(fd, &uinfo);
245 246
		break;
	case HFI1_IOCTL_CTXT_INFO:
247
		ret = get_ctxt_info(fd, (void __user *)(unsigned long)arg,
248 249 250
				    sizeof(struct hfi1_ctxt_info));
		break;
	case HFI1_IOCTL_USER_INFO:
251
		ret = get_base_info(fd, (void __user *)(unsigned long)arg,
252 253 254
				    sizeof(struct hfi1_base_info));
		break;
	case HFI1_IOCTL_CREDIT_UPD:
255
		if (uctxt)
256 257 258 259 260 261 262 263 264
			sc_return_credits(uctxt->sc);
		break;

	case HFI1_IOCTL_TID_UPDATE:
		if (copy_from_user(&tinfo,
				   (struct hfi11_tid_info __user *)arg,
				   sizeof(tinfo)))
			return -EFAULT;

265
		ret = hfi1_user_exp_rcv_setup(fd, &tinfo);
266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286
		if (!ret) {
			/*
			 * Copy the number of tidlist entries we used
			 * and the length of the buffer we registered.
			 * These fields are adjacent in the structure so
			 * we can copy them at the same time.
			 */
			addr = arg + offsetof(struct hfi1_tid_info, tidcnt);
			if (copy_to_user((void __user *)addr, &tinfo.tidcnt,
					 sizeof(tinfo.tidcnt) +
					 sizeof(tinfo.length)))
				ret = -EFAULT;
		}
		break;

	case HFI1_IOCTL_TID_FREE:
		if (copy_from_user(&tinfo,
				   (struct hfi11_tid_info __user *)arg,
				   sizeof(tinfo)))
			return -EFAULT;

287
		ret = hfi1_user_exp_rcv_clear(fd, &tinfo);
288 289 290 291 292 293 294 295 296 297 298 299 300 301
		if (ret)
			break;
		addr = arg + offsetof(struct hfi1_tid_info, tidcnt);
		if (copy_to_user((void __user *)addr, &tinfo.tidcnt,
				 sizeof(tinfo.tidcnt)))
			ret = -EFAULT;
		break;

	case HFI1_IOCTL_TID_INVAL_READ:
		if (copy_from_user(&tinfo,
				   (struct hfi11_tid_info __user *)arg,
				   sizeof(tinfo)))
			return -EFAULT;

302
		ret = hfi1_user_exp_rcv_invalid(fd, &tinfo);
303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413
		if (ret)
			break;
		addr = arg + offsetof(struct hfi1_tid_info, tidcnt);
		if (copy_to_user((void __user *)addr, &tinfo.tidcnt,
				 sizeof(tinfo.tidcnt)))
			ret = -EFAULT;
		break;

	case HFI1_IOCTL_RECV_CTRL:
		ret = get_user(uval, (int __user *)arg);
		if (ret != 0)
			return -EFAULT;
		ret = manage_rcvq(uctxt, fd->subctxt, uval);
		break;

	case HFI1_IOCTL_POLL_TYPE:
		ret = get_user(uval, (int __user *)arg);
		if (ret != 0)
			return -EFAULT;
		uctxt->poll_type = (typeof(uctxt->poll_type))uval;
		break;

	case HFI1_IOCTL_ACK_EVENT:
		ret = get_user(ul_uval, (unsigned long __user *)arg);
		if (ret != 0)
			return -EFAULT;
		ret = user_event_ack(uctxt, fd->subctxt, ul_uval);
		break;

	case HFI1_IOCTL_SET_PKEY:
		ret = get_user(uval16, (u16 __user *)arg);
		if (ret != 0)
			return -EFAULT;
		if (HFI1_CAP_IS_USET(PKEY_CHECK))
			ret = set_ctxt_pkey(uctxt, fd->subctxt, uval16);
		else
			return -EPERM;
		break;

	case HFI1_IOCTL_CTXT_RESET: {
		struct send_context *sc;
		struct hfi1_devdata *dd;

		if (!uctxt || !uctxt->dd || !uctxt->sc)
			return -EINVAL;

		/*
		 * There is no protection here. User level has to
		 * guarantee that no one will be writing to the send
		 * context while it is being re-initialized.
		 * If user level breaks that guarantee, it will break
		 * it's own context and no one else's.
		 */
		dd = uctxt->dd;
		sc = uctxt->sc;
		/*
		 * Wait until the interrupt handler has marked the
		 * context as halted or frozen. Report error if we time
		 * out.
		 */
		wait_event_interruptible_timeout(
			sc->halt_wait, (sc->flags & SCF_HALTED),
			msecs_to_jiffies(SEND_CTXT_HALT_TIMEOUT));
		if (!(sc->flags & SCF_HALTED))
			return -ENOLCK;

		/*
		 * If the send context was halted due to a Freeze,
		 * wait until the device has been "unfrozen" before
		 * resetting the context.
		 */
		if (sc->flags & SCF_FROZEN) {
			wait_event_interruptible_timeout(
				dd->event_queue,
				!(ACCESS_ONCE(dd->flags) & HFI1_FROZEN),
				msecs_to_jiffies(SEND_CTXT_HALT_TIMEOUT));
			if (dd->flags & HFI1_FROZEN)
				return -ENOLCK;

			if (dd->flags & HFI1_FORCED_FREEZE)
				/*
				 * Don't allow context reset if we are into
				 * forced freeze
				 */
				return -ENODEV;

			sc_disable(sc);
			ret = sc_enable(sc);
			hfi1_rcvctrl(dd, HFI1_RCVCTRL_CTXT_ENB,
				     uctxt->ctxt);
		} else {
			ret = sc_restart(sc);
		}
		if (!ret)
			sc_return_credits(sc);
		break;
	}

	case HFI1_IOCTL_GET_VERS:
		uval = HFI1_USER_SWVERSION;
		if (put_user(uval, (int __user *)arg))
			return -EFAULT;
		break;

	default:
		return -EINVAL;
	}

	return ret;
}

M
Mike Marciniszyn 已提交
414 415
static ssize_t hfi1_write_iter(struct kiocb *kiocb, struct iov_iter *from)
{
416 417 418
	struct hfi1_filedata *fd = kiocb->ki_filp->private_data;
	struct hfi1_user_sdma_pkt_q *pq = fd->pq;
	struct hfi1_user_sdma_comp_q *cq = fd->cq;
419
	int done = 0, reqs = 0;
M
Mike Marciniszyn 已提交
420 421
	unsigned long dim = from->nr_segs;

422 423
	if (!cq || !pq)
		return -EIO;
M
Mike Marciniszyn 已提交
424

425 426
	if (!iter_is_iovec(from) || !dim)
		return -EINVAL;
M
Mike Marciniszyn 已提交
427 428

	hfi1_cdbg(SDMA, "SDMA request from %u:%u (%lu)",
429
		  fd->uctxt->ctxt, fd->subctxt, dim);
M
Mike Marciniszyn 已提交
430

431 432
	if (atomic_read(&pq->n_reqs) == pq->n_max_reqs)
		return -ENOSPC;
M
Mike Marciniszyn 已提交
433 434

	while (dim) {
435
		int ret;
M
Mike Marciniszyn 已提交
436 437 438
		unsigned long count = 0;

		ret = hfi1_user_sdma_process_request(
439
			fd, (struct iovec *)(from->iov + done),
M
Mike Marciniszyn 已提交
440
			dim, &count);
441 442 443 444
		if (ret) {
			reqs = ret;
			break;
		}
M
Mike Marciniszyn 已提交
445 446 447 448
		dim -= count;
		done += count;
		reqs++;
	}
449 450

	return reqs;
M
Mike Marciniszyn 已提交
451 452 453 454
}

static int hfi1_file_mmap(struct file *fp, struct vm_area_struct *vma)
{
455 456
	struct hfi1_filedata *fd = fp->private_data;
	struct hfi1_ctxtdata *uctxt = fd->uctxt;
M
Mike Marciniszyn 已提交
457
	struct hfi1_devdata *dd;
458
	unsigned long flags;
M
Mike Marciniszyn 已提交
459 460
	u64 token = vma->vm_pgoff << PAGE_SHIFT,
		memaddr = 0;
461
	void *memvirt = NULL;
M
Mike Marciniszyn 已提交
462 463 464 465 466 467 468 469 470 471 472 473 474 475
	u8 subctxt, mapio = 0, vmf = 0, type;
	ssize_t memlen = 0;
	int ret = 0;
	u16 ctxt;

	if (!is_valid_mmap(token) || !uctxt ||
	    !(vma->vm_flags & VM_SHARED)) {
		ret = -EINVAL;
		goto done;
	}
	dd = uctxt->dd;
	ctxt = HFI1_MMAP_TOKEN_GET(CTXT, token);
	subctxt = HFI1_MMAP_TOKEN_GET(SUBCTXT, token);
	type = HFI1_MMAP_TOKEN_GET(TYPE, token);
476
	if (ctxt != uctxt->ctxt || subctxt != fd->subctxt) {
M
Mike Marciniszyn 已提交
477 478 479 480 481 482 483 484 485 486 487
		ret = -EINVAL;
		goto done;
	}

	flags = vma->vm_flags;

	switch (type) {
	case PIO_BUFS:
	case PIO_BUFS_SOP:
		memaddr = ((dd->physaddr + TXE_PIO_SEND) +
				/* chip pio base */
488
			   (uctxt->sc->hw_context * BIT(16))) +
M
Mike Marciniszyn 已提交
489 490 491 492 493 494 495
				/* 64K PIO space / ctxt */
			(type == PIO_BUFS_SOP ?
				(TXE_PIO_SIZE / 2) : 0); /* sop? */
		/*
		 * Map only the amount allocated to the context, not the
		 * entire available context's PIO space.
		 */
496
		memlen = PAGE_ALIGN(uctxt->sc->credits * PIO_BLOCK_SIZE);
M
Mike Marciniszyn 已提交
497 498 499 500 501 502 503 504 505 506 507 508 509 510 511
		flags &= ~VM_MAYREAD;
		flags |= VM_DONTCOPY | VM_DONTEXPAND;
		vma->vm_page_prot = pgprot_writecombine(vma->vm_page_prot);
		mapio = 1;
		break;
	case PIO_CRED:
		if (flags & VM_WRITE) {
			ret = -EPERM;
			goto done;
		}
		/*
		 * The credit return location for this context could be on the
		 * second or third page allocated for credit returns (if number
		 * of enabled contexts > 64 and 128 respectively).
		 */
512 513
		memvirt = dd->cr_base[uctxt->numa_id].va;
		memaddr = virt_to_phys(memvirt) +
M
Mike Marciniszyn 已提交
514 515 516 517 518 519 520 521 522 523 524 525 526 527 528
			(((u64)uctxt->sc->hw_free -
			  (u64)dd->cr_base[uctxt->numa_id].va) & PAGE_MASK);
		memlen = PAGE_SIZE;
		flags &= ~VM_MAYWRITE;
		flags |= VM_DONTCOPY | VM_DONTEXPAND;
		/*
		 * The driver has already allocated memory for credit
		 * returns and programmed it into the chip. Has that
		 * memory been flagged as non-cached?
		 */
		/* vma->vm_page_prot = pgprot_noncached(vma->vm_page_prot); */
		mapio = 1;
		break;
	case RCV_HDRQ:
		memlen = uctxt->rcvhdrq_size;
529
		memvirt = uctxt->rcvhdrq;
M
Mike Marciniszyn 已提交
530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552
		break;
	case RCV_EGRBUF: {
		unsigned long addr;
		int i;
		/*
		 * The RcvEgr buffer need to be handled differently
		 * as multiple non-contiguous pages need to be mapped
		 * into the user process.
		 */
		memlen = uctxt->egrbufs.size;
		if ((vma->vm_end - vma->vm_start) != memlen) {
			dd_dev_err(dd, "Eager buffer map size invalid (%lu != %lu)\n",
				   (vma->vm_end - vma->vm_start), memlen);
			ret = -EINVAL;
			goto done;
		}
		if (vma->vm_flags & VM_WRITE) {
			ret = -EPERM;
			goto done;
		}
		vma->vm_flags &= ~VM_MAYWRITE;
		addr = vma->vm_start;
		for (i = 0 ; i < uctxt->egrbufs.numbufs; i++) {
553 554
			memlen = uctxt->egrbufs.buffers[i].len;
			memvirt = uctxt->egrbufs.buffers[i].addr;
M
Mike Marciniszyn 已提交
555 556
			ret = remap_pfn_range(
				vma, addr,
557 558 559 560 561 562 563
				/*
				 * virt_to_pfn() does the same, but
				 * it's not available on x86_64
				 * when CONFIG_MMU is enabled.
				 */
				PFN_DOWN(__pa(memvirt)),
				memlen,
M
Mike Marciniszyn 已提交
564 565 566
				vma->vm_page_prot);
			if (ret < 0)
				goto done;
567
			addr += memlen;
M
Mike Marciniszyn 已提交
568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593
		}
		ret = 0;
		goto done;
	}
	case UREGS:
		/*
		 * Map only the page that contains this context's user
		 * registers.
		 */
		memaddr = (unsigned long)
			(dd->physaddr + RXE_PER_CONTEXT_USER)
			+ (uctxt->ctxt * RXE_PER_CONTEXT_SIZE);
		/*
		 * TidFlow table is on the same page as the rest of the
		 * user registers.
		 */
		memlen = PAGE_SIZE;
		flags |= VM_DONTCOPY | VM_DONTEXPAND;
		vma->vm_page_prot = pgprot_noncached(vma->vm_page_prot);
		mapio = 1;
		break;
	case EVENTS:
		/*
		 * Use the page where this context's flags are. User level
		 * knows where it's own bitmap is within the page.
		 */
594
		memaddr = (unsigned long)(dd->events +
595 596
				  ((uctxt->ctxt - dd->first_dyn_alloc_ctxt) *
				   HFI1_MAX_SHARED_CTXTS)) & PAGE_MASK;
M
Mike Marciniszyn 已提交
597 598 599 600 601 602 603 604 605
		memlen = PAGE_SIZE;
		/*
		 * v3.7 removes VM_RESERVED but the effect is kept by
		 * using VM_IO.
		 */
		flags |= VM_IO | VM_DONTEXPAND;
		vmf = 1;
		break;
	case STATUS:
606 607 608 609
		if (flags & (unsigned long)(VM_WRITE | VM_EXEC)) {
			ret = -EPERM;
			goto done;
		}
M
Mike Marciniszyn 已提交
610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627
		memaddr = kvirt_to_phys((void *)dd->status);
		memlen = PAGE_SIZE;
		flags |= VM_IO | VM_DONTEXPAND;
		break;
	case RTAIL:
		if (!HFI1_CAP_IS_USET(DMA_RTAIL)) {
			/*
			 * If the memory allocation failed, the context alloc
			 * also would have failed, so we would never get here
			 */
			ret = -EINVAL;
			goto done;
		}
		if (flags & VM_WRITE) {
			ret = -EPERM;
			goto done;
		}
		memlen = PAGE_SIZE;
628
		memvirt = (void *)uctxt->rcvhdrtail_kvaddr;
M
Mike Marciniszyn 已提交
629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650
		flags &= ~VM_MAYWRITE;
		break;
	case SUBCTXT_UREGS:
		memaddr = (u64)uctxt->subctxt_uregbase;
		memlen = PAGE_SIZE;
		flags |= VM_IO | VM_DONTEXPAND;
		vmf = 1;
		break;
	case SUBCTXT_RCV_HDRQ:
		memaddr = (u64)uctxt->subctxt_rcvhdr_base;
		memlen = uctxt->rcvhdrq_size * uctxt->subctxt_cnt;
		flags |= VM_IO | VM_DONTEXPAND;
		vmf = 1;
		break;
	case SUBCTXT_EGRBUF:
		memaddr = (u64)uctxt->subctxt_rcvegrbuf;
		memlen = uctxt->egrbufs.size * uctxt->subctxt_cnt;
		flags |= VM_IO | VM_DONTEXPAND;
		flags &= ~VM_MAYWRITE;
		vmf = 1;
		break;
	case SDMA_COMP: {
651
		struct hfi1_user_sdma_comp_q *cq = fd->cq;
M
Mike Marciniszyn 已提交
652

653
		if (!cq) {
M
Mike Marciniszyn 已提交
654 655 656 657
			ret = -EFAULT;
			goto done;
		}
		memaddr = (u64)cq->comps;
658
		memlen = PAGE_ALIGN(sizeof(*cq->comps) * cq->nentries);
M
Mike Marciniszyn 已提交
659 660 661 662 663 664 665 666 667 668 669
		flags |= VM_IO | VM_DONTEXPAND;
		vmf = 1;
		break;
	}
	default:
		ret = -EINVAL;
		break;
	}

	if ((vma->vm_end - vma->vm_start) != memlen) {
		hfi1_cdbg(PROC, "%u:%u Memory size mismatch %lu:%lu",
670
			  uctxt->ctxt, fd->subctxt,
M
Mike Marciniszyn 已提交
671 672 673 674 675 676
			  (vma->vm_end - vma->vm_start), memlen);
		ret = -EINVAL;
		goto done;
	}

	vma->vm_flags = flags;
677 678 679
	hfi1_cdbg(PROC,
		  "%u:%u type:%u io/vf:%d/%d, addr:0x%llx, len:%lu(%lu), flags:0x%lx\n",
		    ctxt, subctxt, type, mapio, vmf, memaddr, memlen,
M
Mike Marciniszyn 已提交
680 681
		    vma->vm_end - vma->vm_start, vma->vm_flags);
	if (vmf) {
682
		vma->vm_pgoff = PFN_DOWN(memaddr);
M
Mike Marciniszyn 已提交
683 684 685
		vma->vm_ops = &vm_ops;
		ret = 0;
	} else if (mapio) {
686 687 688
		ret = io_remap_pfn_range(vma, vma->vm_start,
					 PFN_DOWN(memaddr),
					 memlen,
M
Mike Marciniszyn 已提交
689
					 vma->vm_page_prot);
690 691 692 693 694
	} else if (memvirt) {
		ret = remap_pfn_range(vma, vma->vm_start,
				      PFN_DOWN(__pa(memvirt)),
				      memlen,
				      vma->vm_page_prot);
M
Mike Marciniszyn 已提交
695
	} else {
696 697 698
		ret = remap_pfn_range(vma, vma->vm_start,
				      PFN_DOWN(memaddr),
				      memlen,
M
Mike Marciniszyn 已提交
699 700 701 702 703 704 705 706 707 708
				      vma->vm_page_prot);
	}
done:
	return ret;
}

/*
 * Local (non-chip) user memory is not mapped right away but as it is
 * accessed by the user-level code.
 */
709
static int vma_fault(struct vm_fault *vmf)
M
Mike Marciniszyn 已提交
710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727
{
	struct page *page;

	page = vmalloc_to_page((void *)(vmf->pgoff << PAGE_SHIFT));
	if (!page)
		return VM_FAULT_SIGBUS;

	get_page(page);
	vmf->page = page;

	return 0;
}

static unsigned int hfi1_poll(struct file *fp, struct poll_table_struct *pt)
{
	struct hfi1_ctxtdata *uctxt;
	unsigned pollflag;

728
	uctxt = ((struct hfi1_filedata *)fp->private_data)->uctxt;
M
Mike Marciniszyn 已提交
729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744
	if (!uctxt)
		pollflag = POLLERR;
	else if (uctxt->poll_type == HFI1_POLL_TYPE_URGENT)
		pollflag = poll_urgent(fp, pt);
	else  if (uctxt->poll_type == HFI1_POLL_TYPE_ANYRCV)
		pollflag = poll_next(fp, pt);
	else /* invalid */
		pollflag = POLLERR;

	return pollflag;
}

static int hfi1_file_close(struct inode *inode, struct file *fp)
{
	struct hfi1_filedata *fdata = fp->private_data;
	struct hfi1_ctxtdata *uctxt = fdata->uctxt;
745 746 747
	struct hfi1_devdata *dd = container_of(inode->i_cdev,
					       struct hfi1_devdata,
					       user_cdev);
M
Mike Marciniszyn 已提交
748 749 750 751 752 753 754 755 756 757 758 759
	unsigned long flags, *ev;

	fp->private_data = NULL;

	if (!uctxt)
		goto done;

	hfi1_cdbg(PROC, "freeing ctxt %u:%u", uctxt->ctxt, fdata->subctxt);
	mutex_lock(&hfi1_mutex);

	flush_wc();
	/* drain user sdma queue */
760
	hfi1_user_sdma_free_queues(fdata);
M
Mike Marciniszyn 已提交
761

762
	/* release the cpu */
763
	hfi1_put_proc_affinity(fdata->rec_cpu_num);
764

765 766 767
	/* clean up rcv side */
	hfi1_user_exp_rcv_free(fdata);

M
Mike Marciniszyn 已提交
768 769 770 771
	/*
	 * Clear any left over, unhandled events so the next process that
	 * gets this context doesn't get confused.
	 */
772
	ev = dd->events + ((uctxt->ctxt - dd->first_dyn_alloc_ctxt) *
M
Mike Marciniszyn 已提交
773 774 775
			   HFI1_MAX_SHARED_CTXTS) + fdata->subctxt;
	*ev = 0;

776 777
	__clear_bit(fdata->subctxt, uctxt->in_use_ctxts);
	if (!bitmap_empty(uctxt->in_use_ctxts, HFI1_MAX_SHARED_CTXTS)) {
M
Mike Marciniszyn 已提交
778 779 780 781 782 783 784 785 786 787 788 789
		mutex_unlock(&hfi1_mutex);
		goto done;
	}

	spin_lock_irqsave(&dd->uctxt_lock, flags);
	/*
	 * Disable receive context and interrupt available, reset all
	 * RcvCtxtCtrl bits to default values.
	 */
	hfi1_rcvctrl(dd, HFI1_RCVCTRL_CTXT_DIS |
		     HFI1_RCVCTRL_TIDFLOW_DIS |
		     HFI1_RCVCTRL_INTRAVAIL_DIS |
790
		     HFI1_RCVCTRL_TAILUPD_DIS |
M
Mike Marciniszyn 已提交
791 792 793 794 795 796 797 798 799 800 801 802 803 804 805
		     HFI1_RCVCTRL_ONE_PKT_EGR_DIS |
		     HFI1_RCVCTRL_NO_RHQ_DROP_DIS |
		     HFI1_RCVCTRL_NO_EGR_DROP_DIS, uctxt->ctxt);
	/* Clear the context's J_KEY */
	hfi1_clear_ctxt_jkey(dd, uctxt->ctxt);
	/*
	 * Reset context integrity checks to default.
	 * (writes to CSRs probably belong in chip.c)
	 */
	write_kctxt_csr(dd, uctxt->sc->hw_context, SEND_CTXT_CHECK_ENABLE,
			hfi1_pkt_default_send_ctxt_mask(dd, uctxt->sc->type));
	sc_disable(uctxt->sc);
	spin_unlock_irqrestore(&dd->uctxt_lock, flags);

	dd->rcd[uctxt->ctxt] = NULL;
806

807
	hfi1_free_ctxt_rcv_groups(uctxt);
808
	hfi1_clear_ctxt_pkey(dd, uctxt);
809

M
Mike Marciniszyn 已提交
810 811 812 813 814 815 816
	uctxt->rcvwait_to = 0;
	uctxt->piowait_to = 0;
	uctxt->rcvnowait = 0;
	uctxt->pionowait = 0;
	uctxt->event_flags = 0;

	hfi1_stats.sps_ctxts--;
817 818
	if (++dd->freectxts == dd->num_user_contexts)
		aspm_enable_all(dd);
M
Mike Marciniszyn 已提交
819 820 821
	mutex_unlock(&hfi1_mutex);
	hfi1_free_ctxtdata(dd, uctxt);
done:
822
	mmdrop(fdata->mm);
823
	kobject_put(&dd->kobj);
824 825 826 827

	if (atomic_dec_and_test(&dd->user_refcount))
		complete(&dd->user_comp);

M
Mike Marciniszyn 已提交
828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847
	kfree(fdata);
	return 0;
}

/*
 * Convert kernel *virtual* addresses to physical addresses.
 * This is used to vmalloc'ed addresses.
 */
static u64 kvirt_to_phys(void *addr)
{
	struct page *page;
	u64 paddr = 0;

	page = vmalloc_to_page(addr);
	if (page)
		paddr = page_to_pfn(page) << PAGE_SHIFT;

	return paddr;
}

848
static int assign_ctxt(struct hfi1_filedata *fd, struct hfi1_user_info *uinfo)
M
Mike Marciniszyn 已提交
849
{
850
	int ret;
851
	unsigned int swmajor, swminor;
M
Mike Marciniszyn 已提交
852 853

	swmajor = uinfo->userversion >> 16;
854 855
	if (swmajor != HFI1_USER_SWMAJOR)
		return -ENODEV;
M
Mike Marciniszyn 已提交
856 857 858 859

	swminor = uinfo->userversion & 0xffff;

	mutex_lock(&hfi1_mutex);
860 861 862 863 864
	/*
	 * Get a sub context if necessary.
	 * ret < 0 error, 0 no context, 1 sub-context found
	 */
	ret = 0;
865
	if (uinfo->subctxt_cnt) {
866
		ret = find_sub_ctxt(fd, uinfo);
867
		if (ret > 0)
868 869
			fd->rec_cpu_num =
				hfi1_get_proc_affinity(fd->uctxt->numa_id);
870
	}
M
Mike Marciniszyn 已提交
871 872

	/*
873
	 * Allocate a base context if context sharing is not required or we
874
	 * couldn't find a sub context.
M
Mike Marciniszyn 已提交
875
	 */
876 877 878
	if (!ret)
		ret = allocate_ctxt(fd, fd->dd, uinfo);

M
Mike Marciniszyn 已提交
879
	mutex_unlock(&hfi1_mutex);
880 881 882 883 884 885 886 887 888 889

	/* Depending on the context type, do the appropriate init */
	if (ret > 0) {
		/*
		 * sub-context info can only be set up after the base
		 * context has been completed.
		 */
		ret = wait_event_interruptible(fd->uctxt->wait, !test_bit(
					       HFI1_CTXT_BASE_UNINIT,
					       &fd->uctxt->event_flags));
890 891 892 893
		if (test_bit(HFI1_CTXT_BASE_FAILED, &fd->uctxt->event_flags)) {
			clear_bit(fd->subctxt, fd->uctxt->in_use_ctxts);
			return -ENOMEM;
		}
894 895
		/* The only thing a sub context needs is the user_xxx stuff */
		if (!ret)
896 897 898 899
			ret = init_user_ctxt(fd);

		if (ret)
			clear_bit(fd->subctxt, fd->uctxt->in_use_ctxts);
900 901
	} else if (!ret) {
		ret = setup_base_ctxt(fd);
902 903 904 905 906 907 908 909 910
		if (fd->uctxt->subctxt_cnt) {
			/* If there is an error, set the failed bit. */
			if (ret)
				set_bit(HFI1_CTXT_BASE_FAILED,
					&fd->uctxt->event_flags);
			/*
			 * Base context is done, notify anybody using a
			 * sub-context that is waiting for this completion
			 */
911 912 913 914 915 916
			clear_bit(HFI1_CTXT_BASE_UNINIT,
				  &fd->uctxt->event_flags);
			wake_up(&fd->uctxt->wait);
		}
	}

M
Mike Marciniszyn 已提交
917 918 919
	return ret;
}

920 921 922 923
/*
 * The hfi1_mutex must be held when this function is called.  It is
 * necessary to ensure serialized access to the bitmask in_use_ctxts.
 */
924 925
static int find_sub_ctxt(struct hfi1_filedata *fd,
			 const struct hfi1_user_info *uinfo)
M
Mike Marciniszyn 已提交
926
{
927 928
	int i;
	struct hfi1_devdata *dd = fd->dd;
929
	u16 subctxt;
M
Mike Marciniszyn 已提交
930

931 932
	for (i = dd->first_dyn_alloc_ctxt; i < dd->num_rcv_contexts; i++) {
		struct hfi1_ctxtdata *uctxt = dd->rcd[i];
M
Mike Marciniszyn 已提交
933

934
		/* Skip ctxts which are not yet open */
935 936 937
		if (!uctxt ||
		    bitmap_empty(uctxt->in_use_ctxts,
				 HFI1_MAX_SHARED_CTXTS))
938
			continue;
M
Mike Marciniszyn 已提交
939

940 941
		/* Skip dynamically allocted kernel contexts */
		if (uctxt->sc && (uctxt->sc->type == SC_KERNEL))
M
Mike Marciniszyn 已提交
942
			continue;
943 944 945 946 947 948 949 950 951 952

		/* Skip ctxt if it doesn't match the requested one */
		if (memcmp(uctxt->uuid, uinfo->uuid,
			   sizeof(uctxt->uuid)) ||
		    uctxt->jkey != generate_jkey(current_uid()) ||
		    uctxt->subctxt_id != uinfo->subctxt_id ||
		    uctxt->subctxt_cnt != uinfo->subctxt_cnt)
			continue;

		/* Verify the sharing process matches the master */
953
		if (uctxt->userversion != uinfo->userversion)
954
			return -EINVAL;
955 956 957 958 959

		/* Find an unused context */
		subctxt = find_first_zero_bit(uctxt->in_use_ctxts,
					      HFI1_MAX_SHARED_CTXTS);
		if (subctxt >= uctxt->subctxt_cnt)
960
			return -EBUSY;
961

962
		fd->uctxt = uctxt;
963 964 965
		fd->subctxt = subctxt;
		__set_bit(fd->subctxt, uctxt->in_use_ctxts);

966
		return 1;
M
Mike Marciniszyn 已提交
967 968
	}

969
	return 0;
M
Mike Marciniszyn 已提交
970 971
}

972
static int allocate_ctxt(struct hfi1_filedata *fd, struct hfi1_devdata *dd,
M
Mike Marciniszyn 已提交
973 974 975
			 struct hfi1_user_info *uinfo)
{
	struct hfi1_ctxtdata *uctxt;
976
	unsigned int ctxt;
977
	int ret, numa;
M
Mike Marciniszyn 已提交
978 979 980 981 982 983 984 985 986 987 988 989

	if (dd->flags & HFI1_FROZEN) {
		/*
		 * Pick an error that is unique from all other errors
		 * that are returned so the user process knows that
		 * it tried to allocate while the SPC was frozen.  It
		 * it should be able to retry with success in a short
		 * while.
		 */
		return -EIO;
	}

990 991 992 993 994 995 996 997
	/*
	 * This check is sort of redundant to the next EBUSY error. It would
	 * also indicate an inconsistancy in the driver if this value was
	 * zero, but there were still contexts available.
	 */
	if (!dd->freectxts)
		return -EBUSY;

998 999
	for (ctxt = dd->first_dyn_alloc_ctxt;
	     ctxt < dd->num_rcv_contexts; ctxt++)
M
Mike Marciniszyn 已提交
1000 1001 1002 1003 1004 1005
		if (!dd->rcd[ctxt])
			break;

	if (ctxt == dd->num_rcv_contexts)
		return -EBUSY;

1006 1007 1008 1009 1010
	/*
	 * If we don't have a NUMA node requested, preference is towards
	 * device NUMA node.
	 */
	fd->rec_cpu_num = hfi1_get_proc_affinity(dd->node);
1011 1012 1013 1014 1015
	if (fd->rec_cpu_num != -1)
		numa = cpu_to_node(fd->rec_cpu_num);
	else
		numa = numa_node_id();
	uctxt = hfi1_create_ctxtdata(dd->pport, ctxt, numa);
M
Mike Marciniszyn 已提交
1016 1017 1018 1019 1020
	if (!uctxt) {
		dd_dev_err(dd,
			   "Unable to allocate ctxtdata memory, failing open\n");
		return -ENOMEM;
	}
1021 1022 1023 1024
	hfi1_cdbg(PROC, "[%u:%u] pid %u assigned to CPU %d (NUMA %u)",
		  uctxt->ctxt, fd->subctxt, current->pid, fd->rec_cpu_num,
		  uctxt->numa_id);

M
Mike Marciniszyn 已提交
1025 1026 1027 1028
	/*
	 * Allocate and enable a PIO send context.
	 */
	uctxt->sc = sc_alloc(dd, SC_USER, uctxt->rcvhdrqentsize,
1029
			     uctxt->dd->node);
1030 1031 1032 1033
	if (!uctxt->sc) {
		ret = -ENOMEM;
		goto ctxdata_free;
	}
1034 1035
	hfi1_cdbg(PROC, "allocated send context %u(%u)\n", uctxt->sc->sw_index,
		  uctxt->sc->hw_context);
M
Mike Marciniszyn 已提交
1036 1037
	ret = sc_enable(uctxt->sc);
	if (ret)
1038 1039
		goto ctxdata_free;

M
Mike Marciniszyn 已提交
1040
	/*
1041 1042
	 * Setup sub context resources if the user-level has requested
	 * sub contexts.
M
Mike Marciniszyn 已提交
1043 1044 1045
	 * This has to be done here so the rest of the sub-contexts find the
	 * proper master.
	 */
1046
	if (uinfo->subctxt_cnt) {
M
Mike Marciniszyn 已提交
1047 1048 1049 1050 1051 1052
		ret = init_subctxts(uctxt, uinfo);
		/*
		 * On error, we don't need to disable and de-allocate the
		 * send context because it will be done during file close
		 */
		if (ret)
1053
			goto ctxdata_free;
M
Mike Marciniszyn 已提交
1054 1055
	}
	uctxt->userversion = uinfo->userversion;
1056
	uctxt->flags = hfi1_cap_mask; /* save current flag state */
M
Mike Marciniszyn 已提交
1057 1058 1059 1060 1061 1062 1063
	init_waitqueue_head(&uctxt->wait);
	strlcpy(uctxt->comm, current->comm, sizeof(uctxt->comm));
	memcpy(uctxt->uuid, uinfo->uuid, sizeof(uctxt->uuid));
	uctxt->jkey = generate_jkey(current_uid());
	INIT_LIST_HEAD(&uctxt->sdma_queues);
	spin_lock_init(&uctxt->sdma_qlock);
	hfi1_stats.sps_ctxts++;
1064 1065 1066 1067 1068 1069
	/*
	 * Disable ASPM when there are open user/PSM contexts to avoid
	 * issues with ASPM L1 exit latency
	 */
	if (dd->freectxts-- == dd->num_user_contexts)
		aspm_disable_all(dd);
1070
	fd->uctxt = uctxt;
M
Mike Marciniszyn 已提交
1071 1072

	return 0;
1073 1074 1075 1076 1077

ctxdata_free:
	dd->rcd[ctxt] = NULL;
	hfi1_free_ctxtdata(dd, uctxt);
	return ret;
M
Mike Marciniszyn 已提交
1078 1079 1080 1081 1082
}

static int init_subctxts(struct hfi1_ctxtdata *uctxt,
			 const struct hfi1_user_info *uinfo)
{
1083
	u16 num_subctxts;
M
Mike Marciniszyn 已提交
1084 1085

	num_subctxts = uinfo->subctxt_cnt;
1086 1087
	if (num_subctxts > HFI1_MAX_SHARED_CTXTS)
		return -EINVAL;
M
Mike Marciniszyn 已提交
1088 1089 1090 1091

	uctxt->subctxt_cnt = uinfo->subctxt_cnt;
	uctxt->subctxt_id = uinfo->subctxt_id;
	uctxt->redirect_seq_cnt = 1;
1092
	set_bit(HFI1_CTXT_BASE_UNINIT, &uctxt->event_flags);
1093 1094

	return 0;
M
Mike Marciniszyn 已提交
1095 1096 1097 1098 1099
}

static int setup_subctxt(struct hfi1_ctxtdata *uctxt)
{
	int ret = 0;
1100
	u16 num_subctxts = uctxt->subctxt_cnt;
M
Mike Marciniszyn 已提交
1101 1102

	uctxt->subctxt_uregbase = vmalloc_user(PAGE_SIZE);
1103 1104 1105
	if (!uctxt->subctxt_uregbase)
		return -ENOMEM;

M
Mike Marciniszyn 已提交
1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119
	/* We can take the size of the RcvHdr Queue from the master */
	uctxt->subctxt_rcvhdr_base = vmalloc_user(uctxt->rcvhdrq_size *
						  num_subctxts);
	if (!uctxt->subctxt_rcvhdr_base) {
		ret = -ENOMEM;
		goto bail_ureg;
	}

	uctxt->subctxt_rcvegrbuf = vmalloc_user(uctxt->egrbufs.size *
						num_subctxts);
	if (!uctxt->subctxt_rcvegrbuf) {
		ret = -ENOMEM;
		goto bail_rhdr;
	}
1120 1121 1122

	return 0;

M
Mike Marciniszyn 已提交
1123 1124
bail_rhdr:
	vfree(uctxt->subctxt_rcvhdr_base);
1125
	uctxt->subctxt_rcvhdr_base = NULL;
M
Mike Marciniszyn 已提交
1126 1127 1128
bail_ureg:
	vfree(uctxt->subctxt_uregbase);
	uctxt->subctxt_uregbase = NULL;
1129

M
Mike Marciniszyn 已提交
1130 1131 1132
	return ret;
}

1133
static void user_init(struct hfi1_ctxtdata *uctxt)
M
Mike Marciniszyn 已提交
1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158
{
	unsigned int rcvctrl_ops = 0;

	/* initialize poll variables... */
	uctxt->urgent = 0;
	uctxt->urgent_poll = 0;

	/*
	 * Now enable the ctxt for receive.
	 * For chips that are set to DMA the tail register to memory
	 * when they change (and when the update bit transitions from
	 * 0 to 1.  So for those chips, we turn it off and then back on.
	 * This will (very briefly) affect any other open ctxts, but the
	 * duration is very short, and therefore isn't an issue.  We
	 * explicitly set the in-memory tail copy to 0 beforehand, so we
	 * don't have to wait to be sure the DMA update has happened
	 * (chip resets head/tail to 0 on transition to enable).
	 */
	if (uctxt->rcvhdrtail_kvaddr)
		clear_rcvhdrtail(uctxt);

	/* Setup J_KEY before enabling the context */
	hfi1_set_ctxt_jkey(uctxt->dd, uctxt->ctxt, uctxt->jkey);

	rcvctrl_ops = HFI1_RCVCTRL_CTXT_ENB;
1159
	if (HFI1_CAP_UGET_MASK(uctxt->flags, HDRSUPP))
M
Mike Marciniszyn 已提交
1160 1161 1162 1163 1164 1165
		rcvctrl_ops |= HFI1_RCVCTRL_TIDFLOW_ENB;
	/*
	 * Ignore the bit in the flags for now until proper
	 * support for multiple packet per rcv array entry is
	 * added.
	 */
1166
	if (!HFI1_CAP_UGET_MASK(uctxt->flags, MULTI_PKT_EGR))
M
Mike Marciniszyn 已提交
1167
		rcvctrl_ops |= HFI1_RCVCTRL_ONE_PKT_EGR_ENB;
1168
	if (HFI1_CAP_UGET_MASK(uctxt->flags, NODROP_EGR_FULL))
M
Mike Marciniszyn 已提交
1169
		rcvctrl_ops |= HFI1_RCVCTRL_NO_EGR_DROP_ENB;
1170
	if (HFI1_CAP_UGET_MASK(uctxt->flags, NODROP_RHQ_FULL))
M
Mike Marciniszyn 已提交
1171
		rcvctrl_ops |= HFI1_RCVCTRL_NO_RHQ_DROP_ENB;
1172 1173 1174 1175 1176 1177
	/*
	 * The RcvCtxtCtrl.TailUpd bit has to be explicitly written.
	 * We can't rely on the correct value to be set from prior
	 * uses of the chip or ctxt. Therefore, add the rcvctrl op
	 * for both cases.
	 */
1178
	if (HFI1_CAP_UGET_MASK(uctxt->flags, DMA_RTAIL))
M
Mike Marciniszyn 已提交
1179
		rcvctrl_ops |= HFI1_RCVCTRL_TAILUPD_ENB;
1180 1181
	else
		rcvctrl_ops |= HFI1_RCVCTRL_TAILUPD_DIS;
M
Mike Marciniszyn 已提交
1182 1183 1184
	hfi1_rcvctrl(uctxt->dd, rcvctrl_ops, uctxt->ctxt);
}

1185 1186
static int get_ctxt_info(struct hfi1_filedata *fd, void __user *ubase,
			 __u32 len)
M
Mike Marciniszyn 已提交
1187 1188
{
	struct hfi1_ctxt_info cinfo;
1189
	struct hfi1_ctxtdata *uctxt = fd->uctxt;
M
Mike Marciniszyn 已提交
1190 1191
	int ret = 0;

1192
	memset(&cinfo, 0, sizeof(cinfo));
1193 1194 1195 1196
	cinfo.runtime_flags = (((uctxt->flags >> HFI1_CAP_MISC_SHIFT) &
				HFI1_CAP_MISC_MASK) << HFI1_CAP_USER_SHIFT) |
			HFI1_CAP_UGET_MASK(uctxt->flags, MASK) |
			HFI1_CAP_KGET_MASK(uctxt->flags, K2U);
D
Dean Luick 已提交
1197 1198 1199 1200
	/* adjust flag if this fd is not able to cache */
	if (!fd->handler)
		cinfo.runtime_flags |= HFI1_CAP_TID_UNMAP; /* no caching */

M
Mike Marciniszyn 已提交
1201 1202 1203
	cinfo.num_active = hfi1_count_active_units();
	cinfo.unit = uctxt->dd->unit;
	cinfo.ctxt = uctxt->ctxt;
1204
	cinfo.subctxt = fd->subctxt;
M
Mike Marciniszyn 已提交
1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215
	cinfo.rcvtids = roundup(uctxt->egrbufs.alloced,
				uctxt->dd->rcv_entries.group_size) +
		uctxt->expected_count;
	cinfo.credits = uctxt->sc->credits;
	cinfo.numa_node = uctxt->numa_id;
	cinfo.rec_cpu = fd->rec_cpu_num;
	cinfo.send_ctxt = uctxt->sc->hw_context;

	cinfo.egrtids = uctxt->egrbufs.alloced;
	cinfo.rcvhdrq_cnt = uctxt->rcvhdrq_cnt;
	cinfo.rcvhdrq_entsize = uctxt->rcvhdrqentsize << 2;
1216
	cinfo.sdma_ring_size = fd->cq->nentries;
M
Mike Marciniszyn 已提交
1217 1218
	cinfo.rcvegr_size = uctxt->egrbufs.rcvtid_size;

1219
	trace_hfi1_ctxt_info(uctxt->dd, uctxt->ctxt, fd->subctxt, cinfo);
M
Mike Marciniszyn 已提交
1220 1221
	if (copy_to_user(ubase, &cinfo, sizeof(cinfo)))
		ret = -EFAULT;
1222

M
Mike Marciniszyn 已提交
1223 1224 1225
	return ret;
}

1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240
static int init_user_ctxt(struct hfi1_filedata *fd)
{
	struct hfi1_ctxtdata *uctxt = fd->uctxt;
	int ret;

	ret = hfi1_user_sdma_alloc_queues(uctxt, fd);
	if (ret)
		return ret;

	ret = hfi1_user_exp_rcv_init(fd);

	return ret;
}

static int setup_base_ctxt(struct hfi1_filedata *fd)
M
Mike Marciniszyn 已提交
1241
{
1242
	struct hfi1_ctxtdata *uctxt = fd->uctxt;
M
Mike Marciniszyn 已提交
1243 1244 1245
	struct hfi1_devdata *dd = uctxt->dd;
	int ret = 0;

1246
	hfi1_init_ctxt(uctxt->sc);
M
Mike Marciniszyn 已提交
1247

1248 1249 1250
	/* Now allocate the RcvHdr queue and eager buffers. */
	ret = hfi1_create_rcvhdrq(dd, uctxt);
	if (ret)
1251
		return ret;
1252

1253
	ret = hfi1_setup_eagerbufs(uctxt);
1254
	if (ret)
1255
		goto setup_failed;
1256 1257 1258 1259 1260

	/* If sub-contexts are enabled, do the appropriate setup */
	if (uctxt->subctxt_cnt)
		ret = setup_subctxt(uctxt);
	if (ret)
1261
		goto setup_failed;
1262

1263
	ret = hfi1_alloc_ctxt_rcv_groups(uctxt);
1264
	if (ret)
1265
		goto setup_failed;
1266 1267

	ret = init_user_ctxt(fd);
M
Mike Marciniszyn 已提交
1268
	if (ret)
1269
		goto setup_failed;
M
Mike Marciniszyn 已提交
1270

1271 1272 1273 1274 1275 1276
	user_init(uctxt);

	return 0;

setup_failed:
	hfi1_free_ctxtdata(dd, uctxt);
M
Mike Marciniszyn 已提交
1277 1278 1279
	return ret;
}

1280 1281
static int get_base_info(struct hfi1_filedata *fd, void __user *ubase,
			 __u32 len)
M
Mike Marciniszyn 已提交
1282 1283
{
	struct hfi1_base_info binfo;
1284
	struct hfi1_ctxtdata *uctxt = fd->uctxt;
M
Mike Marciniszyn 已提交
1285 1286 1287 1288 1289
	struct hfi1_devdata *dd = uctxt->dd;
	ssize_t sz;
	unsigned offset;
	int ret = 0;

1290
	trace_hfi1_uctxtdata(uctxt->dd, uctxt, fd->subctxt);
M
Mike Marciniszyn 已提交
1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305

	memset(&binfo, 0, sizeof(binfo));
	binfo.hw_version = dd->revision;
	binfo.sw_version = HFI1_KERN_SWVERSION;
	binfo.bthqp = kdeth_qp;
	binfo.jkey = uctxt->jkey;
	/*
	 * If more than 64 contexts are enabled the allocated credit
	 * return will span two or three contiguous pages. Since we only
	 * map the page containing the context's credit return address,
	 * we need to calculate the offset in the proper page.
	 */
	offset = ((u64)uctxt->sc->hw_free -
		  (u64)dd->cr_base[uctxt->numa_id].va) % PAGE_SIZE;
	binfo.sc_credits_addr = HFI1_MMAP_TOKEN(PIO_CRED, uctxt->ctxt,
1306
						fd->subctxt, offset);
M
Mike Marciniszyn 已提交
1307
	binfo.pio_bufbase = HFI1_MMAP_TOKEN(PIO_BUFS, uctxt->ctxt,
1308
					    fd->subctxt,
M
Mike Marciniszyn 已提交
1309 1310 1311
					    uctxt->sc->base_addr);
	binfo.pio_bufbase_sop = HFI1_MMAP_TOKEN(PIO_BUFS_SOP,
						uctxt->ctxt,
1312
						fd->subctxt,
M
Mike Marciniszyn 已提交
1313 1314
						uctxt->sc->base_addr);
	binfo.rcvhdr_bufbase = HFI1_MMAP_TOKEN(RCV_HDRQ, uctxt->ctxt,
1315
					       fd->subctxt,
M
Mike Marciniszyn 已提交
1316 1317
					       uctxt->rcvhdrq);
	binfo.rcvegr_bufbase = HFI1_MMAP_TOKEN(RCV_EGRBUF, uctxt->ctxt,
1318
					       fd->subctxt,
1319
					       uctxt->egrbufs.rcvtids[0].dma);
M
Mike Marciniszyn 已提交
1320
	binfo.sdma_comp_bufbase = HFI1_MMAP_TOKEN(SDMA_COMP, uctxt->ctxt,
1321
						 fd->subctxt, 0);
M
Mike Marciniszyn 已提交
1322 1323 1324 1325 1326
	/*
	 * user regs are at
	 * (RXE_PER_CONTEXT_USER + (ctxt * RXE_PER_CONTEXT_SIZE))
	 */
	binfo.user_regbase = HFI1_MMAP_TOKEN(UREGS, uctxt->ctxt,
1327
					    fd->subctxt, 0);
1328
	offset = offset_in_page((((uctxt->ctxt - dd->first_dyn_alloc_ctxt) *
1329
		    HFI1_MAX_SHARED_CTXTS) + fd->subctxt) *
G
Geliang Tang 已提交
1330
		  sizeof(*dd->events));
M
Mike Marciniszyn 已提交
1331
	binfo.events_bufbase = HFI1_MMAP_TOKEN(EVENTS, uctxt->ctxt,
1332
					      fd->subctxt,
M
Mike Marciniszyn 已提交
1333 1334
					      offset);
	binfo.status_bufbase = HFI1_MMAP_TOKEN(STATUS, uctxt->ctxt,
1335
					      fd->subctxt,
M
Mike Marciniszyn 已提交
1336 1337 1338
					      dd->status);
	if (HFI1_CAP_IS_USET(DMA_RTAIL))
		binfo.rcvhdrtail_base = HFI1_MMAP_TOKEN(RTAIL, uctxt->ctxt,
1339
						       fd->subctxt, 0);
M
Mike Marciniszyn 已提交
1340 1341 1342
	if (uctxt->subctxt_cnt) {
		binfo.subctxt_uregbase = HFI1_MMAP_TOKEN(SUBCTXT_UREGS,
							uctxt->ctxt,
1343
							fd->subctxt, 0);
M
Mike Marciniszyn 已提交
1344 1345
		binfo.subctxt_rcvhdrbuf = HFI1_MMAP_TOKEN(SUBCTXT_RCV_HDRQ,
							 uctxt->ctxt,
1346
							 fd->subctxt, 0);
M
Mike Marciniszyn 已提交
1347 1348
		binfo.subctxt_rcvegrbuf = HFI1_MMAP_TOKEN(SUBCTXT_EGRBUF,
							 uctxt->ctxt,
1349
							 fd->subctxt, 0);
M
Mike Marciniszyn 已提交
1350 1351 1352 1353 1354 1355 1356 1357 1358 1359
	}
	sz = (len < sizeof(binfo)) ? len : sizeof(binfo);
	if (copy_to_user(ubase, &binfo, sz))
		ret = -EFAULT;
	return ret;
}

static unsigned int poll_urgent(struct file *fp,
				struct poll_table_struct *pt)
{
1360 1361
	struct hfi1_filedata *fd = fp->private_data;
	struct hfi1_ctxtdata *uctxt = fd->uctxt;
M
Mike Marciniszyn 已提交
1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382
	struct hfi1_devdata *dd = uctxt->dd;
	unsigned pollflag;

	poll_wait(fp, &uctxt->wait, pt);

	spin_lock_irq(&dd->uctxt_lock);
	if (uctxt->urgent != uctxt->urgent_poll) {
		pollflag = POLLIN | POLLRDNORM;
		uctxt->urgent_poll = uctxt->urgent;
	} else {
		pollflag = 0;
		set_bit(HFI1_CTXT_WAITING_URG, &uctxt->event_flags);
	}
	spin_unlock_irq(&dd->uctxt_lock);

	return pollflag;
}

static unsigned int poll_next(struct file *fp,
			      struct poll_table_struct *pt)
{
1383 1384
	struct hfi1_filedata *fd = fp->private_data;
	struct hfi1_ctxtdata *uctxt = fd->uctxt;
M
Mike Marciniszyn 已提交
1385 1386 1387 1388 1389 1390 1391 1392 1393 1394
	struct hfi1_devdata *dd = uctxt->dd;
	unsigned pollflag;

	poll_wait(fp, &uctxt->wait, pt);

	spin_lock_irq(&dd->uctxt_lock);
	if (hdrqempty(uctxt)) {
		set_bit(HFI1_CTXT_WAITING_RCV, &uctxt->event_flags);
		hfi1_rcvctrl(dd, HFI1_RCVCTRL_INTRAVAIL_ENB, uctxt->ctxt);
		pollflag = 0;
1395
	} else {
M
Mike Marciniszyn 已提交
1396
		pollflag = POLLIN | POLLRDNORM;
1397
	}
M
Mike Marciniszyn 已提交
1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421
	spin_unlock_irq(&dd->uctxt_lock);

	return pollflag;
}

/*
 * Find all user contexts in use, and set the specified bit in their
 * event mask.
 * See also find_ctxt() for a similar use, that is specific to send buffers.
 */
int hfi1_set_uevent_bits(struct hfi1_pportdata *ppd, const int evtbit)
{
	struct hfi1_ctxtdata *uctxt;
	struct hfi1_devdata *dd = ppd->dd;
	unsigned ctxt;
	int ret = 0;
	unsigned long flags;

	if (!dd->events) {
		ret = -EINVAL;
		goto done;
	}

	spin_lock_irqsave(&dd->uctxt_lock, flags);
1422
	for (ctxt = dd->first_dyn_alloc_ctxt; ctxt < dd->num_rcv_contexts;
M
Mike Marciniszyn 已提交
1423 1424 1425 1426
	     ctxt++) {
		uctxt = dd->rcd[ctxt];
		if (uctxt) {
			unsigned long *evs = dd->events +
1427
				(uctxt->ctxt - dd->first_dyn_alloc_ctxt) *
M
Mike Marciniszyn 已提交
1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453
				HFI1_MAX_SHARED_CTXTS;
			int i;
			/*
			 * subctxt_cnt is 0 if not shared, so do base
			 * separately, first, then remaining subctxt, if any
			 */
			set_bit(evtbit, evs);
			for (i = 1; i < uctxt->subctxt_cnt; i++)
				set_bit(evtbit, evs + i);
		}
	}
	spin_unlock_irqrestore(&dd->uctxt_lock, flags);
done:
	return ret;
}

/**
 * manage_rcvq - manage a context's receive queue
 * @uctxt: the context
 * @subctxt: the sub-context
 * @start_stop: action to carry out
 *
 * start_stop == 0 disables receive on the context, for use in queue
 * overflow conditions.  start_stop==1 re-enables, to be used to
 * re-init the software copy of the head register
 */
1454
static int manage_rcvq(struct hfi1_ctxtdata *uctxt, u16 subctxt,
M
Mike Marciniszyn 已提交
1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474
		       int start_stop)
{
	struct hfi1_devdata *dd = uctxt->dd;
	unsigned int rcvctrl_op;

	if (subctxt)
		goto bail;
	/* atomically clear receive enable ctxt. */
	if (start_stop) {
		/*
		 * On enable, force in-memory copy of the tail register to
		 * 0, so that protocol code doesn't have to worry about
		 * whether or not the chip has yet updated the in-memory
		 * copy or not on return from the system call. The chip
		 * always resets it's tail register back to 0 on a
		 * transition from disabled to enabled.
		 */
		if (uctxt->rcvhdrtail_kvaddr)
			clear_rcvhdrtail(uctxt);
		rcvctrl_op = HFI1_RCVCTRL_CTXT_ENB;
1475
	} else {
M
Mike Marciniszyn 已提交
1476
		rcvctrl_op = HFI1_RCVCTRL_CTXT_DIS;
1477
	}
M
Mike Marciniszyn 已提交
1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488
	hfi1_rcvctrl(dd, rcvctrl_op, uctxt->ctxt);
	/* always; new head should be equal to new tail; see above */
bail:
	return 0;
}

/*
 * clear the event notifier events for this context.
 * User process then performs actions appropriate to bit having been
 * set, if desired, and checks again in future.
 */
1489
static int user_event_ack(struct hfi1_ctxtdata *uctxt, u16 subctxt,
M
Mike Marciniszyn 已提交
1490 1491 1492 1493 1494 1495 1496 1497 1498
			  unsigned long events)
{
	int i;
	struct hfi1_devdata *dd = uctxt->dd;
	unsigned long *evs;

	if (!dd->events)
		return 0;

1499
	evs = dd->events + ((uctxt->ctxt - dd->first_dyn_alloc_ctxt) *
M
Mike Marciniszyn 已提交
1500 1501 1502 1503 1504 1505 1506 1507 1508 1509
			    HFI1_MAX_SHARED_CTXTS) + subctxt;

	for (i = 0; i <= _HFI1_MAX_EVENT_BIT; i++) {
		if (!test_bit(i, &events))
			continue;
		clear_bit(i, evs);
	}
	return 0;
}

1510
static int set_ctxt_pkey(struct hfi1_ctxtdata *uctxt, u16 subctxt, u16 pkey)
M
Mike Marciniszyn 已提交
1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544
{
	int ret = -ENOENT, i, intable = 0;
	struct hfi1_pportdata *ppd = uctxt->ppd;
	struct hfi1_devdata *dd = uctxt->dd;

	if (pkey == LIM_MGMT_P_KEY || pkey == FULL_MGMT_P_KEY) {
		ret = -EINVAL;
		goto done;
	}

	for (i = 0; i < ARRAY_SIZE(ppd->pkeys); i++)
		if (pkey == ppd->pkeys[i]) {
			intable = 1;
			break;
		}

	if (intable)
		ret = hfi1_set_ctxt_pkey(dd, uctxt->ctxt, pkey);
done:
	return ret;
}

static void user_remove(struct hfi1_devdata *dd)
{

	hfi1_cdev_cleanup(&dd->user_cdev, &dd->user_device);
}

static int user_add(struct hfi1_devdata *dd)
{
	char name[10];
	int ret;

	snprintf(name, sizeof(name), "%s_%d", class_name(), dd->unit);
1545
	ret = hfi1_cdev_init(dd->unit, name, &hfi1_file_ops,
1546
			     &dd->user_cdev, &dd->user_device,
1547
			     true, &dd->kobj);
M
Mike Marciniszyn 已提交
1548
	if (ret)
1549
		user_remove(dd);
M
Mike Marciniszyn 已提交
1550 1551 1552 1553 1554 1555 1556 1557 1558

	return ret;
}

/*
 * Create per-unit files in /dev
 */
int hfi1_device_create(struct hfi1_devdata *dd)
{
1559
	return user_add(dd);
M
Mike Marciniszyn 已提交
1560 1561 1562 1563 1564 1565 1566 1567 1568 1569
}

/*
 * Remove per-unit files in /dev
 * void, core kernel returns no errors for this stuff
 */
void hfi1_device_remove(struct hfi1_devdata *dd)
{
	user_remove(dd);
}