Merge branch 'master' of /home/davem/src/GIT/linux-2.6/
[linux-2.6] / fs / affs / file.c
1 /*
2  *  linux/fs/affs/file.c
3  *
4  *  (c) 1996  Hans-Joachim Widmaier - Rewritten
5  *
6  *  (C) 1993  Ray Burr - Modified for Amiga FFS filesystem.
7  *
8  *  (C) 1992  Eric Youngdale Modified for ISO 9660 filesystem.
9  *
10  *  (C) 1991  Linus Torvalds - minix filesystem
11  *
12  *  affs regular file handling primitives
13  */
14
15 #include "affs.h"
16
17 #if PAGE_SIZE < 4096
18 #error PAGE_SIZE must be at least 4096
19 #endif
20
21 static int affs_grow_extcache(struct inode *inode, u32 lc_idx);
22 static struct buffer_head *affs_alloc_extblock(struct inode *inode, struct buffer_head *bh, u32 ext);
23 static inline struct buffer_head *affs_get_extblock(struct inode *inode, u32 ext);
24 static struct buffer_head *affs_get_extblock_slow(struct inode *inode, u32 ext);
25 static int affs_file_open(struct inode *inode, struct file *filp);
26 static int affs_file_release(struct inode *inode, struct file *filp);
27
28 const struct file_operations affs_file_operations = {
29         .llseek         = generic_file_llseek,
30         .read           = do_sync_read,
31         .aio_read       = generic_file_aio_read,
32         .write          = do_sync_write,
33         .aio_write      = generic_file_aio_write,
34         .mmap           = generic_file_mmap,
35         .open           = affs_file_open,
36         .release        = affs_file_release,
37         .fsync          = file_fsync,
38         .splice_read    = generic_file_splice_read,
39 };
40
41 const struct inode_operations affs_file_inode_operations = {
42         .truncate       = affs_truncate,
43         .setattr        = affs_notify_change,
44 };
45
46 static int
47 affs_file_open(struct inode *inode, struct file *filp)
48 {
49         pr_debug("AFFS: open(%lu,%d)\n",
50                  inode->i_ino, atomic_read(&AFFS_I(inode)->i_opencnt));
51         atomic_inc(&AFFS_I(inode)->i_opencnt);
52         return 0;
53 }
54
55 static int
56 affs_file_release(struct inode *inode, struct file *filp)
57 {
58         pr_debug("AFFS: release(%lu, %d)\n",
59                  inode->i_ino, atomic_read(&AFFS_I(inode)->i_opencnt));
60
61         if (atomic_dec_and_test(&AFFS_I(inode)->i_opencnt)) {
62                 mutex_lock(&inode->i_mutex);
63                 if (inode->i_size != AFFS_I(inode)->mmu_private)
64                         affs_truncate(inode);
65                 affs_free_prealloc(inode);
66                 mutex_unlock(&inode->i_mutex);
67         }
68
69         return 0;
70 }
71
72 static int
73 affs_grow_extcache(struct inode *inode, u32 lc_idx)
74 {
75         struct super_block      *sb = inode->i_sb;
76         struct buffer_head      *bh;
77         u32 lc_max;
78         int i, j, key;
79
80         if (!AFFS_I(inode)->i_lc) {
81                 char *ptr = (char *)get_zeroed_page(GFP_NOFS);
82                 if (!ptr)
83                         return -ENOMEM;
84                 AFFS_I(inode)->i_lc = (u32 *)ptr;
85                 AFFS_I(inode)->i_ac = (struct affs_ext_key *)(ptr + AFFS_CACHE_SIZE / 2);
86         }
87
88         lc_max = AFFS_LC_SIZE << AFFS_I(inode)->i_lc_shift;
89
90         if (AFFS_I(inode)->i_extcnt > lc_max) {
91                 u32 lc_shift, lc_mask, tmp, off;
92
93                 /* need to recalculate linear cache, start from old size */
94                 lc_shift = AFFS_I(inode)->i_lc_shift;
95                 tmp = (AFFS_I(inode)->i_extcnt / AFFS_LC_SIZE) >> lc_shift;
96                 for (; tmp; tmp >>= 1)
97                         lc_shift++;
98                 lc_mask = (1 << lc_shift) - 1;
99
100                 /* fix idx and old size to new shift */
101                 lc_idx >>= (lc_shift - AFFS_I(inode)->i_lc_shift);
102                 AFFS_I(inode)->i_lc_size >>= (lc_shift - AFFS_I(inode)->i_lc_shift);
103
104                 /* first shrink old cache to make more space */
105                 off = 1 << (lc_shift - AFFS_I(inode)->i_lc_shift);
106                 for (i = 1, j = off; j < AFFS_LC_SIZE; i++, j += off)
107                         AFFS_I(inode)->i_ac[i] = AFFS_I(inode)->i_ac[j];
108
109                 AFFS_I(inode)->i_lc_shift = lc_shift;
110                 AFFS_I(inode)->i_lc_mask = lc_mask;
111         }
112
113         /* fill cache to the needed index */
114         i = AFFS_I(inode)->i_lc_size;
115         AFFS_I(inode)->i_lc_size = lc_idx + 1;
116         for (; i <= lc_idx; i++) {
117                 if (!i) {
118                         AFFS_I(inode)->i_lc[0] = inode->i_ino;
119                         continue;
120                 }
121                 key = AFFS_I(inode)->i_lc[i - 1];
122                 j = AFFS_I(inode)->i_lc_mask + 1;
123                 // unlock cache
124                 for (; j > 0; j--) {
125                         bh = affs_bread(sb, key);
126                         if (!bh)
127                                 goto err;
128                         key = be32_to_cpu(AFFS_TAIL(sb, bh)->extension);
129                         affs_brelse(bh);
130                 }
131                 // lock cache
132                 AFFS_I(inode)->i_lc[i] = key;
133         }
134
135         return 0;
136
137 err:
138         // lock cache
139         return -EIO;
140 }
141
142 static struct buffer_head *
143 affs_alloc_extblock(struct inode *inode, struct buffer_head *bh, u32 ext)
144 {
145         struct super_block *sb = inode->i_sb;
146         struct buffer_head *new_bh;
147         u32 blocknr, tmp;
148
149         blocknr = affs_alloc_block(inode, bh->b_blocknr);
150         if (!blocknr)
151                 return ERR_PTR(-ENOSPC);
152
153         new_bh = affs_getzeroblk(sb, blocknr);
154         if (!new_bh) {
155                 affs_free_block(sb, blocknr);
156                 return ERR_PTR(-EIO);
157         }
158
159         AFFS_HEAD(new_bh)->ptype = cpu_to_be32(T_LIST);
160         AFFS_HEAD(new_bh)->key = cpu_to_be32(blocknr);
161         AFFS_TAIL(sb, new_bh)->stype = cpu_to_be32(ST_FILE);
162         AFFS_TAIL(sb, new_bh)->parent = cpu_to_be32(inode->i_ino);
163         affs_fix_checksum(sb, new_bh);
164
165         mark_buffer_dirty_inode(new_bh, inode);
166
167         tmp = be32_to_cpu(AFFS_TAIL(sb, bh)->extension);
168         if (tmp)
169                 affs_warning(sb, "alloc_ext", "previous extension set (%x)", tmp);
170         AFFS_TAIL(sb, bh)->extension = cpu_to_be32(blocknr);
171         affs_adjust_checksum(bh, blocknr - tmp);
172         mark_buffer_dirty_inode(bh, inode);
173
174         AFFS_I(inode)->i_extcnt++;
175         mark_inode_dirty(inode);
176
177         return new_bh;
178 }
179
180 static inline struct buffer_head *
181 affs_get_extblock(struct inode *inode, u32 ext)
182 {
183         /* inline the simplest case: same extended block as last time */
184         struct buffer_head *bh = AFFS_I(inode)->i_ext_bh;
185         if (ext == AFFS_I(inode)->i_ext_last)
186                 get_bh(bh);
187         else
188                 /* we have to do more (not inlined) */
189                 bh = affs_get_extblock_slow(inode, ext);
190
191         return bh;
192 }
193
194 static struct buffer_head *
195 affs_get_extblock_slow(struct inode *inode, u32 ext)
196 {
197         struct super_block *sb = inode->i_sb;
198         struct buffer_head *bh;
199         u32 ext_key;
200         u32 lc_idx, lc_off, ac_idx;
201         u32 tmp, idx;
202
203         if (ext == AFFS_I(inode)->i_ext_last + 1) {
204                 /* read the next extended block from the current one */
205                 bh = AFFS_I(inode)->i_ext_bh;
206                 ext_key = be32_to_cpu(AFFS_TAIL(sb, bh)->extension);
207                 if (ext < AFFS_I(inode)->i_extcnt)
208                         goto read_ext;
209                 if (ext > AFFS_I(inode)->i_extcnt)
210                         BUG();
211                 bh = affs_alloc_extblock(inode, bh, ext);
212                 if (IS_ERR(bh))
213                         return bh;
214                 goto store_ext;
215         }
216
217         if (ext == 0) {
218                 /* we seek back to the file header block */
219                 ext_key = inode->i_ino;
220                 goto read_ext;
221         }
222
223         if (ext >= AFFS_I(inode)->i_extcnt) {
224                 struct buffer_head *prev_bh;
225
226                 /* allocate a new extended block */
227                 if (ext > AFFS_I(inode)->i_extcnt)
228                         BUG();
229
230                 /* get previous extended block */
231                 prev_bh = affs_get_extblock(inode, ext - 1);
232                 if (IS_ERR(prev_bh))
233                         return prev_bh;
234                 bh = affs_alloc_extblock(inode, prev_bh, ext);
235                 affs_brelse(prev_bh);
236                 if (IS_ERR(bh))
237                         return bh;
238                 goto store_ext;
239         }
240
241 again:
242         /* check if there is an extended cache and whether it's large enough */
243         lc_idx = ext >> AFFS_I(inode)->i_lc_shift;
244         lc_off = ext & AFFS_I(inode)->i_lc_mask;
245
246         if (lc_idx >= AFFS_I(inode)->i_lc_size) {
247                 int err;
248
249                 err = affs_grow_extcache(inode, lc_idx);
250                 if (err)
251                         return ERR_PTR(err);
252                 goto again;
253         }
254
255         /* every n'th key we find in the linear cache */
256         if (!lc_off) {
257                 ext_key = AFFS_I(inode)->i_lc[lc_idx];
258                 goto read_ext;
259         }
260
261         /* maybe it's still in the associative cache */
262         ac_idx = (ext - lc_idx - 1) & AFFS_AC_MASK;
263         if (AFFS_I(inode)->i_ac[ac_idx].ext == ext) {
264                 ext_key = AFFS_I(inode)->i_ac[ac_idx].key;
265                 goto read_ext;
266         }
267
268         /* try to find one of the previous extended blocks */
269         tmp = ext;
270         idx = ac_idx;
271         while (--tmp, --lc_off > 0) {
272                 idx = (idx - 1) & AFFS_AC_MASK;
273                 if (AFFS_I(inode)->i_ac[idx].ext == tmp) {
274                         ext_key = AFFS_I(inode)->i_ac[idx].key;
275                         goto find_ext;
276                 }
277         }
278
279         /* fall back to the linear cache */
280         ext_key = AFFS_I(inode)->i_lc[lc_idx];
281 find_ext:
282         /* read all extended blocks until we find the one we need */
283         //unlock cache
284         do {
285                 bh = affs_bread(sb, ext_key);
286                 if (!bh)
287                         goto err_bread;
288                 ext_key = be32_to_cpu(AFFS_TAIL(sb, bh)->extension);
289                 affs_brelse(bh);
290                 tmp++;
291         } while (tmp < ext);
292         //lock cache
293
294         /* store it in the associative cache */
295         // recalculate ac_idx?
296         AFFS_I(inode)->i_ac[ac_idx].ext = ext;
297         AFFS_I(inode)->i_ac[ac_idx].key = ext_key;
298
299 read_ext:
300         /* finally read the right extended block */
301         //unlock cache
302         bh = affs_bread(sb, ext_key);
303         if (!bh)
304                 goto err_bread;
305         //lock cache
306
307 store_ext:
308         /* release old cached extended block and store the new one */
309         affs_brelse(AFFS_I(inode)->i_ext_bh);
310         AFFS_I(inode)->i_ext_last = ext;
311         AFFS_I(inode)->i_ext_bh = bh;
312         get_bh(bh);
313
314         return bh;
315
316 err_bread:
317         affs_brelse(bh);
318         return ERR_PTR(-EIO);
319 }
320
321 static int
322 affs_get_block(struct inode *inode, sector_t block, struct buffer_head *bh_result, int create)
323 {
324         struct super_block      *sb = inode->i_sb;
325         struct buffer_head      *ext_bh;
326         u32                      ext;
327
328         pr_debug("AFFS: get_block(%u, %lu)\n", (u32)inode->i_ino, (unsigned long)block);
329
330         BUG_ON(block > (sector_t)0x7fffffffUL);
331
332         if (block >= AFFS_I(inode)->i_blkcnt) {
333                 if (block > AFFS_I(inode)->i_blkcnt || !create)
334                         goto err_big;
335         } else
336                 create = 0;
337
338         //lock cache
339         affs_lock_ext(inode);
340
341         ext = (u32)block / AFFS_SB(sb)->s_hashsize;
342         block -= ext * AFFS_SB(sb)->s_hashsize;
343         ext_bh = affs_get_extblock(inode, ext);
344         if (IS_ERR(ext_bh))
345                 goto err_ext;
346         map_bh(bh_result, sb, (sector_t)be32_to_cpu(AFFS_BLOCK(sb, ext_bh, block)));
347
348         if (create) {
349                 u32 blocknr = affs_alloc_block(inode, ext_bh->b_blocknr);
350                 if (!blocknr)
351                         goto err_alloc;
352                 set_buffer_new(bh_result);
353                 AFFS_I(inode)->mmu_private += AFFS_SB(sb)->s_data_blksize;
354                 AFFS_I(inode)->i_blkcnt++;
355
356                 /* store new block */
357                 if (bh_result->b_blocknr)
358                         affs_warning(sb, "get_block", "block already set (%x)", bh_result->b_blocknr);
359                 AFFS_BLOCK(sb, ext_bh, block) = cpu_to_be32(blocknr);
360                 AFFS_HEAD(ext_bh)->block_count = cpu_to_be32(block + 1);
361                 affs_adjust_checksum(ext_bh, blocknr - bh_result->b_blocknr + 1);
362                 bh_result->b_blocknr = blocknr;
363
364                 if (!block) {
365                         /* insert first block into header block */
366                         u32 tmp = be32_to_cpu(AFFS_HEAD(ext_bh)->first_data);
367                         if (tmp)
368                                 affs_warning(sb, "get_block", "first block already set (%d)", tmp);
369                         AFFS_HEAD(ext_bh)->first_data = cpu_to_be32(blocknr);
370                         affs_adjust_checksum(ext_bh, blocknr - tmp);
371                 }
372         }
373
374         affs_brelse(ext_bh);
375         //unlock cache
376         affs_unlock_ext(inode);
377         return 0;
378
379 err_big:
380         affs_error(inode->i_sb,"get_block","strange block request %d", block);
381         return -EIO;
382 err_ext:
383         // unlock cache
384         affs_unlock_ext(inode);
385         return PTR_ERR(ext_bh);
386 err_alloc:
387         brelse(ext_bh);
388         clear_buffer_mapped(bh_result);
389         bh_result->b_bdev = NULL;
390         // unlock cache
391         affs_unlock_ext(inode);
392         return -ENOSPC;
393 }
394
395 static int affs_writepage(struct page *page, struct writeback_control *wbc)
396 {
397         return block_write_full_page(page, affs_get_block, wbc);
398 }
399
400 static int affs_readpage(struct file *file, struct page *page)
401 {
402         return block_read_full_page(page, affs_get_block);
403 }
404
405 static int affs_write_begin(struct file *file, struct address_space *mapping,
406                         loff_t pos, unsigned len, unsigned flags,
407                         struct page **pagep, void **fsdata)
408 {
409         *pagep = NULL;
410         return cont_write_begin(file, mapping, pos, len, flags, pagep, fsdata,
411                                 affs_get_block,
412                                 &AFFS_I(mapping->host)->mmu_private);
413 }
414
415 static sector_t _affs_bmap(struct address_space *mapping, sector_t block)
416 {
417         return generic_block_bmap(mapping,block,affs_get_block);
418 }
419
420 const struct address_space_operations affs_aops = {
421         .readpage = affs_readpage,
422         .writepage = affs_writepage,
423         .sync_page = block_sync_page,
424         .write_begin = affs_write_begin,
425         .write_end = generic_write_end,
426         .bmap = _affs_bmap
427 };
428
429 static inline struct buffer_head *
430 affs_bread_ino(struct inode *inode, int block, int create)
431 {
432         struct buffer_head *bh, tmp_bh;
433         int err;
434
435         tmp_bh.b_state = 0;
436         err = affs_get_block(inode, block, &tmp_bh, create);
437         if (!err) {
438                 bh = affs_bread(inode->i_sb, tmp_bh.b_blocknr);
439                 if (bh) {
440                         bh->b_state |= tmp_bh.b_state;
441                         return bh;
442                 }
443                 err = -EIO;
444         }
445         return ERR_PTR(err);
446 }
447
448 static inline struct buffer_head *
449 affs_getzeroblk_ino(struct inode *inode, int block)
450 {
451         struct buffer_head *bh, tmp_bh;
452         int err;
453
454         tmp_bh.b_state = 0;
455         err = affs_get_block(inode, block, &tmp_bh, 1);
456         if (!err) {
457                 bh = affs_getzeroblk(inode->i_sb, tmp_bh.b_blocknr);
458                 if (bh) {
459                         bh->b_state |= tmp_bh.b_state;
460                         return bh;
461                 }
462                 err = -EIO;
463         }
464         return ERR_PTR(err);
465 }
466
467 static inline struct buffer_head *
468 affs_getemptyblk_ino(struct inode *inode, int block)
469 {
470         struct buffer_head *bh, tmp_bh;
471         int err;
472
473         tmp_bh.b_state = 0;
474         err = affs_get_block(inode, block, &tmp_bh, 1);
475         if (!err) {
476                 bh = affs_getemptyblk(inode->i_sb, tmp_bh.b_blocknr);
477                 if (bh) {
478                         bh->b_state |= tmp_bh.b_state;
479                         return bh;
480                 }
481                 err = -EIO;
482         }
483         return ERR_PTR(err);
484 }
485
486 static int
487 affs_do_readpage_ofs(struct file *file, struct page *page, unsigned from, unsigned to)
488 {
489         struct inode *inode = page->mapping->host;
490         struct super_block *sb = inode->i_sb;
491         struct buffer_head *bh;
492         char *data;
493         u32 bidx, boff, bsize;
494         u32 tmp;
495
496         pr_debug("AFFS: read_page(%u, %ld, %d, %d)\n", (u32)inode->i_ino, page->index, from, to);
497         BUG_ON(from > to || to > PAGE_CACHE_SIZE);
498         kmap(page);
499         data = page_address(page);
500         bsize = AFFS_SB(sb)->s_data_blksize;
501         tmp = (page->index << PAGE_CACHE_SHIFT) + from;
502         bidx = tmp / bsize;
503         boff = tmp % bsize;
504
505         while (from < to) {
506                 bh = affs_bread_ino(inode, bidx, 0);
507                 if (IS_ERR(bh))
508                         return PTR_ERR(bh);
509                 tmp = min(bsize - boff, to - from);
510                 BUG_ON(from + tmp > to || tmp > bsize);
511                 memcpy(data + from, AFFS_DATA(bh) + boff, tmp);
512                 affs_brelse(bh);
513                 bidx++;
514                 from += tmp;
515                 boff = 0;
516         }
517         flush_dcache_page(page);
518         kunmap(page);
519         return 0;
520 }
521
522 static int
523 affs_extent_file_ofs(struct inode *inode, u32 newsize)
524 {
525         struct super_block *sb = inode->i_sb;
526         struct buffer_head *bh, *prev_bh;
527         u32 bidx, boff;
528         u32 size, bsize;
529         u32 tmp;
530
531         pr_debug("AFFS: extent_file(%u, %d)\n", (u32)inode->i_ino, newsize);
532         bsize = AFFS_SB(sb)->s_data_blksize;
533         bh = NULL;
534         size = AFFS_I(inode)->mmu_private;
535         bidx = size / bsize;
536         boff = size % bsize;
537         if (boff) {
538                 bh = affs_bread_ino(inode, bidx, 0);
539                 if (IS_ERR(bh))
540                         return PTR_ERR(bh);
541                 tmp = min(bsize - boff, newsize - size);
542                 BUG_ON(boff + tmp > bsize || tmp > bsize);
543                 memset(AFFS_DATA(bh) + boff, 0, tmp);
544                 be32_add_cpu(&AFFS_DATA_HEAD(bh)->size, tmp);
545                 affs_fix_checksum(sb, bh);
546                 mark_buffer_dirty_inode(bh, inode);
547                 size += tmp;
548                 bidx++;
549         } else if (bidx) {
550                 bh = affs_bread_ino(inode, bidx - 1, 0);
551                 if (IS_ERR(bh))
552                         return PTR_ERR(bh);
553         }
554
555         while (size < newsize) {
556                 prev_bh = bh;
557                 bh = affs_getzeroblk_ino(inode, bidx);
558                 if (IS_ERR(bh))
559                         goto out;
560                 tmp = min(bsize, newsize - size);
561                 BUG_ON(tmp > bsize);
562                 AFFS_DATA_HEAD(bh)->ptype = cpu_to_be32(T_DATA);
563                 AFFS_DATA_HEAD(bh)->key = cpu_to_be32(inode->i_ino);
564                 AFFS_DATA_HEAD(bh)->sequence = cpu_to_be32(bidx);
565                 AFFS_DATA_HEAD(bh)->size = cpu_to_be32(tmp);
566                 affs_fix_checksum(sb, bh);
567                 bh->b_state &= ~(1UL << BH_New);
568                 mark_buffer_dirty_inode(bh, inode);
569                 if (prev_bh) {
570                         u32 tmp = be32_to_cpu(AFFS_DATA_HEAD(prev_bh)->next);
571                         if (tmp)
572                                 affs_warning(sb, "extent_file_ofs", "next block already set for %d (%d)", bidx, tmp);
573                         AFFS_DATA_HEAD(prev_bh)->next = cpu_to_be32(bh->b_blocknr);
574                         affs_adjust_checksum(prev_bh, bh->b_blocknr - tmp);
575                         mark_buffer_dirty_inode(prev_bh, inode);
576                         affs_brelse(prev_bh);
577                 }
578                 size += bsize;
579                 bidx++;
580         }
581         affs_brelse(bh);
582         inode->i_size = AFFS_I(inode)->mmu_private = newsize;
583         return 0;
584
585 out:
586         inode->i_size = AFFS_I(inode)->mmu_private = newsize;
587         return PTR_ERR(bh);
588 }
589
590 static int
591 affs_readpage_ofs(struct file *file, struct page *page)
592 {
593         struct inode *inode = page->mapping->host;
594         u32 to;
595         int err;
596
597         pr_debug("AFFS: read_page(%u, %ld)\n", (u32)inode->i_ino, page->index);
598         to = PAGE_CACHE_SIZE;
599         if (((page->index + 1) << PAGE_CACHE_SHIFT) > inode->i_size) {
600                 to = inode->i_size & ~PAGE_CACHE_MASK;
601                 memset(page_address(page) + to, 0, PAGE_CACHE_SIZE - to);
602         }
603
604         err = affs_do_readpage_ofs(file, page, 0, to);
605         if (!err)
606                 SetPageUptodate(page);
607         unlock_page(page);
608         return err;
609 }
610
611 static int affs_write_begin_ofs(struct file *file, struct address_space *mapping,
612                                 loff_t pos, unsigned len, unsigned flags,
613                                 struct page **pagep, void **fsdata)
614 {
615         struct inode *inode = mapping->host;
616         struct page *page;
617         pgoff_t index;
618         int err = 0;
619
620         pr_debug("AFFS: write_begin(%u, %llu, %llu)\n", (u32)inode->i_ino, (unsigned long long)pos, (unsigned long long)pos + len);
621         if (pos > AFFS_I(inode)->mmu_private) {
622                 /* XXX: this probably leaves a too-big i_size in case of
623                  * failure. Should really be updating i_size at write_end time
624                  */
625                 err = affs_extent_file_ofs(inode, pos);
626                 if (err)
627                         return err;
628         }
629
630         index = pos >> PAGE_CACHE_SHIFT;
631         page = grab_cache_page_write_begin(mapping, index, flags);
632         if (!page)
633                 return -ENOMEM;
634         *pagep = page;
635
636         if (PageUptodate(page))
637                 return 0;
638
639         /* XXX: inefficient but safe in the face of short writes */
640         err = affs_do_readpage_ofs(file, page, 0, PAGE_CACHE_SIZE);
641         if (err) {
642                 unlock_page(page);
643                 page_cache_release(page);
644         }
645         return err;
646 }
647
648 static int affs_write_end_ofs(struct file *file, struct address_space *mapping,
649                                 loff_t pos, unsigned len, unsigned copied,
650                                 struct page *page, void *fsdata)
651 {
652         struct inode *inode = mapping->host;
653         struct super_block *sb = inode->i_sb;
654         struct buffer_head *bh, *prev_bh;
655         char *data;
656         u32 bidx, boff, bsize;
657         unsigned from, to;
658         u32 tmp;
659         int written;
660
661         from = pos & (PAGE_CACHE_SIZE - 1);
662         to = pos + len;
663         /*
664          * XXX: not sure if this can handle short copies (len < copied), but
665          * we don't have to, because the page should always be uptodate here,
666          * due to write_begin.
667          */
668
669         pr_debug("AFFS: write_begin(%u, %llu, %llu)\n", (u32)inode->i_ino, (unsigned long long)pos, (unsigned long long)pos + len);
670         bsize = AFFS_SB(sb)->s_data_blksize;
671         data = page_address(page);
672
673         bh = NULL;
674         written = 0;
675         tmp = (page->index << PAGE_CACHE_SHIFT) + from;
676         bidx = tmp / bsize;
677         boff = tmp % bsize;
678         if (boff) {
679                 bh = affs_bread_ino(inode, bidx, 0);
680                 if (IS_ERR(bh))
681                         return PTR_ERR(bh);
682                 tmp = min(bsize - boff, to - from);
683                 BUG_ON(boff + tmp > bsize || tmp > bsize);
684                 memcpy(AFFS_DATA(bh) + boff, data + from, tmp);
685                 be32_add_cpu(&AFFS_DATA_HEAD(bh)->size, tmp);
686                 affs_fix_checksum(sb, bh);
687                 mark_buffer_dirty_inode(bh, inode);
688                 written += tmp;
689                 from += tmp;
690                 bidx++;
691         } else if (bidx) {
692                 bh = affs_bread_ino(inode, bidx - 1, 0);
693                 if (IS_ERR(bh))
694                         return PTR_ERR(bh);
695         }
696         while (from + bsize <= to) {
697                 prev_bh = bh;
698                 bh = affs_getemptyblk_ino(inode, bidx);
699                 if (IS_ERR(bh))
700                         goto out;
701                 memcpy(AFFS_DATA(bh), data + from, bsize);
702                 if (buffer_new(bh)) {
703                         AFFS_DATA_HEAD(bh)->ptype = cpu_to_be32(T_DATA);
704                         AFFS_DATA_HEAD(bh)->key = cpu_to_be32(inode->i_ino);
705                         AFFS_DATA_HEAD(bh)->sequence = cpu_to_be32(bidx);
706                         AFFS_DATA_HEAD(bh)->size = cpu_to_be32(bsize);
707                         AFFS_DATA_HEAD(bh)->next = 0;
708                         bh->b_state &= ~(1UL << BH_New);
709                         if (prev_bh) {
710                                 u32 tmp = be32_to_cpu(AFFS_DATA_HEAD(prev_bh)->next);
711                                 if (tmp)
712                                         affs_warning(sb, "commit_write_ofs", "next block already set for %d (%d)", bidx, tmp);
713                                 AFFS_DATA_HEAD(prev_bh)->next = cpu_to_be32(bh->b_blocknr);
714                                 affs_adjust_checksum(prev_bh, bh->b_blocknr - tmp);
715                                 mark_buffer_dirty_inode(prev_bh, inode);
716                         }
717                 }
718                 affs_brelse(prev_bh);
719                 affs_fix_checksum(sb, bh);
720                 mark_buffer_dirty_inode(bh, inode);
721                 written += bsize;
722                 from += bsize;
723                 bidx++;
724         }
725         if (from < to) {
726                 prev_bh = bh;
727                 bh = affs_bread_ino(inode, bidx, 1);
728                 if (IS_ERR(bh))
729                         goto out;
730                 tmp = min(bsize, to - from);
731                 BUG_ON(tmp > bsize);
732                 memcpy(AFFS_DATA(bh), data + from, tmp);
733                 if (buffer_new(bh)) {
734                         AFFS_DATA_HEAD(bh)->ptype = cpu_to_be32(T_DATA);
735                         AFFS_DATA_HEAD(bh)->key = cpu_to_be32(inode->i_ino);
736                         AFFS_DATA_HEAD(bh)->sequence = cpu_to_be32(bidx);
737                         AFFS_DATA_HEAD(bh)->size = cpu_to_be32(tmp);
738                         AFFS_DATA_HEAD(bh)->next = 0;
739                         bh->b_state &= ~(1UL << BH_New);
740                         if (prev_bh) {
741                                 u32 tmp = be32_to_cpu(AFFS_DATA_HEAD(prev_bh)->next);
742                                 if (tmp)
743                                         affs_warning(sb, "commit_write_ofs", "next block already set for %d (%d)", bidx, tmp);
744                                 AFFS_DATA_HEAD(prev_bh)->next = cpu_to_be32(bh->b_blocknr);
745                                 affs_adjust_checksum(prev_bh, bh->b_blocknr - tmp);
746                                 mark_buffer_dirty_inode(prev_bh, inode);
747                         }
748                 } else if (be32_to_cpu(AFFS_DATA_HEAD(bh)->size) < tmp)
749                         AFFS_DATA_HEAD(bh)->size = cpu_to_be32(tmp);
750                 affs_brelse(prev_bh);
751                 affs_fix_checksum(sb, bh);
752                 mark_buffer_dirty_inode(bh, inode);
753                 written += tmp;
754                 from += tmp;
755                 bidx++;
756         }
757         SetPageUptodate(page);
758
759 done:
760         affs_brelse(bh);
761         tmp = (page->index << PAGE_CACHE_SHIFT) + from;
762         if (tmp > inode->i_size)
763                 inode->i_size = AFFS_I(inode)->mmu_private = tmp;
764
765         unlock_page(page);
766         page_cache_release(page);
767
768         return written;
769
770 out:
771         bh = prev_bh;
772         if (!written)
773                 written = PTR_ERR(bh);
774         goto done;
775 }
776
777 const struct address_space_operations affs_aops_ofs = {
778         .readpage = affs_readpage_ofs,
779         //.writepage = affs_writepage_ofs,
780         //.sync_page = affs_sync_page_ofs,
781         .write_begin = affs_write_begin_ofs,
782         .write_end = affs_write_end_ofs
783 };
784
785 /* Free any preallocated blocks. */
786
787 void
788 affs_free_prealloc(struct inode *inode)
789 {
790         struct super_block *sb = inode->i_sb;
791
792         pr_debug("AFFS: free_prealloc(ino=%lu)\n", inode->i_ino);
793
794         while (AFFS_I(inode)->i_pa_cnt) {
795                 AFFS_I(inode)->i_pa_cnt--;
796                 affs_free_block(sb, ++AFFS_I(inode)->i_lastalloc);
797         }
798 }
799
800 /* Truncate (or enlarge) a file to the requested size. */
801
802 void
803 affs_truncate(struct inode *inode)
804 {
805         struct super_block *sb = inode->i_sb;
806         u32 ext, ext_key;
807         u32 last_blk, blkcnt, blk;
808         u32 size;
809         struct buffer_head *ext_bh;
810         int i;
811
812         pr_debug("AFFS: truncate(inode=%d, oldsize=%u, newsize=%u)\n",
813                  (u32)inode->i_ino, (u32)AFFS_I(inode)->mmu_private, (u32)inode->i_size);
814
815         last_blk = 0;
816         ext = 0;
817         if (inode->i_size) {
818                 last_blk = ((u32)inode->i_size - 1) / AFFS_SB(sb)->s_data_blksize;
819                 ext = last_blk / AFFS_SB(sb)->s_hashsize;
820         }
821
822         if (inode->i_size > AFFS_I(inode)->mmu_private) {
823                 struct address_space *mapping = inode->i_mapping;
824                 struct page *page;
825                 void *fsdata;
826                 u32 size = inode->i_size;
827                 int res;
828
829                 res = mapping->a_ops->write_begin(NULL, mapping, size, 0, 0, &page, &fsdata);
830                 if (!res)
831                         res = mapping->a_ops->write_end(NULL, mapping, size, 0, 0, page, fsdata);
832                 else
833                         inode->i_size = AFFS_I(inode)->mmu_private;
834                 mark_inode_dirty(inode);
835                 return;
836         } else if (inode->i_size == AFFS_I(inode)->mmu_private)
837                 return;
838
839         // lock cache
840         ext_bh = affs_get_extblock(inode, ext);
841         if (IS_ERR(ext_bh)) {
842                 affs_warning(sb, "truncate", "unexpected read error for ext block %u (%d)",
843                              ext, PTR_ERR(ext_bh));
844                 return;
845         }
846         if (AFFS_I(inode)->i_lc) {
847                 /* clear linear cache */
848                 i = (ext + 1) >> AFFS_I(inode)->i_lc_shift;
849                 if (AFFS_I(inode)->i_lc_size > i) {
850                         AFFS_I(inode)->i_lc_size = i;
851                         for (; i < AFFS_LC_SIZE; i++)
852                                 AFFS_I(inode)->i_lc[i] = 0;
853                 }
854                 /* clear associative cache */
855                 for (i = 0; i < AFFS_AC_SIZE; i++)
856                         if (AFFS_I(inode)->i_ac[i].ext >= ext)
857                                 AFFS_I(inode)->i_ac[i].ext = 0;
858         }
859         ext_key = be32_to_cpu(AFFS_TAIL(sb, ext_bh)->extension);
860
861         blkcnt = AFFS_I(inode)->i_blkcnt;
862         i = 0;
863         blk = last_blk;
864         if (inode->i_size) {
865                 i = last_blk % AFFS_SB(sb)->s_hashsize + 1;
866                 blk++;
867         } else
868                 AFFS_HEAD(ext_bh)->first_data = 0;
869         AFFS_HEAD(ext_bh)->block_count = cpu_to_be32(i);
870         size = AFFS_SB(sb)->s_hashsize;
871         if (size > blkcnt - blk + i)
872                 size = blkcnt - blk + i;
873         for (; i < size; i++, blk++) {
874                 affs_free_block(sb, be32_to_cpu(AFFS_BLOCK(sb, ext_bh, i)));
875                 AFFS_BLOCK(sb, ext_bh, i) = 0;
876         }
877         AFFS_TAIL(sb, ext_bh)->extension = 0;
878         affs_fix_checksum(sb, ext_bh);
879         mark_buffer_dirty_inode(ext_bh, inode);
880         affs_brelse(ext_bh);
881
882         if (inode->i_size) {
883                 AFFS_I(inode)->i_blkcnt = last_blk + 1;
884                 AFFS_I(inode)->i_extcnt = ext + 1;
885                 if (AFFS_SB(sb)->s_flags & SF_OFS) {
886                         struct buffer_head *bh = affs_bread_ino(inode, last_blk, 0);
887                         u32 tmp;
888                         if (IS_ERR(ext_bh)) {
889                                 affs_warning(sb, "truncate", "unexpected read error for last block %u (%d)",
890                                              ext, PTR_ERR(ext_bh));
891                                 return;
892                         }
893                         tmp = be32_to_cpu(AFFS_DATA_HEAD(bh)->next);
894                         AFFS_DATA_HEAD(bh)->next = 0;
895                         affs_adjust_checksum(bh, -tmp);
896                         affs_brelse(bh);
897                 }
898         } else {
899                 AFFS_I(inode)->i_blkcnt = 0;
900                 AFFS_I(inode)->i_extcnt = 1;
901         }
902         AFFS_I(inode)->mmu_private = inode->i_size;
903         // unlock cache
904
905         while (ext_key) {
906                 ext_bh = affs_bread(sb, ext_key);
907                 size = AFFS_SB(sb)->s_hashsize;
908                 if (size > blkcnt - blk)
909                         size = blkcnt - blk;
910                 for (i = 0; i < size; i++, blk++)
911                         affs_free_block(sb, be32_to_cpu(AFFS_BLOCK(sb, ext_bh, i)));
912                 affs_free_block(sb, ext_key);
913                 ext_key = be32_to_cpu(AFFS_TAIL(sb, ext_bh)->extension);
914                 affs_brelse(ext_bh);
915         }
916         affs_free_prealloc(inode);
917 }