Add a limit on blocksize to prevent out-of-memory situation.

c556f390 · Andrius Štikonas · 9dd38744 · c556f390
Commit c556f390 authored Nov 29, 2020 by Andrius Štikonas
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

src/util/externalcommandhelper.cpp src/util/externalcommandhelper.cpp +5 -0

No files found.
--- a/src/util/externalcommandhelper.cpp
+++ b/src/util/externalcommandhelper.cpp
@@ -163,6 +163,11 @@ QVariantMap ExternalCommandHelper::CopyBlocks(const QString& sourceDevice, const
    if (!blockSize) {
        return QVariantMap();
    }
+    constexpr qint64 MiB = 1 << 30;
+    if (blockSize > 100 * MiB) {
+        return QVariantMap();
+    }
+
    QVariantMap reply;
    reply[QStringLiteral("success")] = true;