Add check for available disk space
This commit is contained in:
@@ -94,10 +94,20 @@ bool SocketConnectionReceiver::handlePut(Block<String> &commands)
|
||||
size_t fileLength;
|
||||
size_t totalBytesRead=0;
|
||||
size_t totalPacketsRead=0;
|
||||
size_t availableDiskSpace=0;
|
||||
|
||||
fileName = commands[1];
|
||||
fileLength = commands[commands.size()-1].toLong();
|
||||
std::cout << "PUT" << " " << fileName << " " << fileLength << std::endl;
|
||||
std::cout << "PUT" << " " << fileName << " " << Utility::byteCountToString(fileLength) << std::endl;
|
||||
|
||||
availableDiskSpace=Utility::getAvailableDiskSpace(String("./"));
|
||||
std::cout << Utility::byteCountToString(availableDiskSpace,false) << " available space on disk" << std::endl;
|
||||
if(fileLength > availableDiskSpace)
|
||||
{
|
||||
std::cout << "Unsufficient space on disk. Required space " << Utility::byteCountToString(availableDiskSpace,false) << " , available space" << Utility::byteCountToString(availableDiskSpace,false) << std::endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
FileIO writeFile(fileName, FileIO::ByteOrder::LittleEndian, FileIO::Mode::ReadWrite,FileIO::CreationFlags::CreateAlways);
|
||||
|
||||
if(!writeFile.isOkay())
|
||||
|
||||
Reference in New Issue
Block a user