genalloc.h 5.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171
  1. /*
  2. * Basic general purpose allocator for managing special purpose
  3. * memory, for example, memory that is not managed by the regular
  4. * kmalloc/kfree interface. Uses for this includes on-device special
  5. * memory, uncached memory etc.
  6. *
  7. * It is safe to use the allocator in NMI handlers and other special
  8. * unblockable contexts that could otherwise deadlock on locks. This
  9. * is implemented by using atomic operations and retries on any
  10. * conflicts. The disadvantage is that there may be livelocks in
  11. * extreme cases. For better scalability, one allocator can be used
  12. * for each CPU.
  13. *
  14. * The lockless operation only works if there is enough memory
  15. * available. If new memory is added to the pool a lock has to be
  16. * still taken. So any user relying on locklessness has to ensure
  17. * that sufficient memory is preallocated.
  18. *
  19. * The basic atomic operation of this allocator is cmpxchg on long.
  20. * On architectures that don't have NMI-safe cmpxchg implementation,
  21. * the allocator can NOT be used in NMI handler. So code uses the
  22. * allocator in NMI handler should depend on
  23. * CONFIG_ARCH_HAVE_NMI_SAFE_CMPXCHG.
  24. *
  25. * This source code is licensed under the GNU General Public License,
  26. * Version 2. See the file COPYING for more details.
  27. */
  28. #ifndef __GENALLOC_H__
  29. #define __GENALLOC_H__
  30. #include <linux/types.h>
  31. #include <linux/spinlock_types.h>
  32. #include <linux/atomic.h>
  33. struct device;
  34. struct device_node;
  35. struct gen_pool;
  36. /**
  37. * typedef genpool_algo_t: Allocation callback function type definition
  38. * @map: Pointer to bitmap
  39. * @size: The bitmap size in bits
  40. * @start: The bitnumber to start searching at
  41. * @nr: The number of zeroed bits we're looking for
  42. * @data: optional additional data used by the callback
  43. * @pool: the pool being allocated from
  44. */
  45. typedef unsigned long (*genpool_algo_t)(unsigned long *map,
  46. unsigned long size,
  47. unsigned long start,
  48. unsigned int nr,
  49. void *data, struct gen_pool *pool);
  50. /*
  51. * General purpose special memory pool descriptor.
  52. */
  53. struct gen_pool {
  54. spinlock_t lock;
  55. struct list_head chunks; /* list of chunks in this pool */
  56. int min_alloc_order; /* minimum allocation order */
  57. genpool_algo_t algo; /* allocation function */
  58. void *data;
  59. const char *name;
  60. };
  61. /*
  62. * General purpose special memory pool chunk descriptor.
  63. */
  64. struct gen_pool_chunk {
  65. struct list_head next_chunk; /* next chunk in pool */
  66. atomic_long_t avail;
  67. phys_addr_t phys_addr; /* physical starting address of memory chunk */
  68. unsigned long start_addr; /* start address of memory chunk */
  69. unsigned long end_addr; /* end address of memory chunk (inclusive) */
  70. unsigned long bits[0]; /* bitmap for allocating memory chunk */
  71. };
  72. /*
  73. * gen_pool data descriptor for gen_pool_first_fit_align.
  74. */
  75. struct genpool_data_align {
  76. int align; /* alignment by bytes for starting address */
  77. };
  78. /*
  79. * gen_pool data descriptor for gen_pool_fixed_alloc.
  80. */
  81. struct genpool_data_fixed {
  82. unsigned long offset; /* The offset of the specific region */
  83. };
  84. extern struct gen_pool *gen_pool_create(int, int);
  85. extern phys_addr_t gen_pool_virt_to_phys(struct gen_pool *pool, unsigned long);
  86. extern int gen_pool_add_virt(struct gen_pool *, unsigned long, phys_addr_t,
  87. size_t, int);
  88. /**
  89. * gen_pool_add - add a new chunk of special memory to the pool
  90. * @pool: pool to add new memory chunk to
  91. * @addr: starting address of memory chunk to add to pool
  92. * @size: size in bytes of the memory chunk to add to pool
  93. * @nid: node id of the node the chunk structure and bitmap should be
  94. * allocated on, or -1
  95. *
  96. * Add a new chunk of special memory to the specified pool.
  97. *
  98. * Returns 0 on success or a -ve errno on failure.
  99. */
  100. static inline int gen_pool_add(struct gen_pool *pool, unsigned long addr,
  101. size_t size, int nid)
  102. {
  103. return gen_pool_add_virt(pool, addr, -1, size, nid);
  104. }
  105. extern void gen_pool_destroy(struct gen_pool *);
  106. extern unsigned long gen_pool_alloc(struct gen_pool *, size_t);
  107. extern unsigned long gen_pool_alloc_algo(struct gen_pool *, size_t,
  108. genpool_algo_t algo, void *data);
  109. extern void *gen_pool_dma_alloc(struct gen_pool *pool, size_t size,
  110. dma_addr_t *dma);
  111. extern void gen_pool_free(struct gen_pool *, unsigned long, size_t);
  112. extern void gen_pool_for_each_chunk(struct gen_pool *,
  113. void (*)(struct gen_pool *, struct gen_pool_chunk *, void *), void *);
  114. extern size_t gen_pool_avail(struct gen_pool *);
  115. extern size_t gen_pool_size(struct gen_pool *);
  116. extern void gen_pool_set_algo(struct gen_pool *pool, genpool_algo_t algo,
  117. void *data);
  118. extern unsigned long gen_pool_first_fit(unsigned long *map, unsigned long size,
  119. unsigned long start, unsigned int nr, void *data,
  120. struct gen_pool *pool);
  121. extern unsigned long gen_pool_fixed_alloc(unsigned long *map,
  122. unsigned long size, unsigned long start, unsigned int nr,
  123. void *data, struct gen_pool *pool);
  124. extern unsigned long gen_pool_first_fit_align(unsigned long *map,
  125. unsigned long size, unsigned long start, unsigned int nr,
  126. void *data, struct gen_pool *pool);
  127. extern unsigned long gen_pool_first_fit_order_align(unsigned long *map,
  128. unsigned long size, unsigned long start, unsigned int nr,
  129. void *data, struct gen_pool *pool);
  130. extern unsigned long gen_pool_best_fit(unsigned long *map, unsigned long size,
  131. unsigned long start, unsigned int nr, void *data,
  132. struct gen_pool *pool);
  133. extern struct gen_pool *devm_gen_pool_create(struct device *dev,
  134. int min_alloc_order, int nid, const char *name);
  135. extern struct gen_pool *gen_pool_get(struct device *dev, const char *name);
  136. bool addr_in_gen_pool(struct gen_pool *pool, unsigned long start,
  137. size_t size);
  138. #ifdef CONFIG_OF
  139. extern struct gen_pool *of_gen_pool_get(struct device_node *np,
  140. const char *propname, int index);
  141. #else
  142. static inline struct gen_pool *of_gen_pool_get(struct device_node *np,
  143. const char *propname, int index)
  144. {
  145. return NULL;
  146. }
  147. #endif
  148. #endif /* __GENALLOC_H__ */