Aren/haiku
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

FileRequest: Better error handling for size mismatches

Browse Source
This commit is contained in:
Stephan Aßmus 2014-01-23 10:19:24 +01:00
parent 9e751dda2e
commit d5c2ed6b14
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/kits/network/libnetapi/FileRequest.cpp
View File

@ -65,8 +65,11 @@ BFileRequest::_ProtocolLoop()
// Send all notifications to listener, if any
if (fListener != NULL) {
fListener->ConnectionOpened(this);
off_t size = 0;
file.GetSize(&size);
error = file.GetSize(&size);
if (error != B_OK)
return error;
fResult.SetLength(size);
ssize_t chunkSize;
@ -75,9 +78,13 @@ BFileRequest::_ProtocolLoop()
fListener->DataReceived(this, chunk, chunkSize);
transferredSize += chunkSize;
}
// Return last error if we didn't transfer everything
if (transferredSize != size)
return chunkSize;
// Return error if we didn't transfer everything
if (transferredSize != size) {
if (chunkSize < 0)
return (status_t)chunkSize;
else
return B_IO_ERROR;
}
fListener->DownloadProgress(this, size, size);
}