Merge tag 'fcoe' into for-linus
[linux-2.6-microblaze.git] / fs / ufs / balloc.c
1 /*
2  *  linux/fs/ufs/balloc.c
3  *
4  * Copyright (C) 1998
5  * Daniel Pirkl <daniel.pirkl@email.cz>
6  * Charles University, Faculty of Mathematics and Physics
7  *
8  * UFS2 write support Evgeniy Dushistov <dushistov@mail.ru>, 2007
9  */
10
11 #include <linux/fs.h>
12 #include <linux/stat.h>
13 #include <linux/time.h>
14 #include <linux/string.h>
15 #include <linux/buffer_head.h>
16 #include <linux/capability.h>
17 #include <linux/bitops.h>
18 #include <asm/byteorder.h>
19
20 #include "ufs_fs.h"
21 #include "ufs.h"
22 #include "swab.h"
23 #include "util.h"
24
25 #define INVBLOCK ((u64)-1L)
26
27 static u64 ufs_add_fragments(struct inode *, u64, unsigned, unsigned, int *);
28 static u64 ufs_alloc_fragments(struct inode *, unsigned, u64, unsigned, int *);
29 static u64 ufs_alloccg_block(struct inode *, struct ufs_cg_private_info *, u64, int *);
30 static u64 ufs_bitmap_search (struct super_block *, struct ufs_cg_private_info *, u64, unsigned);
31 static unsigned char ufs_fragtable_8fpb[], ufs_fragtable_other[];
32 static void ufs_clusteracct(struct super_block *, struct ufs_cg_private_info *, unsigned, int);
33
34 /*
35  * Free 'count' fragments from fragment number 'fragment'
36  */
37 void ufs_free_fragments(struct inode *inode, u64 fragment, unsigned count)
38 {
39         struct super_block * sb;
40         struct ufs_sb_private_info * uspi;
41         struct ufs_super_block_first * usb1;
42         struct ufs_cg_private_info * ucpi;
43         struct ufs_cylinder_group * ucg;
44         unsigned cgno, bit, end_bit, bbase, blkmap, i;
45         u64 blkno;
46         
47         sb = inode->i_sb;
48         uspi = UFS_SB(sb)->s_uspi;
49         usb1 = ubh_get_usb_first(uspi);
50         
51         UFSD("ENTER, fragment %llu, count %u\n",
52              (unsigned long long)fragment, count);
53         
54         if (ufs_fragnum(fragment) + count > uspi->s_fpg)
55                 ufs_error (sb, "ufs_free_fragments", "internal error");
56         
57         mutex_lock(&UFS_SB(sb)->s_lock);
58         
59         cgno = ufs_dtog(uspi, fragment);
60         bit = ufs_dtogd(uspi, fragment);
61         if (cgno >= uspi->s_ncg) {
62                 ufs_panic (sb, "ufs_free_fragments", "freeing blocks are outside device");
63                 goto failed;
64         }
65                 
66         ucpi = ufs_load_cylinder (sb, cgno);
67         if (!ucpi) 
68                 goto failed;
69         ucg = ubh_get_ucg (UCPI_UBH(ucpi));
70         if (!ufs_cg_chkmagic(sb, ucg)) {
71                 ufs_panic (sb, "ufs_free_fragments", "internal error, bad magic number on cg %u", cgno);
72                 goto failed;
73         }
74
75         end_bit = bit + count;
76         bbase = ufs_blknum (bit);
77         blkmap = ubh_blkmap (UCPI_UBH(ucpi), ucpi->c_freeoff, bbase);
78         ufs_fragacct (sb, blkmap, ucg->cg_frsum, -1);
79         for (i = bit; i < end_bit; i++) {
80                 if (ubh_isclr (UCPI_UBH(ucpi), ucpi->c_freeoff, i))
81                         ubh_setbit (UCPI_UBH(ucpi), ucpi->c_freeoff, i);
82                 else 
83                         ufs_error (sb, "ufs_free_fragments",
84                                    "bit already cleared for fragment %u", i);
85         }
86         
87         fs32_add(sb, &ucg->cg_cs.cs_nffree, count);
88         uspi->cs_total.cs_nffree += count;
89         fs32_add(sb, &UFS_SB(sb)->fs_cs(cgno).cs_nffree, count);
90         blkmap = ubh_blkmap (UCPI_UBH(ucpi), ucpi->c_freeoff, bbase);
91         ufs_fragacct(sb, blkmap, ucg->cg_frsum, 1);
92
93         /*
94          * Trying to reassemble free fragments into block
95          */
96         blkno = ufs_fragstoblks (bbase);
97         if (ubh_isblockset(UCPI_UBH(ucpi), ucpi->c_freeoff, blkno)) {
98                 fs32_sub(sb, &ucg->cg_cs.cs_nffree, uspi->s_fpb);
99                 uspi->cs_total.cs_nffree -= uspi->s_fpb;
100                 fs32_sub(sb, &UFS_SB(sb)->fs_cs(cgno).cs_nffree, uspi->s_fpb);
101                 if ((UFS_SB(sb)->s_flags & UFS_CG_MASK) == UFS_CG_44BSD)
102                         ufs_clusteracct (sb, ucpi, blkno, 1);
103                 fs32_add(sb, &ucg->cg_cs.cs_nbfree, 1);
104                 uspi->cs_total.cs_nbfree++;
105                 fs32_add(sb, &UFS_SB(sb)->fs_cs(cgno).cs_nbfree, 1);
106                 if (uspi->fs_magic != UFS2_MAGIC) {
107                         unsigned cylno = ufs_cbtocylno (bbase);
108
109                         fs16_add(sb, &ubh_cg_blks(ucpi, cylno,
110                                                   ufs_cbtorpos(bbase)), 1);
111                         fs32_add(sb, &ubh_cg_blktot(ucpi, cylno), 1);
112                 }
113         }
114         
115         ubh_mark_buffer_dirty (USPI_UBH(uspi));
116         ubh_mark_buffer_dirty (UCPI_UBH(ucpi));
117         if (sb->s_flags & MS_SYNCHRONOUS)
118                 ubh_sync_block(UCPI_UBH(ucpi));
119         ufs_mark_sb_dirty(sb);
120         
121         mutex_unlock(&UFS_SB(sb)->s_lock);
122         UFSD("EXIT\n");
123         return;
124
125 failed:
126         mutex_unlock(&UFS_SB(sb)->s_lock);
127         UFSD("EXIT (FAILED)\n");
128         return;
129 }
130
131 /*
132  * Free 'count' fragments from fragment number 'fragment' (free whole blocks)
133  */
134 void ufs_free_blocks(struct inode *inode, u64 fragment, unsigned count)
135 {
136         struct super_block * sb;
137         struct ufs_sb_private_info * uspi;
138         struct ufs_super_block_first * usb1;
139         struct ufs_cg_private_info * ucpi;
140         struct ufs_cylinder_group * ucg;
141         unsigned overflow, cgno, bit, end_bit, i;
142         u64 blkno;
143         
144         sb = inode->i_sb;
145         uspi = UFS_SB(sb)->s_uspi;
146         usb1 = ubh_get_usb_first(uspi);
147
148         UFSD("ENTER, fragment %llu, count %u\n",
149              (unsigned long long)fragment, count);
150         
151         if ((fragment & uspi->s_fpbmask) || (count & uspi->s_fpbmask)) {
152                 ufs_error (sb, "ufs_free_blocks", "internal error, "
153                            "fragment %llu, count %u\n",
154                            (unsigned long long)fragment, count);
155                 goto failed;
156         }
157
158         mutex_lock(&UFS_SB(sb)->s_lock);
159         
160 do_more:
161         overflow = 0;
162         cgno = ufs_dtog(uspi, fragment);
163         bit = ufs_dtogd(uspi, fragment);
164         if (cgno >= uspi->s_ncg) {
165                 ufs_panic (sb, "ufs_free_blocks", "freeing blocks are outside device");
166                 goto failed_unlock;
167         }
168         end_bit = bit + count;
169         if (end_bit > uspi->s_fpg) {
170                 overflow = bit + count - uspi->s_fpg;
171                 count -= overflow;
172                 end_bit -= overflow;
173         }
174
175         ucpi = ufs_load_cylinder (sb, cgno);
176         if (!ucpi) 
177                 goto failed_unlock;
178         ucg = ubh_get_ucg (UCPI_UBH(ucpi));
179         if (!ufs_cg_chkmagic(sb, ucg)) {
180                 ufs_panic (sb, "ufs_free_blocks", "internal error, bad magic number on cg %u", cgno);
181                 goto failed_unlock;
182         }
183
184         for (i = bit; i < end_bit; i += uspi->s_fpb) {
185                 blkno = ufs_fragstoblks(i);
186                 if (ubh_isblockset(UCPI_UBH(ucpi), ucpi->c_freeoff, blkno)) {
187                         ufs_error(sb, "ufs_free_blocks", "freeing free fragment");
188                 }
189                 ubh_setblock(UCPI_UBH(ucpi), ucpi->c_freeoff, blkno);
190                 if ((UFS_SB(sb)->s_flags & UFS_CG_MASK) == UFS_CG_44BSD)
191                         ufs_clusteracct (sb, ucpi, blkno, 1);
192
193                 fs32_add(sb, &ucg->cg_cs.cs_nbfree, 1);
194                 uspi->cs_total.cs_nbfree++;
195                 fs32_add(sb, &UFS_SB(sb)->fs_cs(cgno).cs_nbfree, 1);
196
197                 if (uspi->fs_magic != UFS2_MAGIC) {
198                         unsigned cylno = ufs_cbtocylno(i);
199
200                         fs16_add(sb, &ubh_cg_blks(ucpi, cylno,
201                                                   ufs_cbtorpos(i)), 1);
202                         fs32_add(sb, &ubh_cg_blktot(ucpi, cylno), 1);
203                 }
204         }
205
206         ubh_mark_buffer_dirty (USPI_UBH(uspi));
207         ubh_mark_buffer_dirty (UCPI_UBH(ucpi));
208         if (sb->s_flags & MS_SYNCHRONOUS)
209                 ubh_sync_block(UCPI_UBH(ucpi));
210
211         if (overflow) {
212                 fragment += count;
213                 count = overflow;
214                 goto do_more;
215         }
216
217         ufs_mark_sb_dirty(sb);
218         mutex_unlock(&UFS_SB(sb)->s_lock);
219         UFSD("EXIT\n");
220         return;
221
222 failed_unlock:
223         mutex_unlock(&UFS_SB(sb)->s_lock);
224 failed:
225         UFSD("EXIT (FAILED)\n");
226         return;
227 }
228
229 /*
230  * Modify inode page cache in such way:
231  * have - blocks with b_blocknr equal to oldb...oldb+count-1
232  * get - blocks with b_blocknr equal to newb...newb+count-1
233  * also we suppose that oldb...oldb+count-1 blocks
234  * situated at the end of file.
235  *
236  * We can come here from ufs_writepage or ufs_prepare_write,
237  * locked_page is argument of these functions, so we already lock it.
238  */
239 static void ufs_change_blocknr(struct inode *inode, sector_t beg,
240                                unsigned int count, sector_t oldb,
241                                sector_t newb, struct page *locked_page)
242 {
243         const unsigned blks_per_page =
244                 1 << (PAGE_CACHE_SHIFT - inode->i_blkbits);
245         const unsigned mask = blks_per_page - 1;
246         struct address_space * const mapping = inode->i_mapping;
247         pgoff_t index, cur_index, last_index;
248         unsigned pos, j, lblock;
249         sector_t end, i;
250         struct page *page;
251         struct buffer_head *head, *bh;
252
253         UFSD("ENTER, ino %lu, count %u, oldb %llu, newb %llu\n",
254               inode->i_ino, count,
255              (unsigned long long)oldb, (unsigned long long)newb);
256
257         BUG_ON(!locked_page);
258         BUG_ON(!PageLocked(locked_page));
259
260         cur_index = locked_page->index;
261         end = count + beg;
262         last_index = end >> (PAGE_CACHE_SHIFT - inode->i_blkbits);
263         for (i = beg; i < end; i = (i | mask) + 1) {
264                 index = i >> (PAGE_CACHE_SHIFT - inode->i_blkbits);
265
266                 if (likely(cur_index != index)) {
267                         page = ufs_get_locked_page(mapping, index);
268                         if (!page)/* it was truncated */
269                                 continue;
270                         if (IS_ERR(page)) {/* or EIO */
271                                 ufs_error(inode->i_sb, __func__,
272                                           "read of page %llu failed\n",
273                                           (unsigned long long)index);
274                                 continue;
275                         }
276                 } else
277                         page = locked_page;
278
279                 head = page_buffers(page);
280                 bh = head;
281                 pos = i & mask;
282                 for (j = 0; j < pos; ++j)
283                         bh = bh->b_this_page;
284
285
286                 if (unlikely(index == last_index))
287                         lblock = end & mask;
288                 else
289                         lblock = blks_per_page;
290
291                 do {
292                         if (j >= lblock)
293                                 break;
294                         pos = (i - beg) + j;
295
296                         if (!buffer_mapped(bh))
297                                         map_bh(bh, inode->i_sb, oldb + pos);
298                         if (!buffer_uptodate(bh)) {
299                                 ll_rw_block(READ, 1, &bh);
300                                 wait_on_buffer(bh);
301                                 if (!buffer_uptodate(bh)) {
302                                         ufs_error(inode->i_sb, __func__,
303                                                   "read of block failed\n");
304                                         break;
305                                 }
306                         }
307
308                         UFSD(" change from %llu to %llu, pos %u\n",
309                              (unsigned long long)(pos + oldb),
310                              (unsigned long long)(pos + newb), pos);
311
312                         bh->b_blocknr = newb + pos;
313                         unmap_underlying_metadata(bh->b_bdev,
314                                                   bh->b_blocknr);
315                         mark_buffer_dirty(bh);
316                         ++j;
317                         bh = bh->b_this_page;
318                 } while (bh != head);
319
320                 if (likely(cur_index != index))
321                         ufs_put_locked_page(page);
322         }
323         UFSD("EXIT\n");
324 }
325
326 static void ufs_clear_frags(struct inode *inode, sector_t beg, unsigned int n,
327                             int sync)
328 {
329         struct buffer_head *bh;
330         sector_t end = beg + n;
331
332         for (; beg < end; ++beg) {
333                 bh = sb_getblk(inode->i_sb, beg);
334                 lock_buffer(bh);
335                 memset(bh->b_data, 0, inode->i_sb->s_blocksize);
336                 set_buffer_uptodate(bh);
337                 mark_buffer_dirty(bh);
338                 unlock_buffer(bh);
339                 if (IS_SYNC(inode) || sync)
340                         sync_dirty_buffer(bh);
341                 brelse(bh);
342         }
343 }
344
345 u64 ufs_new_fragments(struct inode *inode, void *p, u64 fragment,
346                            u64 goal, unsigned count, int *err,
347                            struct page *locked_page)
348 {
349         struct super_block * sb;
350         struct ufs_sb_private_info * uspi;
351         struct ufs_super_block_first * usb1;
352         unsigned cgno, oldcount, newcount;
353         u64 tmp, request, result;
354         
355         UFSD("ENTER, ino %lu, fragment %llu, goal %llu, count %u\n",
356              inode->i_ino, (unsigned long long)fragment,
357              (unsigned long long)goal, count);
358         
359         sb = inode->i_sb;
360         uspi = UFS_SB(sb)->s_uspi;
361         usb1 = ubh_get_usb_first(uspi);
362         *err = -ENOSPC;
363
364         mutex_lock(&UFS_SB(sb)->s_lock);
365         tmp = ufs_data_ptr_to_cpu(sb, p);
366
367         if (count + ufs_fragnum(fragment) > uspi->s_fpb) {
368                 ufs_warning(sb, "ufs_new_fragments", "internal warning"
369                             " fragment %llu, count %u",
370                             (unsigned long long)fragment, count);
371                 count = uspi->s_fpb - ufs_fragnum(fragment); 
372         }
373         oldcount = ufs_fragnum (fragment);
374         newcount = oldcount + count;
375
376         /*
377          * Somebody else has just allocated our fragments
378          */
379         if (oldcount) {
380                 if (!tmp) {
381                         ufs_error(sb, "ufs_new_fragments", "internal error, "
382                                   "fragment %llu, tmp %llu\n",
383                                   (unsigned long long)fragment,
384                                   (unsigned long long)tmp);
385                         mutex_unlock(&UFS_SB(sb)->s_lock);
386                         return INVBLOCK;
387                 }
388                 if (fragment < UFS_I(inode)->i_lastfrag) {
389                         UFSD("EXIT (ALREADY ALLOCATED)\n");
390                         mutex_unlock(&UFS_SB(sb)->s_lock);
391                         return 0;
392                 }
393         }
394         else {
395                 if (tmp) {
396                         UFSD("EXIT (ALREADY ALLOCATED)\n");
397                         mutex_unlock(&UFS_SB(sb)->s_lock);
398                         return 0;
399                 }
400         }
401
402         /*
403          * There is not enough space for user on the device
404          */
405         if (!capable(CAP_SYS_RESOURCE) && ufs_freespace(uspi, UFS_MINFREE) <= 0) {
406                 mutex_unlock(&UFS_SB(sb)->s_lock);
407                 UFSD("EXIT (FAILED)\n");
408                 return 0;
409         }
410
411         if (goal >= uspi->s_size) 
412                 goal = 0;
413         if (goal == 0) 
414                 cgno = ufs_inotocg (inode->i_ino);
415         else
416                 cgno = ufs_dtog(uspi, goal);
417          
418         /*
419          * allocate new fragment
420          */
421         if (oldcount == 0) {
422                 result = ufs_alloc_fragments (inode, cgno, goal, count, err);
423                 if (result) {
424                         ufs_cpu_to_data_ptr(sb, p, result);
425                         *err = 0;
426                         UFS_I(inode)->i_lastfrag =
427                                 max(UFS_I(inode)->i_lastfrag, fragment + count);
428                         ufs_clear_frags(inode, result + oldcount,
429                                         newcount - oldcount, locked_page != NULL);
430                 }
431                 mutex_unlock(&UFS_SB(sb)->s_lock);
432                 UFSD("EXIT, result %llu\n", (unsigned long long)result);
433                 return result;
434         }
435
436         /*
437          * resize block
438          */
439         result = ufs_add_fragments (inode, tmp, oldcount, newcount, err);
440         if (result) {
441                 *err = 0;
442                 UFS_I(inode)->i_lastfrag = max(UFS_I(inode)->i_lastfrag,
443                                                 fragment + count);
444                 ufs_clear_frags(inode, result + oldcount, newcount - oldcount,
445                                 locked_page != NULL);
446                 mutex_unlock(&UFS_SB(sb)->s_lock);
447                 UFSD("EXIT, result %llu\n", (unsigned long long)result);
448                 return result;
449         }
450
451         /*
452          * allocate new block and move data
453          */
454         switch (fs32_to_cpu(sb, usb1->fs_optim)) {
455             case UFS_OPTSPACE:
456                 request = newcount;
457                 if (uspi->s_minfree < 5 || uspi->cs_total.cs_nffree
458                     > uspi->s_dsize * uspi->s_minfree / (2 * 100))
459                         break;
460                 usb1->fs_optim = cpu_to_fs32(sb, UFS_OPTTIME);
461                 break;
462             default:
463                 usb1->fs_optim = cpu_to_fs32(sb, UFS_OPTTIME);
464         
465             case UFS_OPTTIME:
466                 request = uspi->s_fpb;
467                 if (uspi->cs_total.cs_nffree < uspi->s_dsize *
468                     (uspi->s_minfree - 2) / 100)
469                         break;
470                 usb1->fs_optim = cpu_to_fs32(sb, UFS_OPTTIME);
471                 break;
472         }
473         result = ufs_alloc_fragments (inode, cgno, goal, request, err);
474         if (result) {
475                 ufs_clear_frags(inode, result + oldcount, newcount - oldcount,
476                                 locked_page != NULL);
477                 ufs_change_blocknr(inode, fragment - oldcount, oldcount,
478                                    uspi->s_sbbase + tmp,
479                                    uspi->s_sbbase + result, locked_page);
480                 ufs_cpu_to_data_ptr(sb, p, result);
481                 *err = 0;
482                 UFS_I(inode)->i_lastfrag = max(UFS_I(inode)->i_lastfrag,
483                                                 fragment + count);
484                 mutex_unlock(&UFS_SB(sb)->s_lock);
485                 if (newcount < request)
486                         ufs_free_fragments (inode, result + newcount, request - newcount);
487                 ufs_free_fragments (inode, tmp, oldcount);
488                 UFSD("EXIT, result %llu\n", (unsigned long long)result);
489                 return result;
490         }
491
492         mutex_unlock(&UFS_SB(sb)->s_lock);
493         UFSD("EXIT (FAILED)\n");
494         return 0;
495 }               
496
497 static u64 ufs_add_fragments(struct inode *inode, u64 fragment,
498                              unsigned oldcount, unsigned newcount, int *err)
499 {
500         struct super_block * sb;
501         struct ufs_sb_private_info * uspi;
502         struct ufs_super_block_first * usb1;
503         struct ufs_cg_private_info * ucpi;
504         struct ufs_cylinder_group * ucg;
505         unsigned cgno, fragno, fragoff, count, fragsize, i;
506         
507         UFSD("ENTER, fragment %llu, oldcount %u, newcount %u\n",
508              (unsigned long long)fragment, oldcount, newcount);
509         
510         sb = inode->i_sb;
511         uspi = UFS_SB(sb)->s_uspi;
512         usb1 = ubh_get_usb_first (uspi);
513         count = newcount - oldcount;
514         
515         cgno = ufs_dtog(uspi, fragment);
516         if (fs32_to_cpu(sb, UFS_SB(sb)->fs_cs(cgno).cs_nffree) < count)
517                 return 0;
518         if ((ufs_fragnum (fragment) + newcount) > uspi->s_fpb)
519                 return 0;
520         ucpi = ufs_load_cylinder (sb, cgno);
521         if (!ucpi)
522                 return 0;
523         ucg = ubh_get_ucg (UCPI_UBH(ucpi));
524         if (!ufs_cg_chkmagic(sb, ucg)) {
525                 ufs_panic (sb, "ufs_add_fragments",
526                         "internal error, bad magic number on cg %u", cgno);
527                 return 0;
528         }
529
530         fragno = ufs_dtogd(uspi, fragment);
531         fragoff = ufs_fragnum (fragno);
532         for (i = oldcount; i < newcount; i++)
533                 if (ubh_isclr (UCPI_UBH(ucpi), ucpi->c_freeoff, fragno + i))
534                         return 0;
535         /*
536          * Block can be extended
537          */
538         ucg->cg_time = cpu_to_fs32(sb, get_seconds());
539         for (i = newcount; i < (uspi->s_fpb - fragoff); i++)
540                 if (ubh_isclr (UCPI_UBH(ucpi), ucpi->c_freeoff, fragno + i))
541                         break;
542         fragsize = i - oldcount;
543         if (!fs32_to_cpu(sb, ucg->cg_frsum[fragsize]))
544                 ufs_panic (sb, "ufs_add_fragments",
545                         "internal error or corrupted bitmap on cg %u", cgno);
546         fs32_sub(sb, &ucg->cg_frsum[fragsize], 1);
547         if (fragsize != count)
548                 fs32_add(sb, &ucg->cg_frsum[fragsize - count], 1);
549         for (i = oldcount; i < newcount; i++)
550                 ubh_clrbit (UCPI_UBH(ucpi), ucpi->c_freeoff, fragno + i);
551
552         fs32_sub(sb, &ucg->cg_cs.cs_nffree, count);
553         fs32_sub(sb, &UFS_SB(sb)->fs_cs(cgno).cs_nffree, count);
554         uspi->cs_total.cs_nffree -= count;
555         
556         ubh_mark_buffer_dirty (USPI_UBH(uspi));
557         ubh_mark_buffer_dirty (UCPI_UBH(ucpi));
558         if (sb->s_flags & MS_SYNCHRONOUS)
559                 ubh_sync_block(UCPI_UBH(ucpi));
560         ufs_mark_sb_dirty(sb);
561
562         UFSD("EXIT, fragment %llu\n", (unsigned long long)fragment);
563         
564         return fragment;
565 }
566
567 #define UFS_TEST_FREE_SPACE_CG \
568         ucg = (struct ufs_cylinder_group *) UFS_SB(sb)->s_ucg[cgno]->b_data; \
569         if (fs32_to_cpu(sb, ucg->cg_cs.cs_nbfree)) \
570                 goto cg_found; \
571         for (k = count; k < uspi->s_fpb; k++) \
572                 if (fs32_to_cpu(sb, ucg->cg_frsum[k])) \
573                         goto cg_found; 
574
575 static u64 ufs_alloc_fragments(struct inode *inode, unsigned cgno,
576                                u64 goal, unsigned count, int *err)
577 {
578         struct super_block * sb;
579         struct ufs_sb_private_info * uspi;
580         struct ufs_super_block_first * usb1;
581         struct ufs_cg_private_info * ucpi;
582         struct ufs_cylinder_group * ucg;
583         unsigned oldcg, i, j, k, allocsize;
584         u64 result;
585         
586         UFSD("ENTER, ino %lu, cgno %u, goal %llu, count %u\n",
587              inode->i_ino, cgno, (unsigned long long)goal, count);
588
589         sb = inode->i_sb;
590         uspi = UFS_SB(sb)->s_uspi;
591         usb1 = ubh_get_usb_first(uspi);
592         oldcg = cgno;
593         
594         /*
595          * 1. searching on preferred cylinder group
596          */
597         UFS_TEST_FREE_SPACE_CG
598
599         /*
600          * 2. quadratic rehash
601          */
602         for (j = 1; j < uspi->s_ncg; j *= 2) {
603                 cgno += j;
604                 if (cgno >= uspi->s_ncg) 
605                         cgno -= uspi->s_ncg;
606                 UFS_TEST_FREE_SPACE_CG
607         }
608
609         /*
610          * 3. brute force search
611          * We start at i = 2 ( 0 is checked at 1.step, 1 at 2.step )
612          */
613         cgno = (oldcg + 1) % uspi->s_ncg;
614         for (j = 2; j < uspi->s_ncg; j++) {
615                 cgno++;
616                 if (cgno >= uspi->s_ncg)
617                         cgno = 0;
618                 UFS_TEST_FREE_SPACE_CG
619         }
620         
621         UFSD("EXIT (FAILED)\n");
622         return 0;
623
624 cg_found:
625         ucpi = ufs_load_cylinder (sb, cgno);
626         if (!ucpi)
627                 return 0;
628         ucg = ubh_get_ucg (UCPI_UBH(ucpi));
629         if (!ufs_cg_chkmagic(sb, ucg)) 
630                 ufs_panic (sb, "ufs_alloc_fragments",
631                         "internal error, bad magic number on cg %u", cgno);
632         ucg->cg_time = cpu_to_fs32(sb, get_seconds());
633
634         if (count == uspi->s_fpb) {
635                 result = ufs_alloccg_block (inode, ucpi, goal, err);
636                 if (result == INVBLOCK)
637                         return 0;
638                 goto succed;
639         }
640
641         for (allocsize = count; allocsize < uspi->s_fpb; allocsize++)
642                 if (fs32_to_cpu(sb, ucg->cg_frsum[allocsize]) != 0)
643                         break;
644         
645         if (allocsize == uspi->s_fpb) {
646                 result = ufs_alloccg_block (inode, ucpi, goal, err);
647                 if (result == INVBLOCK)
648                         return 0;
649                 goal = ufs_dtogd(uspi, result);
650                 for (i = count; i < uspi->s_fpb; i++)
651                         ubh_setbit (UCPI_UBH(ucpi), ucpi->c_freeoff, goal + i);
652                 i = uspi->s_fpb - count;
653
654                 fs32_add(sb, &ucg->cg_cs.cs_nffree, i);
655                 uspi->cs_total.cs_nffree += i;
656                 fs32_add(sb, &UFS_SB(sb)->fs_cs(cgno).cs_nffree, i);
657                 fs32_add(sb, &ucg->cg_frsum[i], 1);
658                 goto succed;
659         }
660
661         result = ufs_bitmap_search (sb, ucpi, goal, allocsize);
662         if (result == INVBLOCK)
663                 return 0;
664         for (i = 0; i < count; i++)
665                 ubh_clrbit (UCPI_UBH(ucpi), ucpi->c_freeoff, result + i);
666         
667         fs32_sub(sb, &ucg->cg_cs.cs_nffree, count);
668         uspi->cs_total.cs_nffree -= count;
669         fs32_sub(sb, &UFS_SB(sb)->fs_cs(cgno).cs_nffree, count);
670         fs32_sub(sb, &ucg->cg_frsum[allocsize], 1);
671
672         if (count != allocsize)
673                 fs32_add(sb, &ucg->cg_frsum[allocsize - count], 1);
674
675 succed:
676         ubh_mark_buffer_dirty (USPI_UBH(uspi));
677         ubh_mark_buffer_dirty (UCPI_UBH(ucpi));
678         if (sb->s_flags & MS_SYNCHRONOUS)
679                 ubh_sync_block(UCPI_UBH(ucpi));
680         ufs_mark_sb_dirty(sb);
681
682         result += cgno * uspi->s_fpg;
683         UFSD("EXIT3, result %llu\n", (unsigned long long)result);
684         return result;
685 }
686
687 static u64 ufs_alloccg_block(struct inode *inode,
688                              struct ufs_cg_private_info *ucpi,
689                              u64 goal, int *err)
690 {
691         struct super_block * sb;
692         struct ufs_sb_private_info * uspi;
693         struct ufs_super_block_first * usb1;
694         struct ufs_cylinder_group * ucg;
695         u64 result, blkno;
696
697         UFSD("ENTER, goal %llu\n", (unsigned long long)goal);
698
699         sb = inode->i_sb;
700         uspi = UFS_SB(sb)->s_uspi;
701         usb1 = ubh_get_usb_first(uspi);
702         ucg = ubh_get_ucg(UCPI_UBH(ucpi));
703
704         if (goal == 0) {
705                 goal = ucpi->c_rotor;
706                 goto norot;
707         }
708         goal = ufs_blknum (goal);
709         goal = ufs_dtogd(uspi, goal);
710         
711         /*
712          * If the requested block is available, use it.
713          */
714         if (ubh_isblockset(UCPI_UBH(ucpi), ucpi->c_freeoff, ufs_fragstoblks(goal))) {
715                 result = goal;
716                 goto gotit;
717         }
718         
719 norot:  
720         result = ufs_bitmap_search (sb, ucpi, goal, uspi->s_fpb);
721         if (result == INVBLOCK)
722                 return INVBLOCK;
723         ucpi->c_rotor = result;
724 gotit:
725         blkno = ufs_fragstoblks(result);
726         ubh_clrblock (UCPI_UBH(ucpi), ucpi->c_freeoff, blkno);
727         if ((UFS_SB(sb)->s_flags & UFS_CG_MASK) == UFS_CG_44BSD)
728                 ufs_clusteracct (sb, ucpi, blkno, -1);
729
730         fs32_sub(sb, &ucg->cg_cs.cs_nbfree, 1);
731         uspi->cs_total.cs_nbfree--;
732         fs32_sub(sb, &UFS_SB(sb)->fs_cs(ucpi->c_cgx).cs_nbfree, 1);
733
734         if (uspi->fs_magic != UFS2_MAGIC) {
735                 unsigned cylno = ufs_cbtocylno((unsigned)result);
736
737                 fs16_sub(sb, &ubh_cg_blks(ucpi, cylno,
738                                           ufs_cbtorpos((unsigned)result)), 1);
739                 fs32_sub(sb, &ubh_cg_blktot(ucpi, cylno), 1);
740         }
741         
742         UFSD("EXIT, result %llu\n", (unsigned long long)result);
743
744         return result;
745 }
746
747 static unsigned ubh_scanc(struct ufs_sb_private_info *uspi,
748                           struct ufs_buffer_head *ubh,
749                           unsigned begin, unsigned size,
750                           unsigned char *table, unsigned char mask)
751 {
752         unsigned rest, offset;
753         unsigned char *cp;
754         
755
756         offset = begin & ~uspi->s_fmask;
757         begin >>= uspi->s_fshift;
758         for (;;) {
759                 if ((offset + size) < uspi->s_fsize)
760                         rest = size;
761                 else
762                         rest = uspi->s_fsize - offset;
763                 size -= rest;
764                 cp = ubh->bh[begin]->b_data + offset;
765                 while ((table[*cp++] & mask) == 0 && --rest)
766                         ;
767                 if (rest || !size)
768                         break;
769                 begin++;
770                 offset = 0;
771         }
772         return (size + rest);
773 }
774
775 /*
776  * Find a block of the specified size in the specified cylinder group.
777  * @sp: pointer to super block
778  * @ucpi: pointer to cylinder group info
779  * @goal: near which block we want find new one
780  * @count: specified size
781  */
782 static u64 ufs_bitmap_search(struct super_block *sb,
783                              struct ufs_cg_private_info *ucpi,
784                              u64 goal, unsigned count)
785 {
786         /*
787          * Bit patterns for identifying fragments in the block map
788          * used as ((map & mask_arr) == want_arr)
789          */
790         static const int mask_arr[9] = {
791                 0x3, 0x7, 0xf, 0x1f, 0x3f, 0x7f, 0xff, 0x1ff, 0x3ff
792         };
793         static const int want_arr[9] = {
794                 0x0, 0x2, 0x6, 0xe, 0x1e, 0x3e, 0x7e, 0xfe, 0x1fe
795         };
796         struct ufs_sb_private_info *uspi = UFS_SB(sb)->s_uspi;
797         struct ufs_super_block_first *usb1;
798         struct ufs_cylinder_group *ucg;
799         unsigned start, length, loc;
800         unsigned pos, want, blockmap, mask, end;
801         u64 result;
802
803         UFSD("ENTER, cg %u, goal %llu, count %u\n", ucpi->c_cgx,
804              (unsigned long long)goal, count);
805
806         usb1 = ubh_get_usb_first (uspi);
807         ucg = ubh_get_ucg(UCPI_UBH(ucpi));
808
809         if (goal)
810                 start = ufs_dtogd(uspi, goal) >> 3;
811         else
812                 start = ucpi->c_frotor >> 3;
813                 
814         length = ((uspi->s_fpg + 7) >> 3) - start;
815         loc = ubh_scanc(uspi, UCPI_UBH(ucpi), ucpi->c_freeoff + start, length,
816                 (uspi->s_fpb == 8) ? ufs_fragtable_8fpb : ufs_fragtable_other,
817                 1 << (count - 1 + (uspi->s_fpb & 7))); 
818         if (loc == 0) {
819                 length = start + 1;
820                 loc = ubh_scanc(uspi, UCPI_UBH(ucpi), ucpi->c_freeoff, length,
821                                 (uspi->s_fpb == 8) ? ufs_fragtable_8fpb :
822                                 ufs_fragtable_other,
823                                 1 << (count - 1 + (uspi->s_fpb & 7)));
824                 if (loc == 0) {
825                         ufs_error(sb, "ufs_bitmap_search",
826                                   "bitmap corrupted on cg %u, start %u,"
827                                   " length %u, count %u, freeoff %u\n",
828                                   ucpi->c_cgx, start, length, count,
829                                   ucpi->c_freeoff);
830                         return INVBLOCK;
831                 }
832                 start = 0;
833         }
834         result = (start + length - loc) << 3;
835         ucpi->c_frotor = result;
836
837         /*
838          * found the byte in the map
839          */
840
841         for (end = result + 8; result < end; result += uspi->s_fpb) {
842                 blockmap = ubh_blkmap(UCPI_UBH(ucpi), ucpi->c_freeoff, result);
843                 blockmap <<= 1;
844                 mask = mask_arr[count];
845                 want = want_arr[count];
846                 for (pos = 0; pos <= uspi->s_fpb - count; pos++) {
847                         if ((blockmap & mask) == want) {
848                                 UFSD("EXIT, result %llu\n",
849                                      (unsigned long long)result);
850                                 return result + pos;
851                         }
852                         mask <<= 1;
853                         want <<= 1;
854                 }
855         }
856
857         ufs_error(sb, "ufs_bitmap_search", "block not in map on cg %u\n",
858                   ucpi->c_cgx);
859         UFSD("EXIT (FAILED)\n");
860         return INVBLOCK;
861 }
862
863 static void ufs_clusteracct(struct super_block * sb,
864         struct ufs_cg_private_info * ucpi, unsigned blkno, int cnt)
865 {
866         struct ufs_sb_private_info * uspi;
867         int i, start, end, forw, back;
868         
869         uspi = UFS_SB(sb)->s_uspi;
870         if (uspi->s_contigsumsize <= 0)
871                 return;
872
873         if (cnt > 0)
874                 ubh_setbit(UCPI_UBH(ucpi), ucpi->c_clusteroff, blkno);
875         else
876                 ubh_clrbit(UCPI_UBH(ucpi), ucpi->c_clusteroff, blkno);
877
878         /*
879          * Find the size of the cluster going forward.
880          */
881         start = blkno + 1;
882         end = start + uspi->s_contigsumsize;
883         if ( end >= ucpi->c_nclusterblks)
884                 end = ucpi->c_nclusterblks;
885         i = ubh_find_next_zero_bit (UCPI_UBH(ucpi), ucpi->c_clusteroff, end, start);
886         if (i > end)
887                 i = end;
888         forw = i - start;
889         
890         /*
891          * Find the size of the cluster going backward.
892          */
893         start = blkno - 1;
894         end = start - uspi->s_contigsumsize;
895         if (end < 0 ) 
896                 end = -1;
897         i = ubh_find_last_zero_bit (UCPI_UBH(ucpi), ucpi->c_clusteroff, start, end);
898         if ( i < end) 
899                 i = end;
900         back = start - i;
901         
902         /*
903          * Account for old cluster and the possibly new forward and
904          * back clusters.
905          */
906         i = back + forw + 1;
907         if (i > uspi->s_contigsumsize)
908                 i = uspi->s_contigsumsize;
909         fs32_add(sb, (__fs32*)ubh_get_addr(UCPI_UBH(ucpi), ucpi->c_clustersumoff + (i << 2)), cnt);
910         if (back > 0)
911                 fs32_sub(sb, (__fs32*)ubh_get_addr(UCPI_UBH(ucpi), ucpi->c_clustersumoff + (back << 2)), cnt);
912         if (forw > 0)
913                 fs32_sub(sb, (__fs32*)ubh_get_addr(UCPI_UBH(ucpi), ucpi->c_clustersumoff + (forw << 2)), cnt);
914 }
915
916
917 static unsigned char ufs_fragtable_8fpb[] = {
918         0x00, 0x01, 0x01, 0x02, 0x01, 0x01, 0x02, 0x04, 0x01, 0x01, 0x01, 0x03, 0x02, 0x03, 0x04, 0x08,
919         0x01, 0x01, 0x01, 0x03, 0x01, 0x01, 0x03, 0x05, 0x02, 0x03, 0x03, 0x02, 0x04, 0x05, 0x08, 0x10,
920         0x01, 0x01, 0x01, 0x03, 0x01, 0x01, 0x03, 0x05, 0x01, 0x01, 0x01, 0x03, 0x03, 0x03, 0x05, 0x09,
921         0x02, 0x03, 0x03, 0x02, 0x03, 0x03, 0x02, 0x06, 0x04, 0x05, 0x05, 0x06, 0x08, 0x09, 0x10, 0x20,
922         0x01, 0x01, 0x01, 0x03, 0x01, 0x01, 0x03, 0x05, 0x01, 0x01, 0x01, 0x03, 0x03, 0x03, 0x05, 0x09, 
923         0x01, 0x01, 0x01, 0x03, 0x01, 0x01, 0x03, 0x05, 0x03, 0x03, 0x03, 0x03, 0x05, 0x05, 0x09, 0x11,
924         0x02, 0x03, 0x03, 0x02, 0x03, 0x03, 0x02, 0x06, 0x03, 0x03, 0x03, 0x03, 0x02, 0x03, 0x06, 0x0A,
925         0x04, 0x05, 0x05, 0x06, 0x05, 0x05, 0x06, 0x04, 0x08, 0x09, 0x09, 0x0A, 0x10, 0x11, 0x20, 0x40,
926         0x01, 0x01, 0x01, 0x03, 0x01, 0x01, 0x03, 0x05, 0x01, 0x01, 0x01, 0x03, 0x03, 0x03, 0x05, 0x09,
927         0x01, 0x01, 0x01, 0x03, 0x01, 0x01, 0x03, 0x05, 0x03, 0x03, 0x03, 0x03, 0x05, 0x05, 0x09, 0x11,
928         0x01, 0x01, 0x01, 0x03, 0x01, 0x01, 0x03, 0x05, 0x01, 0x01, 0x01, 0x03, 0x03, 0x03, 0x05, 0x09,
929         0x03, 0x03, 0x03, 0x03, 0x03, 0x03, 0x03, 0x07, 0x05, 0x05, 0x05, 0x07, 0x09, 0x09, 0x11, 0x21,
930         0x02, 0x03, 0x03, 0x02, 0x03, 0x03, 0x02, 0x06, 0x03, 0x03, 0x03, 0x03, 0x02, 0x03, 0x06, 0x0A,
931         0x03, 0x03, 0x03, 0x03, 0x03, 0x03, 0x03, 0x07, 0x02, 0x03, 0x03, 0x02, 0x06, 0x07, 0x0A, 0x12,
932         0x04, 0x05, 0x05, 0x06, 0x05, 0x05, 0x06, 0x04, 0x05, 0x05, 0x05, 0x07, 0x06, 0x07, 0x04, 0x0C,
933         0x08, 0x09, 0x09, 0x0A, 0x09, 0x09, 0x0A, 0x0C, 0x10, 0x11, 0x11, 0x12, 0x20, 0x21, 0x40, 0x80,
934 };
935
936 static unsigned char ufs_fragtable_other[] = {
937         0x00, 0x16, 0x16, 0x2A, 0x16, 0x16, 0x26, 0x4E, 0x16, 0x16, 0x16, 0x3E, 0x2A, 0x3E, 0x4E, 0x8A,
938         0x16, 0x16, 0x16, 0x3E, 0x16, 0x16, 0x36, 0x5E, 0x16, 0x16, 0x16, 0x3E, 0x3E, 0x3E, 0x5E, 0x9E,
939         0x16, 0x16, 0x16, 0x3E, 0x16, 0x16, 0x36, 0x5E, 0x16, 0x16, 0x16, 0x3E, 0x3E, 0x3E, 0x5E, 0x9E,
940         0x2A, 0x3E, 0x3E, 0x2A, 0x3E, 0x3E, 0x2E, 0x6E, 0x3E, 0x3E, 0x3E, 0x3E, 0x2A, 0x3E, 0x6E, 0xAA,
941         0x16, 0x16, 0x16, 0x3E, 0x16, 0x16, 0x36, 0x5E, 0x16, 0x16, 0x16, 0x3E, 0x3E, 0x3E, 0x5E, 0x9E,
942         0x16, 0x16, 0x16, 0x3E, 0x16, 0x16, 0x36, 0x5E, 0x16, 0x16, 0x16, 0x3E, 0x3E, 0x3E, 0x5E, 0x9E,
943         0x26, 0x36, 0x36, 0x2E, 0x36, 0x36, 0x26, 0x6E, 0x36, 0x36, 0x36, 0x3E, 0x2E, 0x3E, 0x6E, 0xAE,
944         0x4E, 0x5E, 0x5E, 0x6E, 0x5E, 0x5E, 0x6E, 0x4E, 0x5E, 0x5E, 0x5E, 0x7E, 0x6E, 0x7E, 0x4E, 0xCE,
945         0x16, 0x16, 0x16, 0x3E, 0x16, 0x16, 0x36, 0x5E, 0x16, 0x16, 0x16, 0x3E, 0x3E, 0x3E, 0x5E, 0x9E,
946         0x16, 0x16, 0x16, 0x3E, 0x16, 0x16, 0x36, 0x5E, 0x16, 0x16, 0x16, 0x3E, 0x3E, 0x3E, 0x5E, 0x9E,
947         0x16, 0x16, 0x16, 0x3E, 0x16, 0x16, 0x36, 0x5E, 0x16, 0x16, 0x16, 0x3E, 0x3E, 0x3E, 0x5E, 0x9E,
948         0x3E, 0x3E, 0x3E, 0x3E, 0x3E, 0x3E, 0x3E, 0x7E, 0x3E, 0x3E, 0x3E, 0x3E, 0x3E, 0x3E, 0x7E, 0xBE,
949         0x2A, 0x3E, 0x3E, 0x2A, 0x3E, 0x3E, 0x2E, 0x6E, 0x3E, 0x3E, 0x3E, 0x3E, 0x2A, 0x3E, 0x6E, 0xAA,
950         0x3E, 0x3E, 0x3E, 0x3E, 0x3E, 0x3E, 0x3E, 0x7E, 0x3E, 0x3E, 0x3E, 0x3E, 0x3E, 0x3E, 0x7E, 0xBE,
951         0x4E, 0x5E, 0x5E, 0x6E, 0x5E, 0x5E, 0x6E, 0x4E, 0x5E, 0x5E, 0x5E, 0x7E, 0x6E, 0x7E, 0x4E, 0xCE,
952         0x8A, 0x9E, 0x9E, 0xAA, 0x9E, 0x9E, 0xAE, 0xCE, 0x9E, 0x9E, 0x9E, 0xBE, 0xAA, 0xBE, 0xCE, 0x8A,
953 };