1// SPDX-License-Identifier: GPL-2.0
2/*
3 * Copyright (c) 2014-2016 Christoph Hellwig.
4 */
5
6#include <linux/vmalloc.h>
7
8#include "blocklayout.h"
9
10#define NFSDBG_FACILITY NFSDBG_PNFS_LD
11
12static inline struct pnfs_block_extent *
13ext_node(struct rb_node *node)
14{
15 return rb_entry(node, struct pnfs_block_extent, be_node);
16}
17
18static struct pnfs_block_extent *
19ext_tree_first(struct rb_root *root)
20{
21 struct rb_node *node = rb_first(root);
22 return node ? ext_node(node) : NULL;
23}
24
25static struct pnfs_block_extent *
26ext_tree_prev(struct pnfs_block_extent *be)
27{
28 struct rb_node *node = rb_prev(&be->be_node);
29 return node ? ext_node(node) : NULL;
30}
31
32static struct pnfs_block_extent *
33ext_tree_next(struct pnfs_block_extent *be)
34{
35 struct rb_node *node = rb_next(&be->be_node);
36 return node ? ext_node(node) : NULL;
37}
38
39static inline sector_t
40ext_f_end(struct pnfs_block_extent *be)
41{
42 return be->be_f_offset + be->be_length;
43}
44
45static struct pnfs_block_extent *
46__ext_tree_search(struct rb_root *root, sector_t start)
47{
48 struct rb_node *node = root->rb_node;
49 struct pnfs_block_extent *be = NULL;
50
51 while (node) {
52 be = ext_node(node);
53 if (start < be->be_f_offset)
54 node = node->rb_left;
55 else if (start >= ext_f_end(be))
56 node = node->rb_right;
57 else
58 return be;
59 }
60
61 if (be) {
62 if (start < be->be_f_offset)
63 return be;
64
65 if (start >= ext_f_end(be))
66 return ext_tree_next(be);
67 }
68
69 return NULL;
70}
71
72static bool
73ext_can_merge(struct pnfs_block_extent *be1, struct pnfs_block_extent *be2)
74{
75 if (be1->be_state != be2->be_state)
76 return false;
77 if (be1->be_device != be2->be_device)
78 return false;
79
80 if (be1->be_f_offset + be1->be_length != be2->be_f_offset)
81 return false;
82
83 if (be1->be_state != PNFS_BLOCK_NONE_DATA &&
84 (be1->be_v_offset + be1->be_length != be2->be_v_offset))
85 return false;
86
87 if (be1->be_state == PNFS_BLOCK_INVALID_DATA &&
88 be1->be_tag != be2->be_tag)
89 return false;
90
91 return true;
92}
93
94static struct pnfs_block_extent *
95ext_try_to_merge_left(struct rb_root *root, struct pnfs_block_extent *be)
96{
97 struct pnfs_block_extent *left = ext_tree_prev(be);
98
99 if (left && ext_can_merge(be1: left, be2: be)) {
100 left->be_length += be->be_length;
101 rb_erase(&be->be_node, root);
102 nfs4_put_deviceid_node(be->be_device);
103 kfree(objp: be);
104 return left;
105 }
106
107 return be;
108}
109
110static struct pnfs_block_extent *
111ext_try_to_merge_right(struct rb_root *root, struct pnfs_block_extent *be)
112{
113 struct pnfs_block_extent *right = ext_tree_next(be);
114
115 if (right && ext_can_merge(be1: be, be2: right)) {
116 be->be_length += right->be_length;
117 rb_erase(&right->be_node, root);
118 nfs4_put_deviceid_node(right->be_device);
119 kfree(objp: right);
120 }
121
122 return be;
123}
124
125static void __ext_put_deviceids(struct list_head *head)
126{
127 struct pnfs_block_extent *be, *tmp;
128
129 list_for_each_entry_safe(be, tmp, head, be_list) {
130 nfs4_put_deviceid_node(be->be_device);
131 kfree(objp: be);
132 }
133}
134
135static void
136__ext_tree_insert(struct rb_root *root,
137 struct pnfs_block_extent *new, bool merge_ok)
138{
139 struct rb_node **p = &root->rb_node, *parent = NULL;
140 struct pnfs_block_extent *be;
141
142 while (*p) {
143 parent = *p;
144 be = ext_node(node: parent);
145
146 if (new->be_f_offset < be->be_f_offset) {
147 if (merge_ok && ext_can_merge(be1: new, be2: be)) {
148 be->be_f_offset = new->be_f_offset;
149 if (be->be_state != PNFS_BLOCK_NONE_DATA)
150 be->be_v_offset = new->be_v_offset;
151 be->be_length += new->be_length;
152 be = ext_try_to_merge_left(root, be);
153 goto free_new;
154 }
155 p = &(*p)->rb_left;
156 } else if (new->be_f_offset >= ext_f_end(be)) {
157 if (merge_ok && ext_can_merge(be1: be, be2: new)) {
158 be->be_length += new->be_length;
159 be = ext_try_to_merge_right(root, be);
160 goto free_new;
161 }
162 p = &(*p)->rb_right;
163 } else {
164 BUG();
165 }
166 }
167
168 rb_link_node(node: &new->be_node, parent, rb_link: p);
169 rb_insert_color(&new->be_node, root);
170 return;
171free_new:
172 nfs4_put_deviceid_node(new->be_device);
173 kfree(objp: new);
174}
175
176static int
177__ext_tree_remove(struct rb_root *root,
178 sector_t start, sector_t end, struct list_head *tmp)
179{
180 struct pnfs_block_extent *be;
181 sector_t len1 = 0, len2 = 0;
182 sector_t orig_v_offset;
183 sector_t orig_len;
184
185 be = __ext_tree_search(root, start);
186 if (!be)
187 return 0;
188 if (be->be_f_offset >= end)
189 return 0;
190
191 orig_v_offset = be->be_v_offset;
192 orig_len = be->be_length;
193
194 if (start > be->be_f_offset)
195 len1 = start - be->be_f_offset;
196 if (ext_f_end(be) > end)
197 len2 = ext_f_end(be) - end;
198
199 if (len2 > 0) {
200 if (len1 > 0) {
201 struct pnfs_block_extent *new;
202
203 new = kzalloc(size: sizeof(*new), GFP_ATOMIC);
204 if (!new)
205 return -ENOMEM;
206
207 be->be_length = len1;
208
209 new->be_f_offset = end;
210 if (be->be_state != PNFS_BLOCK_NONE_DATA) {
211 new->be_v_offset =
212 orig_v_offset + orig_len - len2;
213 }
214 new->be_length = len2;
215 new->be_state = be->be_state;
216 new->be_tag = be->be_tag;
217 new->be_device = nfs4_get_deviceid(d: be->be_device);
218
219 __ext_tree_insert(root, new, merge_ok: true);
220 } else {
221 be->be_f_offset = end;
222 if (be->be_state != PNFS_BLOCK_NONE_DATA) {
223 be->be_v_offset =
224 orig_v_offset + orig_len - len2;
225 }
226 be->be_length = len2;
227 }
228 } else {
229 if (len1 > 0) {
230 be->be_length = len1;
231 be = ext_tree_next(be);
232 }
233
234 while (be && ext_f_end(be) <= end) {
235 struct pnfs_block_extent *next = ext_tree_next(be);
236
237 rb_erase(&be->be_node, root);
238 list_add_tail(new: &be->be_list, head: tmp);
239 be = next;
240 }
241
242 if (be && be->be_f_offset < end) {
243 len1 = ext_f_end(be) - end;
244 be->be_f_offset = end;
245 if (be->be_state != PNFS_BLOCK_NONE_DATA)
246 be->be_v_offset += be->be_length - len1;
247 be->be_length = len1;
248 }
249 }
250
251 return 0;
252}
253
254int
255ext_tree_insert(struct pnfs_block_layout *bl, struct pnfs_block_extent *new)
256{
257 struct pnfs_block_extent *be;
258 struct rb_root *root;
259 int err = 0;
260
261 switch (new->be_state) {
262 case PNFS_BLOCK_READWRITE_DATA:
263 case PNFS_BLOCK_INVALID_DATA:
264 root = &bl->bl_ext_rw;
265 break;
266 case PNFS_BLOCK_READ_DATA:
267 case PNFS_BLOCK_NONE_DATA:
268 root = &bl->bl_ext_ro;
269 break;
270 default:
271 dprintk("invalid extent type\n");
272 return -EINVAL;
273 }
274
275 spin_lock(lock: &bl->bl_ext_lock);
276retry:
277 be = __ext_tree_search(root, start: new->be_f_offset);
278 if (!be || be->be_f_offset >= ext_f_end(be: new)) {
279 __ext_tree_insert(root, new, merge_ok: true);
280 } else if (new->be_f_offset >= be->be_f_offset) {
281 if (ext_f_end(be: new) <= ext_f_end(be)) {
282 nfs4_put_deviceid_node(new->be_device);
283 kfree(objp: new);
284 } else {
285 sector_t new_len = ext_f_end(be: new) - ext_f_end(be);
286 sector_t diff = new->be_length - new_len;
287
288 new->be_f_offset += diff;
289 new->be_v_offset += diff;
290 new->be_length = new_len;
291 goto retry;
292 }
293 } else if (ext_f_end(be: new) <= ext_f_end(be)) {
294 new->be_length = be->be_f_offset - new->be_f_offset;
295 __ext_tree_insert(root, new, merge_ok: true);
296 } else {
297 struct pnfs_block_extent *split;
298 sector_t new_len = ext_f_end(be: new) - ext_f_end(be);
299 sector_t diff = new->be_length - new_len;
300
301 split = kmemdup(p: new, size: sizeof(*new), GFP_ATOMIC);
302 if (!split) {
303 err = -EINVAL;
304 goto out;
305 }
306
307 split->be_length = be->be_f_offset - split->be_f_offset;
308 split->be_device = nfs4_get_deviceid(d: new->be_device);
309 __ext_tree_insert(root, new: split, merge_ok: true);
310
311 new->be_f_offset += diff;
312 new->be_v_offset += diff;
313 new->be_length = new_len;
314 goto retry;
315 }
316out:
317 spin_unlock(lock: &bl->bl_ext_lock);
318 return err;
319}
320
321static bool
322__ext_tree_lookup(struct rb_root *root, sector_t isect,
323 struct pnfs_block_extent *ret)
324{
325 struct rb_node *node;
326 struct pnfs_block_extent *be;
327
328 node = root->rb_node;
329 while (node) {
330 be = ext_node(node);
331 if (isect < be->be_f_offset)
332 node = node->rb_left;
333 else if (isect >= ext_f_end(be))
334 node = node->rb_right;
335 else {
336 *ret = *be;
337 return true;
338 }
339 }
340
341 return false;
342}
343
344bool
345ext_tree_lookup(struct pnfs_block_layout *bl, sector_t isect,
346 struct pnfs_block_extent *ret, bool rw)
347{
348 bool found = false;
349
350 spin_lock(lock: &bl->bl_ext_lock);
351 if (!rw)
352 found = __ext_tree_lookup(root: &bl->bl_ext_ro, isect, ret);
353 if (!found)
354 found = __ext_tree_lookup(root: &bl->bl_ext_rw, isect, ret);
355 spin_unlock(lock: &bl->bl_ext_lock);
356
357 return found;
358}
359
360int ext_tree_remove(struct pnfs_block_layout *bl, bool rw,
361 sector_t start, sector_t end)
362{
363 int err, err2;
364 LIST_HEAD(tmp);
365
366 spin_lock(lock: &bl->bl_ext_lock);
367 err = __ext_tree_remove(root: &bl->bl_ext_ro, start, end, tmp: &tmp);
368 if (rw) {
369 err2 = __ext_tree_remove(root: &bl->bl_ext_rw, start, end, tmp: &tmp);
370 if (!err)
371 err = err2;
372 }
373 spin_unlock(lock: &bl->bl_ext_lock);
374
375 __ext_put_deviceids(head: &tmp);
376 return err;
377}
378
379static int
380ext_tree_split(struct rb_root *root, struct pnfs_block_extent *be,
381 sector_t split)
382{
383 struct pnfs_block_extent *new;
384 sector_t orig_len = be->be_length;
385
386 new = kzalloc(size: sizeof(*new), GFP_ATOMIC);
387 if (!new)
388 return -ENOMEM;
389
390 be->be_length = split - be->be_f_offset;
391
392 new->be_f_offset = split;
393 if (be->be_state != PNFS_BLOCK_NONE_DATA)
394 new->be_v_offset = be->be_v_offset + be->be_length;
395 new->be_length = orig_len - be->be_length;
396 new->be_state = be->be_state;
397 new->be_tag = be->be_tag;
398 new->be_device = nfs4_get_deviceid(d: be->be_device);
399
400 __ext_tree_insert(root, new, merge_ok: false);
401 return 0;
402}
403
404int
405ext_tree_mark_written(struct pnfs_block_layout *bl, sector_t start,
406 sector_t len, u64 lwb)
407{
408 struct rb_root *root = &bl->bl_ext_rw;
409 sector_t end = start + len;
410 struct pnfs_block_extent *be;
411 int err = 0;
412 LIST_HEAD(tmp);
413
414 spin_lock(lock: &bl->bl_ext_lock);
415 /*
416 * First remove all COW extents or holes from written to range.
417 */
418 err = __ext_tree_remove(root: &bl->bl_ext_ro, start, end, tmp: &tmp);
419 if (err)
420 goto out;
421
422 /*
423 * Then mark all invalid extents in the range as written to.
424 */
425 for (be = __ext_tree_search(root, start); be; be = ext_tree_next(be)) {
426 if (be->be_f_offset >= end)
427 break;
428
429 if (be->be_state != PNFS_BLOCK_INVALID_DATA || be->be_tag)
430 continue;
431
432 if (be->be_f_offset < start) {
433 struct pnfs_block_extent *left = ext_tree_prev(be);
434
435 if (left && ext_can_merge(be1: left, be2: be)) {
436 sector_t diff = start - be->be_f_offset;
437
438 left->be_length += diff;
439
440 be->be_f_offset += diff;
441 be->be_v_offset += diff;
442 be->be_length -= diff;
443 } else {
444 err = ext_tree_split(root, be, split: start);
445 if (err)
446 goto out;
447 }
448 }
449
450 if (ext_f_end(be) > end) {
451 struct pnfs_block_extent *right = ext_tree_next(be);
452
453 if (right && ext_can_merge(be1: be, be2: right)) {
454 sector_t diff = end - be->be_f_offset;
455
456 be->be_length -= diff;
457
458 right->be_f_offset -= diff;
459 right->be_v_offset -= diff;
460 right->be_length += diff;
461 } else {
462 err = ext_tree_split(root, be, split: end);
463 if (err)
464 goto out;
465 }
466 }
467
468 if (be->be_f_offset >= start && ext_f_end(be) <= end) {
469 be->be_tag = EXTENT_WRITTEN;
470 be = ext_try_to_merge_left(root, be);
471 be = ext_try_to_merge_right(root, be);
472 }
473 }
474out:
475 if (bl->bl_lwb < lwb)
476 bl->bl_lwb = lwb;
477 spin_unlock(lock: &bl->bl_ext_lock);
478
479 __ext_put_deviceids(head: &tmp);
480 return err;
481}
482
483static size_t ext_tree_layoutupdate_size(struct pnfs_block_layout *bl, size_t count)
484{
485 if (bl->bl_scsi_layout)
486 return sizeof(__be32) + PNFS_SCSI_RANGE_SIZE * count;
487 else
488 return sizeof(__be32) + PNFS_BLOCK_EXTENT_SIZE * count;
489}
490
491static void ext_tree_free_commitdata(struct nfs4_layoutcommit_args *arg,
492 size_t buffer_size)
493{
494 if (arg->layoutupdate_pages != &arg->layoutupdate_page) {
495 int nr_pages = DIV_ROUND_UP(buffer_size, PAGE_SIZE), i;
496
497 for (i = 0; i < nr_pages; i++)
498 put_page(page: arg->layoutupdate_pages[i]);
499 vfree(addr: arg->start_p);
500 kfree(objp: arg->layoutupdate_pages);
501 } else {
502 put_page(page: arg->layoutupdate_page);
503 }
504}
505
506static __be32 *encode_block_extent(struct pnfs_block_extent *be, __be32 *p)
507{
508 p = xdr_encode_opaque_fixed(p, ptr: be->be_device->deviceid.data,
509 NFS4_DEVICEID4_SIZE);
510 p = xdr_encode_hyper(p, val: be->be_f_offset << SECTOR_SHIFT);
511 p = xdr_encode_hyper(p, val: be->be_length << SECTOR_SHIFT);
512 p = xdr_encode_hyper(p, val: 0LL);
513 *p++ = cpu_to_be32(PNFS_BLOCK_READWRITE_DATA);
514 return p;
515}
516
517static __be32 *encode_scsi_range(struct pnfs_block_extent *be, __be32 *p)
518{
519 p = xdr_encode_hyper(p, val: be->be_f_offset << SECTOR_SHIFT);
520 return xdr_encode_hyper(p, val: be->be_length << SECTOR_SHIFT);
521}
522
523static int ext_tree_encode_commit(struct pnfs_block_layout *bl, __be32 *p,
524 size_t buffer_size, size_t *count, __u64 *lastbyte)
525{
526 struct pnfs_block_extent *be;
527 int ret = 0;
528
529 spin_lock(lock: &bl->bl_ext_lock);
530 for (be = ext_tree_first(root: &bl->bl_ext_rw); be; be = ext_tree_next(be)) {
531 if (be->be_state != PNFS_BLOCK_INVALID_DATA ||
532 be->be_tag != EXTENT_WRITTEN)
533 continue;
534
535 (*count)++;
536 if (ext_tree_layoutupdate_size(bl, count: *count) > buffer_size) {
537 /* keep counting.. */
538 ret = -ENOSPC;
539 continue;
540 }
541
542 if (bl->bl_scsi_layout)
543 p = encode_scsi_range(be, p);
544 else
545 p = encode_block_extent(be, p);
546 be->be_tag = EXTENT_COMMITTING;
547 }
548 *lastbyte = bl->bl_lwb - 1;
549 bl->bl_lwb = 0;
550 spin_unlock(lock: &bl->bl_ext_lock);
551
552 return ret;
553}
554
555int
556ext_tree_prepare_commit(struct nfs4_layoutcommit_args *arg)
557{
558 struct pnfs_block_layout *bl = BLK_LO2EXT(lo: NFS_I(inode: arg->inode)->layout);
559 size_t count = 0, buffer_size = PAGE_SIZE;
560 __be32 *start_p;
561 int ret;
562
563 dprintk("%s enter\n", __func__);
564
565 arg->layoutupdate_page = alloc_page(GFP_NOFS);
566 if (!arg->layoutupdate_page)
567 return -ENOMEM;
568 start_p = page_address(arg->layoutupdate_page);
569 arg->layoutupdate_pages = &arg->layoutupdate_page;
570
571retry:
572 ret = ext_tree_encode_commit(bl, p: start_p + 1, buffer_size, count: &count, lastbyte: &arg->lastbytewritten);
573 if (unlikely(ret)) {
574 ext_tree_free_commitdata(arg, buffer_size);
575
576 buffer_size = ext_tree_layoutupdate_size(bl, count);
577 count = 0;
578
579 arg->layoutupdate_pages =
580 kcalloc(DIV_ROUND_UP(buffer_size, PAGE_SIZE),
581 size: sizeof(struct page *), GFP_NOFS);
582 if (!arg->layoutupdate_pages)
583 return -ENOMEM;
584
585 start_p = __vmalloc(size: buffer_size, GFP_NOFS);
586 if (!start_p) {
587 kfree(objp: arg->layoutupdate_pages);
588 return -ENOMEM;
589 }
590
591 goto retry;
592 }
593
594 *start_p = cpu_to_be32(count);
595 arg->layoutupdate_len = ext_tree_layoutupdate_size(bl, count);
596
597 if (unlikely(arg->layoutupdate_pages != &arg->layoutupdate_page)) {
598 void *p = start_p, *end = p + arg->layoutupdate_len;
599 struct page *page = NULL;
600 int i = 0;
601
602 arg->start_p = start_p;
603 for ( ; p < end; p += PAGE_SIZE) {
604 page = vmalloc_to_page(addr: p);
605 arg->layoutupdate_pages[i++] = page;
606 get_page(page);
607 }
608 }
609
610 dprintk("%s found %zu ranges\n", __func__, count);
611 return 0;
612}
613
614void
615ext_tree_mark_committed(struct nfs4_layoutcommit_args *arg, int status)
616{
617 struct pnfs_block_layout *bl = BLK_LO2EXT(lo: NFS_I(inode: arg->inode)->layout);
618 struct rb_root *root = &bl->bl_ext_rw;
619 struct pnfs_block_extent *be;
620
621 dprintk("%s status %d\n", __func__, status);
622
623 ext_tree_free_commitdata(arg, buffer_size: arg->layoutupdate_len);
624
625 spin_lock(lock: &bl->bl_ext_lock);
626 for (be = ext_tree_first(root); be; be = ext_tree_next(be)) {
627 if (be->be_state != PNFS_BLOCK_INVALID_DATA ||
628 be->be_tag != EXTENT_COMMITTING)
629 continue;
630
631 if (status) {
632 /*
633 * Mark as written and try again.
634 *
635 * XXX: some real error handling here wouldn't hurt..
636 */
637 be->be_tag = EXTENT_WRITTEN;
638 } else {
639 be->be_state = PNFS_BLOCK_READWRITE_DATA;
640 be->be_tag = 0;
641 }
642
643 be = ext_try_to_merge_left(root, be);
644 be = ext_try_to_merge_right(root, be);
645 }
646 spin_unlock(lock: &bl->bl_ext_lock);
647}
648

source code of linux/fs/nfs/blocklayout/extent_tree.c