UpgradeService.php

Go to the documentation of this file.

<?php

class Elgg_UpgradeService {

    public function run() {
        $result = array(
            'failure' => false,
            'reason' => '',
        );

        // prevent someone from running the upgrade script in parallel (see #4643)
        if (!$this->getUpgradeMutex()) {
            $result['failure'] = true;
            $result['reason'] = elgg_echo('upgrade:locked');
            return $result;
        }

        // disable the system log for upgrades to avoid exceptions when the schema changes.
        elgg_unregister_event_handler('log', 'systemlog', 'system_log_default_logger');
        elgg_unregister_event_handler('all', 'all', 'system_log_listener');

        // turn off time limit
        set_time_limit(0);

        if ($this->getUnprocessedUpgrades()) {
            $this->processUpgrades();
        }

        elgg_trigger_event('upgrade', 'system', null);
        elgg_invalidate_simplecache();
        elgg_reset_system_cache();

        $this->releaseUpgradeMutex();

        return $result;
    }

    protected function upgradeCode($version, $quiet = false) {
        $version = (int) $version;
        $upgrade_path = elgg_get_config('path') . 'engine/lib/upgrades/';
        $processed_upgrades = $this->getProcessedUpgrades();

        // upgrading from 1.7 to 1.8. Need to bootstrap.
        if (!$processed_upgrades) {
            $this->bootstrap17to18();

            // grab accurate processed upgrades
            $processed_upgrades = $this->getProcessedUpgrades();
        }

        $upgrade_files = $this->getUpgradeFiles($upgrade_path);

        if ($upgrade_files === false) {
            return false;
        }

        $upgrades = $this->getUnprocessedUpgrades($upgrade_files, $processed_upgrades);

        // Sort and execute
        sort($upgrades);

        foreach ($upgrades as $upgrade) {
            $upgrade_version = $this->getUpgradeFileVersion($upgrade);
            $success = true;

            // hide all errors.
            if ($quiet) {
                // hide include errors as well as any exceptions that might happen
                try {
                    if (!@self::includeCode("$upgrade_path/$upgrade")) {
                        $success = false;
                        error_log("Could not include $upgrade_path/$upgrade");
                    }
                } catch (Exception $e) {
                    $success = false;
                    error_log($e->getMessage());
                }
            } else {
                if (!self::includeCode("$upgrade_path/$upgrade")) {
                    $success = false;
                    error_log("Could not include $upgrade_path/$upgrade");
                }
            }

            if ($success) {
                // don't set the version to a lower number in instances where an upgrade
                // has been merged from a lower version of Elgg
                if ($upgrade_version > $version) {
                    datalist_set('version', $upgrade_version);
                }

                // incrementally set upgrade so we know where to start if something fails.
                $this->setProcessedUpgrade($upgrade);
            } else {
                return false;
            }
        }

        return true;
    }

    protected static function includeCode($file) {
        // do not remove - some upgrade scripts depend on this
        global $CONFIG;

        return include $file;
    }

    protected function setProcessedUpgrade($upgrade) {
        $processed_upgrades = $this->getProcessedUpgrades();
        $processed_upgrades[] = $upgrade;
        $processed_upgrades = array_unique($processed_upgrades);
        return datalist_set('processed_upgrades', serialize($processed_upgrades));
    }

    protected function getProcessedUpgrades() {
        $upgrades = datalist_get('processed_upgrades');
        $unserialized = unserialize($upgrades);
        return $unserialized;
    }

    protected function getUpgradeFileVersion($filename) {
        preg_match('/^([0-9]{10})([\.a-z0-9-_]+)?\.(php)$/i', $filename, $matches);

        if (isset($matches[1])) {
            return (int) $matches[1];
        }

        return false;
    }

    protected function getUpgradeFiles($upgrade_path = null) {
        if (!$upgrade_path) {
            $upgrade_path = elgg_get_config('path') . 'engine/lib/upgrades/';
        }
        $upgrade_path = sanitise_filepath($upgrade_path);
        $handle = opendir($upgrade_path);

        if (!$handle) {
            return false;
        }

        $upgrade_files = array();

        while ($upgrade_file = readdir($handle)) {
            // make sure this is a wellformed upgrade.
            if (is_dir($upgrade_path . '$upgrade_file')) {
                continue;
            }
            $upgrade_version = $this->getUpgradeFileVersion($upgrade_file);
            if (!$upgrade_version) {
                continue;
            }
            $upgrade_files[] = $upgrade_file;
        }

        sort($upgrade_files);

        return $upgrade_files;
    }

    protected function getUnprocessedUpgrades($upgrade_files = null, $processed_upgrades = null) {
        if ($upgrade_files === null) {
            $upgrade_files = $this->getUpgradeFiles();
        }

        if ($processed_upgrades === null) {
            $processed_upgrades = unserialize(datalist_get('processed_upgrades'));
            if (!is_array($processed_upgrades)) {
                $processed_upgrades = array();
            }
        }

        $unprocessed = array_diff($upgrade_files, $processed_upgrades);
        return $unprocessed;
    }

    protected function processUpgrades() {

        $dbversion = (int) datalist_get('version');

        // No version number? Oh snap...this is an upgrade from a clean installation < 1.7.
        // Run all upgrades without error reporting and hope for the best.
        // See https://github.com/elgg/elgg/issues/1432 for more.
        $quiet = !$dbversion;

        // Note: Database upgrades are deprecated as of 1.8.  Use code upgrades.  See #1433
        if ($this->dbUpgrade($dbversion, '', $quiet)) {
            system_message(elgg_echo('upgrade:db'));
        }

        if ($this->upgradeCode($dbversion, $quiet)) {
            system_message(elgg_echo('upgrade:core'));

            // Now we trigger an event to give the option for plugins to do something
            $upgrade_details = new stdClass;
            $upgrade_details->from = $dbversion;
            $upgrade_details->to = elgg_get_version();

            elgg_trigger_event('upgrade', 'upgrade', $upgrade_details);

            return true;
        }

        return false;
    }

    protected function bootstrap17to18() {
        $db_version = (int) datalist_get('version');

        // the 1.8 upgrades before the upgrade system change that are interspersed with 1.7 upgrades.
        $upgrades_18 = array(
            '2010111501.php',
            '2010121601.php',
            '2010121602.php',
            '2010121701.php',
            '2010123101.php',
            '2011010101.php',
        );

        $upgrade_files = $this->getUpgradeFiles();
        $processed_upgrades = array();

        foreach ($upgrade_files as $upgrade_file) {
            // ignore if not in 1.7 format or if it's a 1.8 upgrade
            if (in_array($upgrade_file, $upgrades_18) || !preg_match("/[0-9]{10}\.php/", $upgrade_file)) {
                continue;
            }

            $upgrade_version = $this->getUpgradeFileVersion($upgrade_file);

            // this has already been run in a previous 1.7.X -> 1.7.X upgrade
            if ($upgrade_version < $db_version) {
                $this->setProcessedUpgrade($upgrade_file);
            }
        }
    }

    protected function getUpgradeMutex() {
        global $CONFIG;

        if (!$this->isUpgradeLocked()) {
            // lock it
            insert_data("create table {$CONFIG->dbprefix}upgrade_lock (id INT)");
            elgg_log('Locked for upgrade.', 'NOTICE');
            return true;
        }

        elgg_log('Cannot lock for upgrade: already locked.', 'WARNING');
        return false;
    }

    public function releaseUpgradeMutex() {
        global $CONFIG;
        delete_data("drop table {$CONFIG->dbprefix}upgrade_lock");
        elgg_log('Upgrade unlocked.', 'NOTICE');
    }

    public function isUpgradeLocked() {
        global $CONFIG;

        $is_locked = count(get_data("SHOW TABLES LIKE '{$CONFIG->dbprefix}upgrade_lock'"));

        return (bool)$is_locked;
    }

    protected function dbUpgrade($version, $fromdir = "", $quiet = false) {
        global $CONFIG;

        $version = (int) $version;

        if (!$fromdir) {
            $fromdir = $CONFIG->path . 'engine/schema/upgrades/';
        }

        $i = 0;

        if ($handle = opendir($fromdir)) {
            $sqlupgrades = array();

            while ($sqlfile = readdir($handle)) {
                if (!is_dir($fromdir . $sqlfile)) {
                    if (preg_match('/^([0-9]{10})\.(sql)$/', $sqlfile, $matches)) {
                        $sql_version = (int) $matches[1];
                        if ($sql_version > $version) {
                            $sqlupgrades[] = $sqlfile;
                        }
                    }
                }
            }

            asort($sqlupgrades);

            if (sizeof($sqlupgrades) > 0) {
                foreach ($sqlupgrades as $sqlfile) {

                    // hide all errors.
                    if ($quiet) {
                        try {
                            run_sql_script($fromdir . $sqlfile);
                        } catch (DatabaseException $e) {
                            error_log($e->getmessage());
                        }
                    } else {
                        run_sql_script($fromdir . $sqlfile);
                    }
                    $i++;
                }
            }
        }

        return $i;
    }

}