nextcloud/lib/archive/tar.php

340 lines
7.6 KiB
PHP
Raw Normal View History

<?php
/**
* Copyright (c) 2012 Robin Appelman <icewind@owncloud.com>
* This file is licensed under the Affero General Public License version 3 or
* later.
* See the COPYING-README file.
*/
2012-10-31 03:04:43 +04:00
require_once 'Archive/Tar.php';
class OC_Archive_TAR extends OC_Archive{
const PLAIN=0;
const GZIP=1;
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
*/
private $tar=null;
private $path;
2012-09-07 17:22:01 +04:00
function __construct($source) {
2012-11-04 14:10:46 +04:00
$types=array(null, 'gz', 'bz');
$this->path=$source;
2012-09-10 13:21:54 +04:00
$this->tar=new Archive_Tar($source, $types[self::getTarType($source)]);
}
/**
* try to detect the type of tar compression
* @param string file
* @return str
*/
2012-09-07 17:22:01 +04:00
static public function getTarType($file) {
2012-09-10 13:21:54 +04:00
if(strpos($file, '.')) {
$extension=substr($file, strrpos($file, '.'));
2012-09-07 17:22:01 +04:00
switch($extension) {
case 'gz':
case 'tgz':
return self::GZIP;
case 'bz':
case 'bz2':
return self::BZIP;
default:
return self::PLAIN;
}
}else{
return self::PLAIN;
}
}
/**
* add an empty folder to the archive
* @param string path
* @return bool
*/
2012-09-07 17:22:01 +04:00
function addFolder($path) {
$tmpBase=OC_Helper::tmpFolder();
2012-09-10 13:21:54 +04:00
if(substr($path, -1, 1)!='/') {
$path.='/';
}
2012-09-07 17:22:01 +04:00
if($this->fileExists($path)) {
return false;
}
2012-09-10 13:21:54 +04:00
$parts=explode('/', $path);
$folder=$tmpBase;
2012-09-07 17:22:01 +04:00
foreach($parts as $part) {
$folder.='/'.$part;
2012-09-07 17:22:01 +04:00
if(!is_dir($folder)) {
mkdir($folder);
}
}
2012-09-10 13:21:54 +04:00
$result=$this->tar->addModify(array($tmpBase.$path), '', $tmpBase);
rmdir($tmpBase.$path);
2012-05-18 03:54:02 +04:00
$this->fileList=false;
2012-10-05 14:29:27 +04:00
$this->cachedHeaders=false;
return $result;
}
/**
* add a file to the archive
* @param string path
* @param string source either a local file or string data
* @return bool
*/
2012-11-02 22:53:02 +04:00
function addFile($path, $source='') {
2012-09-07 17:22:01 +04:00
if($this->fileExists($path)) {
$this->remove($path);
}
2012-09-07 17:22:01 +04:00
if($source and $source[0]=='/' and file_exists($source)) {
$header=array();
$dummy='';
$this->tar->_openAppend();
2012-09-10 13:21:54 +04:00
$result=$this->tar->_addfile($source, $header, $dummy, $dummy, $path);
}else{
2012-09-10 13:21:54 +04:00
$result=$this->tar->addString($path, $source);
}
2012-05-18 03:54:02 +04:00
$this->fileList=false;
2012-10-05 14:29:27 +04:00
$this->cachedHeaders=false;
return $result;
}
/**
* rename a file or folder in the archive
* @param string source
* @param string dest
* @return bool
*/
2012-11-02 22:53:02 +04:00
function rename($source, $dest) {
//no proper way to delete, rename entire archive, rename file and remake archive
$tmp=OCP\Files::tmpFolder();
$this->tar->extract($tmp);
2012-09-10 13:21:54 +04:00
rename($tmp.$source, $tmp.$dest);
$this->tar=null;
unlink($this->path);
2012-09-10 13:21:54 +04:00
$types=array(null, 'gz', 'bz');
$this->tar=new Archive_Tar($this->path, $types[self::getTarType($this->path)]);
$this->tar->createModify(array($tmp), '', $tmp.'/');
2012-05-18 03:54:02 +04:00
$this->fileList=false;
2012-10-05 14:29:27 +04:00
$this->cachedHeaders=false;
2012-05-18 03:54:02 +04:00
return true;
}
2012-09-07 17:22:01 +04:00
private function getHeader($file) {
2012-10-05 14:29:27 +04:00
if ( ! $this->cachedHeaders ) {
$this->cachedHeaders = $this->tar->listContent();
}
foreach($this->cachedHeaders as $header) {
2012-10-05 14:39:43 +04:00
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
* @param string path
* @return int
*/
2012-09-07 17:22:01 +04:00
function filesize($path) {
$stat=$this->getHeader($path);
return $stat['size'];
}
/**
* get the last modified time of a file in the archive
* @param string path
* @return int
*/
2012-09-07 17:22:01 +04:00
function mtime($path) {
$stat=$this->getHeader($path);
return $stat['mtime'];
}
/**
* get the files in a folder
* @param path
* @return array
*/
2012-09-07 17:22:01 +04:00
function getFolder($path) {
$files=$this->getFiles();
$folderContent=array();
$pathLength=strlen($path);
2012-09-07 17:22:01 +04:00
foreach($files as $file) {
if($file[0]=='/') {
2012-09-10 13:21:54 +04:00
$file=substr($file, 1);
2012-05-18 03:54:02 +04:00
}
2012-09-10 13:21:54 +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
}
2012-09-10 13:21:54 +04:00
if(array_search($result, $folderContent)===false) {
2012-05-18 03:54:02 +04:00
$folderContent[]=$result;
}
}
}
return $folderContent;
}
/**
*get all files in the archive
* @return array
*/
2012-09-07 17:22:01 +04:00
function getFiles() {
if($this->fileList) {
2012-05-18 03:54:02 +04:00
return $this->fileList;
}
2012-10-05 14:29:27 +04:00
if ( ! $this->cachedHeaders ) {
$this->cachedHeaders = $this->tar->listContent();
}
$files=array();
2012-10-05 14:29:27 +04:00
foreach($this->cachedHeaders as $header) {
$files[]=$header['filename'];
}
2012-05-18 03:54:02 +04:00
$this->fileList=$files;
return $files;
}
/**
* get the content of a file
* @param string path
* @return string
*/
2012-09-07 17:22:01 +04:00
function getFile($path) {
return $this->tar->extractInString($path);
}
/**
* extract a single file from the archive
* @param string path
* @param string dest
* @return bool
*/
2012-11-02 22:53:02 +04:00
function extractFile($path, $dest) {
$tmp=OCP\Files::tmpFolder();
2012-09-07 17:22:01 +04:00
if(!$this->fileExists($path)) {
return false;
}
2012-09-07 17:22:01 +04:00
if($this->fileExists('/'.$path)) {
2012-09-10 13:21:54 +04:00
$success=$this->tar->extractList(array('/'.$path), $tmp);
2012-05-18 03:54:02 +04:00
}else{
2012-09-10 13:21:54 +04:00
$success=$this->tar->extractList(array($path), $tmp);
2012-05-18 03:54:02 +04:00
}
2012-09-07 17:22:01 +04:00
if($success) {
2012-09-10 13:21:54 +04:00
rename($tmp.$path, $dest);
}
OCP\Files::rmdirr($tmp);
return $success;
}
/**
* extract the archive
* @param string path
* @param string dest
* @return bool
*/
2012-09-07 17:22:01 +04:00
function extract($dest) {
return $this->tar->extract($dest);
}
/**
* check if a file or folder exists in the archive
* @param string path
* @return bool
*/
2012-09-07 17:22:01 +04:00
function fileExists($path) {
2012-05-18 03:54:02 +04:00
$files=$this->getFiles();
2012-09-10 13:21:54 +04:00
if((array_search($path, $files)!==false) or (array_search($path.'/', $files)!==false)) {
2012-05-18 03:54:02 +04:00
return true;
}else{
$folderPath=$path;
2012-09-10 13:21:54 +04:00
if(substr($folderPath, -1, 1)!='/') {
2012-05-18 03:54:02 +04:00
$folderPath.='/';
}
$pathLength=strlen($folderPath);
2012-09-07 17:22:01 +04:00
foreach($files as $file) {
2012-09-10 13:21:54 +04:00
if(strlen($file)>$pathLength and substr($file, 0, $pathLength)==$folderPath) {
2012-05-18 03:54:02 +04:00
return true;
}
}
}
2012-09-07 17:22:01 +04:00
if($path[0]!='/') {//not all programs agree on the use of a leading /
2012-05-18 03:54:02 +04:00
return $this->fileExists('/'.$path);
}else{
return false;
}
}
2012-08-29 10:38:33 +04:00
/**
* remove a file or folder from the archive
* @param string path
* @return bool
*/
2012-09-07 17:22:01 +04:00
function remove($path) {
if(!$this->fileExists($path)) {
return false;
}
2012-05-18 03:54:02 +04:00
$this->fileList=false;
2012-10-05 14:29:27 +04:00
$this->cachedHeaders=false;
//no proper way to delete, extract entire archive, delete file and remake archive
$tmp=OCP\Files::tmpFolder();
$this->tar->extract($tmp);
OCP\Files::rmdirr($tmp.$path);
$this->tar=null;
unlink($this->path);
$this->reopen();
2012-09-10 13:21:54 +04:00
$this->tar->createModify(array($tmp), '', $tmp);
return true;
}
/**
* get a file handler
* @param string path
* @param string mode
* @return resource
*/
2012-11-02 22:53:02 +04:00
function getStream($path, $mode) {
2012-09-10 13:21:54 +04:00
if(strrpos($path, '.')!==false) {
$ext=substr($path, strrpos($path, '.'));
}else{
$ext='';
}
$tmpFile=OCP\Files::tmpFile($ext);
2012-09-07 17:22:01 +04:00
if($this->fileExists($path)) {
2012-09-10 13:21:54 +04:00
$this->extractFile($path, $tmpFile);
2012-09-07 17:22:01 +04:00
}elseif($mode=='r' or $mode=='rb') {
return false;
}
2012-09-07 17:22:01 +04:00
if($mode=='r' or $mode=='rb') {
2012-09-10 13:21:54 +04:00
return fopen($tmpFile, $mode);
}else{
2012-11-04 14:10:46 +04:00
OC_CloseStreamWrapper::$callBacks[$tmpFile]=array($this, 'writeBack');
self::$tempFiles[$tmpFile]=$path;
2012-09-10 13:21:54 +04:00
return fopen('close://'.$tmpFile, $mode);
}
}
private static $tempFiles=array();
/**
* write back temporary files
*/
2012-09-07 17:22:01 +04:00
function writeBack($tmpFile) {
if(isset(self::$tempFiles[$tmpFile])) {
2012-09-10 13:21:54 +04:00
$this->addFile(self::$tempFiles[$tmpFile], $tmpFile);
unlink($tmpFile);
}
}
/**
* reopen the archive to ensure everything is written
*/
2012-09-07 17:22:01 +04:00
private function reopen() {
if($this->tar) {
$this->tar->_close();
$this->tar=null;
}
2012-11-04 14:10:46 +04:00
$types=array(null, 'gz', 'bz');
2012-09-10 13:21:54 +04:00
$this->tar=new Archive_Tar($this->path, $types[self::getTarType($this->path)]);
}
}