Clean up TempManager to follow code guidelines
tmpBaseDir can be overridden for unit testing purposes
This commit is contained in:
parent
acae208f2f
commit
a0dfaf9be3
4 changed files with 91 additions and 43 deletions
|
@ -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(
|
||||
|
|
|
@ -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;
|
||||
}
|
||||
|
||||
}
|
||||
|
|
|
@ -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();
|
||||
}
|
||||
|
|
|
@ -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() {
|
||||
|
|
Loading…
Reference in a new issue