From: Fiona Trahe <fiona.tr...@intel.com>

Added functions to allocate and free compression operations.

Signed-off-by: Fiona Trahe <fiona.tr...@intel.com>
Signed-off-by: Pablo de Lara <pablo.de.lara.gua...@intel.com>
Signed-off-by: Shally Verma <shally.ve...@caviumnetworks.com>
Signed-off-by: Ashish Gupta <ashish.gu...@caviumnetworks.com>
---
 lib/librte_compressdev/rte_comp.h                  | 195 +++++++++++++++++++++
 lib/librte_compressdev/rte_compressdev.c           |  72 ++++++++
 lib/librte_compressdev/rte_compressdev_version.map |   1 +
 3 files changed, 268 insertions(+)

diff --git a/lib/librte_compressdev/rte_comp.h 
b/lib/librte_compressdev/rte_comp.h
index cf0f3c999..4cf84c5db 100644
--- a/lib/librte_compressdev/rte_comp.h
+++ b/lib/librte_compressdev/rte_comp.h
@@ -305,6 +305,201 @@ struct rte_comp_op {
         */
 } __rte_cache_aligned;
 
+
+/**
+ * Reset the fields of an operation to their default values.
+ *
+ * @note The private data associated with the operation is not zeroed.
+ *
+ * @param op
+ *   The operation to be reset
+ */
+static inline void
+__rte_comp_op_reset(struct rte_comp_op *op)
+{
+       struct rte_mempool *tmp_mp = op->mempool;
+       rte_iova_t tmp_iova_addr = op->iova_addr;
+
+       memset(op, 0, sizeof(struct rte_comp_op));
+       op->status = RTE_COMP_OP_STATUS_NOT_PROCESSED;
+       op->iova_addr = tmp_iova_addr;
+       op->mempool = tmp_mp;
+}
+
+/**
+ * Private data structure belonging to an operation pool.
+ */
+struct rte_comp_op_pool_private {
+       uint16_t user_size;
+       /**< Size of private user data with each operation. */
+};
+
+
+/**
+ * Returns the size of private user data allocated with each object in
+ * the mempool
+ *
+ * @param mempool
+ *   Mempool for operations
+ * @return
+ *   user data size
+ */
+static inline uint16_t
+__rte_comp_op_get_user_data_size(struct rte_mempool *mempool)
+{
+       struct rte_comp_op_pool_private *priv =
+           (struct rte_comp_op_pool_private *)rte_mempool_get_priv(mempool);
+
+       return priv->user_size;
+}
+
+
+/**
+ * Creates an operation pool
+ *
+ * @param name
+ *   Compress pool name
+ * @param nb_elts
+ *   Number of elements in pool
+ * @param cache_size
+ *   Number of elements to cache on lcore, see
+ *   *rte_mempool_create* for further details about cache size
+ * @param user_size
+ *   Size of private data to allocate for user with each operation
+ * @param socket_id
+ *   Socket to identifier allocate memory on
+ * @return
+ *  - On success pointer to mempool
+ *  - On failure NULL
+ */
+struct rte_mempool *
+rte_comp_op_pool_create(const char *name,
+               unsigned int nb_elts, unsigned int cache_size,
+               uint16_t user_size, int socket_id);
+
+/**
+ * Bulk allocate raw element from mempool and return as comp operations
+ *
+ * @param mempool
+ *   Compress operation mempool
+ * @param ops
+ *   Array to place allocated operations
+ * @param nb_ops
+ *   Number of operations to allocate
+ * @return
+ * - On success returns  number of ops allocated
+ */
+static inline int
+__rte_comp_op_raw_bulk_alloc(struct rte_mempool *mempool,
+               struct rte_comp_op **ops, uint16_t nb_ops)
+{
+       if (rte_mempool_get_bulk(mempool, (void **)ops, nb_ops) == 0)
+               return nb_ops;
+
+       return 0;
+}
+
+/**
+ * Allocate an operation from a mempool with default parameters set
+ *
+ * @param mempool
+ *   Compress operation mempool
+ *
+ * @return
+ * - On success returns a valid rte_comp_op structure
+ * - On failure returns NULL
+ */
+static inline struct rte_comp_op *
+rte_comp_op_alloc(struct rte_mempool *mempool)
+{
+       struct rte_comp_op *op = NULL;
+       int retval;
+
+       retval = __rte_comp_op_raw_bulk_alloc(mempool, &op, 1);
+       if (unlikely(retval != 1))
+               return NULL;
+
+       __rte_comp_op_reset(op);
+
+       return op;
+}
+
+
+/**
+ * Bulk allocate operations from a mempool with default parameters set
+ *
+ * @param mempool
+ *   Compress operation mempool
+ * @param ops
+ *   Array to place allocated operations
+ * @param nb_ops
+ *   Number of operations to allocate
+ * @return
+ * - nb_ops if the number of operations requested were allocated.
+ * - 0 if the requested number of ops are not available.
+ *   None are allocated in this case.
+ */
+static inline unsigned
+rte_comp_op_bulk_alloc(struct rte_mempool *mempool,
+               struct rte_comp_op **ops, uint16_t nb_ops)
+{
+       int i;
+
+       if (unlikely(__rte_comp_op_raw_bulk_alloc(mempool, ops, nb_ops)
+                       != nb_ops))
+               return 0;
+
+       for (i = 0; i < nb_ops; i++)
+               __rte_comp_op_reset(ops[i]);
+
+       return nb_ops;
+}
+
+
+
+/**
+ * Returns a pointer to the private user data of an operation if
+ * that operation has enough capacity for requested size.
+ *
+ * @param op
+ *   Compress operation
+ * @param size
+ *   Size of space requested in private data
+ * @return
+ * - if sufficient space available returns pointer to start of user data
+ * - if insufficient space returns NULL
+ */
+static inline void *
+__rte_comp_op_get_user_data(struct rte_comp_op *op, uint32_t size)
+{
+       uint32_t user_size;
+
+       if (likely(op->mempool != NULL)) {
+               user_size = __rte_comp_op_get_user_data_size(op->mempool);
+
+               if (likely(user_size >= size))
+                       return (void *)(op + 1);
+
+       }
+
+       return NULL;
+}
+
+/**
+ * free operation structure
+ * If operation has been allocate from a rte_mempool, then the operation will
+ * be returned to the mempool.
+ *
+ * @param op
+ *   Compress operation
+ */
+static inline void
+rte_comp_op_free(struct rte_comp_op *op)
+{
+       if (op != NULL && op->mempool != NULL)
+               rte_mempool_put(op->mempool, op);
+}
+
 #ifdef __cplusplus
 }
 #endif
diff --git a/lib/librte_compressdev/rte_compressdev.c 
b/lib/librte_compressdev/rte_compressdev.c
index c90e4beaf..0dab92650 100644
--- a/lib/librte_compressdev/rte_compressdev.c
+++ b/lib/librte_compressdev/rte_compressdev.c
@@ -530,6 +530,78 @@ rte_compressdev_info_get(uint8_t dev_id, struct 
rte_compressdev_info *dev_info)
        dev_info->driver_name = dev->device->driver->name;
 }
 
+/** Initialise rte_comp_op mempool element */
+static void
+rte_comp_op_init(struct rte_mempool *mempool,
+               __rte_unused void *opaque_arg,
+               void *_op_data,
+               __rte_unused unsigned int i)
+{
+       struct rte_comp_op *op = _op_data;
+
+       memset(_op_data, 0, mempool->elt_size);
+
+       op->status = RTE_COMP_OP_STATUS_NOT_PROCESSED;
+       op->iova_addr = rte_mem_virt2iova(_op_data);
+       op->mempool = mempool;
+}
+
+
+struct rte_mempool * __rte_experimental
+rte_comp_op_pool_create(const char *name,
+               unsigned int nb_elts, unsigned int cache_size,
+               uint16_t user_size, int socket_id)
+{
+       struct rte_comp_op_pool_private *priv;
+
+       unsigned int elt_size = sizeof(struct rte_comp_op) + user_size;
+
+       /* lookup mempool in case already allocated */
+       struct rte_mempool *mp = rte_mempool_lookup(name);
+
+       if (mp != NULL) {
+               priv = (struct rte_comp_op_pool_private *)
+                               rte_mempool_get_priv(mp);
+
+               if (mp->elt_size != elt_size ||
+                               mp->cache_size < cache_size ||
+                               mp->size < nb_elts ||
+                               priv->user_size <  user_size) {
+                       mp = NULL;
+                       COMPRESSDEV_LOG(ERR,
+               "Mempool %s already exists but with incompatible parameters",
+                                       name);
+                       return NULL;
+               }
+               return mp;
+       }
+
+       mp = rte_mempool_create(
+                       name,
+                       nb_elts,
+                       elt_size,
+                       cache_size,
+                       sizeof(struct rte_comp_op_pool_private),
+                       NULL,
+                       NULL,
+                       rte_comp_op_init,
+                       NULL,
+                       socket_id,
+                       0);
+
+       if (mp == NULL) {
+               COMPRESSDEV_LOG(ERR, "Failed to create mempool %s", name);
+               return NULL;
+       }
+
+       priv = (struct rte_comp_op_pool_private *)
+                       rte_mempool_get_priv(mp);
+
+       priv->user_size = user_size;
+
+       return mp;
+}
+
 TAILQ_HEAD(compressdev_driver_list, compressdev_driver);
 
 static struct compressdev_driver_list compressdev_driver_list =
diff --git a/lib/librte_compressdev/rte_compressdev_version.map 
b/lib/librte_compressdev/rte_compressdev_version.map
index 78ac8d904..6a3596429 100644
--- a/lib/librte_compressdev/rte_compressdev_version.map
+++ b/lib/librte_compressdev/rte_compressdev_version.map
@@ -28,6 +28,7 @@ EXPERIMENTAL {
        rte_compressdev_socket_id;
        rte_compressdev_start;
        rte_compressdev_stop;
+       rte_comp_op_pool_create;
 
         local: *;
 };
-- 
2.14.3

Reply via email to