util: vircommand: Add wrappers for virCommand error checking

Extract the check and reporting of error from the individual virCommand
APIs into a separate helper. This will aid future refactors.

Signed-off-by: Peter Krempa <pkrempa@redhat.com>
Reviewed-by: Ján Tomko <jtomko@redhat.com>
This commit is contained in:
Peter Krempa 2021-02-23 10:52:14 +01:00
parent ae87dc3d09
commit 047db95770

View File

@ -161,6 +161,27 @@ static void *dryRunOpaque;
static int dryRunStatus;
#endif /* !WIN32 */
static bool
virCommandHasError(virCommandPtr cmd)
{
return !cmd || cmd->has_error != 0;
}
static int
virCommandRaiseError(virCommandPtr cmd)
{
if (!cmd || cmd->has_error != 0) {
virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
_("invalid use of command API"));
return -1;
}
return 0;
}
/*
* virCommandFDIsSet:
* @cmd: pointer to virCommand
@ -982,7 +1003,7 @@ virCommandNewVAList(const char *binary, va_list list)
virCommandPtr cmd = virCommandNew(binary);
const char *arg;
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return cmd;
while ((arg = va_arg(list, const char *)) != NULL)
@ -1076,7 +1097,7 @@ virCommandPassFDGetFDIndex(virCommandPtr cmd, int fd)
{
size_t i = 0;
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return -1;
while (i < cmd->npassfd) {
@ -1100,7 +1121,7 @@ virCommandPassFDGetFDIndex(virCommandPtr cmd, int fd)
void
virCommandSetPidFile(virCommandPtr cmd, const char *pidfile)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
VIR_FREE(cmd->pidfile);
@ -1125,7 +1146,7 @@ virCommandGetUID(virCommandPtr cmd)
void
virCommandSetGID(virCommandPtr cmd, gid_t gid)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
cmd->gid = gid;
@ -1134,7 +1155,7 @@ virCommandSetGID(virCommandPtr cmd, gid_t gid)
void
virCommandSetUID(virCommandPtr cmd, uid_t uid)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
cmd->uid = uid;
@ -1143,7 +1164,7 @@ virCommandSetUID(virCommandPtr cmd, uid_t uid)
void
virCommandSetMaxMemLock(virCommandPtr cmd, unsigned long long bytes)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
cmd->maxMemLock = bytes;
@ -1152,7 +1173,7 @@ virCommandSetMaxMemLock(virCommandPtr cmd, unsigned long long bytes)
void
virCommandSetMaxProcesses(virCommandPtr cmd, unsigned int procs)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
cmd->maxProcesses = procs;
@ -1161,7 +1182,7 @@ virCommandSetMaxProcesses(virCommandPtr cmd, unsigned int procs)
void
virCommandSetMaxFiles(virCommandPtr cmd, unsigned int files)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
cmd->maxFiles = files;
@ -1169,7 +1190,7 @@ virCommandSetMaxFiles(virCommandPtr cmd, unsigned int files)
void virCommandSetMaxCoreSize(virCommandPtr cmd, unsigned long long bytes)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
cmd->maxCore = bytes;
@ -1178,7 +1199,7 @@ void virCommandSetMaxCoreSize(virCommandPtr cmd, unsigned long long bytes)
void virCommandSetUmask(virCommandPtr cmd, int mask)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
cmd->mask = mask;
@ -1193,7 +1214,7 @@ void virCommandSetUmask(virCommandPtr cmd, int mask)
void
virCommandClearCaps(virCommandPtr cmd)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
cmd->flags |= VIR_EXEC_CLEAR_CAPS;
@ -1210,7 +1231,7 @@ void
virCommandAllowCap(virCommandPtr cmd,
int capability)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
cmd->capabilities |= (1ULL << capability);
@ -1231,7 +1252,7 @@ void
virCommandSetSELinuxLabel(virCommandPtr cmd,
const char *label G_GNUC_UNUSED)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
#if defined(WITH_SECDRIVER_SELINUX)
@ -1255,7 +1276,7 @@ void
virCommandSetAppArmorProfile(virCommandPtr cmd,
const char *profile G_GNUC_UNUSED)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
#if defined(WITH_SECDRIVER_APPARMOR)
@ -1277,7 +1298,7 @@ virCommandSetAppArmorProfile(virCommandPtr cmd,
void
virCommandDaemonize(virCommandPtr cmd)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
cmd->flags |= VIR_EXEC_DAEMON;
@ -1293,7 +1314,7 @@ virCommandDaemonize(virCommandPtr cmd)
void
virCommandNonblockingFDs(virCommandPtr cmd)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
cmd->flags |= VIR_EXEC_NONBLOCK;
@ -1312,7 +1333,7 @@ virCommandNonblockingFDs(virCommandPtr cmd)
void
virCommandRawStatus(virCommandPtr cmd)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
cmd->rawStatus = true;
@ -1361,7 +1382,7 @@ virCommandAddEnvFormat(virCommandPtr cmd, const char *format, ...)
char *env;
va_list list;
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
va_start(list, format);
@ -1400,7 +1421,7 @@ virCommandAddEnvString(virCommandPtr cmd, const char *str)
{
char *env;
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
env = g_strdup(str);
@ -1447,7 +1468,7 @@ void
virCommandAddEnvPass(virCommandPtr cmd, const char *name)
{
const char *value;
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
value = getenv(name);
@ -1466,7 +1487,7 @@ virCommandAddEnvPass(virCommandPtr cmd, const char *name)
void
virCommandAddEnvPassCommon(virCommandPtr cmd)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
ignore_value(VIR_RESIZE_N(cmd->env, cmd->maxenv, cmd->nenv, 9));
@ -1486,7 +1507,7 @@ virCommandAddEnvPassCommon(virCommandPtr cmd)
void
virCommandAddEnvXDG(virCommandPtr cmd, const char *baseDir)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
ignore_value(VIR_RESIZE_N(cmd->env, cmd->maxenv, cmd->nenv, 3));
@ -1510,7 +1531,7 @@ virCommandAddEnvXDG(virCommandPtr cmd, const char *baseDir)
void
virCommandAddArg(virCommandPtr cmd, const char *val)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
if (val == NULL) {
@ -1538,7 +1559,7 @@ virCommandAddArgBuffer(virCommandPtr cmd, virBufferPtr buf)
{
g_autofree char *str = virBufferContentAndReset(buf);
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
if (!str)
@ -1566,7 +1587,7 @@ virCommandAddArgFormat(virCommandPtr cmd, const char *format, ...)
char *arg;
va_list list;
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
va_start(list, format);
@ -1609,7 +1630,7 @@ virCommandAddArgSet(virCommandPtr cmd, const char *const*vals)
{
int narg = 0;
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
if (vals[0] == NULL) {
@ -1645,7 +1666,7 @@ virCommandAddArgList(virCommandPtr cmd, ...)
va_list list;
int narg = 0;
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
va_start(list, cmd);
@ -1679,7 +1700,7 @@ virCommandAddArgList(virCommandPtr cmd, ...)
void
virCommandSetWorkingDirectory(virCommandPtr cmd, const char *pwd)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
if (cmd->pwd) {
@ -1727,7 +1748,7 @@ virCommandSetSendBuffer(virCommandPtr cmd,
{
size_t i;
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return -1;
if (fcntl(fd, F_SETFL, O_NONBLOCK) < 0) {
@ -1821,7 +1842,7 @@ virCommandSendBuffersHandlePoll(virCommandPtr cmd,
void
virCommandSetInputBuffer(virCommandPtr cmd, const char *inbuf)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
if (cmd->infd != -1 || cmd->inbuf) {
@ -1851,7 +1872,7 @@ void
virCommandSetOutputBuffer(virCommandPtr cmd, char **outbuf)
{
*outbuf = NULL;
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
if (cmd->outfdptr) {
@ -1885,7 +1906,7 @@ void
virCommandSetErrorBuffer(virCommandPtr cmd, char **errbuf)
{
*errbuf = NULL;
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
if (cmd->errfdptr) {
@ -1909,7 +1930,7 @@ virCommandSetErrorBuffer(virCommandPtr cmd, char **errbuf)
void
virCommandSetInputFD(virCommandPtr cmd, int infd)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
if (cmd->infd != -1 || cmd->inbuf) {
@ -1939,7 +1960,7 @@ virCommandSetInputFD(virCommandPtr cmd, int infd)
void
virCommandSetOutputFD(virCommandPtr cmd, int *outfd)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
if (cmd->outfdptr) {
@ -1965,7 +1986,7 @@ virCommandSetOutputFD(virCommandPtr cmd, int *outfd)
void
virCommandSetErrorFD(virCommandPtr cmd, int *errfd)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
if (cmd->errfdptr) {
@ -1994,7 +2015,7 @@ virCommandSetErrorFD(virCommandPtr cmd, int *errfd)
void
virCommandSetPreExecHook(virCommandPtr cmd, virExecHook hook, void *opaque)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
if (cmd->hook) {
@ -2025,7 +2046,7 @@ virCommandWriteArgLog(virCommandPtr cmd, int logfd)
/* Any errors will be reported later by virCommandRun, which means
* no command will be run, so there is nothing to log. */
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
for (i = 0; i < cmd->nenv; i++) {
@ -2069,9 +2090,8 @@ virCommandToString(virCommandPtr cmd, bool linebreaks)
/* Cannot assume virCommandRun will be called; so report the error
* now. If virCommandRun is called, it will report the same error. */
if (!cmd || cmd->has_error) {
virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
_("invalid use of command API"));
if (virCommandHasError(cmd)) {
virCommandRaiseError(cmd);
return NULL;
}
@ -2118,9 +2138,8 @@ virCommandGetArgList(virCommandPtr cmd,
{
size_t i;
if (cmd->has_error) {
virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
_("invalid use of command API"));
if (virCommandHasError(cmd)) {
virCommandRaiseError(cmd);
return -1;
}
@ -2305,9 +2324,8 @@ virCommandProcessIO(virCommandPtr cmd)
*/
int virCommandExec(virCommandPtr cmd, gid_t *groups, int ngroups)
{
if (!cmd || cmd->has_error) {
virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
_("invalid use of command API"));
if (virCommandHasError(cmd)) {
virCommandRaiseError(cmd);
return -1;
}
@ -2350,9 +2368,8 @@ virCommandRun(virCommandPtr cmd, int *exitstatus)
char *str;
int tmpfd;
if (!cmd || cmd->has_error) {
virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
_("invalid use of command API"));
if (virCommandHasError(cmd)) {
virCommandRaiseError(cmd);
return -1;
}
@ -2497,9 +2514,8 @@ virCommandRunAsync(virCommandPtr cmd, pid_t *pid)
bool synchronous = false;
int infd[2] = {-1, -1};
if (!cmd || cmd->has_error) {
virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
_("invalid use of command API"));
if (virCommandHasError(cmd)) {
virCommandRaiseError(cmd);
return -1;
}
@ -2646,9 +2662,8 @@ virCommandWait(virCommandPtr cmd, int *exitstatus)
int ret;
int status = 0;
if (!cmd || cmd->has_error) {
virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
_("invalid use of command API"));
if (virCommandHasError(cmd)) {
virCommandRaiseError(cmd);
return -1;
}
@ -2743,7 +2758,7 @@ virCommandAbort(virCommandPtr cmd)
*/
void virCommandRequireHandshake(virCommandPtr cmd)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
if (cmd->handshake) {
@ -2786,7 +2801,12 @@ int virCommandHandshakeWait(virCommandPtr cmd)
char c;
int rv;
if (!cmd || cmd->has_error || !cmd->handshake) {
if (virCommandHasError(cmd)) {
virCommandRaiseError(cmd);
return -1;
}
if (!cmd->handshake) {
virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
_("invalid use of command API"));
return -1;
@ -2844,7 +2864,12 @@ int virCommandHandshakeNotify(virCommandPtr cmd)
{
char c = '1';
if (!cmd || cmd->has_error || !cmd->handshake) {
if (virCommandHasError(cmd)) {
virCommandRaiseError(cmd);
return -1;
}
if (!cmd->handshake) {
virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
_("invalid use of command API"));
return -1;
@ -2872,7 +2897,7 @@ virCommandSetSendBuffer(virCommandPtr cmd,
unsigned char *buffer G_GNUC_UNUSED,
size_t buflen G_GNUC_UNUSED)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return -1;
cmd->has_error = ENOTSUP;
@ -2929,7 +2954,7 @@ virCommandAbort(virCommandPtr cmd G_GNUC_UNUSED)
void virCommandRequireHandshake(virCommandPtr cmd)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
cmd->has_error = ENOSYS;
@ -3056,7 +3081,7 @@ virCommandFree(virCommandPtr cmd)
void
virCommandDoAsyncIO(virCommandPtr cmd)
{
if (!cmd || cmd->has_error)
if (virCommandHasError(cmd))
return;
cmd->flags |= VIR_EXEC_ASYNC_IO | VIR_EXEC_NONBLOCK;