Btrfs: Fix hole insertion corner cases
[safe/jmp/linux-2.6] / fs / btrfs / file.c
1 /*
2  * Copyright (C) 2007 Oracle.  All rights reserved.
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public
6  * License v2 as published by the Free Software Foundation.
7  *
8  * This program is distributed in the hope that it will be useful,
9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11  * General Public License for more details.
12  *
13  * You should have received a copy of the GNU General Public
14  * License along with this program; if not, write to the
15  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
16  * Boston, MA 021110-1307, USA.
17  */
18
19 #include <linux/fs.h>
20 #include <linux/pagemap.h>
21 #include <linux/highmem.h>
22 #include <linux/time.h>
23 #include <linux/init.h>
24 #include <linux/string.h>
25 #include <linux/smp_lock.h>
26 #include <linux/backing-dev.h>
27 #include <linux/mpage.h>
28 #include <linux/swap.h>
29 #include <linux/writeback.h>
30 #include <linux/statfs.h>
31 #include <linux/compat.h>
32 #include <linux/version.h>
33 #include "ctree.h"
34 #include "disk-io.h"
35 #include "transaction.h"
36 #include "btrfs_inode.h"
37 #include "ordered-data.h"
38 #include "ioctl.h"
39 #include "print-tree.h"
40
41
42 static int btrfs_copy_from_user(loff_t pos, int num_pages, int write_bytes,
43                                 struct page **prepared_pages,
44                                 const char __user * buf)
45 {
46         long page_fault = 0;
47         int i;
48         int offset = pos & (PAGE_CACHE_SIZE - 1);
49
50         for (i = 0; i < num_pages && write_bytes > 0; i++, offset = 0) {
51                 size_t count = min_t(size_t,
52                                      PAGE_CACHE_SIZE - offset, write_bytes);
53                 struct page *page = prepared_pages[i];
54                 fault_in_pages_readable(buf, count);
55
56                 /* Copy data from userspace to the current page */
57                 kmap(page);
58                 page_fault = __copy_from_user(page_address(page) + offset,
59                                               buf, count);
60                 /* Flush processor's dcache for this page */
61                 flush_dcache_page(page);
62                 kunmap(page);
63                 buf += count;
64                 write_bytes -= count;
65
66                 if (page_fault)
67                         break;
68         }
69         return page_fault ? -EFAULT : 0;
70 }
71
72 static void btrfs_drop_pages(struct page **pages, size_t num_pages)
73 {
74         size_t i;
75         for (i = 0; i < num_pages; i++) {
76                 if (!pages[i])
77                         break;
78                 unlock_page(pages[i]);
79                 mark_page_accessed(pages[i]);
80                 page_cache_release(pages[i]);
81         }
82 }
83
84 static int noinline insert_inline_extent(struct btrfs_trans_handle *trans,
85                                 struct btrfs_root *root, struct inode *inode,
86                                 u64 offset, size_t size,
87                                 struct page **pages, size_t page_offset,
88                                 int num_pages)
89 {
90         struct btrfs_key key;
91         struct btrfs_path *path;
92         struct extent_buffer *leaf;
93         char *kaddr;
94         unsigned long ptr;
95         struct btrfs_file_extent_item *ei;
96         struct page *page;
97         u32 datasize;
98         int err = 0;
99         int ret;
100         int i;
101         ssize_t cur_size;
102
103         path = btrfs_alloc_path();
104         if (!path)
105                 return -ENOMEM;
106
107         btrfs_set_trans_block_group(trans, inode);
108
109         key.objectid = inode->i_ino;
110         key.offset = offset;
111         btrfs_set_key_type(&key, BTRFS_EXTENT_DATA_KEY);
112
113         ret = btrfs_search_slot(trans, root, &key, path, 0, 1);
114         if (ret < 0) {
115                 err = ret;
116                 goto fail;
117         }
118         if (ret == 1) {
119                 struct btrfs_key found_key;
120
121                 if (path->slots[0] == 0)
122                         goto insert;
123
124                 path->slots[0]--;
125                 leaf = path->nodes[0];
126                 btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
127
128                 if (found_key.objectid != inode->i_ino)
129                         goto insert;
130
131                 if (found_key.type != BTRFS_EXTENT_DATA_KEY)
132                         goto insert;
133                 ei = btrfs_item_ptr(leaf, path->slots[0],
134                                     struct btrfs_file_extent_item);
135
136                 if (btrfs_file_extent_type(leaf, ei) !=
137                     BTRFS_FILE_EXTENT_INLINE) {
138                         goto insert;
139                 }
140                 btrfs_item_key_to_cpu(leaf, &key, path->slots[0]);
141                 ret = 0;
142         }
143         if (ret == 0) {
144                 u32 found_size;
145                 u64 found_end;
146
147                 leaf = path->nodes[0];
148                 ei = btrfs_item_ptr(leaf, path->slots[0],
149                                     struct btrfs_file_extent_item);
150
151                 if (btrfs_file_extent_type(leaf, ei) !=
152                     BTRFS_FILE_EXTENT_INLINE) {
153                         err = ret;
154                         btrfs_print_leaf(root, leaf);
155                         printk("found wasn't inline offset %Lu inode %lu\n",
156                                offset, inode->i_ino);
157                         goto fail;
158                 }
159                 found_size = btrfs_file_extent_inline_len(leaf,
160                                           btrfs_item_nr(leaf, path->slots[0]));
161                 found_end = key.offset + found_size;
162
163                 if (found_end < offset + size) {
164                         btrfs_release_path(root, path);
165                         ret = btrfs_search_slot(trans, root, &key, path,
166                                                 offset + size - found_end, 1);
167                         BUG_ON(ret != 0);
168
169                         ret = btrfs_extend_item(trans, root, path,
170                                                 offset + size - found_end);
171                         if (ret) {
172                                 err = ret;
173                                 goto fail;
174                         }
175                         leaf = path->nodes[0];
176                         ei = btrfs_item_ptr(leaf, path->slots[0],
177                                             struct btrfs_file_extent_item);
178                 }
179                 if (found_end < offset) {
180                         ptr = btrfs_file_extent_inline_start(ei) + found_size;
181                         memset_extent_buffer(leaf, 0, ptr, offset - found_end);
182                 }
183         } else {
184 insert:
185                 btrfs_release_path(root, path);
186                 datasize = offset + size - key.offset;
187                 datasize = btrfs_file_extent_calc_inline_size(datasize);
188                 ret = btrfs_insert_empty_item(trans, root, path, &key,
189                                               datasize);
190                 if (ret) {
191                         err = ret;
192                         printk("got bad ret %d\n", ret);
193                         goto fail;
194                 }
195                 leaf = path->nodes[0];
196                 ei = btrfs_item_ptr(leaf, path->slots[0],
197                                     struct btrfs_file_extent_item);
198                 btrfs_set_file_extent_generation(leaf, ei, trans->transid);
199                 btrfs_set_file_extent_type(leaf, ei, BTRFS_FILE_EXTENT_INLINE);
200         }
201         ptr = btrfs_file_extent_inline_start(ei) + offset - key.offset;
202
203         cur_size = size;
204         i = 0;
205         while (size > 0) {
206                 page = pages[i];
207                 kaddr = kmap_atomic(page, KM_USER0);
208                 cur_size = min_t(size_t, PAGE_CACHE_SIZE - page_offset, size);
209                 write_extent_buffer(leaf, kaddr + page_offset, ptr, cur_size);
210                 kunmap_atomic(kaddr, KM_USER0);
211                 page_offset = 0;
212                 ptr += cur_size;
213                 size -= cur_size;
214                 if (i >= num_pages) {
215                         printk("i %d num_pages %d\n", i, num_pages);
216                 }
217                 i++;
218         }
219         btrfs_mark_buffer_dirty(leaf);
220 fail:
221         btrfs_free_path(path);
222         return err;
223 }
224
225 static int noinline dirty_and_release_pages(struct btrfs_trans_handle *trans,
226                                    struct btrfs_root *root,
227                                    struct file *file,
228                                    struct page **pages,
229                                    size_t num_pages,
230                                    loff_t pos,
231                                    size_t write_bytes)
232 {
233         int err = 0;
234         int i;
235         struct inode *inode = fdentry(file)->d_inode;
236         struct extent_map *em;
237         struct extent_map_tree *em_tree = &BTRFS_I(inode)->extent_tree;
238         u64 hint_byte;
239         u64 num_bytes;
240         u64 start_pos;
241         u64 end_of_last_block;
242         u64 end_pos = pos + write_bytes;
243         u64 inline_size;
244         loff_t isize = i_size_read(inode);
245         em = alloc_extent_map(GFP_NOFS);
246         if (!em)
247                 return -ENOMEM;
248
249         em->bdev = inode->i_sb->s_bdev;
250
251         start_pos = pos & ~((u64)root->sectorsize - 1);
252         num_bytes = (write_bytes + pos - start_pos +
253                     root->sectorsize - 1) & ~((u64)root->sectorsize - 1);
254
255         end_of_last_block = start_pos + num_bytes - 1;
256
257         lock_extent(em_tree, start_pos, end_of_last_block, GFP_NOFS);
258         mutex_lock(&root->fs_info->fs_mutex);
259         trans = btrfs_start_transaction(root, 1);
260         if (!trans) {
261                 err = -ENOMEM;
262                 goto out_unlock;
263         }
264         btrfs_set_trans_block_group(trans, inode);
265         inode->i_blocks += num_bytes >> 9;
266         hint_byte = 0;
267
268         if ((end_of_last_block & 4095) == 0) {
269                 printk("strange end of last %Lu %zu %Lu\n", start_pos, write_bytes, end_of_last_block);
270         }
271         set_extent_uptodate(em_tree, start_pos, end_of_last_block, GFP_NOFS);
272
273         /* FIXME...EIEIO, ENOSPC and more */
274
275         /* insert any holes we need to create */
276         if (inode->i_size < start_pos) {
277                 u64 last_pos_in_file;
278                 u64 hole_size;
279                 u64 mask = root->sectorsize - 1;
280                 last_pos_in_file = (isize + mask) & ~mask;
281                 hole_size = (end_pos - last_pos_in_file + mask) & ~mask;
282
283                 if (last_pos_in_file < start_pos) {
284                         err = btrfs_drop_extents(trans, root, inode,
285                                                  last_pos_in_file,
286                                                  last_pos_in_file + hole_size,
287                                                  last_pos_in_file,
288                                                  &hint_byte);
289                         if (err)
290                                 goto failed;
291
292                         err = btrfs_insert_file_extent(trans, root,
293                                                        inode->i_ino,
294                                                        last_pos_in_file,
295                                                        0, 0, hole_size);
296                         btrfs_check_file(root, inode);
297                 }
298                 if (err)
299                         goto failed;
300         }
301
302         /*
303          * either allocate an extent for the new bytes or setup the key
304          * to show we are doing inline data in the extent
305          */
306         inline_size = end_pos;
307         if (isize >= BTRFS_MAX_INLINE_DATA_SIZE(root) ||
308             inline_size > 8192 ||
309             inline_size >= BTRFS_MAX_INLINE_DATA_SIZE(root)) {
310                 u64 last_end;
311                 u64 existing_delalloc = 0;
312
313                 for (i = 0; i < num_pages; i++) {
314                         struct page *p = pages[i];
315                         SetPageUptodate(p);
316                         set_page_dirty(p);
317                 }
318                 last_end = (u64)(pages[num_pages -1]->index) <<
319                                 PAGE_CACHE_SHIFT;
320                 last_end += PAGE_CACHE_SIZE - 1;
321                 if (start_pos < isize) {
322                         u64 delalloc_start = start_pos;
323                         existing_delalloc = count_range_bits(em_tree,
324                                              &delalloc_start,
325                                              end_of_last_block, (u64)-1,
326                                              EXTENT_DELALLOC);
327                 }
328                 set_extent_delalloc(em_tree, start_pos, end_of_last_block,
329                                  GFP_NOFS);
330                 spin_lock(&root->fs_info->delalloc_lock);
331                 root->fs_info->delalloc_bytes += (end_of_last_block + 1 -
332                                           start_pos) - existing_delalloc;
333                 spin_unlock(&root->fs_info->delalloc_lock);
334                 btrfs_add_ordered_inode(inode);
335         } else {
336                 u64 aligned_end;
337                 /* step one, delete the existing extents in this range */
338                 aligned_end = (pos + write_bytes + root->sectorsize - 1) &
339                         ~((u64)root->sectorsize - 1);
340                 err = btrfs_drop_extents(trans, root, inode, start_pos,
341                                          aligned_end, aligned_end, &hint_byte);
342                 if (err)
343                         goto failed;
344                 if (isize > inline_size)
345                         inline_size = min_t(u64, isize, aligned_end);
346                 inline_size -= start_pos;
347                 err = insert_inline_extent(trans, root, inode, start_pos,
348                                            inline_size, pages, 0, num_pages);
349                 BUG_ON(err);
350         }
351         if (end_pos > isize) {
352                 i_size_write(inode, end_pos);
353                 btrfs_update_inode(trans, root, inode);
354         }
355 failed:
356         err = btrfs_end_transaction(trans, root);
357 out_unlock:
358         mutex_unlock(&root->fs_info->fs_mutex);
359         unlock_extent(em_tree, start_pos, end_of_last_block, GFP_NOFS);
360         free_extent_map(em);
361         return err;
362 }
363
364 int btrfs_drop_extent_cache(struct inode *inode, u64 start, u64 end)
365 {
366         struct extent_map *em;
367         struct extent_map_tree *em_tree = &BTRFS_I(inode)->extent_tree;
368
369         while(1) {
370                 em = lookup_extent_mapping(em_tree, start, end);
371                 if (!em)
372                         break;
373                 remove_extent_mapping(em_tree, em);
374                 /* once for us */
375                 free_extent_map(em);
376                 /* once for the tree*/
377                 free_extent_map(em);
378         }
379         return 0;
380 }
381
382 int btrfs_check_file(struct btrfs_root *root, struct inode *inode)
383 {
384         return 0;
385 #if 0
386         struct btrfs_path *path;
387         struct btrfs_key found_key;
388         struct extent_buffer *leaf;
389         struct btrfs_file_extent_item *extent;
390         u64 last_offset = 0;
391         int nritems;
392         int slot;
393         int found_type;
394         int ret;
395         int err = 0;
396         u64 extent_end = 0;
397
398         path = btrfs_alloc_path();
399         ret = btrfs_lookup_file_extent(NULL, root, path, inode->i_ino,
400                                        last_offset, 0);
401         while(1) {
402                 nritems = btrfs_header_nritems(path->nodes[0]);
403                 if (path->slots[0] >= nritems) {
404                         ret = btrfs_next_leaf(root, path);
405                         if (ret)
406                                 goto out;
407                         nritems = btrfs_header_nritems(path->nodes[0]);
408                 }
409                 slot = path->slots[0];
410                 leaf = path->nodes[0];
411                 btrfs_item_key_to_cpu(leaf, &found_key, slot);
412                 if (found_key.objectid != inode->i_ino)
413                         break;
414                 if (found_key.type != BTRFS_EXTENT_DATA_KEY)
415                         goto out;
416
417                 if (found_key.offset != last_offset) {
418                         WARN_ON(1);
419                         btrfs_print_leaf(root, leaf);
420                         printk("inode %lu found offset %Lu expected %Lu\n",
421                                inode->i_ino, found_key.offset, last_offset);
422                         err = 1;
423                         goto out;
424                 }
425                 extent = btrfs_item_ptr(leaf, slot,
426                                         struct btrfs_file_extent_item);
427                 found_type = btrfs_file_extent_type(leaf, extent);
428                 if (found_type == BTRFS_FILE_EXTENT_REG) {
429                         extent_end = found_key.offset +
430                              btrfs_file_extent_num_bytes(leaf, extent);
431                 } else if (found_type == BTRFS_FILE_EXTENT_INLINE) {
432                         struct btrfs_item *item;
433                         item = btrfs_item_nr(leaf, slot);
434                         extent_end = found_key.offset +
435                              btrfs_file_extent_inline_len(leaf, item);
436                         extent_end = (extent_end + root->sectorsize - 1) &
437                                 ~((u64)root->sectorsize -1 );
438                 }
439                 last_offset = extent_end;
440                 path->slots[0]++;
441         }
442         if (last_offset < inode->i_size) {
443                 WARN_ON(1);
444                 btrfs_print_leaf(root, leaf);
445                 printk("inode %lu found offset %Lu size %Lu\n", inode->i_ino,
446                        last_offset, inode->i_size);
447                 err = 1;
448
449         }
450 out:
451         btrfs_free_path(path);
452         return err;
453 #endif
454 }
455
456 /*
457  * this is very complex, but the basic idea is to drop all extents
458  * in the range start - end.  hint_block is filled in with a block number
459  * that would be a good hint to the block allocator for this file.
460  *
461  * If an extent intersects the range but is not entirely inside the range
462  * it is either truncated or split.  Anything entirely inside the range
463  * is deleted from the tree.
464  */
465 int btrfs_drop_extents(struct btrfs_trans_handle *trans,
466                        struct btrfs_root *root, struct inode *inode,
467                        u64 start, u64 end, u64 inline_limit, u64 *hint_byte)
468 {
469         u64 extent_end = 0;
470         u64 search_start = start;
471         struct extent_buffer *leaf;
472         struct btrfs_file_extent_item *extent;
473         struct btrfs_path *path;
474         struct btrfs_key key;
475         struct btrfs_file_extent_item old;
476         int keep;
477         int slot;
478         int bookend;
479         int found_type;
480         int found_extent;
481         int found_inline;
482         int recow;
483         int ret;
484
485         btrfs_drop_extent_cache(inode, start, end - 1);
486
487         path = btrfs_alloc_path();
488         if (!path)
489                 return -ENOMEM;
490         while(1) {
491                 recow = 0;
492                 btrfs_release_path(root, path);
493                 ret = btrfs_lookup_file_extent(trans, root, path, inode->i_ino,
494                                                search_start, -1);
495                 if (ret < 0)
496                         goto out;
497                 if (ret > 0) {
498                         if (path->slots[0] == 0) {
499                                 ret = 0;
500                                 goto out;
501                         }
502                         path->slots[0]--;
503                 }
504 next_slot:
505                 keep = 0;
506                 bookend = 0;
507                 found_extent = 0;
508                 found_inline = 0;
509                 extent = NULL;
510                 leaf = path->nodes[0];
511                 slot = path->slots[0];
512                 ret = 0;
513                 btrfs_item_key_to_cpu(leaf, &key, slot);
514
515                 if (key.offset >= end || key.objectid != inode->i_ino) {
516                         goto out;
517                 }
518                 if (btrfs_key_type(&key) > BTRFS_EXTENT_DATA_KEY) {
519                         goto out;
520                 }
521                 if (recow) {
522                         search_start = key.offset;
523                         continue;
524                 }
525                 if (btrfs_key_type(&key) == BTRFS_EXTENT_DATA_KEY) {
526                         extent = btrfs_item_ptr(leaf, slot,
527                                                 struct btrfs_file_extent_item);
528                         found_type = btrfs_file_extent_type(leaf, extent);
529                         if (found_type == BTRFS_FILE_EXTENT_REG) {
530                                 extent_end =
531                                      btrfs_file_extent_disk_bytenr(leaf,
532                                                                    extent);
533                                 if (extent_end)
534                                         *hint_byte = extent_end;
535
536                                 extent_end = key.offset +
537                                      btrfs_file_extent_num_bytes(leaf, extent);
538                                 found_extent = 1;
539                         } else if (found_type == BTRFS_FILE_EXTENT_INLINE) {
540                                 struct btrfs_item *item;
541                                 item = btrfs_item_nr(leaf, slot);
542                                 found_inline = 1;
543                                 extent_end = key.offset +
544                                      btrfs_file_extent_inline_len(leaf, item);
545                         }
546                 } else {
547                         extent_end = search_start;
548                 }
549
550                 /* we found nothing we can drop */
551                 if ((!found_extent && !found_inline) ||
552                     search_start >= extent_end) {
553                         int nextret;
554                         u32 nritems;
555                         nritems = btrfs_header_nritems(leaf);
556                         if (slot >= nritems - 1) {
557                                 nextret = btrfs_next_leaf(root, path);
558                                 if (nextret)
559                                         goto out;
560                                 recow = 1;
561                         } else {
562                                 path->slots[0]++;
563                         }
564                         goto next_slot;
565                 }
566
567                 if (found_inline) {
568                         u64 mask = root->sectorsize - 1;
569                         search_start = (extent_end + mask) & ~mask;
570                 } else
571                         search_start = extent_end;
572                 if (end <= extent_end && start >= key.offset && found_inline) {
573                         *hint_byte = EXTENT_MAP_INLINE;
574                         continue;
575                 }
576                 if (end < extent_end && end >= key.offset) {
577                         if (found_extent) {
578                                 u64 disk_bytenr =
579                                     btrfs_file_extent_disk_bytenr(leaf, extent);
580                                 u64 disk_num_bytes =
581                                     btrfs_file_extent_disk_num_bytes(leaf,
582                                                                       extent);
583                                 read_extent_buffer(leaf, &old,
584                                                    (unsigned long)extent,
585                                                    sizeof(old));
586                                 if (disk_bytenr != 0) {
587                                         ret = btrfs_inc_extent_ref(trans, root,
588                                                  disk_bytenr, disk_num_bytes,
589                                                  root->root_key.objectid,
590                                                  trans->transid,
591                                                  key.objectid, end);
592                                         BUG_ON(ret);
593                                 }
594                         }
595                         bookend = 1;
596                         if (found_inline && start <= key.offset &&
597                             inline_limit < extent_end)
598                                 keep = 1;
599                 }
600                 /* truncate existing extent */
601                 if (start > key.offset) {
602                         u64 new_num;
603                         u64 old_num;
604                         keep = 1;
605                         WARN_ON(start & (root->sectorsize - 1));
606                         if (found_extent) {
607                                 new_num = start - key.offset;
608                                 old_num = btrfs_file_extent_num_bytes(leaf,
609                                                                       extent);
610                                 *hint_byte =
611                                         btrfs_file_extent_disk_bytenr(leaf,
612                                                                       extent);
613                                 if (btrfs_file_extent_disk_bytenr(leaf,
614                                                                   extent)) {
615                                         inode->i_blocks -=
616                                                 (old_num - new_num) >> 9;
617                                 }
618                                 btrfs_set_file_extent_num_bytes(leaf, extent,
619                                                                 new_num);
620                                 btrfs_mark_buffer_dirty(leaf);
621                         } else if (key.offset < inline_limit &&
622                                    (end > extent_end) &&
623                                    (inline_limit < extent_end)) {
624                                 u32 new_size;
625                                 new_size = btrfs_file_extent_calc_inline_size(
626                                                    inline_limit - key.offset);
627                                 btrfs_truncate_item(trans, root, path,
628                                                     new_size, 1);
629                         }
630                 }
631                 /* delete the entire extent */
632                 if (!keep) {
633                         u64 disk_bytenr = 0;
634                         u64 disk_num_bytes = 0;
635                         u64 extent_num_bytes = 0;
636                         u64 root_gen;
637                         u64 root_owner;
638
639                         root_gen = btrfs_header_generation(leaf);
640                         root_owner = btrfs_header_owner(leaf);
641                         if (found_extent) {
642                                 disk_bytenr =
643                                       btrfs_file_extent_disk_bytenr(leaf,
644                                                                      extent);
645                                 disk_num_bytes =
646                                       btrfs_file_extent_disk_num_bytes(leaf,
647                                                                        extent);
648                                 extent_num_bytes =
649                                       btrfs_file_extent_num_bytes(leaf, extent);
650                                 *hint_byte =
651                                         btrfs_file_extent_disk_bytenr(leaf,
652                                                                       extent);
653                         }
654                         ret = btrfs_del_item(trans, root, path);
655                         /* TODO update progress marker and return */
656                         BUG_ON(ret);
657                         btrfs_release_path(root, path);
658                         extent = NULL;
659                         if (found_extent && disk_bytenr != 0) {
660                                 inode->i_blocks -= extent_num_bytes >> 9;
661                                 ret = btrfs_free_extent(trans, root,
662                                                 disk_bytenr,
663                                                 disk_num_bytes,
664                                                 root_owner,
665                                                 root_gen, inode->i_ino,
666                                                 key.offset, 0);
667                         }
668
669                         BUG_ON(ret);
670                         if (!bookend && search_start >= end) {
671                                 ret = 0;
672                                 goto out;
673                         }
674                         if (!bookend)
675                                 continue;
676                 }
677                 if (bookend && found_inline && start <= key.offset &&
678                     inline_limit < extent_end && key.offset <= inline_limit) {
679                         u32 new_size;
680                         new_size = btrfs_file_extent_calc_inline_size(
681                                                    extent_end - inline_limit);
682                         btrfs_truncate_item(trans, root, path, new_size, 0);
683                 }
684                 /* create bookend, splitting the extent in two */
685                 if (bookend && found_extent) {
686                         struct btrfs_key ins;
687                         ins.objectid = inode->i_ino;
688                         ins.offset = end;
689                         btrfs_set_key_type(&ins, BTRFS_EXTENT_DATA_KEY);
690                         btrfs_release_path(root, path);
691                         ret = btrfs_insert_empty_item(trans, root, path, &ins,
692                                                       sizeof(*extent));
693
694                         leaf = path->nodes[0];
695                         if (ret) {
696                                 btrfs_print_leaf(root, leaf);
697                                 printk("got %d on inserting %Lu %u %Lu start %Lu end %Lu found %Lu %Lu keep was %d\n", ret , ins.objectid, ins.type, ins.offset, start, end, key.offset, extent_end, keep);
698                         }
699                         BUG_ON(ret);
700                         extent = btrfs_item_ptr(leaf, path->slots[0],
701                                                 struct btrfs_file_extent_item);
702                         write_extent_buffer(leaf, &old,
703                                             (unsigned long)extent, sizeof(old));
704
705                         btrfs_set_file_extent_offset(leaf, extent,
706                                     le64_to_cpu(old.offset) + end - key.offset);
707                         WARN_ON(le64_to_cpu(old.num_bytes) <
708                                 (extent_end - end));
709                         btrfs_set_file_extent_num_bytes(leaf, extent,
710                                                         extent_end - end);
711                         btrfs_set_file_extent_type(leaf, extent,
712                                                    BTRFS_FILE_EXTENT_REG);
713
714                         btrfs_mark_buffer_dirty(path->nodes[0]);
715                         if (le64_to_cpu(old.disk_bytenr) != 0) {
716                                 inode->i_blocks +=
717                                       btrfs_file_extent_num_bytes(leaf,
718                                                                   extent) >> 9;
719                         }
720                         ret = 0;
721                         goto out;
722                 }
723         }
724 out:
725         btrfs_free_path(path);
726         return ret;
727 }
728
729 /*
730  * this gets pages into the page cache and locks them down
731  */
732 static int prepare_pages(struct btrfs_root *root, struct file *file,
733                          struct page **pages, size_t num_pages,
734                          loff_t pos, unsigned long first_index,
735                          unsigned long last_index, size_t write_bytes)
736 {
737         int i;
738         unsigned long index = pos >> PAGE_CACHE_SHIFT;
739         struct inode *inode = fdentry(file)->d_inode;
740         int err = 0;
741         u64 start_pos;
742
743         start_pos = pos & ~((u64)root->sectorsize - 1);
744
745         memset(pages, 0, num_pages * sizeof(struct page *));
746
747         for (i = 0; i < num_pages; i++) {
748                 pages[i] = grab_cache_page(inode->i_mapping, index + i);
749                 if (!pages[i]) {
750                         err = -ENOMEM;
751                         BUG_ON(1);
752                 }
753 #if LINUX_VERSION_CODE <= KERNEL_VERSION(2,6,18)
754                 ClearPageDirty(pages[i]);
755 #else
756                 cancel_dirty_page(pages[i], PAGE_CACHE_SIZE);
757 #endif
758                 wait_on_page_writeback(pages[i]);
759                 set_page_extent_mapped(pages[i]);
760                 WARN_ON(!PageLocked(pages[i]));
761         }
762         return 0;
763 }
764
765 static ssize_t btrfs_file_write(struct file *file, const char __user *buf,
766                                 size_t count, loff_t *ppos)
767 {
768         loff_t pos;
769         loff_t start_pos;
770         ssize_t num_written = 0;
771         ssize_t err = 0;
772         int ret = 0;
773         struct inode *inode = fdentry(file)->d_inode;
774         struct btrfs_root *root = BTRFS_I(inode)->root;
775         struct page **pages = NULL;
776         int nrptrs;
777         struct page *pinned[2];
778         unsigned long first_index;
779         unsigned long last_index;
780
781         nrptrs = min((count + PAGE_CACHE_SIZE - 1) / PAGE_CACHE_SIZE,
782                      PAGE_CACHE_SIZE / (sizeof(struct page *)));
783         pinned[0] = NULL;
784         pinned[1] = NULL;
785         if (file->f_flags & O_DIRECT)
786                 return -EINVAL;
787
788         pos = *ppos;
789         start_pos = pos;
790
791         vfs_check_frozen(inode->i_sb, SB_FREEZE_WRITE);
792         current->backing_dev_info = inode->i_mapping->backing_dev_info;
793         err = generic_write_checks(file, &pos, &count, S_ISBLK(inode->i_mode));
794         if (err)
795                 goto out_nolock;
796         if (count == 0)
797                 goto out_nolock;
798         err = remove_suid(fdentry(file));
799         if (err)
800                 goto out_nolock;
801         file_update_time(file);
802
803         pages = kmalloc(nrptrs * sizeof(struct page *), GFP_KERNEL);
804
805         mutex_lock(&inode->i_mutex);
806         first_index = pos >> PAGE_CACHE_SHIFT;
807         last_index = (pos + count) >> PAGE_CACHE_SHIFT;
808
809         /*
810          * there are lots of better ways to do this, but this code
811          * makes sure the first and last page in the file range are
812          * up to date and ready for cow
813          */
814         if ((pos & (PAGE_CACHE_SIZE - 1))) {
815                 pinned[0] = grab_cache_page(inode->i_mapping, first_index);
816                 if (!PageUptodate(pinned[0])) {
817                         ret = btrfs_readpage(NULL, pinned[0]);
818                         BUG_ON(ret);
819                         wait_on_page_locked(pinned[0]);
820                 } else {
821                         unlock_page(pinned[0]);
822                 }
823         }
824         if ((pos + count) & (PAGE_CACHE_SIZE - 1)) {
825                 pinned[1] = grab_cache_page(inode->i_mapping, last_index);
826                 if (!PageUptodate(pinned[1])) {
827                         ret = btrfs_readpage(NULL, pinned[1]);
828                         BUG_ON(ret);
829                         wait_on_page_locked(pinned[1]);
830                 } else {
831                         unlock_page(pinned[1]);
832                 }
833         }
834
835         while(count > 0) {
836                 size_t offset = pos & (PAGE_CACHE_SIZE - 1);
837                 size_t write_bytes = min(count, nrptrs *
838                                         (size_t)PAGE_CACHE_SIZE -
839                                          offset);
840                 size_t num_pages = (write_bytes + PAGE_CACHE_SIZE - 1) >>
841                                         PAGE_CACHE_SHIFT;
842
843                 WARN_ON(num_pages > nrptrs);
844                 memset(pages, 0, sizeof(pages));
845
846                 mutex_lock(&root->fs_info->fs_mutex);
847                 ret = btrfs_check_free_space(root, write_bytes, 0);
848                 mutex_unlock(&root->fs_info->fs_mutex);
849                 if (ret)
850                         goto out;
851
852                 ret = prepare_pages(root, file, pages, num_pages,
853                                     pos, first_index, last_index,
854                                     write_bytes);
855                 if (ret)
856                         goto out;
857
858                 ret = btrfs_copy_from_user(pos, num_pages,
859                                            write_bytes, pages, buf);
860                 if (ret) {
861                         btrfs_drop_pages(pages, num_pages);
862                         goto out;
863                 }
864
865                 ret = dirty_and_release_pages(NULL, root, file, pages,
866                                               num_pages, pos, write_bytes);
867                 btrfs_drop_pages(pages, num_pages);
868                 if (ret)
869                         goto out;
870
871                 buf += write_bytes;
872                 count -= write_bytes;
873                 pos += write_bytes;
874                 num_written += write_bytes;
875
876                 balance_dirty_pages_ratelimited_nr(inode->i_mapping, num_pages);
877                 if (num_pages < (root->leafsize >> PAGE_CACHE_SHIFT) + 1)
878                         btrfs_btree_balance_dirty(root, 1);
879                 btrfs_throttle(root);
880                 cond_resched();
881         }
882 out:
883         mutex_unlock(&inode->i_mutex);
884
885 out_nolock:
886         kfree(pages);
887         if (pinned[0])
888                 page_cache_release(pinned[0]);
889         if (pinned[1])
890                 page_cache_release(pinned[1]);
891         *ppos = pos;
892
893         if (num_written > 0 && ((file->f_flags & O_SYNC) || IS_SYNC(inode))) {
894                 err = sync_page_range(inode, inode->i_mapping,
895                                       start_pos, num_written);
896                 if (err < 0)
897                         num_written = err;
898         }
899         current->backing_dev_info = NULL;
900         return num_written ? num_written : err;
901 }
902
903 static int btrfs_sync_file(struct file *file,
904                            struct dentry *dentry, int datasync)
905 {
906         struct inode *inode = dentry->d_inode;
907         struct btrfs_root *root = BTRFS_I(inode)->root;
908         int ret = 0;
909         struct btrfs_trans_handle *trans;
910
911         /*
912          * check the transaction that last modified this inode
913          * and see if its already been committed
914          */
915         mutex_lock(&root->fs_info->fs_mutex);
916         if (!BTRFS_I(inode)->last_trans)
917                 goto out;
918         mutex_lock(&root->fs_info->trans_mutex);
919         if (BTRFS_I(inode)->last_trans <=
920             root->fs_info->last_trans_committed) {
921                 BTRFS_I(inode)->last_trans = 0;
922                 mutex_unlock(&root->fs_info->trans_mutex);
923                 goto out;
924         }
925         mutex_unlock(&root->fs_info->trans_mutex);
926
927         /*
928          * ok we haven't committed the transaction yet, lets do a commit
929          */
930         trans = btrfs_start_transaction(root, 1);
931         if (!trans) {
932                 ret = -ENOMEM;
933                 goto out;
934         }
935         ret = btrfs_commit_transaction(trans, root);
936 out:
937         mutex_unlock(&root->fs_info->fs_mutex);
938         return ret > 0 ? EIO : ret;
939 }
940
941 static struct vm_operations_struct btrfs_file_vm_ops = {
942 #if LINUX_VERSION_CODE < KERNEL_VERSION(2,6,23)
943         .nopage         = filemap_nopage,
944         .populate       = filemap_populate,
945 #else
946         .fault          = filemap_fault,
947 #endif
948         .page_mkwrite   = btrfs_page_mkwrite,
949 };
950
951 static int btrfs_file_mmap(struct file  *filp, struct vm_area_struct *vma)
952 {
953         vma->vm_ops = &btrfs_file_vm_ops;
954         file_accessed(filp);
955         return 0;
956 }
957
958 struct file_operations btrfs_file_operations = {
959         .llseek         = generic_file_llseek,
960         .read           = do_sync_read,
961         .aio_read       = generic_file_aio_read,
962         .splice_read    = generic_file_splice_read,
963 #if LINUX_VERSION_CODE <= KERNEL_VERSION(2,6,18)
964         .sendfile       = generic_file_sendfile,
965 #endif
966         .write          = btrfs_file_write,
967         .mmap           = btrfs_file_mmap,
968         .open           = generic_file_open,
969         .fsync          = btrfs_sync_file,
970         .unlocked_ioctl = btrfs_ioctl,
971 #ifdef CONFIG_COMPAT
972         .compat_ioctl   = btrfs_ioctl,
973 #endif
974 };
975