1 /* SPDX-License-Identifier: MIT */ 2 #ifndef __NVKM_MM_H__ 3 #define __NVKM_MM_H__ 4 #include <core/os.h> 5 6 struct nvkm_mm_node { 7 struct list_head nl_entry; 8 struct list_head fl_entry; 9 struct nvkm_mm_node *next; 10 11 #define NVKM_MM_HEAP_ANY 0x00 12 u8 heap; 13 #define NVKM_MM_TYPE_NONE 0x00 14 #define NVKM_MM_TYPE_HOLE 0xff 15 u8 type; 16 u32 offset; 17 u32 length; 18 }; 19 20 struct nvkm_mm { 21 struct list_head nodes; 22 struct list_head free; 23 24 u32 block_size; 25 int heap_nodes; 26 }; 27 28 static inline bool 29 nvkm_mm_initialised(struct nvkm_mm *mm) 30 { 31 return mm->heap_nodes; 32 } 33 34 int nvkm_mm_init(struct nvkm_mm *, u8 heap, u32 offset, u32 length, u32 block); 35 int nvkm_mm_fini(struct nvkm_mm *); 36 int nvkm_mm_head(struct nvkm_mm *, u8 heap, u8 type, u32 size_max, 37 u32 size_min, u32 align, struct nvkm_mm_node **); 38 int nvkm_mm_tail(struct nvkm_mm *, u8 heap, u8 type, u32 size_max, 39 u32 size_min, u32 align, struct nvkm_mm_node **); 40 void nvkm_mm_free(struct nvkm_mm *, struct nvkm_mm_node **); 41 void nvkm_mm_dump(struct nvkm_mm *, const char *); 42 43 static inline u32 44 nvkm_mm_heap_size(struct nvkm_mm *mm, u8 heap) 45 { 46 struct nvkm_mm_node *node; 47 u32 size = 0; 48 list_for_each_entry(node, &mm->nodes, nl_entry) { 49 if (node->heap == heap) 50 size += node->length; 51 } 52 return size; 53 } 54 55 static inline bool 56 nvkm_mm_contiguous(struct nvkm_mm_node *node) 57 { 58 return !node->next; 59 } 60 61 static inline u32 62 nvkm_mm_addr(struct nvkm_mm_node *node) 63 { 64 if (WARN_ON(!nvkm_mm_contiguous(node))) 65 return 0; 66 return node->offset; 67 } 68 69 static inline u32 70 nvkm_mm_size(struct nvkm_mm_node *node) 71 { 72 u32 size = 0; 73 do { 74 size += node->length; 75 } while ((node = node->next)); 76 return size; 77 } 78 #endif 79