2 #include <QCoreApplication>
8 #include <QNetworkCookie>
9 #include <QAuthenticator>
10 #include <QTextStream>
11 #include <QNetworkProxy>
12 #include <QMutexLocker>
26 #include "mythversion.h"
34 #define LOC QString("DownloadManager: ")
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;
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)")
399 .arg(url,
dest, QString::number((
long long)caller)));
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",
730 QByteArray(
"MythTV v") + MYTH_BINARY_VERSION +
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 connect(dlInfo->
m_reply, &QNetworkReply::errorOccurred,
771 connect(dlInfo->
m_reply, &QNetworkReply::downloadProgress,
780 QAuthenticator *authenticator)
792 LOG(VB_FILE, LOG_DEBUG,
"Calling auth callback");
814 if (dlInfo->
m_url.startsWith(
"http://[fe80::",Qt::CaseInsensitive))
815 return downloadNowLinkLocal(dlInfo, deleteInfo);
827 while ((!dlInfo->
IsDone()) &&
829 (((!dlInfo->
m_url.startsWith(
"myth://")) &&
831 ((dlInfo->
m_url.startsWith(
"myth://")) &&
840 bool done = dlInfo->
IsDone();
842 done && (dlInfo->
m_errorCode == QNetworkReply::NoError);
851 LOG(VB_FILE, LOG_DEBUG,
852 LOC + QString(
"Aborting download - lack of data transfer"));
885 bool MythDownloadManager::downloadNowLinkLocal(
MythDownloadInfo *dlInfo,
bool deleteInfo)
892 LOG(VB_GENERAL, LOG_ERR,
LOC + QString(
"No data buffer provided for %1").arg(dlInfo->
m_url));
899 LOG(VB_GENERAL, LOG_ERR,
LOC + QString(
"Unsupported authentication for %1").arg(dlInfo->
m_url));
905 LOG(VB_GENERAL, LOG_ERR,
LOC + QString(
"Unsupported File output %1 for %2")
914 LOG(VB_GENERAL, LOG_ERR,
LOC + QString(
"Unsupported link-local operation %1")
915 .arg(dlInfo->
m_url));
919 QUrl url(dlInfo->
m_url);
920 QString host(url.host());
921 int port(url.port(80));
935 QByteArray* buffer = dlInfo->
m_data;
936 QHash<QByteArray, QByteArray>
headers;
939 if (!
headers.contains(
"User-Agent"))
940 headers.insert(
"User-Agent", QByteArray(
"MythDownloadManager v") +
941 MYTH_BINARY_VERSION);
942 headers.insert(
"Connection",
"close");
943 headers.insert(
"Accept-Encoding",
"identity");
944 if (!buffer->isEmpty())
945 headers.insert(
"Content-Length", QString::number(buffer->size()).toUtf8());
946 headers.insert(
"Host", (url.host() +
":" + QString::number(port)).toUtf8());
948 QByteArray requestMessage;
949 QString path (url.path());
950 requestMessage.append(
"POST ");
951 requestMessage.append(path.toLatin1());
952 requestMessage.append(
" HTTP/1.1\r\n");
953 QHashIterator<QByteArray, QByteArray> it(
headers);
957 requestMessage.append(it.key());
958 requestMessage.append(
": ");
959 requestMessage.append(it.value());
960 requestMessage.append(
"\r\n");
962 requestMessage.append(
"\r\n");
963 if (!buffer->isEmpty())
964 requestMessage.append(*buffer);
967 socket.connectToHost(host,
static_cast<uint16_t>(port));
969 if (!socket.waitForConnected(5000))
972 ok = socket.write(requestMessage) > 0;
975 ok = socket.waitForDisconnected(5000);
978 *buffer = socket.readAll();
980 QByteArray delim(
"\r\n\r\n");
981 int delimLoc = buffer->indexOf(delim);
983 *buffer = buffer->right(buffer->size() - delimLoc - 4);
1000 LOG(VB_GENERAL, LOG_ERR,
LOC + QString(
"Link Local request failed: %1").arg(url.toString()));
1021 for (
const auto& url : std::as_const(urls))
1024 while (lit.hasNext())
1028 if (dlInfo->
m_url == url)
1051 if (QThread::currentThread() == this->thread())
1074 while (lit.hasNext())
1082 LOG(VB_FILE, LOG_DEBUG,
1083 LOC + QString(
"Aborting download - user request"));
1092 dlInfo->
m_errorCode = QNetworkReply::OperationCanceledError;
1114 dlInfo->
m_data =
nullptr;
1118 QMap <QString, MythDownloadInfo*>::iterator mit =
m_downloadInfos.begin();
1126 dlInfo->
m_data =
nullptr;
1136 auto *reply = qobject_cast<QNetworkReply *>(sender());
1137 if (reply ==
nullptr)
1140 LOG(VB_FILE, LOG_DEBUG,
LOC + QString(
"downloadError %1 ")
1141 .arg(errorCode) + reply->errorString() );
1146 reply->deleteLater();
1164 const QUrl& oldRedirectUrl)
1166 LOG(VB_FILE, LOG_DEBUG,
LOC + QString(
"redirectUrl()"));
1169 if(!possibleRedirectUrl.isEmpty() && possibleRedirectUrl != oldRedirectUrl)
1180 LOG(VB_FILE, LOG_DEBUG,
LOC + QString(
"downloadFinished(%1)")
1181 .arg((
long long)reply));
1186 reply->deleteLater();
1192 if (!dlInfo || !dlInfo->
m_reply)
1206 int statusCode = -1;
1207 static const QString kDateFormat =
"ddd, dd MMM yyyy hh:mm:ss 'GMT'";
1208 QNetworkReply *reply = dlInfo->
m_reply;
1212 QUrl possibleRedirectUrl =
1213 reply->attribute(QNetworkRequest::RedirectionTargetAttribute).toUrl();
1215 if (!possibleRedirectUrl.isEmpty() &&
1216 possibleRedirectUrl.isValid() &&
1217 possibleRedirectUrl.isRelative())
1218 possibleRedirectUrl = reply->url().resolved(possibleRedirectUrl);
1220 if (!possibleRedirectUrl.isEmpty() && dlInfo->
m_finalUrl !=
nullptr)
1221 *dlInfo->
m_finalUrl = QString(possibleRedirectUrl.toString());
1227 reply->attribute(QNetworkRequest::HttpStatusCodeAttribute);
1228 if (status.isValid())
1229 statusCode = status.toInt();
1234 (statusCode == 301 || statusCode == 302 ||
1235 statusCode == 303)))
1237 LOG(VB_FILE, LOG_DEBUG,
LOC +
1238 QString(
"downloadFinished(%1): Redirect: %2 -> %3")
1239 .arg(QString::number((
long long)dlInfo),
1240 reply->url().toString(),
1241 dlInfo->m_redirectedTo.toString()));
1244 dlInfo->m_data->clear();
1246 dlInfo->m_bytesReceived = 0;
1247 dlInfo->m_bytesTotal = 0;
1249 QNetworkRequest request(dlInfo->m_redirectedTo);
1251 if (dlInfo->m_reload)
1253 request.setAttribute(QNetworkRequest::CacheLoadControlAttribute,
1254 QNetworkRequest::AlwaysNetwork);
1256 else if (dlInfo->m_preferCache)
1258 request.setAttribute(QNetworkRequest::CacheLoadControlAttribute,
1259 QNetworkRequest::PreferCache);
1262 request.setRawHeader(
"User-Agent",
1263 "MythDownloadManager v" +
1264 QByteArray(MYTH_BINARY_VERSION));
1266 switch (dlInfo->m_requestType)
1269 dlInfo->m_reply =
m_manager->head(request);
1273 dlInfo->m_reply =
m_manager->get(request);
1279 connect(dlInfo->m_reply, &QNetworkReply::errorOccurred,
1281 connect(dlInfo->m_reply, &QNetworkReply::downloadProgress,
1285 reply->deleteLater();
1289 LOG(VB_FILE, LOG_DEBUG, QString(
"downloadFinished(%1): COMPLETE: %2")
1290 .arg((
long long)dlInfo).arg(dlInfo->m_url));
1294 QUrl fileUrl = dlInfo->m_url;
1295 QString redirectLoc;
1297 while (!(redirectLoc =
getHeader(fileUrl,
"Location")).isNull())
1301 LOG(VB_GENERAL, LOG_WARNING, QString(
"Cache Redirection limit "
1303 .arg(fileUrl.toString()));
1306 fileUrl.setUrl(redirectLoc);
1311 QNetworkCacheMetaData urlData =
m_manager->cache()->metaData(fileUrl);
1313 if (
getHeader(urlData,
"Date").isNull())
1315 QNetworkCacheMetaData::RawHeaderList
headers = urlData.rawHeaders();
1316 QNetworkCacheMetaData::RawHeader newheader;
1318 newheader = QNetworkCacheMetaData::RawHeader(
"Date",
1319 now.toString(kDateFormat).toLatin1());
1321 urlData.setRawHeaders(
headers);
1323 m_manager->cache()->updateMetaData(urlData);
1328 dlInfo->m_redirectedTo.clear();
1334 if (reply && dlInfo->m_processReply)
1336 bool append = (!dlInfo->m_syncMode && dlInfo->m_caller);
1337 QByteArray data = reply->readAll();
1338 dataSize = data.size();
1341 dlInfo->m_bytesReceived += dataSize;
1343 dlInfo->m_bytesReceived = dataSize;
1345 dlInfo->m_bytesTotal = dlInfo->m_bytesReceived;
1350 dlInfo->m_data->append(data);
1352 *dlInfo->m_data = data;
1354 else if (!dlInfo->m_outFile.isEmpty())
1356 saveFile(dlInfo->m_outFile, data, append);
1363 (*dlInfo->m_data) = dlInfo->m_privData;
1365 else if (!dlInfo->m_outFile.isEmpty())
1367 saveFile(dlInfo->m_outFile, dlInfo->m_privData);
1369 dlInfo->m_bytesReceived += dataSize;
1370 dlInfo->m_bytesTotal = dlInfo->m_bytesReceived;
1378 LOG(VB_GENERAL, LOG_ERR,
LOC +
1379 QString(
"ERROR download finished but failed to remove url: %1")
1380 .arg(dlInfo->m_url));
1387 dlInfo->SetDone(
true);
1389 if (!dlInfo->m_syncMode)
1391 if (dlInfo->m_caller)
1393 LOG(VB_FILE, LOG_DEBUG, QString(
"downloadFinished(%1): "
1394 "COMPLETE: %2, sending event to caller")
1395 .arg((
long long)dlInfo).arg(dlInfo->m_url));
1398 args << dlInfo->m_url;
1399 args << dlInfo->m_outFile;
1400 args << QString::number(dlInfo->m_bytesTotal);
1402 args << (reply ? reply->errorString() : QString());
1403 args << QString::number((
int)(reply ? reply->error() :
1404 dlInfo->m_errorCode));
1406 QCoreApplication::postEvent(dlInfo->m_caller,
1425 auto *reply = qobject_cast<QNetworkReply *>(sender());
1426 if (reply ==
nullptr)
1429 LOG(VB_FILE, LOG_DEBUG,
LOC +
1430 QString(
"downloadProgress(%1, %2) (for reply %3)")
1431 .arg(bytesReceived).arg(bytesTotal).arg((
long long)reply));
1444 LOG(VB_FILE, LOG_DEBUG,
LOC +
1445 QString(
"downloadProgress: %1 to %2 is at %3 of %4 bytes downloaded")
1447 .arg(bytesReceived).arg(bytesTotal));
1451 LOG(VB_FILE, LOG_DEBUG, QString(
"downloadProgress(%1): "
1452 "sending event to caller")
1453 .arg(reply->url().toString()));
1456 QByteArray data = reply->readAll();
1461 dlInfo->
m_data->append(data);
1469 args << QString::number(bytesReceived);
1470 args << QString::number(bytesTotal);
1472 QCoreApplication::postEvent(dlInfo->
m_caller,
1485 const QByteArray &data,
1488 if (outFile.isEmpty() || data.isEmpty())
1491 QFile
file(outFile);
1492 QFileInfo fileInfo(outFile);
1493 QDir qdir(fileInfo.absolutePath());
1495 if (!qdir.exists() && !qdir.mkpath(fileInfo.absolutePath()))
1497 LOG(VB_GENERAL, LOG_ERR, QString(
"Failed to create: '%1'")
1498 .arg(fileInfo.absolutePath()));
1502 QIODevice::OpenMode mode = QIODevice::Unbuffered|QIODevice::WriteOnly;
1504 mode |= QIODevice::Append;
1506 if (!
file.open(mode))
1508 LOG(VB_GENERAL, LOG_ERR, QString(
"Failed to open: '%1'") .arg(outFile));
1513 size_t remaining = data.size();
1514 uint failure_cnt = 0;
1515 while ((remaining > 0) && (failure_cnt < 5))
1517 ssize_t written =
file.write(data.data() + offset, remaining);
1527 remaining -= written;
1530 return remaining <= 0;
1544 static const QString kDateFormat =
"ddd, dd MMM yyyy hh:mm:ss 'GMT'";
1545 LOG(VB_FILE, LOG_DEBUG,
LOC + QString(
"GetLastModified('%1')").arg(url));
1550 QUrl cacheUrl = QUrl(url);
1553 QString redirectLoc;
1555 while (!(redirectLoc =
getHeader(cacheUrl,
"Location")).isNull())
1559 LOG(VB_GENERAL, LOG_WARNING, QString(
"Cache Redirection limit "
1561 .arg(cacheUrl.toString()));
1564 cacheUrl.setUrl(redirectLoc);
1569 QNetworkCacheMetaData urlData =
m_manager->cache()->metaData(cacheUrl);
1572 if (urlData.isValid() &&
1573 ((!urlData.expirationDate().isValid()) ||
1574 (urlData.expirationDate().secsTo(now) < 0)))
1576 if (urlData.lastModified().toUTC().secsTo(now) <= 3600)
1578 result = urlData.lastModified().toUTC();
1582 QString date =
getHeader(urlData,
"Date");
1585 QDateTime loadDate =
1587 loadDate.setTimeSpec(Qt::UTC);
1588 if (loadDate.secsTo(now) <= 1200)
1590 result = urlData.lastModified().toUTC();
1596 if (!result.isValid())
1599 dlInfo->
m_url = url;
1610 QNetworkRequest::LastModifiedHeader);
1611 if (lastMod.isValid())
1612 result = lastMod.toDateTime().toUTC();
1621 LOG(VB_FILE, LOG_DEBUG,
LOC + QString(
"GetLastModified('%1'): Result %2")
1622 .arg(url, result.toString()));
1650 auto *jar = qobject_cast<MythCookieJar *>(
m_manager->cookieJar());
1672 auto *inJar = qobject_cast<MythCookieJar *>(
m_manager->cookieJar());
1673 if (inJar ==
nullptr)
1689 auto *inJar = qobject_cast<MythCookieJar *>(jar);
1690 if (inJar ==
nullptr)
1708 if (inJar !=
nullptr)
1725 QNetworkCacheMetaData metadata =
m_manager->cache()->metaData(url);
1737 const QString& header)
1739 auto headers = cacheData.rawHeaders();
1740 for (
const auto& rh : std::as_const(
headers))
1741 if (QString(rh.first) == header)
1752 const QList<QNetworkCookie> cookieList = old.allCookies();
1753 setAllCookies(cookieList);
1761 LOG(VB_GENERAL, LOG_DEBUG, QString(
"MythCookieJar: loading cookies from: %1").arg(
filename));
1764 if (!f.open(QIODevice::ReadOnly))
1766 LOG(VB_GENERAL, LOG_WARNING, QString(
"MythCookieJar::load() failed to open file for reading: %1").arg(
filename));
1770 QList<QNetworkCookie> cookieList;
1771 QTextStream stream(&f);
1772 while (!stream.atEnd())
1774 QString cookie = stream.readLine();
1775 cookieList << QNetworkCookie::parseCookies(cookie.toLocal8Bit());
1778 setAllCookies(cookieList);
1786 LOG(VB_GENERAL, LOG_DEBUG, QString(
"MythCookieJar: saving cookies to: %1").arg(
filename));
1789 if (!f.open(QIODevice::WriteOnly))
1791 LOG(VB_GENERAL, LOG_ERR, QString(
"MythCookieJar::save() failed to open file for writing: %1").arg(
filename));
1795 QList<QNetworkCookie> cookieList = allCookies();
1796 QTextStream stream(&f);
1798 for (
const auto& cookie : std::as_const(cookieList))
1799 stream << cookie.toRawForm() << Qt::endl;