Add tarball support to multipastes

Signed-off-by: Florian Pritz <bluewind@xinu.at>

5 files changed, 254 insertions, 0 deletions

diff --git a/application/config/config.php b/application/config/config.php
index 2191b21df..2748e97c0 100644
--- a/application/config/config.php
+++ b/application/config/config.php
@@ -384,6 +384,13 @@ $config['actions_max_age'] = 60*60*24*5; // 5 days
 // Files smaller than this won't be deleted (even if they are old enough)
 $config['small_upload_size'] = 1024*10; // 10KiB
 
+// Maximum size for multipaste tarballs. 0 disables the feature
+$config['tarball_max_size'] = 1024*1024*50; // 50MiB
+
+// Multipaste tarballs older than this will be deleted by the cron job
+// Changing this is not recommended
+$config['tarball_cache_time'] = 60*5; // 5 minutes
+
 
 // Possible values:
 // - apc: needs the apc module and is only useful on long running php processes
diff --git a/application/controllers/file.php b/application/controllers/file.php
index 62cf342b1..292605ff0 100644
--- a/application/controllers/file.php
+++ b/application/controllers/file.php
@@ -108,6 +108,11 @@ class File extends MY_Controller {
 		case "info":
 			return $this->_display_info($id);
 
+		case "tar":
+			if ($is_multipaste) {
+				return $this->_tarball($id);
+			}
+
 		default:
 			if ($is_multipaste) {
 				show_error("Invalid action \"".htmlspecialchars($lexer)."\"");
@@ -370,6 +375,56 @@ class File extends MY_Controller {
 		}
 	}
 
+	private function _tarball($id)
+	{
+		if ($this->mmultipaste->id_exists($id)) {
+			$seen = array();
+			$path = $this->mmultipaste->get_tarball_path($id);
+			$archive = new \service\storage($path);
+
+			if (!$archive->exists()) {
+				$files = $this->mmultipaste->get_files($id);
+
+				$total_size = 0;
+				foreach ($files as $filedata) {
+					$total_size += $filedata["filesize"];
+				}
+
+				if ($total_size > $this->config->item("tarball_max_size")) {
+					show_error("Tarball too large, refusing to create.");
+				}
+
+				$tmpfile = $archive->begin();
+				// create empty tar archive so PharData has something to open
+				file_put_contents($tmpfile, str_repeat("\0", 1024*10));
+				$a = new PharData($tmpfile);
+
+				foreach ($files as $filedata) {
+					$filename = $filedata["filename"];
+					if (isset($seen[$filename]) && $seen[$filename]) {
+						$filename = $filedata["id"]."-".$filedata["filename"];
+					}
+					assert(!isset($seen[$filename]));
+					$a->addFile($this->mfile->file($filedata["hash"]), $filename);
+					$seen[$filename] = true;
+				}
+				$archive->gzip_compress();
+				$archive->commit();
+			}
+
+			// update mtime so the cronjob will keep the file for longer
+			$lock = fopen($archive->get_file(), "r+");
+			flock($lock, LOCK_SH);
+			touch($archive->get_file());
+			flock($lock, LOCK_UN);
+
+			assert(filesize($archive->get_file()) > 0);
+
+			$this->load->driver("ddownload");
+			$this->ddownload->serveFile($archive->get_file(), "$id.tar.gz", "application/x-gzip");
+		}
+	}
+
 	function _non_existent()
 	{
 		$this->data["title"] .= " - Not Found";
@@ -997,6 +1052,24 @@ class File extends MY_Controller {
 	{
 		if (!$this->input->is_cli_request()) return;
 
+		$tarball_dir = $this->config->item("upload_path")."/special/multipaste-tarballs";
+		if (is_dir($tarball_dir)) {
+			$tarball_cache_time = $this->config->item("tarball_cache_time");
+			$it = new RecursiveIteratorIterator(
+				new RecursiveDirectoryIterator($tarball_dir), RecursiveIteratorIterator::SELF_FIRST);
+
+			foreach ($it as $file) {
+				if ($file->isFile()) {
+					if ($file->getMTime() < time() - $tarball_cache_time) {
+						$lock = fopen($file, "r+");
+						flock($lock, LOCK_EX);
+						unlink($file);
+						flock($lock, LOCK_UN);
+					}
+				}
+			}
+		}
+
 		// 0 age disables age checks
 		if ($this->config->item('upload_max_age') == 0) return;
 
@@ -1075,6 +1148,9 @@ class File extends MY_Controller {
 			}
 		}
 		closedir($outer_dh);
+
+		// TODO: clean up special/multipaste-tarballs? cron() already expires
+		// after a rather short time, do we really need this here then?
 	}
 
 	function nuke_id()
diff --git a/application/models/mmultipaste.php b/application/models/mmultipaste.php
index 367e74787..29fc183b6 100644
--- a/application/models/mmultipaste.php
+++ b/application/models/mmultipaste.php
@@ -88,11 +88,20 @@ class Mmultipaste extends CI_Model {
 		return true;
 	}
 
+	public function get_tarball_path($id)
+	{
+		return $this->config->item("upload_path")."/special/multipaste-tarballs/".substr(md5($id), 0, 3)."/$id.tar.gz";
+	}
+
 	public function delete_id($id)
 	{
 		$this->db->where('url_id', $id)
 			->delete('multipaste');
 
+		$path = $this->get_tarball_path($id);
+		$f = new \service\storage($this->config->item("upload_path"), $path);
+		$f->unlink();
+
 		if ($this->id_exists($id))  {
 			return false;
 		}
diff --git a/application/service/storage.php b/application/service/storage.php
new file mode 100644
index 000000000..925827d46
--- /dev/null
+++ b/application/service/storage.php
@@ -0,0 +1,161 @@
+<?php
+/*
+ * Copyright 2014 Florian "Bluewind" Pritz <bluewind@server-speed.net>
+ *
+ * Licensed under AGPLv3
+ * (see COPYING for full license text)
+ *
+ */
+
+namespace service;
+
+/**
+ * This class allows to change a temporary file and replace the original one atomically
+ */
+class storage {
+	private $path;
+	private $tempfile = NULL;
+
+	public function __construct($path)
+	{
+		assert(!is_dir($path));
+
+		$this->path = $path;
+	}
+
+	/**
+	 * Create a temp file which can be written to.
+	 *
+	 * Call commit() once you are done writing.
+	 * Call rollback() to remove the file and throw away any data written.
+	 *
+	 * Calling this multiple times will automatically rollback previous calls.
+	 *
+	 * @return temp file path
+	 */
+	public function begin()
+	{
+		if($this->tempfile !== NULL) {
+			$this->rollback();
+		}
+
+		$this->tempfile = $this->create_tempfile();
+
+		return $this->tempfile;
+	}
+
+	/**
+	 * Create a temporary file. You'll need to remove it yourself when no longer needed.
+	 *
+	 * @return path to the temporary file
+	 */
+	private function create_tempfile()
+	{
+		$dir = dirname($this->get_file());
+		$prefix = basename($this->get_file());
+
+		if (!is_dir($dir)) {
+			mkdir($dir, 0777, true);
+		}
+		assert(is_dir($dir));
+
+		return tempnam($dir, $prefix);
+	}
+
+	/**
+	 * Save the temporary file returned by begin() to the permanent path
+	 * (supplied to the constructor) in an atomic fashion.
+	 */
+	public function commit()
+	{
+		$ret = rename($this->tempfile, $this->get_file());
+		if ($ret) {
+			$this->tempfile = NULL;
+		}
+
+		return $ret;
+	}
+
+	public function exists()
+	{
+		return file_exists($this->get_file());
+	}
+
+	public function get_file()
+	{
+		return $this->path;
+	}
+
+	/**
+	 * Throw away any changes made to the temporary file returned by begin()
+	 */
+	public function rollback()
+	{
+		if ($this->tempfile !== NULL) {
+			unlink($this->tempfile);
+			$this->tempfile = NULL;
+		}
+	}
+
+	public function __destruct()
+	{
+		$this->rollback();
+	}
+
+	/**
+	 * GZIPs the temp file
+	 *
+	 * From http://stackoverflow.com/questions/6073397/how-do-you-create-a-gz-file-using-php
+	 * Based on function by Kioob at:
+	 * http://www.php.net/manual/en/function.gzwrite.php#34955
+	 *
+	 * @param string $source Path to file that should be compressed
+	 * @param integer $level GZIP compression level (default: 6)
+	 * @return boolean true if operation succeeds, false on error
+	 */
+	public function gzip_compress($level = 6){
+		if ($this->tempfile === NULL) {
+			return;
+		}
+
+		$source = $this->tempfile;
+		$file = new storage($source);
+		$dest = $file->begin();
+		$mode = 'wb' . $level;
+		$error = false;
+		$chunk_size = 1024*512;
+
+		if ($fp_out = gzopen($dest, $mode)) {
+			if ($fp_in = fopen($source,'rb')) {
+				while (!feof($fp_in)) {
+					gzwrite($fp_out, fread($fp_in, $chunk_size));
+				}
+				fclose($fp_in);
+			} else {
+				$error = true;
+			}
+			gzclose($fp_out);
+		} else {
+			$error = true;
+		}
+
+		if ($error) {
+			return false;
+		} else {
+			$file->commit();
+			return true;
+		}
+	}
+
+	/**
+	 * Delete the file if it exists.
+	 */
+	public function unlink()
+	{
+		if ($this->exists()) {
+			unlink($this->get_file());
+		}
+	}
+}
+
+# vim: set noet:
diff --git a/application/views/file/upload_form.php b/application/views/file/upload_form.php
index 5051f689b..4434a53cf 100644
--- a/application/views/file/upload_form.php
+++ b/application/views/file/upload_form.php
@@ -97,6 +97,7 @@
 			<dt>/&lt;ID&gt;/</dt><dd>automatically display everything in a sensible way</dd>
 			<dt>/&lt;ID&gt;/qr</dt><dd>display a qr code containing a link to <span class="example">/&lt;ID&gt;/</span></dd>
 			<dt>/&lt;ID&gt;/info</dt><dd>display some information about the multipaste</dd>
+			<dt>/&lt;ID&gt;/tar</dt><dd>download a tarball of all files in the multipaste (files may be renamed to avoid conflicts)</dd>
 		</dl>
 	</div>
 	<div class="col-lg-6">