Add check for available disk space

This commit is contained in:
2025-08-17 10:08:49 -04:00
parent 2e4a713b1c
commit f7facd3319
4 changed files with 37 additions and 5 deletions

View File

@@ -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())