149 lines
3.0 KiB
PHP
149 lines
3.0 KiB
PHP
<?php
|
|
|
|
/**
|
|
* Copyright (c) 2014 Robin Appelman <icewind@owncloud.com>
|
|
* This file is licensed under the Affero General Public License version 3 or
|
|
* later.
|
|
* See the COPYING-README file.
|
|
*/
|
|
|
|
namespace OC;
|
|
|
|
use OCP\ILogger;
|
|
use OCP\ITempManager;
|
|
|
|
class TempManager implements ITempManager {
|
|
/**
|
|
* Current temporary files and folders
|
|
*
|
|
* @var string[]
|
|
*/
|
|
protected $current = array();
|
|
|
|
/**
|
|
* i.e. /tmp on linux systems
|
|
*
|
|
* @var string
|
|
*/
|
|
protected $tmpBaseDir;
|
|
|
|
/**
|
|
* @var \OCP\ILogger
|
|
*/
|
|
protected $log;
|
|
|
|
/**
|
|
* @param string $baseDir
|
|
* @param \OCP\ILogger $logger
|
|
*/
|
|
public function __construct($baseDir, ILogger $logger) {
|
|
$this->tmpBaseDir = $baseDir;
|
|
$this->log = $logger;
|
|
}
|
|
|
|
protected function generatePath($postFix) {
|
|
return $this->tmpBaseDir . '/oc_tmp_' . md5(time() . rand()) . $postFix;
|
|
}
|
|
|
|
/**
|
|
* Create a temporary file and return the path
|
|
*
|
|
* @param string $postFix
|
|
* @return string
|
|
*/
|
|
public function getTemporaryFile($postFix = '') {
|
|
$file = $this->generatePath($postFix);
|
|
if (is_writable($this->tmpBaseDir)) {
|
|
touch($file);
|
|
$this->current[] = $file;
|
|
return $file;
|
|
} else {
|
|
$this->log->warning(
|
|
'Can not create a temporary file in directory {dir}. Check it exists and has correct permissions',
|
|
array(
|
|
'dir' => $this->tmpBaseDir
|
|
)
|
|
);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Create a temporary folder and return the path
|
|
*
|
|
* @param string $postFix
|
|
* @return string
|
|
*/
|
|
public function getTemporaryFolder($postFix = '') {
|
|
$path = $this->generatePath($postFix);
|
|
if (is_writable($this->tmpBaseDir)) {
|
|
mkdir($path);
|
|
$this->current[] = $path;
|
|
return $path . '/';
|
|
} else {
|
|
$this->log->warning(
|
|
'Can not create a temporary folder in directory {dir}. Check it exists and has correct permissions',
|
|
array(
|
|
'dir' => $this->tmpBaseDir
|
|
)
|
|
);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Remove the temporary files and folders generated during this request
|
|
*/
|
|
public function clean() {
|
|
$this->cleanFiles($this->current);
|
|
}
|
|
|
|
protected function cleanFiles($files) {
|
|
foreach ($files as $file) {
|
|
if (file_exists($file)) {
|
|
try {
|
|
\OC_Helper::rmdirr($file);
|
|
} catch (\UnexpectedValueException $ex) {
|
|
$this->log->warning(
|
|
"Error deleting temporary file/folder: {file} - Reason: {error}",
|
|
array(
|
|
'file' => $file,
|
|
'error' => $ex->getMessage()
|
|
)
|
|
);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Remove old temporary files and folders that were failed to be cleaned
|
|
*/
|
|
public function cleanOld() {
|
|
$this->cleanFiles($this->getOldFiles());
|
|
}
|
|
|
|
/**
|
|
* Get all temporary files and folders generated by oc older than an hour
|
|
*
|
|
* @return string[]
|
|
*/
|
|
protected function getOldFiles() {
|
|
$cutOfTime = time() - 3600;
|
|
$files = array();
|
|
$dh = opendir($this->tmpBaseDir);
|
|
if ($dh) {
|
|
while (($file = readdir($dh)) !== false) {
|
|
if (substr($file, 0, 7) === 'oc_tmp_') {
|
|
$path = $this->tmpBaseDir . '/' . $file;
|
|
$mtime = filemtime($path);
|
|
if ($mtime < $cutOfTime) {
|
|
$files[] = $path;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return $files;
|
|
}
|
|
}
|