Mirror/zeek
1
0
Fork 0
mirror of https://github.com/zeek/zeek.git synced 2025-10-06 16:48:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Fix checking of fwrite return values

Browse source
This commit is contained in:
Johanna Amann 2014-10-28 07:20:26 -07:00
parent 832a2b7bab
commit ed73c83b61
1 changed files with 7 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/File.cc
View file

@ -708,10 +708,10 @@ void BroFile::InitEncrypt(const char* keyfile)
secret_len = htonl(secret_len);
if ( ! (fwrite("BROENC1", 7, 1, f) &&
fwrite(&secret_len, sizeof(secret_len), 1, f) &&
fwrite(secret, ntohl(secret_len), 1, f) &&
fwrite(iv, iv_len, 1, f)) )
if ( fwrite("BROENC1", 7, 1, f) < 7 ||
fwrite(&secret_len, sizeof(secret_len), 1, f) < sizeof(secret_len) ||
fwrite(secret, ntohl(secret_len), 1, f) < ntohl(secret_len) ||
fwrite(iv, iv_len, 1, f) < iv_len )
{
reporter->Error("can't write header to log file %s: %s",
name, strerror(errno));
@ -736,7 +736,7 @@ void BroFile::FinishEncrypt()
int outl;
EVP_SealFinal(cipher_ctx, cipher_buffer, &outl);
if ( outl && ! fwrite(cipher_buffer, outl, 1, f) )
if ( outl && fwrite(cipher_buffer, outl, 1, f) < outl )
{
reporter->Error("write error for %s: %s",
name, strerror(errno));
@ -777,7 +777,7 @@ int BroFile::Write(const char* data, int len)
return 0;
}
if ( outl && ! fwrite(cipher_buffer, outl, 1, f) )
if ( outl && fwrite(cipher_buffer, outl, 1, f) < outl )
{
reporter->Error("write error for %s: %s",
name, strerror(errno));
@ -792,8 +792,7 @@ int BroFile::Write(const char* data, int len)
return 1;
}
len = fwrite(data, 1, len, f);
if ( len <= 0 )
if ( fwrite(data, 1, len, f) < len )
return false;
if ( rotate_size && current_size < rotate_size && current_size + len >= rotate_size )