One Hat Cyber Team

Edit File: Reader.php

<?php
/**
 * Copyright 2017 Horde LLC (http://www.horde.org/)
 *
 * See the enclosed file LICENSE for license information (BSD). If you
 * did not receive this file, see http://www.horde.org/licenses/bsd.
 *
 * @author   Jan Schneider <jan@horde.org>
 * @category Horde
 * @license  http://www.horde.org/licenses/bsd BSD
 * @package  Backup
 */

namespace Horde\Backup;

use ArrayIterator;
use Horde_Compress_Tar as Tar;
use Horde_Compress_Zip as Zip;
use Horde_Pack_Driver_Json as Json;
use Horde\Backup\Exception;
use Horde\Backup\Reader\CompressIterator\Tar as TarIterator;
use Horde\Backup\Reader\CompressIterator\Zip as ZipIterator;
use Horde\Backup\Translation;

/**
 * The backup reader class that reads backups from the backup directory.
 *
 * @author    Jan Schneider <jan@horde.org>
 * @category  Horde
 * @copyright 2017 Horde LLC
 * @license   http://www.horde.org/licenses/bsd BSD
 * @package   Backup
 */
class Reader
{
    /**
     * Backup directory.
     *
     * @var string
     */
    protected $_dir;

/**
     * Constructor.
     *
     * @param string $directory  Backup directory.
     */
    public function __construct($directory)
    {
        if (!Json::supported()) {
            throw new Exception(Translation::t("JSON serializer not available"));
        }

$this->_dir = $directory;
        if (!strlen($this->_dir)) {
            throw new Exception(Translation::t("Empty directory name"));
        }
        if (!is_dir($this->_dir)) {
            throw new Exception(
                sprintf(Translation::t("%s is not a directory"), $this->_dir)
            );
        }
        if (!is_readable($this->_dir) || !is_writable($this->_dir)) {
            throw new Exception(
                sprintf(Translation::t("Access denied to %s"), $this->_dir)
            );
        }
    }

/**
     * Returns the available user backups.
     *
     * @return Iterator  A list of user backups.
     */
    public function listBackups()
    {
        return new \RecursiveIteratorIterator(
            new \RecursiveDirectoryIterator(
                $this->_dir,
                \FilesystemIterator::CURRENT_AS_PATHNAME | \FilesystemIterator::SKIP_DOTS
            )
        );
    }

/**
     * Returns user data from backups.
     *
     * @param array $users         A list of users to restore. Defaults to all
     *                             backups.
     * @param array $applications  A list of applications to restore. Defaults
     *                             to all backups.
     *
     * @return \Horde\Backup\Collection[]  All restored object collections.
     */
    public function restore(
        array $applications = array(), $users = array()
    )
    {
        if ($users) {
            $backups = new ArrayIterator($this->_getBackupFiles($users));
        } else {
            $backups = $this->listBackups();
        }

$data = array();
        foreach ($backups as $file) {
            switch (substr($file, -4)) {
            case '.zip':
                $data = array_merge_recursive(
                    $data,
                    $this->_restoreFromZip($file, $applications)
                );
                break;
            case '.tar':
                $data = array_merge_recursive(
                    $data,
                    $this->_restoreFromTar($file, $applications)
                );
                break;
            }
        }

return $data;
    }

/**
     * Restores user data from a ZIP file.
     *
     * @param string $file         Pathname to a ZIP backup file.
     * @param array $applications  A list of applications to restore. Defaults
     *                             to all backups.
     *
     * @return \Horde\Backup\Collection[]  All restored object collections.
     */
    protected function _restoreFromZip($file, $applications)
    {
        $user = basename($file, '.zip');
        $contents = file_get_contents($file);
        $compress = new Zip();
        $files = $compress->decompress(
            $contents, array('action' => Zip::ZIP_LIST)
        );

return $this->_buildCollections(
            $files,
            $applications,
            $contents,
            $user,
            function ($application, $resource, $files, $contents)
            {
                return new ZipIterator(
                    $application, $resource, $files, $contents
                );
            }
        );
    }

/**
     * Restores user data from a TAR file.
     *
     * @param string $file         Pathname to a TAR backup file.
     * @param array $applications  A list of applications to restore. Defaults
     *                             to all backups.
     *
     * @return \Horde\Backup\Collection[]  All restored object collections.
     */
    protected function _restoreFromTar($file, $applications)
    {
        $user = basename($file, '.tar');
        $contents = file_get_contents($file);
        $compress = new Tar();
        $files = $compress->decompress($contents);

return $this->_buildCollections(
            $files,
            $applications,
            $contents,
            $user,
            function ($application, $resource, $files, $contents)
            {
                return new TarIterator($application, $resource, $files);
            }
        );
    }

/**
     * Builds a list of object collections from any Horde_Compress backend.
     *
     * @param array $files         Archive info from Horde_Compress.
     * @param array $applications  A list of applications to restore. Defaults
     *                             to all backups.
     * @param string $contents     The archive file contents.
     * @param string $user         A user name.
     * @param callable $factory    A factory for iterators that are passed to
     *                             \Horde\Backup\Collection.
     *
     * @return \Horde\Backup\Collection[]  All restored object collections.
     */
    protected function _buildCollections(
        $files, $applications, $contents, $user, $factory
    )
    {
        $data = array();
        foreach ($files as $key => $info) {
            $path = explode('/', $info['name']);
            if (!$applications || in_array($path[0], $applications)) {
                $data[$path[0]][$path[1]] = true;
            }
        }

$collections = array();
        foreach ($data as $application => $resources) {
            $collections[$application] = array();
            foreach (array_keys($resources) as $resource) {
                $collections[$application][] = new Collection(
                    $factory($application, $resource, $files, $contents),
                    $user,
                    $resource
                );
            }
        }

return $collections;
    }

/**
     * Builds a backup iterator for individual users.
     *
     * @param array $users  A list of users.
     *
     * @return array  A list of backup files.
     */
    protected function _getBackupFiles(array $users)
    {
        $files = array();
        foreach ($users as $user) {
            $file = $this->_dir . '/' . $user;
            if (file_exists($file . '.zip')) {
                $files[] = $file . '.zip';
            } elseif (file_exists($file . '.tar')) {
                $files[] = $file . '.tar';
            }
        }
        return $files;
    }
}