lscpu: fix cpu map array sizes

Completely virtualized architectures like s390 may have multiple
virtual sockets and/or cores where each of them has a different
number of cores and cpus.
So the general assumption within the allocation scheme that e.g.
each socket contains the same number of cores is not necessarily
true. To make sure the arrays are always large enough we simply
allocate enough memory so that each array could hold cpu masks
for all present cpus.

Signed-off-by: Heiko Carstens <heiko.carstens@de.ibm.com>

diff --git a/sys-utils/lscpu.c b/sys-utils/lscpu.c
index feece73efd5028becbc494df9c689a06ce82ffb9..210a4c2431823bae5655792ddb83666442468f51 100644 (file)
--- a/sys-utils/lscpu.c
+++ b/sys-utils/lscpu.c
@@ -660,11 +660,16 @@ read_topology(struct lscpu_desc *desc, int num)
                 */
                if (!desc->nthreads)
                        desc->nthreads = nsockets * ncores * nthreads;
+               /* For each map we make sure that it can have up to ncpus
+                * entries. This is because we cannot reliably calculate the
+                * number of cores, sockets and books on all architectures.
+                * E.g. completely virtualized architectures like s390 may
+                * have multiple sockets of different sizes.
+                */
+               desc->coremaps = xcalloc(desc->ncpus, sizeof(cpu_set_t *));
+               desc->socketmaps = xcalloc(desc->ncpus, sizeof(cpu_set_t *));
                if (book_siblings)
-                       desc->bookmaps = xcalloc(nbooks, sizeof(cpu_set_t *));
-
-               desc->socketmaps = xcalloc(nsockets, sizeof(cpu_set_t *));
-               desc->coremaps = xcalloc(ncores * nsockets, sizeof(cpu_set_t *));
+                       desc->bookmaps = xcalloc(desc->ncpus, sizeof(cpu_set_t *));
        }
 
        add_cpuset_to_array(desc->socketmaps, &desc->nsockets, core_siblings);