[nrf fromtree] soc: nordic: common: dmm: Add optional usage stats

Add support for getting usage statistics for DMM.

Signed-off-by: Krzysztof Chruściński <[email protected]>
(cherry picked from commit d10ee98)

Author: nordic-krch

soc/nordic/common/Kconfig

@@ -60,6 +60,9 @@ config DMM_HEAP_CHUNKS
 	  Number of chunks is a trade-off between performance and granularity.
 	  Must be multiply of 32.
 
+config DMM_STATS
+	bool "Usage statistics"
+
 endif # HAS_NORDIC_DMM
 
 rsource "vpr/Kconfig"

soc/nordic/common/dmm.c

@@ -52,6 +52,11 @@ struct dmm_heap {
 	size_t blk_size;
 	const struct dmm_region *region;
 	sys_bitarray_t bitarray;
+#ifdef CONFIG_DMM_STATS
+	atomic_t curr_use;
+	uint32_t max_use;
+	struct k_spinlock lock;
+#endif
 };
 
 static const struct dmm_region dmm_regions[] = {
@@ -235,6 +240,15 @@ static void *dmm_buffer_alloc(struct dmm_heap *dh, size_t length)
 
 	tail_mask_set(dh->tail_mask, num_bits, off);
 
+#ifdef CONFIG_DMM_STATS
+	k_spinlock_key_t key;
+
+	key = k_spin_lock(&dh->lock);
+	dh->curr_use += num_bits;
+	dh->max_use = MAX(dh->max_use, dh->curr_use);
+	k_spin_unlock(&dh->lock, key);
+#endif
+
 	return (void *)(dh->ptr + dh->blk_size * off);
 }
 
@@ -244,6 +258,9 @@ static void dmm_buffer_free(struct dmm_heap *dh, void *buffer)
 	size_t num_bits = num_bits_get(dh->tail_mask, offset);
 	int rv;
 
+#ifdef CONFIG_DMM_STATS
+	atomic_sub(&dh->curr_use, num_bits);
+#endif
 	rv = sys_bitarray_free(&dh->bitarray, num_bits, offset);
 	(void)rv;
 	__ASSERT_NO_MSG(rv == 0);
@@ -431,6 +448,34 @@ int dmm_buffer_in_release(void *region, void *user_buffer, size_t user_length, v
 	return 0;
 }
 
+int dmm_stats_get(void *region, uintptr_t *start_addr, uint32_t *curr_use, uint32_t *max_use)
+{
+#ifdef CONFIG_DMM_STATS
+	struct dmm_heap *dh;
+
+	dh = dmm_heap_find(region);
+	if (dh == NULL) {
+		return -EINVAL;
+	}
+
+	if (start_addr) {
+		*start_addr = dh->ptr;
+	}
+
+	if (curr_use) {
+		*curr_use = (100 * dh->curr_use) / dh->bitarray.num_bits;
+	}
+
+	if (max_use) {
+		*max_use = (100 * dh->max_use) / dh->bitarray.num_bits;
+	}
+
+	return 0;
+#else
+	return -ENOTSUP;
+#endif
+}
+
 int dmm_init(void)
 {
 	struct dmm_heap *dh;

soc/nordic/common/dmm.h

@@ -163,6 +163,22 @@ int dmm_buffer_in_prepare(void *region, void *user_buffer, size_t user_length, v
  */
 int dmm_buffer_in_release(void *region, void *user_buffer, size_t user_length, void *buffer_in);
 
+/**
+ * @brief Get statistics.
+ *
+ * Must be enabled with CONFIG_DMM_STATS.
+ *
+ * @param[in] region DMM memory region.
+ * @param[out] start_addr Location where starting address of the memory region is set. Can be null.
+ * @param[out] curr_use Location where current use in percent is written. Can be null.
+ * @param[out] max_use Location where maximum use in percent is written. Can be null.
+ *
+ * @retval 0 on success.
+ * @retval -EINVAL Invalid region.
+ * @retval -ENOTSUP Feature is disabled.
+ */
+int dmm_stats_get(void *region, uintptr_t *start_addr, uint32_t *curr_use, uint32_t *max_use);
+
 /**
  * @brief Initialize DMM.
  *
@@ -210,6 +226,17 @@ static ALWAYS_INLINE int dmm_buffer_in_release(void *region, void *user_buffer,
 	return 0;
 }
 
+static ALWAYS_INLINE int dmm_stats_get(void *region, uintptr_t *start_addr,
+				       uint32_t *curr_use, uint32_t *max_use)
+{
+	ARG_UNUSED(region);
+	ARG_UNUSED(start_addr);
+	ARG_UNUSED(curr_use);
+	ARG_UNUSED(max_use);
+
+	return 0;
+}
+
 static ALWAYS_INLINE int dmm_init(void)
 {
 	return 0;