xref: /linux/fs/ufs/balloc.c (revision f884ab15afdc5514e88105c92a4e2e1e6539869a)
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 };
954