service/HtmlInlineImageService.cpp
| Line | Branch | Exec | Source |
|---|---|---|---|
| 1 | #include "HtmlInlineImageService.h" | ||
| 2 | #include "MimeParser.h" | ||
| 3 | #include "util/Utf8ByteCount.h" | ||
| 4 | #include <QCoreApplication> | ||
| 5 | #include <QFile> | ||
| 6 | #include <QFileInfo> | ||
| 7 | #include <QFutureWatcher> | ||
| 8 | #include <QSqlDatabase> | ||
| 9 | #include <QSqlDriver> | ||
| 10 | #include <QSqlError> | ||
| 11 | #include <QSqlQuery> | ||
| 12 | #include <QScopeGuard> | ||
| 13 | #include <QUuid> | ||
| 14 | #include <optional> | ||
| 15 | #include <QtConcurrent> | ||
| 16 | #include <QCryptographicHash> | ||
| 17 | #include <sqlite3.h> | ||
| 18 | #ifdef Q_OS_UNIX | ||
| 19 | #include <sys/stat.h> | ||
| 20 | #endif | ||
| 21 | |||
| 22 | namespace { | ||
| 23 | // S106-040: the retained connection is reused only for the same file — | ||
| 24 | // same path AND (on Unix) the same device and inode, like | ||
| 25 | // NativeImageResources. A cache database replaced at the same path (rebuild, | ||
| 26 | // restore) used to be read through the deleted inode until restart. | ||
| 27 | struct FileIdentity { | ||
| 28 | 10 | quint64 device = 0; | |
| 29 | 10 | quint64 inode = 0; | |
| 30 |
3/4✗ Branch 2 → 3 not taken.
✓ Branch 2 → 4 taken 10 times.
✓ Branch 4 → 5 taken 1 time.
✓ Branch 4 → 6 taken 9 times.
|
10 | bool operator==(const FileIdentity &) const = default; |
| 31 | }; | ||
| 32 | 20 | std::optional<FileIdentity> fileIdentity(const QString &path) { | |
| 33 | #ifdef Q_OS_UNIX | ||
| 34 | 20 | struct stat file{}; | |
| 35 |
5/11✓ Branch 2 → 3 taken 20 times.
✓ Branch 5 → 6 taken 20 times.
✗ Branch 5 → 7 not taken.
✗ Branch 6 → 7 not taken.
✓ Branch 6 → 8 taken 20 times.
✓ Branch 9 → 10 taken 20 times.
✗ Branch 9 → 11 not taken.
✗ Branch 11 → 12 not taken.
✓ Branch 11 → 13 taken 20 times.
✗ Branch 17 → 18 not taken.
✗ Branch 17 → 19 not taken.
|
20 | if (::stat(QFile::encodeName(path).constData(), &file) != 0 || !S_ISREG(file.st_mode)) |
| 36 | ✗ | return std::nullopt; | |
| 37 | 20 | return FileIdentity{quint64(file.st_dev), quint64(file.st_ino)}; | |
| 38 | #else | ||
| 39 | return QFileInfo(path).isFile() ? std::optional<FileIdentity>(FileIdentity{}) : std::nullopt; | ||
| 40 | #endif | ||
| 41 | } | ||
| 42 | |||
| 43 | struct InlineReadConnection { | ||
| 44 | QString path; | ||
| 45 | QString name; | ||
| 46 | QSqlDatabase db; | ||
| 47 | FileIdentity identity; | ||
| 48 | |||
| 49 | 3 | ~InlineReadConnection() { reset(); } | |
| 50 | |||
| 51 | 20 | void reset() { | |
| 52 |
2/2✓ Branch 3 → 4 taken 11 times.
✓ Branch 3 → 9 taken 9 times.
|
20 | if (db.isValid()) { |
| 53 | 11 | db.close(); | |
| 54 |
2/2✓ Branch 5 → 6 taken 11 times.
✓ Branch 6 → 7 taken 11 times.
|
11 | db = QSqlDatabase(); |
| 55 | } | ||
| 56 |
2/2✓ Branch 10 → 11 taken 11 times.
✓ Branch 10 → 12 taken 9 times.
|
20 | if (!name.isEmpty()) QSqlDatabase::removeDatabase(name); |
| 57 | 20 | path.clear(); | |
| 58 | 20 | name.clear(); | |
| 59 | 20 | } | |
| 60 | |||
| 61 | 20 | bool ensure(const QString &databasePath, QString *error) { | |
| 62 |
1/1✓ Branch 2 → 3 taken 20 times.
|
20 | const auto current = fileIdentity(databasePath); |
| 63 |
1/2✗ Branch 4 → 5 not taken.
✓ Branch 4 → 13 taken 20 times.
|
20 | if (!current) { |
| 64 | ✗ | reset(); | |
| 65 | ✗ | *error = QStringLiteral("Inline image cache is unavailable"); | |
| 66 | ✗ | return false; | |
| 67 | } | ||
| 68 |
9/9✓ Branch 13 → 14 taken 20 times.
✓ Branch 14 → 15 taken 12 times.
✓ Branch 14 → 21 taken 8 times.
✓ Branch 16 → 17 taken 10 times.
✓ Branch 16 → 21 taken 2 times.
✓ Branch 19 → 20 taken 9 times.
✓ Branch 19 → 21 taken 1 time.
✓ Branch 22 → 23 taken 9 times.
✓ Branch 22 → 24 taken 11 times.
|
20 | if (db.isOpen() && path == databasePath && identity == *current) return true; |
| 69 |
1/1✓ Branch 24 → 25 taken 11 times.
|
11 | reset(); |
| 70 |
3/3✓ Branch 25 → 26 taken 11 times.
✓ Branch 26 → 27 taken 11 times.
✓ Branch 30 → 31 taken 11 times.
|
22 | name = QStringLiteral("html-inline-") + QUuid::createUuid().toString(); |
| 71 |
2/2✓ Branch 39 → 40 taken 11 times.
✓ Branch 40 → 41 taken 11 times.
|
22 | db = QSqlDatabase::addDatabase(QStringLiteral("QSQLITE"), name); |
| 72 |
1/1✓ Branch 44 → 45 taken 11 times.
|
11 | db.setDatabaseName(databasePath); |
| 73 |
1/1✓ Branch 48 → 49 taken 11 times.
|
22 | db.setConnectOptions(QStringLiteral("QSQLITE_OPEN_READONLY;QSQLITE_BUSY_TIMEOUT=50")); |
| 74 |
2/3✓ Branch 51 → 52 taken 11 times.
✗ Branch 52 → 53 not taken.
✓ Branch 52 → 60 taken 11 times.
|
11 | if (!db.open()) { |
| 75 | ✗ | *error = db.lastError().text(); | |
| 76 | ✗ | reset(); | |
| 77 | ✗ | return false; | |
| 78 | } | ||
| 79 | 11 | path = databasePath; | |
| 80 | 11 | identity = *current; | |
| 81 | 11 | return true; | |
| 82 | } | ||
| 83 | }; | ||
| 84 | } | ||
| 85 | |||
| 86 |
1/1✓ Branch 3 → 4 taken 23 times.
|
23 | HtmlInlineImageService::HtmlInlineImageService(QObject *parent) : QObject(parent) { |
| 87 |
1/1✓ Branch 6 → 7 taken 23 times.
|
23 | m_pool.setMaxThreadCount(1); |
| 88 |
1/1✓ Branch 7 → 8 taken 23 times.
|
23 | m_pool.setExpiryTimeout(-1); |
| 89 |
1/1✓ Branch 9 → 10 taken 23 times.
|
24 | connect(QCoreApplication::instance(), &QCoreApplication::aboutToQuit, this, [this] { stop(); }); |
| 90 | 23 | } | |
| 91 | 44 | HtmlInlineImageService::~HtmlInlineImageService() { stop(); } | |
| 92 | 8691 | HtmlInlineImageService *HtmlInlineImageService::shared() { | |
| 93 |
3/4✓ Branch 2 → 3 taken 21 times.
✓ Branch 2 → 7 taken 8670 times.
✓ Branch 4 → 5 taken 21 times.
✗ Branch 4 → 7 not taken.
|
8691 | static QPointer<HtmlInlineImageService> service; |
| 94 |
5/8✓ Branch 8 → 9 taken 21 times.
✓ Branch 8 → 16 taken 8670 times.
✓ Branch 11 → 12 taken 21 times.
✓ Branch 12 → 13 taken 21 times.
✗ Branch 13 → 14 not taken.
✓ Branch 13 → 15 taken 21 times.
✗ Branch 18 → 19 not taken.
✗ Branch 18 → 20 not taken.
|
8691 | if (!service) service = new HtmlInlineImageService(QCoreApplication::instance()); |
| 95 | 8691 | return service; | |
| 96 | } | ||
| 97 | 24 | void HtmlInlineImageService::stop() { | |
| 98 | 24 | m_stopping = true; | |
| 99 |
3/5✓ Branch 2 → 3 taken 24 times.
✓ Branch 3 → 4 taken 24 times.
✗ Branch 6 → 7 not taken.
✗ Branch 9 → 5 not taken.
✓ Branch 9 → 10 taken 24 times.
|
24 | for (const auto &request : m_pending) disconnect(request.destroyed); |
| 100 | 24 | m_pending.clear(); | |
| 101 |
1/2✗ Branch 12 → 13 not taken.
✓ Branch 12 → 18 taken 24 times.
|
24 | if (m_active) { m_active->cancelled->store(true); disconnect(m_active->destroyed); } |
| 102 |
1/1✓ Branch 19 → 20 taken 24 times.
|
24 | m_pool.waitForDone(); |
| 103 | 24 | } | |
| 104 | 8721 | void HtmlInlineImageService::cancel(QObject *owner) { | |
| 105 |
6/6✓ Branch 3 → 4 taken 12 times.
✓ Branch 3 → 8 taken 8709 times.
✓ Branch 6 → 7 taken 1 time.
✓ Branch 6 → 8 taken 11 times.
✓ Branch 9 → 10 taken 1 time.
✓ Branch 9 → 13 taken 8720 times.
|
8721 | if (m_active && m_active->owner == owner) m_active->cancelled->store(true); |
| 106 | 8721 | m_pending.removeIf([this, owner](const Request &request) { | |
| 107 |
5/6✓ Branch 3 → 4 taken 78 times.
✓ Branch 3 → 7 taken 1 time.
✓ Branch 5 → 6 taken 78 times.
✗ Branch 5 → 7 not taken.
✓ Branch 8 → 9 taken 78 times.
✓ Branch 8 → 10 taken 1 time.
|
79 | if (request.owner && request.owner != owner) return false; |
| 108 | 1 | disconnect(request.destroyed); | |
| 109 | 1 | return true; | |
| 110 | }); | ||
| 111 | 8721 | } | |
| 112 | 15 | void HtmlInlineImageService::load(QObject *owner, HtmlInlineImageRequest input, Callback callback) { | |
| 113 |
1/1✓ Branch 2 → 3 taken 15 times.
|
15 | cancel(owner); |
| 114 |
1/2✗ Branch 3 → 4 not taken.
✓ Branch 3 → 5 taken 15 times.
|
17 | if (!owner) return; |
| 115 | // Match the sanitizer's UTF-8 admission ceiling without cutting the existing | ||
| 116 | // full-HTML character capacity in half merely because QString uses UTF-16. | ||
| 117 | 15 | const qsizetype inputBytes = input.expectedHtml.size() > MaxInputBytes | |
| 118 |
2/2✓ Branch 6 → 7 taken 14 times.
✓ Branch 6 → 10 taken 1 time.
|
15 | ? MaxInputBytes + 1 |
| 119 |
1/1✓ Branch 8 → 9 taken 14 times.
|
14 | : MailJD::utf8ByteCountUpTo(input.expectedHtml, MaxInputBytes); |
| 120 | 15 | qsizetype bytes = inputBytes; | |
| 121 |
2/2✓ Branch 12 → 13 taken 12 times.
✓ Branch 12 → 15 taken 3 times.
|
15 | if (m_active) bytes += m_active->inputBytes; |
| 122 |
4/4✓ Branch 15 → 16 taken 15 times.
✓ Branch 16 → 17 taken 15 times.
✓ Branch 21 → 18 taken 38 times.
✓ Branch 21 → 22 taken 15 times.
|
53 | for (const auto &request : m_pending) bytes += request.inputBytes; |
| 123 |
4/4✓ Branch 24 → 25 taken 14 times.
✓ Branch 24 → 37 taken 1 time.
✓ Branch 25 → 26 taken 13 times.
✓ Branch 25 → 37 taken 1 time.
|
15 | if (m_stopping || m_pending.size() >= MaxPending || bytes > MaxInputBytes || |
| 124 |
3/6✓ Branch 27 → 28 taken 13 times.
✗ Branch 27 → 37 not taken.
✓ Branch 29 → 30 taken 13 times.
✗ Branch 29 → 37 not taken.
✓ Branch 31 → 32 taken 13 times.
✗ Branch 31 → 37 not taken.
|
26 | !input.mailId.isValid() || input.mailId.accountId.size() > 4096 || |
| 125 |
6/9✓ Branch 22 → 23 taken 15 times.
✗ Branch 22 → 37 not taken.
✓ Branch 33 → 34 taken 13 times.
✗ Branch 33 → 37 not taken.
✓ Branch 35 → 36 taken 13 times.
✗ Branch 36 → 37 not taken.
✓ Branch 36 → 38 taken 13 times.
✓ Branch 39 → 40 taken 2 times.
✓ Branch 39 → 47 taken 13 times.
|
43 | input.databasePath.isEmpty() || input.databasePath.size() > 4096 || input.databasePath.contains(QChar::Null)) { |
| 126 | 2 | callback({{}, QStringLiteral("Inline image request is unavailable or exceeds its budget")}); | |
| 127 | 2 | return; | |
| 128 | } | ||
| 129 |
3/11✓ Branch 47 → 48 taken 13 times.
✓ Branch 52 → 53 taken 13 times.
✓ Branch 53 → 54 taken 13 times.
✗ Branch 83 → 84 not taken.
✗ Branch 83 → 85 not taken.
✗ Branch 86 → 87 not taken.
✗ Branch 86 → 88 not taken.
✗ Branch 89 → 90 not taken.
✗ Branch 89 → 91 not taken.
✗ Branch 92 → 93 not taken.
✗ Branch 92 → 94 not taken.
|
13 | Request request{owner, std::move(input), std::move(callback), inputBytes}; |
| 130 | 13 | const auto cancelled = request.cancelled; | |
| 131 |
1/1✓ Branch 56 → 57 taken 13 times.
|
14 | request.destroyed = connect(owner, &QObject::destroyed, this, [cancelled] { cancelled->store(true); }); |
| 132 |
1/1✓ Branch 61 → 62 taken 13 times.
|
13 | m_pending.append(std::move(request)); |
| 133 |
1/1✓ Branch 62 → 63 taken 13 times.
|
13 | startNext(); |
| 134 | 17 | } | |
| 135 | 25 | void HtmlInlineImageService::startNext() { | |
| 136 |
5/6✓ Branch 2 → 3 taken 25 times.
✗ Branch 2 → 5 not taken.
✓ Branch 4 → 5 taken 10 times.
✓ Branch 4 → 6 taken 15 times.
✓ Branch 7 → 8 taken 10 times.
✓ Branch 7 → 9 taken 15 times.
|
28 | if (m_stopping || m_active) return; |
| 137 |
1/1✓ Branch 9 → 10 taken 15 times.
|
15 | cancel(nullptr); |
| 138 |
2/2✓ Branch 11 → 12 taken 3 times.
✓ Branch 11 → 13 taken 12 times.
|
15 | if (m_pending.isEmpty()) return; |
| 139 |
1/1✓ Branch 13 → 14 taken 12 times.
|
12 | m_active = m_pending.takeFirst(); |
| 140 |
3/6✓ Branch 16 → 17 taken 12 times.
✓ Branch 17 → 18 taken 12 times.
✗ Branch 18 → 19 not taken.
✓ Branch 18 → 20 taken 12 times.
✗ Branch 38 → 39 not taken.
✗ Branch 38 → 40 not taken.
|
12 | auto *watcher = new QFutureWatcher<HtmlInlineImageResult>(this); |
| 141 |
1/1✓ Branch 20 → 21 taken 12 times.
|
12 | connect(watcher, &QFutureWatcher<HtmlInlineImageResult>::finished, this, [this, watcher] { |
| 142 | 12 | auto request = std::move(*m_active); | |
| 143 | 12 | m_active.reset(); | |
| 144 |
1/1✓ Branch 6 → 7 taken 12 times.
|
12 | auto result = watcher->result(); |
| 145 |
1/1✓ Branch 7 → 8 taken 12 times.
|
12 | watcher->deleteLater(); |
| 146 |
1/1✓ Branch 8 → 9 taken 12 times.
|
12 | disconnect(request.destroyed); |
| 147 |
1/1✓ Branch 9 → 10 taken 12 times.
|
12 | const QPointer<HtmlInlineImageService> guard(this); |
| 148 |
7/9✓ Branch 10 → 11 taken 12 times.
✗ Branch 10 → 17 not taken.
✓ Branch 12 → 13 taken 12 times.
✗ Branch 12 → 17 not taken.
✓ Branch 15 → 16 taken 11 times.
✓ Branch 15 → 17 taken 1 time.
✓ Branch 18 → 19 taken 11 times.
✓ Branch 18 → 24 taken 1 time.
✓ Branch 21 → 22 taken 11 times.
|
12 | if (!m_stopping && request.owner && !request.cancelled->load()) request.callback(std::move(result)); |
| 149 |
2/3✓ Branch 25 → 26 taken 12 times.
✗ Branch 25 → 27 not taken.
✓ Branch 26 → 27 taken 12 times.
|
12 | if (guard) startNext(); |
| 150 | 12 | }); | |
| 151 | 12 | const auto input = m_active->input; | |
| 152 | 12 | const auto cancelled = m_active->cancelled; | |
| 153 |
2/4✓ Branch 28 → 29 taken 12 times.
✓ Branch 29 → 30 taken 12 times.
✗ Branch 47 → 48 not taken.
✗ Branch 47 → 49 not taken.
|
24 | watcher->setFuture(QtConcurrent::run(&m_pool, [input, cancelled] { return read(input, *cancelled); })); |
| 154 | 12 | } | |
| 155 | 22 | HtmlInlineImageResult HtmlInlineImageService::read(const HtmlInlineImageRequest &input, const std::atomic_bool &cancelled) { | |
| 156 |
2/2✓ Branch 3 → 4 taken 1 time.
✓ Branch 3 → 5 taken 21 times.
|
22 | if (cancelled.load()) return {}; |
| 157 | // This pool has one persistent worker. Reuse its read-only connection but | ||
| 158 | // finish each transaction, including cancelled/error reads, before the next | ||
| 159 | // message so WAL checkpoints and newer mailbox epochs remain visible. | ||
| 160 |
3/3✓ Branch 5 → 6 taken 3 times.
✓ Branch 5 → 8 taken 18 times.
✓ Branch 6 → 7 taken 3 times.
|
21 | static thread_local InlineReadConnection connection; |
| 161 |
4/4✓ Branch 8 → 9 taken 21 times.
✓ Branch 9 → 10 taken 21 times.
✓ Branch 11 → 12 taken 1 time.
✓ Branch 11 → 19 taken 20 times.
|
21 | if (!QFileInfo(input.databasePath).isFile()) { |
| 162 |
1/1✓ Branch 12 → 13 taken 1 time.
|
1 | connection.reset(); |
| 163 | 1 | return {{}, QStringLiteral("Inline image cache is unavailable")}; | |
| 164 | } | ||
| 165 | 20 | QString error; | |
| 166 |
2/3✓ Branch 19 → 20 taken 20 times.
✗ Branch 20 → 21 not taken.
✓ Branch 20 → 23 taken 20 times.
|
20 | if (!connection.ensure(input.databasePath, &error)) return {{}, error}; |
| 167 | 20 | auto &db = connection.db; | |
| 168 |
2/3✓ Branch 23 → 24 taken 20 times.
✗ Branch 24 → 25 not taken.
✓ Branch 24 → 33 taken 20 times.
|
20 | if (!db.transaction()) { |
| 169 | ✗ | error = db.lastError().text(); | |
| 170 | ✗ | connection.reset(); | |
| 171 | ✗ | return {{}, error}; | |
| 172 | } | ||
| 173 | 20 | HtmlInlineImageResult result; | |
| 174 | { | ||
| 175 | // A superseded mail must not keep this single foreground lane occupied | ||
| 176 | // while SQLite compares a large text_html value or scans CID metadata. | ||
| 177 | // Remove the handler before rollback or a connection reset can close the | ||
| 178 | // native handle; the cancellation token outlives this worker invocation. | ||
| 179 | 20 | sqlite3 *sqlite = nullptr; | |
| 180 |
2/2✓ Branch 33 → 34 taken 20 times.
✓ Branch 34 → 35 taken 20 times.
|
20 | if (auto handle = db.driver()->handle(); |
| 181 |
7/13✓ Branch 35 → 36 taken 20 times.
✓ Branch 36 → 37 taken 20 times.
✗ Branch 36 → 42 not taken.
✓ Branch 37 → 38 taken 20 times.
✓ Branch 38 → 39 taken 20 times.
✓ Branch 40 → 41 taken 20 times.
✗ Branch 40 → 42 not taken.
✓ Branch 43 → 44 taken 20 times.
✗ Branch 43 → 45 not taken.
✓ Branch 45 → 46 taken 20 times.
✗ Branch 45 → 48 not taken.
✗ Branch 92 → 93 not taken.
✗ Branch 92 → 94 not taken.
|
20 | handle.isValid() && QByteArray(handle.typeName()) == "sqlite3*") |
| 182 |
1/1✓ Branch 46 → 47 taken 20 times.
|
20 | sqlite = *static_cast<sqlite3 **>(handle.data()); |
| 183 |
2/3✓ Branch 49 → 50 taken 20 times.
✗ Branch 49 → 53 not taken.
✓ Branch 51 → 52 taken 20 times.
|
20 | if (sqlite) sqlite3_progress_handler(sqlite, 1000, [](void *context) { |
| 184 | ✗ | return static_cast<const std::atomic_bool *>(context)->load( | |
| 185 | ✗ | std::memory_order_relaxed) ? 1 : 0; | |
| 186 | }, const_cast<std::atomic_bool *>(&cancelled)); | ||
| 187 | 60 | const auto clearProgress = qScopeGuard([sqlite] { | |
| 188 |
1/2✓ Branch 2 → 3 taken 20 times.
✗ Branch 2 → 4 not taken.
|
20 | if (sqlite) sqlite3_progress_handler(sqlite, 0, nullptr, nullptr); |
| 189 | 20 | }); | |
| 190 | 20 | const auto readSnapshot = [&]() -> HtmlInlineImageResult { | |
| 191 |
1/1✓ Branch 2 → 3 taken 20 times.
|
20 | QSqlQuery header(db); |
| 192 |
1/1✓ Branch 6 → 7 taken 20 times.
|
20 | header.prepare(QStringLiteral("SELECT h.id FROM headers h JOIN folders f ON f.id=h.folder_id " |
| 193 | "JOIN bodies b ON b.header_id=h.id WHERE f.account=? AND h.folder_id=? AND h.uid=? " | ||
| 194 | "AND f.uidvalidity=? AND b.text_html=?")); | ||
| 195 |
1/1✓ Branch 11 → 12 taken 20 times.
|
20 | header.addBindValue(input.mailId.accountId); |
| 196 |
1/1✓ Branch 15 → 16 taken 20 times.
|
20 | header.addBindValue(input.mailId.folderId); |
| 197 |
1/1✓ Branch 19 → 20 taken 20 times.
|
20 | header.addBindValue(input.mailId.uid); |
| 198 |
1/1✓ Branch 23 → 24 taken 20 times.
|
20 | header.addBindValue(input.uidValidity); |
| 199 |
1/1✓ Branch 27 → 28 taken 20 times.
|
20 | header.addBindValue(input.expectedHtml); |
| 200 |
2/3✓ Branch 29 → 30 taken 20 times.
✗ Branch 30 → 31 not taken.
✓ Branch 30 → 36 taken 20 times.
|
20 | if (!header.exec()) return {{}, header.lastError().text()}; |
| 201 |
3/3✓ Branch 36 → 37 taken 20 times.
✓ Branch 37 → 38 taken 5 times.
✓ Branch 37 → 44 taken 15 times.
|
25 | if (!header.next()) return {{}, QStringLiteral("Inline image message identity or content changed")}; |
| 202 |
2/2✓ Branch 44 → 45 taken 15 times.
✓ Branch 45 → 46 taken 15 times.
|
15 | const auto headerId = header.value(0).toLongLong(); |
| 203 |
1/1✓ Branch 47 → 48 taken 15 times.
|
15 | header.finish(); |
| 204 | struct Metadata { qint64 id = 0, bytes = 0; QString type; int count = 0; }; | ||
| 205 | 15 | QHash<QString, Metadata> metadata; | |
| 206 | 15 | QStringList order; | |
| 207 |
1/1✓ Branch 48 → 49 taken 15 times.
|
15 | QSqlQuery rows(db); |
| 208 |
1/1✓ Branch 52 → 53 taken 15 times.
|
15 | rows.prepare(QStringLiteral("SELECT id,substr(content_id,1,4097),substr(content_type,1,257),length(data) " |
| 209 | "FROM attachments WHERE header_id=? AND presentation=1 ORDER BY id LIMIT 1001")); | ||
| 210 |
1/1✓ Branch 57 → 58 taken 15 times.
|
15 | rows.addBindValue(headerId); |
| 211 |
2/3✓ Branch 59 → 60 taken 15 times.
✗ Branch 60 → 61 not taken.
✓ Branch 60 → 66 taken 15 times.
|
15 | if (!rows.exec()) return {{}, rows.lastError().text()}; |
| 212 | 15 | int count = 0; | |
| 213 |
3/3✓ Branch 118 → 119 taken 34 times.
✓ Branch 119 → 67 taken 19 times.
✓ Branch 119 → 120 taken 15 times.
|
34 | while (rows.next()) { |
| 214 |
1/2✗ Branch 68 → 69 not taken.
✓ Branch 68 → 70 taken 19 times.
|
19 | if (cancelled.load()) return {}; |
| 215 |
1/2✗ Branch 70 → 71 not taken.
✓ Branch 70 → 77 taken 19 times.
|
19 | if (++count > 1000) return {{}, QStringLiteral("Inline image metadata exceeds its budget")}; |
| 216 |
2/2✓ Branch 77 → 78 taken 19 times.
✓ Branch 78 → 79 taken 19 times.
|
19 | const auto rawId = rows.value(1).toString(); |
| 217 |
1/2✗ Branch 81 → 82 not taken.
✓ Branch 81 → 88 taken 19 times.
|
19 | if (rawId.size() > 4096) return {{}, QStringLiteral("Inline image identifier exceeds its budget")}; |
| 218 |
1/1✓ Branch 88 → 89 taken 19 times.
|
19 | const auto id = MimeParser::normalizeContentId(rawId); |
| 219 |
1/2✗ Branch 90 → 91 not taken.
✓ Branch 90 → 92 taken 19 times.
|
19 | if (id.isEmpty()) continue; |
| 220 |
3/3✓ Branch 93 → 94 taken 18 times.
✓ Branch 93 → 95 taken 1 time.
✓ Branch 94 → 95 taken 18 times.
|
19 | if (!metadata.contains(id)) order.append(id); |
| 221 |
1/1✓ Branch 95 → 96 taken 19 times.
|
19 | auto &item = metadata[id]; |
| 222 | 19 | ++item.count; | |
| 223 |
2/2✓ Branch 96 → 97 taken 19 times.
✓ Branch 97 → 98 taken 19 times.
|
19 | item.id = rows.value(0).toLongLong(); |
| 224 |
2/2✓ Branch 99 → 100 taken 19 times.
✓ Branch 100 → 101 taken 19 times.
|
19 | item.type = rows.value(2).toString(); |
| 225 |
2/2✓ Branch 104 → 105 taken 19 times.
✓ Branch 105 → 106 taken 19 times.
|
19 | item.bytes = rows.value(3).toLongLong(); |
| 226 | 19 | } | |
| 227 |
3/4✓ Branch 120 → 121 taken 15 times.
✓ Branch 121 → 122 taken 15 times.
✗ Branch 123 → 124 not taken.
✓ Branch 123 → 129 taken 15 times.
|
15 | if (rows.lastError().isValid()) return {{}, rows.lastError().text()}; |
| 228 |
1/1✓ Branch 129 → 130 taken 15 times.
|
15 | rows.finish(); |
| 229 | 15 | HtmlInlineImageResult output; | |
| 230 |
2/2✓ Branch 130 → 131 taken 15 times.
✓ Branch 132 → 133 taken 15 times.
|
15 | output.contentDigest = QCryptographicHash::hash(input.expectedHtml.toUtf8(), QCryptographicHash::Sha256); |
| 231 | 15 | qint64 aggregate = 0; | |
| 232 |
4/4✓ Branch 136 → 137 taken 15 times.
✓ Branch 137 → 138 taken 15 times.
✓ Branch 230 → 139 taken 18 times.
✓ Branch 230 → 231 taken 15 times.
|
33 | for (const auto &id : order) { |
| 233 |
1/2✗ Branch 141 → 142 not taken.
✓ Branch 141 → 143 taken 18 times.
|
18 | if (cancelled.load()) return {}; |
| 234 | 18 | const auto item = metadata.value(id); | |
| 235 |
2/4✓ Branch 146 → 147 taken 17 times.
✗ Branch 146 → 150 not taken.
✓ Branch 147 → 148 taken 17 times.
✗ Branch 147 → 150 not taken.
|
17 | if (item.count != 1 || item.type.size() > 256 || item.bytes <= 0 || |
| 236 |
5/6✓ Branch 144 → 145 taken 17 times.
✓ Branch 144 → 150 taken 1 time.
✓ Branch 148 → 149 taken 17 times.
✗ Branch 148 → 150 not taken.
✓ Branch 152 → 153 taken 1 time.
✓ Branch 152 → 154 taken 17 times.
|
52 | item.bytes > MimeParser::MaxInlineImageBytes || |
| 237 |
1/2✗ Branch 149 → 150 not taken.
✓ Branch 149 → 151 taken 17 times.
|
18 | item.bytes > MimeParser::MaxInlineImageAggregateBytes - aggregate) continue; |
| 238 |
1/1✓ Branch 154 → 155 taken 17 times.
|
17 | QSqlQuery data(db); |
| 239 |
1/1✓ Branch 158 → 159 taken 17 times.
|
17 | data.prepare(QStringLiteral("SELECT data FROM attachments WHERE id=? AND header_id=?")); |
| 240 |
1/1✓ Branch 163 → 164 taken 17 times.
|
17 | data.addBindValue(item.id); |
| 241 |
1/1✓ Branch 167 → 168 taken 17 times.
|
17 | data.addBindValue(headerId); |
| 242 |
2/3✓ Branch 169 → 170 taken 17 times.
✗ Branch 170 → 171 not taken.
✓ Branch 170 → 176 taken 17 times.
|
17 | if (!data.exec()) return {{}, data.lastError().text()}; |
| 243 |
2/3✓ Branch 176 → 177 taken 17 times.
✗ Branch 177 → 178 not taken.
✓ Branch 177 → 184 taken 17 times.
|
17 | if (!data.next()) return {{}, QStringLiteral("Inline image disappeared inside its snapshot")}; |
| 244 |
2/2✓ Branch 184 → 185 taken 17 times.
✓ Branch 185 → 186 taken 17 times.
|
17 | const auto bytes = data.value(0).toByteArray(); |
| 245 | 17 | aggregate += bytes.size(); // Rejected bytes also consume the read budget. | |
| 246 |
3/3✓ Branch 188 → 189 taken 17 times.
✓ Branch 189 → 190 taken 2 times.
✓ Branch 189 → 191 taken 15 times.
|
17 | if (!MimeParser::isAllowedInlineImage(item.type, bytes)) continue; |
| 247 |
3/3✓ Branch 193 → 194 taken 15 times.
✓ Branch 194 → 195 taken 15 times.
✓ Branch 195 → 196 taken 15 times.
|
15 | const auto type = item.type.section(';', 0, 0).trimmed().toLower(); |
| 248 |
4/4✓ Branch 202 → 203 taken 15 times.
✓ Branch 203 → 204 taken 15 times.
✓ Branch 204 → 205 taken 15 times.
✓ Branch 205 → 206 taken 15 times.
|
30 | output.images.insert(id, QStringLiteral("data:%1;base64,%2").arg(type, QString::fromLatin1(bytes.toBase64()))); |
| 249 | 22 | } | |
| 250 | 15 | return output; | |
| 251 |
0/16✗ Branch 31 → 32 not taken.
✗ Branch 32 → 33 not taken.
✗ Branch 61 → 62 not taken.
✗ Branch 62 → 63 not taken.
✗ Branch 124 → 125 not taken.
✗ Branch 125 → 126 not taken.
✗ Branch 171 → 172 not taken.
✗ Branch 172 → 173 not taken.
✗ Branch 268 → 269 not taken.
✗ Branch 268 → 270 not taken.
✗ Branch 287 → 288 not taken.
✗ Branch 287 → 289 not taken.
✗ Branch 314 → 315 not taken.
✗ Branch 314 → 316 not taken.
✗ Branch 339 → 340 not taken.
✗ Branch 339 → 341 not taken.
|
30 | }; |
| 252 |
1/1✓ Branch 54 → 55 taken 20 times.
|
20 | result = readSnapshot(); |
| 253 | 20 | } | |
| 254 | // S106-040: after a failed read the next request starts on a fresh | ||
| 255 | // connection (a handle broken by an SQL error was kept for the process life). | ||
| 256 |
3/4✓ Branch 59 → 60 taken 5 times.
✓ Branch 59 → 63 taken 15 times.
✓ Branch 61 → 62 taken 5 times.
✗ Branch 61 → 63 not taken.
|
20 | const bool sqlError = !result.error.isEmpty() && !cancelled.load(); |
| 257 |
2/3✓ Branch 64 → 65 taken 20 times.
✗ Branch 65 → 66 not taken.
✓ Branch 65 → 74 taken 20 times.
|
20 | if (!db.rollback()) { |
| 258 | ✗ | error = db.lastError().text(); | |
| 259 | ✗ | connection.reset(); | |
| 260 | ✗ | return {{}, error}; | |
| 261 | } | ||
| 262 |
2/2✓ Branch 74 → 75 taken 5 times.
✓ Branch 74 → 76 taken 15 times.
|
20 | if (sqlError) |
| 263 |
1/1✓ Branch 75 → 76 taken 5 times.
|
5 | connection.reset(); |
| 264 |
1/2✗ Branch 77 → 78 not taken.
✓ Branch 77 → 79 taken 20 times.
|
20 | return cancelled.load() ? HtmlInlineImageResult{} : result; |
| 265 | 22 | } | |
| 266 |