nextcloud/lib/private/Archive/TAR.php

385 lines
9.0 KiB
PHP
Raw Normal View History

<?php
/**
2016-07-21 18:07:57 +03:00
* @copyright Copyright (c) 2016, ownCloud, Inc.
*
2015-03-26 13:44:34 +03:00
* @author Bart Visscher <bartv@thisnet.nl>
* @author Christopher Schäpers <kondou@ts.unde.re>
* @author Christoph Wurst <christoph@winzerhof-wurst.at>
2016-07-21 18:07:57 +03:00
* @author Joas Schilling <coding@schilljs.com>
2015-03-26 13:44:34 +03:00
* @author Jörn Friedrich Dreyer <jfd@butonic.de>
* @author Morris Jobke <hey@morrisjobke.de>
* @author Remco Brenninkmeijer <requist1@starmail.nl>
2016-07-21 19:13:36 +03:00
* @author Robin Appelman <robin@icewind.nl>
2016-01-12 17:02:16 +03:00
* @author Robin McCorkell <robin@mccorkell.me.uk>
2016-07-21 18:07:57 +03:00
* @author Roeland Jago Douma <roeland@famdouma.nl>
2015-03-26 13:44:34 +03:00
* @author Thomas Müller <thomas.mueller@tmit.eu>
*
* @license AGPL-3.0
*
* This code is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License, version 3,
* as published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License, version 3,
* along with this program. If not, see <http://www.gnu.org/licenses/>
2015-03-26 13:44:34 +03:00
*
*/
namespace OC\Archive;
use Icewind\Streams\CallbackWrapper;
class TAR extends Archive {
public const PLAIN = 0;
public const GZIP = 1;
public const BZIP = 2;
2012-05-18 03:54:02 +04:00
private $fileList;
2012-10-05 14:29:27 +04:00
private $cachedHeaders;
2012-08-29 10:38:33 +04:00
/**
* @var \Archive_Tar tar
*/
2014-08-17 00:13:53 +04:00
private $tar = null;
private $path;
/**
* @param string $source
*/
public function __construct($source) {
$types = [null, 'gz', 'bz2'];
2014-08-17 00:13:53 +04:00
$this->path = $source;
$this->tar = new \Archive_Tar($source, $types[self::getTarType($source)]);
}
/**
* try to detect the type of tar compression
2014-08-17 00:13:53 +04:00
*
* @param string $file
* @return integer
*/
public static function getTarType($file) {
2014-08-17 00:13:53 +04:00
if (strpos($file, '.')) {
$extension = substr($file, strrpos($file, '.'));
switch ($extension) {
case '.gz':
case '.tgz':
return self::GZIP;
case '.bz':
case '.bz2':
return self::BZIP;
case '.tar':
return self::PLAIN;
default:
return self::PLAIN;
}
2014-08-17 00:13:53 +04:00
} else {
return self::PLAIN;
}
}
/**
* add an empty folder to the archive
2014-08-17 00:13:53 +04:00
*
* @param string $path
* @return bool
*/
public function addFolder($path) {
2015-12-18 13:19:53 +03:00
$tmpBase = \OC::$server->getTempManager()->getTemporaryFolder();
$path = rtrim($path, '/') . '/';
2014-08-17 00:13:53 +04:00
if ($this->fileExists($path)) {
return false;
}
2014-08-17 00:13:53 +04:00
$parts = explode('/', $path);
$folder = $tmpBase;
foreach ($parts as $part) {
$folder .= '/' . $part;
if (!is_dir($folder)) {
mkdir($folder);
}
}
$result = $this->tar->addModify([$tmpBase . $path], '', $tmpBase);
2014-08-17 00:13:53 +04:00
rmdir($tmpBase . $path);
$this->fileList = false;
$this->cachedHeaders = false;
return $result;
}
2014-08-17 00:13:53 +04:00
/**
* add a file to the archive
2014-08-17 00:13:53 +04:00
*
* @param string $path
* @param string $source either a local file or string data
* @return bool
*/
public function addFile($path, $source = '') {
2014-08-17 00:13:53 +04:00
if ($this->fileExists($path)) {
$this->remove($path);
}
2014-08-17 00:13:53 +04:00
if ($source and $source[0] == '/' and file_exists($source)) {
2014-08-17 00:19:01 +04:00
$source = file_get_contents($source);
}
2014-08-17 00:19:01 +04:00
$result = $this->tar->addString($path, $source);
2014-08-17 00:13:53 +04:00
$this->fileList = false;
$this->cachedHeaders = false;
return $result;
}
/**
* rename a file or folder in the archive
2014-08-17 00:13:53 +04:00
*
* @param string $source
* @param string $dest
* @return bool
*/
public function rename($source, $dest) {
//no proper way to delete, rename entire archive, rename file and remake archive
$tmp = \OC::$server->getTempManager()->getTemporaryFolder();
$this->tar->extract($tmp);
2014-08-17 00:13:53 +04:00
rename($tmp . $source, $tmp . $dest);
$this->tar = null;
unlink($this->path);
$types = [null, 'gz', 'bz'];
$this->tar = new \Archive_Tar($this->path, $types[self::getTarType($this->path)]);
$this->tar->createModify([$tmp], '', $tmp . '/');
2014-08-17 00:13:53 +04:00
$this->fileList = false;
$this->cachedHeaders = false;
2012-05-18 03:54:02 +04:00
return true;
}
/**
* @param string $file
*/
2012-09-07 17:22:01 +04:00
private function getHeader($file) {
2014-08-17 00:13:53 +04:00
if (!$this->cachedHeaders) {
2012-10-05 14:29:27 +04:00
$this->cachedHeaders = $this->tar->listContent();
}
2014-08-17 00:13:53 +04:00
foreach ($this->cachedHeaders as $header) {
if ($file == $header['filename']
or $file . '/' == $header['filename']
or '/' . $file . '/' == $header['filename']
or '/' . $file == $header['filename']
) {
return $header;
}
}
return null;
}
2012-08-29 10:38:33 +04:00
/**
* get the uncompressed size of a file in the archive
2014-08-17 00:13:53 +04:00
*
* @param string $path
* @return int
*/
public function filesize($path) {
2014-08-17 00:13:53 +04:00
$stat = $this->getHeader($path);
return $stat['size'];
}
2014-08-17 00:13:53 +04:00
/**
* get the last modified time of a file in the archive
2014-08-17 00:13:53 +04:00
*
* @param string $path
* @return int
*/
public function mtime($path) {
2014-08-17 00:13:53 +04:00
$stat = $this->getHeader($path);
return $stat['mtime'];
}
/**
* get the files in a folder
2014-08-17 00:13:53 +04:00
*
* @param string $path
* @return array
*/
public function getFolder($path) {
2014-08-17 00:13:53 +04:00
$files = $this->getFiles();
$folderContent = [];
2014-08-17 00:13:53 +04:00
$pathLength = strlen($path);
foreach ($files as $file) {
if ($file[0] == '/') {
$file = substr($file, 1);
2012-05-18 03:54:02 +04:00
}
2014-08-17 00:13:53 +04:00
if (substr($file, 0, $pathLength) == $path and $file != $path) {
$result = substr($file, $pathLength);
if ($pos = strpos($result, '/')) {
$result = substr($result, 0, $pos + 1);
2012-05-18 03:54:02 +04:00
}
2014-08-17 00:13:53 +04:00
if (array_search($result, $folderContent) === false) {
$folderContent[] = $result;
}
}
}
return $folderContent;
}
2014-08-17 00:13:53 +04:00
/**
2013-07-16 07:56:52 +04:00
* get all files in the archive
2014-08-17 00:13:53 +04:00
*
* @return array
*/
public function getFiles() {
2014-08-17 00:13:53 +04:00
if ($this->fileList) {
2012-05-18 03:54:02 +04:00
return $this->fileList;
}
2014-08-17 00:13:53 +04:00
if (!$this->cachedHeaders) {
2012-10-05 14:29:27 +04:00
$this->cachedHeaders = $this->tar->listContent();
}
$files = [];
2014-08-17 00:13:53 +04:00
foreach ($this->cachedHeaders as $header) {
$files[] = $header['filename'];
}
2014-08-17 00:13:53 +04:00
$this->fileList = $files;
return $files;
}
2014-08-17 00:13:53 +04:00
/**
* get the content of a file
2014-08-17 00:13:53 +04:00
*
* @param string $path
* @return string
*/
public function getFile($path) {
return $this->tar->extractInString($path);
}
2014-08-17 00:13:53 +04:00
/**
* extract a single file from the archive
2014-08-17 00:13:53 +04:00
*
* @param string $path
* @param string $dest
* @return bool
*/
public function extractFile($path, $dest) {
$tmp = \OC::$server->getTempManager()->getTemporaryFolder();
2014-08-17 00:13:53 +04:00
if (!$this->fileExists($path)) {
return false;
}
2014-08-17 00:13:53 +04:00
if ($this->fileExists('/' . $path)) {
$success = $this->tar->extractList(['/' . $path], $tmp);
2014-08-17 00:13:53 +04:00
} else {
$success = $this->tar->extractList([$path], $tmp);
2012-05-18 03:54:02 +04:00
}
2014-08-17 00:13:53 +04:00
if ($success) {
rename($tmp . $path, $dest);
}
\OCP\Files::rmdirr($tmp);
return $success;
}
2014-08-17 00:13:53 +04:00
/**
* extract the archive
2014-08-17 00:13:53 +04:00
*
* @param string $dest
* @return bool
*/
public function extract($dest) {
return $this->tar->extract($dest);
}
2014-08-17 00:13:53 +04:00
/**
* check if a file or folder exists in the archive
2014-08-17 00:13:53 +04:00
*
* @param string $path
* @return bool
*/
public function fileExists($path) {
2014-08-17 00:13:53 +04:00
$files = $this->getFiles();
if ((array_search($path, $files) !== false) or (array_search($path . '/', $files) !== false)) {
2012-05-18 03:54:02 +04:00
return true;
2014-08-17 00:13:53 +04:00
} else {
$folderPath = rtrim($path, '/') . '/';
2014-08-17 00:13:53 +04:00
$pathLength = strlen($folderPath);
foreach ($files as $file) {
if (strlen($file) > $pathLength and substr($file, 0, $pathLength) == $folderPath) {
2012-05-18 03:54:02 +04:00
return true;
}
}
}
2014-08-17 00:13:53 +04:00
if ($path[0] != '/') { //not all programs agree on the use of a leading /
return $this->fileExists('/' . $path);
} else {
2012-05-18 03:54:02 +04:00
return false;
}
}
2012-08-29 10:38:33 +04:00
/**
* remove a file or folder from the archive
2014-08-17 00:13:53 +04:00
*
* @param string $path
* @return bool
*/
public function remove($path) {
2014-08-17 00:13:53 +04:00
if (!$this->fileExists($path)) {
return false;
}
2014-08-17 00:13:53 +04:00
$this->fileList = false;
$this->cachedHeaders = false;
//no proper way to delete, extract entire archive, delete file and remake archive
$tmp = \OC::$server->getTempManager()->getTemporaryFolder();
$this->tar->extract($tmp);
\OCP\Files::rmdirr($tmp . $path);
2014-08-17 00:13:53 +04:00
$this->tar = null;
unlink($this->path);
$this->reopen();
$this->tar->createModify([$tmp], '', $tmp);
return true;
}
2014-08-17 00:13:53 +04:00
/**
* get a file handler
2014-08-17 00:13:53 +04:00
*
* @param string $path
* @param string $mode
* @return resource
*/
public function getStream($path, $mode) {
2014-08-17 00:13:53 +04:00
if (strrpos($path, '.') !== false) {
$ext = substr($path, strrpos($path, '.'));
} else {
$ext = '';
}
$tmpFile = \OC::$server->getTempManager()->getTemporaryFile($ext);
2014-08-17 00:13:53 +04:00
if ($this->fileExists($path)) {
2012-09-10 13:21:54 +04:00
$this->extractFile($path, $tmpFile);
2014-08-17 00:13:53 +04:00
} elseif ($mode == 'r' or $mode == 'rb') {
return false;
}
2014-08-17 00:13:53 +04:00
if ($mode == 'r' or $mode == 'rb') {
2012-09-10 13:21:54 +04:00
return fopen($tmpFile, $mode);
2014-08-17 00:13:53 +04:00
} else {
$handle = fopen($tmpFile, $mode);
return CallbackWrapper::wrap($handle, null, null, function () use ($path, $tmpFile) {
$this->writeBack($tmpFile, $path);
});
}
}
/**
* write back temporary files
*/
public function writeBack($tmpFile, $path) {
$this->addFile($path, $tmpFile);
unlink($tmpFile);
}
/**
* reopen the archive to ensure everything is written
*/
2012-09-07 17:22:01 +04:00
private function reopen() {
2014-08-17 00:13:53 +04:00
if ($this->tar) {
$this->tar->_close();
2014-08-17 00:13:53 +04:00
$this->tar = null;
}
$types = [null, 'gz', 'bz'];
$this->tar = new \Archive_Tar($this->path, $types[self::getTarType($this->path)]);
}
}