mirror of
https://github.com/libvirt/libvirt.git
synced 2025-02-25 18:55:26 -06:00
0dffca8f95
While virCommandSetDryRun is used in tests only, there were some cases when error paths would not call the function with NULL arguments to reset the dry run infrastructure. Introduce virCommandDryRunToken type which must be allocated via virCommandDryRunTokenNew and passed to virCommandSetDryRun. This way we can use automatic variable cleaning to trigger the cleanup of virCommandSetDryRun parameters and also the use of the token variable ensures that all callers of virCommandSetDryRun clean up after themselves and also that the token isn't left unused in the code. Signed-off-by: Peter Krempa <pkrempa@redhat.com> Reviewed-by: Ján Tomko <jtomko@redhat.com> Reviewed-by: Pavel Hrdina <phrdina@redhat.com>
134 lines
2.9 KiB
C
134 lines
2.9 KiB
C
/*
|
|
* Copyright (C) 2014 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"
|
|
|
|
#ifdef __linux__
|
|
|
|
# define LIBVIRT_VIRCOMMANDPRIV_H_ALLOW
|
|
# include "vircommandpriv.h"
|
|
# include "virkmod.h"
|
|
# include "virstring.h"
|
|
|
|
# define MODNAME "vfio-pci"
|
|
|
|
# define VIR_FROM_THIS VIR_FROM_NONE
|
|
|
|
static int
|
|
checkOutput(virBufferPtr buf, const char *exp_cmd)
|
|
{
|
|
int ret = -1;
|
|
char *actual_cmd = NULL;
|
|
|
|
if (!(actual_cmd = virBufferContentAndReset(buf))) {
|
|
fprintf(stderr, "cannot compare buffer to exp: %s", exp_cmd);
|
|
goto cleanup;
|
|
}
|
|
|
|
if (STRNEQ(exp_cmd, actual_cmd)) {
|
|
virTestDifference(stderr, exp_cmd, actual_cmd);
|
|
goto cleanup;
|
|
}
|
|
|
|
ret = 0;
|
|
|
|
cleanup:
|
|
VIR_FREE(actual_cmd);
|
|
return ret;
|
|
}
|
|
|
|
|
|
static int
|
|
testKModLoad(const void *args G_GNUC_UNUSED)
|
|
{
|
|
int ret = -1;
|
|
char *errbuf = NULL;
|
|
g_auto(virBuffer) buf = VIR_BUFFER_INITIALIZER;
|
|
g_autoptr(virCommandDryRunToken) dryRunToken = virCommandDryRunTokenNew();
|
|
|
|
virCommandSetDryRun(dryRunToken, &buf, NULL, NULL);
|
|
|
|
errbuf = virKModLoad(MODNAME);
|
|
if (errbuf) {
|
|
fprintf(stderr, "Failed to load, error: %s\n", errbuf);
|
|
goto cleanup;
|
|
}
|
|
|
|
if (checkOutput(&buf, MODPROBE " -b " MODNAME "\n") < 0)
|
|
goto cleanup;
|
|
|
|
ret = 0;
|
|
|
|
cleanup:
|
|
VIR_FREE(errbuf);
|
|
return ret;
|
|
}
|
|
|
|
|
|
static int
|
|
testKModUnload(const void *args G_GNUC_UNUSED)
|
|
{
|
|
int ret = -1;
|
|
char *errbuf = NULL;
|
|
g_auto(virBuffer) buf = VIR_BUFFER_INITIALIZER;
|
|
g_autoptr(virCommandDryRunToken) dryRunToken = virCommandDryRunTokenNew();
|
|
|
|
virCommandSetDryRun(dryRunToken, &buf, NULL, NULL);
|
|
|
|
errbuf = virKModUnload(MODNAME);
|
|
if (errbuf) {
|
|
fprintf(stderr, "Failed to unload, error: %s\n", errbuf);
|
|
goto cleanup;
|
|
}
|
|
|
|
if (checkOutput(&buf, RMMOD " " MODNAME "\n") < 0)
|
|
goto cleanup;
|
|
|
|
ret = 0;
|
|
|
|
cleanup:
|
|
VIR_FREE(errbuf);
|
|
return ret;
|
|
}
|
|
|
|
|
|
static int
|
|
mymain(void)
|
|
{
|
|
int ret = 0;
|
|
|
|
if (virTestRun("load", testKModLoad, NULL) < 0)
|
|
ret = -1;
|
|
if (virTestRun("unload", testKModUnload, NULL) < 0)
|
|
ret = -1;
|
|
|
|
return ret == 0 ? EXIT_SUCCESS : EXIT_FAILURE;
|
|
|
|
}
|
|
|
|
VIR_TEST_MAIN(mymain);
|
|
#else
|
|
int
|
|
main(void)
|
|
{
|
|
return EXIT_AM_SKIP;
|
|
}
|
|
#endif
|