This patch supports backup memory mechanism, the memarea object could use another memarea object as a backup.
Signed-off-by: Chengwen Feng <fengcheng...@huawei.com> --- doc/guides/prog_guide/memarea_lib.rst | 3 +++ lib/memarea/memarea_private.h | 2 ++ lib/memarea/rte_memarea.c | 22 ++++++++++++++++++++++ lib/memarea/rte_memarea.h | 6 ++++++ 4 files changed, 33 insertions(+) diff --git a/doc/guides/prog_guide/memarea_lib.rst b/doc/guides/prog_guide/memarea_lib.rst index 7686625c90..c52ab95aeb 100644 --- a/doc/guides/prog_guide/memarea_lib.rst +++ b/doc/guides/prog_guide/memarea_lib.rst @@ -26,6 +26,9 @@ The main features are as follows: * It supports MT-safe as long as it's specified at creation time. +* It provides backup memory mechanism, the memory object could use another + memarea object as a backup. + Library API Overview -------------------- diff --git a/lib/memarea/memarea_private.h b/lib/memarea/memarea_private.h index f71e01e7b5..ce765ed801 100644 --- a/lib/memarea/memarea_private.h +++ b/lib/memarea/memarea_private.h @@ -23,11 +23,13 @@ struct memarea_private { struct rte_memarea_param init; rte_spinlock_t lock; void *area_addr; + void *top_addr; struct memarea_elem_list elem_list; struct memarea_elem_list free_list; uint64_t alloc_fails; uint64_t refcnt_check_fails; + uint64_t bak_alloc_fails; } __rte_cache_aligned; #endif /* MEMAREA_PRIVATE_H */ diff --git a/lib/memarea/rte_memarea.c b/lib/memarea/rte_memarea.c index e40715d16a..3efe8d4819 100644 --- a/lib/memarea/rte_memarea.c +++ b/lib/memarea/rte_memarea.c @@ -116,6 +116,7 @@ rte_memarea_create(const struct rte_memarea_param *init) TAILQ_INIT(&priv->elem_list); TAILQ_INIT(&priv->free_list); priv->area_addr = addr; + priv->top_addr = (void *)((uintptr_t)addr + init->total_sz - 1); elem = addr; elem->size = init->total_sz - sizeof(struct memarea_elem); elem->cookie = MEMAREA_FREE_ELEM_COOKIE; @@ -183,6 +184,15 @@ memarea_add_node(struct memarea_private *priv, struct memarea_elem *elem, size_t elem->size = align_size; } +static inline void * +memarea_alloc_backup(struct memarea_private *priv, size_t size, uint32_t cookie) +{ + void *ptr = rte_memarea_alloc(priv->init.bak_memarea, size, cookie); + if (unlikely(ptr == NULL)) + priv->bak_alloc_fails++; + return ptr; +} + void * rte_memarea_alloc(struct rte_memarea *ma, size_t size, uint32_t cookie) { @@ -206,6 +216,8 @@ rte_memarea_alloc(struct rte_memarea *ma, size_t size, uint32_t cookie) ptr = (void *)((uintptr_t)elem + sizeof(struct memarea_elem)); break; } + if (ptr == NULL && priv->init.bak_memarea != NULL) + ptr = memarea_alloc_backup(priv, size, cookie); if (unlikely(ptr == NULL)) priv->alloc_fails++; memarea_unlock(priv); @@ -264,6 +276,12 @@ rte_memarea_update_refcnt(struct rte_memarea *ma, void *ptr, int16_t value) struct memarea_private *priv = ma->private_data; memarea_lock(priv); + if (ptr < priv->area_addr || ptr > priv->top_addr) { + rte_memarea_update_refcnt(priv->init.bak_memarea, ptr, value); + memarea_unlock(priv); + return; + } + if (unlikely(elem->refcnt <= 0 || elem->refcnt + value < 0)) { RTE_LOG(ERR, MEMAREA, "memarea cookie: %u curr refcnt: %d update refcnt: %d check fail!\n", @@ -274,6 +292,7 @@ rte_memarea_update_refcnt(struct rte_memarea *ma, void *ptr, int16_t value) memarea_unlock(priv); return; } + elem->refcnt += value; if (elem->refcnt == 0) memarea_free_elem(priv, elem); @@ -351,10 +370,13 @@ rte_memarea_dump(struct rte_memarea *ma, FILE *f, bool dump_all) fprintf(f, " source-user-memarea: %s\n", memarea_name(priv->init.user_memarea)); fprintf(f, " total-size: 0x%lx\n", priv->init.total_sz); fprintf(f, " mt-safe: %s\n", priv->init.mt_safe ? "yes" : "no"); + if (priv->init.bak_memarea) + fprintf(f, " backup-memarea-name: %s\n", memarea_name(priv->init.bak_memarea)); fprintf(f, " total-regions: %u\n", memarea_elem_list_num(priv)); fprintf(f, " total-free-regions: %u\n", memarea_free_list_num(priv)); fprintf(f, " alloc_fails: %" PRIu64 "\n", priv->alloc_fails); fprintf(f, " refcnt_check_fails: %" PRIu64 "\n", priv->refcnt_check_fails); + fprintf(f, " backup_alloc_fails: %" PRIu64 "\n", priv->bak_alloc_fails); if (dump_all) memarea_dump_all(priv, f); memarea_unlock(priv); diff --git a/lib/memarea/rte_memarea.h b/lib/memarea/rte_memarea.h index d1383f18c3..a467627523 100644 --- a/lib/memarea/rte_memarea.h +++ b/lib/memarea/rte_memarea.h @@ -41,6 +41,8 @@ * specified, all the functions of the memarea API are lock-free, and * assume to not be invoked in parallel on different logical cores to work * on the same memarea object. + * f) It provides backup memory mechanism, the memarea object could use + * another memarea object as a backup. */ #include <stdbool.h> @@ -91,6 +93,10 @@ struct rte_memarea_param { */ struct rte_memarea *user_memarea; }; + /** Backup memarea, which is used to handle the scenario where the + * current memarea object allocation failure. + */ + struct rte_memarea *bak_memarea; }; /** -- 2.17.1