[PATCH v1] graph: fix out of bounds access when re-allocate node objs

[Zhirun Yan <zhirun.yan@intel.com>]

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: 40d4f51403ec ("graph: implement fastpath routines")

Signed-off-by: Zhirun Yan <zhirun.yan@intel.com>
Signed-off-by: Liang, Cunming <cunming.liang@intel.com>
---
 lib/graph/rte_graph_worker.h | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/lib/graph/rte_graph_worker.h b/lib/graph/rte_graph_worker.h
index 0c0b9c095a..b7d145c3cb 100644
--- a/lib/graph/rte_graph_worker.h
+++ b/lib/graph/rte_graph_worker.h
@@ -218,13 +218,16 @@ static __rte_always_inline void
 __rte_node_enqueue_prologue(struct rte_graph *graph, struct rte_node *node,
 			    const uint16_t idx, const uint16_t space)
 {
+	uint32_t req_size;
 
 	/* Add to the pending stream list if the node is new */
 	if (idx == 0)
 		__rte_node_enqueue_tail_update(graph, node);
 
-	if (unlikely(node->size < (idx + space)))
-		__rte_node_stream_alloc(graph, node);
+	if (unlikely(node->size < (idx + space))) {
+		req_size = rte_align32pow2(node->size + space);
+		__rte_node_stream_alloc_size(graph, node, req_size);
+	}
 }
 
 /**
@@ -430,9 +433,12 @@ rte_node_next_stream_get(struct rte_graph *graph, struct rte_node *node,
 	node = __rte_node_next_node_get(node, next);
 	const uint16_t idx = node->idx;
 	uint16_t free_space = node->size - idx;
+	uint32_t req_size;
 
-	if (unlikely(free_space < nb_objs))
-		__rte_node_stream_alloc_size(graph, node, nb_objs);
+	if (unlikely(free_space < nb_objs)) {
+		req_size = rte_align32pow2(node->size + nb_objs);
+		__rte_node_stream_alloc_size(graph, node, req_size);
+	}
 
 	return &node->objs[idx];
 }
-- 
2.25.1