mirror of
https://github.com/qbittorrent/qBittorrent.git
synced 2024-10-22 18:56:08 +03:00
531 lines
16 KiB
C++
531 lines
16 KiB
C++
/*
|
|
* Bittorrent Client using Qt4 and libtorrent.
|
|
* Copyright (C) 2012 Christophe Dumez
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License
|
|
* as published by the Free Software Foundation; either version 2
|
|
* of the License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
*
|
|
* In addition, as a special exception, the copyright holders give permission to
|
|
* link this program with the OpenSSL project's "OpenSSL" library (or with
|
|
* modified versions of it that use the same license as the "OpenSSL" library),
|
|
* and distribute the linked executables. You must obey the GNU General Public
|
|
* License in all respects for all of the code used other than "OpenSSL". If you
|
|
* modify file(s), you may extend this exception to your version of the file(s),
|
|
* but you are not obligated to do so. If you do not wish to do so, delete this
|
|
* exception statement from your version.
|
|
*
|
|
* Contact : chris@qbittorrent.org
|
|
*/
|
|
|
|
#include "fs_utils.h"
|
|
#include "misc.h"
|
|
|
|
#include <QDebug>
|
|
#include <QDir>
|
|
#include <QFile>
|
|
#include <QFileInfo>
|
|
#include <QSettings>
|
|
#ifdef DISABLE_GUI
|
|
#include <QCoreApplication>
|
|
#else
|
|
#include <QApplication>
|
|
#endif
|
|
#include <libtorrent/torrent_info.hpp>
|
|
|
|
#ifdef Q_OS_MAC
|
|
#include <CoreServices/CoreServices.h>
|
|
#include <Carbon/Carbon.h>
|
|
#endif
|
|
|
|
#ifndef Q_OS_WIN
|
|
#if defined(Q_OS_MAC) || defined(Q_OS_FREEBSD)
|
|
#include <sys/param.h>
|
|
#include <sys/mount.h>
|
|
#elif defined(Q_OS_HAIKU)
|
|
#include <kernel/fs_info.h>
|
|
#else
|
|
#include <sys/vfs.h>
|
|
#endif
|
|
#else
|
|
#include <shlobj.h>
|
|
#include <winbase.h>
|
|
#endif
|
|
|
|
#if defined(Q_OS_WIN) || defined(Q_OS_OS2)
|
|
#if (QT_VERSION < QT_VERSION_CHECK(5, 0, 0))
|
|
#include <QDesktopServices>
|
|
#else
|
|
#include <QStandardPaths>
|
|
#endif
|
|
#endif
|
|
|
|
using namespace libtorrent;
|
|
|
|
/**
|
|
* Converts a path to a string suitable for display.
|
|
* This function makes sure the directory separator used is consistent
|
|
* with the OS being run.
|
|
*/
|
|
QString fsutils::toNativePath(const QString& path) {
|
|
return QDir::toNativeSeparators(path);
|
|
}
|
|
|
|
QString fsutils::fromNativePath(const QString &path) {
|
|
return QDir::fromNativeSeparators(path);
|
|
}
|
|
|
|
/**
|
|
* Returns the file extension part of a file name.
|
|
*/
|
|
QString fsutils::fileExtension(const QString &filename) {
|
|
QString ext = QString(filename).remove(".!qB");
|
|
const int point_index = ext.lastIndexOf(".");
|
|
return (point_index >= 0) ? ext.mid(point_index + 1) : QString();
|
|
}
|
|
|
|
QString fsutils::fileName(const QString& file_path) {
|
|
QString path = fsutils::fromNativePath(file_path);
|
|
const int slash_index = path.lastIndexOf("/");
|
|
if (slash_index == -1)
|
|
return path;
|
|
return path.mid(slash_index + 1);
|
|
}
|
|
|
|
QString fsutils::folderName(const QString& file_path) {
|
|
QString path = fsutils::fromNativePath(file_path);
|
|
const int slash_index = path.lastIndexOf("/");
|
|
if (slash_index == -1)
|
|
return path;
|
|
return path.left(slash_index);
|
|
}
|
|
|
|
bool fsutils::isValidTorrentFile(const QString& torrent_path) {
|
|
try {
|
|
boost::intrusive_ptr<libtorrent::torrent_info> t = new torrent_info(fsutils::toNativePath(torrent_path).toUtf8().constData());
|
|
if (!t->is_valid() || t->num_files() == 0)
|
|
return false;
|
|
} catch(std::exception&) {
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
* Remove an empty folder tree.
|
|
*
|
|
* This function will also remove .DS_Store files on Mac OS and
|
|
* Thumbs.db on Windows.
|
|
*/
|
|
bool fsutils::smartRemoveEmptyFolderTree(const QString& dir_path) {
|
|
qDebug() << Q_FUNC_INFO << dir_path;
|
|
if (dir_path.isEmpty())
|
|
return false;
|
|
|
|
QDir dir(dir_path);
|
|
if (!dir.exists())
|
|
return true;
|
|
|
|
// Remove Files created by the OS
|
|
#if defined Q_OS_MAC
|
|
fsutils::forceRemove(dir_path + QLatin1String("/.DS_Store"));
|
|
#elif defined Q_OS_WIN
|
|
fsutils::forceRemove(dir_path + QLatin1String("/Thumbs.db"));
|
|
#endif
|
|
|
|
QFileInfoList sub_files = dir.entryInfoList();
|
|
foreach (const QFileInfo& info, sub_files) {
|
|
QString sub_name = info.fileName();
|
|
if (sub_name == "." || sub_name == "..")
|
|
continue;
|
|
|
|
QString sub_path = info.absoluteFilePath();
|
|
qDebug() << Q_FUNC_INFO << "sub file: " << sub_path;
|
|
if (info.isDir()) {
|
|
if (!smartRemoveEmptyFolderTree(sub_path)) {
|
|
qWarning() << Q_FUNC_INFO << "Failed to remove folder: " << sub_path;
|
|
return false;
|
|
}
|
|
} else {
|
|
if (info.isHidden()) {
|
|
qDebug() << Q_FUNC_INFO << "Removing hidden file: " << sub_path;
|
|
if (!fsutils::forceRemove(sub_path)) {
|
|
qWarning() << Q_FUNC_INFO << "Failed to remove " << sub_path;
|
|
return false;
|
|
}
|
|
} else {
|
|
qWarning() << Q_FUNC_INFO << "Folder is not empty, aborting. Found: " << sub_path;
|
|
}
|
|
}
|
|
}
|
|
qDebug() << Q_FUNC_INFO << "Calling rmdir on " << dir_path;
|
|
return QDir().rmdir(dir_path);
|
|
}
|
|
|
|
/**
|
|
* Removes the file with the given file_path.
|
|
*
|
|
* This function will try to fix the file permissions before removing it.
|
|
*/
|
|
bool fsutils::forceRemove(const QString& file_path) {
|
|
QFile f(file_path);
|
|
if (!f.exists())
|
|
return true;
|
|
// Make sure we have read/write permissions
|
|
f.setPermissions(f.permissions()|QFile::ReadOwner|QFile::WriteOwner|QFile::ReadUser|QFile::WriteUser);
|
|
// Remove the file
|
|
return f.remove();
|
|
}
|
|
|
|
/**
|
|
* Returns the size of a file.
|
|
* If the file is a folder, it will compute its size based on its content.
|
|
*
|
|
* Returns -1 in case of error.
|
|
*/
|
|
qint64 fsutils::computePathSize(const QString& path) {
|
|
// Check if it is a file
|
|
QFileInfo fi(path);
|
|
if (!fi.exists()) return -1;
|
|
if (fi.isFile()) return fi.size();
|
|
// Compute folder size based on its content
|
|
qint64 size = 0;
|
|
foreach (const QFileInfo &subfi, QDir(path).entryInfoList(QDir::Dirs|QDir::Files)) {
|
|
if (subfi.fileName().startsWith(".")) continue;
|
|
if (subfi.isDir())
|
|
size += fsutils::computePathSize(subfi.absoluteFilePath());
|
|
else
|
|
size += subfi.size();
|
|
}
|
|
return size;
|
|
}
|
|
|
|
/**
|
|
* Makes deep comparison of two files to make sure they are identical.
|
|
*/
|
|
bool fsutils::sameFiles(const QString& path1, const QString& path2) {
|
|
QFile f1(path1), f2(path2);
|
|
if (!f1.exists() || !f2.exists()) return false;
|
|
if (f1.size() != f2.size()) return false;
|
|
if (!f1.open(QIODevice::ReadOnly)) return false;
|
|
if (!f2.open(QIODevice::ReadOnly)) {
|
|
f1.close();
|
|
return false;
|
|
}
|
|
bool same = true;
|
|
while(!f1.atEnd() && !f2.atEnd()) {
|
|
if (f1.read(1024) != f2.read(1024)) {
|
|
same = false;
|
|
break;
|
|
}
|
|
}
|
|
f1.close(); f2.close();
|
|
return same;
|
|
}
|
|
|
|
QString fsutils::updateLabelInSavePath(const QString& defaultSavePath, const QString& save_path, const QString& old_label, const QString& new_label) {
|
|
if (old_label == new_label) return fsutils::fromNativePath(save_path);
|
|
QString defaultPath = fsutils::fromNativePath(defaultSavePath);
|
|
QString path = fsutils::fromNativePath(save_path);
|
|
qDebug("UpdateLabelInSavePath(%s, %s, %s)", qPrintable(path), qPrintable(old_label), qPrintable(new_label));
|
|
if (!path.startsWith(defaultPath)) return path;
|
|
QString new_save_path = path;
|
|
new_save_path.remove(defaultPath);
|
|
QStringList path_parts = new_save_path.split("/", QString::SkipEmptyParts);
|
|
if (path_parts.empty()) {
|
|
if (!new_label.isEmpty())
|
|
path_parts << new_label;
|
|
} else {
|
|
if (old_label.isEmpty() || path_parts.first() != old_label) {
|
|
if (path_parts.first() != new_label)
|
|
path_parts.prepend(new_label);
|
|
} else {
|
|
if (new_label.isEmpty()) {
|
|
path_parts.removeAt(0);
|
|
} else {
|
|
if (path_parts.first() != new_label)
|
|
path_parts.replace(0, new_label);
|
|
}
|
|
}
|
|
}
|
|
new_save_path = defaultPath;
|
|
if (!new_save_path.endsWith("/")) new_save_path += "/";
|
|
new_save_path += path_parts.join("/");
|
|
qDebug("New save path is %s", qPrintable(new_save_path));
|
|
return new_save_path;
|
|
}
|
|
|
|
QString fsutils::toValidFileSystemName(QString filename) {
|
|
qDebug("toValidFSName: %s", qPrintable(filename));
|
|
const QRegExp regex("[\\\\/:?\"*<>|]");
|
|
filename.replace(regex, " ");
|
|
qDebug("toValidFSName, result: %s", qPrintable(filename));
|
|
return filename.trimmed();
|
|
}
|
|
|
|
bool fsutils::isValidFileSystemName(const QString& filename) {
|
|
if (filename.isEmpty()) return false;
|
|
const QRegExp regex("[\\\\/:?\"*<>|]");
|
|
return !filename.contains(regex);
|
|
}
|
|
|
|
long long fsutils::freeDiskSpaceOnPath(QString path) {
|
|
if (path.isEmpty()) return -1;
|
|
QDir dir_path(path);
|
|
if (!dir_path.exists()) {
|
|
QStringList parts = path.split("/");
|
|
while (parts.size() > 1 && !QDir(parts.join("/")).exists()) {
|
|
parts.removeLast();
|
|
}
|
|
dir_path = QDir(parts.join("/"));
|
|
if (!dir_path.exists()) return -1;
|
|
}
|
|
Q_ASSERT(dir_path.exists());
|
|
|
|
#ifndef Q_OS_WIN
|
|
unsigned long long available;
|
|
#ifdef Q_OS_HAIKU
|
|
const QString statfs_path = dir_path.path()+"/.";
|
|
dev_t device = dev_for_path (qPrintable(statfs_path));
|
|
if (device >= 0) {
|
|
fs_info info;
|
|
if(fs_stat_dev(device, &info)==B_OK){
|
|
available = ((unsigned long long)(info.free_blocks*info.block_size));
|
|
return available;
|
|
}
|
|
}
|
|
return -1;
|
|
#else
|
|
struct statfs stats;
|
|
const QString statfs_path = dir_path.path()+"/.";
|
|
const int ret = statfs (qPrintable(statfs_path), &stats) ;
|
|
if (ret == 0) {
|
|
available = ((unsigned long long)stats.f_bavail) *
|
|
((unsigned long long)stats.f_bsize) ;
|
|
return available;
|
|
} else {
|
|
return -1;
|
|
}
|
|
#endif
|
|
#else
|
|
typedef BOOL (WINAPI *GetDiskFreeSpaceEx_t)(LPCTSTR,
|
|
PULARGE_INTEGER,
|
|
PULARGE_INTEGER,
|
|
PULARGE_INTEGER);
|
|
GetDiskFreeSpaceEx_t
|
|
pGetDiskFreeSpaceEx = (GetDiskFreeSpaceEx_t)::GetProcAddress
|
|
(
|
|
::GetModuleHandle(TEXT("kernel32.dll")),
|
|
"GetDiskFreeSpaceExW"
|
|
);
|
|
if ( pGetDiskFreeSpaceEx )
|
|
{
|
|
ULARGE_INTEGER bytesFree, bytesTotal;
|
|
unsigned long long *ret;
|
|
if (pGetDiskFreeSpaceEx((LPCTSTR)(fsutils::toNativePath(dir_path.path())).utf16(), &bytesFree, &bytesTotal, NULL)) {
|
|
ret = (unsigned long long*)&bytesFree;
|
|
return *ret;
|
|
} else {
|
|
return -1;
|
|
}
|
|
} else {
|
|
return -1;
|
|
}
|
|
#endif
|
|
}
|
|
|
|
QString fsutils::branchPath(const QString& file_path, QString* removed) {
|
|
QString ret = fsutils::fromNativePath(file_path);
|
|
if (ret.endsWith("/"))
|
|
ret.chop(1);
|
|
const int slashIndex = ret.lastIndexOf("/");
|
|
if (slashIndex >= 0) {
|
|
if (removed)
|
|
*removed = ret.mid(slashIndex + 1);
|
|
ret = ret.left(slashIndex);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
bool fsutils::sameFileNames(const QString &first, const QString &second) {
|
|
#if defined(Q_OS_UNIX) || defined(Q_WS_QWS)
|
|
return QString::compare(first, second, Qt::CaseSensitive) == 0;
|
|
#else
|
|
return QString::compare(first, second, Qt::CaseInsensitive) == 0;
|
|
#endif
|
|
}
|
|
|
|
QString fsutils::expandPath(const QString &path) {
|
|
QString ret = fsutils::fromNativePath(path.trimmed());
|
|
if (ret.isEmpty())
|
|
return ret;
|
|
|
|
return QDir::cleanPath(ret);
|
|
}
|
|
|
|
QString fsutils::expandPathAbs(const QString& path) {
|
|
QString ret = fsutils::expandPath(path);
|
|
|
|
if (!QDir::isAbsolutePath(ret))
|
|
ret = QDir(ret).absolutePath();
|
|
|
|
return ret;
|
|
}
|
|
|
|
QString fsutils::QDesktopServicesDataLocation() {
|
|
QString result;
|
|
#ifdef Q_OS_WIN
|
|
LPWSTR path=new WCHAR[256];
|
|
if (SHGetSpecialFolderPath(0, path, CSIDL_LOCAL_APPDATA, FALSE))
|
|
result = fsutils::fromNativePath(QString::fromWCharArray(path));
|
|
if (!QCoreApplication::applicationName().isEmpty())
|
|
result += QLatin1String("/") + qApp->applicationName();
|
|
#else
|
|
#ifdef Q_OS_MAC
|
|
FSRef ref;
|
|
OSErr err = FSFindFolder(kUserDomain, kApplicationSupportFolderType, false, &ref);
|
|
if (err)
|
|
return QString();
|
|
QString path;
|
|
QByteArray ba(2048, 0);
|
|
if (FSRefMakePath(&ref, reinterpret_cast<UInt8 *>(ba.data()), ba.size()) == noErr)
|
|
result = QString::fromUtf8(ba).normalized(QString::NormalizationForm_C);
|
|
result += QLatin1Char('/') + qApp->applicationName();
|
|
#else
|
|
QString xdgDataHome = QLatin1String(qgetenv("XDG_DATA_HOME"));
|
|
if (xdgDataHome.isEmpty())
|
|
xdgDataHome = QDir::homePath() + QLatin1String("/.local/share");
|
|
xdgDataHome += QLatin1String("/data/")
|
|
+ qApp->applicationName();
|
|
result = xdgDataHome;
|
|
#endif
|
|
#endif
|
|
if (!result.endsWith("/"))
|
|
result += "/";
|
|
return result;
|
|
}
|
|
|
|
QString fsutils::QDesktopServicesCacheLocation() {
|
|
QString result;
|
|
#if defined(Q_OS_WIN) || defined(Q_OS_OS2)
|
|
result = QDesktopServicesDataLocation() + QLatin1String("cache");
|
|
#else
|
|
#ifdef Q_OS_MAC
|
|
// http://developer.apple.com/documentation/Carbon/Reference/Folder_Manager/Reference/reference.html
|
|
FSRef ref;
|
|
OSErr err = FSFindFolder(kUserDomain, kCachedDataFolderType, false, &ref);
|
|
if (err)
|
|
return QString();
|
|
QByteArray ba(2048, 0);
|
|
if (FSRefMakePath(&ref, reinterpret_cast<UInt8 *>(ba.data()), ba.size()) == noErr)
|
|
result = QString::fromUtf8(ba).normalized(QString::NormalizationForm_C);
|
|
result += QLatin1Char('/') + qApp->applicationName();
|
|
#else
|
|
QString xdgCacheHome = QLatin1String(qgetenv("XDG_CACHE_HOME"));
|
|
if (xdgCacheHome.isEmpty())
|
|
xdgCacheHome = QDir::homePath() + QLatin1String("/.cache");
|
|
xdgCacheHome += QLatin1Char('/') + QCoreApplication::applicationName();
|
|
result = xdgCacheHome;
|
|
#endif
|
|
#endif
|
|
if (!result.endsWith("/"))
|
|
result += "/";
|
|
return result;
|
|
}
|
|
|
|
QString fsutils::QDesktopServicesDownloadLocation() {
|
|
#if defined(Q_OS_WIN) || defined(Q_OS_OS2)
|
|
// as long as it stays WinXP like we do the same on OS/2
|
|
// TODO: Use IKnownFolderManager to get path of FOLDERID_Downloads
|
|
// instead of hardcoding "Downloads"
|
|
// Unfortunately, this would break compatibility with WinXP
|
|
#if (QT_VERSION < QT_VERSION_CHECK(5, 0, 0))
|
|
return QDir(QDesktopServices::storageLocation(QDesktopServices::DocumentsLocation)).absoluteFilePath(
|
|
QCoreApplication::translate("fsutils", "Downloads"));
|
|
#else
|
|
return QDir(QStandardPaths::writableLocation(QStandardPaths::DocumentsLocation)).absoluteFilePath(
|
|
QCoreApplication::translate("fsutils", "Downloads"));
|
|
#endif
|
|
#endif
|
|
|
|
#if (defined(Q_OS_UNIX) && !defined(Q_OS_MAC))
|
|
QString save_path;
|
|
// Default save path on Linux
|
|
QString config_path = QString::fromLocal8Bit(qgetenv("XDG_CONFIG_HOME").constData());
|
|
if (config_path.isEmpty())
|
|
config_path = QDir::home().absoluteFilePath(".config");
|
|
|
|
QString user_dirs_file = config_path + "/user-dirs.dirs";
|
|
if (QFile::exists(user_dirs_file)) {
|
|
QSettings settings(user_dirs_file, QSettings::IniFormat);
|
|
// We need to force UTF-8 encoding here since this is not
|
|
// the default for Ini files.
|
|
settings.setIniCodec("UTF-8");
|
|
QString xdg_download_dir = settings.value("XDG_DOWNLOAD_DIR").toString();
|
|
if (!xdg_download_dir.isEmpty()) {
|
|
// Resolve $HOME environment variables
|
|
xdg_download_dir.replace("$HOME", QDir::homePath());
|
|
save_path = xdg_download_dir;
|
|
qDebug() << Q_FUNC_INFO << "SUCCESS: Using XDG path for downloads: " << save_path;
|
|
}
|
|
}
|
|
|
|
// Fallback
|
|
if (!save_path.isEmpty() && !QFile::exists(save_path)) {
|
|
QDir().mkpath(save_path);
|
|
}
|
|
|
|
if (save_path.isEmpty() || !QFile::exists(save_path)) {
|
|
save_path = QDir::home().absoluteFilePath(QCoreApplication::translate("fsutils", "Downloads"));
|
|
qDebug() << Q_FUNC_INFO << "using" << save_path << "as fallback since the XDG detection did not work";
|
|
}
|
|
|
|
return save_path;
|
|
#endif
|
|
|
|
#ifdef Q_OS_MAC
|
|
// TODO: How to support this on Mac OS X?
|
|
#endif
|
|
|
|
// Fallback
|
|
return QDir::home().absoluteFilePath(QCoreApplication::translate("fsutils", "Downloads"));
|
|
}
|
|
|
|
QString fsutils::searchEngineLocation() {
|
|
QString folder = "nova";
|
|
if (misc::pythonVersion() >= 3)
|
|
folder = "nova3";
|
|
const QString location = fsutils::expandPathAbs(QDesktopServicesDataLocation()
|
|
+ folder);
|
|
QDir locationDir(location);
|
|
if (!locationDir.exists())
|
|
locationDir.mkpath(locationDir.absolutePath());
|
|
return location;
|
|
}
|
|
|
|
QString fsutils::BTBackupLocation() {
|
|
const QString location = fsutils::expandPathAbs(QDesktopServicesDataLocation()
|
|
+ "BT_backup");
|
|
QDir locationDir(location);
|
|
if (!locationDir.exists())
|
|
locationDir.mkpath(locationDir.absolutePath());
|
|
return location;
|
|
}
|
|
|
|
QString fsutils::cacheLocation() {
|
|
QString location = fsutils::expandPathAbs(QDesktopServicesCacheLocation());
|
|
QDir locationDir(location);
|
|
if (!locationDir.exists())
|
|
locationDir.mkpath(locationDir.absolutePath());
|
|
return location;
|
|
}
|