External/libvirt
1
0
Fork 0
mirror of https://gitlab.com/libvirt/libvirt.git synced 2025-02-22 03:12:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

virStream*All: Call virStreamAbort() more frequently

Browse Source 
Our documentation to the virStreamRecvAll, virStreamSendAll,
virStreamSparseRecvAll, and virStreamSparseSendAll functions
indicates that if these functions fail, then virStreamAbort is
called. But that is not necessarily true. For instance all of
these functions allocate a buffer to work with. If the allocation
fails, no virStreamAbort() is called despite -1 being returned.
It's the same story with argument sanity checks and a lot of
other checks.

Signed-off-by: Michal Privoznik <mprivozn@redhat.com>
Reviewed-by: John Ferlan <jferlan@redhat.com>
This commit is contained in:
Michal Privoznik 2017-06-01 08:20:31 +02:00
parent 0fe4aa149f
commit 6f8aa8e8da
1 changed files with 23 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
src/libvirt-stream.c
View File

@ -217,8 +217,10 @@ virStreamSend(virStreamPtr stream,
* while (1) {
* char buf[1024];
* int got = virStreamRecv(st, buf, 1024);
* if (got < 0)
* if (got < 0) {
* virStreamAbort(st);
* break;
* }
* if (got == 0) {
* virStreamFinish(st);
* break;
@ -596,10 +598,8 @@ virStreamSendAll(virStreamPtr stream,
for (;;) {
int got, offset = 0;
got = (handler)(stream, bytes, want, opaque);
if (got < 0) {
virStreamAbort(stream);
if (got < 0)
goto cleanup;
}
if (got == 0)
break;
while (offset < got) {
@ -615,8 +615,10 @@ virStreamSendAll(virStreamPtr stream,
cleanup:
VIR_FREE(bytes);
if (ret != 0)
if (ret != 0) {
virStreamAbort(stream);
virDispatchError(stream->conn);
}
return ret;
}
@ -728,21 +730,16 @@ int virStreamSparseSendAll(virStreamPtr stream,
const unsigned int skipFlags = 0;
if (!dataLen) {
if (holeHandler(stream, &inData, &sectionLen, opaque) < 0) {
virStreamAbort(stream);
if (holeHandler(stream, &inData, &sectionLen, opaque) < 0)
goto cleanup;
}
if (!inData && sectionLen) {
if (virStreamSendHole(stream, sectionLen, skipFlags) < 0) {
virStreamAbort(stream);
if (virStreamSendHole(stream, sectionLen, skipFlags) < 0)
goto cleanup;
}
if (skipHandler(stream, sectionLen, opaque) < 0) {
virReportSystemError(errno, "%s",
_("unable to skip hole"));
virStreamAbort(stream);
goto cleanup;
}
continue;
@ -755,10 +752,8 @@ int virStreamSparseSendAll(virStreamPtr stream,
want = dataLen;
got = (handler)(stream, bytes, want, opaque);
if (got < 0) {
virStreamAbort(stream);
if (got < 0)
goto cleanup;
}
if (got == 0)
break;
while (offset < got) {
@ -775,8 +770,10 @@ int virStreamSparseSendAll(virStreamPtr stream,
cleanup:
VIR_FREE(bytes);
if (ret != 0)
if (ret != 0) {
virStreamAbort(stream);
virDispatchError(stream->conn);
}
return ret;
}
@ -857,10 +854,8 @@ virStreamRecvAll(virStreamPtr stream,
while (offset < got) {
int done;
done = (handler)(stream, bytes + offset, got - offset, opaque);
if (done < 0) {
virStreamAbort(stream);
if (done < 0)
goto cleanup;
}
offset += done;
}
}
@ -869,8 +864,10 @@ virStreamRecvAll(virStreamPtr stream,
cleanup:
VIR_FREE(bytes);
if (ret != 0)
if (ret != 0) {
virStreamAbort(stream);
virDispatchError(stream->conn);
}
return ret;
}
@ -963,15 +960,11 @@ virStreamSparseRecvAll(virStreamPtr stream,
got = virStreamRecvFlags(stream, bytes, want, flags);
if (got == -3) {
if (virStreamRecvHole(stream, &holeLen, holeFlags) < 0) {
virStreamAbort(stream);
if (virStreamRecvHole(stream, &holeLen, holeFlags) < 0)
goto cleanup;
}
if (holeHandler(stream, holeLen, opaque) < 0) {
virStreamAbort(stream);
if (holeHandler(stream, holeLen, opaque) < 0)
goto cleanup;
}
continue;
} else if (got < 0) {
goto cleanup;
@ -981,10 +974,8 @@ virStreamSparseRecvAll(virStreamPtr stream,
while (offset < got) {
int done;
done = (handler)(stream, bytes + offset, got - offset, opaque);
if (done < 0) {
virStreamAbort(stream);
if (done < 0)
goto cleanup;
}
offset += done;
}
}
@ -993,8 +984,10 @@ virStreamSparseRecvAll(virStreamPtr stream,
cleanup:
VIR_FREE(bytes);
if (ret != 0)
if (ret != 0) {
virStreamAbort(stream);
virDispatchError(stream->conn);
}
return ret;
}