Add check for available disk space
This commit is contained in:
4
sstp/.vscode/settings.json
vendored
4
sstp/.vscode/settings.json
vendored
@@ -53,6 +53,8 @@
|
||||
"stop_token": "cpp",
|
||||
"thread": "cpp",
|
||||
"*.tpp": "cpp",
|
||||
"csignal": "cpp"
|
||||
"csignal": "cpp",
|
||||
"codecvt": "cpp",
|
||||
"iomanip": "cpp"
|
||||
}
|
||||
}
|
||||
@@ -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())
|
||||
|
||||
@@ -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