]>
Commit | Line | Data |
---|---|---|
93cf2076 GW |
1 | /* |
2 | * CDDL HEADER START | |
3 | * | |
4 | * The contents of this file are subject to the terms of the | |
5 | * Common Development and Distribution License (the "License"). | |
6 | * You may not use this file except in compliance with the License. | |
7 | * | |
8 | * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE | |
9 | * or http://www.opensolaris.org/os/licensing. | |
10 | * See the License for the specific language governing permissions | |
11 | * and limitations under the License. | |
12 | * | |
13 | * When distributing Covered Code, include this CDDL HEADER in each | |
14 | * file and include the License file at usr/src/OPENSOLARIS.LICENSE. | |
15 | * If applicable, add the following below this CDDL HEADER, with the | |
16 | * fields enclosed by brackets "[]" replaced with your own identifying | |
17 | * information: Portions Copyright [yyyy] [name of copyright owner] | |
18 | * | |
19 | * CDDL HEADER END | |
20 | */ | |
21 | /* | |
22 | * Copyright 2009 Sun Microsystems, Inc. All rights reserved. | |
23 | * Use is subject to license terms. | |
24 | */ | |
25 | ||
26 | /* | |
9bd274dd | 27 | * Copyright (c) 2013, 2014 by Delphix. All rights reserved. |
93cf2076 GW |
28 | */ |
29 | ||
30 | #ifndef _SYS_RANGE_TREE_H | |
31 | #define _SYS_RANGE_TREE_H | |
32 | ||
33 | #include <sys/avl.h> | |
34 | #include <sys/dmu.h> | |
35 | ||
36 | #ifdef __cplusplus | |
37 | extern "C" { | |
38 | #endif | |
39 | ||
40 | #define RANGE_TREE_HISTOGRAM_SIZE 64 | |
41 | ||
42 | typedef struct range_tree_ops range_tree_ops_t; | |
43 | ||
44 | typedef struct range_tree { | |
45 | avl_tree_t rt_root; /* offset-ordered segment AVL tree */ | |
46 | uint64_t rt_space; /* sum of all segments in the map */ | |
d4a72f23 | 47 | uint64_t rt_gap; /* allowable inter-segment gap */ |
93cf2076 | 48 | range_tree_ops_t *rt_ops; |
d4a72f23 TC |
49 | |
50 | /* rt_avl_compare should only be set if rt_arg is an AVL tree */ | |
93cf2076 | 51 | void *rt_arg; |
d4a72f23 TC |
52 | int (*rt_avl_compare)(const void *, const void *); |
53 | ||
93cf2076 GW |
54 | |
55 | /* | |
56 | * The rt_histogram maintains a histogram of ranges. Each bucket, | |
57 | * rt_histogram[i], contains the number of ranges whose size is: | |
58 | * 2^i <= size of range in bytes < 2^(i+1) | |
59 | */ | |
60 | uint64_t rt_histogram[RANGE_TREE_HISTOGRAM_SIZE]; | |
61 | kmutex_t *rt_lock; /* pointer to lock that protects map */ | |
62 | } range_tree_t; | |
63 | ||
64 | typedef struct range_seg { | |
65 | avl_node_t rs_node; /* AVL node */ | |
66 | avl_node_t rs_pp_node; /* AVL picker-private node */ | |
67 | uint64_t rs_start; /* starting offset of this segment */ | |
68 | uint64_t rs_end; /* ending offset (non-inclusive) */ | |
d4a72f23 | 69 | uint64_t rs_fill; /* actual fill if gap mode is on */ |
93cf2076 GW |
70 | } range_seg_t; |
71 | ||
72 | struct range_tree_ops { | |
73 | void (*rtop_create)(range_tree_t *rt, void *arg); | |
74 | void (*rtop_destroy)(range_tree_t *rt, void *arg); | |
75 | void (*rtop_add)(range_tree_t *rt, range_seg_t *rs, void *arg); | |
76 | void (*rtop_remove)(range_tree_t *rt, range_seg_t *rs, void *arg); | |
77 | void (*rtop_vacate)(range_tree_t *rt, void *arg); | |
78 | }; | |
79 | ||
80 | typedef void range_tree_func_t(void *arg, uint64_t start, uint64_t size); | |
81 | ||
82 | void range_tree_init(void); | |
83 | void range_tree_fini(void); | |
d4a72f23 TC |
84 | range_tree_t *range_tree_create_impl(range_tree_ops_t *ops, void *arg, |
85 | int (*avl_compare) (const void *, const void *), kmutex_t *lp, | |
86 | uint64_t gap); | |
93cf2076 GW |
87 | range_tree_t *range_tree_create(range_tree_ops_t *ops, void *arg, kmutex_t *lp); |
88 | void range_tree_destroy(range_tree_t *rt); | |
89 | boolean_t range_tree_contains(range_tree_t *rt, uint64_t start, uint64_t size); | |
d4a72f23 TC |
90 | range_seg_t *range_tree_find(range_tree_t *rt, uint64_t start, uint64_t size); |
91 | void range_tree_resize_segment(range_tree_t *rt, range_seg_t *rs, | |
92 | uint64_t newstart, uint64_t newsize); | |
93cf2076 GW |
93 | uint64_t range_tree_space(range_tree_t *rt); |
94 | void range_tree_verify(range_tree_t *rt, uint64_t start, uint64_t size); | |
95 | void range_tree_swap(range_tree_t **rtsrc, range_tree_t **rtdst); | |
96 | void range_tree_stat_verify(range_tree_t *rt); | |
d4a72f23 | 97 | void range_tree_set_lock(range_tree_t *rt, kmutex_t *lp); |
93cf2076 GW |
98 | |
99 | void range_tree_add(void *arg, uint64_t start, uint64_t size); | |
100 | void range_tree_remove(void *arg, uint64_t start, uint64_t size); | |
d4a72f23 TC |
101 | void range_tree_remove_fill(range_tree_t *rt, uint64_t start, uint64_t size); |
102 | void range_tree_adjust_fill(range_tree_t *rt, range_seg_t *rs, int64_t delta); | |
9bd274dd | 103 | void range_tree_clear(range_tree_t *rt, uint64_t start, uint64_t size); |
93cf2076 GW |
104 | |
105 | void range_tree_vacate(range_tree_t *rt, range_tree_func_t *func, void *arg); | |
106 | void range_tree_walk(range_tree_t *rt, range_tree_func_t *func, void *arg); | |
d4a72f23 TC |
107 | range_seg_t *range_tree_first(range_tree_t *rt); |
108 | ||
109 | void rt_avl_create(range_tree_t *rt, void *arg); | |
110 | void rt_avl_destroy(range_tree_t *rt, void *arg); | |
111 | void rt_avl_add(range_tree_t *rt, range_seg_t *rs, void *arg); | |
112 | void rt_avl_remove(range_tree_t *rt, range_seg_t *rs, void *arg); | |
113 | void rt_avl_vacate(range_tree_t *rt, void *arg); | |
114 | extern struct range_tree_ops rt_avl_ops; | |
93cf2076 GW |
115 | |
116 | #ifdef __cplusplus | |
117 | } | |
118 | #endif | |
119 | ||
120 | #endif /* _SYS_RANGE_TREE_H */ |