diff options
-rw-r--r-- | lib/private/server.php | 5 | ||||
-rw-r--r-- | lib/private/tempmanager.php | 104 | ||||
-rw-r--r-- | lib/public/itempmanager.php | 8 | ||||
-rw-r--r-- | tests/lib/tempmanager.php | 17 |
4 files changed, 91 insertions, 43 deletions
diff --git a/lib/private/server.php b/lib/private/server.php index 229fae843cc..518b0cb2d92 100644 --- a/lib/private/server.php +++ b/lib/private/server.php @@ -344,7 +344,10 @@ class Server extends SimpleContainer implements IServerContainer { } }); $this->registerService('TempManager', function (Server $c) { - return new TempManager($c->getLogger()); + return new TempManager( + $c->getLogger(), + $c->getConfig() + ); }); $this->registerService('AppManager', function(Server $c) { return new \OC\App\AppManager( diff --git a/lib/private/tempmanager.php b/lib/private/tempmanager.php index b1b98751607..cc7c51d8568 100644 --- a/lib/private/tempmanager.php +++ b/lib/private/tempmanager.php @@ -25,6 +25,7 @@ namespace OC; use OCP\ILogger; +use OCP\IConfig; use OCP\ITempManager; class TempManager implements ITempManager { @@ -34,15 +35,20 @@ class TempManager implements ITempManager { protected $tmpBaseDir; /** @var ILogger */ protected $log; + /** @var IConfig */ + protected $config; + /** Prefix */ const TMP_PREFIX = 'oc_tmp_'; /** * @param \OCP\ILogger $logger + * @param \OCP\IConfig $config */ - public function __construct(ILogger $logger) { - $this->tmpBaseDir = $this->t_get_temp_dir(); + public function __construct(ILogger $logger, IConfig $config) { $this->log = $logger; + $this->config = $config; + $this->tmpBaseDir = $this->getTempBaseDir(); } /** @@ -191,57 +197,77 @@ class TempManager implements ITempManager { } /** - * Get the temporary directory to store transfer data - * @return null|string Path to the temporary directory or null + * Get the temporary base directory configured on the server + * + * @return string Path to the temporary directory or null + * @throws \UnexpectedValueException */ - public function t_get_temp_dir() { - // Get the temporary directory and log the path if loglevel is set to debug - // Info: based on the temp dir, further directories may be created unique to the instance - $temp = self::gather_temp_dir(); - \OCP\Util::writeLog('Core', 'Temporary directory set to: ' . ($temp ? $temp : 'NULL'), \OCP\Util::DEBUG); - return $temp; - } + public function getTempBaseDir() { + if ($this->tmpBaseDir) { + return $this->tmpBaseDir; + } - /** - * Get a temporary directory from possible sources - * If a temporary directory is set in config.php, use this one - * @return null|string Path to the temporary directory or null - */ - private function gather_temp_dir() { - if ($temp = self::get_config_temp_dir()) return $temp; - if ($temp = ini_get('upload_tmp_dir')) return $temp; - if ($temp = getenv('TMP')) return $temp; - if ($temp = getenv('TEMP')) return $temp; - if ($temp = getenv('TMPDIR')) return $temp; + $directories = []; + if ($temp = $this->config->getSystemValue('tempdirectory', null)) { + $directories[] = $temp; + } + if ($temp = ini_get('upload_tmp_dir')) { + $directories[] = $temp; + } + if ($temp = getenv('TMP')) { + $directories[] = $temp; + } + if ($temp = getenv('TEMP')) { + $directories[] = $temp; + } + if ($temp = getenv('TMPDIR')) { + $directories[] = $temp; + } $temp = tempnam(__FILE__, ''); if (file_exists($temp)) { unlink($temp); - return dirname($temp); + $directories[] = dirname($temp); } - if ($temp = sys_get_temp_dir()) return $temp; - return null; + if ($temp = sys_get_temp_dir()) { + $directories[] = $temp; + } + + foreach ($directories as $dir) { + if ($this->checkTemporaryDirectory($dir)) { + return $dir; + } + } + throw new \UnexpectedValueException('Unable to detect system temporary directory'); } /** - * Check if the temporary directory is defined in config.php and is present and writable - * @return bool|string Path to the temporary directory or false + * Check if a temporary directory is ready for use + * + * @param mixed $directory + * @return bool */ - private function get_config_temp_dir() { - $temp = \OC::$server->getConfig()->getSystemValue('tempdirectory', false); + private function checkTemporaryDirectory($directory) { // surpress any possible errors caused by is_writable // checks missing or invalid path or characters, wrong permissions ect - if ($temp) { - try { - if (is_writeable($temp)) { - return $temp; - } else { - \OCP\Util::writeLog('Core', 'Manually set temporary directory in config.php is not present or writable: ' . $temp, \OCP\Util::WARN); - return false; - } - } catch (Exception $e) { - return false; + try { + if (is_writeable($directory)) { + return true; } + } catch (Exception $e) { } + $this->log->warning('Temporary directory {dir} is not present or writable', + ['dir' => $directory] + ); + return false; + } + + /** + * Override the temporary base directory + * + * @param string $directory + */ + public function overrideTempBaseDir($directory) { + $this->tmpBaseDir = $directory; } } diff --git a/lib/public/itempmanager.php b/lib/public/itempmanager.php index 7ba5b1e7bff..6364fa71913 100644 --- a/lib/public/itempmanager.php +++ b/lib/public/itempmanager.php @@ -58,4 +58,12 @@ interface ITempManager { * @since 8.0.0 */ public function cleanOld(); + + /** + * Get the temporary base directory + * + * @return string + * @since 8.2.0 + */ + public function getTempBaseDir(); } diff --git a/tests/lib/tempmanager.php b/tests/lib/tempmanager.php index 89625ac62f2..04e14c335bd 100644 --- a/tests/lib/tempmanager.php +++ b/tests/lib/tempmanager.php @@ -23,10 +23,12 @@ class NullLogger extends Log { class TempManager extends \Test\TestCase { + protected $baseDir = null; + protected function setUp() { parent::setUp(); - $this->baseDir = $this->getManager()->t_get_temp_dir() . $this->getUniqueID('/oc_tmp_test'); + $this->baseDir = $this->getManager()->getTempBaseDir() . $this->getUniqueID('/oc_tmp_test'); if (!is_dir($this->baseDir)) { mkdir($this->baseDir); } @@ -34,18 +36,27 @@ class TempManager extends \Test\TestCase { protected function tearDown() { \OC_Helper::rmdirr($this->baseDir); + $this->baseDir = null; parent::tearDown(); } /** * @param \OCP\ILogger $logger + * @param \OCP\IConfig $config * @return \OC\TempManager */ - protected function getManager($logger = null) { + protected function getManager($logger = null, $config = null) { if (!$logger) { $logger = new NullLogger(); } - return new \OC\TempManager($logger); + if (!$config) { + $config = \OC::$server->getConfig(); + } + $manager = new \OC\TempManager($logger, $config); + if ($this->baseDir) { + $manager->overrideTempBaseDir($this->baseDir); + } + return $manager; } public function testGetFile() { |