aboutsummaryrefslogtreecommitdiffstats
path: root/lib/private/Archive/TAR.php
diff options
context:
space:
mode:
Diffstat (limited to 'lib/private/Archive/TAR.php')
-rw-r--r--lib/private/Archive/TAR.php338
1 files changed, 338 insertions, 0 deletions
diff --git a/lib/private/Archive/TAR.php b/lib/private/Archive/TAR.php
new file mode 100644
index 00000000000..e08720a75a5
--- /dev/null
+++ b/lib/private/Archive/TAR.php
@@ -0,0 +1,338 @@
+<?php
+
+/**
+ * SPDX-FileCopyrightText: 2016-2024 Nextcloud GmbH and Nextcloud contributors
+ * SPDX-FileCopyrightText: 2016 ownCloud, Inc.
+ * SPDX-License-Identifier: AGPL-3.0-only
+ */
+namespace OC\Archive;
+
+use Icewind\Streams\CallbackWrapper;
+
+class TAR extends Archive {
+ public const PLAIN = 0;
+ public const GZIP = 1;
+ public const BZIP = 2;
+
+ /**
+ * @var string[]|false
+ */
+ private $fileList = false;
+ /**
+ * @var array|false
+ */
+ private $cachedHeaders = false;
+
+ private \Archive_Tar $tar;
+
+ private string $path;
+
+ public function __construct(string $source) {
+ $types = [null, 'gz', 'bz2'];
+ $this->path = $source;
+ $this->tar = new \Archive_Tar($source, $types[self::getTarType($source)]);
+ }
+
+ /**
+ * try to detect the type of tar compression
+ */
+ public static function getTarType(string $file): int {
+ 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;
+ }
+ } else {
+ return self::PLAIN;
+ }
+ }
+
+ /**
+ * add an empty folder to the archive
+ */
+ public function addFolder(string $path): bool {
+ $tmpBase = \OC::$server->getTempManager()->getTemporaryFolder();
+ $path = rtrim($path, '/') . '/';
+ if ($this->fileExists($path)) {
+ return false;
+ }
+ $parts = explode('/', $path);
+ $folder = $tmpBase;
+ foreach ($parts as $part) {
+ $folder .= '/' . $part;
+ if (!is_dir($folder)) {
+ mkdir($folder);
+ }
+ }
+ $result = $this->tar->addModify([$tmpBase . $path], '', $tmpBase);
+ rmdir($tmpBase . $path);
+ $this->fileList = false;
+ $this->cachedHeaders = false;
+ return $result;
+ }
+
+ /**
+ * add a file to the archive
+ *
+ * @param string $source either a local file or string data
+ */
+ public function addFile(string $path, string $source = ''): bool {
+ if ($this->fileExists($path)) {
+ $this->remove($path);
+ }
+ if ($source and $source[0] == '/' and file_exists($source)) {
+ $source = file_get_contents($source);
+ }
+ $result = $this->tar->addString($path, $source);
+ $this->fileList = false;
+ $this->cachedHeaders = false;
+ return $result;
+ }
+
+ /**
+ * rename a file or folder in the archive
+ */
+ public function rename(string $source, string $dest): bool {
+ //no proper way to delete, rename entire archive, rename file and remake archive
+ $tmp = \OC::$server->getTempManager()->getTemporaryFolder();
+ $this->tar->extract($tmp);
+ rename($tmp . $source, $tmp . $dest);
+ $this->tar->_close();
+ unlink($this->path);
+ $types = [null, 'gz', 'bz'];
+ $this->tar = new \Archive_Tar($this->path, $types[self::getTarType($this->path)]);
+ $this->tar->createModify([$tmp], '', $tmp . '/');
+ $this->fileList = false;
+ $this->cachedHeaders = false;
+ return true;
+ }
+
+ private function getHeader(string $file): ?array {
+ if (!$this->cachedHeaders) {
+ $this->cachedHeaders = $this->tar->listContent();
+ }
+ 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;
+ }
+
+ /**
+ * get the uncompressed size of a file in the archive
+ */
+ public function filesize(string $path): false|int|float {
+ $stat = $this->getHeader($path);
+ return $stat['size'] ?? false;
+ }
+
+ /**
+ * get the last modified time of a file in the archive
+ *
+ * @return int|false
+ */
+ public function mtime(string $path) {
+ $stat = $this->getHeader($path);
+ return $stat['mtime'] ?? false;
+ }
+
+ /**
+ * get the files in a folder
+ */
+ public function getFolder(string $path): array {
+ $files = $this->getFiles();
+ $folderContent = [];
+ $pathLength = strlen($path);
+ foreach ($files as $file) {
+ if ($file[0] == '/') {
+ $file = substr($file, 1);
+ }
+ if (substr($file, 0, $pathLength) == $path and $file != $path) {
+ $result = substr($file, $pathLength);
+ if ($pos = strpos($result, '/')) {
+ $result = substr($result, 0, $pos + 1);
+ }
+ if (!in_array($result, $folderContent)) {
+ $folderContent[] = $result;
+ }
+ }
+ }
+ return $folderContent;
+ }
+
+ /**
+ * get all files in the archive
+ */
+ public function getFiles(): array {
+ if ($this->fileList !== false) {
+ return $this->fileList;
+ }
+ if ($this->cachedHeaders === false) {
+ $headers = $this->tar->listContent();
+ if (is_array($headers)) {
+ $this->cachedHeaders = $headers;
+ } else {
+ return [];
+ }
+ }
+ $files = [];
+ foreach ($this->cachedHeaders as $header) {
+ $files[] = $header['filename'];
+ }
+ $this->fileList = $files;
+ return $files;
+ }
+
+ /**
+ * get the content of a file
+ *
+ * @return string|false
+ */
+ public function getFile(string $path) {
+ $string = $this->tar->extractInString($path);
+ /** @var ?string $string */
+ if (is_string($string)) {
+ return $string;
+ } else {
+ return false;
+ }
+ }
+
+ /**
+ * extract a single file from the archive
+ */
+ public function extractFile(string $path, string $dest): bool {
+ $tmp = \OC::$server->getTempManager()->getTemporaryFolder();
+ if (!$this->fileExists($path)) {
+ return false;
+ }
+ if ($this->fileExists('/' . $path)) {
+ $success = $this->tar->extractList(['/' . $path], $tmp);
+ } else {
+ $success = $this->tar->extractList([$path], $tmp);
+ }
+ if ($success) {
+ rename($tmp . $path, $dest);
+ }
+ \OCP\Files::rmdirr($tmp);
+ return $success;
+ }
+
+ /**
+ * extract the archive
+ */
+ public function extract(string $dest): bool {
+ return $this->tar->extract($dest);
+ }
+
+ /**
+ * check if a file or folder exists in the archive
+ */
+ public function fileExists(string $path): bool {
+ $files = $this->getFiles();
+ if ((in_array($path, $files)) or (in_array($path . '/', $files))) {
+ return true;
+ } else {
+ $folderPath = rtrim($path, '/') . '/';
+ $pathLength = strlen($folderPath);
+ foreach ($files as $file) {
+ if (strlen($file) > $pathLength and substr($file, 0, $pathLength) == $folderPath) {
+ return true;
+ }
+ }
+ }
+ if ($path[0] != '/') { //not all programs agree on the use of a leading /
+ return $this->fileExists('/' . $path);
+ } else {
+ return false;
+ }
+ }
+
+ /**
+ * remove a file or folder from the archive
+ */
+ public function remove(string $path): bool {
+ if (!$this->fileExists($path)) {
+ return false;
+ }
+ $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);
+ unlink($this->path);
+ $this->reopen();
+ $this->tar->createModify([$tmp], '', $tmp);
+ return true;
+ }
+
+ /**
+ * get a file handler
+ *
+ * @return bool|resource
+ */
+ public function getStream(string $path, string $mode) {
+ $lastPoint = strrpos($path, '.');
+ if ($lastPoint !== false) {
+ $ext = substr($path, $lastPoint);
+ } else {
+ $ext = '';
+ }
+ $tmpFile = \OC::$server->getTempManager()->getTemporaryFile($ext);
+ if ($this->fileExists($path)) {
+ $this->extractFile($path, $tmpFile);
+ } elseif ($mode == 'r' or $mode == 'rb') {
+ return false;
+ }
+ if ($mode == 'r' or $mode == 'rb') {
+ return fopen($tmpFile, $mode);
+ } 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(string $tmpFile, string $path): void {
+ $this->addFile($path, $tmpFile);
+ unlink($tmpFile);
+ }
+
+ /**
+ * reopen the archive to ensure everything is written
+ */
+ private function reopen(): void {
+ $this->tar->_close();
+ $types = [null, 'gz', 'bz'];
+ $this->tar = new \Archive_Tar($this->path, $types[self::getTarType($this->path)]);
+ }
+
+ /**
+ * Get error object from archive_tar.
+ */
+ public function getError(): ?\PEAR_Error {
+ if ($this->tar->error_object instanceof \PEAR_Error) {
+ return $this->tar->error_object;
+ }
+ return null;
+ }
+}