graph: fix node objects allocation

For __rte_node_enqueue_prologue(), if the number of objs is more than
the node->size * 2, the extra objs will write out of bounds memory.
It should use __rte_node_stream_alloc_size() to request enough memory.

And for rte_node_next_stream_put(), it will re-allocate a small size,
when the node free space is small and new objs is less than the current
node->size. Some objs pointers behind new size may be lost. And it will
cause memory leak. It should request enough size of memory, containing
the original objs and new objs at least.

Fixes: 40d4f51403 ("graph: implement fastpath routines")
Cc: stable@dpdk.org

Signed-off-by: Zhirun Yan <zhirun.yan@intel.com>
Signed-off-by: Cunming Liang <cunming.liang@intel.com>
Acked-by: Jerin Jacob <jerinj@marvell.com>
This commit is contained in:
Zhirun Yan 2022-08-04 14:02:41 +08:00 committed by Thomas Monjalon
parent 90cf759aaf
commit afe67d1414

View File

@ -224,7 +224,7 @@ __rte_node_enqueue_prologue(struct rte_graph *graph, struct rte_node *node,
__rte_node_enqueue_tail_update(graph, node);
if (unlikely(node->size < (idx + space)))
__rte_node_stream_alloc(graph, node);
__rte_node_stream_alloc_size(graph, node, node->size + space);
}
/**
@ -432,7 +432,7 @@ rte_node_next_stream_get(struct rte_graph *graph, struct rte_node *node,
uint16_t free_space = node->size - idx;
if (unlikely(free_space < nb_objs))
__rte_node_stream_alloc_size(graph, node, nb_objs);
__rte_node_stream_alloc_size(graph, node, node->size + nb_objs);
return &node->objs[idx];
}