2 * Copyright (c) 2003, 2004, 2005, 2006 PathScale, Inc. All rights reserved.
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:
10 * Redistribution and use in source and binary forms, with or
11 * without modification, are permitted provided that the following
14 * - Redistributions of source code must retain the above
15 * copyright notice, this list of conditions and the following
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.
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
34 #include <linux/device.h>
35 #include <linux/vs_memory.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 ||
76 !vx_vmlocked_avail(current->mm, num_pages)) {
81 ipath_cdbg(VERBOSE, "pin %lx pages from vaddr %lx\n",
82 (unsigned long) num_pages, start_page);
84 for (got = 0; got < num_pages; got += ret) {
85 ret = get_user_pages(current, current->mm,
86 start_page + got * PAGE_SIZE,
87 num_pages - got, 1, 1,
93 vx_vmlocked_add(current->mm, num_pages);
99 __ipath_release_user_pages(p, got, 0);
105 * ipath_get_user_pages - lock user pages into memory
106 * @start_page: the start page
107 * @num_pages: the number of pages
108 * @p: the output page structures
110 * This function takes a given start page (page aligned user virtual
111 * address) and pins it and the following specified number of pages. For
112 * now, num_pages is always 1, but that will probably change at some point
113 * (because caller is doing expected sends on a single virtually contiguous
114 * buffer, so we can do all pages at once).
116 int ipath_get_user_pages(unsigned long start_page, size_t num_pages,
121 down_write(¤t->mm->mmap_sem);
123 ret = __get_user_pages(start_page, num_pages, p, NULL);
125 up_write(¤t->mm->mmap_sem);
131 * ipath_get_user_pages_nocopy - lock a single page for I/O and mark shared
132 * @start_page: the page to lock
133 * @p: the output page structure
135 * This is similar to ipath_get_user_pages, but it's always one page, and we
136 * mark the page as locked for I/O, and shared. This is used for the user
137 * process page that contains the destination address for the rcvhdrq tail
138 * update, so we need to have the vma. If we don't do this, the page can be
139 * taken away from us on fork, even if the child never touches it, and then
140 * the user process never sees the tail register updates.
142 int ipath_get_user_pages_nocopy(unsigned long page, struct page **p)
144 struct vm_area_struct *vma;
147 down_write(¤t->mm->mmap_sem);
149 ret = __get_user_pages(page, 1, p, &vma);
151 up_write(¤t->mm->mmap_sem);
156 void ipath_release_user_pages(struct page **p, size_t num_pages)
158 down_write(¤t->mm->mmap_sem);
160 __ipath_release_user_pages(p, num_pages, 1);
162 vx_vmlocked_sub(current->mm, num_pages);
164 up_write(¤t->mm->mmap_sem);
167 struct ipath_user_pages_work {
168 struct work_struct work;
169 struct mm_struct *mm;
170 unsigned long num_pages;
173 static void user_pages_account(void *ptr)
175 struct ipath_user_pages_work *work = ptr;
177 down_write(&work->mm->mmap_sem);
178 vx_vmlocked_sub(work->mm, work->num_pages);
179 up_write(&work->mm->mmap_sem);
184 void ipath_release_user_pages_on_close(struct page **p, size_t num_pages)
186 struct ipath_user_pages_work *work;
187 struct mm_struct *mm;
189 __ipath_release_user_pages(p, num_pages, 1);
191 mm = get_task_mm(current);
195 work = kmalloc(sizeof(*work), GFP_KERNEL);
201 INIT_WORK(&work->work, user_pages_account, work);
203 work->num_pages = num_pages;