nvme/streams: avoid allocating too large a buffer

vincentkfu · vincentkfu · commit 8ac0eec77e9c · 2024-08-05T18:15:24.000Z
We only need to allocate as many data structures as there were stream
IDs provided.

Signed-off-by: Vincent Fu &lt;vincent.fu@samsung.com&gt;
diff --git a/dataplacement.c b/dataplacement.c
@@ -52,7 +52,7 @@ static int init_ruh_info(struct thread_data *td, struct fio_file *f)
 			log_err("fio: stream IDs must be provided for dataplacement=streams\n");
 			return -EINVAL;
 		}
-		ruhs = scalloc(1, sizeof(*ruhs) + FIO_MAX_DP_IDS * sizeof(*ruhs->plis));
+		ruhs = scalloc(1, sizeof(*ruhs) + td->o.dp_nr_ids * sizeof(*ruhs->plis));
 		if (!ruhs)
 			return -ENOMEM;
 

Original file line number	Diff line number	Diff line change
`@@ -52,7 +52,7 @@ static int init_ruh_info(struct thread_data td, struct fio_file f)`
`52`	`52`	`log_err("fio: stream IDs must be provided for dataplacement=streams\n");`
`53`	`53`	`return -EINVAL;`
`54`	`54`	`}`
`55`		`- ruhs = scalloc(1, sizeof(ruhs) + FIO_MAX_DP_IDS sizeof(*ruhs->plis));`
	`55`	`+ ruhs = scalloc(1, sizeof(ruhs) + td->o.dp_nr_ids sizeof(*ruhs->plis));`
`56`	`56`	`if (!ruhs)`
`57`	`57`	`return -ENOMEM;`
`58`	`58`