Revert "Check for local URLs more strictly."

This reverts commit c3fda41224.
This commit is contained in:
John Preston 2024-09-10 14:13:50 +04:00
parent ca3c179b75
commit d1e3b9f15d
7 changed files with 16 additions and 36 deletions

View file

@ -32,7 +32,6 @@ https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
#include "boxes/premium_limits_box.h" #include "boxes/premium_limits_box.h"
#include "boxes/premium_preview_box.h" #include "boxes/premium_preview_box.h"
#include "boxes/send_credits_box.h" #include "boxes/send_credits_box.h"
#include "platform/platform_file_utilities.h"
#include "ui/effects/scroll_content_shadow.h" #include "ui/effects/scroll_content_shadow.h"
#include "ui/widgets/fields/number_input.h" #include "ui/widgets/fields/number_input.h"
#include "ui/widgets/checkbox.h" #include "ui/widgets/checkbox.h"
@ -72,7 +71,7 @@ constexpr auto kMaxMessageLength = 4096;
using Ui::SendFilesWay; using Ui::SendFilesWay;
[[nodiscard]] inline bool CanAddUrls(const QList<QUrl> &urls) { [[nodiscard]] inline bool CanAddUrls(const QList<QUrl> &urls) {
return !urls.isEmpty() && ranges::all_of(urls, Core::UrlIsLocal); return !urls.isEmpty() && ranges::all_of(urls, &QUrl::isLocalFile);
} }
[[nodiscard]] bool CanAddFiles(not_null<const QMimeData*> data) { [[nodiscard]] bool CanAddFiles(not_null<const QMimeData*> data) {

View file

@ -7,10 +7,6 @@ https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
*/ */
#pragma once #pragma once
namespace Core {
bool UrlIsLocal(const QUrl &url);
} // namespace Core
namespace Main { namespace Main {
class Session; class Session;
} // namespace Main } // namespace Main
@ -49,7 +45,7 @@ void ShowInFolder(const QString &filepath);
namespace internal { namespace internal {
inline QString UrlToLocalDefault(const QUrl &url) { inline QString UrlToLocalDefault(const QUrl &url) {
return Core::UrlIsLocal(url) ? url.toLocalFile() : QString(); return url.toLocalFile();
} }
void UnsafeOpenUrlDefault(const QString &url); void UnsafeOpenUrlDefault(const QString &url);

View file

@ -226,24 +226,13 @@ bool CanSendFiles(not_null<const QMimeData*> data) {
if (data->hasImage()) { if (data->hasImage()) {
return true; return true;
} else if (const auto urls = ReadMimeUrls(data); !urls.empty()) { } else if (const auto urls = ReadMimeUrls(data); !urls.empty()) {
if (ranges::all_of(urls, UrlIsLocal)) { if (ranges::all_of(urls, &QUrl::isLocalFile)) {
return true; return true;
} }
} }
return false; return false;
} }
bool UrlIsLocal(const QUrl &url) {
if (!url.isLocalFile()) {
return false;
}
const auto result = url.toLocalFile();
if (result.startsWith("//")) {
return false;
}
return !result.isEmpty();
}
QString FileExtension(const QString &filepath) { QString FileExtension(const QString &filepath) {
const auto reversed = ranges::views::reverse(filepath); const auto reversed = ranges::views::reverse(filepath);
const auto last = ranges::find_first_of(reversed, ".\\/"); const auto last = ranges::find_first_of(reversed, ".\\/");

View file

@ -68,7 +68,6 @@ struct MimeImageData {
[[nodiscard]] QString ReadMimeText(not_null<const QMimeData*> data); [[nodiscard]] QString ReadMimeText(not_null<const QMimeData*> data);
[[nodiscard]] QList<QUrl> ReadMimeUrls(not_null<const QMimeData*> data); [[nodiscard]] QList<QUrl> ReadMimeUrls(not_null<const QMimeData*> data);
[[nodiscard]] bool CanSendFiles(not_null<const QMimeData*> data); [[nodiscard]] bool CanSendFiles(not_null<const QMimeData*> data);
[[nodiscard]] bool UrlIsLocal(const QUrl &url);
enum class NameType : uchar { enum class NameType : uchar {
Unknown, Unknown,

View file

@ -5786,7 +5786,7 @@ bool HistoryWidget::canSendFiles(not_null<const QMimeData*> data) const {
} else if (data->hasImage()) { } else if (data->hasImage()) {
return true; return true;
} else if (const auto urls = Core::ReadMimeUrls(data); !urls.empty()) { } else if (const auto urls = Core::ReadMimeUrls(data); !urls.empty()) {
if (ranges::all_of(urls, Core::UrlIsLocal)) { if (ranges::all_of(urls, &QUrl::isLocalFile)) {
return true; return true;
} }
} }

View file

@ -8,7 +8,6 @@ https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
#include "platform/mac/file_utilities_mac.h" #include "platform/mac/file_utilities_mac.h"
#include "base/platform/mac/base_utilities_mac.h" #include "base/platform/mac/base_utilities_mac.h"
#include "core/mime_type.h"
#include "lang/lang_keys.h" #include "lang/lang_keys.h"
#include "styles/style_window.h" #include "styles/style_window.h"
@ -380,9 +379,6 @@ namespace Platform {
namespace File { namespace File {
QString UrlToLocal(const QUrl &url) { QString UrlToLocal(const QUrl &url) {
if (!Core::UrlIsLocal(url)) {
return QString();
}
auto result = url.toLocalFile(); auto result = url.toLocalFile();
if (result.startsWith(u"/.file/id="_q)) { if (result.startsWith(u"/.file/id="_q)) {
NSString *nsurl = [[[NSURL URLWithString: [NSString stringWithUTF8String: (u"file://"_q + result).toUtf8().constData()]] filePathURL] path]; NSString *nsurl = [[[NSURL URLWithString: [NSString stringWithUTF8String: (u"file://"_q + result).toUtf8().constData()]] filePathURL] path];

View file

@ -83,9 +83,10 @@ bool ValidatePhotoEditorMediaDragData(not_null<const QMimeData*> data) {
} }
if (!urls.isEmpty()) { if (!urls.isEmpty()) {
using namespace Core; const auto url = urls.front();
const auto file = Platform::File::UrlToLocal(urls.front()); if (url.isLocalFile()) {
if (!file.isEmpty()) { using namespace Core;
const auto file = Platform::File::UrlToLocal(url);
const auto info = QFileInfo(file); const auto info = QFileInfo(file);
return FileIsImage(file, MimeTypeForFile(info).name()) return FileIsImage(file, MimeTypeForFile(info).name())
&& QImageReader(file).canRead(); && QImageReader(file).canRead();
@ -106,10 +107,10 @@ bool ValidateEditMediaDragData(
} }
if (albumType == Ui::AlbumType::PhotoVideo && !urls.isEmpty()) { if (albumType == Ui::AlbumType::PhotoVideo && !urls.isEmpty()) {
using namespace Core; const auto url = urls.front();
const auto file = Platform::File::UrlToLocal(urls.front()); if (url.isLocalFile()) {
if (!file.isEmpty()) { using namespace Core;
const auto info = QFileInfo(file); const auto info = QFileInfo(Platform::File::UrlToLocal(url));
return IsMimeAcceptedForPhotoVideoAlbum(MimeTypeForFile(info).name()); return IsMimeAcceptedForPhotoVideoAlbum(MimeTypeForFile(info).name());
} }
} }
@ -133,10 +134,10 @@ MimeDataState ComputeMimeDataState(const QMimeData *data) {
auto allAreSmallImages = true; auto allAreSmallImages = true;
for (const auto &url : urls) { for (const auto &url : urls) {
const auto file = Platform::File::UrlToLocal(url); if (!url.isLocalFile()) {
if (file.isEmpty()) {
return MimeDataState::None; return MimeDataState::None;
} }
const auto file = Platform::File::UrlToLocal(url);
const auto info = QFileInfo(file); const auto info = QFileInfo(file);
if (info.isDir()) { if (info.isDir()) {
@ -174,13 +175,13 @@ PreparedList PrepareMediaList(
auto locals = QStringList(); auto locals = QStringList();
locals.reserve(files.size()); locals.reserve(files.size());
for (const auto &url : files) { for (const auto &url : files) {
locals.push_back(Platform::File::UrlToLocal(url)); if (!url.isLocalFile()) {
if (locals.back().isEmpty()) {
return { return {
PreparedList::Error::NonLocalUrl, PreparedList::Error::NonLocalUrl,
url.toDisplayString() url.toDisplayString()
}; };
} }
locals.push_back(Platform::File::UrlToLocal(url));
} }
return PrepareMediaList(locals, previewWidth, premium); return PrepareMediaList(locals, previewWidth, premium);
} }