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

@@ -53,6 +53,8 @@
"stop_token": "cpp",
"thread": "cpp",
"*.tpp": "cpp",
"csignal": "cpp"
"csignal": "cpp",
"codecvt": "cpp",
"iomanip": "cpp"
}
}

View File

@@ -15,8 +15,6 @@ void signalHandler(int signal); // The Control-C handler
SmartPointer<SSTP> sstp;
// ****************************************************************************************************************************************************************
/// @brief
/// @param argc
/// @param argv [0] program. [1] SERVERMODE {port} [2] CLIENTMODE {serveripaddress} {serverport} SEND {FileName}
@@ -25,7 +23,7 @@ int main(int argc, char **argv)
{
int returnCode(0);
Profiler profiler;
String version = "0.1.0.4"; // major version, minor version, patch, build
String version = "0.1.0.5"; // major version, minor version, patch, build
std::cout << "sstp version " << version.str() << std::endl;
if(!registerSignalHandler())

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