oom: add helpers for setting and clearing TIF_MEMDIE
[cascardo/linux.git] / mm / filemap_xip.c
1 /*
2  *      linux/mm/filemap_xip.c
3  *
4  * Copyright (C) 2005 IBM Corporation
5  * Author: Carsten Otte <cotte@de.ibm.com>
6  *
7  * derived from linux/mm/filemap.c - Copyright (C) Linus Torvalds
8  *
9  */
10
11 #include <linux/fs.h>
12 #include <linux/pagemap.h>
13 #include <linux/export.h>
14 #include <linux/uio.h>
15 #include <linux/rmap.h>
16 #include <linux/mmu_notifier.h>
17 #include <linux/sched.h>
18 #include <linux/seqlock.h>
19 #include <linux/mutex.h>
20 #include <linux/gfp.h>
21 #include <asm/tlbflush.h>
22 #include <asm/io.h>
23
24 /*
25  * We do use our own empty page to avoid interference with other users
26  * of ZERO_PAGE(), such as /dev/zero
27  */
28 static DEFINE_MUTEX(xip_sparse_mutex);
29 static seqcount_t xip_sparse_seq = SEQCNT_ZERO(xip_sparse_seq);
30 static struct page *__xip_sparse_page;
31
32 /* called under xip_sparse_mutex */
33 static struct page *xip_sparse_page(void)
34 {
35         if (!__xip_sparse_page) {
36                 struct page *page = alloc_page(GFP_HIGHUSER | __GFP_ZERO);
37
38                 if (page)
39                         __xip_sparse_page = page;
40         }
41         return __xip_sparse_page;
42 }
43
44 /*
45  * This is a file read routine for execute in place files, and uses
46  * the mapping->a_ops->get_xip_mem() function for the actual low-level
47  * stuff.
48  *
49  * Note the struct file* is not used at all.  It may be NULL.
50  */
51 static ssize_t
52 do_xip_mapping_read(struct address_space *mapping,
53                     struct file_ra_state *_ra,
54                     struct file *filp,
55                     char __user *buf,
56                     size_t len,
57                     loff_t *ppos)
58 {
59         struct inode *inode = mapping->host;
60         pgoff_t index, end_index;
61         unsigned long offset;
62         loff_t isize, pos;
63         size_t copied = 0, error = 0;
64
65         BUG_ON(!mapping->a_ops->get_xip_mem);
66
67         pos = *ppos;
68         index = pos >> PAGE_CACHE_SHIFT;
69         offset = pos & ~PAGE_CACHE_MASK;
70
71         isize = i_size_read(inode);
72         if (!isize)
73                 goto out;
74
75         end_index = (isize - 1) >> PAGE_CACHE_SHIFT;
76         do {
77                 unsigned long nr, left;
78                 void *xip_mem;
79                 unsigned long xip_pfn;
80                 int zero = 0;
81
82                 /* nr is the maximum number of bytes to copy from this page */
83                 nr = PAGE_CACHE_SIZE;
84                 if (index >= end_index) {
85                         if (index > end_index)
86                                 goto out;
87                         nr = ((isize - 1) & ~PAGE_CACHE_MASK) + 1;
88                         if (nr <= offset) {
89                                 goto out;
90                         }
91                 }
92                 nr = nr - offset;
93                 if (nr > len - copied)
94                         nr = len - copied;
95
96                 error = mapping->a_ops->get_xip_mem(mapping, index, 0,
97                                                         &xip_mem, &xip_pfn);
98                 if (unlikely(error)) {
99                         if (error == -ENODATA) {
100                                 /* sparse */
101                                 zero = 1;
102                         } else
103                                 goto out;
104                 }
105
106                 /* If users can be writing to this page using arbitrary
107                  * virtual addresses, take care about potential aliasing
108                  * before reading the page on the kernel side.
109                  */
110                 if (mapping_writably_mapped(mapping))
111                         /* address based flush */ ;
112
113                 /*
114                  * Ok, we have the mem, so now we can copy it to user space...
115                  *
116                  * The actor routine returns how many bytes were actually used..
117                  * NOTE! This may not be the same as how much of a user buffer
118                  * we filled up (we may be padding etc), so we can only update
119                  * "pos" here (the actor routine has to update the user buffer
120                  * pointers and the remaining count).
121                  */
122                 if (!zero)
123                         left = __copy_to_user(buf+copied, xip_mem+offset, nr);
124                 else
125                         left = __clear_user(buf + copied, nr);
126
127                 if (left) {
128                         error = -EFAULT;
129                         goto out;
130                 }
131
132                 copied += (nr - left);
133                 offset += (nr - left);
134                 index += offset >> PAGE_CACHE_SHIFT;
135                 offset &= ~PAGE_CACHE_MASK;
136         } while (copied < len);
137
138 out:
139         *ppos = pos + copied;
140         if (filp)
141                 file_accessed(filp);
142
143         return (copied ? copied : error);
144 }
145
146 ssize_t
147 xip_file_read(struct file *filp, char __user *buf, size_t len, loff_t *ppos)
148 {
149         if (!access_ok(VERIFY_WRITE, buf, len))
150                 return -EFAULT;
151
152         return do_xip_mapping_read(filp->f_mapping, &filp->f_ra, filp,
153                             buf, len, ppos);
154 }
155 EXPORT_SYMBOL_GPL(xip_file_read);
156
157 /*
158  * __xip_unmap is invoked from xip_unmap and xip_write
159  *
160  * This function walks all vmas of the address_space and unmaps the
161  * __xip_sparse_page when found at pgoff.
162  */
163 static void __xip_unmap(struct address_space * mapping, unsigned long pgoff)
164 {
165         struct vm_area_struct *vma;
166         struct page *page;
167         unsigned count;
168         int locked = 0;
169
170         count = read_seqcount_begin(&xip_sparse_seq);
171
172         page = __xip_sparse_page;
173         if (!page)
174                 return;
175
176 retry:
177         i_mmap_lock_read(mapping);
178         vma_interval_tree_foreach(vma, &mapping->i_mmap, pgoff, pgoff) {
179                 pte_t *pte, pteval;
180                 spinlock_t *ptl;
181                 struct mm_struct *mm = vma->vm_mm;
182                 unsigned long address = vma->vm_start +
183                         ((pgoff - vma->vm_pgoff) << PAGE_SHIFT);
184
185                 BUG_ON(address < vma->vm_start || address >= vma->vm_end);
186                 pte = page_check_address(page, mm, address, &ptl, 1);
187                 if (pte) {
188                         /* Nuke the page table entry. */
189                         flush_cache_page(vma, address, pte_pfn(*pte));
190                         pteval = ptep_clear_flush(vma, address, pte);
191                         page_remove_rmap(page);
192                         dec_mm_counter(mm, MM_FILEPAGES);
193                         BUG_ON(pte_dirty(pteval));
194                         pte_unmap_unlock(pte, ptl);
195                         /* must invalidate_page _before_ freeing the page */
196                         mmu_notifier_invalidate_page(mm, address);
197                         page_cache_release(page);
198                 }
199         }
200         i_mmap_unlock_read(mapping);
201
202         if (locked) {
203                 mutex_unlock(&xip_sparse_mutex);
204         } else if (read_seqcount_retry(&xip_sparse_seq, count)) {
205                 mutex_lock(&xip_sparse_mutex);
206                 locked = 1;
207                 goto retry;
208         }
209 }
210
211 /*
212  * xip_fault() is invoked via the vma operations vector for a
213  * mapped memory region to read in file data during a page fault.
214  *
215  * This function is derived from filemap_fault, but used for execute in place
216  */
217 static int xip_file_fault(struct vm_area_struct *vma, struct vm_fault *vmf)
218 {
219         struct file *file = vma->vm_file;
220         struct address_space *mapping = file->f_mapping;
221         struct inode *inode = mapping->host;
222         pgoff_t size;
223         void *xip_mem;
224         unsigned long xip_pfn;
225         struct page *page;
226         int error;
227
228         /* XXX: are VM_FAULT_ codes OK? */
229 again:
230         size = (i_size_read(inode) + PAGE_CACHE_SIZE - 1) >> PAGE_CACHE_SHIFT;
231         if (vmf->pgoff >= size)
232                 return VM_FAULT_SIGBUS;
233
234         error = mapping->a_ops->get_xip_mem(mapping, vmf->pgoff, 0,
235                                                 &xip_mem, &xip_pfn);
236         if (likely(!error))
237                 goto found;
238         if (error != -ENODATA)
239                 return VM_FAULT_OOM;
240
241         /* sparse block */
242         if ((vma->vm_flags & (VM_WRITE | VM_MAYWRITE)) &&
243             (vma->vm_flags & (VM_SHARED | VM_MAYSHARE)) &&
244             (!(mapping->host->i_sb->s_flags & MS_RDONLY))) {
245                 int err;
246
247                 /* maybe shared writable, allocate new block */
248                 mutex_lock(&xip_sparse_mutex);
249                 error = mapping->a_ops->get_xip_mem(mapping, vmf->pgoff, 1,
250                                                         &xip_mem, &xip_pfn);
251                 mutex_unlock(&xip_sparse_mutex);
252                 if (error)
253                         return VM_FAULT_SIGBUS;
254                 /* unmap sparse mappings at pgoff from all other vmas */
255                 __xip_unmap(mapping, vmf->pgoff);
256
257 found:
258                 err = vm_insert_mixed(vma, (unsigned long)vmf->virtual_address,
259                                                         xip_pfn);
260                 if (err == -ENOMEM)
261                         return VM_FAULT_OOM;
262                 /*
263                  * err == -EBUSY is fine, we've raced against another thread
264                  * that faulted-in the same page
265                  */
266                 if (err != -EBUSY)
267                         BUG_ON(err);
268                 return VM_FAULT_NOPAGE;
269         } else {
270                 int err, ret = VM_FAULT_OOM;
271
272                 mutex_lock(&xip_sparse_mutex);
273                 write_seqcount_begin(&xip_sparse_seq);
274                 error = mapping->a_ops->get_xip_mem(mapping, vmf->pgoff, 0,
275                                                         &xip_mem, &xip_pfn);
276                 if (unlikely(!error)) {
277                         write_seqcount_end(&xip_sparse_seq);
278                         mutex_unlock(&xip_sparse_mutex);
279                         goto again;
280                 }
281                 if (error != -ENODATA)
282                         goto out;
283                 /* not shared and writable, use xip_sparse_page() */
284                 page = xip_sparse_page();
285                 if (!page)
286                         goto out;
287                 err = vm_insert_page(vma, (unsigned long)vmf->virtual_address,
288                                                         page);
289                 if (err == -ENOMEM)
290                         goto out;
291
292                 ret = VM_FAULT_NOPAGE;
293 out:
294                 write_seqcount_end(&xip_sparse_seq);
295                 mutex_unlock(&xip_sparse_mutex);
296
297                 return ret;
298         }
299 }
300
301 static const struct vm_operations_struct xip_file_vm_ops = {
302         .fault  = xip_file_fault,
303         .page_mkwrite   = filemap_page_mkwrite,
304 };
305
306 int xip_file_mmap(struct file * file, struct vm_area_struct * vma)
307 {
308         BUG_ON(!file->f_mapping->a_ops->get_xip_mem);
309
310         file_accessed(file);
311         vma->vm_ops = &xip_file_vm_ops;
312         vma->vm_flags |= VM_MIXEDMAP;
313         return 0;
314 }
315 EXPORT_SYMBOL_GPL(xip_file_mmap);
316
317 static ssize_t
318 __xip_file_write(struct file *filp, const char __user *buf,
319                   size_t count, loff_t pos, loff_t *ppos)
320 {
321         struct address_space * mapping = filp->f_mapping;
322         const struct address_space_operations *a_ops = mapping->a_ops;
323         struct inode    *inode = mapping->host;
324         long            status = 0;
325         size_t          bytes;
326         ssize_t         written = 0;
327
328         BUG_ON(!mapping->a_ops->get_xip_mem);
329
330         do {
331                 unsigned long index;
332                 unsigned long offset;
333                 size_t copied;
334                 void *xip_mem;
335                 unsigned long xip_pfn;
336
337                 offset = (pos & (PAGE_CACHE_SIZE -1)); /* Within page */
338                 index = pos >> PAGE_CACHE_SHIFT;
339                 bytes = PAGE_CACHE_SIZE - offset;
340                 if (bytes > count)
341                         bytes = count;
342
343                 status = a_ops->get_xip_mem(mapping, index, 0,
344                                                 &xip_mem, &xip_pfn);
345                 if (status == -ENODATA) {
346                         /* we allocate a new page unmap it */
347                         mutex_lock(&xip_sparse_mutex);
348                         status = a_ops->get_xip_mem(mapping, index, 1,
349                                                         &xip_mem, &xip_pfn);
350                         mutex_unlock(&xip_sparse_mutex);
351                         if (!status)
352                                 /* unmap page at pgoff from all other vmas */
353                                 __xip_unmap(mapping, index);
354                 }
355
356                 if (status)
357                         break;
358
359                 copied = bytes -
360                         __copy_from_user_nocache(xip_mem + offset, buf, bytes);
361
362                 if (likely(copied > 0)) {
363                         status = copied;
364
365                         if (status >= 0) {
366                                 written += status;
367                                 count -= status;
368                                 pos += status;
369                                 buf += status;
370                         }
371                 }
372                 if (unlikely(copied != bytes))
373                         if (status >= 0)
374                                 status = -EFAULT;
375                 if (status < 0)
376                         break;
377         } while (count);
378         *ppos = pos;
379         /*
380          * No need to use i_size_read() here, the i_size
381          * cannot change under us because we hold i_mutex.
382          */
383         if (pos > inode->i_size) {
384                 i_size_write(inode, pos);
385                 mark_inode_dirty(inode);
386         }
387
388         return written ? written : status;
389 }
390
391 ssize_t
392 xip_file_write(struct file *filp, const char __user *buf, size_t len,
393                loff_t *ppos)
394 {
395         struct address_space *mapping = filp->f_mapping;
396         struct inode *inode = mapping->host;
397         size_t count;
398         loff_t pos;
399         ssize_t ret;
400
401         mutex_lock(&inode->i_mutex);
402
403         if (!access_ok(VERIFY_READ, buf, len)) {
404                 ret=-EFAULT;
405                 goto out_up;
406         }
407
408         pos = *ppos;
409         count = len;
410
411         /* We can write back this queue in page reclaim */
412         current->backing_dev_info = mapping->backing_dev_info;
413
414         ret = generic_write_checks(filp, &pos, &count, S_ISBLK(inode->i_mode));
415         if (ret)
416                 goto out_backing;
417         if (count == 0)
418                 goto out_backing;
419
420         ret = file_remove_suid(filp);
421         if (ret)
422                 goto out_backing;
423
424         ret = file_update_time(filp);
425         if (ret)
426                 goto out_backing;
427
428         ret = __xip_file_write (filp, buf, count, pos, ppos);
429
430  out_backing:
431         current->backing_dev_info = NULL;
432  out_up:
433         mutex_unlock(&inode->i_mutex);
434         return ret;
435 }
436 EXPORT_SYMBOL_GPL(xip_file_write);
437
438 /*
439  * truncate a page used for execute in place
440  * functionality is analog to block_truncate_page but does use get_xip_mem
441  * to get the page instead of page cache
442  */
443 int
444 xip_truncate_page(struct address_space *mapping, loff_t from)
445 {
446         pgoff_t index = from >> PAGE_CACHE_SHIFT;
447         unsigned offset = from & (PAGE_CACHE_SIZE-1);
448         unsigned blocksize;
449         unsigned length;
450         void *xip_mem;
451         unsigned long xip_pfn;
452         int err;
453
454         BUG_ON(!mapping->a_ops->get_xip_mem);
455
456         blocksize = 1 << mapping->host->i_blkbits;
457         length = offset & (blocksize - 1);
458
459         /* Block boundary? Nothing to do */
460         if (!length)
461                 return 0;
462
463         length = blocksize - length;
464
465         err = mapping->a_ops->get_xip_mem(mapping, index, 0,
466                                                 &xip_mem, &xip_pfn);
467         if (unlikely(err)) {
468                 if (err == -ENODATA)
469                         /* Hole? No need to truncate */
470                         return 0;
471                 else
472                         return err;
473         }
474         memset(xip_mem + offset, 0, length);
475         return 0;
476 }
477 EXPORT_SYMBOL_GPL(xip_truncate_page);