2 #include <QCoreApplication>
8 #include <QNetworkCookie>
9 #include <QAuthenticator>
10 #include <QTextStream>
11 #include <QNetworkProxy>
12 #include <QMutexLocker>
34 #define LOC QString("DownloadManager: ")
35 #define CACHE_REDIRECTION_LIMIT 10
49 qRegisterMetaType<QNetworkReply::NetworkError>(
"QNetworkReply::NetworkError");
61 QMutexLocker lock(&
m_lock);
67 QMutexLocker lock(&
m_lock);
91 const QHash<QByteArray, QByteArray> *
m_headers {
nullptr};
93 QNetworkReply::NetworkError
m_errorCode {QNetworkReply::NoError};
159 while (!tmpDLM->getQueueThread())
162 tmpDLM->moveToThread(tmpDLM->getQueueThread());
163 tmpDLM->setRunThread();
165 while (!tmpDLM->isRunning())
195 bool downloading =
false;
196 bool itemsInQueue =
false;
197 bool itemsInCancellationQueue =
false;
198 bool waitAnyway =
false;
205 m_manager =
new QNetworkAccessManager(
this);
209 QCoreApplication::applicationName() +
"-" +
219 m_manager->cookieJar()->setParent(
nullptr);
221 QObject::connect(
m_manager, &QNetworkAccessManager::finished,
229 LOG(VB_GENERAL, LOG_DEBUG,
"Updating DLManager's Cookie Jar");
239 if (itemsInCancellationQueue)
244 QCoreApplication::processEvents();
250 if (!itemsInQueue || waitAnyway)
257 LOG(VB_FILE, LOG_DEBUG,
LOC + QString(
"waiting 200ms"));
262 LOG(VB_FILE, LOG_DEBUG,
LOC + QString(
"waiting for more items to download"));
294 if (dlInfo->
m_url.startsWith(
"myth://"))
322 const QString &
dest, QByteArray *data,
329 dlInfo->m_request = req;
330 dlInfo->m_outFile =
dest;
331 dlInfo->m_data = data;
332 dlInfo->m_caller = caller;
333 dlInfo->m_requestType = reqType;
334 dlInfo->m_reload = reload;
354 const QString &
dest, QByteArray *data,
358 const QHash<QByteArray, QByteArray> *headers,
364 dlInfo->m_request = req;
365 dlInfo->m_outFile =
dest;
366 dlInfo->m_data = data;
367 dlInfo->m_requestType = reqType;
368 dlInfo->m_reload = reload;
369 dlInfo->m_syncMode =
true;
370 dlInfo->m_authCallback = authCallbackFn;
371 dlInfo->m_authArg = authArg;
372 dlInfo->m_headers = headers;
373 dlInfo->m_finalUrl = finalUrl;
383 LOG(VB_FILE, LOG_DEBUG,
LOC + QString(
"preCache('%1')").
arg(url));
384 queueItem(url,
nullptr, QString(),
nullptr,
nullptr);
398 LOG(VB_FILE, LOG_DEBUG,
LOC + QString(
"queueDownload('%1', '%2', %3)")
413 LOG(VB_FILE, LOG_DEBUG,
LOC + QString(
"queueDownload('%1', '%2', %3)")
414 .
arg(req->url().toString()).arg((
long long)data)
415 .arg((
long long)caller));
417 queueItem(req->url().toString(), req, QString(), data, caller,
419 (QNetworkRequest::AlwaysNetwork == req->attribute(
420 QNetworkRequest::CacheLoadControlAttribute,
421 QNetworkRequest::PreferNetwork).toInt()));
443 const bool reload, QString *finalUrl)
447 nullptr,
nullptr,
nullptr, &redirected))
449 if (!redirected.isEmpty() && finalUrl !=
nullptr)
450 *finalUrl = redirected;
464 QNetworkReply *reply =
nullptr;
467 dlInfo->m_reload = reload;
468 dlInfo->m_syncMode =
true;
469 dlInfo->m_processReply =
false;
475 reply = dlInfo->m_reply;
477 dlInfo->m_reply =
nullptr;
497 LOG(VB_FILE, LOG_DEBUG,
LOC + QString(
"download('%1', '%2')")
498 .
arg(req->url().toString()).arg((
long long)data));
499 return processItem(req->url().toString(), req, QString(), data,
501 (QNetworkRequest::AlwaysNetwork == req->attribute(
502 QNetworkRequest::CacheLoadControlAttribute,
503 QNetworkRequest::PreferNetwork).toInt()));
516 const bool reload,
AuthCallback authCallbackFn,
void *authArg,
517 const QHash<QByteArray, QByteArray> *headers)
533 LOG(VB_FILE, LOG_DEBUG,
LOC + QString(
"queuePost('%1', '%2')")
534 .
arg(url).
arg((
long long)data));
538 LOG(VB_GENERAL, LOG_ERR,
LOC +
"queuePost(), data is NULL!");
554 LOG(VB_FILE, LOG_DEBUG,
LOC + QString(
"queuePost('%1', '%2')")
555 .
arg(req->url().toString()).arg((
long long)data));
559 LOG(VB_GENERAL, LOG_ERR,
LOC +
"queuePost(), data is NULL!");
563 queueItem(req->url().toString(), req, QString(), data, caller,
565 (QNetworkRequest::AlwaysNetwork == req->attribute(
566 QNetworkRequest::CacheLoadControlAttribute,
567 QNetworkRequest::PreferNetwork).toInt()));
578 LOG(VB_FILE, LOG_DEBUG,
LOC + QString(
"post('%1', '%2')")
579 .
arg(url).
arg((
long long)data));
583 LOG(VB_GENERAL, LOG_ERR,
LOC +
"post(), data is NULL!");
597 LOG(VB_FILE, LOG_DEBUG,
LOC + QString(
"post('%1', '%2')")
598 .
arg(req->url().toString()).arg((
long long)data));
602 LOG(VB_GENERAL, LOG_ERR,
LOC +
"post(), data is NULL!");
606 return processItem(req->url().toString(), req, QString(), data,
608 (QNetworkRequest::AlwaysNetwork == req->attribute(
609 QNetworkRequest::CacheLoadControlAttribute,
610 QNetworkRequest::PreferNetwork).toInt()));
624 const QHash<QByteArray, QByteArray> *headers)
626 LOG(VB_FILE, LOG_DEBUG,
LOC + QString(
"postAuth('%1', '%2')")
627 .
arg(url).
arg((
long long)data));
631 LOG(VB_GENERAL, LOG_ERR,
LOC +
"postAuth(), data is NULL!");
656 static const QString kDateFormat =
"ddd, dd MMM yyyy hh:mm:ss 'GMT'";
657 QUrl qurl(dlInfo->
m_url);
658 QNetworkRequest request;
667 request.setUrl(qurl);
671 request.setAttribute(QNetworkRequest::CacheLoadControlAttribute,
672 QNetworkRequest::AlwaysNetwork);
683 while (!(redirectLoc =
getHeader(qurl,
"Location")).isNull())
687 LOG(VB_GENERAL, LOG_WARNING, QString(
"Cache Redirection limit "
689 .
arg(qurl.toString()));
692 qurl.setUrl(redirectLoc);
696 LOG(VB_NETWORK, LOG_DEBUG, QString(
"Checking cache for %1")
697 .
arg(qurl.toString()));
700 QNetworkCacheMetaData urlData =
m_manager->cache()->metaData(qurl);
702 if ((urlData.isValid()) &&
703 ((!urlData.expirationDate().isValid()) ||
704 (urlData.expirationDate().toUTC().secsTo(now) < 10)))
706 QString dateString =
getHeader(urlData,
"Date");
708 if (!dateString.isNull())
712 loadDate.setTimeSpec(Qt::UTC);
713 if (loadDate.secsTo(now) <= 720)
716 LOG(VB_NETWORK, LOG_DEBUG, QString(
"Preferring cache for %1")
717 .
arg(qurl.toString()));
724 request.setAttribute(QNetworkRequest::CacheLoadControlAttribute,
725 QNetworkRequest::PreferCache);
727 if (!request.hasRawHeader(
"User-Agent"))
729 request.setRawHeader(
"User-Agent",
731 " MythDownloadManager");
736 QHash<QByteArray, QByteArray>::const_iterator it =
738 for ( ; it != dlInfo->
m_headers->constEnd(); ++it )
740 if (!it.key().isEmpty() && !it.value().isEmpty())
742 request.setRawHeader(it.key(), it.value());
765 connect(
m_manager, &QNetworkAccessManager::authenticationRequired,
769 #if QT_VERSION < QT_VERSION_CHECK(5,15,0)
773 connect(dlInfo->
m_reply, &QNetworkReply::errorOccurred,
776 connect(dlInfo->
m_reply, &QNetworkReply::downloadProgress,
785 QAuthenticator *authenticator)
797 LOG(VB_FILE, LOG_DEBUG,
"Calling auth callback");
819 if (dlInfo->
m_url.startsWith(
"http://[fe80::",Qt::CaseInsensitive))
820 return downloadNowLinkLocal(dlInfo, deleteInfo);
832 while ((!dlInfo->
IsDone()) &&
834 (((!dlInfo->
m_url.startsWith(
"myth://")) &&
836 ((dlInfo->
m_url.startsWith(
"myth://")) &&
845 bool done = dlInfo->
IsDone();
847 done && (dlInfo->
m_errorCode == QNetworkReply::NoError);
856 LOG(VB_FILE, LOG_DEBUG,
857 LOC + QString(
"Aborting download - lack of data transfer"));
890 bool MythDownloadManager::downloadNowLinkLocal(
MythDownloadInfo *dlInfo,
bool deleteInfo)
897 LOG(VB_GENERAL, LOG_ERR,
LOC + QString(
"No data buffer provided for %1").
arg(dlInfo->
m_url));
904 LOG(VB_GENERAL, LOG_ERR,
LOC + QString(
"Unsupported authentication for %1").
arg(dlInfo->
m_url));
910 LOG(VB_GENERAL, LOG_ERR,
LOC + QString(
"Unsupported File output %1 for %2")
919 LOG(VB_GENERAL, LOG_ERR,
LOC + QString(
"Unsupported link-local operation %1")
924 QUrl url(dlInfo->
m_url);
925 QString host(url.host());
926 int port(url.port(80));
940 QByteArray* buffer = dlInfo->
m_data;
941 QHash<QByteArray, QByteArray> headers;
944 if (!headers.contains(
"User-Agent"))
946 headers.insert(
"Connection",
"close");
947 headers.insert(
"Accept-Encoding",
"identity");
948 if (!buffer->isEmpty())
949 headers.insert(
"Content-Length", QString::number(buffer->size()).toUtf8());
950 headers.insert(
"Host", (url.host() +
":" + QString::number(port)).toUtf8());
952 QByteArray requestMessage;
953 QString path (url.path());
954 requestMessage.append(
"POST ");
955 requestMessage.append(path);
956 requestMessage.append(
" HTTP/1.1\r\n");
957 QHashIterator<QByteArray, QByteArray> it(headers);
961 requestMessage.append(it.key());
962 requestMessage.append(
": ");
963 requestMessage.append(it.value());
964 requestMessage.append(
"\r\n");
966 requestMessage.append(
"\r\n");
967 if (!buffer->isEmpty())
968 requestMessage.append(*buffer);
971 socket.connectToHost(host,
static_cast<uint16_t>(port));
973 if (!socket.waitForConnected(5000))
976 ok = socket.write(requestMessage) > 0;
979 ok = socket.waitForDisconnected(5000);
982 *buffer = socket.readAll();
984 QByteArray delim(
"\r\n\r\n");
985 int delimLoc = buffer->indexOf(delim);
987 *buffer = buffer->right(buffer->size() - delimLoc - 4);
1004 LOG(VB_GENERAL, LOG_ERR,
LOC + QString(
"Link Local request failed: %1").
arg(url.toString()));
1025 for (
const auto& url : qAsConst(urls))
1028 while (lit.hasNext())
1032 if (dlInfo->
m_url == url)
1055 if (QThread::currentThread() == this->thread())
1078 while (lit.hasNext())
1086 LOG(VB_FILE, LOG_DEBUG,
1087 LOC + QString(
"Aborting download - user request"));
1096 dlInfo->
m_errorCode = QNetworkReply::OperationCanceledError;
1118 dlInfo->
m_data =
nullptr;
1122 QMap <QString, MythDownloadInfo*>::iterator mit =
m_downloadInfos.begin();
1130 dlInfo->
m_data =
nullptr;
1140 auto *reply = qobject_cast<QNetworkReply *>(sender());
1141 if (reply ==
nullptr)
1144 LOG(VB_FILE, LOG_DEBUG,
LOC + QString(
"downloadError %1 ")
1145 .
arg(errorCode) + reply->errorString() );
1150 reply->deleteLater();
1168 const QUrl& oldRedirectUrl)
1170 LOG(VB_FILE, LOG_DEBUG,
LOC + QString(
"redirectUrl()"));
1173 if(!possibleRedirectUrl.isEmpty() && possibleRedirectUrl != oldRedirectUrl)
1184 LOG(VB_FILE, LOG_DEBUG,
LOC + QString(
"downloadFinished(%1)")
1185 .
arg((
long long)reply));
1190 reply->deleteLater();
1196 if (!dlInfo || !dlInfo->
m_reply)
1210 int statusCode = -1;
1211 static const QString kDateFormat =
"ddd, dd MMM yyyy hh:mm:ss 'GMT'";
1212 QNetworkReply *reply = dlInfo->
m_reply;
1216 QUrl possibleRedirectUrl =
1217 reply->attribute(QNetworkRequest::RedirectionTargetAttribute).toUrl();
1219 if (!possibleRedirectUrl.isEmpty() &&
1220 possibleRedirectUrl.isValid() &&
1221 possibleRedirectUrl.isRelative())
1222 possibleRedirectUrl = reply->url().resolved(possibleRedirectUrl);
1224 if (!possibleRedirectUrl.isEmpty() && dlInfo->
m_finalUrl !=
nullptr)
1225 *dlInfo->
m_finalUrl = QString(possibleRedirectUrl.toString());
1231 reply->attribute(QNetworkRequest::HttpStatusCodeAttribute);
1232 if (status.isValid())
1233 statusCode = status.toInt();
1238 (statusCode == 301 || statusCode == 302 ||
1239 statusCode == 303)))
1241 LOG(VB_FILE, LOG_DEBUG,
LOC +
1242 QString(
"downloadFinished(%1): Redirect: %2 -> %3")
1243 .
arg((
long long)dlInfo)
1244 .
arg(reply->url().toString())
1245 .arg(dlInfo->m_redirectedTo.toString()));
1248 dlInfo->m_data->clear();
1250 dlInfo->m_bytesReceived = 0;
1251 dlInfo->m_bytesTotal = 0;
1253 QNetworkRequest request(dlInfo->m_redirectedTo);
1255 if (dlInfo->m_reload)
1257 request.setAttribute(QNetworkRequest::CacheLoadControlAttribute,
1258 QNetworkRequest::AlwaysNetwork);
1260 else if (dlInfo->m_preferCache)
1262 request.setAttribute(QNetworkRequest::CacheLoadControlAttribute,
1263 QNetworkRequest::PreferCache);
1266 request.setRawHeader(
"User-Agent",
1269 switch (dlInfo->m_requestType)
1272 dlInfo->m_reply =
m_manager->head(request);
1276 dlInfo->m_reply =
m_manager->get(request);
1282 #if QT_VERSION < QT_VERSION_CHECK(5,15,0)
1286 connect(dlInfo->m_reply, &QNetworkReply::errorOccurred,
1289 connect(dlInfo->m_reply, &QNetworkReply::downloadProgress,
1293 reply->deleteLater();
1297 LOG(VB_FILE, LOG_DEBUG, QString(
"downloadFinished(%1): COMPLETE: %2")
1298 .
arg((
long long)dlInfo).
arg(dlInfo->m_url));
1302 QUrl fileUrl = dlInfo->m_url;
1303 QString redirectLoc;
1305 while (!(redirectLoc =
getHeader(fileUrl,
"Location")).isNull())
1309 LOG(VB_GENERAL, LOG_WARNING, QString(
"Cache Redirection limit "
1311 .
arg(fileUrl.toString()));
1314 fileUrl.setUrl(redirectLoc);
1319 QNetworkCacheMetaData urlData =
m_manager->cache()->metaData(fileUrl);
1321 if (
getHeader(urlData,
"Date").isNull())
1323 QNetworkCacheMetaData::RawHeaderList headers = urlData.rawHeaders();
1324 QNetworkCacheMetaData::RawHeader newheader;
1326 newheader = QNetworkCacheMetaData::RawHeader(
"Date",
1327 now.toString(kDateFormat).toLatin1());
1328 headers.append(newheader);
1329 urlData.setRawHeaders(headers);
1331 m_manager->cache()->updateMetaData(urlData);
1336 dlInfo->m_redirectedTo.clear();
1342 if (reply && dlInfo->m_processReply)
1344 bool append = (!dlInfo->m_syncMode && dlInfo->m_caller);
1345 QByteArray data = reply->readAll();
1346 dataSize = data.size();
1349 dlInfo->m_bytesReceived += dataSize;
1351 dlInfo->m_bytesReceived = dataSize;
1353 dlInfo->m_bytesTotal = dlInfo->m_bytesReceived;
1358 dlInfo->m_data->append(data);
1360 *dlInfo->m_data = data;
1362 else if (!dlInfo->m_outFile.isEmpty())
1364 saveFile(dlInfo->m_outFile, data, append);
1371 (*dlInfo->m_data) = dlInfo->m_privData;
1373 else if (!dlInfo->m_outFile.isEmpty())
1375 saveFile(dlInfo->m_outFile, dlInfo->m_privData);
1377 dlInfo->m_bytesReceived += dataSize;
1378 dlInfo->m_bytesTotal = dlInfo->m_bytesReceived;
1386 LOG(VB_GENERAL, LOG_ERR,
LOC +
1387 QString(
"ERROR download finished but failed to remove url: %1")
1388 .
arg(dlInfo->m_url));
1395 dlInfo->SetDone(
true);
1397 if (!dlInfo->m_syncMode)
1399 if (dlInfo->m_caller)
1401 LOG(VB_FILE, LOG_DEBUG, QString(
"downloadFinished(%1): "
1402 "COMPLETE: %2, sending event to caller")
1403 .
arg((
long long)dlInfo).
arg(dlInfo->m_url));
1406 args << dlInfo->m_url;
1407 args << dlInfo->m_outFile;
1408 args << QString::number(dlInfo->m_bytesTotal);
1410 args << (reply ? reply->errorString() : QString());
1411 args << QString::number((
int)(reply ? reply->error() :
1412 dlInfo->m_errorCode));
1414 QCoreApplication::postEvent(dlInfo->m_caller,
1433 auto *reply = qobject_cast<QNetworkReply *>(sender());
1434 if (reply ==
nullptr)
1437 LOG(VB_FILE, LOG_DEBUG,
LOC +
1438 QString(
"downloadProgress(%1, %2) (for reply %3)")
1439 .
arg(bytesReceived).
arg(bytesTotal).
arg((
long long)reply));
1452 LOG(VB_FILE, LOG_DEBUG,
LOC +
1453 QString(
"downloadProgress: %1 to %2 is at %3 of %4 bytes downloaded")
1455 .arg(bytesReceived).arg(bytesTotal));
1459 LOG(VB_FILE, LOG_DEBUG, QString(
"downloadProgress(%1): "
1460 "sending event to caller")
1461 .
arg(reply->url().toString()));
1464 QByteArray data = reply->readAll();
1469 dlInfo->
m_data->append(data);
1477 args << QString::number(bytesReceived);
1478 args << QString::number(bytesTotal);
1480 QCoreApplication::postEvent(dlInfo->
m_caller,
1493 const QByteArray &data,
1496 if (outFile.isEmpty() || data.isEmpty())
1499 QFile
file(outFile);
1501 QDir qdir(
fileInfo.absolutePath());
1503 if (!qdir.exists() && !qdir.mkpath(
fileInfo.absolutePath()))
1505 LOG(VB_GENERAL, LOG_ERR, QString(
"Failed to create: '%1'")
1510 QIODevice::OpenMode mode = QIODevice::Unbuffered|QIODevice::WriteOnly;
1512 mode |= QIODevice::Append;
1514 if (!
file.open(mode))
1516 LOG(VB_GENERAL, LOG_ERR, QString(
"Failed to open: '%1'") .
arg(outFile));
1521 size_t remaining = data.size();
1522 uint failure_cnt = 0;
1523 while ((remaining > 0) && (failure_cnt < 5))
1525 ssize_t written =
file.write(data.data() + offset, remaining);
1535 remaining -= written;
1538 return remaining <= 0;
1552 static const QString kDateFormat =
"ddd, dd MMM yyyy hh:mm:ss 'GMT'";
1553 LOG(VB_FILE, LOG_DEBUG,
LOC + QString(
"GetLastModified('%1')").
arg(url));
1558 QUrl cacheUrl = QUrl(url);
1561 QString redirectLoc;
1563 while (!(redirectLoc =
getHeader(cacheUrl,
"Location")).isNull())
1567 LOG(VB_GENERAL, LOG_WARNING, QString(
"Cache Redirection limit "
1569 .
arg(cacheUrl.toString()));
1572 cacheUrl.setUrl(redirectLoc);
1577 QNetworkCacheMetaData urlData =
m_manager->cache()->metaData(cacheUrl);
1580 if (urlData.isValid() &&
1581 ((!urlData.expirationDate().isValid()) ||
1582 (urlData.expirationDate().secsTo(now) < 0)))
1584 if (urlData.lastModified().toUTC().secsTo(now) <= 3600)
1586 result = urlData.lastModified().toUTC();
1590 QString date =
getHeader(urlData,
"Date");
1593 QDateTime loadDate =
1595 loadDate.setTimeSpec(Qt::UTC);
1596 if (loadDate.secsTo(now) <= 1200)
1598 result = urlData.lastModified().toUTC();
1604 if (!result.isValid())
1607 dlInfo->
m_url = url;
1618 QNetworkRequest::LastModifiedHeader);
1619 if (lastMod.isValid())
1620 result = lastMod.toDateTime().toUTC();
1629 LOG(VB_FILE, LOG_DEBUG,
LOC + QString(
"GetLastModified('%1'): Result %2")
1630 .
arg(url).
arg(result.toString()));
1658 auto *jar = qobject_cast<MythCookieJar *>(
m_manager->cookieJar());
1680 auto *inJar = qobject_cast<MythCookieJar *>(
m_manager->cookieJar());
1681 if (inJar ==
nullptr)
1697 auto *inJar = qobject_cast<MythCookieJar *>(jar);
1698 if (inJar ==
nullptr)
1716 if (inJar !=
nullptr)
1733 QNetworkCacheMetaData metadata =
m_manager->cache()->metaData(url);
1745 const QString& header)
1747 auto headers = cacheData.rawHeaders();
1748 for (
const auto& rh : qAsConst(headers))
1749 if (QString(rh.first) == header)
1750 return QString(rh.second);
1760 const QList<QNetworkCookie> cookieList = old.allCookies();
1761 setAllCookies(cookieList);
1769 LOG(VB_GENERAL, LOG_DEBUG, QString(
"MythCookieJar: loading cookies from: %1").
arg(
filename));
1772 if (!
f.open(QIODevice::ReadOnly))
1774 LOG(VB_GENERAL, LOG_WARNING, QString(
"MythCookieJar::load() failed to open file for reading: %1").
arg(
filename));
1778 QList<QNetworkCookie> cookieList;
1779 QTextStream stream(&
f);
1780 while (!stream.atEnd())
1782 QString cookie = stream.readLine();
1783 cookieList << QNetworkCookie::parseCookies(cookie.toLocal8Bit());
1786 setAllCookies(cookieList);
1794 LOG(VB_GENERAL, LOG_DEBUG, QString(
"MythCookieJar: saving cookies to: %1").
arg(
filename));
1797 if (!
f.open(QIODevice::WriteOnly))
1799 LOG(VB_GENERAL, LOG_ERR, QString(
"MythCookieJar::save() failed to open file for writing: %1").
arg(
filename));
1803 QList<QNetworkCookie> cookieList = allCookies();
1804 QTextStream stream(&
f);
1806 #if QT_VERSION < QT_VERSION_CHECK(5,14,0)
1807 for (
const auto& cookie : qAsConst(cookieList))
1808 stream << cookie.toRawForm() << endl;
1810 for (
const auto& cookie : qAsConst(cookieList))
1811 stream << cookie.toRawForm() << Qt::endl;