diff mbox series

[conntrack-tools,v2,1/3] conntrackd: prevent memory loss if reallocation fails

Message ID 20240302160802.7309-2-donald.yandt@gmail.com
State Accepted
Headers show
Series fix potential memory loss and exit codes | expand

Commit Message

Donald Yandt March 2, 2024, 4:08 p.m. UTC 
Signed-off-by: Donald Yandt <donald.yandt@gmail.com>
---
 src/vector.c | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)
diff mbox series

Patch

diff --git a/src/vector.c b/src/vector.c
index c81e7ce..0af8db7 100644
--- a/src/vector.c
+++ b/src/vector.c
@@ -60,13 +60,16 @@  void vector_destroy(struct vector *v)
 
 int vector_add(struct vector *v, void *data)
 {
+	void *ptr;
+
 	if (v->cur_elems >= v->max_elems) {
 		v->max_elems += DEFAULT_VECTOR_GROWTH;
-		v->data = realloc(v->data, v->max_elems * v->size);
-		if (v->data == NULL) {
+		ptr = realloc(v->data, v->max_elems * v->size);
+		if (ptr == NULL) {
 			v->max_elems -= DEFAULT_VECTOR_GROWTH;
 			return -1;
 		}
+		v->data = ptr;
 	}
 	memcpy(v->data + (v->size * v->cur_elems), data, v->size);
 	v->cur_elems++;