mythcorecontext.cpp

Go to the documentation of this file.

// Qt
#include <QtGlobal>
#include <QCoreApplication>
#include <QUrl>
#include <QDir>
#include <QFileInfo>
#include <QDebug>
#include <QMutex>
#include <QRunnable>
#include <QWaitCondition>
#include <QAbstractSocket>
#include <QHostAddress>
#include <QHostInfo>
#include <QNetworkInterface>
#include <QNetworkAddressEntry>
#include <QLocale>
#include <QPair>
#include <QDateTime>
 
// Std
#include <algorithm>
#include <cmath>
#include <cstdarg>
#include <queue>
#include <unistd.h>       // for usleep()
 
#ifdef _WIN32
#include <winsock2.h>
#else
#include <clocale>
#include <utility>
#endif
 
// MythTV
#include "compat.h"
#include "mythdownloadmanager.h"
#include "mythcorecontext.h"
#include "mythsocket.h"
#include "mythsystemlegacy.h"
#include "mthreadpool.h"
#include "exitcodes.h"
#include "mythlogging.h"
#include "mythversion.h"
#include "logging.h"
#include "mthread.h"
#include "serverpool.h"
#include "mythdate.h"
#include "mythplugin.h"
#include "mythmiscutil.h"
#include "mythpower.h"
 
#define LOC      QString("MythCoreContext::%1(): ").arg(__func__)
 
MythCoreContext *gCoreContext = nullptr;
 
class MythCoreContextPrivate : public QObject
{
  public:
    MythCoreContextPrivate(MythCoreContext *lparent, QString binversion,
                           QObject *guicontext);
   ~MythCoreContextPrivate() override;
 
    bool WaitForWOL(std::chrono::milliseconds timeout = std::chrono::milliseconds::max());
 
  public:
    MythCoreContext *m_parent     { nullptr };
    QObject         *m_guiContext { nullptr };
    QObject         *m_guiObject  { nullptr };
    QString          m_appBinaryVersion;
 
    QMutex  m_localHostLock;        
    QString m_localHostname;        
    QMutex  m_masterHostLock;       
    QString m_masterHostname;       
    QMutex  m_scopesLock;           
    QMap<QString, QString> m_scopes;
 
    QMutex      m_sockLock;         
    MythSocket *m_serverSock { nullptr }; 
    MythSocket *m_eventSock  { nullptr }; 
 
    QMutex         m_wolInProgressLock;
    QWaitCondition m_wolInProgressWaitCondition;
    bool           m_wolInProgress { false };
    bool           m_isWOLAllowed  { true };
 
    bool m_backend  { false };
    bool m_frontend { false };
 
    MythDB *m_database { nullptr };
 
    QThread *m_uiThread { nullptr };
 
    MythLocale *m_locale { nullptr };
    QString m_language;
 
    MythScheduler *m_scheduler { nullptr };
 
    bool m_blockingClient { true };
 
    QMap<QObject *, MythCoreContext::PlaybackStartCb> m_playbackClients;
    QMutex m_playbackLock;
    bool m_inwanting   { false };
    bool m_intvwanting { false };
 
    bool m_announcedProtocol { false };
 
    MythPluginManager *m_pluginmanager { nullptr };
 
    bool m_isexiting { false };
 
    QMap<QString, QPair<int64_t, uint64_t> >  m_fileswritten;
    QMutex m_fileslock;
 
    MythSessionManager *m_sessionManager { nullptr };
 
    QList<QHostAddress> m_approvedIps;
    QList<QHostAddress> m_deniedIps;
 
    MythPower *m_power { nullptr };
};
 
MythCoreContextPrivate::MythCoreContextPrivate(MythCoreContext *lparent,
                                               QString binversion,
                                               QObject *guicontext)
    : m_parent(lparent),
      m_guiContext(guicontext),
      m_appBinaryVersion(std::move(binversion)),
      m_database(GetMythDB()),
      m_uiThread(QThread::currentThread())
{
    MThread::ThreadSetup("CoreContext");
}
 
static void delete_sock(
#if QT_VERSION < QT_VERSION_CHECK(6,0,0)
    QMutexLocker &locker,
#else
    QMutexLocker<QMutex> &locker,
#endif
    MythSocket **s)
{
    if (*s)
    {
        MythSocket *tmp = *s;
        *s = nullptr;
        locker.unlock();
        tmp->DecrRef();
        locker.relock();
    }
}
 
MythCoreContextPrivate::~MythCoreContextPrivate()
{
    if (m_power)
        MythPower::AcquireRelease(this, false);
 
    MThreadPool::StopAllPools();
 
    {
        QMutexLocker locker(&m_sockLock);
        delete_sock(locker, &m_serverSock);
        delete_sock(locker, &m_eventSock);
    }
 
    delete m_locale;
 
    delete m_sessionManager;
 
    MThreadPool::ShutdownAllPools();
 
    ShutdownMythSystemLegacy();
 
    ShutdownMythDownloadManager();
 
    // This has already been run in the MythContext dtor.  Do we need it here
    // too?
#if 0
    logStop(); // need to shutdown db logger before we kill db
#endif
 
    MThread::Cleanup();
 
    GetMythDB()->GetDBManager()->CloseDatabases();
 
    if (m_database) {
        DestroyMythDB();
        m_database = nullptr;
    }
 
    loggingDeregisterThread();
}
 
bool MythCoreContextPrivate::WaitForWOL(std::chrono::milliseconds timeout)
{
    std::chrono::milliseconds timeout_remaining = timeout;
    while (m_wolInProgress && (timeout_remaining > 0ms))
    {
        LOG(VB_GENERAL, LOG_INFO, LOC + "Wake-On-LAN in progress, waiting...");
 
        std::chrono::milliseconds max_wait = std::min(1000ms, timeout_remaining);
        m_wolInProgressWaitCondition.wait(&m_wolInProgressLock, max_wait.count());
        timeout_remaining -= max_wait;
    }
 
    return !m_wolInProgress;
}
 
MythCoreContext::MythCoreContext(const QString &binversion,
                                 QObject *guiContext)
{
    d = new MythCoreContextPrivate(this, binversion, guiContext);
}
 
bool MythCoreContext::Init(void)
{
    if (!d)
    {
        LOG(VB_GENERAL, LOG_EMERG, LOC + "Out-of-memory");
        return false;
    }
 
    if (d->m_appBinaryVersion != MYTH_BINARY_VERSION)
    {
        LOG(VB_GENERAL, LOG_CRIT,
            QString("Application binary version (%1) does not "
                    "match libraries (%2)")
                .arg(d->m_appBinaryVersion, MYTH_BINARY_VERSION));
 
        QString warning = tr("This application is not compatible with the "
                             "installed MythTV libraries. Please recompile "
                             "after a make distclean");
        LOG(VB_GENERAL, LOG_WARNING, warning);
 
        return false;
    }
 
#ifndef _WIN32
    QString lang_variables("");
    QString lc_value = setlocale(LC_CTYPE, nullptr);
    if (lc_value.isEmpty())
    {
        // try fallback to environment variables for non-glibc systems
        // LC_ALL, then LC_CTYPE
        lc_value = qEnvironmentVariable("LC_ALL");
        if (lc_value.isEmpty())
            lc_value = qEnvironmentVariable("LC_CTYPE");
    }
    if (!lc_value.contains("UTF-8", Qt::CaseInsensitive))
        lang_variables.append("LC_ALL or LC_CTYPE");
    lc_value = qEnvironmentVariable("LANG");
    if (!lc_value.contains("UTF-8", Qt::CaseInsensitive))
    {
        if (!lang_variables.isEmpty())
            lang_variables.append(", and ");
        lang_variables.append("LANG");
    }
    LOG(VB_GENERAL, LOG_INFO, QString("Assumed character encoding: %1")
                                     .arg(lc_value));
    if (!lang_variables.isEmpty())
    {
        LOG(VB_GENERAL, LOG_WARNING, QString("This application expects to "
            "be running a locale that specifies a UTF-8 codeset, and many "
            "features may behave improperly with your current language "
            "settings. Please set the %1 variable(s) in the environment "
            "in which this program is executed to include a UTF-8 codeset "
            "(such as 'en_US.UTF-8').").arg(lang_variables));
    }
#endif
 
    return true;
}
 
MythCoreContext::~MythCoreContext()
{
    delete d;
    d = nullptr;
}
 
void MythCoreContext::setTestIntSettings(QMap<QString,int> &overrides)
{
    m_testOverrideInts = std::move(overrides);
}
 
void MythCoreContext::setTestFloatSettings(QMap<QString,double> &overrides)
{
    m_testOverrideFloats = std::move(overrides);
}
 
void MythCoreContext::setTestStringSettings(QMap<QString,QString> &overrides)
{
    m_testOverrideStrings = std::move(overrides);
}
 
bool MythCoreContext::SetupCommandSocket(MythSocket *serverSock,
                                         const QString &announcement,
                                         std::chrono::milliseconds timeout,
                                         bool &proto_mismatch)
{
    proto_mismatch = false;
 
#ifndef IGNORE_PROTO_VER_MISMATCH
    if (!CheckProtoVersion(serverSock, timeout, true))
    {
        proto_mismatch = true;
        return false;
    }
#else
    Q_UNUSED(timeout);
#endif
 
    QStringList strlist(announcement);
 
    if (!serverSock->WriteStringList(strlist))
    {
        LOG(VB_GENERAL, LOG_ERR, LOC + "Connecting server socket to "
                                       "master backend, socket write failed");
        return false;
    }
 
    if (!serverSock->ReadStringList(strlist, MythSocket::kShortTimeout) ||
        strlist.empty() || (strlist[0] == "ERROR"))
    {
        if (!strlist.empty())
        {
            LOG(VB_GENERAL, LOG_ERR, LOC + "Problem connecting "
                                           "server socket to master backend");
        }
        else
        {
            LOG(VB_GENERAL, LOG_ERR, LOC + "Timeout connecting "
                                           "server socket to master backend");
        }
        return false;
    }
 
    return true;
}
 
// Connects to master server safely (i.e. by taking m_sockLock)
bool MythCoreContext::SafeConnectToMasterServer(bool blockingClient,
                                                bool openEventSocket)
{
    QMutexLocker locker(&d->m_sockLock);
    bool success = ConnectToMasterServer(blockingClient, openEventSocket);
 
    return success;
}
 
// Assumes that either m_sockLock is held, or the app is still single
// threaded (i.e. during startup).
bool MythCoreContext::ConnectToMasterServer(bool blockingClient,
                                            bool openEventSocket)
{
    if (IsMasterBackend())
    {
        // Should never get here unless there is a bug in the code somewhere.
        // If this happens, it can cause endless event loops.
        LOG(VB_GENERAL, LOG_ERR, LOC + "ERROR: Master backend tried to connect back "
                "to itself!");
        return false;
    }
    if (IsExiting())
        return false;
 
    QString server = GetMasterServerIP();
    if (server.isEmpty())
        return false;
 
    int     port   = GetMasterServerPort();
    bool    proto_mismatch = false;
 
    if (d->m_serverSock && !d->m_serverSock->IsConnected())
    {
        d->m_serverSock->DecrRef();
        d->m_serverSock = nullptr;
    }
 
    if (!d->m_serverSock)
    {
        QString type = IsFrontend() ? "Frontend" : (blockingClient ? "Playback" : "Monitor");
        QString ann = QString("ANN %1 %2 %3")
            .arg(type, d->m_localHostname, QString::number(static_cast<int>(false)));
        d->m_serverSock = ConnectCommandSocket(
            server, port, ann, &proto_mismatch);
    }
 
    if (!d->m_serverSock)
        return false;
 
    d->m_blockingClient = blockingClient;
 
    if (!openEventSocket)
        return true;
 
 
    if (!IsBackend())
    {
        if (d->m_eventSock && !d->m_eventSock->IsConnected())
        {
            d->m_eventSock->DecrRef();
            d->m_eventSock = nullptr;
        }
        if (!d->m_eventSock)
            d->m_eventSock = ConnectEventSocket(server, port);
 
        if (!d->m_eventSock)
        {
            d->m_serverSock->DecrRef();
            d->m_serverSock = nullptr;
 
            QCoreApplication::postEvent(
                d->m_guiContext, new MythEvent("CONNECTION_FAILURE"));
 
            return false;
        }
    }
 
    return true;
}
 
MythSocket *MythCoreContext::ConnectCommandSocket(
    const QString &hostname, int port, const QString &announce,
    bool *p_proto_mismatch, int maxConnTry, std::chrono::milliseconds setup_timeout)
{
    MythSocket *serverSock = nullptr;
 
    {
        QMutexLocker locker(&d->m_wolInProgressLock);
        d->WaitForWOL();
    }
 
    QString WOLcmd;
    if (IsWOLAllowed())
        WOLcmd = GetSetting("WOLbackendCommand", "");
 
    if (maxConnTry < 1)
        maxConnTry = std::max(GetNumSetting("BackendConnectRetry", 1), 1);
 
    std::chrono::seconds WOLsleepTime = 0s;
    int WOLmaxConnTry = 0;
    if (!WOLcmd.isEmpty())
    {
        WOLsleepTime  = GetDurSetting<std::chrono::seconds>("WOLbackendReconnectWaitTime", 0s);
        WOLmaxConnTry = std::max(GetNumSetting("WOLbackendConnectRetry", 1), 1);
        maxConnTry    = std::max(maxConnTry, WOLmaxConnTry);
    }
 
    bool we_attempted_wol = false;
 
    if (setup_timeout <= 0ms)
        setup_timeout = MythSocket::kShortTimeout;
 
    bool proto_mismatch = false;
    for (int cnt = 1; cnt <= maxConnTry; cnt++)
    {
        LOG(VB_GENERAL, LOG_INFO, LOC +
            QString("Connecting to backend server: %1:%2 (try %3 of %4)")
                .arg(hostname).arg(port).arg(cnt).arg(maxConnTry));
 
        serverSock = new MythSocket();
 
        std::chrono::microseconds sleepus = 0us;
        if (serverSock->ConnectToHost(hostname, port))
        {
            if (SetupCommandSocket(
                    serverSock, announce, setup_timeout, proto_mismatch))
            {
                break;
            }
 
            if (proto_mismatch)
            {
                if (p_proto_mismatch)
                    *p_proto_mismatch = true;
 
                serverSock->DecrRef();
                serverSock = nullptr;
                break;
            }
 
            setup_timeout += setup_timeout / 2;
        }
        else if (!WOLcmd.isEmpty() && (cnt < maxConnTry))
        {
            if (!we_attempted_wol)
            {
                QMutexLocker locker(&d->m_wolInProgressLock);
                if (d->m_wolInProgress)
                {
                    d->WaitForWOL();
                    continue;
                }
 
                d->m_wolInProgress = we_attempted_wol = true;
            }
 
            MythWakeup(WOLcmd, kMSDontDisableDrawing | kMSDontBlockInputDevs |
                                kMSProcessEvents);
            sleepus = WOLsleepTime;
        }
 
        serverSock->DecrRef();
        serverSock = nullptr;
 
        if (cnt == 1)
        {
            QCoreApplication::postEvent(
                d->m_guiContext, new MythEvent("CONNECTION_FAILURE"));
        }
 
        if (sleepus != 0us)
            usleep(sleepus.count());
    }
 
    if (we_attempted_wol)
    {
        QMutexLocker locker(&d->m_wolInProgressLock);
        d->m_wolInProgress = false;
        d->m_wolInProgressWaitCondition.wakeAll();
    }
 
    if (!serverSock && !proto_mismatch)
    {
        LOG(VB_GENERAL, LOG_ERR,
                "Connection to master server timed out.\n\t\t\t"
                "Either the server is down or the master server settings"
                "\n\t\t\t"
                "in mythtv-settings does not contain the proper IP address\n");
    }
    else
    {
        QCoreApplication::postEvent(
            d->m_guiContext, new MythEvent("CONNECTION_RESTABLISHED"));
    }
 
    return serverSock;
}
 
MythSocket *MythCoreContext::ConnectEventSocket(const QString &hostname,
                                                int port)
{
    auto *eventSock = new MythSocket(-1, this);
 
    // Assume that since we _just_ connected the command socket,
    // this one won't need multiple retries to work...
    if (!eventSock->ConnectToHost(hostname, port))
    {
        LOG(VB_GENERAL, LOG_ERR, LOC + "Failed to connect event "
                                       "socket to master backend");
        eventSock->DecrRef();
        return nullptr;
    }
 
    QString str = QString("ANN Monitor %1 %2")
        .arg(d->m_localHostname).arg(static_cast<int>(true));
    QStringList strlist(str);
    eventSock->WriteStringList(strlist);
    bool ok = true;
    if (!eventSock->ReadStringList(strlist) || strlist.empty() ||
        (strlist[0] == "ERROR"))
    {
        if (!strlist.empty())
        {
            LOG(VB_GENERAL, LOG_ERR, LOC +
                "Problem connecting event socket to master backend");
        }
        else
        {
            LOG(VB_GENERAL, LOG_ERR, LOC +
                "Timeout connecting event socket to master backend");
        }
        ok = false;
    }
 
    if (!ok)
    {
        eventSock->DecrRef();
        eventSock = nullptr;
    }
 
    return eventSock;
}
 
bool MythCoreContext::IsConnectedToMaster(void)
{
    QMutexLocker locker(&d->m_sockLock);
    return d->m_serverSock;
}
 
void MythCoreContext::BlockShutdown(void)
{
    QStringList strlist;
 
    QMutexLocker locker(&d->m_sockLock);
    if (d->m_serverSock == nullptr)
        return;
 
    strlist << "BLOCK_SHUTDOWN";
    d->m_serverSock->SendReceiveStringList(strlist);
 
    d->m_blockingClient = true;
}
 
void MythCoreContext::AllowShutdown(void)
{
    QStringList strlist;
 
    QMutexLocker locker(&d->m_sockLock);
    if (d->m_serverSock == nullptr)
        return;
 
    strlist << "ALLOW_SHUTDOWN";
    d->m_serverSock->SendReceiveStringList(strlist);
 
    d->m_blockingClient = false;
}
 
bool MythCoreContext::IsBlockingClient(void) const
{
    return d->m_blockingClient;
}
 
void MythCoreContext::SetWOLAllowed(bool allow)
{
    d->m_isWOLAllowed = allow;
}
 
bool MythCoreContext::IsWOLAllowed() const
{
    return d->m_isWOLAllowed;
}
 
void MythCoreContext::SetAsBackend(bool backend)
{
    d->m_backend = backend;
}
 
bool MythCoreContext::IsBackend(void) const
{
    return d->m_backend;
}
 
void MythCoreContext::SetAsFrontend(bool frontend)
{
    d->m_frontend = frontend;
}
 
bool MythCoreContext::IsFrontend(void) const
{
    return d->m_frontend;
}
 
bool MythCoreContext::IsMasterHost(void)
{
    QString host = GetHostName();
    return IsMasterHost(host);
}
 
bool MythCoreContext::IsMasterHost(const QString &host)
{
    // Temporary code here only to facilitate the upgrade
    // from 1346 or earlier. The way of determining master host is
    // changing, and the new way of determning master host
    // will not work with earlier databases.
    // This code can be removed when updating from prior to
    // 1347 is no longer allowed.
    // Note that we are deprecating some settings including
    // MasterServerIP, and can remove them at a future time.
    if (GetNumSetting("DBSchemaVer") < 1347)
    {
        // Temporary copy of code from old version of
        // IsThisHost(Qstring&,QString&)
        QString addr(resolveSettingAddress("MasterServerIP"));
        if (addr.toLower() == host.toLower())
            return true;
        QHostAddress addrfix(addr);
        addrfix.setScopeId(QString());
        QString addrstr = addrfix.toString();
        if (addrfix.isNull())
            addrstr = resolveAddress(addr);
        QString thisip  = GetBackendServerIP4(host);
        QString thisip6 = GetBackendServerIP6(host);
        return !addrstr.isEmpty()
                && ((addrstr == thisip) || (addrstr == thisip6));
    }
    return GetSetting("MasterServerName") ==  host;
}
 
bool MythCoreContext::IsMasterBackend(void)
{
    return (IsBackend() && IsMasterHost());
}
 
bool MythCoreContext::BackendIsRunning(void)
{
#if defined(Q_OS_DARWIN) || defined(__FreeBSD__) || defined(__OpenBSD__)
    const char *command = "ps -axc | grep -i mythbackend | grep -v grep > /dev/null";
#elif defined _WIN32
    const char *command = "%systemroot%\\system32\\tasklist.exe "
       " | %systemroot%\\system32\\find.exe /i \"mythbackend.exe\" ";
#else
    const char *command = "ps ch -C mythbackend -o pid > /dev/null";
#endif
    uint res = myth_system(command, kMSDontBlockInputDevs |
                                    kMSDontDisableDrawing |
                                    kMSProcessEvents);
    return (res == GENERIC_EXIT_OK);
}
 
bool MythCoreContext::IsThisBackend(const QString &addr)
{
    return IsBackend() && IsThisHost(addr);
}
 
bool MythCoreContext::IsThisHost(const QString &addr)
{
    return IsThisHost(addr, GetHostName());
}
 
bool MythCoreContext::IsThisHost(const QString &addr, const QString &host)
{
    if (addr.toLower() == host.toLower())
        return true;
 
    QHostAddress addrfix(addr);
    addrfix.setScopeId(QString());
    QString addrstr = addrfix.toString();
 
    if (addrfix.isNull())
    {
        addrstr = resolveAddress(addr);
    }
 
    QString thisip  = GetBackendServerIP(host);
 
    return !addrstr.isEmpty() && ((addrstr == thisip));
}
 
bool MythCoreContext::IsFrontendOnly(void)
{
    // find out if a backend runs on this host...
    bool backendOnLocalhost = false;
 
    QStringList strlist("QUERY_IS_ACTIVE_BACKEND");
    strlist << GetHostName();
 
    SendReceiveStringList(strlist);
 
    backendOnLocalhost = strlist[0] != "FALSE";
 
    return !backendOnLocalhost;
}
 
QString MythCoreContext::GenMythURL(const QString& host, int port, QString path, const QString& storageGroup)
{
    QUrl ret;
 
    QString m_host;
 
    QHostAddress addr(host);
    if (!addr.isNull())
    {
        LOG(VB_GENERAL, LOG_CRIT, LOC + QString("(%1/%2): Given "
                                          "IP address instead of hostname "
                                          "(ID). This is invalid.").arg(host, path));
    }
 
    m_host = host;
 
    // Basically if it appears to be an IPv6 IP surround the IP with [] otherwise don't bother
    if (!addr.isNull() && addr.protocol() == QAbstractSocket::IPv6Protocol)
        m_host = "[" + addr.toString().toLower() + "]";
 
    ret.setScheme("myth");
    if (!storageGroup.isEmpty())
        ret.setUserName(storageGroup);
    ret.setHost(m_host);
    if (port > 0 && port != 6543)
        ret.setPort(port);
  if (!path.startsWith("/"))
          path = QString("/") + path;
    ret.setPath(path);
 
#if 0
    LOG(VB_GENERAL, LOG_DEBUG, LOC +
        QString("GenMythURL returning %1").arg(ret.toString()));
#endif
 
    return ret.toString();
}
 
QString MythCoreContext::GetMasterHostPrefix(const QString &storageGroup,
                                             const QString &path)
{
    return GenMythURL(GetMasterHostName(),
                      GetMasterServerPort(),
                      path,
                      storageGroup);
}
 
QString MythCoreContext::GetMasterHostName(void)
{
    QMutexLocker locker(&d->m_masterHostLock);
 
    if (d->m_masterHostname.isEmpty())
    {
 
        if (IsMasterBackend())
            d->m_masterHostname = d->m_localHostname;
        else
        {
            QStringList strlist("QUERY_HOSTNAME");
 
            if (SendReceiveStringList(strlist))
                d->m_masterHostname = strlist[0];
        }
    }
 
    return d->m_masterHostname;
}
 
void MythCoreContext::ClearSettingsCache(const QString &myKey)
{
    d->m_database->ClearSettingsCache(myKey);
}
 
void MythCoreContext::ActivateSettingsCache(bool activate)
{
    d->m_database->ActivateSettingsCache(activate);
}
 
QString MythCoreContext::GetHostName(void)
{
    QMutexLocker locker(&d->m_localHostLock);
    return d->m_localHostname;
}
 
QString MythCoreContext::GetFilePrefix(void)
{
    return GetSetting("RecordFilePrefix");
}
 
void MythCoreContext::GetResolutionSetting(const QString &type,
                                           int &width, int &height,
                                           double &forced_aspect,
                                           double &refresh_rate,
                                           int index)
{
    d->m_database->GetResolutionSetting(type, width, height, forced_aspect,
                                        refresh_rate, index);
}
 
void MythCoreContext::GetResolutionSetting(const QString &t, int &w,
                                           int &h, int i)
{
    d->m_database->GetResolutionSetting(t, w, h, i);
}
 
MDBManager *MythCoreContext::GetDBManager(void)
{
    return d->m_database->GetDBManager();
}
 
bool MythCoreContext::IsDatabaseIgnored(void) const
{
    return d->m_database->IsDatabaseIgnored();
}
 
void MythCoreContext::SaveSetting(const QString &key, int newValue)
{
    d->m_database->SaveSetting(key, newValue);
}
 
void MythCoreContext::SaveSetting(const QString &key, const QString &newValue)
{
    d->m_database->SaveSetting(key, newValue);
}
 
bool MythCoreContext::SaveSettingOnHost(const QString &key,
                                        const QString &newValue,
                                        const QString &host)
{
    return d->m_database->SaveSettingOnHost(key, newValue, host);
}
 
QString MythCoreContext::GetSetting(const QString &key,
                                    const QString &defaultval)
{
    if (!m_testOverrideStrings.empty())
        return m_testOverrideStrings[key];
    return d->m_database->GetSetting(key, defaultval);
}
 
bool MythCoreContext::GetBoolSetting(const QString &key, bool defaultval)
{
    int result = GetNumSetting(key, static_cast<int>(defaultval));
    return result > 0;
}
 
int MythCoreContext::GetNumSetting(const QString &key, int defaultval)
{
    if (!m_testOverrideInts.empty())
        return m_testOverrideInts[key];
    return d->m_database->GetNumSetting(key, defaultval);
}
 
double MythCoreContext::GetFloatSetting(const QString &key, double defaultval)
{
    if (!m_testOverrideFloats.empty())
        return m_testOverrideFloats[key];
    return d->m_database->GetFloatSetting(key, defaultval);
}
 
QString MythCoreContext::GetSettingOnHost(const QString &key,
                                          const QString &host,
                                          const QString &defaultval)
{
    if (!m_testOverrideStrings.empty())
        return m_testOverrideStrings[key];
    return d->m_database->GetSettingOnHost(key, host, defaultval);
}
 
bool MythCoreContext::GetBoolSettingOnHost(const QString &key,
                                           const QString &host,
                                           bool defaultval)
{
    int result = GetNumSettingOnHost(key, host, static_cast<int>(defaultval));
    return result > 0;
}
 
int MythCoreContext::GetNumSettingOnHost(const QString &key,
                                         const QString &host,
                                         int defaultval)
{
    if (!m_testOverrideInts.empty())
        return m_testOverrideInts[key];
    return d->m_database->GetNumSettingOnHost(key, host, defaultval);
}
 
double MythCoreContext::GetFloatSettingOnHost(const QString &key,
                                              const QString &host,
                                              double defaultval)
{
    if (!m_testOverrideFloats.empty())
        return m_testOverrideFloats[key];
    return d->m_database->GetFloatSettingOnHost(key, host, defaultval);
}
 
QString MythCoreContext::GetMasterServerIP(void)
{
    QString masterserver = gCoreContext->GetSetting("MasterServerName");
    QString masterip = resolveSettingAddress("BackendServerAddr",masterserver);
    // Even if empty, return it here if we were to assume that localhost
    // should be used it just causes a lot of unnecessary error messages.
    return masterip;
}
 
int MythCoreContext::GetMasterServerPort(void)
{
    QString masterserver = gCoreContext->GetSetting
        ("MasterServerName");
    return gCoreContext->GetNumSettingOnHost
        ("BackendServerPort", masterserver, 6543);
}
 
int MythCoreContext::GetMasterServerStatusPort(void)
{
    QString masterhost = GetMasterHostName();
 
    return GetBackendStatusPort(masterhost);
}
 
QString MythCoreContext::GetBackendServerIP(void)
{
    return GetBackendServerIP(d->m_localHostname);
}
 
QString MythCoreContext::GetBackendServerIP(const QString &host)
{
    return resolveSettingAddress("BackendServerAddr",host);
}
 
QString MythCoreContext::GetBackendServerIP4(void)
{
    return GetBackendServerIP4(d->m_localHostname);
}
 
QString MythCoreContext::GetBackendServerIP4(const QString &host)
{
    return resolveSettingAddress("BackendServerIP", host, ResolveIPv4);
}
 
QString MythCoreContext::GetBackendServerIP6(void)
{
    return GetBackendServerIP6(d->m_localHostname);
}
 
QString MythCoreContext::GetBackendServerIP6(const QString &host)
{
    return resolveSettingAddress("BackendServerIP6", host, ResolveIPv6);
}
 
int MythCoreContext::GetBackendServerPort(void)
{
    return GetBackendServerPort(d->m_localHostname);
}
 
QHash<QString,int> MythCoreContext::s_serverPortCache;
 
void MythCoreContext::ClearBackendServerPortCache()
{
    s_serverPortCache.clear();
}
 
int MythCoreContext::GetBackendServerPort(const QString &host)
{
    int port = s_serverPortCache.value(host, -1);
    if (port != -1)
        return port;
    port = GetNumSettingOnHost("BackendServerPort", host, 6543);
    s_serverPortCache[host] = port;
    return port;
}
 
int MythCoreContext::GetBackendStatusPort(void)
{
    return GetBackendStatusPort(d->m_localHostname);
}
 
int MythCoreContext::GetBackendStatusPort(const QString &host)
{
    return GetNumSettingOnHost("BackendStatusPort", host, 6544);
}
 
bool MythCoreContext::GetScopeForAddress(QHostAddress &addr) const
{
    QHostAddress addr1  = addr;
    addr1.setScopeId(QString());
    QString addrstr     = addr1.toString();
    QMutexLocker lock(&d->m_scopesLock);
 
    if (!d->m_scopes.contains(addrstr))
        return false;
 
    addr.setScopeId(d->m_scopes[addrstr]);
    return true;
}
 
void MythCoreContext::SetScopeForAddress(const QHostAddress &addr)
{
    QHostAddress addr1 = addr;
    addr1.setScopeId(QString());
    QMutexLocker lock(&d->m_scopesLock);
 
    d->m_scopes.insert(addr1.toString(), addr.scopeId());
}
 
void MythCoreContext::SetScopeForAddress(const QHostAddress &addr, int scope)
{
    QHostAddress addr1 = addr;
    addr1.setScopeId(QString());
    QMutexLocker lock(&d->m_scopesLock);
 
    d->m_scopes.insert(addr1.toString(), QString::number(scope));
}
 
QString MythCoreContext::resolveSettingAddress(const QString &name,
                                               const QString &host,
                                               ResolveType type, bool keepscope)
{
    QString value;
 
    if (host.isEmpty())
    {
        value = GetSetting(name);
    }
    else
    {
        value = GetSettingOnHost(name, host);
    }
 
    return resolveAddress(value, type, keepscope);
}
 
QString MythCoreContext::resolveAddress(const QString &host, ResolveType type,
                                        bool keepscope)
{
    QHostAddress addr(host);
 
    if (!host.isEmpty() && addr.isNull())
    {
        // address is likely a hostname, attempts to resolve it
        QHostInfo info = QHostInfo::fromName(host);
        QList<QHostAddress> list = info.addresses();
 
        if (list.isEmpty())
        {
            LOG(VB_GENERAL, LOG_WARNING, LOC +
                QString("Can't resolve hostname:'%1', using localhost").arg(host));
            return type == ResolveIPv4 ? "127.0.0.1" : "::1";
        }
        QHostAddress v4;
        QHostAddress v6;
 
        // Return the first address fitting the type critera
        for (const auto& item : qAsConst(list))
        {
            addr = item;
            QAbstractSocket::NetworkLayerProtocol prot = addr.protocol();
 
            if (prot == QAbstractSocket::IPv4Protocol)
            {
                v4 = addr;
                if (type == 0)
                    break;
            }
            else if (prot == QAbstractSocket::IPv6Protocol)
            {
                v6 = addr;
                if (type != 0)
                    break;
            }
        }
        switch (type)
        {
            case ResolveIPv4:
                addr = v4.isNull() ? QHostAddress::LocalHost : v4;
                break;
            case ResolveIPv6:
                addr = v6.isNull() ? QHostAddress::LocalHostIPv6 : v6;
                break;
            default:
                addr = v6.isNull() ?
                    (v4.isNull() ? QHostAddress::LocalHostIPv6 : v4) : v6;
                break;
        }
    }
    else if (host.isEmpty())
    {
        return {};
    }
 
    if (!keepscope)
    {
        addr.setScopeId(QString());
    }
    return addr.toString();
}
 
bool MythCoreContext::CheckSubnet(const QAbstractSocket *socket)
{
    QHostAddress peer = socket->peerAddress();
    return CheckSubnet(peer);
}
 
bool MythCoreContext::CheckSubnet(const QHostAddress &peer)
{
    static const QHostAddress kLinkLocal("fe80::");
    if (GetBoolSetting("AllowConnFromAll",false))
        return true;
    if (d->m_approvedIps.contains(peer))
        return true;
    if (d->m_deniedIps.contains(peer))
    {
        LOG(VB_GENERAL, LOG_WARNING, LOC +
          QString("Repeat denied connection from ip address: %1")
          .arg(peer.toString()));
        return false;
    }
 
    // allow all link-local
    if (peer.isInSubnet(kLinkLocal,10))
    {
        d->m_approvedIps.append(peer);
        return true;
    }
 
    // loop through all available interfaces
    QList<QNetworkInterface> IFs = QNetworkInterface::allInterfaces();
    for (const auto & qni : qAsConst(IFs))
    {
        if ((qni.flags() & QNetworkInterface::IsRunning) == 0)
            continue;
 
        QList<QNetworkAddressEntry> IPs = qni.addressEntries();
        for (const auto & qnai : qAsConst(IPs))
        {
            int pfxlen = qnai.prefixLength();
            // Set this to test rejection without having an extra
            // network.
            if (GetBoolSetting("DebugSubnet"))
                pfxlen += 4;
            if (peer.isInSubnet(qnai.ip(),pfxlen))
            {
                d->m_approvedIps.append(peer);
                return true;
            }
        }
    }
    d->m_deniedIps.append(peer);
    LOG(VB_GENERAL, LOG_WARNING, LOC +
        QString("Denied connection from ip address: %1")
        .arg(peer.toString()));
    return false;
}
 
 
void MythCoreContext::OverrideSettingForSession(const QString &key,
                                                const QString &value)
{
    d->m_database->OverrideSettingForSession(key, value);
}
 
void MythCoreContext::ClearOverrideSettingForSession(const QString &key)
{
    d->m_database->ClearOverrideSettingForSession(key);
}
 
bool MythCoreContext::IsUIThread(void)
{
    return is_current_thread(d->m_uiThread);
}
 
bool MythCoreContext::SendReceiveStringList(
    QStringList &strlist, bool quickTimeout, bool block)
{
    QString msg;
    if (HasGUI() && IsUIThread())
    {
        msg = "SendReceiveStringList(";
        for (uint i=0; i<(uint)strlist.size() && i<2; i++)
            msg += (i?",":"") + strlist[i];
        msg += (strlist.size() > 2) ? "...)" : ")";
        LOG(VB_GENERAL, LOG_DEBUG, LOC + msg + " called from UI thread");
    }
 
    QString query_type = "UNKNOWN";
 
    if (!strlist.isEmpty())
        query_type = strlist[0];
 
    QMutexLocker locker(&d->m_sockLock);
    if (!d->m_serverSock)
    {
        bool blockingClient = d->m_blockingClient &&
                             (GetNumSetting("idleTimeoutSecs",0) > 0);
        ConnectToMasterServer(blockingClient);
    }
 
    bool ok = false;
 
    if (d->m_serverSock)
    {
        std::chrono::milliseconds timeout = quickTimeout ?
            MythSocket::kShortTimeout : MythSocket::kLongTimeout;
        ok = d->m_serverSock->SendReceiveStringList(strlist, 0, timeout);
 
        if (!ok)
        {
            LOG(VB_GENERAL, LOG_NOTICE, LOC +
                QString("Connection to backend server lost"));
            d->m_serverSock->DecrRef();
            d->m_serverSock = nullptr;
 
            if (d->m_eventSock)
            {
                d->m_eventSock->DecrRef();
                d->m_eventSock = nullptr;
            }
 
            if (block)
            {
                ConnectToMasterServer(d->m_blockingClient);
 
                if (d->m_serverSock)
                {
                    ok = d->m_serverSock->SendReceiveStringList(
                        strlist, 0, timeout);
                }
            }
        }
 
        // this should not happen
        while (ok && strlist[0] == "BACKEND_MESSAGE")
        {
            // oops, not for us
            LOG(VB_GENERAL, LOG_EMERG, LOC + "SRSL you shouldn't see this!!");
            QString message = strlist[1];
            strlist.pop_front(); strlist.pop_front();
 
            MythEvent me(message, strlist);
            dispatch(me);
 
            ok = d->m_serverSock->ReadStringList(strlist, timeout);
        }
 
        if (!ok)
        {
            if (d->m_serverSock)
            {
                d->m_serverSock->DecrRef();
                d->m_serverSock = nullptr;
            }
 
            LOG(VB_GENERAL, LOG_CRIT, LOC +
                QString("Reconnection to backend server failed"));
 
            QCoreApplication::postEvent(d->m_guiContext,
                                new MythEvent("PERSISTENT_CONNECTION_FAILURE"));
        }
    }
 
    if (ok)
    {
        if (strlist.isEmpty())
            ok = false;
        else if (strlist[0] == "ERROR")
        {
            if (strlist.size() == 2)
            {
                LOG(VB_GENERAL, LOG_INFO, LOC +
                    QString("Protocol query '%1' responded with the error '%2'")
                        .arg(query_type, strlist[1]));
            }
            else
            {
                LOG(VB_GENERAL, LOG_INFO, LOC +
                    QString("Protocol query '%1' responded with an error, but "
                            "no error message.") .arg(query_type));
            }
 
            ok = false;
        }
        else if (strlist[0] == "UNKNOWN_COMMAND")
        {
            LOG(VB_GENERAL, LOG_ERR, LOC +
                QString("Protocol query '%1' responded with the error 'UNKNOWN_COMMAND'")
                        .arg(query_type));
 
            ok = false;
        }
    }
 
    return ok;
}
 
class SendAsyncMessage : public QRunnable
{
  public:
    SendAsyncMessage(QString msg, QStringList extra) :
        m_message(std::move(msg)), m_extraData(std::move(extra))
    {
    }
 
    explicit SendAsyncMessage(QString msg) : m_message(std::move(msg)) { }
 
    void run(void) override // QRunnable
    {
        QStringList strlist("MESSAGE");
        strlist << m_message;
        strlist << m_extraData;
        gCoreContext->SendReceiveStringList(strlist);
    }
 
  private:
    QString m_message;
    QStringList m_extraData;
};
 
void MythCoreContext::SendMessage(const QString &message)
{
    if (IsBackend())
    {
        dispatch(MythEvent(message));
    }
    else
    {
        MThreadPool::globalInstance()->start(
            new SendAsyncMessage(message), "SendMessage");
    }
}
 
void MythCoreContext::SendEvent(const MythEvent &event)
{
    if (IsBackend())
    {
        dispatch(event);
    }
    else
    {
        MThreadPool::globalInstance()->start(
            new SendAsyncMessage(event.Message(), event.ExtraDataList()),
            "SendEvent");
    }
}
 
void MythCoreContext::SendSystemEvent(const QString &msg)
{
    if (QCoreApplication::applicationName() == MYTH_APPNAME_MYTHTV_SETUP)
        return;
 
    SendMessage(QString("SYSTEM_EVENT %1 SENDER %2")
                        .arg(msg, GetHostName()));
}
 
void MythCoreContext::SendHostSystemEvent(const QString &msg,
                                          const QString &hostname,
                                          const QString &args)
{
    SendSystemEvent(QString("%1 HOST %2 %3").arg(msg, hostname, args));
}
 
 
void MythCoreContext::readyRead(MythSocket *sock)
{
    do
    {
        QStringList strlist;
        if (!sock->ReadStringList(strlist))
            continue;
 
        if (strlist.size() < 2)
            continue;
 
        QString prefix = strlist[0];
        QString message = strlist[1];
#if QT_VERSION < QT_VERSION_CHECK(5,14,0)
        QStringList tokens = message.split(" ", QString::SkipEmptyParts);
#else
        QStringList tokens = message.split(" ", Qt::SkipEmptyParts);
#endif
 
        if (prefix == "OK")
        {
        }
        else if (prefix != "BACKEND_MESSAGE")
        {
            LOG(VB_NETWORK, LOG_ERR, LOC +
                    QString("Received a: %1 message from the backend "
                            "but I don't know what to do with it.")
                        .arg(prefix));
        }
        else if (message == "CLEAR_SETTINGS_CACHE")
        {
            // No need to dispatch this message to ourself, so handle it
            LOG(VB_NETWORK, LOG_INFO, LOC + "Received remote 'Clear Cache' request");
            ClearSettingsCache();
        }
        else if (message.startsWith("FILE_WRITTEN"))
        {
            QString file;
            uint64_t size = 0;
            int NUMTOKENS = 3; // Number of tokens expected
 
            if (tokens.size() == NUMTOKENS)
            {
                file = tokens[1];
                size = tokens[2].toULongLong();
            }
            else
            {
                LOG(VB_NETWORK, LOG_ERR, LOC +
                    QString("FILE_WRITTEN event received "
                            "with invalid number of arguments, "
                            "%1 expected, %2 actual")
                    .arg(NUMTOKENS-1)
                    .arg(tokens.size()-1));
                return;
            }
            // No need to dispatch this message to ourself, so handle it
            LOG(VB_NETWORK, LOG_INFO, LOC +
                QString("Received remote 'FILE_WRITTEN %1' request").arg(file));
            RegisterFileForWrite(file, size);
        }
        else if (message.startsWith("FILE_CLOSED"))
        {
            QString file;
            int NUMTOKENS = 2; // Number of tokens expected
 
            if (tokens.size() == NUMTOKENS)
            {
                file = tokens[1];
            }
            else
            {
                LOG(VB_NETWORK, LOG_ERR, LOC +
                    QString("FILE_CLOSED event received "
                            "with invalid number of arguments, "
                            "%1 expected, %2 actual")
                    .arg(NUMTOKENS-1)
                    .arg(tokens.size()-1));
                return;
            }
            // No need to dispatch this message to ourself, so handle it
            LOG(VB_NETWORK, LOG_INFO, LOC +
                QString("Received remote 'FILE_CLOSED %1' request").arg(file));
            UnregisterFileForWrite(file);
        }
        else
        {
            strlist.pop_front();
            strlist.pop_front();
            MythEvent me(message, strlist);
            dispatch(me);
        }
    }
    while (sock->IsDataAvailable());
}
 
void MythCoreContext::connectionClosed(MythSocket *sock)
{
    (void)sock;
 
    LOG(VB_GENERAL, LOG_NOTICE, LOC +
        "Event socket closed.  No connection to the backend.");
 
    dispatch(MythEvent("BACKEND_SOCKETS_CLOSED"));
}
 
bool MythCoreContext::CheckProtoVersion(MythSocket *socket,
                                        std::chrono::milliseconds timeout,
                                        bool error_dialog_desired)
{
    if (!socket)
        return false;
 
    QStringList strlist(QString("MYTH_PROTO_VERSION %1 %2")
                        .arg(MYTH_PROTO_VERSION,
                             QString::fromUtf8(MYTH_PROTO_TOKEN)));
    socket->WriteStringList(strlist);
 
    if (!socket->ReadStringList(strlist, timeout) || strlist.empty())
    {
        LOG(VB_GENERAL, LOG_CRIT, "Protocol version check failure.\n\t\t\t"
                "The response to MYTH_PROTO_VERSION was empty.\n\t\t\t"
                "This happens when the backend is too busy to respond,\n\t\t\t"
                "or has deadlocked due to bugs or hardware failure.");
 
        return false;
    }
    if (strlist[0] == "REJECT" && strlist.size() >= 2)
    {
        LOG(VB_GENERAL, LOG_CRIT, LOC + QString("Protocol version or token mismatch "
                                          "(frontend=%1/%2,backend=%3/\?\?)\n")
                                      .arg(MYTH_PROTO_VERSION,
                                           QString::fromUtf8(MYTH_PROTO_TOKEN),
                                           strlist[1]));
 
        if (error_dialog_desired && d->m_guiContext)
        {
            QStringList list(strlist[1]);
            QCoreApplication::postEvent(
                d->m_guiContext, new MythEvent("VERSION_MISMATCH", list));
        }
 
        return false;
    }
    if (strlist[0] == "ACCEPT")
    {
        if (!d->m_announcedProtocol)
        {
            d->m_announcedProtocol = true;
            LOG(VB_GENERAL, LOG_INFO, LOC +
                            QString("Using protocol version %1 %2")
                                .arg(MYTH_PROTO_VERSION,
                                     QString::fromUtf8(MYTH_PROTO_TOKEN)));
        }
 
        return true;
    }
 
    LOG(VB_GENERAL, LOG_ERR, LOC +
        QString("Unexpected response to MYTH_PROTO_VERSION: %1")
            .arg(strlist[0]));
    return false;
}
 
void MythCoreContext::dispatch(const MythEvent &event)
{
    LOG(VB_NETWORK, LOG_INFO, LOC + QString("MythEvent: %1").arg(event.Message()));
 
    MythObservable::dispatch(event);
}
 
void MythCoreContext::SetLocalHostname(const QString &hostname)
{
    QMutexLocker locker(&d->m_localHostLock);
    d->m_localHostname = hostname;
    d->m_database->SetLocalHostname(hostname);
}
 
void MythCoreContext::SetGUIObject(QObject *gui)
{
    d->m_guiObject = gui;
}
 
bool MythCoreContext::HasGUI(void) const
{
    return d->m_guiObject;
}
 
QObject *MythCoreContext::GetGUIObject(void)
{
    return d->m_guiObject;
}
 
QObject *MythCoreContext::GetGUIContext(void)
{
    return d->m_guiContext;
}
 
MythDB *MythCoreContext::GetDB(void)
{
    return d->m_database;
}
 
MythLocale *MythCoreContext::GetLocale(void) const
{
    return d->m_locale;
}
 
QString MythCoreContext::GetLanguage(void)
{
    return GetLanguageAndVariant().left(2);
}
 
QString MythCoreContext::GetLanguageAndVariant(void)
{
    if (d->m_language.isEmpty())
        d->m_language = GetSetting("Language", "en_US").toLower();
 
    return d->m_language;
}
 
void MythCoreContext::ResetLanguage(void)
{
    d->m_language.clear();
}
 
void MythCoreContext::ResetSockets(void)
{
    QMutexLocker locker(&d->m_sockLock);
    LOG(VB_GENERAL, LOG_INFO, "Restarting Backend Connections");
    if (d->m_serverSock)
        d->m_serverSock->DisconnectFromHost();
    if (d->m_eventSock)
        d->m_eventSock->DisconnectFromHost();
    dispatch(MythEvent("BACKEND_SOCKETS_CLOSED"));
}
 
void MythCoreContext::InitPower(bool Create)
{
    if (Create && !d->m_power)
    {
        d->m_power = MythPower::AcquireRelease(d, true);
    }
    else if (!Create && d->m_power)
    {
        MythPower::AcquireRelease(d, false);
        d->m_power = nullptr;
    }
}
 
void MythCoreContext::InitLocale(void)
{
    if (!d->m_locale)
        d->m_locale = new MythLocale();
 
    QString localeCode = d->m_locale->GetLocaleCode();
    LOG(VB_GENERAL, LOG_NOTICE, QString("Setting QT default locale to %1")
                                                            .arg(localeCode));
    QLocale::setDefault(d->m_locale->ToQLocale());
}
 
void MythCoreContext::ReInitLocale(void)
{
    if (!d->m_locale)
        d->m_locale = new MythLocale();
    else
        d->m_locale->ReInit();
 
    QString localeCode = d->m_locale->GetLocaleCode();
    LOG(VB_GENERAL, LOG_NOTICE, QString("Setting QT default locale to %1")
                                                            .arg(localeCode));
    QLocale::setDefault(d->m_locale->ToQLocale());
}
 
QLocale MythCoreContext::GetQLocale(void)
{
    if (!d->m_locale)
        InitLocale();
 
    return d->m_locale->ToQLocale();
}
 
void MythCoreContext::SaveLocaleDefaults(void)
{
    if (!d->m_locale)
        InitLocale();
 
    if (!d->m_locale->GetLocaleCode().isEmpty())
    {
        LOG(VB_GENERAL, LOG_INFO,
            QString("Current locale %1") .arg(d->m_locale->GetLocaleCode()));
 
        d->m_locale->SaveLocaleDefaults();
        return;
    }
 
    LOG(VB_GENERAL, LOG_ERR, LOC +
        "No locale defined! We weren't able to set locale defaults.");
}
 
void MythCoreContext::SetScheduler(MythScheduler *sched)
{
    d->m_scheduler = sched;
}
 
MythScheduler *MythCoreContext::GetScheduler(void)
{
    return d->m_scheduler;
}
 
void MythCoreContext::WaitUntilSignals(std::vector<CoreWaitInfo> & sigs) const
{
    if (sigs.empty())
        return;
 
    QEventLoop eventLoop;
    for (const auto& s : sigs)
    {
        LOG(VB_GENERAL, LOG_DEBUG, LOC +
            QString("Waiting for signal %1")
            .arg(s.name));
        connect(this, s.fn, &eventLoop, &QEventLoop::quit);
    }
 
    eventLoop.exec(QEventLoop::ExcludeUserInputEvents | QEventLoop::ExcludeSocketNotifiers);
}
 
void MythCoreContext::RegisterForPlayback(QObject *sender, PlaybackStartCb method)
{
    if (!sender || !method)
        return;
 
    QMutexLocker lock(&d->m_playbackLock);
 
    if (!d->m_playbackClients.contains(sender))
    {
        d->m_playbackClients.insert(sender, method);
        connect(this, &MythCoreContext::TVPlaybackAboutToStart,
                sender, method,
                Qt::BlockingQueuedConnection);
    }
}
 
void MythCoreContext::UnregisterForPlayback(QObject *sender)
{
    QMutexLocker lock(&d->m_playbackLock);
 
    if (d->m_playbackClients.contains(sender))
    {
        PlaybackStartCb method = d->m_playbackClients.value(sender);
        disconnect(this, &MythCoreContext::TVPlaybackAboutToStart,
                   sender, method);
        d->m_playbackClients.remove(sender);
    }
}
 
void MythCoreContext::WantingPlayback(QObject *sender)
{
    QMutexLocker lock(&d->m_playbackLock);
    PlaybackStartCb method { nullptr };
    d->m_inwanting = true;
 
    // If any registered client are in the same thread, they will deadlock, so rebuild
    // connections for any clients in the same thread as non-blocking connection
    QThread *currentThread = QThread::currentThread();
 
    QMap<QObject *, PlaybackStartCb>::iterator it = d->m_playbackClients.begin();
    for (; it != d->m_playbackClients.end(); ++it)
    {
        if (it.key() == sender)
            continue;   // will be done separately, no need to do it again
 
        QThread *thread = it.key()->thread();
 
        if (thread != currentThread)
            continue;
 
        disconnect(this, &MythCoreContext::TVPlaybackAboutToStart, it.key(), it.value());
        connect(this, &MythCoreContext::TVPlaybackAboutToStart, it.key(), it.value());
    }
 
    // disconnect sender so it won't receive the message
    if (d->m_playbackClients.contains(sender))
    {
        method = d->m_playbackClients.value(sender);
        disconnect(this, &MythCoreContext::TVPlaybackAboutToStart, sender, method);
    }
 
    // emit signal
    emit TVPlaybackAboutToStart();
 
    // reconnect sender
    if (method)
    {
        connect(this, &MythCoreContext::TVPlaybackAboutToStart,
                sender, method,
                Qt::BlockingQueuedConnection);
    }
    // Restore blocking connections
    for (; it != d->m_playbackClients.end(); ++it)
    {
        if (it.key() == sender)
            continue;   // already done above, no need to do it again
 
        QThread *thread = it.key()->thread();
 
        if (thread != currentThread)
            continue;
 
        disconnect(this, &MythCoreContext::TVPlaybackAboutToStart,
                   it.key(), it.value());
        connect(this, &MythCoreContext::TVPlaybackAboutToStart,
                it.key(), it.value(), Qt::BlockingQueuedConnection);
    }
    d->m_inwanting = false;
}
 
void MythCoreContext::TVInWantingPlayback(bool b)
{
    // when called, it will be while the m_playbackLock is held
    // following a call to WantingPlayback
    d->m_intvwanting = b;
}
 
bool MythCoreContext::InWantingPlayback(void)
{
    bool locked = d->m_playbackLock.tryLock();
    bool intvplayback = d->m_intvwanting;
 
    if (!locked && d->m_inwanting)
        return true; // we're in the middle of WantingPlayback
 
    if (!locked)
        return false;
 
    d->m_playbackLock.unlock();
 
    return intvplayback;
}
 
MythSessionManager* MythCoreContext::GetSessionManager(void)
{
    if (!d->m_sessionManager)
        d->m_sessionManager = new MythSessionManager();
 
    return d->m_sessionManager;
}
 
bool MythCoreContext::TestPluginVersion(const QString &name,
                                   const QString &libversion,
                                   const QString &pluginversion)
{
    if (libversion == pluginversion)
        return true;
 
    LOG(VB_GENERAL, LOG_EMERG, LOC +
             QString("Plugin %1 (%2) binary version does not "
                     "match libraries (%3)")
                 .arg(name, pluginversion, libversion));
    return false;
}
 
void MythCoreContext::SetPluginManager(MythPluginManager *pmanager)
{
    if (d->m_pluginmanager == pmanager)
        return;
 
    if (d->m_pluginmanager)
    {
        delete d->m_pluginmanager;
        d->m_pluginmanager = nullptr;
    }
 
    d->m_pluginmanager = pmanager;
}
 
MythPluginManager *MythCoreContext::GetPluginManager(void)
{
    return d->m_pluginmanager;
}
 
void MythCoreContext::SetExiting(bool exiting)
{
    d->m_isexiting = exiting;
}
 
bool MythCoreContext::IsExiting(void)
{
    return d->m_isexiting;
}
 
void MythCoreContext::RegisterFileForWrite(const QString& file, uint64_t size)
{
    QMutexLocker lock(&d->m_fileslock);
 
    QPair<int64_t, uint64_t> pair(QDateTime::currentMSecsSinceEpoch(), size);
    d->m_fileswritten.insert(file, pair);
 
    if (IsBackend())
    {
        QString message = QString("FILE_WRITTEN %1 %2").arg(file).arg(size);
        MythEvent me(message);
        dispatch(me);
    }
 
    LOG(VB_FILE, LOG_DEBUG, LOC +
        QString("%1").arg(file));
}
 
void MythCoreContext::UnregisterFileForWrite(const QString& file)
{
    QMutexLocker lock(&d->m_fileslock);
 
    d->m_fileswritten.remove(file);
 
    if (IsBackend())
    {
        QString message = QString("FILE_CLOSED %1").arg(file);
        MythEvent me(message);
        dispatch(me);
    }
 
    LOG(VB_FILE, LOG_DEBUG, LOC +
        QString("%1").arg(file));
}
 
bool MythCoreContext::IsRegisteredFileForWrite(const QString& file)
{
    QMutexLocker lock(&d->m_fileslock);
 
    return d->m_fileswritten.contains(file);
}
 
/* vim: set expandtab tabstop=4 shiftwidth=4: */