dbutil.cpp

Go to the documentation of this file.

#include <climits>
#include <cstdio>
#include <cstdlib>
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
 
#include <QDir>
#include <QFile>
#include <QRegularExpression>
#include <QDateTime>
#include <QSqlError>
#include <QSqlRecord>
 
#include "dbutil.h"
#include "mythcorecontext.h"
#include "storagegroup.h"
#include "mythmiscutil.h"
#include "mythdate.h"
#include "mythdb.h"
#include "mythdirs.h"
#include "mythlogging.h"
#include "mythsystemlegacy.h"
#include "exitcodes.h"
 
#define LOC QString("DBUtil: ")
 
const int DBUtil::kUnknownVersionNumber = INT_MIN;
 
QString DBUtil::GetDBMSVersion(void)
{
    if (m_versionString.isEmpty())
        QueryDBMSVersion();
    return m_versionString;
}
 
int DBUtil::CompareDBMSVersion(int major, int minor, int point)
{
    if (m_versionMajor < 0)
        if (!ParseDBMSVersion())
           return kUnknownVersionNumber;
 
    int result = 0;
    std::array<int,3> version {m_versionMajor, m_versionMinor, m_versionPoint};
    std::array<int,3> compareto {major, minor, point};
    for (int i = 0; i < 3 && !result; i++)
    {
        if ((version[i] > -1) || (compareto[i] != 0))
            result = version[i] - compareto[i];
    }
 
    return result;
}
 
bool DBUtil::IsNewDatabase(void)
{
    const QStringList tables = GetTables();
    const int size = tables.size();
    // Usually there will be a single table called schemalock, but check for
    // no tables, also, just in case.
    return (((size == 1) && (tables.at(0).endsWith(".`schemalock`"))) ||
            (size == 0));
}
 
bool DBUtil::IsBackupInProgress(void)
{
    QString backupStartTimeStr =
        gCoreContext->GetSetting("BackupDBLastRunStart");
    QString backupEndTimeStr = gCoreContext->GetSetting("BackupDBLastRunEnd");
 
    if (backupStartTimeStr.isEmpty())
    {
        LOG(VB_DATABASE, LOG_ERR, "DBUtil::BackupInProgress(): No start time "
                                  "found, database backup is not in progress.");
        return false;
    }
 
    backupStartTimeStr.replace(" ", "T");
 
    QDateTime backupStartTime = MythDate::fromString(backupStartTimeStr);
    auto backupElapsed = MythDate::secsInPast(backupStartTime);
 
    // No end time set
    if (backupEndTimeStr.isEmpty())
    {
        // If DB Backup started less then 10 minutes ago, assume still running
        if (backupElapsed < 10min)
        {
            LOG(VB_DATABASE, LOG_INFO,
                QString("DBUtil::BackupInProgress(): Found "
                    "database backup start time of %1 which was %2 seconds "
                    "ago, therefore it appears the backup is still running.")
                    .arg(backupStartTimeStr)
                    .arg(backupElapsed.count()));
            return true;
        }
        LOG(VB_DATABASE, LOG_ERR, QString("DBUtil::BackupInProgress(): "
                "Database backup started at %1, but no end time was found. "
                "The backup started %2 seconds ago and should have "
                "finished by now therefore it appears it is not running .")
                .arg(backupStartTimeStr)
                .arg(backupElapsed.count()));
        return false;
    }
 
    backupEndTimeStr.replace(" ", "T");
 
    QDateTime backupEndTime = MythDate::fromString(backupEndTimeStr);
 
    if (backupEndTime >= backupStartTime)
    {
        LOG(VB_DATABASE, LOG_ERR,
            QString("DBUtil::BackupInProgress(): Found "
                    "database backup end time of %1 later than start time "
                    "of %2, therefore backup is not running.")
            .arg(backupEndTimeStr, backupStartTimeStr));
        return false;
    }
    if (backupElapsed > 10min)
    {
        LOG(VB_DATABASE, LOG_ERR,
            QString("DBUtil::BackupInProgress(): "
                    "Database backup started at %1, but has not ended yet.  "
                    "The backup started %2 seconds ago and should have "
                    "finished by now therefore it appears it is not running")
            .arg(backupStartTimeStr)
            .arg(backupElapsed.count()));
        return false;
    }
 
    // start > end and started less than 10 minutes ago
    LOG(VB_DATABASE, LOG_INFO, QString("DBUtil::BackupInProgress(): "
                                       "Database backup started at %1, and is still running.")
        .arg(backupStartTimeStr));
    return true;
}
 
MythDBBackupStatus DBUtil::BackupDB(QString &filename,
                                    [[maybe_unused]] bool disableRotation)
{
    filename = QString();
 
#ifdef _WIN32
    LOG(VB_GENERAL, LOG_CRIT, "Database backups disabled on Windows.");
    return kDB_Backup_Disabled;
#else
 
    if (gCoreContext->GetBoolSetting("DisableAutomaticBackup", false))
    {
        LOG(VB_GENERAL, LOG_CRIT,
            "Database backups disabled.  Skipping backup.");
        return kDB_Backup_Disabled;
    }
 
    if (IsNewDatabase())
    {
        LOG(VB_GENERAL, LOG_CRIT, "New database detected.  Skipping backup.");
        return kDB_Backup_Empty_DB;
    }
 
    QString backupScript = GetShareDir() + "mythconverg_backup.pl";
    backupScript = gCoreContext->GetSetting("DatabaseBackupScript",
                                            backupScript);
 
    if (!QFile::exists(backupScript))
    {
        LOG(VB_GENERAL, LOG_CRIT, QString("Database backup script does "
                                          "not exist: %1").arg(backupScript));
        backupScript.clear();
    }
 
    bool result = false;
    MSqlQuery query(MSqlQuery::InitCon());
 
    gCoreContext->SaveSettingOnHost(
        "BackupDBLastRunStart",
        MythDate::toString(MythDate::current(), MythDate::kDatabase), nullptr);
 
    if (!backupScript.isEmpty())
    {
        result = DoBackup(backupScript, filename, disableRotation);
        if (!result)
            LOG(VB_GENERAL, LOG_CRIT, "Script-based database backup failed. "
                                      "Retrying with internal backup.");
    }
 
    if (!result)
        result = DoBackup(filename);
 
    gCoreContext->SaveSettingOnHost(
        "BackupDBLastRunEnd",
        MythDate::toString(MythDate::current(), MythDate::kDatabase), nullptr);
 
    if (query.isConnected())
    {
        QString dbTag("BackupDB");
        query.prepare("DELETE FROM housekeeping WHERE tag = :TAG ;");
        query.bindValue(":TAG", dbTag);
        if (!query.exec())
            MythDB::DBError("DBUtil::BackupDB", query);
 
        query.prepare("INSERT INTO housekeeping(tag,lastrun) "
                       "values(:TAG ,now()) ;");
        query.bindValue(":TAG", dbTag);
        if (!query.exec())
            MythDB::DBError("DBUtil::BackupDB", query);
    }
 
    if (result)
        return kDB_Backup_Completed;
 
    return kDB_Backup_Failed;
#endif // _WIN32
}
 
bool DBUtil::CheckTables(const bool repair, const QString &options)
{
    MSqlQuery query(MSqlQuery::InitCon());
    if (!query.isConnected())
        return false;
 
    const QStringList all_tables = GetTables(QStringList("MyISAM"));
 
    if (all_tables.empty())
        return true;
 
    QString sql = QString("CHECK TABLE %1 %2;")
        .arg(all_tables.join(", "), options);
 
    LOG(VB_GENERAL, LOG_CRIT, "Checking database tables.");
    if (!query.exec(sql))
    {
        MythDB::DBError("DBUtil Checking Tables", query);
        return false;
    }
 
    QStringList tables = CheckRepairStatus(query);
    bool result = true;
    if (!tables.empty())
    {
        LOG(VB_GENERAL, LOG_CRIT, QString("Found crashed database table(s): %1")
                                      .arg(tables.join(", ")));
        if (repair)
        {
            // If RepairTables() repairs the crashed tables, return true
            result = RepairTables(tables);
        }
        else
        {
            result = false;
        }
    }
 
    return result;
}
 
bool DBUtil::RepairTables(const QStringList &tables)
{
    MSqlQuery query(MSqlQuery::InitCon());
    if (!query.isConnected())
        return false;
 
    QString all_tables = tables.join(", ");
    LOG(VB_GENERAL, LOG_CRIT, QString("Repairing database tables: %1")
                                  .arg(all_tables));
 
    QString sql = QString("REPAIR TABLE %1;").arg(all_tables);
    if (!query.exec(sql))
    {
        MythDB::DBError("DBUtil Repairing Tables", query);
        return false;
    }
 
    QStringList bad_tables = CheckRepairStatus(query);
    bool result = true;
    if (!bad_tables.empty())
    {
        LOG(VB_GENERAL, LOG_CRIT,
            QString("Unable to repair crashed table(s): %1")
                .arg(bad_tables.join(", ")));
        result = false;
    }
    return result;
}
 
QStringList DBUtil::CheckRepairStatus(MSqlQuery &query)
{
    QStringList tables;
    QSqlRecord record = query.record();
    int table_index = record.indexOf("Table");
    int type_index = record.indexOf("Msg_type");
    int text_index = record.indexOf("Msg_text");
    QString table;
    QString type;
    QString text;
    QString previous_table;
    bool ok = true;
    while (query.next())
    {
        table = query.value(table_index).toString();
        type = query.value(type_index).toString();
        text = query.value(text_index).toString();
        if (table != previous_table)
        {
            if (!ok)
            {
                tables.append(previous_table);
                ok = true;
            }
            previous_table = table;
        }
        // If the final row shows status OK, the table is now good
        if ("status" == type.toLower() && "ok" == text.toLower())
            ok = true;
        else if ("error" == type.toLower() ||
                 ("status" == type.toLower() && "ok" != text.toLower()))
            ok = false;
    }
    // Check the last table in the list
    if (!ok)
        tables.append(table);
    return tables;
}
 
QStringList DBUtil::GetTables(const QStringList &engines)
{
    QStringList result;
 
    MSqlQuery query(MSqlQuery::InitCon());
    if (!query.isConnected())
        return result;
 
    QString sql = "SELECT CONCAT('`', TABLE_SCHEMA, "
                  "              '`.`', TABLE_NAME, "
                  "              '`') AS `TABLE_NAME` "
                  "  FROM INFORMATION_SCHEMA.TABLES "
                  " WHERE TABLE_SCHEMA = DATABASE() "
                  "   AND TABLE_TYPE = 'BASE TABLE'";
    if (!engines.empty())
        sql.append(QString("   AND ENGINE IN ('%1')")
                           .arg(engines.join("', '")));
    if (!query.exec(sql))
    {
        MythDB::DBError("DBUtil Finding Tables", query);
        return result;
    }
 
    while (query.next())
    {
        result.append(query.value(0).toString());
    }
 
    return result;
}
 
QString DBUtil::CreateBackupFilename(const QString& prefix, const QString& extension)
{
    QString time = MythDate::toString(MythDate::current(), MythDate::kFilename);
    return QString("%1-%2%3").arg(prefix, time, extension);
}
 
QString DBUtil::GetBackupDirectory()
{
    QString directory;
    StorageGroup sgroup("DB Backups", gCoreContext->GetHostName());
    QStringList dirList = sgroup.GetDirList();
    if (!dirList.empty())
    {
        directory = sgroup.FindNextDirMostFree();
 
        if (!QDir(directory).exists())
        {
            LOG(VB_FILE, LOG_INFO, "GetBackupDirectory() - ignoring " +
                                   directory + ", using /tmp");
            directory.clear();
        }
    }
 
    if (directory.isNull())
    {
        // Rather than use kDefaultStorageDir, the default for
        // FindNextDirMostFree() when no dirs are defined for the StorageGroup,
        // use /tmp as it's possible that kDefaultStorageDir doesn't exist
        // and (at least on *nix) less possible that /tmp doesn't exist
        directory = "/tmp";
    }
 
    return directory;
}
 
bool DBUtil::CreateTemporaryDBConf(
    const QString &privateinfo, QString &filename)
{
    bool ok = true;
    filename = createTempFile("/tmp/mythtv_db_backup_conf_XXXXXX");
    const QByteArray     tmpfile     = filename.toLocal8Bit();
 
    FILE *fp = fopen(tmpfile.constData(), "w");
    if (!fp)
    {
        LOG(VB_GENERAL, LOG_ERR, LOC +
            QString("Unable to create temporary "
                    "configuration file for creating DB backup: %1")
                .arg(tmpfile.constData()));
        filename = "";
        ok = false;
    }
    else
    {
        if (chmod(tmpfile.constData(), S_IRUSR))
        {
            LOG(VB_GENERAL, LOG_ERR, LOC + QString("Error changing permissions '%1'")
                    .arg(tmpfile.constData()) + ENO);
        }
 
        QByteArray outarr = privateinfo.toLocal8Bit();
        fprintf(fp, "%s", outarr.constData());
 
        if (fclose(fp))
        {
            LOG(VB_GENERAL, LOG_ERR, LOC + QString("Error closing '%1'")
                    .arg(tmpfile.constData()) + ENO);
        }
    }
 
    return ok;
}
 
bool DBUtil::DoBackup(const QString &backupScript, QString &filename,
                      bool disableRotation)
{
    DatabaseParams dbParams = GetMythDB()->GetDatabaseParams();
    QString     dbSchemaVer = gCoreContext->GetSetting("DBSchemaVer");
    QString backupDirectory = GetBackupDirectory();
    QString  backupFilename = CreateBackupFilename(dbParams.m_dbName + "-" +
                                                   dbSchemaVer, ".sql");
    QString      scriptArgs = gCoreContext->GetSetting("BackupDBScriptArgs");
    QString rotate = "";
    if (disableRotation)
    {
        if (!(scriptArgs.contains("rotate", Qt::CaseInsensitive)))
            rotate = "rotate=-1";
    }
 
 
    QString privateinfo =
        QString("DBHostName=%1\nDBPort=%2\n"
                "DBUserName=%3\nDBPassword=%4\n"
                "DBName=%5\nDBSchemaVer=%6\n"
                "DBBackupDirectory=%7\nDBBackupFilename=%8\n%9\n")
        .arg(dbParams.m_dbHostName, QString::number(dbParams.m_dbPort),
             dbParams.m_dbUserName, dbParams.m_dbPassword,
             dbParams.m_dbName,     dbSchemaVer,
             backupDirectory,       backupFilename,
             rotate);
    QString tempDatabaseConfFile;
    bool hastemp = CreateTemporaryDBConf(privateinfo, tempDatabaseConfFile);
    if (!hastemp)
        LOG(VB_GENERAL, LOG_ERR, LOC + "Attempting backup, anyway.");
 
    LOG(VB_GENERAL, LOG_ERR, QString("Backing up database with script: '%1'")
            .arg(backupScript));
 
    QString command = backupScript + " " + scriptArgs + " " +
                      tempDatabaseConfFile;
    uint status = myth_system(command, kMSDontBlockInputDevs|kMSAnonLog);
 
    if (hastemp)
    {
        QByteArray tmpfile = tempDatabaseConfFile.toLocal8Bit();
        unlink(tmpfile.constData());
    }
 
    if (status != GENERIC_EXIT_OK)
    {
        LOG(VB_GENERAL, LOG_ERR, LOC +
            QString("Error backing up database: %1 (%2)")
                .arg(command).arg(status));
        filename = "__FAILED__";
        return false;
    }
 
    LOG(VB_GENERAL, LOG_CRIT, "Database Backup complete.");
 
    QDir dir(backupDirectory, backupFilename + "*");
    uint numfiles = dir.count();
    if (numfiles < 1)
    {
        // If no file begins with the suggested filename, don't show the backup
        // filename in the GUI message -- the script probably used some other
        // filename
        filename = "";
        LOG(VB_FILE, LOG_ERR, LOC +
            QString("No files beginning with the suggested database backup "
                    "filename '%1' were found in '%2'.")
                .arg(backupFilename, backupDirectory));
    }
    else
    {
        filename = dir.path() + "/" + dir[0];;
        if (numfiles > 1)
        {
            LOG(VB_FILE, LOG_ERR, LOC +
                QString("Multiple files beginning with the suggested database "
                        "backup filename '%1' were found in '%2'. "
                        "Assuming the first is the backup.")
                    .arg(backupFilename, backupDirectory));
        }
    }
 
    if (!filename.isEmpty())
    {
        LOG(VB_GENERAL, LOG_CRIT, QString("Backed up database to file: '%1'")
                .arg(filename));
    }
 
    return true;
}
 
bool DBUtil::DoBackup(QString &filename)
{
    DatabaseParams dbParams = GetMythDB()->GetDatabaseParams();
    QString     dbSchemaVer = gCoreContext->GetSetting("DBSchemaVer");
    QString backupDirectory = GetBackupDirectory();
 
    QString command;
    QString compressCommand("");
    QString extension = ".sql";
    if (QFile::exists("/bin/gzip"))
        compressCommand = "/bin/gzip";
    else if (QFile::exists("/usr/bin/gzip"))
        compressCommand = "/usr/bin/gzip";
    else
        LOG(VB_GENERAL, LOG_CRIT, "Neither /bin/gzip nor /usr/bin/gzip exist. "
                                  "The database backup will be uncompressed.");
 
    QString backupFilename = CreateBackupFilename(
        dbParams.m_dbName + "-" + dbSchemaVer, extension);
    QString backupPathname = backupDirectory + "/" + backupFilename;
 
    QString privateinfo = QString(
        "[client]\npassword=%1\n[mysqldump]\npassword=%2\n")
        .arg(dbParams.m_dbPassword, dbParams.m_dbPassword);
    QString tempExtraConfFile;
    if (!CreateTemporaryDBConf(privateinfo, tempExtraConfFile))
        return false;
 
    QString portArg = "";
    if (dbParams.m_dbPort > 0)
        portArg = QString(" --port='%1'").arg(dbParams.m_dbPort);
    command = QString("mysqldump --defaults-extra-file='%1' --host='%2'%3"
                      " --user='%4' --add-drop-table --add-locks"
                      " --allow-keywords --complete-insert"
                      " --extended-insert --lock-tables --no-create-db --quick"
                      " '%5' > '%6' 2>/dev/null")
                      .arg(tempExtraConfFile, dbParams.m_dbHostName,
                           portArg,           dbParams.m_dbUserName,
                           dbParams.m_dbName, backupPathname);
 
    LOG(VB_FILE, LOG_INFO, QString("Backing up database with command: '%1'")
            .arg(command));
    LOG(VB_GENERAL, LOG_CRIT, QString("Backing up database to file: '%1'")
            .arg(backupPathname));
 
    uint status = myth_system(command, kMSDontBlockInputDevs|kMSAnonLog);
 
    QByteArray tmpfile = tempExtraConfFile.toLocal8Bit();
    unlink(tmpfile.constData());
 
    if (status != GENERIC_EXIT_OK)
    {
        LOG(VB_GENERAL, LOG_ERR, LOC +
            QString("Error backing up database: '%1' (%2)")
                .arg(command).arg(status));
        filename = "__FAILED__";
        return false;
    }
 
    if (compressCommand != "")
    {
        LOG(VB_GENERAL, LOG_CRIT, "Compressing database backup file.");
        compressCommand += " " + backupPathname;
        status = myth_system(compressCommand, kMSDontBlockInputDevs);
 
        if (status != GENERIC_EXIT_OK)
        {
            LOG(VB_GENERAL, LOG_CRIT,
                   "Compression failed, backup file will remain uncompressed.");
        }
        else
        {
            backupPathname += ".gz";
 
            LOG(VB_GENERAL, LOG_CRIT, QString("Database Backup filename: '%1'")
                    .arg(backupPathname));
        }
    }
 
    LOG(VB_GENERAL, LOG_CRIT, "Database Backup complete.");
 
    filename = backupPathname;
    return true;
}
 
bool DBUtil::QueryDBMSVersion(void)
{
    // Allow users to override the string provided by the database server in
    // case the value was changed to an unrecognizable string by whomever
    // compiled the MySQL server
    QString dbmsVersion = gCoreContext->GetSetting("DBMSVersionOverride");
 
    if (dbmsVersion.isEmpty())
    {
        MSqlQuery query(MSqlQuery::InitCon());
        query.prepare("SELECT VERSION();");
        if (!query.exec() || !query.next())
        {
            LOG(VB_GENERAL, LOG_ERR, LOC +
                "Unable to determine MySQL version.");
            MythDB::DBError("DBUtil Querying DBMS version", query);
            dbmsVersion.clear();
        }
        else
            dbmsVersion = query.value(0).toString();
    }
    m_versionString = dbmsVersion;
 
    return !m_versionString.isEmpty();
}
 
bool DBUtil::ParseDBMSVersion()
{
    if (m_versionString.isEmpty())
        if (!QueryDBMSVersion())
            return false;
 
    static const QRegularExpression parseVersion
        { R"(^(\d+)(?:\.(\d+)(?:\.(\d+))?)?)" };
    auto match = parseVersion.match(m_versionString);
    if (!match.hasMatch())
        return false;
 
    // If any of these wasn't matched, the captured string will be
    // empty and toInt will parse it as a zero.
    m_versionMajor = match.capturedView(1).toInt(nullptr);
    m_versionMinor = match.capturedView(2).toInt(nullptr);
    m_versionPoint = match.capturedView(3).toInt(nullptr);
 
    return m_versionMajor > -1;
}
 
int DBUtil::CountClients(void)
{
    int count = 0;
 
    MSqlQuery query(MSqlQuery::InitCon());
    if (!query.isConnected())
    {
        LOG(VB_GENERAL, LOG_DEBUG, "Not connected to DB");
        return count;
    }
 
    if (!query.exec("SHOW PROCESSLIST;"))
    {
        MythDB::DBError("DBUtil CountClients", query);
        return count;
    }
 
    QSqlRecord record = query.record();
    int db_index = record.indexOf("db");
    QString dbName = GetMythDB()->GetDatabaseName();
    QString inUseDB;
 
    while (query.next())
    {
        inUseDB = query.value(db_index).toString();
        if (inUseDB == dbName)
            ++count;
    }
 
    // On average, each myth program has 4 database connections,
    // but we round up just in case a new program is loading:
    count = (count + 3)/4;
 
    LOG(VB_GENERAL, LOG_DEBUG,
            QString("DBUtil::CountClients() found %1").arg(count));
 
    return count;
}
 
bool DBUtil::TryLockSchema(MSqlQuery &query, uint timeout_secs)
{
    query.prepare("SELECT GET_LOCK('schemaLock', :TIMEOUT)");
    query.bindValue(":TIMEOUT", timeout_secs);
    return query.exec() && query.first() && query.value(0).toBool();
}
 
void DBUtil::UnlockSchema(MSqlQuery &query)
{
    query.prepare("SELECT RELEASE_LOCK('schemaLock')");
    if (!query.exec())
    {
        MythDB::DBError("DBUtil UnlockSchema", query);
    }
}
 
bool DBUtil::CheckTimeZoneSupport(void)
{
    MSqlQuery query(MSqlQuery::InitCon());
    query.prepare("SELECT CONVERT_TZ(NOW(), 'SYSTEM', 'Etc/UTC')");
    if (!query.exec() || !query.next())
    {
        LOG(VB_GENERAL, LOG_ERR, "MySQL time zone support check failed");
        return false;
    }
 
    return !query.value(0).isNull();
}
 
bool DBUtil::CheckTableColumnExists(const QString &tableName, const QString &columnName)
{
    MSqlQuery query(MSqlQuery::InitCon());
    if (!query.isConnected())
        return false;
 
    QString sql = QString("SELECT COUNT(*) FROM information_schema.columns "
                          "WHERE table_schema = DATABASE() AND "
                                "table_name = '%1' AND column_name = '%2';")
                          .arg(tableName, columnName);
    LOG(VB_GENERAL, LOG_DEBUG,
            QString("DBUtil::CheckTableColumnExists() SQL: %1").arg(sql));
 
    if (!query.exec(sql))
    {
        MythDB::DBError("DBUtil Check Table Column Exists", query);
        return false;
    }
 
    bool result = false;
    if (query.next())
    {
        result = (query.value(0).toInt() > 0);
    }
    else
    {
        LOG(VB_GENERAL, LOG_ERR,
                            QString("DBUtil::CheckTableColumnExists() - Empty result set"));
    }
 
    LOG(VB_GENERAL, LOG_DEBUG,
            QString("DBUtil::CheckTableColumnExists('%1', '%2') result: %3").arg(tableName,
                    columnName, QVariant(result).toString()));
 
    return result;
}
 
/* vim: set expandtab tabstop=4 shiftwidth=4: */