/* * This file is part of nzbget * * Copyright (C) 2007-2013 Andrey Prygunkov * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. * * $Revision$ * $Date$ * */ #ifdef HAVE_CONFIG_H #include "config.h" #endif #ifdef WIN32 #include "win32.h" #endif #include #include #include #ifdef WIN32 #include #else #include #endif #include #include "nzbget.h" #include "PrePostProcessor.h" #include "Options.h" #include "Log.h" #include "QueueCoordinator.h" #include "ScriptController.h" #include "DiskState.h" #include "Util.h" #include "Scheduler.h" #include "Scanner.h" #include "Unpack.h" #include "NZBFile.h" extern QueueCoordinator* g_pQueueCoordinator; extern Options* g_pOptions; extern DiskState* g_pDiskState; extern Scheduler* g_pScheduler; extern Scanner* g_pScanner; PrePostProcessor::PrePostProcessor() { debug("Creating PrePostProcessor"); m_bHasMoreJobs = false; m_bPostPause = false; m_QueueCoordinatorObserver.m_pOwner = this; g_pQueueCoordinator->Attach(&m_QueueCoordinatorObserver); #ifndef DISABLE_PARCHECK m_ParCoordinator.m_pOwner = this; #endif } PrePostProcessor::~PrePostProcessor() { debug("Destroying PrePostProcessor"); } void PrePostProcessor::Cleanup() { debug("Cleaning up PrePostProcessor"); DownloadQueue* pDownloadQueue = g_pQueueCoordinator->LockQueue(); for (PostQueue::iterator it = pDownloadQueue->GetPostQueue()->begin(); it != pDownloadQueue->GetPostQueue()->end(); it++) { delete *it; } pDownloadQueue->GetPostQueue()->clear(); for (FileQueue::iterator it = pDownloadQueue->GetParkedFiles()->begin(); it != pDownloadQueue->GetParkedFiles()->end(); it++) { delete *it; } pDownloadQueue->GetParkedFiles()->clear(); for (HistoryList::iterator it = pDownloadQueue->GetHistoryList()->begin(); it != pDownloadQueue->GetHistoryList()->end(); it++) { delete *it; } pDownloadQueue->GetHistoryList()->clear(); g_pQueueCoordinator->UnlockQueue(); } void PrePostProcessor::Run() { debug("Entering PrePostProcessor-loop"); if (g_pOptions->GetServerMode() && g_pOptions->GetSaveQueue() && g_pOptions->GetReloadQueue() && g_pOptions->GetReloadPostQueue()) { DownloadQueue* pDownloadQueue = g_pQueueCoordinator->LockQueue(); SanitisePostQueue(pDownloadQueue->GetPostQueue()); g_pQueueCoordinator->UnlockQueue(); } g_pScheduler->FirstCheck(); ApplySchedulerState(); int iDiskSpaceInterval = 1000; int iSchedulerInterval = 1000; int iHistoryInterval = 600000; const int iStepMSec = 200; while (!IsStopped()) { // check incoming nzb directory g_pScanner->Check(); if (!(g_pOptions->GetPauseDownload() || g_pOptions->GetPauseDownload2()) && g_pOptions->GetDiskSpace() > 0 && !g_pQueueCoordinator->GetStandBy() && iDiskSpaceInterval >= 1000) { // check free disk space every 1 second CheckDiskSpace(); iDiskSpaceInterval = 0; } iDiskSpaceInterval += iStepMSec; // check post-queue every 200 msec CheckPostQueue(); if (iSchedulerInterval >= 1000) { // check scheduler tasks every 1 second g_pScheduler->IntervalCheck(); ApplySchedulerState(); iSchedulerInterval = 0; CheckScheduledResume(); } iSchedulerInterval += iStepMSec; if (iHistoryInterval >= 600000) { // check history (remove old entries) every 10 minutes CheckHistory(); iHistoryInterval = 0; } iHistoryInterval += iStepMSec; usleep(iStepMSec * 1000); } Cleanup(); debug("Exiting PrePostProcessor-loop"); } void PrePostProcessor::Stop() { Thread::Stop(); DownloadQueue* pDownloadQueue = g_pQueueCoordinator->LockQueue(); #ifndef DISABLE_PARCHECK m_ParCoordinator.Stop(); #endif if (!pDownloadQueue->GetPostQueue()->empty()) { PostInfo* pPostInfo = pDownloadQueue->GetPostQueue()->front(); if ((pPostInfo->GetStage() == PostInfo::ptUnpacking || pPostInfo->GetStage() == PostInfo::ptExecutingScript) && pPostInfo->GetPostThread()) { Thread* pPostThread = pPostInfo->GetPostThread(); pPostInfo->SetPostThread(NULL); pPostThread->SetAutoDestroy(true); pPostThread->Stop(); } } g_pQueueCoordinator->UnlockQueue(); } void PrePostProcessor::QueueCoordinatorUpdate(Subject * Caller, void * Aspect) { if (IsStopped()) { return; } QueueCoordinator::Aspect* pAspect = (QueueCoordinator::Aspect*)Aspect; if (pAspect->eAction == QueueCoordinator::eaNZBFileFound) { NZBFound(pAspect->pDownloadQueue, pAspect->pNZBInfo); } else if (pAspect->eAction == QueueCoordinator::eaNZBFileAdded) { NZBAdded(pAspect->pDownloadQueue, pAspect->pNZBInfo); } else if ((pAspect->eAction == QueueCoordinator::eaFileCompleted || pAspect->eAction == QueueCoordinator::eaFileDeleted)) { if ( #ifndef DISABLE_PARCHECK !m_ParCoordinator.AddPar(pAspect->pFileInfo, pAspect->eAction == QueueCoordinator::eaFileDeleted) && #endif IsNZBFileCompleted(pAspect->pDownloadQueue, pAspect->pNZBInfo, true, false) && (!pAspect->pFileInfo->GetPaused() || IsNZBFileCompleted(pAspect->pDownloadQueue, pAspect->pNZBInfo, false, false))) { if ((pAspect->eAction == QueueCoordinator::eaFileCompleted || (pAspect->pFileInfo->GetAutoDeleted() && IsNZBFileCompleted(pAspect->pDownloadQueue, pAspect->pNZBInfo, false, true))) && pAspect->pFileInfo->GetNZBInfo()->GetDeleteStatus() != NZBInfo::dsHealth) { info("Collection %s completely downloaded", pAspect->pNZBInfo->GetName()); NZBDownloaded(pAspect->pDownloadQueue, pAspect->pNZBInfo); } else if ((pAspect->eAction == QueueCoordinator::eaFileDeleted || (pAspect->eAction == QueueCoordinator::eaFileCompleted && pAspect->pFileInfo->GetNZBInfo()->GetDeleteStatus() > NZBInfo::dsNone)) && !pAspect->pNZBInfo->GetParCleanup() && !pAspect->pNZBInfo->GetPostProcess() && IsNZBFileCompleted(pAspect->pDownloadQueue, pAspect->pNZBInfo, false, true)) { info("Collection %s deleted from queue", pAspect->pNZBInfo->GetName()); NZBDeleted(pAspect->pDownloadQueue, pAspect->pNZBInfo); } } } } void PrePostProcessor::NZBFound(DownloadQueue* pDownloadQueue, NZBInfo* pNZBInfo) { if (g_pOptions->GetDupeCheck() && !pNZBInfo->GetNoDupeCheck()) { CheckDupeFound(pDownloadQueue, pNZBInfo); } } void PrePostProcessor::NZBAdded(DownloadQueue* pDownloadQueue, NZBInfo* pNZBInfo) { if (g_pOptions->GetParCheck() != Options::pcForce) { m_ParCoordinator.PausePars(pDownloadQueue, pNZBInfo); } if (g_pOptions->GetDupeCheck() && !pNZBInfo->GetNoDupeCheck()) { CheckDupeAdded(pDownloadQueue, pNZBInfo); } if (strlen(g_pOptions->GetNZBAddedProcess()) > 0) { NZBAddedScriptController::StartScript(pDownloadQueue, pNZBInfo, g_pOptions->GetNZBAddedProcess()); } } void PrePostProcessor::NZBDownloaded(DownloadQueue* pDownloadQueue, NZBInfo* pNZBInfo) { if (!pNZBInfo->GetPostProcess() && g_pOptions->GetDecode()) { info("Queueing %s for post-processing", pNZBInfo->GetName()); PostInfo* pPostInfo = new PostInfo(); pPostInfo->SetNZBInfo(pNZBInfo); pPostInfo->SetInfoName(pNZBInfo->GetName()); if (pNZBInfo->GetParStatus() == NZBInfo::psNone && g_pOptions->GetParCheck() != Options::pcForce) { pNZBInfo->SetParStatus(NZBInfo::psSkipped); } if (pNZBInfo->GetRenameStatus() == NZBInfo::rsNone && !g_pOptions->GetParRename()) { pNZBInfo->SetRenameStatus(NZBInfo::rsSkipped); } if (pNZBInfo->GetDeleteStatus() != NZBInfo::dsNone) { pNZBInfo->SetParStatus(NZBInfo::psFailure); pNZBInfo->SetUnpackStatus(NZBInfo::usFailure); pNZBInfo->SetRenameStatus(NZBInfo::rsFailure); pNZBInfo->SetMoveStatus(NZBInfo::msFailure); } pNZBInfo->SetPostProcess(true); pDownloadQueue->GetPostQueue()->push_back(pPostInfo); SaveQueue(pDownloadQueue); m_bHasMoreJobs = true; } else { NZBCompleted(pDownloadQueue, pNZBInfo, true); } } void PrePostProcessor::NZBDeleted(DownloadQueue* pDownloadQueue, NZBInfo* pNZBInfo) { if (pNZBInfo->GetDeleteStatus() == NZBInfo::dsNone) { pNZBInfo->SetDeleteStatus(NZBInfo::dsManual); } pNZBInfo->SetDeleting(false); if (g_pOptions->GetDeleteCleanupDisk() && pNZBInfo->GetCleanupDisk()) { // download was cancelled, deleting already downloaded files from disk for (NZBInfo::Files::reverse_iterator it = pNZBInfo->GetCompletedFiles()->rbegin(); it != pNZBInfo->GetCompletedFiles()->rend(); it++) { char* szFilename = *it; if (Util::FileExists(szFilename)) { detail("Deleting file %s", Util::BaseFileName(szFilename)); remove(szFilename); } } // delete .out.tmp-files and _brokenlog.txt DirBrowser dir(pNZBInfo->GetDestDir()); while (const char* szFilename = dir.Next()) { int iLen = strlen(szFilename); if ((iLen > 8 && !strcmp(szFilename + iLen - 8, ".out.tmp")) || !strcmp(szFilename, "_brokenlog.txt")) { char szFullFilename[1024]; snprintf(szFullFilename, 1024, "%s%c%s", pNZBInfo->GetDestDir(), PATH_SEPARATOR, szFilename); szFullFilename[1024-1] = '\0'; detail("Deleting file %s", szFilename); remove(szFullFilename); } } // delete old directory (if empty) if (Util::DirEmpty(pNZBInfo->GetDestDir())) { rmdir(pNZBInfo->GetDestDir()); } if (g_pOptions->GetNzbCleanupDisk()) { DeleteQueuedFile(pNZBInfo->GetQueuedFilename()); } } if (pNZBInfo->GetDeleteStatus() == NZBInfo::dsHealth) { NZBDownloaded(pDownloadQueue, pNZBInfo); } else { NZBCompleted(pDownloadQueue, pNZBInfo, true); } } void PrePostProcessor::NZBCompleted(DownloadQueue* pDownloadQueue, NZBInfo* pNZBInfo, bool bSaveQueue) { bool bNeedSave = false; if (g_pOptions->GetKeepHistory() > 0) { //remove old item for the same NZB for (HistoryList::iterator it = pDownloadQueue->GetHistoryList()->begin(); it != pDownloadQueue->GetHistoryList()->end(); it++) { HistoryInfo* pHistoryInfo = *it; if (pHistoryInfo->GetNZBInfo() == pNZBInfo) { delete pHistoryInfo; pDownloadQueue->GetHistoryList()->erase(it); break; } } HistoryInfo* pHistoryInfo = new HistoryInfo(pNZBInfo); pHistoryInfo->SetTime(time(NULL)); pDownloadQueue->GetHistoryList()->push_front(pHistoryInfo); // park files int iParkedFiles = 0; int index = 0; for (FileQueue::iterator it = pDownloadQueue->GetFileQueue()->begin(); it != pDownloadQueue->GetFileQueue()->end(); ) { FileInfo* pFileInfo = *it; if (pFileInfo->GetNZBInfo() == pNZBInfo && !pFileInfo->GetDeleted()) { detail("Park file %s", pFileInfo->GetFilename()); g_pQueueCoordinator->DiscardDiskFile(pFileInfo); pDownloadQueue->GetFileQueue()->erase(it); pDownloadQueue->GetParkedFiles()->push_back(pFileInfo); it = pDownloadQueue->GetFileQueue()->begin() + index; iParkedFiles++; } else { it++; index++; } } pNZBInfo->SetParkedFileCount(iParkedFiles); info("Collection %s added to history", pNZBInfo->GetName()); bNeedSave = true; } if (pNZBInfo->GetDupe() && pNZBInfo->GetDeleteStatus() != NZBInfo::dsManual) { DupeCompleted(pDownloadQueue, pNZBInfo); bNeedSave = true; } if (bSaveQueue && bNeedSave) { SaveQueue(pDownloadQueue); } } /** - if download of an item completes successfully and there are (paused) duplicates to this item in the queue, they all are deleted from queue; - if download of an item fails and there are (paused) duplicates to this item in the queue the first of them is unpaused; this repeats for every duplicate of the title until one of them can be downloaded or all fail; */ void PrePostProcessor::DupeCompleted(DownloadQueue* pDownloadQueue, NZBInfo* pNZBInfo) { debug("Processing duplicates for %s", pNZBInfo->GetName()); bool bFailure = pNZBInfo->GetParStatus() == NZBInfo::psFailure || pNZBInfo->GetUnpackStatus() == NZBInfo::usFailure || (pNZBInfo->GetParStatus() == NZBInfo::psSkipped && pNZBInfo->GetUnpackStatus() == NZBInfo::usSkipped && pNZBInfo->CalcHealth() < pNZBInfo->CalcCriticalHealth()); IDList groupIDList; std::set groupNZBs; FileInfo* pDupeFileInfo = NULL; for (FileQueue::iterator it = pDownloadQueue->GetFileQueue()->begin(); it != pDownloadQueue->GetFileQueue()->end(); it++) { FileInfo* pFileInfo = *it; if (pFileInfo->GetNZBInfo() != pNZBInfo && pFileInfo->GetNZBInfo()->GetDupe() && !strcmp(pFileInfo->GetNZBInfo()->GetDupeKey(), pNZBInfo->GetDupeKey())) { if (bFailure) { // find nzb with highest DupeScore if (!pDupeFileInfo || pFileInfo->GetNZBInfo()->GetDupeScore() > pDupeFileInfo->GetNZBInfo()->GetDupeScore()) { pDupeFileInfo = pFileInfo; } } else if (groupNZBs.find(pFileInfo->GetNZBInfo()) == groupNZBs.end()) { groupIDList.push_back(pFileInfo->GetID()); groupNZBs.insert(pFileInfo->GetNZBInfo()); pFileInfo->GetNZBInfo()->SetDeleteStatus(NZBInfo::dsDupe); } } } if (bFailure && pDupeFileInfo) { info("Unpausing duplicate %s", pDupeFileInfo->GetNZBInfo()->GetName()); g_pQueueCoordinator->GetQueueEditor()->LockedEditEntry(pDownloadQueue, pDupeFileInfo->GetID(), false, QueueEditor::eaGroupResume, 0, NULL); g_pQueueCoordinator->GetQueueEditor()->LockedEditEntry(pDownloadQueue, pDupeFileInfo->GetID(), false, QueueEditor::eaGroupPauseExtraPars, 0, NULL); } if (!bFailure && !groupIDList.empty()) { info("Removing duplicates for %s from queue", pNZBInfo->GetName()); g_pQueueCoordinator->GetQueueEditor()->LockedEditList(pDownloadQueue, &groupIDList, false, QueueEditor::eaGroupDelete, 0, NULL); } } /** Check if the title was already downloaded or is already queued. */ void PrePostProcessor::CheckDupeFound(DownloadQueue* pDownloadQueue, NZBInfo* pNZBInfo) { debug("Checking duplicates for %s", pNZBInfo->GetName()); bool bHasDupeKey = !Util::EmptyStr(pNZBInfo->GetDupeKey()); // find duplicates in download queue having exactly same content GroupQueue groupQueue; pDownloadQueue->BuildGroups(&groupQueue); for (GroupQueue::iterator it = groupQueue.begin(); it != groupQueue.end(); it++) { GroupInfo* pGroupInfo = *it; NZBInfo* pGroupNZBInfo = pGroupInfo->GetNZBInfo(); bool bSameContent = (pNZBInfo->GetFullContentHash() > 0 && pNZBInfo->GetFullContentHash() == pGroupNZBInfo->GetFullContentHash()) || (pNZBInfo->GetFilteredContentHash() > 0 && pNZBInfo->GetFilteredContentHash() == pGroupNZBInfo->GetFilteredContentHash()); if (pGroupNZBInfo != pNZBInfo && bSameContent) { if (!strcmp(pNZBInfo->GetName(), pGroupNZBInfo->GetName())) { warn("Skipping duplicate %s, already queued", pNZBInfo->GetName()); } else { warn("Skipping duplicate %s, already queued as %s", pNZBInfo->GetName(), pGroupNZBInfo->GetName()); } pNZBInfo->SetDeleteStatus(NZBInfo::dsDupe); // Flag saying QueueCoordinator to skip nzb-file return; } } // find duplicates in post queue having exactly same content for (PostQueue::iterator it = pDownloadQueue->GetPostQueue()->begin(); it != pDownloadQueue->GetPostQueue()->end(); it++) { PostInfo* pPostInfo = *it; bool bSameContent = (pNZBInfo->GetFullContentHash() > 0 && pNZBInfo->GetFullContentHash() == pPostInfo->GetNZBInfo()->GetFullContentHash()) || (pNZBInfo->GetFilteredContentHash() > 0 && pNZBInfo->GetFilteredContentHash() == pPostInfo->GetNZBInfo()->GetFilteredContentHash()); if (bSameContent) { if (!strcmp(pNZBInfo->GetName(), pPostInfo->GetNZBInfo()->GetName())) { warn("Skipping duplicate %s, already queued", pNZBInfo->GetName()); } else { warn("Skipping duplicate %s, already queued as %s", pNZBInfo->GetName(), pPostInfo->GetNZBInfo()->GetName()); } pNZBInfo->SetDeleteStatus(NZBInfo::dsDupe); // Flag saying QueueCoordinator to skip nzb-file return; } } // find duplicates in history for (HistoryList::iterator it = pDownloadQueue->GetHistoryList()->begin(); it != pDownloadQueue->GetHistoryList()->end(); it++) { HistoryInfo* pHistoryInfo = *it; bool bSkip = false; const char* szDupeName = NULL; bool bSameContent = false; if (pHistoryInfo->GetKind() == HistoryInfo::hkNZBInfo && (!strcmp(pHistoryInfo->GetNZBInfo()->GetName(), pNZBInfo->GetName()) || (bHasDupeKey && !strcmp(pHistoryInfo->GetNZBInfo()->GetDupeKey(), pNZBInfo->GetDupeKey())))) { bool bFailure = pHistoryInfo->GetNZBInfo()->GetDeleteStatus() != NZBInfo::dsNone || pHistoryInfo->GetNZBInfo()->GetParStatus() == NZBInfo::psFailure || pHistoryInfo->GetNZBInfo()->GetUnpackStatus() == NZBInfo::usFailure || (pHistoryInfo->GetNZBInfo()->GetParStatus() == NZBInfo::psSkipped && pHistoryInfo->GetNZBInfo()->GetUnpackStatus() == NZBInfo::usSkipped && pHistoryInfo->GetNZBInfo()->CalcHealth() < pHistoryInfo->GetNZBInfo()->CalcCriticalHealth()); bSkip = (!bFailure && pNZBInfo->GetDupeScore() <= pHistoryInfo->GetNZBInfo()->GetDupeScore()); szDupeName = pHistoryInfo->GetNZBInfo()->GetName(); } if (!bSkip && pHistoryInfo->GetKind() == HistoryInfo::hkDupInfo && (!strcmp(pHistoryInfo->GetDupInfo()->GetName(), pNZBInfo->GetName()) || (bHasDupeKey && !strcmp(pHistoryInfo->GetDupInfo()->GetDupeKey(), pNZBInfo->GetDupeKey())))) { bool bFailure = pHistoryInfo->GetDupInfo()->GetStatus() != DupInfo::dsSuccess; bSkip = (!bFailure && pNZBInfo->GetDupeScore() <= pHistoryInfo->GetDupInfo()->GetDupeScore()); szDupeName = pHistoryInfo->GetDupInfo()->GetName(); } if (!bSkip && pHistoryInfo->GetKind() == HistoryInfo::hkNZBInfo && ((pNZBInfo->GetFullContentHash() > 0 && pNZBInfo->GetFullContentHash() == pHistoryInfo->GetNZBInfo()->GetFullContentHash()) || (pNZBInfo->GetFilteredContentHash() > 0 && pNZBInfo->GetFilteredContentHash() == pHistoryInfo->GetNZBInfo()->GetFilteredContentHash()))) { bSkip = true; bSameContent = true; szDupeName = pHistoryInfo->GetNZBInfo()->GetName(); } if (!bSkip && pHistoryInfo->GetKind() == HistoryInfo::hkDupInfo && ((pNZBInfo->GetFullContentHash() > 0 && pNZBInfo->GetFullContentHash() == pHistoryInfo->GetDupInfo()->GetFullContentHash()) || (pNZBInfo->GetFilteredContentHash() > 0 && pNZBInfo->GetFilteredContentHash() == pHistoryInfo->GetDupInfo()->GetFilteredContentHash()))) { bSkip = true; bSameContent = true; szDupeName = pHistoryInfo->GetDupInfo()->GetName(); } if (bSkip) { if (!strcmp(pNZBInfo->GetName(), szDupeName)) { warn("Skipping duplicate %s, found in history with %s", pNZBInfo->GetName(), bSameContent ? "exactly same content" : "success status"); } else { warn("Skipping duplicate %s, found in history %s with %s", pNZBInfo->GetName(), szDupeName, bSameContent ? "exactly same content" : "success status"); } pNZBInfo->SetDeleteStatus(NZBInfo::dsDupe); // Flag saying QueueCoordinator to skip nzb-file return; } } } /** - If download queue or post-queue contain a duplicate the existing item and the newly added item are marked as duplicates to each other. The newly added item is paused. */ void PrePostProcessor::CheckDupeAdded(DownloadQueue* pDownloadQueue, NZBInfo* pNZBInfo) { debug("Checking duplicates for %s", pNZBInfo->GetName()); bool bHasDupeKey = !Util::EmptyStr(pNZBInfo->GetDupeKey()); bool bHigherScore = true; NZBInfo* pDupeNZBInfo = NULL; // find all duplicates in post queue std::set postDupes; for (PostQueue::iterator it = pDownloadQueue->GetPostQueue()->begin(); it != pDownloadQueue->GetPostQueue()->end(); it++) { PostInfo* pPostInfo = *it; if (!strcmp(pPostInfo->GetNZBInfo()->GetName(), pNZBInfo->GetName()) || (bHasDupeKey && !strcmp(pPostInfo->GetNZBInfo()->GetDupeKey(), pNZBInfo->GetDupeKey()))) { postDupes.insert(pPostInfo->GetNZBInfo()); if (!pDupeNZBInfo) { pDupeNZBInfo = pPostInfo->GetNZBInfo(); } bHigherScore = bHigherScore && pPostInfo->GetNZBInfo()->GetDupeScore() < pNZBInfo->GetDupeScore(); } } // find all duplicates in download queue GroupQueue groupQueue; pDownloadQueue->BuildGroups(&groupQueue); std::list queueDupes; GroupInfo* pNewGroupInfo = NULL; for (GroupQueue::iterator it = groupQueue.begin(); it != groupQueue.end(); it++) { GroupInfo* pGroupInfo = *it; NZBInfo* pGroupNZBInfo = pGroupInfo->GetNZBInfo(); if (pGroupNZBInfo != pNZBInfo && (!strcmp(pGroupNZBInfo->GetName(), pNZBInfo->GetName()) || (bHasDupeKey && !strcmp(pGroupNZBInfo->GetDupeKey(), pNZBInfo->GetDupeKey())))) { queueDupes.push_back(pGroupInfo); if (!pDupeNZBInfo) { pDupeNZBInfo = pGroupNZBInfo; } bHigherScore = bHigherScore && pGroupNZBInfo->GetDupeScore() < pNZBInfo->GetDupeScore(); } if (pGroupNZBInfo == pNZBInfo) { pNewGroupInfo = pGroupInfo; } } if (pDupeNZBInfo) { info("Marking collection %s as duplicate to %s", pNZBInfo->GetName(), pDupeNZBInfo->GetName()); pNZBInfo->SetDupe(true); pDupeNZBInfo->SetDupe(true); if (!bHasDupeKey) { if (!Util::EmptyStr(pDupeNZBInfo->GetDupeKey())) { pNZBInfo->SetDupeKey(pDupeNZBInfo->GetDupeKey()); } else { // taking ID of the first NZB as DupeKey char szDupeKey[20]; snprintf(szDupeKey, 20, "nzb=%i", pDupeNZBInfo->GetID()); szDupeKey[20-1] = '\0'; pNZBInfo->SetDupeKey(szDupeKey); pDupeNZBInfo->SetDupeKey(szDupeKey); } } // pause all duplicates with lower DupeScore, which are not in post-processing for (std::list::iterator it = queueDupes.begin(); it != queueDupes.end(); it++) { GroupInfo* pGroupInfo = *it; NZBInfo* pDupeNZB = pGroupInfo->GetNZBInfo(); if (pDupeNZB->GetDupeScore() < pNZBInfo->GetDupeScore() && postDupes.find(pDupeNZB) == postDupes.end() && pGroupInfo->GetPausedFileCount() < pGroupInfo->GetRemainingFileCount()) { info("Pausing collection %s with lower duplicate score", pDupeNZB->GetName()); g_pQueueCoordinator->GetQueueEditor()->LockedEditEntry(pDownloadQueue, pGroupInfo->GetLastID(), false, QueueEditor::eaGroupPause, 0, NULL); } } if (!bHigherScore) { g_pQueueCoordinator->GetQueueEditor()->LockedEditEntry(pDownloadQueue, pNewGroupInfo->GetLastID(), false, QueueEditor::eaGroupPause, 0, NULL); } } } /** * Removes old entries from (recent) history */ void PrePostProcessor::CheckHistory() { DownloadQueue* pDownloadQueue = g_pQueueCoordinator->LockQueue(); time_t tMinTime = time(NULL) - g_pOptions->GetKeepHistory() * 60000; bool bChanged = false; int index = 0; // traversing in a reverse order to delete items in order they were added to history // (just to produce the log-messages in a more logical order) for (HistoryList::reverse_iterator it = pDownloadQueue->GetHistoryList()->rbegin(); it != pDownloadQueue->GetHistoryList()->rend(); ) { HistoryInfo* pHistoryInfo = *it; if (pHistoryInfo->GetKind() != HistoryInfo::hkDupInfo && pHistoryInfo->GetTime() < tMinTime) { char szNiceName[1024]; pHistoryInfo->GetName(szNiceName, 1024); if (g_pOptions->GetDupeCheck() && pHistoryInfo->GetKind() == HistoryInfo::hkNZBInfo) { // replace history element DupInfo* pDupInfo = new DupInfo(); pDupInfo->SetName(pHistoryInfo->GetNZBInfo()->GetName()); pDupInfo->SetDupeKey(pHistoryInfo->GetNZBInfo()->GetDupeKey()); pDupInfo->SetDupeScore(pHistoryInfo->GetNZBInfo()->GetDupeScore()); pDupInfo->SetSize(pHistoryInfo->GetNZBInfo()->GetSize()); pDupInfo->SetFullContentHash(pHistoryInfo->GetNZBInfo()->GetFullContentHash()); pDupInfo->SetFilteredContentHash(pHistoryInfo->GetNZBInfo()->GetFilteredContentHash()); bool bFailure = pHistoryInfo->GetNZBInfo()->GetDeleteStatus() != NZBInfo::dsNone || pHistoryInfo->GetNZBInfo()->GetParStatus() == NZBInfo::psFailure || pHistoryInfo->GetNZBInfo()->GetUnpackStatus() == NZBInfo::usFailure || (pHistoryInfo->GetNZBInfo()->GetParStatus() == NZBInfo::psSkipped && pHistoryInfo->GetNZBInfo()->GetUnpackStatus() == NZBInfo::usSkipped && pHistoryInfo->GetNZBInfo()->CalcHealth() < pHistoryInfo->GetNZBInfo()->CalcCriticalHealth()); pDupInfo->SetStatus( pHistoryInfo->GetNZBInfo()->GetDeleteStatus() == NZBInfo::dsManual || pHistoryInfo->GetNZBInfo()->GetDeleteStatus() == NZBInfo::dsDupe ? DupInfo::dsDeleted : bFailure ? DupInfo::dsFailed : DupInfo::dsSuccess); HistoryInfo* pNewHistoryInfo = new HistoryInfo(pDupInfo); pNewHistoryInfo->SetTime(pHistoryInfo->GetTime()); (*pDownloadQueue->GetHistoryList())[pDownloadQueue->GetHistoryList()->size() - 1 - index] = pNewHistoryInfo; info("Collection %s removed from recent history", szNiceName); index++; } else { pDownloadQueue->GetHistoryList()->erase(pDownloadQueue->GetHistoryList()->end() - 1 - index); info("Collection %s removed from history", szNiceName); } delete pHistoryInfo; it = pDownloadQueue->GetHistoryList()->rbegin() + index; bChanged = true; } else { it++; index++; } } if (bChanged) { SaveQueue(pDownloadQueue); } g_pQueueCoordinator->UnlockQueue(); } void PrePostProcessor::DeleteQueuedFile(const char* szQueuedFile) { // szQueuedFile may contain one filename or several filenames separated // with "|"-character (for merged groups) char* szFilename = strdup(szQueuedFile); char* szEnd = szFilename - 1; while (szEnd) { char* szName1 = szEnd + 1; szEnd = strchr(szName1, '|'); if (szEnd) *szEnd = '\0'; if (Util::FileExists(szName1)) { info("Deleting file %s", szName1); remove(szName1); } } free(szFilename); } /** * Find ID of any file in the nzb-file */ int PrePostProcessor::FindGroupID(DownloadQueue* pDownloadQueue, NZBInfo* pNZBInfo) { for (FileQueue::iterator it = pDownloadQueue->GetFileQueue()->begin(); it != pDownloadQueue->GetFileQueue()->end(); it++) { FileInfo* pFileInfo = *it; if (pFileInfo->GetNZBInfo() == pNZBInfo) { return pFileInfo->GetID(); break; } } return 0; } void PrePostProcessor::CheckDiskSpace() { long long lFreeSpace = Util::FreeDiskSize(g_pOptions->GetDestDir()); if (lFreeSpace > -1 && lFreeSpace / 1024 / 1024 < g_pOptions->GetDiskSpace()) { warn("Low disk space. Pausing download"); g_pOptions->SetPauseDownload(true); } } void PrePostProcessor::CheckPostQueue() { DownloadQueue* pDownloadQueue = g_pQueueCoordinator->LockQueue(); if (!pDownloadQueue->GetPostQueue()->empty()) { PostInfo* pPostInfo = pDownloadQueue->GetPostQueue()->front(); if (!pPostInfo->GetWorking()) { #ifndef DISABLE_PARCHECK if (pPostInfo->GetRequestParCheck() && pPostInfo->GetNZBInfo()->GetParStatus() <= NZBInfo::psSkipped && g_pOptions->GetParCheck() != Options::pcManual) { pPostInfo->GetNZBInfo()->SetParStatus(NZBInfo::psNone); pPostInfo->SetRequestParCheck(false); pPostInfo->SetStage(PostInfo::ptQueued); pPostInfo->GetNZBInfo()->GetScriptStatuses()->Clear(); DeletePostThread(pPostInfo); } else if (pPostInfo->GetRequestParCheck() && pPostInfo->GetNZBInfo()->GetParStatus() <= NZBInfo::psSkipped && g_pOptions->GetParCheck() == Options::pcManual) { pPostInfo->SetRequestParCheck(false); pPostInfo->GetNZBInfo()->SetParStatus(NZBInfo::psManual); DeletePostThread(pPostInfo); FileInfo* pFileInfo = GetQueueGroup(pDownloadQueue, pPostInfo->GetNZBInfo()); if (pFileInfo) { info("Downloading all remaining files for manual par-check for %s", pPostInfo->GetNZBInfo()->GetName()); g_pQueueCoordinator->GetQueueEditor()->LockedEditEntry(pDownloadQueue, pFileInfo->GetID(), false, QueueEditor::eaGroupResume, 0, NULL); pPostInfo->SetStage(PostInfo::ptFinished); pPostInfo->GetNZBInfo()->SetPostProcess(false); } else { info("There are no par-files remain for download for %s", pPostInfo->GetNZBInfo()->GetName()); pPostInfo->SetStage(PostInfo::ptQueued); } } else if (pPostInfo->GetRequestParRename()) { pPostInfo->GetNZBInfo()->SetRenameStatus(NZBInfo::rsNone); pPostInfo->SetRequestParRename(false); pPostInfo->SetStage(PostInfo::ptQueued); DeletePostThread(pPostInfo); } #endif if (pPostInfo->GetDeleted()) { pPostInfo->SetStage(PostInfo::ptFinished); } if (pPostInfo->GetStage() == PostInfo::ptQueued && !g_pOptions->GetPausePostProcess()) { DeletePostThread(pPostInfo); StartJob(pDownloadQueue, pPostInfo); } else if (pPostInfo->GetStage() == PostInfo::ptFinished) { UpdatePauseState(false, NULL); JobCompleted(pDownloadQueue, pPostInfo); } else if (!g_pOptions->GetPausePostProcess()) { error("Internal error: invalid state in post-processor"); } } } g_pQueueCoordinator->UnlockQueue(); } void PrePostProcessor::SaveQueue(DownloadQueue* pDownloadQueue) { if (g_pOptions->GetSaveQueue() && g_pOptions->GetServerMode()) { g_pDiskState->SaveDownloadQueue(pDownloadQueue); } } /** * Reset the state of items after reloading from disk and * delete items which could not be resumed. */ void PrePostProcessor::SanitisePostQueue(PostQueue* pPostQueue) { for (PostQueue::iterator it = pPostQueue->begin(); it != pPostQueue->end(); it++) { PostInfo* pPostInfo = *it; if (pPostInfo->GetStage() == PostInfo::ptExecutingScript || !Util::DirectoryExists(pPostInfo->GetNZBInfo()->GetDestDir())) { pPostInfo->SetStage(PostInfo::ptFinished); } else { pPostInfo->SetStage(PostInfo::ptQueued); } } } void PrePostProcessor::DeletePostThread(PostInfo* pPostInfo) { if (pPostInfo->GetPostThread()) { delete pPostInfo->GetPostThread(); pPostInfo->SetPostThread(NULL); } } void PrePostProcessor::StartJob(DownloadQueue* pDownloadQueue, PostInfo* pPostInfo) { #ifndef DISABLE_PARCHECK if (pPostInfo->GetNZBInfo()->GetRenameStatus() == NZBInfo::rsNone) { UpdatePauseState(g_pOptions->GetParPauseQueue(), "par-rename"); m_ParCoordinator.StartParRenameJob(pPostInfo); return; } else if (pPostInfo->GetNZBInfo()->GetParStatus() == NZBInfo::psNone) { if (m_ParCoordinator.FindMainPars(pPostInfo->GetNZBInfo()->GetDestDir(), NULL)) { UpdatePauseState(g_pOptions->GetParPauseQueue(), "par-check"); m_ParCoordinator.StartParCheckJob(pPostInfo); } else { info("Nothing to par-check for %s", pPostInfo->GetInfoName()); pPostInfo->GetNZBInfo()->SetParStatus(NZBInfo::psSkipped); pPostInfo->SetWorking(false); pPostInfo->SetStage(PostInfo::ptQueued); } return; } else if (pPostInfo->GetNZBInfo()->GetParStatus() == NZBInfo::psSkipped && pPostInfo->GetNZBInfo()->CalcHealth() < pPostInfo->GetNZBInfo()->CalcCriticalHealth() && m_ParCoordinator.FindMainPars(pPostInfo->GetNZBInfo()->GetDestDir(), NULL)) { warn("Skipping par-check for %s due to health %.1f%% below critical %.1f%%", pPostInfo->GetInfoName(), pPostInfo->GetNZBInfo()->CalcHealth() / 10.0, pPostInfo->GetNZBInfo()->CalcCriticalHealth() / 10.0); pPostInfo->GetNZBInfo()->SetParStatus(NZBInfo::psFailure); return; } else if (pPostInfo->GetNZBInfo()->GetParStatus() == NZBInfo::psSkipped && pPostInfo->GetNZBInfo()->GetFailedSize() - pPostInfo->GetNZBInfo()->GetParFailedSize() > 0 && m_ParCoordinator.FindMainPars(pPostInfo->GetNZBInfo()->GetDestDir(), NULL)) { info("Collection %s with health %.1f%% needs par-check", pPostInfo->GetInfoName(), pPostInfo->GetNZBInfo()->CalcHealth() / 10.0); pPostInfo->SetRequestParCheck(true); return; } #endif NZBParameter* pUnpackParameter = pPostInfo->GetNZBInfo()->GetParameters()->Find("*Unpack:", false); bool bUnpackParam = !(pUnpackParameter && !strcasecmp(pUnpackParameter->GetValue(), "no")); bool bUnpack = bUnpackParam && (pPostInfo->GetNZBInfo()->GetUnpackStatus() == NZBInfo::usNone); bool bParFailed = pPostInfo->GetNZBInfo()->GetParStatus() == NZBInfo::psFailure || pPostInfo->GetNZBInfo()->GetParStatus() == NZBInfo::psRepairPossible || pPostInfo->GetNZBInfo()->GetParStatus() == NZBInfo::psManual; bool bCleanup = !bUnpack && pPostInfo->GetNZBInfo()->GetCleanupStatus() == NZBInfo::csNone && (pPostInfo->GetNZBInfo()->GetParStatus() == NZBInfo::psSuccess || (pPostInfo->GetNZBInfo()->GetUnpackStatus() == NZBInfo::usSuccess && pPostInfo->GetNZBInfo()->GetParStatus() != NZBInfo::psFailure)) && strlen(g_pOptions->GetExtCleanupDisk()) > 0; bool bMoveInter = !bUnpack && pPostInfo->GetNZBInfo()->GetMoveStatus() == NZBInfo::msNone && pPostInfo->GetNZBInfo()->GetUnpackStatus() != NZBInfo::usFailure && pPostInfo->GetNZBInfo()->GetParStatus() != NZBInfo::psFailure && pPostInfo->GetNZBInfo()->GetParStatus() != NZBInfo::psManual && strlen(g_pOptions->GetInterDir()) > 0 && !strncmp(pPostInfo->GetNZBInfo()->GetDestDir(), g_pOptions->GetInterDir(), strlen(g_pOptions->GetInterDir())); // TODO: check if download has pp-scripts defined bool bPostScript = true; if (bUnpack && bParFailed) { warn("Skipping unpack for %s due to %s", pPostInfo->GetInfoName(), pPostInfo->GetNZBInfo()->GetParStatus() == NZBInfo::psManual ? "required par-repair" : "par-failure"); pPostInfo->GetNZBInfo()->SetUnpackStatus(NZBInfo::usSkipped); bUnpack = false; } if (!bUnpack && !bMoveInter && !bPostScript) { pPostInfo->SetStage(PostInfo::ptFinished); return; } pPostInfo->SetProgressLabel(bUnpack ? "Unpacking" : bMoveInter ? "Moving" : "Executing post-process-script"); pPostInfo->SetWorking(true); pPostInfo->SetStage(bUnpack ? PostInfo::ptUnpacking : bMoveInter ? PostInfo::ptMoving : PostInfo::ptExecutingScript); pPostInfo->SetFileProgress(0); pPostInfo->SetStageProgress(0); SaveQueue(pDownloadQueue); if (!pPostInfo->GetStartTime()) { pPostInfo->SetStartTime(time(NULL)); } pPostInfo->SetStageTime(time(NULL)); if (bUnpack) { UpdatePauseState(g_pOptions->GetUnpackPauseQueue(), "unpack"); UnpackController::StartJob(pPostInfo); } else if (bCleanup) { UpdatePauseState(g_pOptions->GetUnpackPauseQueue() || g_pOptions->GetScriptPauseQueue(), "cleanup"); CleanupController::StartJob(pPostInfo); } else if (bMoveInter) { UpdatePauseState(g_pOptions->GetUnpackPauseQueue() || g_pOptions->GetScriptPauseQueue(), "move"); MoveController::StartJob(pPostInfo); } else { UpdatePauseState(g_pOptions->GetScriptPauseQueue(), "post-process-script"); PostScriptController::StartJob(pPostInfo); } } void PrePostProcessor::JobCompleted(DownloadQueue* pDownloadQueue, PostInfo* pPostInfo) { pPostInfo->SetWorking(false); pPostInfo->SetProgressLabel(""); pPostInfo->SetStage(PostInfo::ptFinished); DeletePostThread(pPostInfo); if (IsNZBFileCompleted(pDownloadQueue, pPostInfo->GetNZBInfo(), true, false)) { // Cleaning up queue if par-check was successful or unpack was successful or // script was successful (if unpack was not performed) bool bCanCleanupQueue = pPostInfo->GetNZBInfo()->GetParStatus() == NZBInfo::psSuccess || pPostInfo->GetNZBInfo()->GetParStatus() == NZBInfo::psRepairPossible || pPostInfo->GetNZBInfo()->GetUnpackStatus() == NZBInfo::usSuccess || (pPostInfo->GetNZBInfo()->GetUnpackStatus() == NZBInfo::usNone && pPostInfo->GetNZBInfo()->GetScriptStatuses()->CalcTotalStatus() == ScriptStatus::srSuccess); if ((g_pOptions->GetParCleanupQueue() || g_pOptions->GetNzbCleanupDisk()) && bCanCleanupQueue) { if (g_pOptions->GetParCleanupQueue()) { FileInfo* pFileInfo = GetQueueGroup(pDownloadQueue, pPostInfo->GetNZBInfo()); if (pFileInfo) { info("Cleaning up download queue for %s", pPostInfo->GetNZBInfo()->GetName()); pFileInfo->GetNZBInfo()->ClearCompletedFiles(); pFileInfo->GetNZBInfo()->SetParCleanup(true); g_pQueueCoordinator->GetQueueEditor()->LockedEditEntry(pDownloadQueue, pFileInfo->GetID(), false, QueueEditor::eaGroupDelete, 0, NULL); } } if (g_pOptions->GetNzbCleanupDisk()) { DeleteQueuedFile(pPostInfo->GetNZBInfo()->GetQueuedFilename()); } } NZBCompleted(pDownloadQueue, pPostInfo->GetNZBInfo(), false); } for (PostQueue::iterator it = pDownloadQueue->GetPostQueue()->begin(); it != pDownloadQueue->GetPostQueue()->end(); it++) { if (pPostInfo == *it) { pDownloadQueue->GetPostQueue()->erase(it); break; } } delete pPostInfo; SaveQueue(pDownloadQueue); m_bHasMoreJobs = !pDownloadQueue->GetPostQueue()->empty(); } bool PrePostProcessor::IsNZBFileCompleted(DownloadQueue* pDownloadQueue, NZBInfo* pNZBInfo, bool bIgnorePausedPars, bool bAllowOnlyOneDeleted) { bool bNZBFileCompleted = true; int iDeleted = 0; for (FileQueue::iterator it = pDownloadQueue->GetFileQueue()->begin(); it != pDownloadQueue->GetFileQueue()->end(); it++) { FileInfo* pFileInfo = *it; if (pFileInfo->GetNZBInfo() == pNZBInfo) { if (pFileInfo->GetDeleted()) { iDeleted++; } if (((!pFileInfo->GetPaused() || !bIgnorePausedPars || !(m_ParCoordinator.ParseParFilename(pFileInfo->GetFilename(), NULL, NULL))) && !pFileInfo->GetDeleted()) || (bAllowOnlyOneDeleted && iDeleted > 1)) { bNZBFileCompleted = false; break; } } } return bNZBFileCompleted; } /** * Returns the first FileInfo belonging to given NZBInfo. */ FileInfo* PrePostProcessor::GetQueueGroup(DownloadQueue* pDownloadQueue, NZBInfo* pNZBInfo) { for (FileQueue::iterator it = pDownloadQueue->GetFileQueue()->begin(); it != pDownloadQueue->GetFileQueue()->end(); it++) { FileInfo* pFileInfo = *it; if (pFileInfo->GetNZBInfo() == pNZBInfo) { return pFileInfo; } } return NULL; } void PrePostProcessor::ApplySchedulerState() { if (g_pScheduler->GetDownloadRateChanged()) { info("Scheduler: set download rate to %i KB/s", g_pScheduler->GetDownloadRate() / 1024); g_pOptions->SetDownloadRate(g_pScheduler->GetDownloadRate()); } if (g_pScheduler->GetPauseDownloadChanged()) { info("Scheduler: %s download queue", g_pScheduler->GetPauseDownload() ? "pause" : "unpause"); m_bSchedulerPauseChanged = true; m_bSchedulerPause = g_pScheduler->GetPauseDownload(); if (!m_bPostPause) { g_pOptions->SetPauseDownload(m_bSchedulerPause); } } if (g_pScheduler->GetPauseScanChanged()) { info("Scheduler: %s scan", g_pScheduler->GetPauseScan() ? "pause" : "unpause"); g_pOptions->SetPauseScan(g_pScheduler->GetPauseScan()); } } void PrePostProcessor::UpdatePauseState(bool bNeedPause, const char* szReason) { if (bNeedPause) { if (PauseDownload()) { info("Pausing queue before %s", szReason); } } else if (m_bPostPause) { if (UnpauseDownload()) { info("Unpausing queue after %s", m_szPauseReason); } } m_szPauseReason = szReason; } bool PrePostProcessor::PauseDownload() { debug("PrePostProcessor::PauseDownload()"); if (m_bPostPause && g_pOptions->GetPauseDownload()) { return false; } m_bPostPause = !g_pOptions->GetPauseDownload(); m_bSchedulerPauseChanged = false; g_pOptions->SetPauseDownload(true); return m_bPostPause; } bool PrePostProcessor::UnpauseDownload() { debug("PrePostProcessor::UnpauseDownload()"); bool bPause = true; if (m_bPostPause) { m_bPostPause = false; bPause = m_bSchedulerPauseChanged && m_bSchedulerPause; g_pOptions->SetPauseDownload(bPause); } return !bPause; } void PrePostProcessor::CheckScheduledResume() { time_t tResumeTime = g_pOptions->GetResumeTime(); time_t tCurrentTime = time(NULL); if (tResumeTime > 0 && tCurrentTime >= tResumeTime) { info("Autoresume"); g_pOptions->SetResumeTime(0); g_pOptions->SetPauseDownload2(false); g_pOptions->SetPausePostProcess(false); g_pOptions->SetPauseScan(false); } } bool PrePostProcessor::QueueEditList(IDList* pIDList, EEditAction eAction, int iOffset, const char* szText) { debug("Edit-command for post-processor received"); switch (eAction) { case eaPostMoveOffset: case eaPostMoveTop: case eaPostMoveBottom: return PostQueueMove(pIDList, eAction, iOffset); case eaPostDelete: return PostQueueDelete(pIDList); case eaHistoryDelete: case eaHistoryReturn: case eaHistoryProcess: case eaHistorySetParameter: return HistoryEdit(pIDList, eAction, iOffset, szText); default: return false; } } bool PrePostProcessor::PostQueueDelete(IDList* pIDList) { bool bOK = false; DownloadQueue* pDownloadQueue = g_pQueueCoordinator->LockQueue(); for (IDList::iterator itID = pIDList->begin(); itID != pIDList->end(); itID++) { int iID = *itID; for (PostQueue::iterator itPost = pDownloadQueue->GetPostQueue()->begin(); itPost != pDownloadQueue->GetPostQueue()->end(); itPost++) { PostInfo* pPostInfo = *itPost; if (pPostInfo->GetID() == iID) { if (pPostInfo->GetWorking()) { info("Deleting active post-job %s", pPostInfo->GetInfoName()); pPostInfo->SetDeleted(true); #ifndef DISABLE_PARCHECK if (PostInfo::ptLoadingPars <= pPostInfo->GetStage() && pPostInfo->GetStage() <= PostInfo::ptRenaming) { if (m_ParCoordinator.Cancel()) { bOK = true; } } else #endif if (pPostInfo->GetPostThread()) { debug("Terminating %s for %s", (pPostInfo->GetStage() == PostInfo::ptUnpacking ? "unpack" : "post-process-script"), pPostInfo->GetInfoName()); pPostInfo->GetPostThread()->Stop(); bOK = true; } else { error("Internal error in PrePostProcessor::QueueDelete"); } } else { info("Deleting queued post-job %s", pPostInfo->GetInfoName()); JobCompleted(pDownloadQueue, pPostInfo); bOK = true; } break; } } } g_pQueueCoordinator->UnlockQueue(); return bOK; } bool PrePostProcessor::PostQueueMove(IDList* pIDList, EEditAction eAction, int iOffset) { if (pIDList->size() != 1) { //NOTE: Only one post-job can be moved at once return false; } DownloadQueue* pDownloadQueue = g_pQueueCoordinator->LockQueue(); bool bOK = false; int iID = pIDList->front(); unsigned int iIndex = 0; PostInfo* pPostInfo = NULL; for (PostQueue::iterator it = pDownloadQueue->GetPostQueue()->begin(); it != pDownloadQueue->GetPostQueue()->end(); it++) { PostInfo* pPostInfo1 = *it; if (pPostInfo1->GetID() == iID) { pPostInfo = pPostInfo1; break; } iIndex++; } if (pPostInfo) { // NOTE: only items which are not currently being processed can be moved unsigned int iNewIndex = 0; switch (eAction) { case eaPostMoveTop: iNewIndex = 1; break; case eaPostMoveBottom: iNewIndex = pDownloadQueue->GetPostQueue()->size() - 1; break; case eaPostMoveOffset: iNewIndex = iIndex + iOffset; break; default: ; // suppress compiler warning } if (iNewIndex < 1) { iNewIndex = 1; } else if (iNewIndex > pDownloadQueue->GetPostQueue()->size() - 1) { iNewIndex = pDownloadQueue->GetPostQueue()->size() - 1; } if (0 < iNewIndex && iNewIndex < pDownloadQueue->GetPostQueue()->size() && iNewIndex != iIndex) { pDownloadQueue->GetPostQueue()->erase(pDownloadQueue->GetPostQueue()->begin() + iIndex); pDownloadQueue->GetPostQueue()->insert(pDownloadQueue->GetPostQueue()->begin() + iNewIndex, pPostInfo); SaveQueue(pDownloadQueue); bOK = true; } } g_pQueueCoordinator->UnlockQueue(); return bOK; } bool PrePostProcessor::HistoryEdit(IDList* pIDList, EEditAction eAction, int iOffset, const char* szText) { bool bOK = false; DownloadQueue* pDownloadQueue = g_pQueueCoordinator->LockQueue(); for (IDList::iterator itID = pIDList->begin(); itID != pIDList->end(); itID++) { int iID = *itID; for (HistoryList::iterator itHistory = pDownloadQueue->GetHistoryList()->begin(); itHistory != pDownloadQueue->GetHistoryList()->end(); itHistory++) { HistoryInfo* pHistoryInfo = *itHistory; if (pHistoryInfo->GetID() == iID) { switch (eAction) { case eaHistoryDelete: HistoryDelete(pDownloadQueue, itHistory, pHistoryInfo); break; case eaHistoryReturn: case eaHistoryProcess: HistoryReturn(pDownloadQueue, itHistory, pHistoryInfo, eAction == eaHistoryProcess); break; case eaHistorySetParameter: HistorySetParameter(pHistoryInfo, szText); break; default: // nothing, just to avoid compiler warning break; } bOK = true; break; } } } if (bOK) { SaveQueue(pDownloadQueue); } g_pQueueCoordinator->UnlockQueue(); return bOK; } void PrePostProcessor::HistoryDelete(DownloadQueue* pDownloadQueue, HistoryList::iterator itHistory, HistoryInfo* pHistoryInfo) { char szNiceName[1024]; pHistoryInfo->GetName(szNiceName, 1024); info("Deleting %s from history", szNiceName); if (pHistoryInfo->GetKind() == HistoryInfo::hkNZBInfo) { NZBInfo* pNZBInfo = pHistoryInfo->GetNZBInfo(); // delete parked files int index = 0; for (FileQueue::iterator it = pDownloadQueue->GetParkedFiles()->begin(); it != pDownloadQueue->GetParkedFiles()->end(); ) { FileInfo* pFileInfo = *it; if (pFileInfo->GetNZBInfo() == pNZBInfo) { pDownloadQueue->GetParkedFiles()->erase(it); if (g_pOptions->GetSaveQueue() && g_pOptions->GetServerMode()) { g_pDiskState->DiscardFile(pFileInfo); } delete pFileInfo; it = pDownloadQueue->GetParkedFiles()->begin() + index; } else { it++; index++; } } } pDownloadQueue->GetHistoryList()->erase(itHistory); delete pHistoryInfo; } void PrePostProcessor::HistoryReturn(DownloadQueue* pDownloadQueue, HistoryList::iterator itHistory, HistoryInfo* pHistoryInfo, bool bReprocess) { char szNiceName[1024]; pHistoryInfo->GetName(szNiceName, 1024); debug("Returning %s from history back to download queue", szNiceName); bool bUnparked = false; if (bReprocess && pHistoryInfo->GetKind() != HistoryInfo::hkNZBInfo) { error("Could not restart postprocessing for %s: history item has wrong type", szNiceName); return; } if (pHistoryInfo->GetKind() == HistoryInfo::hkNZBInfo) { NZBInfo* pNZBInfo = pHistoryInfo->GetNZBInfo(); // unpark files int index = 0; for (FileQueue::reverse_iterator it = pDownloadQueue->GetParkedFiles()->rbegin(); it != pDownloadQueue->GetParkedFiles()->rend(); ) { FileInfo* pFileInfo = *it; if (pFileInfo->GetNZBInfo() == pNZBInfo) { detail("Unpark file %s", pFileInfo->GetFilename()); pDownloadQueue->GetParkedFiles()->erase(pDownloadQueue->GetParkedFiles()->end() - 1 - index); pDownloadQueue->GetFileQueue()->push_front(pFileInfo); bUnparked = true; it = pDownloadQueue->GetParkedFiles()->rbegin() + index; } else { it++; index++; } } // reset postprocessing status variables pNZBInfo->SetPostProcess(false); pNZBInfo->SetParCleanup(false); if (!pNZBInfo->GetUnpackCleanedUpDisk()) { pNZBInfo->SetUnpackStatus(NZBInfo::usNone); pNZBInfo->SetCleanupStatus(NZBInfo::csNone); if (m_ParCoordinator.FindMainPars(pNZBInfo->GetDestDir(), NULL)) { pNZBInfo->SetParStatus(NZBInfo::psNone); pNZBInfo->SetRenameStatus(NZBInfo::rsNone); } } pNZBInfo->GetScriptStatuses()->Clear(); pNZBInfo->SetParkedFileCount(0); } if (pHistoryInfo->GetKind() == HistoryInfo::hkUrlInfo) { UrlInfo* pUrlInfo = pHistoryInfo->GetUrlInfo(); pHistoryInfo->DiscardUrlInfo(); pUrlInfo->SetStatus(UrlInfo::aiUndefined); pDownloadQueue->GetUrlQueue()->push_back(pUrlInfo); bUnparked = true; } if (bUnparked || bReprocess) { pDownloadQueue->GetHistoryList()->erase(itHistory); // the object "pHistoryInfo" is released few lines later, after the call to "NZBDownloaded" info("%s returned from history back to download queue", szNiceName); } else { warn("Could not return %s back from history to download queue: history item does not have any files left for download", szNiceName); } if (bReprocess) { // start postprocessing debug("Restarting postprocessing for %s", szNiceName); NZBDownloaded(pDownloadQueue, pHistoryInfo->GetNZBInfo()); } if (bUnparked || bReprocess) { delete pHistoryInfo; } } void PrePostProcessor::HistorySetParameter(HistoryInfo* pHistoryInfo, const char* szText) { char szNiceName[1024]; pHistoryInfo->GetName(szNiceName, 1024); debug("Setting post-process-parameter '%s' for '%s'", szText, szNiceName); if (pHistoryInfo->GetKind() != HistoryInfo::hkNZBInfo) { error("Could not set post-process-parameter for %s: history item has wrong type", szNiceName); return; } char* szStr = strdup(szText); char* szValue = strchr(szStr, '='); if (szValue) { *szValue = '\0'; szValue++; pHistoryInfo->GetNZBInfo()->GetParameters()->SetParameter(szStr, szValue); } else { error("Could not set post-process-parameter for %s: invalid argument: %s", pHistoryInfo->GetNZBInfo()->GetName(), szText); } free(szStr); }