Download.php

Go to the documentation of this file.

<?php
namespace TYPO3\CMS\Core\Http\Observer;

/*
 * This file is part of the TYPO3 CMS project.
 *
 * It is free software; you can redistribute it and/or modify it under
 * the terms of the GNU General Public License, either version 2
 * of the License, or any later version.
 *
 * For the full copyright and license information, please read the
 * LICENSE.txt file that was distributed with this source code.
 *
 * The TYPO3 project - inspiring people to share!
 */

class Download implements \SplObserver
{
    protected $filePointer = false;

    protected $targetFilePath = '';

    protected $targetDirectory = '';

    protected $targetFilename = '';

    public function __construct($directory, $filename = '')
    {
        $this->setDirectory($directory);
        $this->setFilename($filename);
    }

    public function update(\SplSubject $request)
    {
        $event = $request->getLastEvent();
        switch ($event['name']) {
            case 'receivedHeaders':
                if ($this->targetFilename === '') {
                    $this->determineFilename($request, $event['data']);
                }
                $this->openFile();
                break;
            case 'receivedBodyPart':
            case 'receivedEncodedBodyPart':
                fwrite($this->filePointer, $event['data']);
                break;
            case 'receivedBody':
                $this->closeFile();
                break;
            default:
                // Do nothing
        }
    }

    public function setDirectory($directory)
    {
        if (!is_dir($directory)) {
            throw new \InvalidArgumentException($directory . ' is not a directory', 1312223779);
        }
        if (!\TYPO3\CMS\Core\Utility\GeneralUtility::isAllowedAbsPath($directory)) {
            throw new \InvalidArgumentException($directory . ' is not within the PATH_site' . ' OR within the lockRootPath', 1328734617);
        }
        $this->targetDirectory = ($directory = rtrim($directory, DIRECTORY_SEPARATOR));
    }

    public function setFilename($filename = '')
    {
        $this->targetFilename = basename($filename);
    }

    protected function determineFilename(\HTTP_Request2 $request, \HTTP_Request2_Response $response)
    {
        $matches = [];
        $disposition = $response->getHeader('content-disposition');
        if ($disposition !== null && 0 === strpos($disposition, 'attachment') && 1 === preg_match('/filename="([^"]+)"/', $disposition, $matches)) {
            $filename = basename($matches[1]);
        } else {
            $filename = basename($request->getUrl()->getPath());
        }
        $this->setFilename($filename);
    }

    protected function openFile()
    {
        if ($this->targetFilename === '') {
            throw new \UnexpectedValueException('The file name must not be empty', 1321113658);
        }
        $this->targetFilePath = $this->targetDirectory . DIRECTORY_SEPARATOR . $this->targetFilename;
        $this->filePointer = @fopen($this->targetFilePath, 'wb');
        if ($this->filePointer === false) {
            throw new \TYPO3\CMS\Core\Exception('Cannot open target file ' . $this->targetFilePath, 1320833203);
        }
    }

    protected function closeFile()
    {
        fclose($this->filePointer);
        $this->filePointer = false;
        \TYPO3\CMS\Core\Utility\GeneralUtility::fixPermissions($this->targetFilePath);
    }
}