12:51 a.m.
This test creates a Fake NUMA topology with non-sequential cell ids
to check if libvirt properly handles the same
Signed-off-by: Shivaprasad G Bhat <sbhat(a)linux.vnet.ibm.com>
Signed-off-by: Pradipta Kr. Banerjee <bpradip(a)in.ibm.com>
---
tests/Makefile.am | 5 ++
tests/vircapstest.c | 129 ++++++++++++++++++++++++++++++++++++++++++++++++++++
2 files changed, 134 insertions(+)
diff --git a/tests/Makefile.am b/tests/Makefile.am
index eb96f38..bb50ac5 100644
--- a/tests/Makefile.am
+++ b/tests/Makefile.am
@@ -144,6 +144,7 @@ test_programs = virshtest sockettest \
virstoragetest \
virnetdevbandwidthtest \
virkmodtest \
+ vircapstest \
$(NULL)
if WITH_REMOTE
@@ -672,6 +673,10 @@ virkmodtest_SOURCES = \
virkmodtest.c testutils.h testutils.c
virkmodtest_LDADD = $(LDADDS)
+vircapstest_SOURCES = \
+ vircapstest.c testutils.h testutils.c
+vircapstest_LDADD = $(LDADDS)
+
if WITH_LIBVIRTD
libvirtdconftest_SOURCES = \
libvirtdconftest.c testutils.h testutils.c \
diff --git a/tests/vircapstest.c b/tests/vircapstest.c
new file mode 100644
index 0000000..dab8f3b
--- /dev/null
+++ b/tests/vircapstest.c
@@ -0,0 +1,129 @@
+/*
+ * Copyright (C) IBM Corp 2014
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library. If not, see
+ * <http://www.gnu.org/licenses/>;.
+ *
+ */
+
+#include <config.h>
+#include <stdlib.h>
+
+#include "testutils.h"
+#include "capabilities.h"
+#include "virbitmap.h"
+
+
+#define VIR_FROM_THIS VIR_FROM_NONE
+
+#define MAX_CELLS 4
+#define MAX_CPUS_IN_CELL 2
+#define MAX_MEM_IN_CELL 2097152
+
+
+/*
+ * Build NUMA Toplogy with cell id starting from (0 + seq)
+ * for testing
+*/
+static virCapsPtr
+buildNUMATopology(int seq)
+{
+ virCapsPtr caps;
+ virCapsHostNUMACellCPUPtr cell_cpus;
+ int core_id, cell_id;
+ int id;
+
+ if ((caps = virCapabilitiesNew(VIR_ARCH_X86_64, 0, 0)) == NULL)
+ goto error;
+
+ id = 0;
+ for (cell_id = 0; cell_id < MAX_CELLS; cell_id++) {
+ if (VIR_ALLOC_N(cell_cpus, MAX_CPUS_IN_CELL) < 0)
+ goto error;
+
+ for (core_id = 0; core_id < MAX_CPUS_IN_CELL; core_id++) {
+ cell_cpus[core_id].id = id + core_id;
+ cell_cpus[core_id].socket_id = cell_id + seq;
+ cell_cpus[core_id].core_id = id + core_id;
+ if (!(cell_cpus[core_id].siblings =
+ virBitmapNew(MAX_CPUS_IN_CELL)))
+ goto error;
+ ignore_value(virBitmapSetBit(cell_cpus[core_id].siblings, id));
+ }
+ id++;
+
+ if (virCapabilitiesAddHostNUMACell(caps, cell_id + seq,
+ MAX_CPUS_IN_CELL,
+ MAX_MEM_IN_CELL,
+ cell_cpus) < 0)
+ goto error;
+
+ cell_cpus = NULL;
+ }
+
+ return caps;
+
+error:
+ virObjectUnref(caps);
+ return NULL;
+
+}
+
+
+static int
+test_virCapabilitiesGetCpusForNodemask(const void *data ATTRIBUTE_UNUSED)
+{
+ const char *nodestr = "3,4,5,6";
+ virBitmapPtr nodemask = NULL;
+ virBitmapPtr cpumap = NULL;
+ virCapsPtr caps;
+ int mask_size = 8;
+ int ret = -1;
+
+ //Build a NUMA topology with cell_id (NUMA node id
+ //being 3(0 + 3),4(1 + 3), 5 and 6
+ if (!(caps = buildNUMATopology(3)))
+ goto error;
+
+ if (virBitmapParse(nodestr, 0, &nodemask, mask_size) < 0)
+ goto error;
+
+ if (!(cpumap = virCapabilitiesGetCpusForNodemask(caps, nodemask)))
+ goto error;
+
+ ret = 0;
+
+error:
+ virBitmapFree(nodemask);
+ virBitmapFree(cpumap);
+ return ret;
+
+}
+
+
+static int
+mymain(void)
+{
+ int ret = 0;
+
+ if (virtTestRun("test_virCapabilitiesGetCpusForNodemask",
+ test_virCapabilitiesGetCpusForNodemask, NULL) < 0)
+ ret = -1;
+
+ return ret;
+
+
+}
+
+VIRT_TEST_MAIN(mymain)
--
1.8.5.3