From 6c3c1a4cb8e4973b8e603edfbe608e1b1587f3f2 Mon Sep 17 00:00:00 2001 From: Ivan Komissarov Date: Tue, 6 May 2014 16:10:10 +0000 Subject: Add the QStorageInfo class Allows to retrieve information about mounted volumes such as label, total/available size, filesystem type and so on. Possible use cases are: - allows to do checks about filesystem before performing actual operation (such as available/maximum volume size) - allows to retrive information about volume that can be shown in file dialogs - allows to retrieve volume for specific path and check if two or more paths belong to the same volume or not [ChangeLog][QtCore] Added QStorageInfo class to retrive information about mounted volumes and drives Change-Id: Ibf9c2e6b53ef39c5605894a4422acdbbca4030c4 Reviewed-by: Shawn Rutledge --- src/corelib/io/qstorageinfo_win.cpp | 193 ++++++++++++++++++++++++++++++++++++ 1 file changed, 193 insertions(+) create mode 100644 src/corelib/io/qstorageinfo_win.cpp (limited to 'src/corelib/io/qstorageinfo_win.cpp') diff --git a/src/corelib/io/qstorageinfo_win.cpp b/src/corelib/io/qstorageinfo_win.cpp new file mode 100644 index 0000000000..aa970778e6 --- /dev/null +++ b/src/corelib/io/qstorageinfo_win.cpp @@ -0,0 +1,193 @@ +/**************************************************************************** +** +** Copyright (C) 2014 Ivan Komissarov +** Contact: http://www.qt-project.org/legal +** +** This file is part of the QtCore module of the Qt Toolkit. +** +** $QT_BEGIN_LICENSE:LGPL$ +** Commercial License Usage +** Licensees holding valid commercial Qt licenses may use this file in +** accordance with the commercial license agreement provided with the +** Software or, alternatively, in accordance with the terms contained in +** a written agreement between you and Digia. For licensing terms and +** conditions see http://qt.digia.com/licensing. For further information +** use the contact form at http://qt.digia.com/contact-us. +** +** GNU Lesser General Public License Usage +** Alternatively, this file may be used under the terms of the GNU Lesser +** General Public License version 2.1 as published by the Free Software +** Foundation and appearing in the file LICENSE.LGPL included in the +** packaging of this file. Please review the following information to +** ensure the GNU Lesser General Public License version 2.1 requirements +** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html. +** +** In addition, as a special exception, Digia gives you certain additional +** rights. These rights are described in the Digia Qt LGPL Exception +** version 1.1, included in the file LGPL_EXCEPTION.txt in this package. +** +** GNU General Public License Usage +** Alternatively, this file may be used under the terms of the GNU +** General Public License version 3.0 as published by the Free Software +** Foundation and appearing in the file LICENSE.GPL included in the +** packaging of this file. Please review the following information to +** ensure the GNU General Public License version 3.0 requirements will be +** met: http://www.gnu.org/copyleft/gpl.html. +** +** +** $QT_END_LICENSE$ +** +****************************************************************************/ + +#include "qstorageinfo_p.h" + +#include +#include +#include + +#include + +QT_BEGIN_NAMESPACE + +static const int defaultBufferSize = MAX_PATH + 1; + +void QStorageInfoPrivate::initRootPath() +{ + rootPath = QFileInfo(rootPath).canonicalFilePath(); + + if (rootPath.isEmpty()) + return; + + QString path = QDir::toNativeSeparators(rootPath); + rootPath.clear(); + + if (path.startsWith(QStringLiteral("\\\\?\\"))) + path.remove(0, 4); + if (path.length() < 2 || path.at(1) != QLatin1Char(':')) + return; + path[0] = path[0].toUpper(); + if (!(path.at(0).unicode() >= 'A' && path.at(0).unicode() <= 'Z')) + return; + if (!path.endsWith(QLatin1Char('\\'))) + path.append(QLatin1Char('\\')); + + // ### test if disk mounted to folder on other disk + wchar_t buffer[defaultBufferSize]; + if (::GetVolumePathName(reinterpret_cast(path.utf16()), buffer, defaultBufferSize)) + rootPath = QDir::fromNativeSeparators(QString::fromWCharArray(buffer)); +} + +static inline QByteArray getDevice(const QString &rootPath) +{ + const QString path = QDir::toNativeSeparators(rootPath); + const UINT type = ::GetDriveType(reinterpret_cast(path.utf16())); + if (type == DRIVE_REMOTE) { + QVarLengthArray buffer(256); + DWORD bufferLength = buffer.size(); + DWORD result; + UNIVERSAL_NAME_INFO *remoteNameInfo; + do { + buffer.resize(bufferLength); + remoteNameInfo = reinterpret_cast(buffer.data()); + result = ::WNetGetUniversalName(reinterpret_cast(path.utf16()), + UNIVERSAL_NAME_INFO_LEVEL, + remoteNameInfo, + &bufferLength); + } while (result == ERROR_MORE_DATA); + if (result == NO_ERROR) + return QString::fromWCharArray(remoteNameInfo->lpUniversalName).toUtf8(); + return QByteArray(); + } + + wchar_t deviceBuffer[51]; + if (::GetVolumeNameForVolumeMountPoint(reinterpret_cast(path.utf16()), + deviceBuffer, + sizeof(deviceBuffer) / sizeof(wchar_t))) { + return QString::fromWCharArray(deviceBuffer).toLatin1(); + } + return QByteArray(); +} + +void QStorageInfoPrivate::doStat() +{ + initRootPath(); + if (rootPath.isEmpty()) + return; + + retreiveVolumeInfo(); + device = getDevice(rootPath); + retreiveDiskFreeSpace(); +} + +void QStorageInfoPrivate::retreiveVolumeInfo() +{ + const UINT oldmode = ::SetErrorMode(SEM_FAILCRITICALERRORS | SEM_NOOPENFILEERRORBOX); + + const QString path = QDir::toNativeSeparators(rootPath); + wchar_t nameBuffer[defaultBufferSize]; + wchar_t fileSystemTypeBuffer[defaultBufferSize]; + DWORD fileSystemFlags = 0; + const bool result = ::GetVolumeInformation(reinterpret_cast(path.utf16()), + nameBuffer, + defaultBufferSize, + Q_NULLPTR, + Q_NULLPTR, + &fileSystemFlags, + fileSystemTypeBuffer, + defaultBufferSize); + if (!result) { + ready = false; + valid = ::GetLastError() == ERROR_NOT_READY; + } else { + ready = true; + valid = true; + + fileSystemType = QString::fromWCharArray(fileSystemTypeBuffer).toLatin1(); + name = QString::fromWCharArray(nameBuffer); + + readOnly = (fileSystemFlags & FILE_READ_ONLY_VOLUME) != 0; + } + + ::SetErrorMode(oldmode); +} + +void QStorageInfoPrivate::retreiveDiskFreeSpace() +{ + const UINT oldmode = ::SetErrorMode(SEM_FAILCRITICALERRORS | SEM_NOOPENFILEERRORBOX); + + const QString path = QDir::toNativeSeparators(rootPath); + ::GetDiskFreeSpaceEx(reinterpret_cast(path.utf16()), + PULARGE_INTEGER(&bytesAvailable), + PULARGE_INTEGER(&bytesTotal), + PULARGE_INTEGER(&bytesFree)); + + ::SetErrorMode(oldmode); +} + +QList QStorageInfoPrivate::mountedVolumes() +{ + QList volumes; + + QString driveName = QStringLiteral("A:/"); + const UINT oldmode = ::SetErrorMode(SEM_FAILCRITICALERRORS | SEM_NOOPENFILEERRORBOX); + quint32 driveBits = quint32(::GetLogicalDrives()) & 0x3ffffff; + ::SetErrorMode(oldmode); + while (driveBits) { + if (driveBits & 1) { + QStorageInfo drive(driveName); + if (!drive.rootPath().isEmpty()) // drive exists, but not mounted + volumes.append(drive); + } + driveName[0] = driveName[0].unicode() + 1; + driveBits = driveBits >> 1; + } + + return volumes; +} + +QStorageInfo QStorageInfoPrivate::root() +{ + return QStorageInfo(QDir::fromNativeSeparators(QFile::decodeName(qgetenv("SystemDrive")))); +} + +QT_END_NAMESPACE -- cgit v1.2.3