Как узнать, хватит ли места на диске при копировании файла?

Question

[Павел]

Задача следующая: копируется файл, если место на диске закончилось, то остановить копирование и выдать предупреждение пользователю. необходима кроссплатформенная реализация (виндовс — линукс) на Qt.

видел эту функцию в интернете:

#ifdef _WIN32
    #include <windows.h>
#else // linux stuff
    #include <sys/vfs.h>
    #include <sys/stat.h>
#endif // _WIN32

bool getFreeTotalSpace(const QString& sDirPath,double& fTotal, double& fFree)
{
#ifdef _WIN32

    QString sCurDir = QDir::current().absPath();
    QDir::setCurrent( sDirPath );
    
    ULARGE_INTEGER free,total;
    bool bRes = ::GetDiskFreeSpaceExA( 0 , &free , &total , NULL );
    if ( !bRes ) return false;

    QDir::setCurrent( sCurDir );

    fFree = static_cast<double>( static_cast<__int64>(free.QuadPart) ) / fKB;
    fTotal = static_cast<double>( static_cast<__int64>(total.QuadPart) ) / fKB

#else //linux

    struct stat stst;
    struct statfs stfs;

    if ( ::stat(sDirPath.local8Bit(),&stst) == -1 ) return false;
    if ( ::statfs(sDirPath.local8Bit(),&stfs) == -1 ) return false;

    fFree = stfs.f_bavail * ( stst.st_blksize / fKB);
    fTotal = stfs.f_blocks * ( stst.st_blksize / fKB );

#endif // _WIN32

    return true;
}

но не могу в ней разобраться. подскажите как ее использовать. что именно непонятно: возвращается булевое значение — это ответ «места на диске нет\есть»? и что такое fKB.

заранее спасибо

Comments

[rasa]

Скобка выпала из строки " fFree = stfs.f_bavail * ( stst.st_blksize / fKB;"

Answer 1

[agmt]

Почему бы не заиспользовать posix_fallocate? Ну или более кроссовый QFile::seek(needSize-1); QFile::write("", 1);

Comments

[Павел]

не совсем понял что Вы этим добились?

Answer 2

[rasa]

Булевое — потому что функция на С++, результат возвращается в fTotal и в fFree
fKB = 1024

Comments

[Павел]

странное объяснения. функции на с++ могут возвращать не только булевые значения.

и я так понял fTotal надо задавать вручную? размер проверяемого диска?

[Павел]

ой, все, понял. спасибо.

[Павел]

то есть я к тому, что почему было бы не возвратить само значение значение? или структуру с полями fTotal и fFree? Ваш ответ «возвращает булевое потому что С++» мне не понятен

[rasa]

Я о выходах из функции через return — return true/return false.
В данном случае автор возвращает из своего кода то же, что и функция GetDiskFreeSpaceExA. Т.е. если ф-ция GetDiskFreeSpaceExA вернула ошибку — FluffyMan понимает, что нет смысла пытаться работать с fTotal и fFree

[Павел]

FluffyMan понял))

Answer 3

[vScherba]

Можно также стандартным std::ofstream ofs(имя_файла); ofs.seekp(нужный_размер); ofs << '\0';
Этот код, как и версия от agmt резервирует файл на диске нужного размера. Если места не хватит, обрабатывайте ошибку. Если ошибки не возникло, у Вас есть гарантированно выделенное место на диске, которое никто не займет.
В Вашем же примере, после определения достаточности места, другое приложение может заполнить диск, и об этом Вы узнаете только во время непосредственной записи в файл.

Comments

[vScherba]

Я не пользовался Qt, но мне кажется, что там должен быть готовый метод для копирования файла с резервированием места на диске, возможно реализованный на базе API для каждой ОС. Полагаю, никто не скопирует файл быстрее самой ОС.

[Павел]

возможно стандартная QFile::copy() и работает с резервированием места. но в перечне ошибок, который возвращает метод QFile::error() нет ошибки о том, что кончилось место.

[agmt]

, я не знаю системного вызова в linux для копирования файла. Хотя, вероятно, через sendfile (linux-only, начиная с ядра 2.6.33) будет быстрее. Но в какой момент он выдаст предупреждение о нехватке места — не знаю.