From 52a27b09fc6b7384c7dd0d7697ce5fe100d9652c Mon Sep 17 00:00:00 2001 From: Alex Vainman Date: Mon, 1 Feb 2010 07:57:45 +0200 Subject: [PATCH] Fix incorrect splits/merges in the memory tree when madvise() fails. ibv_madvise_range() first manages (splits or merges) memory ranges in the tree and only then calls madvise(). If madvise() fails, the tree's memory range may contain incorrectly split or merged ranges. The patch undoes the split and merge operations performed on the node which caused the madvise() failure as well as on that node's neighbors. Signed-off-by: Alex Vainman Signed-off-by: Roland Dreier --- src/memory.c | 37 ++++++++++++++++++++++++++++++++++++- 1 file changed, 36 insertions(+), 1 deletion(-) diff --git a/src/memory.c b/src/memory.c index 6a3305f..03f49c8 100644 --- a/src/memory.c +++ b/src/memory.c @@ -489,6 +489,39 @@ static struct ibv_mem_node *get_start_node(uintptr_t start, uintptr_t end, return node; } +/* + * This function is called if madvise() fails to undo merging/splitting + * operations performed on the node. + */ +static struct ibv_mem_node *undo_node(struct ibv_mem_node *node, + uintptr_t start, int inc) +{ + struct ibv_mem_node *tmp = NULL; + + /* + * This condition can be true only if we merged this + * node with the previous one, so we need to split them. + */ + if (start > node->start) { + tmp = split_range(node, start); + if (tmp) { + node->refcnt += inc; + node = tmp; + } else + return NULL; + } + + tmp = __mm_prev(node); + if (tmp && tmp->refcnt == node->refcnt) + node = merge_ranges(node, tmp); + + tmp = __mm_next(node); + if (tmp && tmp->refcnt == node->refcnt) + node = merge_ranges(tmp, node); + + return node; +} + static int ibv_madvise_range(void *base, size_t size, int advice) { uintptr_t start, end; @@ -541,8 +574,10 @@ static int ibv_madvise_range(void *base, size_t size, int advice) ret = madvise((void *) node->start, node->end - node->start + 1, advice); - if (ret) + if (ret) { + node = undo_node(node, start, inc); goto out; + } } node->refcnt += inc; -- 2.41.0