From 743308e8828d36ce53fbf25a6fd815dad74b7d59 Mon Sep 17 00:00:00 2001 From: Dominik Riebeling Date: Tue, 26 Jul 2011 20:54:44 +0000 Subject: Take cluster size into account when calculating zip extracted size. Allow passing an (optional) cluster size to round up all file sizes when calculating the total size of an extracted zip archive. This allows to check if the space on disk is actually sufficient without relying on an arbitrary headroom value which might be wrong. Addresses FS#12195. git-svn-id: svn://svn.rockbox.org/rockbox/trunk@30214 a1c6a512-1295-4272-9138-f99709370657 --- rbutil/rbutilqt/base/utils.cpp | 31 ++++++++++++++++++++++++++++--- rbutil/rbutilqt/base/utils.h | 4 +++- rbutil/rbutilqt/base/zipinstaller.cpp | 3 ++- rbutil/rbutilqt/base/ziputil.cpp | 21 +++++++++++++++++---- rbutil/rbutilqt/base/ziputil.h | 2 +- 5 files changed, 51 insertions(+), 10 deletions(-) diff --git a/rbutil/rbutilqt/base/utils.cpp b/rbutil/rbutilqt/base/utils.cpp index 1fdf627378..ca3a6d2256 100644 --- a/rbutil/rbutilqt/base/utils.cpp +++ b/rbutil/rbutilqt/base/utils.cpp @@ -187,13 +187,25 @@ QString Utils::filesystemName(QString path) //! @return size in bytes qulonglong Utils::filesystemFree(QString path) { - return filesystemSize(path, FilesystemFree); + qulonglong size = filesystemSize(path, FilesystemFree); + qDebug() << "[Utils] free disk space for" << path << size; + return size; } qulonglong Utils::filesystemTotal(QString path) { - return filesystemSize(path, FilesystemTotal); + qulonglong size = filesystemSize(path, FilesystemTotal); + qDebug() << "[Utils] total disk space for" << path << size; + return size; +} + + +qulonglong Utils::filesystemClusterSize(QString path) +{ + qulonglong size = filesystemSize(path, FilesystemClusterSize); + qDebug() << "[Utils] cluster size for" << path << size; + return size; } @@ -214,6 +226,9 @@ qulonglong Utils::filesystemSize(QString path, enum Utils::Size type) if(type == FilesystemTotal) { size = (qulonglong)fs.f_frsize * (qulonglong)fs.f_blocks; } + if(type == FilesystemClusterSize) { + size = (qulonglong)fs.f_frsize; + } } #endif #if defined(Q_OS_WIN32) @@ -230,9 +245,19 @@ qulonglong Utils::filesystemSize(QString path, enum Utils::Size type) if(type == FilesystemTotal) { size = totalNumberBytes.QuadPart; } + if(type == FilesystemClusterSize) { + DWORD sectorsPerCluster; + DWORD bytesPerSector; + DWORD freeClusters; + DWORD totalClusters; + ret = GetDiskFreeSpaceW((LPCTSTR)path.utf16(), §orsPerCluster, + &bytesPerSector, &freeClusters, &totalClusters); + if(ret) { + size = bytesPerSector * sectorsPerCluster; + } + } } #endif - qDebug() << "[Utils] Filesystem:" << path << size; return size; } diff --git a/rbutil/rbutilqt/base/utils.h b/rbutil/rbutilqt/base/utils.h index 659bbc4ed4..b4477699fd 100644 --- a/rbutil/rbutilqt/base/utils.h +++ b/rbutil/rbutilqt/base/utils.h @@ -33,13 +33,15 @@ class Utils : public QObject public: enum Size { FilesystemTotal, - FilesystemFree + FilesystemFree, + FilesystemClusterSize, }; static bool recursiveRmdir(const QString &dirName); static QString resolvePathCase(QString path); static qulonglong filesystemFree(QString path); static qulonglong filesystemTotal(QString path); + static qulonglong filesystemClusterSize(QString path); static qulonglong filesystemSize(QString path, enum Size type); static QString findExecutable(QString name); static QString checkEnvironment(bool permission); diff --git a/rbutil/rbutilqt/base/zipinstaller.cpp b/rbutil/rbutilqt/base/zipinstaller.cpp index 76f673ff81..c450f3015f 100644 --- a/rbutil/rbutilqt/base/zipinstaller.cpp +++ b/rbutil/rbutilqt/base/zipinstaller.cpp @@ -140,7 +140,8 @@ void ZipInstaller::downloadDone(bool error) // some room for operating (also includes calculation mistakes due to // cluster sizes on the player). if((qint64)Utils::filesystemFree(m_mountpoint) - < (zip.totalUncompressedSize() + 1000000)) { + < (zip.totalUncompressedSize(Utils::filesystemClusterSize(m_mountpoint)) + + 1000000)) { emit logItem(tr("Not enough disk space! Aborting."), LOGERROR); emit logProgress(1, 1); emit done(true); diff --git a/rbutil/rbutilqt/base/ziputil.cpp b/rbutil/rbutilqt/base/ziputil.cpp index ed8f17eefe..4396fa5e87 100644 --- a/rbutil/rbutilqt/base/ziputil.cpp +++ b/rbutil/rbutilqt/base/ziputil.cpp @@ -205,7 +205,7 @@ bool ZipUtil::appendFileToArchive(QString& file, QString& basedir) //! @brief calculate total size of extracted files -qint64 ZipUtil::totalUncompressedSize(void) +qint64 ZipUtil::totalUncompressedSize(unsigned int clustersize) { qint64 uncompressed = 0; @@ -214,10 +214,23 @@ qint64 ZipUtil::totalUncompressedSize(void) return -1; } int max = items.size(); - for(int i = 0; i < max; ++i) { - uncompressed += items.at(i).uncompressedSize; + if(clustersize > 0) { + for(int i = 0; i < max; ++i) { + qint64 item = items.at(i).uncompressedSize; + uncompressed += (item + clustersize - (item % clustersize)); + } + } + else { + for(int i = 0; i < max; ++i) { + uncompressed += items.at(i).uncompressedSize; + } + } + if(clustersize > 0) { + qDebug() << "[ZipUtil] calculation rounded to cluster size for each file:" + << clustersize; } - qDebug() << "[ZipUtil] size of archive files uncompressed:" << uncompressed; + qDebug() << "[ZipUtil] size of archive files uncompressed:" + << uncompressed; return uncompressed; } diff --git a/rbutil/rbutilqt/base/ziputil.h b/rbutil/rbutilqt/base/ziputil.h index cfafb96566..dc1b986c29 100644 --- a/rbutil/rbutilqt/base/ziputil.h +++ b/rbutil/rbutilqt/base/ziputil.h @@ -37,7 +37,7 @@ class ZipUtil : public QObject bool extractArchive(QString& dest); bool appendDirToArchive(QString& source, QString& basedir); bool appendFileToArchive(QString& file, QString& basedir); - qint64 totalUncompressedSize(void); + qint64 totalUncompressedSize(unsigned int clustersize = 0); QStringList files(void); signals: -- cgit v1.2.3