Browse Source

Merge pull request #11293 from Chocobo1/qt

Code clean up
adaptive-webui-19844
Mike Tzou 5 years ago committed by GitHub
parent
commit
c062a31f12
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 13
      src/app/application.cpp
  2. 7
      src/app/qtlocalpeer/qtlocalpeer.cpp
  3. 66
      src/base/bittorrent/session.cpp
  4. 26
      src/base/bittorrent/session.h
  5. 10
      src/base/utils/misc.cpp
  6. 4
      winconf.pri

13
src/app/application.cpp

@ -685,11 +685,8 @@ void Application::cleanup()
m_window->hide(); m_window->hide();
#ifdef Q_OS_WIN #ifdef Q_OS_WIN
typedef BOOL (WINAPI *PSHUTDOWNBRCREATE)(HWND, LPCWSTR); ::ShutdownBlockReasonCreate(reinterpret_cast<HWND>(m_window->effectiveWinId())
const auto shutdownBRCreate = Utils::Misc::loadWinAPI<PSHUTDOWNBRCREATE>("User32.dll", "ShutdownBlockReasonCreate"); , tr("Saving torrent progress...").toStdWString().c_str());
// Only available on Vista+
if (shutdownBRCreate)
shutdownBRCreate((HWND)m_window->effectiveWinId(), tr("Saving torrent progress...").toStdWString().c_str());
#endif // Q_OS_WIN #endif // Q_OS_WIN
// Do manual cleanup in MainWindow to force widgets // Do manual cleanup in MainWindow to force widgets
@ -728,11 +725,7 @@ void Application::cleanup()
#ifndef DISABLE_GUI #ifndef DISABLE_GUI
if (m_window) { if (m_window) {
#ifdef Q_OS_WIN #ifdef Q_OS_WIN
using PSHUTDOWNBRDESTROY = BOOL (WINAPI *)(HWND); ::ShutdownBlockReasonDestroy(reinterpret_cast<HWND>(m_window->effectiveWinId()));
const auto shutdownBRDestroy = Utils::Misc::loadWinAPI<PSHUTDOWNBRDESTROY>("User32.dll", "ShutdownBlockReasonDestroy");
// Only available on Vista+
if (shutdownBRDestroy)
shutdownBRDestroy(reinterpret_cast<HWND>(m_window->effectiveWinId()));
#endif // Q_OS_WIN #endif // Q_OS_WIN
delete m_window; delete m_window;
UIThemeManager::freeInstance(); UIThemeManager::freeInstance();

7
src/app/qtlocalpeer/qtlocalpeer.cpp

@ -116,14 +116,9 @@ QtLocalPeer::QtLocalPeer(QObject* parent, const QString &appId)
+ QLatin1Char('-') + QString::number(idNum, 16); + QLatin1Char('-') + QString::number(idNum, 16);
#if defined(Q_OS_WIN) #if defined(Q_OS_WIN)
using PPROCESSIDTOSESSIONID = BOOL (WINAPI *)(DWORD, DWORD *);
const auto processIdToSessionId = Utils::Misc::loadWinAPI<PPROCESSIDTOSESSIONID>("kernel32.dll", "ProcessIdToSessionId");
if (processIdToSessionId) {
DWORD sessionId = 0; DWORD sessionId = 0;
processIdToSessionId(GetCurrentProcessId(), &sessionId); ::ProcessIdToSessionId(GetCurrentProcessId(), &sessionId);
socketName += (QLatin1Char('-') + QString::number(sessionId, 16)); socketName += (QLatin1Char('-') + QString::number(sessionId, 16));
}
#else #else
socketName += (QLatin1Char('-') + QString::number(::getuid(), 16)); socketName += (QLatin1Char('-') + QString::number(::getuid(), 16));
#endif #endif

66
src/base/bittorrent/session.cpp

@ -1625,7 +1625,7 @@ void Session::banIP(const QString &ip)
// Delete a torrent from the session, given its hash // Delete a torrent from the session, given its hash
// and from the disk, if the corresponding deleteOption is chosen // and from the disk, if the corresponding deleteOption is chosen
bool Session::deleteTorrent(const QString &hash, const DeleteOption deleteOption) bool Session::deleteTorrent(const InfoHash &hash, const DeleteOption deleteOption)
{ {
TorrentHandle *const torrent = m_torrents.take(hash); TorrentHandle *const torrent = m_torrents.take(hash);
if (!torrent) return false; if (!torrent) return false;
@ -1680,9 +1680,10 @@ bool Session::deleteTorrent(const QString &hash, const DeleteOption deleteOption
bool Session::cancelLoadMetadata(const InfoHash &hash) bool Session::cancelLoadMetadata(const InfoHash &hash)
{ {
if (!m_loadedMetadata.contains(hash)) return false; const auto loadedMetadataIter = m_loadedMetadata.find(hash);
if (loadedMetadataIter == m_loadedMetadata.end()) return false;
m_loadedMetadata.remove(hash); m_loadedMetadata.erase(loadedMetadataIter);
const lt::torrent_handle torrent = m_nativeSession->find_torrent(hash); const lt::torrent_handle torrent = m_nativeSession->find_torrent(hash);
if (!torrent.is_valid()) return false; if (!torrent.is_valid()) return false;
@ -4008,14 +4009,17 @@ void Session::handleTorrentRemovedAlert(const lt::torrent_removed_alert *p)
{ {
const InfoHash infoHash {p->info_hash}; const InfoHash infoHash {p->info_hash};
if (m_loadedMetadata.contains(infoHash)) const auto loadedMetadataIter = m_loadedMetadata.find(infoHash);
emit metadataLoaded(m_loadedMetadata.take(infoHash)); if (loadedMetadataIter != m_loadedMetadata.end()) {
emit metadataLoaded(*loadedMetadataIter);
m_loadedMetadata.erase(loadedMetadataIter);
}
if (m_removingTorrents.contains(infoHash)) { const auto removingTorrentDataIter = m_removingTorrents.find(infoHash);
const RemovingTorrentData tmpRemovingTorrentData = m_removingTorrents[infoHash]; if (removingTorrentDataIter != m_removingTorrents.end()) {
if (tmpRemovingTorrentData.deleteOption == Torrent) { if (removingTorrentDataIter->deleteOption == Torrent) {
LogMsg(tr("'%1' was removed from the transfer list.", "'xxx.avi' was removed...").arg(tmpRemovingTorrentData.name)); LogMsg(tr("'%1' was removed from the transfer list.", "'xxx.avi' was removed...").arg(removingTorrentDataIter->name));
m_removingTorrents.remove(infoHash); m_removingTorrents.erase(removingTorrentDataIter);
} }
} }
} }
@ -4023,44 +4027,48 @@ void Session::handleTorrentRemovedAlert(const lt::torrent_removed_alert *p)
void Session::handleTorrentDeletedAlert(const lt::torrent_deleted_alert *p) void Session::handleTorrentDeletedAlert(const lt::torrent_deleted_alert *p)
{ {
const InfoHash infoHash {p->info_hash}; const InfoHash infoHash {p->info_hash};
const auto removingTorrentDataIter = m_removingTorrents.find(infoHash);
if (!m_removingTorrents.contains(infoHash)) if (removingTorrentDataIter == m_removingTorrents.end())
return; return;
const RemovingTorrentData tmpRemovingTorrentData = m_removingTorrents.take(infoHash);
Utils::Fs::smartRemoveEmptyFolderTree(tmpRemovingTorrentData.savePathToRemove);
LogMsg(tr("'%1' was removed from the transfer list and hard disk.", "'xxx.avi' was removed...").arg(tmpRemovingTorrentData.name)); Utils::Fs::smartRemoveEmptyFolderTree(removingTorrentDataIter->savePathToRemove);
LogMsg(tr("'%1' was removed from the transfer list and hard disk.", "'xxx.avi' was removed...").arg(removingTorrentDataIter->name));
m_removingTorrents.erase(removingTorrentDataIter);
} }
void Session::handleTorrentDeleteFailedAlert(const lt::torrent_delete_failed_alert *p) void Session::handleTorrentDeleteFailedAlert(const lt::torrent_delete_failed_alert *p)
{ {
const InfoHash infoHash {p->info_hash}; const InfoHash infoHash {p->info_hash};
const auto removingTorrentDataIter = m_removingTorrents.find(infoHash);
if (!m_removingTorrents.contains(infoHash)) if (removingTorrentDataIter == m_removingTorrents.end())
return; return;
const RemovingTorrentData tmpRemovingTorrentData = m_removingTorrents.take(infoHash);
// libtorrent won't delete the directory if it contains files not listed in the torrent, // libtorrent won't delete the directory if it contains files not listed in the torrent,
// so we remove the directory ourselves // so we remove the directory ourselves
Utils::Fs::smartRemoveEmptyFolderTree(tmpRemovingTorrentData.savePathToRemove); Utils::Fs::smartRemoveEmptyFolderTree(removingTorrentDataIter->savePathToRemove);
if (p->error) { if (p->error) {
LogMsg(tr("'%1' was removed from the transfer list but the files couldn't be deleted. Error: %2", "'xxx.avi' was removed...") LogMsg(tr("'%1' was removed from the transfer list but the files couldn't be deleted. Error: %2", "'xxx.avi' was removed...")
.arg(tmpRemovingTorrentData.name, QString::fromLocal8Bit(p->error.message().c_str())) .arg(removingTorrentDataIter->name, QString::fromLocal8Bit(p->error.message().c_str()))
, Log::WARNING); , Log::WARNING);
} }
else { else {
LogMsg(tr("'%1' was removed from the transfer list.", "'xxx.avi' was removed...").arg(tmpRemovingTorrentData.name)); LogMsg(tr("'%1' was removed from the transfer list.", "'xxx.avi' was removed...").arg(removingTorrentDataIter->name));
} }
m_removingTorrents.erase(removingTorrentDataIter);
} }
void Session::handleMetadataReceivedAlert(const lt::metadata_received_alert *p) void Session::handleMetadataReceivedAlert(const lt::metadata_received_alert *p)
{ {
const InfoHash hash {p->handle.info_hash()}; const InfoHash hash {p->handle.info_hash()};
const auto loadedMetadataIter = m_loadedMetadata.find(hash);
if (m_loadedMetadata.contains(hash)) { if (loadedMetadataIter != m_loadedMetadata.end()) {
--m_extraLimit; --m_extraLimit;
adjustLimits(); adjustLimits();
m_loadedMetadata[hash] = TorrentInfo(p->handle.torrent_file()); *loadedMetadataIter = TorrentInfo(p->handle.torrent_file());
m_nativeSession->remove_torrent(p->handle, lt::session::delete_files); m_nativeSession->remove_torrent(p->handle, lt::session::delete_files);
} }
} }
@ -4531,23 +4539,15 @@ namespace
QString convertIfaceNameToGuid(const QString &name) QString convertIfaceNameToGuid(const QString &name)
{ {
// Under Windows XP or on Qt version <= 5.5 'name' will be a GUID already. // Under Windows XP or on Qt version <= 5.5 'name' will be a GUID already.
QUuid uuid(name); const QUuid uuid(name);
if (!uuid.isNull()) if (!uuid.isNull())
return uuid.toString().toUpper(); // Libtorrent expects the GUID in uppercase return uuid.toString().toUpper(); // Libtorrent expects the GUID in uppercase
using PCONVERTIFACENAMETOLUID = NETIO_STATUS (WINAPI *)(const WCHAR *, PNET_LUID); NET_LUID luid {};
const auto ConvertIfaceNameToLuid = Utils::Misc::loadWinAPI<PCONVERTIFACENAMETOLUID>("Iphlpapi.dll", "ConvertInterfaceNameToLuidW"); const LONG res = ::ConvertInterfaceNameToLuidW(name.toStdWString().c_str(), &luid);
if (!ConvertIfaceNameToLuid) return {};
using PCONVERTIFACELUIDTOGUID = NETIO_STATUS (WINAPI *)(const NET_LUID *, GUID *);
const auto ConvertIfaceLuidToGuid = Utils::Misc::loadWinAPI<PCONVERTIFACELUIDTOGUID>("Iphlpapi.dll", "ConvertInterfaceLuidToGuid");
if (!ConvertIfaceLuidToGuid) return {};
NET_LUID luid;
const LONG res = ConvertIfaceNameToLuid(name.toStdWString().c_str(), &luid);
if (res == 0) { if (res == 0) {
GUID guid; GUID guid;
if (ConvertIfaceLuidToGuid(&luid, &guid) == 0) if (::ConvertInterfaceLuidToGuid(&luid, &guid) == 0)
return QUuid(guid).toString().toUpper(); return QUuid(guid).toString().toUpper();
} }

26
src/base/bittorrent/session.h

@ -95,18 +95,19 @@ namespace BitTorrent
class TrackerEntry; class TrackerEntry;
struct CreateTorrentParams; struct CreateTorrentParams;
class SessionSettingsEnums // Using `Q_ENUM_NS()` without a wrapper namespace in our case is not advised
// since `Q_NAMESPACE` cannot be used when the same namespace resides at different files.
// https://www.kdab.com/new-qt-5-8-meta-object-support-namespaces/#comment-143779
namespace SessionSettingsEnums
{ {
Q_GADGET Q_NAMESPACE
public:
// TODO: remove `SessionSettingsEnums` wrapper when we can use `Q_ENUM_NS` directly (QT >= 5.8 only)
enum class ChokingAlgorithm : int enum class ChokingAlgorithm : int
{ {
FixedSlots = 0, FixedSlots = 0,
RateBased = 1 RateBased = 1
}; };
Q_ENUM(ChokingAlgorithm) Q_ENUM_NS(ChokingAlgorithm)
enum class SeedChokingAlgorithm : int enum class SeedChokingAlgorithm : int
{ {
@ -114,14 +115,14 @@ namespace BitTorrent
FastestUpload = 1, FastestUpload = 1,
AntiLeech = 2 AntiLeech = 2
}; };
Q_ENUM(SeedChokingAlgorithm) Q_ENUM_NS(SeedChokingAlgorithm)
enum class MixedModeAlgorithm : int enum class MixedModeAlgorithm : int
{ {
TCP = 0, TCP = 0,
Proportional = 1 Proportional = 1
}; };
Q_ENUM(MixedModeAlgorithm) Q_ENUM_NS(MixedModeAlgorithm)
enum class BTProtocol : int enum class BTProtocol : int
{ {
@ -129,12 +130,9 @@ namespace BitTorrent
TCP = 1, TCP = 1,
UTP = 2 UTP = 2
}; };
Q_ENUM(BTProtocol) Q_ENUM_NS(BTProtocol)
}; }
using ChokingAlgorithm = SessionSettingsEnums::ChokingAlgorithm; using namespace SessionSettingsEnums;
using SeedChokingAlgorithm = SessionSettingsEnums::SeedChokingAlgorithm;
using MixedModeAlgorithm = SessionSettingsEnums::MixedModeAlgorithm;
using BTProtocol = SessionSettingsEnums::BTProtocol;
struct SessionMetricIndices struct SessionMetricIndices
{ {
@ -409,7 +407,7 @@ namespace BitTorrent
bool isKnownTorrent(const InfoHash &hash) const; bool isKnownTorrent(const InfoHash &hash) const;
bool addTorrent(const QString &source, const AddTorrentParams &params = AddTorrentParams()); bool addTorrent(const QString &source, const AddTorrentParams &params = AddTorrentParams());
bool addTorrent(const TorrentInfo &torrentInfo, const AddTorrentParams &params = AddTorrentParams()); bool addTorrent(const TorrentInfo &torrentInfo, const AddTorrentParams &params = AddTorrentParams());
bool deleteTorrent(const QString &hash, DeleteOption deleteOption = Torrent); bool deleteTorrent(const InfoHash &hash, DeleteOption deleteOption = Torrent);
bool loadMetadata(const MagnetUri &magnetUri); bool loadMetadata(const MagnetUri &magnetUri);
bool cancelLoadMetadata(const InfoHash &hash); bool cancelLoadMetadata(const InfoHash &hash);

10
src/base/utils/misc.cpp

@ -30,6 +30,7 @@
#ifdef Q_OS_WIN #ifdef Q_OS_WIN
#include <windows.h> #include <windows.h>
#include <powrprof.h>
#include <Shlobj.h> #include <Shlobj.h>
#else #else
#include <sys/types.h> #include <sys/types.h>
@ -117,16 +118,11 @@ void Utils::Misc::shutdownComputer(const ShutdownDialogAction &action)
if (GetLastError() != ERROR_SUCCESS) if (GetLastError() != ERROR_SUCCESS)
return; return;
using PSETSUSPENDSTATE = BOOLEAN (WINAPI *)(BOOLEAN, BOOLEAN, BOOLEAN);
const auto setSuspendState = Utils::Misc::loadWinAPI<PSETSUSPENDSTATE>("PowrProf.dll", "SetSuspendState");
if (action == ShutdownDialogAction::Suspend) { if (action == ShutdownDialogAction::Suspend) {
if (setSuspendState) ::SetSuspendState(false, false, false);
setSuspendState(false, false, false);
} }
else if (action == ShutdownDialogAction::Hibernate) { else if (action == ShutdownDialogAction::Hibernate) {
if (setSuspendState) ::SetSuspendState(true, false, false);
setSuspendState(true, false, false);
} }
else { else {
const QString msg = QCoreApplication::translate("misc", "qBittorrent will shutdown the computer now because all downloads are complete."); const QString msg = QCoreApplication::translate("misc", "qBittorrent will shutdown the computer now because all downloads are complete.");

4
winconf.pri

@ -34,7 +34,7 @@ win32-g++* {
RC_FILE = qbittorrent_mingw.rc RC_FILE = qbittorrent_mingw.rc
LIBS += libadvapi32 libshell32 libuser32 libole32 libwsock32 libws2_32 LIBS += libadvapi32 libiphlpapi libole32 libpowrprof libshell32 libuser32 libwsock32 libws2_32
} }
else:win32-msvc* { else:win32-msvc* {
CONFIG -= embed_manifest_exe CONFIG -= embed_manifest_exe
@ -42,7 +42,7 @@ else:win32-msvc* {
RC_FILE = qbittorrent.rc RC_FILE = qbittorrent.rc
LIBS += advapi32.lib shell32.lib crypt32.lib User32.lib ole32.lib LIBS += advapi32.lib crypt32.lib Iphlpapi.lib ole32.lib PowrProf.lib shell32.lib User32.lib
} }
# See an example build configuration in "conf.pri.windows" # See an example build configuration in "conf.pri.windows"

Loading…
Cancel
Save