Replace bintree with rbtree implementation

[scan-pagemap] / pagemap.h

diff --git a/pagemap.h b/pagemap.h
index 34bcd75271ffe2597bff6580686ffbc6b2512df0..bf431a8de6b1118f5214cf843f629b8703f56e36 100644 (file)
--- a/pagemap.h
+++ b/pagemap.h
@@ -19,10 +19,11 @@
 #define _PAGEMAP_H
 
 #include <stdlib.h>
+#include <string.h>
 
 #include "utils.h"
 #include "list.h"
-#include "bintree.h"
+#include "rbtree.h"
 
 #define PAGE_SHIFT     12
 #define PAGE_SIZE      (1 << PAGE_SHIFT)
@@ -39,7 +40,7 @@ struct maps_list {
 #define BITRANGE(first, last) (((2ll << (last - first)) - 1) << first)
 
 struct pageframe {
-       struct bintree tree;
+       struct rb_node tree;
        struct list_head ml;    /* List of mappings which refer to this pfn */
 
        unsigned long long pf;  /* page frame entry from /proc/pid/pagemap */
@@ -49,7 +50,10 @@ struct pageframe {
        int refcount;
 };
 
-#define tree_to_pageframe(tree_struct)                         \
+struct pageframe *pf_insert(struct rb_root *root, struct pageframe *pf);
+struct pageframe *pf_search(struct rb_root *root, struct pageframe *pf);
+
+#define rb_to_pageframe(tree_struct)                           \
        container_of((tree_struct), struct pageframe, tree)
 
 static inline void clear_pageframe(struct pageframe *pf)