rtla/utils: Use calloc and check the potential memory allocation failure

[ Upstream commit b5f37a0b6f667f5c72340ca9dcd7703f261cb981 ]

Replace malloc with calloc and add memory allocating check
of mon_cpus before used.

Link: https://lkml.kernel.org/r/20220615073348.6891-1-jianchunfu@cmss.chinamobile.com
Fixes: 7d0dc9576dc3 ("rtla/timerlat: Add --dma-latency option")
Signed-off-by: jianchunfu <jianchunfu@cmss.chinamobile.com>
Acked-by: Daniel Bristot de Oliveira <bristot@kernel.org>
Signed-off-by: Steven Rostedt (Google) <rostedt@goodmis.org>
Signed-off-by: Sasha Levin <sashal@kernel.org>

diff --git a/tools/tracing/rtla/src/utils.c b/tools/tracing/rtla/src/utils.c
index 5352167a1e751d297d3b5888d0e6e4a205fc892a..5ae2fa96fde1e2fb2832e6735677790ef90364c3 100644 (file)
--- a/tools/tracing/rtla/src/utils.c
+++ b/tools/tracing/rtla/src/utils.c
@@ -106,8 +106,9 @@ int parse_cpu_list(char *cpu_list, char **monitored_cpus)
 
        nr_cpus = sysconf(_SC_NPROCESSORS_CONF);
 
-       mon_cpus = malloc(nr_cpus * sizeof(char));
-       memset(mon_cpus, 0, (nr_cpus * sizeof(char)));
+       mon_cpus = calloc(nr_cpus, sizeof(char));
+       if (!mon_cpus)
+               goto err;
 
        for (p = cpu_list; *p; ) {
                cpu = atoi(p);