Root/fs/ntfs/bitmap.c

1/*
2 * bitmap.c - NTFS kernel bitmap handling. Part of the Linux-NTFS project.
3 *
4 * Copyright (c) 2004-2005 Anton Altaparmakov
5 *
6 * This program/include file is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License as published
8 * by the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
10 *
11 * This program/include file is distributed in the hope that it will be
12 * useful, but WITHOUT ANY WARRANTY; without even the implied warranty
13 * of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License
17 * along with this program (in the main directory of the Linux-NTFS
18 * distribution in the file COPYING); if not, write to the Free Software
19 * Foundation,Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20 */
21
22#ifdef NTFS_RW
23
24#include <linux/pagemap.h>
25
26#include "bitmap.h"
27#include "debug.h"
28#include "aops.h"
29#include "ntfs.h"
30
31/**
32 * __ntfs_bitmap_set_bits_in_run - set a run of bits in a bitmap to a value
33 * @vi: vfs inode describing the bitmap
34 * @start_bit: first bit to set
35 * @count: number of bits to set
36 * @value: value to set the bits to (i.e. 0 or 1)
37 * @is_rollback: if 'true' this is a rollback operation
38 *
39 * Set @count bits starting at bit @start_bit in the bitmap described by the
40 * vfs inode @vi to @value, where @value is either 0 or 1.
41 *
42 * @is_rollback should always be 'false', it is for internal use to rollback
43 * errors. You probably want to use ntfs_bitmap_set_bits_in_run() instead.
44 *
45 * Return 0 on success and -errno on error.
46 */
47int __ntfs_bitmap_set_bits_in_run(struct inode *vi, const s64 start_bit,
48        const s64 count, const u8 value, const bool is_rollback)
49{
50    s64 cnt = count;
51    pgoff_t index, end_index;
52    struct address_space *mapping;
53    struct page *page;
54    u8 *kaddr;
55    int pos, len;
56    u8 bit;
57
58    BUG_ON(!vi);
59    ntfs_debug("Entering for i_ino 0x%lx, start_bit 0x%llx, count 0x%llx, "
60            "value %u.%s", vi->i_ino, (unsigned long long)start_bit,
61            (unsigned long long)cnt, (unsigned int)value,
62            is_rollback ? " (rollback)" : "");
63    BUG_ON(start_bit < 0);
64    BUG_ON(cnt < 0);
65    BUG_ON(value > 1);
66    /*
67     * Calculate the indices for the pages containing the first and last
68     * bits, i.e. @start_bit and @start_bit + @cnt - 1, respectively.
69     */
70    index = start_bit >> (3 + PAGE_CACHE_SHIFT);
71    end_index = (start_bit + cnt - 1) >> (3 + PAGE_CACHE_SHIFT);
72
73    /* Get the page containing the first bit (@start_bit). */
74    mapping = vi->i_mapping;
75    page = ntfs_map_page(mapping, index);
76    if (IS_ERR(page)) {
77        if (!is_rollback)
78            ntfs_error(vi->i_sb, "Failed to map first page (error "
79                    "%li), aborting.", PTR_ERR(page));
80        return PTR_ERR(page);
81    }
82    kaddr = page_address(page);
83
84    /* Set @pos to the position of the byte containing @start_bit. */
85    pos = (start_bit >> 3) & ~PAGE_CACHE_MASK;
86
87    /* Calculate the position of @start_bit in the first byte. */
88    bit = start_bit & 7;
89
90    /* If the first byte is partial, modify the appropriate bits in it. */
91    if (bit) {
92        u8 *byte = kaddr + pos;
93        while ((bit & 7) && cnt) {
94            cnt--;
95            if (value)
96                *byte |= 1 << bit++;
97            else
98                *byte &= ~(1 << bit++);
99        }
100        /* If we are done, unmap the page and return success. */
101        if (!cnt)
102            goto done;
103
104        /* Update @pos to the new position. */
105        pos++;
106    }
107    /*
108     * Depending on @value, modify all remaining whole bytes in the page up
109     * to @cnt.
110     */
111    len = min_t(s64, cnt >> 3, PAGE_CACHE_SIZE - pos);
112    memset(kaddr + pos, value ? 0xff : 0, len);
113    cnt -= len << 3;
114
115    /* Update @len to point to the first not-done byte in the page. */
116    if (cnt < 8)
117        len += pos;
118
119    /* If we are not in the last page, deal with all subsequent pages. */
120    while (index < end_index) {
121        BUG_ON(cnt <= 0);
122
123        /* Update @index and get the next page. */
124        flush_dcache_page(page);
125        set_page_dirty(page);
126        ntfs_unmap_page(page);
127        page = ntfs_map_page(mapping, ++index);
128        if (IS_ERR(page))
129            goto rollback;
130        kaddr = page_address(page);
131        /*
132         * Depending on @value, modify all remaining whole bytes in the
133         * page up to @cnt.
134         */
135        len = min_t(s64, cnt >> 3, PAGE_CACHE_SIZE);
136        memset(kaddr, value ? 0xff : 0, len);
137        cnt -= len << 3;
138    }
139    /*
140     * The currently mapped page is the last one. If the last byte is
141     * partial, modify the appropriate bits in it. Note, @len is the
142     * position of the last byte inside the page.
143     */
144    if (cnt) {
145        u8 *byte;
146
147        BUG_ON(cnt > 7);
148
149        bit = cnt;
150        byte = kaddr + len;
151        while (bit--) {
152            if (value)
153                *byte |= 1 << bit;
154            else
155                *byte &= ~(1 << bit);
156        }
157    }
158done:
159    /* We are done. Unmap the page and return success. */
160    flush_dcache_page(page);
161    set_page_dirty(page);
162    ntfs_unmap_page(page);
163    ntfs_debug("Done.");
164    return 0;
165rollback:
166    /*
167     * Current state:
168     * - no pages are mapped
169     * - @count - @cnt is the number of bits that have been modified
170     */
171    if (is_rollback)
172        return PTR_ERR(page);
173    if (count != cnt)
174        pos = __ntfs_bitmap_set_bits_in_run(vi, start_bit, count - cnt,
175                value ? 0 : 1, true);
176    else
177        pos = 0;
178    if (!pos) {
179        /* Rollback was successful. */
180        ntfs_error(vi->i_sb, "Failed to map subsequent page (error "
181                "%li), aborting.", PTR_ERR(page));
182    } else {
183        /* Rollback failed. */
184        ntfs_error(vi->i_sb, "Failed to map subsequent page (error "
185                "%li) and rollback failed (error %i). "
186                "Aborting and leaving inconsistent metadata. "
187                "Unmount and run chkdsk.", PTR_ERR(page), pos);
188        NVolSetErrors(NTFS_SB(vi->i_sb));
189    }
190    return PTR_ERR(page);
191}
192
193#endif /* NTFS_RW */
194

Archive Download this file



interactive