libvirt/tests/virnetdevopenvswitchtest.c

338 lines
12 KiB
C

/*
* Copyright (C) 2019 Red Hat, Inc.
*
* 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 "testutils.h"
#define LIBVIRT_VIRCOMMANDPRIV_H_ALLOW
#include "vircommandpriv.h"
#include "virnetdevbandwidth.h"
#include "virnetdevopenvswitch.h"
#include "netdev_bandwidth_conf.c"
#define VIR_FROM_THIS VIR_FROM_NONE
typedef struct _InterfaceParseStatsData InterfaceParseStatsData;
struct _InterfaceParseStatsData {
const char *filename;
const virDomainInterfaceStatsStruct stats;
};
struct testSetQosStruct {
const char *band;
const char *exp_cmd;
const char *iface;
};
struct testClearQosStruct {
const char *exp_cmd;
const char *iface;
const unsigned char *vmid;
};
static int
testVirNetDevBandwidthParse(virNetDevBandwidth **var,
const char *xml)
{
g_autoptr(xmlDoc) doc = NULL;
g_autoptr(xmlXPathContext) ctxt = NULL;
if (!xml)
return 0;
if (!(doc = virXMLParseStringCtxt((xml),
"bandwidth definition",
&ctxt)))
return -1;
return virNetDevBandwidthParse(var,
NULL,
ctxt->node,
true);
}
static const unsigned char vm_id[VIR_UUID_BUFLEN] = "fakeuuid";
static int
testInterfaceParseStats(const void *opaque)
{
const InterfaceParseStatsData *data = opaque;
g_autofree char *filename = NULL;
g_autofree char *buf = NULL;
virDomainInterfaceStatsStruct actual;
filename = g_strdup_printf("%s/virnetdevopenvswitchdata/%s", abs_srcdir,
data->filename);
if (virFileReadAll(filename, 1024, &buf) < 0)
return -1;
if (virNetDevOpenvswitchInterfaceParseStats(buf, &actual) < 0)
return -1;
if (memcmp(&actual, &data->stats, sizeof(actual)) != 0) {
fprintf(stderr,
"Expected stats: %lld %lld %lld %lld %lld %lld %lld %lld\n"
"Actual stats: %lld %lld %lld %lld %lld %lld %lld %lld",
data->stats.rx_bytes,
data->stats.rx_packets,
data->stats.rx_errs,
data->stats.rx_drop,
data->stats.tx_bytes,
data->stats.tx_packets,
data->stats.tx_errs,
data->stats.tx_drop,
actual.rx_bytes,
actual.rx_packets,
actual.rx_errs,
actual.rx_drop,
actual.tx_bytes,
actual.tx_packets,
actual.tx_errs,
actual.tx_drop);
return -1;
}
return 0;
}
typedef struct _escapeData escapeData;
struct _escapeData {
const char *input;
const char *expect;
};
static int
testNameEscape(const void *opaque)
{
const escapeData *data = opaque;
g_autofree char *reply = g_strdup(data->input);
int rv;
rv = virNetDevOpenvswitchMaybeUnescapeReply(reply);
if (data->expect) {
if (rv < 0 || STRNEQ(reply, data->expect)) {
fprintf(stderr,
"Unexpected failure, expected: %s for input %s got %s\n",
data->expect, data->input, reply);
return -1;
}
} else {
if (rv >= 0) {
fprintf(stderr,
"Unexpected success, input %s got %s\n",
data->input, reply);
return -1;
}
}
return 0;
}
static int
testVirNetDevOpenvswitchInterfaceSetQos(const void *data)
{
const struct testSetQosStruct *info = data;
const char *iface = info->iface;
g_autoptr(virNetDevBandwidth) band = NULL;
g_auto(virBuffer) buf = VIR_BUFFER_INITIALIZER;
g_autofree char *actual_cmd = NULL;
g_autoptr(virCommandDryRunToken) dryRunToken = virCommandDryRunTokenNew();
if (testVirNetDevBandwidthParse(&band, info->band) < 0)
return -1;
if (!iface)
iface = "tap-fake";
virCommandSetDryRun(dryRunToken, &buf, false, false, NULL, NULL);
if (virNetDevOpenvswitchInterfaceSetQos(iface, band, vm_id, true) < 0)
return -1;
if (!(actual_cmd = virBufferContentAndReset(&buf))) {
/* This is interesting, no command has been executed.
* Maybe that's expected, actually. */
}
if (STRNEQ_NULLABLE(info->exp_cmd, actual_cmd)) {
virTestDifference(stderr,
NULLSTR(info->exp_cmd),
NULLSTR(actual_cmd));
return -1;
}
return 0;
}
static int
testVirNetDevOpenvswitchInterfaceClearQos(const void *data)
{
const struct testClearQosStruct *info = data;
g_auto(virBuffer) buf = VIR_BUFFER_INITIALIZER;
g_autofree char *actual_cmd = NULL;
const char *iface = info->iface;
const unsigned char *vmid = info->vmid;
g_autoptr(virCommandDryRunToken) dryRunToken = virCommandDryRunTokenNew();
virCommandSetDryRun(dryRunToken, &buf, false, false, NULL, NULL);
if (virNetDevOpenvswitchInterfaceClearQos(iface, vmid) < 0)
return -1;
if (!(actual_cmd = virBufferContentAndReset(&buf))) {
/* This is interesting, no command has been executed.
* Maybe that's expected, actually. */
}
if (STRNEQ_NULLABLE(info->exp_cmd, actual_cmd)) {
virTestDifference(stderr,
NULLSTR(info->exp_cmd),
NULLSTR(actual_cmd));
return -1;
}
return 0;
}
static int
mymain(void)
{
int ret = 0;
#define TEST_INTERFACE_STATS(file, \
rxBytes, rxPackets, rxErrs, rxDrop, \
txBytes, txPackets, txErrs, txDrop) \
do { \
const InterfaceParseStatsData data = {.filename = file, .stats = { \
rxBytes, rxPackets, rxErrs, rxDrop, \
txBytes, txPackets, txErrs, txDrop}}; \
if (virTestRun("Interface stats " file, testInterfaceParseStats, &data) < 0) \
ret = -1; \
} while (0)
TEST_INTERFACE_STATS("stats1.json", 9, 12, 11, 10, 2, 8, 5, 4);
TEST_INTERFACE_STATS("stats2.json", 12406, 173, 0, 0, 0, 0, 0, 0);
#define TEST_NAME_ESCAPE(str, fail) \
do { \
const escapeData data = {str, fail};\
if (virTestRun("Name escape " str, testNameEscape, &data) < 0) \
ret = -1; \
} while (0)
TEST_NAME_ESCAPE("", "");
TEST_NAME_ESCAPE("\"\"", "");
TEST_NAME_ESCAPE("vhost-user1", "vhost-user1");
TEST_NAME_ESCAPE("\"vhost-user1\"", "vhost-user1");
TEST_NAME_ESCAPE("\"vhost_user-name.to.escape1", NULL);
TEST_NAME_ESCAPE("\"vhost_user-name.to\\\"escape1\"", "vhost_user-name.to\"escape1");
TEST_NAME_ESCAPE("\"vhost\"user1\"", NULL);
TEST_NAME_ESCAPE("\"\\\\", NULL);
#define DO_TEST_SET(Band, Exp_cmd, ...) \
do { \
struct testSetQosStruct data = {.band = Band, \
.exp_cmd = Exp_cmd, \
__VA_ARGS__}; \
if (virTestRun("virNetDevOpenvswitchInterfaceSetQos", \
testVirNetDevOpenvswitchInterfaceSetQos, \
&data) < 0) \
ret = -1; \
} while (0)
DO_TEST_SET(("<bandwidth>"
" <inbound average='20000'/>"
"</bandwidth>"),
(OVS_VSCTL " --timeout=5 --no-heading --columns=_uuid find queue"
" 'external-ids:vm-id=\"66616b65-7575-6964-0000-000000000000\"'"
" 'external-ids:ifname=\"tap-fake\"'\n"
OVS_VSCTL " --timeout=5 --no-heading --columns=_uuid find qos"
" 'external-ids:vm-id=\"66616b65-7575-6964-0000-000000000000\"'"
" 'external-ids:ifname=\"tap-fake\"'\n"
OVS_VSCTL " --timeout=5 set port tap-fake qos=@qos1"
" 'external-ids:vm-id=\"66616b65-7575-6964-0000-000000000000\"'"
" 'external-ids:ifname=\"tap-fake\"'"
" -- --id=@qos1 create qos type=linux-htb other_config:min-rate=160000000"
" queues:0=@queue0 'external-ids:vm-id=\"66616b65-7575-6964-0000-000000000000\"'"
" 'external-ids:ifname=\"tap-fake\"'"
" -- --id=@queue0 create queue other_config:min-rate=160000000 "
"'external-ids:vm-id=\"66616b65-7575-6964-0000-000000000000\"'"
" 'external-ids:ifname=\"tap-fake\"'\n"
OVS_VSCTL " --timeout=5 set Interface tap-fake ingress_policing_rate=0 ingress_policing_burst=0\n"));
DO_TEST_SET(NULL, NULL);
DO_TEST_SET("<bandwidth/>", NULL);
DO_TEST_SET(("<bandwidth>"
" <inbound average='0' />"
"</bandwidth>"),
(OVS_VSCTL " --timeout=5 --no-heading --columns=_uuid find queue"
" 'external-ids:vm-id=\"66616b65-7575-6964-0000-000000000000\"'"
" 'external-ids:ifname=\"tap-fake\"'\n"
OVS_VSCTL " --timeout=5 --no-heading --columns=_uuid find qos"
" 'external-ids:vm-id=\"66616b65-7575-6964-0000-000000000000\"'"
" 'external-ids:ifname=\"tap-fake\"'\n"
OVS_VSCTL " --timeout=5 set Interface tap-fake ingress_policing_rate=0 ingress_policing_burst=0\n"));
DO_TEST_SET(("<bandwidth>"
" <inbound average='0' />"
" <outbound average='5000' />"
"</bandwidth>"),
(OVS_VSCTL " --timeout=5 --no-heading --columns=_uuid find queue"
" 'external-ids:vm-id=\"66616b65-7575-6964-0000-000000000000\"'"
" 'external-ids:ifname=\"tap-fake\"'\n"
OVS_VSCTL " --timeout=5 --no-heading --columns=_uuid find qos"
" 'external-ids:vm-id=\"66616b65-7575-6964-0000-000000000000\"'"
" 'external-ids:ifname=\"tap-fake\"'\n"
OVS_VSCTL " --timeout=5 set Interface tap-fake ingress_policing_rate=40000\n"));
#define DO_TEST_CLEAR_QOS(Iface, Vmid, Exp_cmd, ...) \
do { \
struct testClearQosStruct data = {.iface = Iface, \
.vmid = Vmid, \
.exp_cmd = Exp_cmd, \
__VA_ARGS__}; \
if (virTestRun("virNetDevOpenvswitchInterfaceClearQos", \
testVirNetDevOpenvswitchInterfaceClearQos, \
&data) < 0) \
ret = -1; \
} while (0)
DO_TEST_CLEAR_QOS(("fake-iface"), vm_id,
(OVS_VSCTL " --timeout=5 --no-heading --columns=_uuid find queue"
" 'external-ids:vm-id=\"66616b65-7575-6964-0000-000000000000\"'"
" 'external-ids:ifname=\"fake-iface\"'\n"
OVS_VSCTL " --timeout=5 --no-heading --columns=_uuid find qos"
" 'external-ids:vm-id=\"66616b65-7575-6964-0000-000000000000\"'"
" 'external-ids:ifname=\"fake-iface\"'\n"
OVS_VSCTL " --timeout=5 set Interface fake-iface ingress_policing_rate=0 ingress_policing_burst=0\n"));
return ret == 0 ? EXIT_SUCCESS : EXIT_FAILURE;
}
VIR_TEST_MAIN_PRELOAD(mymain,
VIR_TEST_MOCK("virnetdevbandwidth"))