Improve cake allocator's memory utilisation (#43)

[lunaix-os.git] / lunaix-os / includes / lunaix / mm / cake.h

diff --git a/lunaix-os/includes/lunaix/mm/cake.h b/lunaix-os/includes/lunaix/mm/cake.h
index 0312c45d381ac7854a802cb248eb792ac496cbf6..d880531399d1d06d42cad3a59c6f463e3e14afc3 100644 (file)
--- a/lunaix-os/includes/lunaix/mm/cake.h
+++ b/lunaix-os/includes/lunaix/mm/cake.h
@@ -2,10 +2,16 @@
 #define __LUNAIX_CAKE_H
 
 #include <lunaix/ds/llist.h>
 #define __LUNAIX_CAKE_H
 
 #include <lunaix/ds/llist.h>

+#include <lunaix/spike.h>

 
 #define PILE_NAME_MAXLEN 20
 
 
 #define PILE_NAME_MAXLEN 20
 

-#define PILE_CACHELINE 1
+#define PILE_ALIGN_CACHE 0b0001
+#define PILE_FL_EXTERN   0b0010
+
+struct cake_pile;
+
+typedef void (*pile_cb)(struct cake_pile*, void*);

 
 struct cake_pile
 {
 
 struct cake_pile
 {


@@ -13,26 +19,45 @@ struct cake_pile
     struct llist_header full;
     struct llist_header partial;
     struct llist_header free;
     struct llist_header full;
     struct llist_header partial;
     struct llist_header free;

-    unsigned int offset;
-    unsigned int piece_size;
-    unsigned int cakes_count;
-    unsigned int alloced_pieces;
-    unsigned int pieces_per_cake;
-    unsigned int pg_per_cake;
-    char pile_name[PILE_NAME_MAXLEN];
+    u32_t offset;
+    u32_t piece_size;
+    u32_t cakes_count;
+    u32_t alloced_pieces;
+    u32_t pieces_per_cake;
+    u32_t pg_per_cake;
+    u32_t options;
+    char pile_name[PILE_NAME_MAXLEN+1];
+
+    pile_cb ctor;

 };
 
 };
 

-typedef unsigned int piece_index_t;
+typedef unsigned short piece_t;

 
 

-#define EO_FREE_PIECE (-1)
+#define EO_FREE_PIECE ((piece_t)-1)
+
+#define CAKE_FL_SIZE        128 
+#define CAKE_FL_MAXLEN      \
+    ((unsigned int)((CAKE_FL_SIZE - sizeof(ptr_t)) / sizeof(piece_t)))
+struct cake_fl
+{
+    piece_t indices[CAKE_FL_MAXLEN];
+    struct cake_fl* next;
+} align(CAKE_FL_SIZE);

 
 struct cake_s
 {
     struct llist_header cakes;
 
 struct cake_s
 {
     struct llist_header cakes;

+    struct cake_pile* owner;

     void* first_piece;
     unsigned int used_pieces;
     unsigned int next_free;
     void* first_piece;
     unsigned int used_pieces;
     unsigned int next_free;

-    piece_index_t free_list[0];
+    union {
+        struct cake_fl* fl;
+        struct {
+            void* rsvd;
+            piece_t free_list[0];
+        };
+    };

 };
 
 /**
 };
 
 /**


@@ -49,6 +74,9 @@ cake_new_pile(char* name,
               unsigned int pg_per_cake,
               int options);
 
               unsigned int pg_per_cake,
               int options);
 

+void
+cake_set_constructor(struct cake_pile* pile, pile_cb ctor);
+

 /**
  * @brief 拿一块儿蛋糕
  *
 /**
  * @brief 拿一块儿蛋糕
  *


@@ -70,11 +98,24 @@ cake_release(struct cake_pile* pile, void* area);
 void
 cake_init();
 
 void
 cake_init();
 

-/**
- * @brief 统计蛋糕数量 - 问问Pinkie :D
- *
- */

 void
 void

-cake_stats();
+cake_export();
+
+void
+cake_reclaim_freed();
+
+/********** some handy constructor ***********/
+
+void
+cake_ctor_zeroing(struct cake_pile* pile, void* piece);
+
+#define DEADCAKE_MARK 0xdeadcafeUL
+
+static inline void
+cake_ensure_valid(void* area) {
+    if (unlikely(*(unsigned int*)area == DEADCAKE_MARK)) {
+        fail("access to freed cake piece");
+    }
+}

 
 #endif /* __LUNAIX_VALLOC_H */
 
 #endif /* __LUNAIX_VALLOC_H */