|
Kmods SIG |
d83023 |
From 3f3b442b5ad2455507c9bfdacf39a3792eb3a6d0 Mon Sep 17 00:00:00 2001
|
|
Kmods SIG |
d83023 |
From: Konstantin Komarov <almaz.alexandrovich@paragon-software.com>
|
|
Kmods SIG |
d83023 |
Date: Fri, 13 Aug 2021 17:21:29 +0300
|
|
Kmods SIG |
d83023 |
Subject: [Backport 3f3b442b5ad2] src: Add bitmap
|
|
Kmods SIG |
d83023 |
|
|
Kmods SIG |
d83023 |
This adds bitmap
|
|
Kmods SIG |
d83023 |
|
|
Kmods SIG |
d83023 |
Signed-off-by: Konstantin Komarov <almaz.alexandrovich@paragon-software.com>
|
|
Kmods SIG |
d83023 |
---
|
|
Kmods SIG |
d83023 |
src/bitfunc.c | 135 ++++
|
|
Kmods SIG |
d83023 |
src/bitmap.c | 1519 ++++++++++++++++++++++++++++++++++++++++++++
|
|
Kmods SIG |
d83023 |
2 files changed, 1654 insertions(+)
|
|
Kmods SIG |
d83023 |
create mode 100644 src/bitfunc.c
|
|
Kmods SIG |
d83023 |
create mode 100644 src/bitmap.c
|
|
Kmods SIG |
d83023 |
|
|
Kmods SIG |
d83023 |
diff --git a/src/bitfunc.c b/src/bitfunc.c
|
|
Kmods SIG |
d83023 |
new file mode 100644
|
|
Kmods SIG |
d83023 |
index 0000000000000000000000000000000000000000..2de5faef2721d5bac9caf7ff81adec167ae21dea
|
|
Kmods SIG |
d83023 |
--- /dev/null
|
|
Kmods SIG |
d83023 |
+++ b/src/bitfunc.c
|
|
Kmods SIG |
d83023 |
@@ -0,0 +1,135 @@
|
|
Kmods SIG |
d83023 |
+// SPDX-License-Identifier: GPL-2.0
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ *
|
|
Kmods SIG |
d83023 |
+ * Copyright (C) 2019-2021 Paragon Software GmbH, All rights reserved.
|
|
Kmods SIG |
d83023 |
+ *
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+#include <linux/blkdev.h>
|
|
Kmods SIG |
d83023 |
+#include <linux/buffer_head.h>
|
|
Kmods SIG |
d83023 |
+#include <linux/fs.h>
|
|
Kmods SIG |
d83023 |
+#include <linux/nls.h>
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+#include "debug.h"
|
|
Kmods SIG |
d83023 |
+#include "ntfs.h"
|
|
Kmods SIG |
d83023 |
+#include "ntfs_fs.h"
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+#define BITS_IN_SIZE_T (sizeof(size_t) * 8)
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ * fill_mask[i] - first i bits are '1' , i = 0,1,2,3,4,5,6,7,8
|
|
Kmods SIG |
d83023 |
+ * fill_mask[i] = 0xFF >> (8-i)
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+static const u8 fill_mask[] = { 0x00, 0x01, 0x03, 0x07, 0x0F,
|
|
Kmods SIG |
d83023 |
+ 0x1F, 0x3F, 0x7F, 0xFF };
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ * zero_mask[i] - first i bits are '0' , i = 0,1,2,3,4,5,6,7,8
|
|
Kmods SIG |
d83023 |
+ * zero_mask[i] = 0xFF << i
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+static const u8 zero_mask[] = { 0xFF, 0xFE, 0xFC, 0xF8, 0xF0,
|
|
Kmods SIG |
d83023 |
+ 0xE0, 0xC0, 0x80, 0x00 };
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ * are_bits_clear
|
|
Kmods SIG |
d83023 |
+ *
|
|
Kmods SIG |
d83023 |
+ * Returns true if all bits [bit, bit+nbits) are zeros "0"
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+bool are_bits_clear(const ulong *lmap, size_t bit, size_t nbits)
|
|
Kmods SIG |
d83023 |
+{
|
|
Kmods SIG |
d83023 |
+ size_t pos = bit & 7;
|
|
Kmods SIG |
d83023 |
+ const u8 *map = (u8 *)lmap + (bit >> 3);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (pos) {
|
|
Kmods SIG |
d83023 |
+ if (8 - pos >= nbits)
|
|
Kmods SIG |
d83023 |
+ return !nbits || !(*map & fill_mask[pos + nbits] &
|
|
Kmods SIG |
d83023 |
+ zero_mask[pos]);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (*map++ & zero_mask[pos])
|
|
Kmods SIG |
d83023 |
+ return false;
|
|
Kmods SIG |
d83023 |
+ nbits -= 8 - pos;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ pos = ((size_t)map) & (sizeof(size_t) - 1);
|
|
Kmods SIG |
d83023 |
+ if (pos) {
|
|
Kmods SIG |
d83023 |
+ pos = sizeof(size_t) - pos;
|
|
Kmods SIG |
d83023 |
+ if (nbits >= pos * 8) {
|
|
Kmods SIG |
d83023 |
+ for (nbits -= pos * 8; pos; pos--, map++) {
|
|
Kmods SIG |
d83023 |
+ if (*map)
|
|
Kmods SIG |
d83023 |
+ return false;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ for (pos = nbits / BITS_IN_SIZE_T; pos; pos--, map += sizeof(size_t)) {
|
|
Kmods SIG |
d83023 |
+ if (*((size_t *)map))
|
|
Kmods SIG |
d83023 |
+ return false;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ for (pos = (nbits % BITS_IN_SIZE_T) >> 3; pos; pos--, map++) {
|
|
Kmods SIG |
d83023 |
+ if (*map)
|
|
Kmods SIG |
d83023 |
+ return false;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ pos = nbits & 7;
|
|
Kmods SIG |
d83023 |
+ if (pos && (*map & fill_mask[pos]))
|
|
Kmods SIG |
d83023 |
+ return false;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ // All bits are zero
|
|
Kmods SIG |
d83023 |
+ return true;
|
|
Kmods SIG |
d83023 |
+}
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ * are_bits_set
|
|
Kmods SIG |
d83023 |
+ *
|
|
Kmods SIG |
d83023 |
+ * Returns true if all bits [bit, bit+nbits) are ones "1"
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+bool are_bits_set(const ulong *lmap, size_t bit, size_t nbits)
|
|
Kmods SIG |
d83023 |
+{
|
|
Kmods SIG |
d83023 |
+ u8 mask;
|
|
Kmods SIG |
d83023 |
+ size_t pos = bit & 7;
|
|
Kmods SIG |
d83023 |
+ const u8 *map = (u8 *)lmap + (bit >> 3);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (pos) {
|
|
Kmods SIG |
d83023 |
+ if (8 - pos >= nbits) {
|
|
Kmods SIG |
d83023 |
+ mask = fill_mask[pos + nbits] & zero_mask[pos];
|
|
Kmods SIG |
d83023 |
+ return !nbits || (*map & mask) == mask;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ mask = zero_mask[pos];
|
|
Kmods SIG |
d83023 |
+ if ((*map++ & mask) != mask)
|
|
Kmods SIG |
d83023 |
+ return false;
|
|
Kmods SIG |
d83023 |
+ nbits -= 8 - pos;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ pos = ((size_t)map) & (sizeof(size_t) - 1);
|
|
Kmods SIG |
d83023 |
+ if (pos) {
|
|
Kmods SIG |
d83023 |
+ pos = sizeof(size_t) - pos;
|
|
Kmods SIG |
d83023 |
+ if (nbits >= pos * 8) {
|
|
Kmods SIG |
d83023 |
+ for (nbits -= pos * 8; pos; pos--, map++) {
|
|
Kmods SIG |
d83023 |
+ if (*map != 0xFF)
|
|
Kmods SIG |
d83023 |
+ return false;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ for (pos = nbits / BITS_IN_SIZE_T; pos; pos--, map += sizeof(size_t)) {
|
|
Kmods SIG |
d83023 |
+ if (*((size_t *)map) != MINUS_ONE_T)
|
|
Kmods SIG |
d83023 |
+ return false;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ for (pos = (nbits % BITS_IN_SIZE_T) >> 3; pos; pos--, map++) {
|
|
Kmods SIG |
d83023 |
+ if (*map != 0xFF)
|
|
Kmods SIG |
d83023 |
+ return false;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ pos = nbits & 7;
|
|
Kmods SIG |
d83023 |
+ if (pos) {
|
|
Kmods SIG |
d83023 |
+ u8 mask = fill_mask[pos];
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if ((*map & mask) != mask)
|
|
Kmods SIG |
d83023 |
+ return false;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ // All bits are ones
|
|
Kmods SIG |
d83023 |
+ return true;
|
|
Kmods SIG |
d83023 |
+}
|
|
Kmods SIG |
d83023 |
diff --git a/src/bitmap.c b/src/bitmap.c
|
|
Kmods SIG |
d83023 |
new file mode 100644
|
|
Kmods SIG |
d83023 |
index 0000000000000000000000000000000000000000..32aab0031221db3ee67b09c239b960158370d238
|
|
Kmods SIG |
d83023 |
--- /dev/null
|
|
Kmods SIG |
d83023 |
+++ b/src/bitmap.c
|
|
Kmods SIG |
d83023 |
@@ -0,0 +1,1519 @@
|
|
Kmods SIG |
d83023 |
+// SPDX-License-Identifier: GPL-2.0
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ *
|
|
Kmods SIG |
d83023 |
+ * Copyright (C) 2019-2021 Paragon Software GmbH, All rights reserved.
|
|
Kmods SIG |
d83023 |
+ *
|
|
Kmods SIG |
d83023 |
+ * This code builds two trees of free clusters extents.
|
|
Kmods SIG |
d83023 |
+ * Trees are sorted by start of extent and by length of extent.
|
|
Kmods SIG |
d83023 |
+ * NTFS_MAX_WND_EXTENTS defines the maximum number of elements in trees.
|
|
Kmods SIG |
d83023 |
+ * In extreme case code reads on-disk bitmap to find free clusters
|
|
Kmods SIG |
d83023 |
+ *
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+#include <linux/blkdev.h>
|
|
Kmods SIG |
d83023 |
+#include <linux/buffer_head.h>
|
|
Kmods SIG |
d83023 |
+#include <linux/fs.h>
|
|
Kmods SIG |
d83023 |
+#include <linux/nls.h>
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+#include "debug.h"
|
|
Kmods SIG |
d83023 |
+#include "ntfs.h"
|
|
Kmods SIG |
d83023 |
+#include "ntfs_fs.h"
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ * Maximum number of extents in tree.
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+#define NTFS_MAX_WND_EXTENTS (32u * 1024u)
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+struct rb_node_key {
|
|
Kmods SIG |
d83023 |
+ struct rb_node node;
|
|
Kmods SIG |
d83023 |
+ size_t key;
|
|
Kmods SIG |
d83023 |
+};
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ * Tree is sorted by start (key)
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+struct e_node {
|
|
Kmods SIG |
d83023 |
+ struct rb_node_key start; /* Tree sorted by start */
|
|
Kmods SIG |
d83023 |
+ struct rb_node_key count; /* Tree sorted by len*/
|
|
Kmods SIG |
d83023 |
+};
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+static int wnd_rescan(struct wnd_bitmap *wnd);
|
|
Kmods SIG |
d83023 |
+static struct buffer_head *wnd_map(struct wnd_bitmap *wnd, size_t iw);
|
|
Kmods SIG |
d83023 |
+static bool wnd_is_free_hlp(struct wnd_bitmap *wnd, size_t bit, size_t bits);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+static struct kmem_cache *ntfs_enode_cachep;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+int __init ntfs3_init_bitmap(void)
|
|
Kmods SIG |
d83023 |
+{
|
|
Kmods SIG |
d83023 |
+ ntfs_enode_cachep =
|
|
Kmods SIG |
d83023 |
+ kmem_cache_create("ntfs3_enode_cache", sizeof(struct e_node), 0,
|
|
Kmods SIG |
d83023 |
+ SLAB_RECLAIM_ACCOUNT, NULL);
|
|
Kmods SIG |
d83023 |
+ return ntfs_enode_cachep ? 0 : -ENOMEM;
|
|
Kmods SIG |
d83023 |
+}
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+void ntfs3_exit_bitmap(void)
|
|
Kmods SIG |
d83023 |
+{
|
|
Kmods SIG |
d83023 |
+ kmem_cache_destroy(ntfs_enode_cachep);
|
|
Kmods SIG |
d83023 |
+}
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+static inline u32 wnd_bits(const struct wnd_bitmap *wnd, size_t i)
|
|
Kmods SIG |
d83023 |
+{
|
|
Kmods SIG |
d83023 |
+ return i + 1 == wnd->nwnd ? wnd->bits_last : wnd->sb->s_blocksize * 8;
|
|
Kmods SIG |
d83023 |
+}
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ * b_pos + b_len - biggest fragment
|
|
Kmods SIG |
d83023 |
+ * Scan range [wpos wbits) window 'buf'
|
|
Kmods SIG |
d83023 |
+ * Returns -1 if not found
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+static size_t wnd_scan(const ulong *buf, size_t wbit, u32 wpos, u32 wend,
|
|
Kmods SIG |
d83023 |
+ size_t to_alloc, size_t *prev_tail, size_t *b_pos,
|
|
Kmods SIG |
d83023 |
+ size_t *b_len)
|
|
Kmods SIG |
d83023 |
+{
|
|
Kmods SIG |
d83023 |
+ while (wpos < wend) {
|
|
Kmods SIG |
d83023 |
+ size_t free_len;
|
|
Kmods SIG |
d83023 |
+ u32 free_bits, end;
|
|
Kmods SIG |
d83023 |
+ u32 used = find_next_zero_bit(buf, wend, wpos);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (used >= wend) {
|
|
Kmods SIG |
d83023 |
+ if (*b_len < *prev_tail) {
|
|
Kmods SIG |
d83023 |
+ *b_pos = wbit - *prev_tail;
|
|
Kmods SIG |
d83023 |
+ *b_len = *prev_tail;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ *prev_tail = 0;
|
|
Kmods SIG |
d83023 |
+ return -1;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (used > wpos) {
|
|
Kmods SIG |
d83023 |
+ wpos = used;
|
|
Kmods SIG |
d83023 |
+ if (*b_len < *prev_tail) {
|
|
Kmods SIG |
d83023 |
+ *b_pos = wbit - *prev_tail;
|
|
Kmods SIG |
d83023 |
+ *b_len = *prev_tail;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ *prev_tail = 0;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /*
|
|
Kmods SIG |
d83023 |
+ * Now we have a fragment [wpos, wend) staring with 0
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+ end = wpos + to_alloc - *prev_tail;
|
|
Kmods SIG |
d83023 |
+ free_bits = find_next_bit(buf, min(end, wend), wpos);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ free_len = *prev_tail + free_bits - wpos;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (*b_len < free_len) {
|
|
Kmods SIG |
d83023 |
+ *b_pos = wbit + wpos - *prev_tail;
|
|
Kmods SIG |
d83023 |
+ *b_len = free_len;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (free_len >= to_alloc)
|
|
Kmods SIG |
d83023 |
+ return wbit + wpos - *prev_tail;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (free_bits >= wend) {
|
|
Kmods SIG |
d83023 |
+ *prev_tail += free_bits - wpos;
|
|
Kmods SIG |
d83023 |
+ return -1;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ wpos = free_bits + 1;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ *prev_tail = 0;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ return -1;
|
|
Kmods SIG |
d83023 |
+}
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ * wnd_close
|
|
Kmods SIG |
d83023 |
+ *
|
|
Kmods SIG |
d83023 |
+ * Frees all resources
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+void wnd_close(struct wnd_bitmap *wnd)
|
|
Kmods SIG |
d83023 |
+{
|
|
Kmods SIG |
d83023 |
+ struct rb_node *node, *next;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ ntfs_free(wnd->free_bits);
|
|
Kmods SIG |
d83023 |
+ run_close(&wnd->run);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ node = rb_first(&wnd->start_tree);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ while (node) {
|
|
Kmods SIG |
d83023 |
+ next = rb_next(node);
|
|
Kmods SIG |
d83023 |
+ rb_erase(node, &wnd->start_tree);
|
|
Kmods SIG |
d83023 |
+ kmem_cache_free(ntfs_enode_cachep,
|
|
Kmods SIG |
d83023 |
+ rb_entry(node, struct e_node, start.node));
|
|
Kmods SIG |
d83023 |
+ node = next;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+}
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+static struct rb_node *rb_lookup(struct rb_root *root, size_t v)
|
|
Kmods SIG |
d83023 |
+{
|
|
Kmods SIG |
d83023 |
+ struct rb_node **p = &root->rb_node;
|
|
Kmods SIG |
d83023 |
+ struct rb_node *r = NULL;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ while (*p) {
|
|
Kmods SIG |
d83023 |
+ struct rb_node_key *k;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ k = rb_entry(*p, struct rb_node_key, node);
|
|
Kmods SIG |
d83023 |
+ if (v < k->key) {
|
|
Kmods SIG |
d83023 |
+ p = &(*p)->rb_left;
|
|
Kmods SIG |
d83023 |
+ } else if (v > k->key) {
|
|
Kmods SIG |
d83023 |
+ r = &k->node;
|
|
Kmods SIG |
d83023 |
+ p = &(*p)->rb_right;
|
|
Kmods SIG |
d83023 |
+ } else {
|
|
Kmods SIG |
d83023 |
+ return &k->node;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ return r;
|
|
Kmods SIG |
d83023 |
+}
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ * rb_insert_count
|
|
Kmods SIG |
d83023 |
+ *
|
|
Kmods SIG |
d83023 |
+ * Helper function to insert special kind of 'count' tree
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+static inline bool rb_insert_count(struct rb_root *root, struct e_node *e)
|
|
Kmods SIG |
d83023 |
+{
|
|
Kmods SIG |
d83023 |
+ struct rb_node **p = &root->rb_node;
|
|
Kmods SIG |
d83023 |
+ struct rb_node *parent = NULL;
|
|
Kmods SIG |
d83023 |
+ size_t e_ckey = e->count.key;
|
|
Kmods SIG |
d83023 |
+ size_t e_skey = e->start.key;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ while (*p) {
|
|
Kmods SIG |
d83023 |
+ struct e_node *k =
|
|
Kmods SIG |
d83023 |
+ rb_entry(parent = *p, struct e_node, count.node);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (e_ckey > k->count.key) {
|
|
Kmods SIG |
d83023 |
+ p = &(*p)->rb_left;
|
|
Kmods SIG |
d83023 |
+ } else if (e_ckey < k->count.key) {
|
|
Kmods SIG |
d83023 |
+ p = &(*p)->rb_right;
|
|
Kmods SIG |
d83023 |
+ } else if (e_skey < k->start.key) {
|
|
Kmods SIG |
d83023 |
+ p = &(*p)->rb_left;
|
|
Kmods SIG |
d83023 |
+ } else if (e_skey > k->start.key) {
|
|
Kmods SIG |
d83023 |
+ p = &(*p)->rb_right;
|
|
Kmods SIG |
d83023 |
+ } else {
|
|
Kmods SIG |
d83023 |
+ WARN_ON(1);
|
|
Kmods SIG |
d83023 |
+ return false;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ rb_link_node(&e->count.node, parent, p);
|
|
Kmods SIG |
d83023 |
+ rb_insert_color(&e->count.node, root);
|
|
Kmods SIG |
d83023 |
+ return true;
|
|
Kmods SIG |
d83023 |
+}
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ * inline bool rb_insert_start
|
|
Kmods SIG |
d83023 |
+ *
|
|
Kmods SIG |
d83023 |
+ * Helper function to insert special kind of 'start' tree
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+static inline bool rb_insert_start(struct rb_root *root, struct e_node *e)
|
|
Kmods SIG |
d83023 |
+{
|
|
Kmods SIG |
d83023 |
+ struct rb_node **p = &root->rb_node;
|
|
Kmods SIG |
d83023 |
+ struct rb_node *parent = NULL;
|
|
Kmods SIG |
d83023 |
+ size_t e_skey = e->start.key;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ while (*p) {
|
|
Kmods SIG |
d83023 |
+ struct e_node *k;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ parent = *p;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ k = rb_entry(parent, struct e_node, start.node);
|
|
Kmods SIG |
d83023 |
+ if (e_skey < k->start.key) {
|
|
Kmods SIG |
d83023 |
+ p = &(*p)->rb_left;
|
|
Kmods SIG |
d83023 |
+ } else if (e_skey > k->start.key) {
|
|
Kmods SIG |
d83023 |
+ p = &(*p)->rb_right;
|
|
Kmods SIG |
d83023 |
+ } else {
|
|
Kmods SIG |
d83023 |
+ WARN_ON(1);
|
|
Kmods SIG |
d83023 |
+ return false;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ rb_link_node(&e->start.node, parent, p);
|
|
Kmods SIG |
d83023 |
+ rb_insert_color(&e->start.node, root);
|
|
Kmods SIG |
d83023 |
+ return true;
|
|
Kmods SIG |
d83023 |
+}
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ * wnd_add_free_ext
|
|
Kmods SIG |
d83023 |
+ *
|
|
Kmods SIG |
d83023 |
+ * adds a new extent of free space
|
|
Kmods SIG |
d83023 |
+ * build = 1 when building tree
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+static void wnd_add_free_ext(struct wnd_bitmap *wnd, size_t bit, size_t len,
|
|
Kmods SIG |
d83023 |
+ bool build)
|
|
Kmods SIG |
d83023 |
+{
|
|
Kmods SIG |
d83023 |
+ struct e_node *e, *e0 = NULL;
|
|
Kmods SIG |
d83023 |
+ size_t ib, end_in = bit + len;
|
|
Kmods SIG |
d83023 |
+ struct rb_node *n;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (build) {
|
|
Kmods SIG |
d83023 |
+ /* Use extent_min to filter too short extents */
|
|
Kmods SIG |
d83023 |
+ if (wnd->count >= NTFS_MAX_WND_EXTENTS &&
|
|
Kmods SIG |
d83023 |
+ len <= wnd->extent_min) {
|
|
Kmods SIG |
d83023 |
+ wnd->uptodated = -1;
|
|
Kmods SIG |
d83023 |
+ return;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ } else {
|
|
Kmods SIG |
d83023 |
+ /* Try to find extent before 'bit' */
|
|
Kmods SIG |
d83023 |
+ n = rb_lookup(&wnd->start_tree, bit);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (!n) {
|
|
Kmods SIG |
d83023 |
+ n = rb_first(&wnd->start_tree);
|
|
Kmods SIG |
d83023 |
+ } else {
|
|
Kmods SIG |
d83023 |
+ e = rb_entry(n, struct e_node, start.node);
|
|
Kmods SIG |
d83023 |
+ n = rb_next(n);
|
|
Kmods SIG |
d83023 |
+ if (e->start.key + e->count.key == bit) {
|
|
Kmods SIG |
d83023 |
+ /* Remove left */
|
|
Kmods SIG |
d83023 |
+ bit = e->start.key;
|
|
Kmods SIG |
d83023 |
+ len += e->count.key;
|
|
Kmods SIG |
d83023 |
+ rb_erase(&e->start.node, &wnd->start_tree);
|
|
Kmods SIG |
d83023 |
+ rb_erase(&e->count.node, &wnd->count_tree);
|
|
Kmods SIG |
d83023 |
+ wnd->count -= 1;
|
|
Kmods SIG |
d83023 |
+ e0 = e;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ while (n) {
|
|
Kmods SIG |
d83023 |
+ size_t next_end;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ e = rb_entry(n, struct e_node, start.node);
|
|
Kmods SIG |
d83023 |
+ next_end = e->start.key + e->count.key;
|
|
Kmods SIG |
d83023 |
+ if (e->start.key > end_in)
|
|
Kmods SIG |
d83023 |
+ break;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* Remove right */
|
|
Kmods SIG |
d83023 |
+ n = rb_next(n);
|
|
Kmods SIG |
d83023 |
+ len += next_end - end_in;
|
|
Kmods SIG |
d83023 |
+ end_in = next_end;
|
|
Kmods SIG |
d83023 |
+ rb_erase(&e->start.node, &wnd->start_tree);
|
|
Kmods SIG |
d83023 |
+ rb_erase(&e->count.node, &wnd->count_tree);
|
|
Kmods SIG |
d83023 |
+ wnd->count -= 1;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (!e0)
|
|
Kmods SIG |
d83023 |
+ e0 = e;
|
|
Kmods SIG |
d83023 |
+ else
|
|
Kmods SIG |
d83023 |
+ kmem_cache_free(ntfs_enode_cachep, e);
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (wnd->uptodated != 1) {
|
|
Kmods SIG |
d83023 |
+ /* Check bits before 'bit' */
|
|
Kmods SIG |
d83023 |
+ ib = wnd->zone_bit == wnd->zone_end ||
|
|
Kmods SIG |
d83023 |
+ bit < wnd->zone_end
|
|
Kmods SIG |
d83023 |
+ ? 0
|
|
Kmods SIG |
d83023 |
+ : wnd->zone_end;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ while (bit > ib && wnd_is_free_hlp(wnd, bit - 1, 1)) {
|
|
Kmods SIG |
d83023 |
+ bit -= 1;
|
|
Kmods SIG |
d83023 |
+ len += 1;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* Check bits after 'end_in' */
|
|
Kmods SIG |
d83023 |
+ ib = wnd->zone_bit == wnd->zone_end ||
|
|
Kmods SIG |
d83023 |
+ end_in > wnd->zone_bit
|
|
Kmods SIG |
d83023 |
+ ? wnd->nbits
|
|
Kmods SIG |
d83023 |
+ : wnd->zone_bit;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ while (end_in < ib && wnd_is_free_hlp(wnd, end_in, 1)) {
|
|
Kmods SIG |
d83023 |
+ end_in += 1;
|
|
Kmods SIG |
d83023 |
+ len += 1;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ /* Insert new fragment */
|
|
Kmods SIG |
d83023 |
+ if (wnd->count >= NTFS_MAX_WND_EXTENTS) {
|
|
Kmods SIG |
d83023 |
+ if (e0)
|
|
Kmods SIG |
d83023 |
+ kmem_cache_free(ntfs_enode_cachep, e0);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ wnd->uptodated = -1;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* Compare with smallest fragment */
|
|
Kmods SIG |
d83023 |
+ n = rb_last(&wnd->count_tree);
|
|
Kmods SIG |
d83023 |
+ e = rb_entry(n, struct e_node, count.node);
|
|
Kmods SIG |
d83023 |
+ if (len <= e->count.key)
|
|
Kmods SIG |
d83023 |
+ goto out; /* Do not insert small fragments */
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (build) {
|
|
Kmods SIG |
d83023 |
+ struct e_node *e2;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ n = rb_prev(n);
|
|
Kmods SIG |
d83023 |
+ e2 = rb_entry(n, struct e_node, count.node);
|
|
Kmods SIG |
d83023 |
+ /* smallest fragment will be 'e2->count.key' */
|
|
Kmods SIG |
d83023 |
+ wnd->extent_min = e2->count.key;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* Replace smallest fragment by new one */
|
|
Kmods SIG |
d83023 |
+ rb_erase(&e->start.node, &wnd->start_tree);
|
|
Kmods SIG |
d83023 |
+ rb_erase(&e->count.node, &wnd->count_tree);
|
|
Kmods SIG |
d83023 |
+ wnd->count -= 1;
|
|
Kmods SIG |
d83023 |
+ } else {
|
|
Kmods SIG |
d83023 |
+ e = e0 ? e0 : kmem_cache_alloc(ntfs_enode_cachep, GFP_ATOMIC);
|
|
Kmods SIG |
d83023 |
+ if (!e) {
|
|
Kmods SIG |
d83023 |
+ wnd->uptodated = -1;
|
|
Kmods SIG |
d83023 |
+ goto out;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (build && len <= wnd->extent_min)
|
|
Kmods SIG |
d83023 |
+ wnd->extent_min = len;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ e->start.key = bit;
|
|
Kmods SIG |
d83023 |
+ e->count.key = len;
|
|
Kmods SIG |
d83023 |
+ if (len > wnd->extent_max)
|
|
Kmods SIG |
d83023 |
+ wnd->extent_max = len;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ rb_insert_start(&wnd->start_tree, e);
|
|
Kmods SIG |
d83023 |
+ rb_insert_count(&wnd->count_tree, e);
|
|
Kmods SIG |
d83023 |
+ wnd->count += 1;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+out:;
|
|
Kmods SIG |
d83023 |
+}
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ * wnd_remove_free_ext
|
|
Kmods SIG |
d83023 |
+ *
|
|
Kmods SIG |
d83023 |
+ * removes a run from the cached free space
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+static void wnd_remove_free_ext(struct wnd_bitmap *wnd, size_t bit, size_t len)
|
|
Kmods SIG |
d83023 |
+{
|
|
Kmods SIG |
d83023 |
+ struct rb_node *n, *n3;
|
|
Kmods SIG |
d83023 |
+ struct e_node *e, *e3;
|
|
Kmods SIG |
d83023 |
+ size_t end_in = bit + len;
|
|
Kmods SIG |
d83023 |
+ size_t end3, end, new_key, new_len, max_new_len;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* Try to find extent before 'bit' */
|
|
Kmods SIG |
d83023 |
+ n = rb_lookup(&wnd->start_tree, bit);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (!n)
|
|
Kmods SIG |
d83023 |
+ return;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ e = rb_entry(n, struct e_node, start.node);
|
|
Kmods SIG |
d83023 |
+ end = e->start.key + e->count.key;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ new_key = new_len = 0;
|
|
Kmods SIG |
d83023 |
+ len = e->count.key;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* Range [bit,end_in) must be inside 'e' or outside 'e' and 'n' */
|
|
Kmods SIG |
d83023 |
+ if (e->start.key > bit)
|
|
Kmods SIG |
d83023 |
+ ;
|
|
Kmods SIG |
d83023 |
+ else if (end_in <= end) {
|
|
Kmods SIG |
d83023 |
+ /* Range [bit,end_in) inside 'e' */
|
|
Kmods SIG |
d83023 |
+ new_key = end_in;
|
|
Kmods SIG |
d83023 |
+ new_len = end - end_in;
|
|
Kmods SIG |
d83023 |
+ len = bit - e->start.key;
|
|
Kmods SIG |
d83023 |
+ } else if (bit > end) {
|
|
Kmods SIG |
d83023 |
+ bool bmax = false;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ n3 = rb_next(n);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ while (n3) {
|
|
Kmods SIG |
d83023 |
+ e3 = rb_entry(n3, struct e_node, start.node);
|
|
Kmods SIG |
d83023 |
+ if (e3->start.key >= end_in)
|
|
Kmods SIG |
d83023 |
+ break;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (e3->count.key == wnd->extent_max)
|
|
Kmods SIG |
d83023 |
+ bmax = true;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ end3 = e3->start.key + e3->count.key;
|
|
Kmods SIG |
d83023 |
+ if (end3 > end_in) {
|
|
Kmods SIG |
d83023 |
+ e3->start.key = end_in;
|
|
Kmods SIG |
d83023 |
+ rb_erase(&e3->count.node, &wnd->count_tree);
|
|
Kmods SIG |
d83023 |
+ e3->count.key = end3 - end_in;
|
|
Kmods SIG |
d83023 |
+ rb_insert_count(&wnd->count_tree, e3);
|
|
Kmods SIG |
d83023 |
+ break;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ n3 = rb_next(n3);
|
|
Kmods SIG |
d83023 |
+ rb_erase(&e3->start.node, &wnd->start_tree);
|
|
Kmods SIG |
d83023 |
+ rb_erase(&e3->count.node, &wnd->count_tree);
|
|
Kmods SIG |
d83023 |
+ wnd->count -= 1;
|
|
Kmods SIG |
d83023 |
+ kmem_cache_free(ntfs_enode_cachep, e3);
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ if (!bmax)
|
|
Kmods SIG |
d83023 |
+ return;
|
|
Kmods SIG |
d83023 |
+ n3 = rb_first(&wnd->count_tree);
|
|
Kmods SIG |
d83023 |
+ wnd->extent_max =
|
|
Kmods SIG |
d83023 |
+ n3 ? rb_entry(n3, struct e_node, count.node)->count.key
|
|
Kmods SIG |
d83023 |
+ : 0;
|
|
Kmods SIG |
d83023 |
+ return;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (e->count.key != wnd->extent_max) {
|
|
Kmods SIG |
d83023 |
+ ;
|
|
Kmods SIG |
d83023 |
+ } else if (rb_prev(&e->count.node)) {
|
|
Kmods SIG |
d83023 |
+ ;
|
|
Kmods SIG |
d83023 |
+ } else {
|
|
Kmods SIG |
d83023 |
+ n3 = rb_next(&e->count.node);
|
|
Kmods SIG |
d83023 |
+ max_new_len = len > new_len ? len : new_len;
|
|
Kmods SIG |
d83023 |
+ if (!n3) {
|
|
Kmods SIG |
d83023 |
+ wnd->extent_max = max_new_len;
|
|
Kmods SIG |
d83023 |
+ } else {
|
|
Kmods SIG |
d83023 |
+ e3 = rb_entry(n3, struct e_node, count.node);
|
|
Kmods SIG |
d83023 |
+ wnd->extent_max = max(e3->count.key, max_new_len);
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (!len) {
|
|
Kmods SIG |
d83023 |
+ if (new_len) {
|
|
Kmods SIG |
d83023 |
+ e->start.key = new_key;
|
|
Kmods SIG |
d83023 |
+ rb_erase(&e->count.node, &wnd->count_tree);
|
|
Kmods SIG |
d83023 |
+ e->count.key = new_len;
|
|
Kmods SIG |
d83023 |
+ rb_insert_count(&wnd->count_tree, e);
|
|
Kmods SIG |
d83023 |
+ } else {
|
|
Kmods SIG |
d83023 |
+ rb_erase(&e->start.node, &wnd->start_tree);
|
|
Kmods SIG |
d83023 |
+ rb_erase(&e->count.node, &wnd->count_tree);
|
|
Kmods SIG |
d83023 |
+ wnd->count -= 1;
|
|
Kmods SIG |
d83023 |
+ kmem_cache_free(ntfs_enode_cachep, e);
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ goto out;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ rb_erase(&e->count.node, &wnd->count_tree);
|
|
Kmods SIG |
d83023 |
+ e->count.key = len;
|
|
Kmods SIG |
d83023 |
+ rb_insert_count(&wnd->count_tree, e);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (!new_len)
|
|
Kmods SIG |
d83023 |
+ goto out;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (wnd->count >= NTFS_MAX_WND_EXTENTS) {
|
|
Kmods SIG |
d83023 |
+ wnd->uptodated = -1;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* Get minimal extent */
|
|
Kmods SIG |
d83023 |
+ e = rb_entry(rb_last(&wnd->count_tree), struct e_node,
|
|
Kmods SIG |
d83023 |
+ count.node);
|
|
Kmods SIG |
d83023 |
+ if (e->count.key > new_len)
|
|
Kmods SIG |
d83023 |
+ goto out;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* Replace minimum */
|
|
Kmods SIG |
d83023 |
+ rb_erase(&e->start.node, &wnd->start_tree);
|
|
Kmods SIG |
d83023 |
+ rb_erase(&e->count.node, &wnd->count_tree);
|
|
Kmods SIG |
d83023 |
+ wnd->count -= 1;
|
|
Kmods SIG |
d83023 |
+ } else {
|
|
Kmods SIG |
d83023 |
+ e = kmem_cache_alloc(ntfs_enode_cachep, GFP_ATOMIC);
|
|
Kmods SIG |
d83023 |
+ if (!e)
|
|
Kmods SIG |
d83023 |
+ wnd->uptodated = -1;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (e) {
|
|
Kmods SIG |
d83023 |
+ e->start.key = new_key;
|
|
Kmods SIG |
d83023 |
+ e->count.key = new_len;
|
|
Kmods SIG |
d83023 |
+ rb_insert_start(&wnd->start_tree, e);
|
|
Kmods SIG |
d83023 |
+ rb_insert_count(&wnd->count_tree, e);
|
|
Kmods SIG |
d83023 |
+ wnd->count += 1;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+out:
|
|
Kmods SIG |
d83023 |
+ if (!wnd->count && 1 != wnd->uptodated)
|
|
Kmods SIG |
d83023 |
+ wnd_rescan(wnd);
|
|
Kmods SIG |
d83023 |
+}
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ * wnd_rescan
|
|
Kmods SIG |
d83023 |
+ *
|
|
Kmods SIG |
d83023 |
+ * Scan all bitmap. used while initialization.
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+static int wnd_rescan(struct wnd_bitmap *wnd)
|
|
Kmods SIG |
d83023 |
+{
|
|
Kmods SIG |
d83023 |
+ int err = 0;
|
|
Kmods SIG |
d83023 |
+ size_t prev_tail = 0;
|
|
Kmods SIG |
d83023 |
+ struct super_block *sb = wnd->sb;
|
|
Kmods SIG |
d83023 |
+ struct ntfs_sb_info *sbi = sb->s_fs_info;
|
|
Kmods SIG |
d83023 |
+ u64 lbo, len = 0;
|
|
Kmods SIG |
d83023 |
+ u32 blocksize = sb->s_blocksize;
|
|
Kmods SIG |
d83023 |
+ u8 cluster_bits = sbi->cluster_bits;
|
|
Kmods SIG |
d83023 |
+ u32 wbits = 8 * sb->s_blocksize;
|
|
Kmods SIG |
d83023 |
+ u32 used, frb;
|
|
Kmods SIG |
d83023 |
+ const ulong *buf;
|
|
Kmods SIG |
d83023 |
+ size_t wpos, wbit, iw, vbo;
|
|
Kmods SIG |
d83023 |
+ struct buffer_head *bh = NULL;
|
|
Kmods SIG |
d83023 |
+ CLST lcn, clen;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ wnd->uptodated = 0;
|
|
Kmods SIG |
d83023 |
+ wnd->extent_max = 0;
|
|
Kmods SIG |
d83023 |
+ wnd->extent_min = MINUS_ONE_T;
|
|
Kmods SIG |
d83023 |
+ wnd->total_zeroes = 0;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ vbo = 0;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ for (iw = 0; iw < wnd->nwnd; iw++) {
|
|
Kmods SIG |
d83023 |
+ if (iw + 1 == wnd->nwnd)
|
|
Kmods SIG |
d83023 |
+ wbits = wnd->bits_last;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (wnd->inited) {
|
|
Kmods SIG |
d83023 |
+ if (!wnd->free_bits[iw]) {
|
|
Kmods SIG |
d83023 |
+ /* all ones */
|
|
Kmods SIG |
d83023 |
+ if (prev_tail) {
|
|
Kmods SIG |
d83023 |
+ wnd_add_free_ext(wnd,
|
|
Kmods SIG |
d83023 |
+ vbo * 8 - prev_tail,
|
|
Kmods SIG |
d83023 |
+ prev_tail, true);
|
|
Kmods SIG |
d83023 |
+ prev_tail = 0;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ goto next_wnd;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ if (wbits == wnd->free_bits[iw]) {
|
|
Kmods SIG |
d83023 |
+ /* all zeroes */
|
|
Kmods SIG |
d83023 |
+ prev_tail += wbits;
|
|
Kmods SIG |
d83023 |
+ wnd->total_zeroes += wbits;
|
|
Kmods SIG |
d83023 |
+ goto next_wnd;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (!len) {
|
|
Kmods SIG |
d83023 |
+ u32 off = vbo & sbi->cluster_mask;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (!run_lookup_entry(&wnd->run, vbo >> cluster_bits,
|
|
Kmods SIG |
d83023 |
+ &lcn, &clen, NULL)) {
|
|
Kmods SIG |
d83023 |
+ err = -ENOENT;
|
|
Kmods SIG |
d83023 |
+ goto out;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ lbo = ((u64)lcn << cluster_bits) + off;
|
|
Kmods SIG |
d83023 |
+ len = ((u64)clen << cluster_bits) - off;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ bh = ntfs_bread(sb, lbo >> sb->s_blocksize_bits);
|
|
Kmods SIG |
d83023 |
+ if (!bh) {
|
|
Kmods SIG |
d83023 |
+ err = -EIO;
|
|
Kmods SIG |
d83023 |
+ goto out;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ buf = (ulong *)bh->b_data;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ used = __bitmap_weight(buf, wbits);
|
|
Kmods SIG |
d83023 |
+ if (used < wbits) {
|
|
Kmods SIG |
d83023 |
+ frb = wbits - used;
|
|
Kmods SIG |
d83023 |
+ wnd->free_bits[iw] = frb;
|
|
Kmods SIG |
d83023 |
+ wnd->total_zeroes += frb;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ wpos = 0;
|
|
Kmods SIG |
d83023 |
+ wbit = vbo * 8;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (wbit + wbits > wnd->nbits)
|
|
Kmods SIG |
d83023 |
+ wbits = wnd->nbits - wbit;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ do {
|
|
Kmods SIG |
d83023 |
+ used = find_next_zero_bit(buf, wbits, wpos);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (used > wpos && prev_tail) {
|
|
Kmods SIG |
d83023 |
+ wnd_add_free_ext(wnd, wbit + wpos - prev_tail,
|
|
Kmods SIG |
d83023 |
+ prev_tail, true);
|
|
Kmods SIG |
d83023 |
+ prev_tail = 0;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ wpos = used;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (wpos >= wbits) {
|
|
Kmods SIG |
d83023 |
+ /* No free blocks */
|
|
Kmods SIG |
d83023 |
+ prev_tail = 0;
|
|
Kmods SIG |
d83023 |
+ break;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ frb = find_next_bit(buf, wbits, wpos);
|
|
Kmods SIG |
d83023 |
+ if (frb >= wbits) {
|
|
Kmods SIG |
d83023 |
+ /* keep last free block */
|
|
Kmods SIG |
d83023 |
+ prev_tail += frb - wpos;
|
|
Kmods SIG |
d83023 |
+ break;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ wnd_add_free_ext(wnd, wbit + wpos - prev_tail,
|
|
Kmods SIG |
d83023 |
+ frb + prev_tail - wpos, true);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* Skip free block and first '1' */
|
|
Kmods SIG |
d83023 |
+ wpos = frb + 1;
|
|
Kmods SIG |
d83023 |
+ /* Reset previous tail */
|
|
Kmods SIG |
d83023 |
+ prev_tail = 0;
|
|
Kmods SIG |
d83023 |
+ } while (wpos < wbits);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+next_wnd:
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (bh)
|
|
Kmods SIG |
d83023 |
+ put_bh(bh);
|
|
Kmods SIG |
d83023 |
+ bh = NULL;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ vbo += blocksize;
|
|
Kmods SIG |
d83023 |
+ if (len) {
|
|
Kmods SIG |
d83023 |
+ len -= blocksize;
|
|
Kmods SIG |
d83023 |
+ lbo += blocksize;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* Add last block */
|
|
Kmods SIG |
d83023 |
+ if (prev_tail)
|
|
Kmods SIG |
d83023 |
+ wnd_add_free_ext(wnd, wnd->nbits - prev_tail, prev_tail, true);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /*
|
|
Kmods SIG |
d83023 |
+ * Before init cycle wnd->uptodated was 0
|
|
Kmods SIG |
d83023 |
+ * If any errors or limits occurs while initialization then
|
|
Kmods SIG |
d83023 |
+ * wnd->uptodated will be -1
|
|
Kmods SIG |
d83023 |
+ * If 'uptodated' is still 0 then Tree is really updated
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+ if (!wnd->uptodated)
|
|
Kmods SIG |
d83023 |
+ wnd->uptodated = 1;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (wnd->zone_bit != wnd->zone_end) {
|
|
Kmods SIG |
d83023 |
+ size_t zlen = wnd->zone_end - wnd->zone_bit;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ wnd->zone_end = wnd->zone_bit;
|
|
Kmods SIG |
d83023 |
+ wnd_zone_set(wnd, wnd->zone_bit, zlen);
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+out:
|
|
Kmods SIG |
d83023 |
+ return err;
|
|
Kmods SIG |
d83023 |
+}
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ * wnd_init
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+int wnd_init(struct wnd_bitmap *wnd, struct super_block *sb, size_t nbits)
|
|
Kmods SIG |
d83023 |
+{
|
|
Kmods SIG |
d83023 |
+ int err;
|
|
Kmods SIG |
d83023 |
+ u32 blocksize = sb->s_blocksize;
|
|
Kmods SIG |
d83023 |
+ u32 wbits = blocksize * 8;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ init_rwsem(&wnd->rw_lock);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ wnd->sb = sb;
|
|
Kmods SIG |
d83023 |
+ wnd->nbits = nbits;
|
|
Kmods SIG |
d83023 |
+ wnd->total_zeroes = nbits;
|
|
Kmods SIG |
d83023 |
+ wnd->extent_max = MINUS_ONE_T;
|
|
Kmods SIG |
d83023 |
+ wnd->zone_bit = wnd->zone_end = 0;
|
|
Kmods SIG |
d83023 |
+ wnd->nwnd = bytes_to_block(sb, bitmap_size(nbits));
|
|
Kmods SIG |
d83023 |
+ wnd->bits_last = nbits & (wbits - 1);
|
|
Kmods SIG |
d83023 |
+ if (!wnd->bits_last)
|
|
Kmods SIG |
d83023 |
+ wnd->bits_last = wbits;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ wnd->free_bits = ntfs_zalloc(wnd->nwnd * sizeof(u16));
|
|
Kmods SIG |
d83023 |
+ if (!wnd->free_bits)
|
|
Kmods SIG |
d83023 |
+ return -ENOMEM;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ err = wnd_rescan(wnd);
|
|
Kmods SIG |
d83023 |
+ if (err)
|
|
Kmods SIG |
d83023 |
+ return err;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ wnd->inited = true;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ return 0;
|
|
Kmods SIG |
d83023 |
+}
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ * wnd_map
|
|
Kmods SIG |
d83023 |
+ *
|
|
Kmods SIG |
d83023 |
+ * call sb_bread for requested window
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+static struct buffer_head *wnd_map(struct wnd_bitmap *wnd, size_t iw)
|
|
Kmods SIG |
d83023 |
+{
|
|
Kmods SIG |
d83023 |
+ size_t vbo;
|
|
Kmods SIG |
d83023 |
+ CLST lcn, clen;
|
|
Kmods SIG |
d83023 |
+ struct super_block *sb = wnd->sb;
|
|
Kmods SIG |
d83023 |
+ struct ntfs_sb_info *sbi;
|
|
Kmods SIG |
d83023 |
+ struct buffer_head *bh;
|
|
Kmods SIG |
d83023 |
+ u64 lbo;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ sbi = sb->s_fs_info;
|
|
Kmods SIG |
d83023 |
+ vbo = (u64)iw << sb->s_blocksize_bits;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (!run_lookup_entry(&wnd->run, vbo >> sbi->cluster_bits, &lcn, &clen,
|
|
Kmods SIG |
d83023 |
+ NULL)) {
|
|
Kmods SIG |
d83023 |
+ return ERR_PTR(-ENOENT);
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ lbo = ((u64)lcn << sbi->cluster_bits) + (vbo & sbi->cluster_mask);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ bh = ntfs_bread(wnd->sb, lbo >> sb->s_blocksize_bits);
|
|
Kmods SIG |
d83023 |
+ if (!bh)
|
|
Kmods SIG |
d83023 |
+ return ERR_PTR(-EIO);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ return bh;
|
|
Kmods SIG |
d83023 |
+}
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ * wnd_set_free
|
|
Kmods SIG |
d83023 |
+ *
|
|
Kmods SIG |
d83023 |
+ * Marks the bits range from bit to bit + bits as free
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+int wnd_set_free(struct wnd_bitmap *wnd, size_t bit, size_t bits)
|
|
Kmods SIG |
d83023 |
+{
|
|
Kmods SIG |
d83023 |
+ int err = 0;
|
|
Kmods SIG |
d83023 |
+ struct super_block *sb = wnd->sb;
|
|
Kmods SIG |
d83023 |
+ size_t bits0 = bits;
|
|
Kmods SIG |
d83023 |
+ u32 wbits = 8 * sb->s_blocksize;
|
|
Kmods SIG |
d83023 |
+ size_t iw = bit >> (sb->s_blocksize_bits + 3);
|
|
Kmods SIG |
d83023 |
+ u32 wbit = bit & (wbits - 1);
|
|
Kmods SIG |
d83023 |
+ struct buffer_head *bh;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ while (iw < wnd->nwnd && bits) {
|
|
Kmods SIG |
d83023 |
+ u32 tail, op;
|
|
Kmods SIG |
d83023 |
+ ulong *buf;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (iw + 1 == wnd->nwnd)
|
|
Kmods SIG |
d83023 |
+ wbits = wnd->bits_last;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ tail = wbits - wbit;
|
|
Kmods SIG |
d83023 |
+ op = tail < bits ? tail : bits;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ bh = wnd_map(wnd, iw);
|
|
Kmods SIG |
d83023 |
+ if (IS_ERR(bh)) {
|
|
Kmods SIG |
d83023 |
+ err = PTR_ERR(bh);
|
|
Kmods SIG |
d83023 |
+ break;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ buf = (ulong *)bh->b_data;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ lock_buffer(bh);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ __bitmap_clear(buf, wbit, op);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ wnd->free_bits[iw] += op;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ set_buffer_uptodate(bh);
|
|
Kmods SIG |
d83023 |
+ mark_buffer_dirty(bh);
|
|
Kmods SIG |
d83023 |
+ unlock_buffer(bh);
|
|
Kmods SIG |
d83023 |
+ put_bh(bh);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ wnd->total_zeroes += op;
|
|
Kmods SIG |
d83023 |
+ bits -= op;
|
|
Kmods SIG |
d83023 |
+ wbit = 0;
|
|
Kmods SIG |
d83023 |
+ iw += 1;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ wnd_add_free_ext(wnd, bit, bits0, false);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ return err;
|
|
Kmods SIG |
d83023 |
+}
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ * wnd_set_used
|
|
Kmods SIG |
d83023 |
+ *
|
|
Kmods SIG |
d83023 |
+ * Marks the bits range from bit to bit + bits as used
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+int wnd_set_used(struct wnd_bitmap *wnd, size_t bit, size_t bits)
|
|
Kmods SIG |
d83023 |
+{
|
|
Kmods SIG |
d83023 |
+ int err = 0;
|
|
Kmods SIG |
d83023 |
+ struct super_block *sb = wnd->sb;
|
|
Kmods SIG |
d83023 |
+ size_t bits0 = bits;
|
|
Kmods SIG |
d83023 |
+ size_t iw = bit >> (sb->s_blocksize_bits + 3);
|
|
Kmods SIG |
d83023 |
+ u32 wbits = 8 * sb->s_blocksize;
|
|
Kmods SIG |
d83023 |
+ u32 wbit = bit & (wbits - 1);
|
|
Kmods SIG |
d83023 |
+ struct buffer_head *bh;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ while (iw < wnd->nwnd && bits) {
|
|
Kmods SIG |
d83023 |
+ u32 tail, op;
|
|
Kmods SIG |
d83023 |
+ ulong *buf;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (unlikely(iw + 1 == wnd->nwnd))
|
|
Kmods SIG |
d83023 |
+ wbits = wnd->bits_last;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ tail = wbits - wbit;
|
|
Kmods SIG |
d83023 |
+ op = tail < bits ? tail : bits;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ bh = wnd_map(wnd, iw);
|
|
Kmods SIG |
d83023 |
+ if (IS_ERR(bh)) {
|
|
Kmods SIG |
d83023 |
+ err = PTR_ERR(bh);
|
|
Kmods SIG |
d83023 |
+ break;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ buf = (ulong *)bh->b_data;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ lock_buffer(bh);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ __bitmap_set(buf, wbit, op);
|
|
Kmods SIG |
d83023 |
+ wnd->free_bits[iw] -= op;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ set_buffer_uptodate(bh);
|
|
Kmods SIG |
d83023 |
+ mark_buffer_dirty(bh);
|
|
Kmods SIG |
d83023 |
+ unlock_buffer(bh);
|
|
Kmods SIG |
d83023 |
+ put_bh(bh);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ wnd->total_zeroes -= op;
|
|
Kmods SIG |
d83023 |
+ bits -= op;
|
|
Kmods SIG |
d83023 |
+ wbit = 0;
|
|
Kmods SIG |
d83023 |
+ iw += 1;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (!RB_EMPTY_ROOT(&wnd->start_tree))
|
|
Kmods SIG |
d83023 |
+ wnd_remove_free_ext(wnd, bit, bits0);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ return err;
|
|
Kmods SIG |
d83023 |
+}
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ * wnd_is_free_hlp
|
|
Kmods SIG |
d83023 |
+ *
|
|
Kmods SIG |
d83023 |
+ * Returns true if all clusters [bit, bit+bits) are free (bitmap only)
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+static bool wnd_is_free_hlp(struct wnd_bitmap *wnd, size_t bit, size_t bits)
|
|
Kmods SIG |
d83023 |
+{
|
|
Kmods SIG |
d83023 |
+ struct super_block *sb = wnd->sb;
|
|
Kmods SIG |
d83023 |
+ size_t iw = bit >> (sb->s_blocksize_bits + 3);
|
|
Kmods SIG |
d83023 |
+ u32 wbits = 8 * sb->s_blocksize;
|
|
Kmods SIG |
d83023 |
+ u32 wbit = bit & (wbits - 1);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ while (iw < wnd->nwnd && bits) {
|
|
Kmods SIG |
d83023 |
+ u32 tail, op;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (unlikely(iw + 1 == wnd->nwnd))
|
|
Kmods SIG |
d83023 |
+ wbits = wnd->bits_last;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ tail = wbits - wbit;
|
|
Kmods SIG |
d83023 |
+ op = tail < bits ? tail : bits;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (wbits != wnd->free_bits[iw]) {
|
|
Kmods SIG |
d83023 |
+ bool ret;
|
|
Kmods SIG |
d83023 |
+ struct buffer_head *bh = wnd_map(wnd, iw);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (IS_ERR(bh))
|
|
Kmods SIG |
d83023 |
+ return false;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ ret = are_bits_clear((ulong *)bh->b_data, wbit, op);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ put_bh(bh);
|
|
Kmods SIG |
d83023 |
+ if (!ret)
|
|
Kmods SIG |
d83023 |
+ return false;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ bits -= op;
|
|
Kmods SIG |
d83023 |
+ wbit = 0;
|
|
Kmods SIG |
d83023 |
+ iw += 1;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ return true;
|
|
Kmods SIG |
d83023 |
+}
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ * wnd_is_free
|
|
Kmods SIG |
d83023 |
+ *
|
|
Kmods SIG |
d83023 |
+ * Returns true if all clusters [bit, bit+bits) are free
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+bool wnd_is_free(struct wnd_bitmap *wnd, size_t bit, size_t bits)
|
|
Kmods SIG |
d83023 |
+{
|
|
Kmods SIG |
d83023 |
+ bool ret;
|
|
Kmods SIG |
d83023 |
+ struct rb_node *n;
|
|
Kmods SIG |
d83023 |
+ size_t end;
|
|
Kmods SIG |
d83023 |
+ struct e_node *e;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (RB_EMPTY_ROOT(&wnd->start_tree))
|
|
Kmods SIG |
d83023 |
+ goto use_wnd;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ n = rb_lookup(&wnd->start_tree, bit);
|
|
Kmods SIG |
d83023 |
+ if (!n)
|
|
Kmods SIG |
d83023 |
+ goto use_wnd;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ e = rb_entry(n, struct e_node, start.node);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ end = e->start.key + e->count.key;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (bit < end && bit + bits <= end)
|
|
Kmods SIG |
d83023 |
+ return true;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+use_wnd:
|
|
Kmods SIG |
d83023 |
+ ret = wnd_is_free_hlp(wnd, bit, bits);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ return ret;
|
|
Kmods SIG |
d83023 |
+}
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ * wnd_is_used
|
|
Kmods SIG |
d83023 |
+ *
|
|
Kmods SIG |
d83023 |
+ * Returns true if all clusters [bit, bit+bits) are used
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+bool wnd_is_used(struct wnd_bitmap *wnd, size_t bit, size_t bits)
|
|
Kmods SIG |
d83023 |
+{
|
|
Kmods SIG |
d83023 |
+ bool ret = false;
|
|
Kmods SIG |
d83023 |
+ struct super_block *sb = wnd->sb;
|
|
Kmods SIG |
d83023 |
+ size_t iw = bit >> (sb->s_blocksize_bits + 3);
|
|
Kmods SIG |
d83023 |
+ u32 wbits = 8 * sb->s_blocksize;
|
|
Kmods SIG |
d83023 |
+ u32 wbit = bit & (wbits - 1);
|
|
Kmods SIG |
d83023 |
+ size_t end;
|
|
Kmods SIG |
d83023 |
+ struct rb_node *n;
|
|
Kmods SIG |
d83023 |
+ struct e_node *e;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (RB_EMPTY_ROOT(&wnd->start_tree))
|
|
Kmods SIG |
d83023 |
+ goto use_wnd;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ end = bit + bits;
|
|
Kmods SIG |
d83023 |
+ n = rb_lookup(&wnd->start_tree, end - 1);
|
|
Kmods SIG |
d83023 |
+ if (!n)
|
|
Kmods SIG |
d83023 |
+ goto use_wnd;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ e = rb_entry(n, struct e_node, start.node);
|
|
Kmods SIG |
d83023 |
+ if (e->start.key + e->count.key > bit)
|
|
Kmods SIG |
d83023 |
+ return false;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+use_wnd:
|
|
Kmods SIG |
d83023 |
+ while (iw < wnd->nwnd && bits) {
|
|
Kmods SIG |
d83023 |
+ u32 tail, op;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (unlikely(iw + 1 == wnd->nwnd))
|
|
Kmods SIG |
d83023 |
+ wbits = wnd->bits_last;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ tail = wbits - wbit;
|
|
Kmods SIG |
d83023 |
+ op = tail < bits ? tail : bits;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (wnd->free_bits[iw]) {
|
|
Kmods SIG |
d83023 |
+ bool ret;
|
|
Kmods SIG |
d83023 |
+ struct buffer_head *bh = wnd_map(wnd, iw);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (IS_ERR(bh))
|
|
Kmods SIG |
d83023 |
+ goto out;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ ret = are_bits_set((ulong *)bh->b_data, wbit, op);
|
|
Kmods SIG |
d83023 |
+ put_bh(bh);
|
|
Kmods SIG |
d83023 |
+ if (!ret)
|
|
Kmods SIG |
d83023 |
+ goto out;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ bits -= op;
|
|
Kmods SIG |
d83023 |
+ wbit = 0;
|
|
Kmods SIG |
d83023 |
+ iw += 1;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ ret = true;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+out:
|
|
Kmods SIG |
d83023 |
+ return ret;
|
|
Kmods SIG |
d83023 |
+}
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ * wnd_find
|
|
Kmods SIG |
d83023 |
+ * - flags - BITMAP_FIND_XXX flags
|
|
Kmods SIG |
d83023 |
+ *
|
|
Kmods SIG |
d83023 |
+ * looks for free space
|
|
Kmods SIG |
d83023 |
+ * Returns 0 if not found
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+size_t wnd_find(struct wnd_bitmap *wnd, size_t to_alloc, size_t hint,
|
|
Kmods SIG |
d83023 |
+ size_t flags, size_t *allocated)
|
|
Kmods SIG |
d83023 |
+{
|
|
Kmods SIG |
d83023 |
+ struct super_block *sb;
|
|
Kmods SIG |
d83023 |
+ u32 wbits, wpos, wzbit, wzend;
|
|
Kmods SIG |
d83023 |
+ size_t fnd, max_alloc, b_len, b_pos;
|
|
Kmods SIG |
d83023 |
+ size_t iw, prev_tail, nwnd, wbit, ebit, zbit, zend;
|
|
Kmods SIG |
d83023 |
+ size_t to_alloc0 = to_alloc;
|
|
Kmods SIG |
d83023 |
+ const ulong *buf;
|
|
Kmods SIG |
d83023 |
+ const struct e_node *e;
|
|
Kmods SIG |
d83023 |
+ const struct rb_node *pr, *cr;
|
|
Kmods SIG |
d83023 |
+ u8 log2_bits;
|
|
Kmods SIG |
d83023 |
+ bool fbits_valid;
|
|
Kmods SIG |
d83023 |
+ struct buffer_head *bh;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* fast checking for available free space */
|
|
Kmods SIG |
d83023 |
+ if (flags & BITMAP_FIND_FULL) {
|
|
Kmods SIG |
d83023 |
+ size_t zeroes = wnd_zeroes(wnd);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ zeroes -= wnd->zone_end - wnd->zone_bit;
|
|
Kmods SIG |
d83023 |
+ if (zeroes < to_alloc0)
|
|
Kmods SIG |
d83023 |
+ goto no_space;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (to_alloc0 > wnd->extent_max)
|
|
Kmods SIG |
d83023 |
+ goto no_space;
|
|
Kmods SIG |
d83023 |
+ } else {
|
|
Kmods SIG |
d83023 |
+ if (to_alloc > wnd->extent_max)
|
|
Kmods SIG |
d83023 |
+ to_alloc = wnd->extent_max;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (wnd->zone_bit <= hint && hint < wnd->zone_end)
|
|
Kmods SIG |
d83023 |
+ hint = wnd->zone_end;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ max_alloc = wnd->nbits;
|
|
Kmods SIG |
d83023 |
+ b_len = b_pos = 0;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (hint >= max_alloc)
|
|
Kmods SIG |
d83023 |
+ hint = 0;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (RB_EMPTY_ROOT(&wnd->start_tree)) {
|
|
Kmods SIG |
d83023 |
+ if (wnd->uptodated == 1) {
|
|
Kmods SIG |
d83023 |
+ /* extents tree is updated -> no free space */
|
|
Kmods SIG |
d83023 |
+ goto no_space;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ goto scan_bitmap;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ e = NULL;
|
|
Kmods SIG |
d83023 |
+ if (!hint)
|
|
Kmods SIG |
d83023 |
+ goto allocate_biggest;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* Use hint: enumerate extents by start >= hint */
|
|
Kmods SIG |
d83023 |
+ pr = NULL;
|
|
Kmods SIG |
d83023 |
+ cr = wnd->start_tree.rb_node;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ for (;;) {
|
|
Kmods SIG |
d83023 |
+ e = rb_entry(cr, struct e_node, start.node);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (e->start.key == hint)
|
|
Kmods SIG |
d83023 |
+ break;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (e->start.key < hint) {
|
|
Kmods SIG |
d83023 |
+ pr = cr;
|
|
Kmods SIG |
d83023 |
+ cr = cr->rb_right;
|
|
Kmods SIG |
d83023 |
+ if (!cr)
|
|
Kmods SIG |
d83023 |
+ break;
|
|
Kmods SIG |
d83023 |
+ continue;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ cr = cr->rb_left;
|
|
Kmods SIG |
d83023 |
+ if (!cr) {
|
|
Kmods SIG |
d83023 |
+ e = pr ? rb_entry(pr, struct e_node, start.node) : NULL;
|
|
Kmods SIG |
d83023 |
+ break;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (!e)
|
|
Kmods SIG |
d83023 |
+ goto allocate_biggest;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (e->start.key + e->count.key > hint) {
|
|
Kmods SIG |
d83023 |
+ /* We have found extension with 'hint' inside */
|
|
Kmods SIG |
d83023 |
+ size_t len = e->start.key + e->count.key - hint;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (len >= to_alloc && hint + to_alloc <= max_alloc) {
|
|
Kmods SIG |
d83023 |
+ fnd = hint;
|
|
Kmods SIG |
d83023 |
+ goto found;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (!(flags & BITMAP_FIND_FULL)) {
|
|
Kmods SIG |
d83023 |
+ if (len > to_alloc)
|
|
Kmods SIG |
d83023 |
+ len = to_alloc;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (hint + len <= max_alloc) {
|
|
Kmods SIG |
d83023 |
+ fnd = hint;
|
|
Kmods SIG |
d83023 |
+ to_alloc = len;
|
|
Kmods SIG |
d83023 |
+ goto found;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+allocate_biggest:
|
|
Kmods SIG |
d83023 |
+ /* Allocate from biggest free extent */
|
|
Kmods SIG |
d83023 |
+ e = rb_entry(rb_first(&wnd->count_tree), struct e_node, count.node);
|
|
Kmods SIG |
d83023 |
+ if (e->count.key != wnd->extent_max)
|
|
Kmods SIG |
d83023 |
+ wnd->extent_max = e->count.key;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (e->count.key < max_alloc) {
|
|
Kmods SIG |
d83023 |
+ if (e->count.key >= to_alloc) {
|
|
Kmods SIG |
d83023 |
+ ;
|
|
Kmods SIG |
d83023 |
+ } else if (flags & BITMAP_FIND_FULL) {
|
|
Kmods SIG |
d83023 |
+ if (e->count.key < to_alloc0) {
|
|
Kmods SIG |
d83023 |
+ /* Biggest free block is less then requested */
|
|
Kmods SIG |
d83023 |
+ goto no_space;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ to_alloc = e->count.key;
|
|
Kmods SIG |
d83023 |
+ } else if (-1 != wnd->uptodated) {
|
|
Kmods SIG |
d83023 |
+ to_alloc = e->count.key;
|
|
Kmods SIG |
d83023 |
+ } else {
|
|
Kmods SIG |
d83023 |
+ /* Check if we can use more bits */
|
|
Kmods SIG |
d83023 |
+ size_t op, max_check;
|
|
Kmods SIG |
d83023 |
+ struct rb_root start_tree;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ memcpy(&start_tree, &wnd->start_tree,
|
|
Kmods SIG |
d83023 |
+ sizeof(struct rb_root));
|
|
Kmods SIG |
d83023 |
+ memset(&wnd->start_tree, 0, sizeof(struct rb_root));
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ max_check = e->start.key + to_alloc;
|
|
Kmods SIG |
d83023 |
+ if (max_check > max_alloc)
|
|
Kmods SIG |
d83023 |
+ max_check = max_alloc;
|
|
Kmods SIG |
d83023 |
+ for (op = e->start.key + e->count.key; op < max_check;
|
|
Kmods SIG |
d83023 |
+ op++) {
|
|
Kmods SIG |
d83023 |
+ if (!wnd_is_free(wnd, op, 1))
|
|
Kmods SIG |
d83023 |
+ break;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ memcpy(&wnd->start_tree, &start_tree,
|
|
Kmods SIG |
d83023 |
+ sizeof(struct rb_root));
|
|
Kmods SIG |
d83023 |
+ to_alloc = op - e->start.key;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* Prepare to return */
|
|
Kmods SIG |
d83023 |
+ fnd = e->start.key;
|
|
Kmods SIG |
d83023 |
+ if (e->start.key + to_alloc > max_alloc)
|
|
Kmods SIG |
d83023 |
+ to_alloc = max_alloc - e->start.key;
|
|
Kmods SIG |
d83023 |
+ goto found;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (wnd->uptodated == 1) {
|
|
Kmods SIG |
d83023 |
+ /* extents tree is updated -> no free space */
|
|
Kmods SIG |
d83023 |
+ goto no_space;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ b_len = e->count.key;
|
|
Kmods SIG |
d83023 |
+ b_pos = e->start.key;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+scan_bitmap:
|
|
Kmods SIG |
d83023 |
+ sb = wnd->sb;
|
|
Kmods SIG |
d83023 |
+ log2_bits = sb->s_blocksize_bits + 3;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* At most two ranges [hint, max_alloc) + [0, hint) */
|
|
Kmods SIG |
d83023 |
+Again:
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* TODO: optimize request for case nbits > wbits */
|
|
Kmods SIG |
d83023 |
+ iw = hint >> log2_bits;
|
|
Kmods SIG |
d83023 |
+ wbits = sb->s_blocksize * 8;
|
|
Kmods SIG |
d83023 |
+ wpos = hint & (wbits - 1);
|
|
Kmods SIG |
d83023 |
+ prev_tail = 0;
|
|
Kmods SIG |
d83023 |
+ fbits_valid = true;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (max_alloc == wnd->nbits) {
|
|
Kmods SIG |
d83023 |
+ nwnd = wnd->nwnd;
|
|
Kmods SIG |
d83023 |
+ } else {
|
|
Kmods SIG |
d83023 |
+ size_t t = max_alloc + wbits - 1;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ nwnd = likely(t > max_alloc) ? (t >> log2_bits) : wnd->nwnd;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* Enumerate all windows */
|
|
Kmods SIG |
d83023 |
+ for (; iw < nwnd; iw++) {
|
|
Kmods SIG |
d83023 |
+ wbit = iw << log2_bits;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (!wnd->free_bits[iw]) {
|
|
Kmods SIG |
d83023 |
+ if (prev_tail > b_len) {
|
|
Kmods SIG |
d83023 |
+ b_pos = wbit - prev_tail;
|
|
Kmods SIG |
d83023 |
+ b_len = prev_tail;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* Skip full used window */
|
|
Kmods SIG |
d83023 |
+ prev_tail = 0;
|
|
Kmods SIG |
d83023 |
+ wpos = 0;
|
|
Kmods SIG |
d83023 |
+ continue;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (unlikely(iw + 1 == nwnd)) {
|
|
Kmods SIG |
d83023 |
+ if (max_alloc == wnd->nbits) {
|
|
Kmods SIG |
d83023 |
+ wbits = wnd->bits_last;
|
|
Kmods SIG |
d83023 |
+ } else {
|
|
Kmods SIG |
d83023 |
+ size_t t = max_alloc & (wbits - 1);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (t) {
|
|
Kmods SIG |
d83023 |
+ wbits = t;
|
|
Kmods SIG |
d83023 |
+ fbits_valid = false;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (wnd->zone_end > wnd->zone_bit) {
|
|
Kmods SIG |
d83023 |
+ ebit = wbit + wbits;
|
|
Kmods SIG |
d83023 |
+ zbit = max(wnd->zone_bit, wbit);
|
|
Kmods SIG |
d83023 |
+ zend = min(wnd->zone_end, ebit);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* Here we have a window [wbit, ebit) and zone [zbit, zend) */
|
|
Kmods SIG |
d83023 |
+ if (zend <= zbit) {
|
|
Kmods SIG |
d83023 |
+ /* Zone does not overlap window */
|
|
Kmods SIG |
d83023 |
+ } else {
|
|
Kmods SIG |
d83023 |
+ wzbit = zbit - wbit;
|
|
Kmods SIG |
d83023 |
+ wzend = zend - wbit;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* Zone overlaps window */
|
|
Kmods SIG |
d83023 |
+ if (wnd->free_bits[iw] == wzend - wzbit) {
|
|
Kmods SIG |
d83023 |
+ prev_tail = 0;
|
|
Kmods SIG |
d83023 |
+ wpos = 0;
|
|
Kmods SIG |
d83023 |
+ continue;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* Scan two ranges window: [wbit, zbit) and [zend, ebit) */
|
|
Kmods SIG |
d83023 |
+ bh = wnd_map(wnd, iw);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (IS_ERR(bh)) {
|
|
Kmods SIG |
d83023 |
+ /* TODO: error */
|
|
Kmods SIG |
d83023 |
+ prev_tail = 0;
|
|
Kmods SIG |
d83023 |
+ wpos = 0;
|
|
Kmods SIG |
d83023 |
+ continue;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ buf = (ulong *)bh->b_data;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* Scan range [wbit, zbit) */
|
|
Kmods SIG |
d83023 |
+ if (wpos < wzbit) {
|
|
Kmods SIG |
d83023 |
+ /* Scan range [wpos, zbit) */
|
|
Kmods SIG |
d83023 |
+ fnd = wnd_scan(buf, wbit, wpos, wzbit,
|
|
Kmods SIG |
d83023 |
+ to_alloc, &prev_tail,
|
|
Kmods SIG |
d83023 |
+ &b_pos, &b_len);
|
|
Kmods SIG |
d83023 |
+ if (fnd != MINUS_ONE_T) {
|
|
Kmods SIG |
d83023 |
+ put_bh(bh);
|
|
Kmods SIG |
d83023 |
+ goto found;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ prev_tail = 0;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* Scan range [zend, ebit) */
|
|
Kmods SIG |
d83023 |
+ if (wzend < wbits) {
|
|
Kmods SIG |
d83023 |
+ fnd = wnd_scan(buf, wbit,
|
|
Kmods SIG |
d83023 |
+ max(wzend, wpos), wbits,
|
|
Kmods SIG |
d83023 |
+ to_alloc, &prev_tail,
|
|
Kmods SIG |
d83023 |
+ &b_pos, &b_len);
|
|
Kmods SIG |
d83023 |
+ if (fnd != MINUS_ONE_T) {
|
|
Kmods SIG |
d83023 |
+ put_bh(bh);
|
|
Kmods SIG |
d83023 |
+ goto found;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ wpos = 0;
|
|
Kmods SIG |
d83023 |
+ put_bh(bh);
|
|
Kmods SIG |
d83023 |
+ continue;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* Current window does not overlap zone */
|
|
Kmods SIG |
d83023 |
+ if (!wpos && fbits_valid && wnd->free_bits[iw] == wbits) {
|
|
Kmods SIG |
d83023 |
+ /* window is empty */
|
|
Kmods SIG |
d83023 |
+ if (prev_tail + wbits >= to_alloc) {
|
|
Kmods SIG |
d83023 |
+ fnd = wbit + wpos - prev_tail;
|
|
Kmods SIG |
d83023 |
+ goto found;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* Increase 'prev_tail' and process next window */
|
|
Kmods SIG |
d83023 |
+ prev_tail += wbits;
|
|
Kmods SIG |
d83023 |
+ wpos = 0;
|
|
Kmods SIG |
d83023 |
+ continue;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* read window */
|
|
Kmods SIG |
d83023 |
+ bh = wnd_map(wnd, iw);
|
|
Kmods SIG |
d83023 |
+ if (IS_ERR(bh)) {
|
|
Kmods SIG |
d83023 |
+ // TODO: error
|
|
Kmods SIG |
d83023 |
+ prev_tail = 0;
|
|
Kmods SIG |
d83023 |
+ wpos = 0;
|
|
Kmods SIG |
d83023 |
+ continue;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ buf = (ulong *)bh->b_data;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* Scan range [wpos, eBits) */
|
|
Kmods SIG |
d83023 |
+ fnd = wnd_scan(buf, wbit, wpos, wbits, to_alloc, &prev_tail,
|
|
Kmods SIG |
d83023 |
+ &b_pos, &b_len);
|
|
Kmods SIG |
d83023 |
+ put_bh(bh);
|
|
Kmods SIG |
d83023 |
+ if (fnd != MINUS_ONE_T)
|
|
Kmods SIG |
d83023 |
+ goto found;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (b_len < prev_tail) {
|
|
Kmods SIG |
d83023 |
+ /* The last fragment */
|
|
Kmods SIG |
d83023 |
+ b_len = prev_tail;
|
|
Kmods SIG |
d83023 |
+ b_pos = max_alloc - prev_tail;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (hint) {
|
|
Kmods SIG |
d83023 |
+ /*
|
|
Kmods SIG |
d83023 |
+ * We have scanned range [hint max_alloc)
|
|
Kmods SIG |
d83023 |
+ * Prepare to scan range [0 hint + to_alloc)
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+ size_t nextmax = hint + to_alloc;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (likely(nextmax >= hint) && nextmax < max_alloc)
|
|
Kmods SIG |
d83023 |
+ max_alloc = nextmax;
|
|
Kmods SIG |
d83023 |
+ hint = 0;
|
|
Kmods SIG |
d83023 |
+ goto Again;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (!b_len)
|
|
Kmods SIG |
d83023 |
+ goto no_space;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ wnd->extent_max = b_len;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (flags & BITMAP_FIND_FULL)
|
|
Kmods SIG |
d83023 |
+ goto no_space;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ fnd = b_pos;
|
|
Kmods SIG |
d83023 |
+ to_alloc = b_len;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+found:
|
|
Kmods SIG |
d83023 |
+ if (flags & BITMAP_FIND_MARK_AS_USED) {
|
|
Kmods SIG |
d83023 |
+ /* TODO optimize remove extent (pass 'e'?) */
|
|
Kmods SIG |
d83023 |
+ if (wnd_set_used(wnd, fnd, to_alloc))
|
|
Kmods SIG |
d83023 |
+ goto no_space;
|
|
Kmods SIG |
d83023 |
+ } else if (wnd->extent_max != MINUS_ONE_T &&
|
|
Kmods SIG |
d83023 |
+ to_alloc > wnd->extent_max) {
|
|
Kmods SIG |
d83023 |
+ wnd->extent_max = to_alloc;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ *allocated = fnd;
|
|
Kmods SIG |
d83023 |
+ return to_alloc;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+no_space:
|
|
Kmods SIG |
d83023 |
+ return 0;
|
|
Kmods SIG |
d83023 |
+}
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ * wnd_extend
|
|
Kmods SIG |
d83023 |
+ *
|
|
Kmods SIG |
d83023 |
+ * Extend bitmap ($MFT bitmap)
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+int wnd_extend(struct wnd_bitmap *wnd, size_t new_bits)
|
|
Kmods SIG |
d83023 |
+{
|
|
Kmods SIG |
d83023 |
+ int err;
|
|
Kmods SIG |
d83023 |
+ struct super_block *sb = wnd->sb;
|
|
Kmods SIG |
d83023 |
+ struct ntfs_sb_info *sbi = sb->s_fs_info;
|
|
Kmods SIG |
d83023 |
+ u32 blocksize = sb->s_blocksize;
|
|
Kmods SIG |
d83023 |
+ u32 wbits = blocksize * 8;
|
|
Kmods SIG |
d83023 |
+ u32 b0, new_last;
|
|
Kmods SIG |
d83023 |
+ size_t bits, iw, new_wnd;
|
|
Kmods SIG |
d83023 |
+ size_t old_bits = wnd->nbits;
|
|
Kmods SIG |
d83023 |
+ u16 *new_free;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (new_bits <= old_bits)
|
|
Kmods SIG |
d83023 |
+ return -EINVAL;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* align to 8 byte boundary */
|
|
Kmods SIG |
d83023 |
+ new_wnd = bytes_to_block(sb, bitmap_size(new_bits));
|
|
Kmods SIG |
d83023 |
+ new_last = new_bits & (wbits - 1);
|
|
Kmods SIG |
d83023 |
+ if (!new_last)
|
|
Kmods SIG |
d83023 |
+ new_last = wbits;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (new_wnd != wnd->nwnd) {
|
|
Kmods SIG |
d83023 |
+ new_free = ntfs_malloc(new_wnd * sizeof(u16));
|
|
Kmods SIG |
d83023 |
+ if (!new_free)
|
|
Kmods SIG |
d83023 |
+ return -ENOMEM;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (new_free != wnd->free_bits)
|
|
Kmods SIG |
d83023 |
+ memcpy(new_free, wnd->free_bits,
|
|
Kmods SIG |
d83023 |
+ wnd->nwnd * sizeof(short));
|
|
Kmods SIG |
d83023 |
+ memset(new_free + wnd->nwnd, 0,
|
|
Kmods SIG |
d83023 |
+ (new_wnd - wnd->nwnd) * sizeof(short));
|
|
Kmods SIG |
d83023 |
+ ntfs_free(wnd->free_bits);
|
|
Kmods SIG |
d83023 |
+ wnd->free_bits = new_free;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* Zero bits [old_bits,new_bits) */
|
|
Kmods SIG |
d83023 |
+ bits = new_bits - old_bits;
|
|
Kmods SIG |
d83023 |
+ b0 = old_bits & (wbits - 1);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ for (iw = old_bits >> (sb->s_blocksize_bits + 3); bits; iw += 1) {
|
|
Kmods SIG |
d83023 |
+ u32 op;
|
|
Kmods SIG |
d83023 |
+ size_t frb;
|
|
Kmods SIG |
d83023 |
+ u64 vbo, lbo, bytes;
|
|
Kmods SIG |
d83023 |
+ struct buffer_head *bh;
|
|
Kmods SIG |
d83023 |
+ ulong *buf;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (iw + 1 == new_wnd)
|
|
Kmods SIG |
d83023 |
+ wbits = new_last;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ op = b0 + bits > wbits ? wbits - b0 : bits;
|
|
Kmods SIG |
d83023 |
+ vbo = (u64)iw * blocksize;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ err = ntfs_vbo_to_lbo(sbi, &wnd->run, vbo, &lbo, &bytes);
|
|
Kmods SIG |
d83023 |
+ if (err)
|
|
Kmods SIG |
d83023 |
+ break;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ bh = ntfs_bread(sb, lbo >> sb->s_blocksize_bits);
|
|
Kmods SIG |
d83023 |
+ if (!bh)
|
|
Kmods SIG |
d83023 |
+ return -EIO;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ lock_buffer(bh);
|
|
Kmods SIG |
d83023 |
+ buf = (ulong *)bh->b_data;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ __bitmap_clear(buf, b0, blocksize * 8 - b0);
|
|
Kmods SIG |
d83023 |
+ frb = wbits - __bitmap_weight(buf, wbits);
|
|
Kmods SIG |
d83023 |
+ wnd->total_zeroes += frb - wnd->free_bits[iw];
|
|
Kmods SIG |
d83023 |
+ wnd->free_bits[iw] = frb;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ set_buffer_uptodate(bh);
|
|
Kmods SIG |
d83023 |
+ mark_buffer_dirty(bh);
|
|
Kmods SIG |
d83023 |
+ unlock_buffer(bh);
|
|
Kmods SIG |
d83023 |
+ /*err = sync_dirty_buffer(bh);*/
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ b0 = 0;
|
|
Kmods SIG |
d83023 |
+ bits -= op;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ wnd->nbits = new_bits;
|
|
Kmods SIG |
d83023 |
+ wnd->nwnd = new_wnd;
|
|
Kmods SIG |
d83023 |
+ wnd->bits_last = new_last;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ wnd_add_free_ext(wnd, old_bits, new_bits - old_bits, false);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ return 0;
|
|
Kmods SIG |
d83023 |
+}
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+/*
|
|
Kmods SIG |
d83023 |
+ * wnd_zone_set
|
|
Kmods SIG |
d83023 |
+ */
|
|
Kmods SIG |
d83023 |
+void wnd_zone_set(struct wnd_bitmap *wnd, size_t lcn, size_t len)
|
|
Kmods SIG |
d83023 |
+{
|
|
Kmods SIG |
d83023 |
+ size_t zlen;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ zlen = wnd->zone_end - wnd->zone_bit;
|
|
Kmods SIG |
d83023 |
+ if (zlen)
|
|
Kmods SIG |
d83023 |
+ wnd_add_free_ext(wnd, wnd->zone_bit, zlen, false);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (!RB_EMPTY_ROOT(&wnd->start_tree) && len)
|
|
Kmods SIG |
d83023 |
+ wnd_remove_free_ext(wnd, lcn, len);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ wnd->zone_bit = lcn;
|
|
Kmods SIG |
d83023 |
+ wnd->zone_end = lcn + len;
|
|
Kmods SIG |
d83023 |
+}
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+int ntfs_trim_fs(struct ntfs_sb_info *sbi, struct fstrim_range *range)
|
|
Kmods SIG |
d83023 |
+{
|
|
Kmods SIG |
d83023 |
+ int err = 0;
|
|
Kmods SIG |
d83023 |
+ struct super_block *sb = sbi->sb;
|
|
Kmods SIG |
d83023 |
+ struct wnd_bitmap *wnd = &sbi->used.bitmap;
|
|
Kmods SIG |
d83023 |
+ u32 wbits = 8 * sb->s_blocksize;
|
|
Kmods SIG |
d83023 |
+ CLST len = 0, lcn = 0, done = 0;
|
|
Kmods SIG |
d83023 |
+ CLST minlen = bytes_to_cluster(sbi, range->minlen);
|
|
Kmods SIG |
d83023 |
+ CLST lcn_from = bytes_to_cluster(sbi, range->start);
|
|
Kmods SIG |
d83023 |
+ size_t iw = lcn_from >> (sb->s_blocksize_bits + 3);
|
|
Kmods SIG |
d83023 |
+ u32 wbit = lcn_from & (wbits - 1);
|
|
Kmods SIG |
d83023 |
+ const ulong *buf;
|
|
Kmods SIG |
d83023 |
+ CLST lcn_to;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (!minlen)
|
|
Kmods SIG |
d83023 |
+ minlen = 1;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (range->len == (u64)-1)
|
|
Kmods SIG |
d83023 |
+ lcn_to = wnd->nbits;
|
|
Kmods SIG |
d83023 |
+ else
|
|
Kmods SIG |
d83023 |
+ lcn_to = bytes_to_cluster(sbi, range->start + range->len);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ down_read_nested(&wnd->rw_lock, BITMAP_MUTEX_CLUSTERS);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ for (; iw < wnd->nbits; iw++, wbit = 0) {
|
|
Kmods SIG |
d83023 |
+ CLST lcn_wnd = iw * wbits;
|
|
Kmods SIG |
d83023 |
+ struct buffer_head *bh;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (lcn_wnd > lcn_to)
|
|
Kmods SIG |
d83023 |
+ break;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (!wnd->free_bits[iw])
|
|
Kmods SIG |
d83023 |
+ continue;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (iw + 1 == wnd->nwnd)
|
|
Kmods SIG |
d83023 |
+ wbits = wnd->bits_last;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ if (lcn_wnd + wbits > lcn_to)
|
|
Kmods SIG |
d83023 |
+ wbits = lcn_to - lcn_wnd;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ bh = wnd_map(wnd, iw);
|
|
Kmods SIG |
d83023 |
+ if (IS_ERR(bh)) {
|
|
Kmods SIG |
d83023 |
+ err = PTR_ERR(bh);
|
|
Kmods SIG |
d83023 |
+ break;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ buf = (ulong *)bh->b_data;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ for (; wbit < wbits; wbit++) {
|
|
Kmods SIG |
d83023 |
+ if (!test_bit(wbit, buf)) {
|
|
Kmods SIG |
d83023 |
+ if (!len)
|
|
Kmods SIG |
d83023 |
+ lcn = lcn_wnd + wbit;
|
|
Kmods SIG |
d83023 |
+ len += 1;
|
|
Kmods SIG |
d83023 |
+ continue;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ if (len >= minlen) {
|
|
Kmods SIG |
d83023 |
+ err = ntfs_discard(sbi, lcn, len);
|
|
Kmods SIG |
d83023 |
+ if (err)
|
|
Kmods SIG |
d83023 |
+ goto out;
|
|
Kmods SIG |
d83023 |
+ done += len;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ len = 0;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+ put_bh(bh);
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ /* Process the last fragment */
|
|
Kmods SIG |
d83023 |
+ if (len >= minlen) {
|
|
Kmods SIG |
d83023 |
+ err = ntfs_discard(sbi, lcn, len);
|
|
Kmods SIG |
d83023 |
+ if (err)
|
|
Kmods SIG |
d83023 |
+ goto out;
|
|
Kmods SIG |
d83023 |
+ done += len;
|
|
Kmods SIG |
d83023 |
+ }
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+out:
|
|
Kmods SIG |
d83023 |
+ range->len = (u64)done << sbi->cluster_bits;
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ up_read(&wnd->rw_lock);
|
|
Kmods SIG |
d83023 |
+
|
|
Kmods SIG |
d83023 |
+ return err;
|
|
Kmods SIG |
d83023 |
+}
|
|
Kmods SIG |
d83023 |
--
|
|
Kmods SIG |
d83023 |
2.31.1
|
|
Kmods SIG |
d83023 |
|