2 * Copyright (c) 2006 QLogic, Inc. All rights reserved.
3 * Copyright (c) 2003, 2004, 2005, 2006 PathScale, Inc. All rights reserved.
5 * This software is available to you under a choice of one of two
6 * licenses. You may choose to be licensed under the terms of the GNU
7 * General Public License (GPL) Version 2, available from the file
8 * COPYING in the main directory of this source tree, or the
9 * OpenIB.org BSD license below:
11 * Redistribution and use in source and binary forms, with or
12 * without modification, are permitted provided that the following
15 * - Redistributions of source code must retain the above
16 * copyright notice, this list of conditions and the following
19 * - Redistributions in binary form must reproduce the above
20 * copyright notice, this list of conditions and the following
21 * disclaimer in the documentation and/or other materials
22 * provided with the distribution.
24 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
25 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
26 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
27 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
28 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
29 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
30 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
35 #include <linux/device.h>
37 #include "ipath_kernel.h"
39 static void __ipath_release_user_pages(struct page **p, size_t num_pages,
44 for (i = 0; i < num_pages; i++) {
45 ipath_cdbg(MM, "%lu/%lu put_page %p\n", (unsigned long) i,
46 (unsigned long) num_pages, p[i]);
48 set_page_dirty_lock(p[i]);
53 /* call with current->mm->mmap_sem held */
54 static int __get_user_pages(unsigned long start_page, size_t num_pages,
55 struct page **p, struct vm_area_struct **vma)
57 unsigned long lock_limit;
63 * XXX - causes MPI programs to fail, haven't had time to check
66 if (!capable(CAP_IPC_LOCK)) {
72 lock_limit = current->signal->rlim[RLIMIT_MEMLOCK].rlim_cur >>
75 if (num_pages > lock_limit) {
80 ipath_cdbg(VERBOSE, "pin %lx pages from vaddr %lx\n",
81 (unsigned long) num_pages, start_page);
83 for (got = 0; got < num_pages; got += ret) {
84 ret = get_user_pages(current, current->mm,
85 start_page + got * PAGE_SIZE,
86 num_pages - got, 1, 1,
92 current->mm->locked_vm += num_pages;
98 __ipath_release_user_pages(p, got, 0);
104 * ipath_get_user_pages - lock user pages into memory
105 * @start_page: the start page
106 * @num_pages: the number of pages
107 * @p: the output page structures
109 * This function takes a given start page (page aligned user virtual
110 * address) and pins it and the following specified number of pages. For
111 * now, num_pages is always 1, but that will probably change at some point
112 * (because caller is doing expected sends on a single virtually contiguous
113 * buffer, so we can do all pages at once).
115 int ipath_get_user_pages(unsigned long start_page, size_t num_pages,
120 down_write(¤t->mm->mmap_sem);
122 ret = __get_user_pages(start_page, num_pages, p, NULL);
124 up_write(¤t->mm->mmap_sem);
130 * ipath_get_user_pages_nocopy - lock a single page for I/O and mark shared
131 * @start_page: the page to lock
132 * @p: the output page structure
134 * This is similar to ipath_get_user_pages, but it's always one page, and we
135 * mark the page as locked for I/O, and shared. This is used for the user
136 * process page that contains the destination address for the rcvhdrq tail
137 * update, so we need to have the vma. If we don't do this, the page can be
138 * taken away from us on fork, even if the child never touches it, and then
139 * the user process never sees the tail register updates.
141 int ipath_get_user_pages_nocopy(unsigned long page, struct page **p)
143 struct vm_area_struct *vma;
146 down_write(¤t->mm->mmap_sem);
148 ret = __get_user_pages(page, 1, p, &vma);
150 up_write(¤t->mm->mmap_sem);
155 void ipath_release_user_pages(struct page **p, size_t num_pages)
157 down_write(¤t->mm->mmap_sem);
159 __ipath_release_user_pages(p, num_pages, 1);
161 current->mm->locked_vm -= num_pages;
163 up_write(¤t->mm->mmap_sem);
166 struct ipath_user_pages_work {
167 struct work_struct work;
168 struct mm_struct *mm;
169 unsigned long num_pages;
172 static void user_pages_account(void *ptr)
174 struct ipath_user_pages_work *work = ptr;
176 down_write(&work->mm->mmap_sem);
177 work->mm->locked_vm -= work->num_pages;
178 up_write(&work->mm->mmap_sem);
183 void ipath_release_user_pages_on_close(struct page **p, size_t num_pages)
185 struct ipath_user_pages_work *work;
186 struct mm_struct *mm;
188 __ipath_release_user_pages(p, num_pages, 1);
190 mm = get_task_mm(current);
194 work = kmalloc(sizeof(*work), GFP_KERNEL);
200 INIT_WORK(&work->work, user_pages_account, work);
202 work->num_pages = num_pages;