mirrors/libguestfs
1
0
Fork 0
mirror of https://github.com/libguestfs/libguestfs.git synced 2026-03-21 22:53:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Carefully check return values from xwrite.

Browse Source
This commit is contained in:
Richard Jones
2009-04-26 09:51:30 +01:00
parent e65e1d5d00
commit 38f9222d21
1 changed files with 18 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
daemon/proto.c
View File

@@ -79,6 +79,7 @@ main_loop (int _sock)
xread (sock, buf, len);
#if 0
if (verbose) {
int i, j;
@@ -99,6 +100,7 @@ main_loop (int _sock)
printf ("|\n");
}
}
#endif
/* Decode the message header. */
xdrmem_create (&xdr, buf, len, XDR_DECODE);
@@ -209,8 +211,14 @@ send_error (const char *msg)
xdr_uint32_t (&xdr, &len);
xdr_destroy (&xdr);
(void) xwrite (sock, lenbuf, 4);
(void) xwrite (sock, buf, len);
if (xwrite (sock, lenbuf, 4) == -1) {
fprintf (stderr, "xwrite failed\n");
exit (1);
}
if (xwrite (sock, buf, len) == -1) {
fprintf (stderr, "xwrite failed\n");
exit (1);
}
}
void
@@ -250,8 +258,14 @@ reply (xdrproc_t xdrp, char *ret)
xdr_uint32_t (&xdr, &len);
xdr_destroy (&xdr);
(void) xwrite (sock, lenbuf, 4);
(void) xwrite (sock, buf, len);
if (xwrite (sock, lenbuf, 4) == -1) {
fprintf (stderr, "xwrite failed\n");
exit (1);
}
if (xwrite (sock, buf, len) == len) {
fprintf (stderr, "xwrite failed\n");
exit (1);
}
}
/* Receive file chunks, repeatedly calling 'cb'. */