From 0398f1f363d5affba0633c104eaf22c1100f8e80 Mon Sep 17 00:00:00 2001
From: Alexander Sulfrian <alexander@sulfrian.net>
Date: Fri, 11 Dec 2009 03:05:45 +0100
Subject: added storage engine framework
---
paste/include/createpage.inc | 139 +++++++++++++++++++++++++++++++++----------
1 file changed, 108 insertions(+), 31 deletions(-)
diff --git a/paste/include/createpage.inc b/paste/include/createpage.inc
index 0e47b2e..f1a16aa 100644
--- a/paste/include/createpage.inc
+++ b/paste/include/createpage.inc
@@ -33,45 +33,122 @@ THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
-function CreatePage($input)
+abstract class StorageEngine
{
- list($text) = $input;
+ abstract public function setContent($content);
- $result_dir = $Config['results_dir'];
- $tmpfname = tempnam ("$result_dir/", "");
+ public function generateToken($name)
+ {
+ global $config;
- # I can't remember why I tacked on 2 random digits.
- $outbase = $tmpfname . rand(10, 99);
- $outfile = $outbase . ".html";
+ $ipAddr = $_SERVER["REMOTE_ADDR"];
+ return sha1($name . $ipAddr . $config['token_salt']);
+ }
- $ipAddr = $_SERVER["REMOTE_ADDR"];
- $pasteID = basename($outbase);
+ public function prepareContent($name, $baseUrl, $content)
+ {
+ $content = preg_replace("/=\{this-paste-url\}/",
+ $baseUrl . '/' . $name, $content);
- $token = sha1($pasteID . $ipAddr . $Config['token_salt']);
+ $content = preg_replace("/=\{remove-paste-url\}/", $baseUrl
+ . '/' . 'remove.php?id=' . $name
+ . '&t=' . $this->generateToken(),
+ $content);
- # There is probably a better way to do this. tempnam() creates the
- # temporary file in the results directory, but we then need to strip
- # this off the front of the filename to make a shorter url.
- $rawout = $outfile;
- $rawout = preg_replace("/$result_dir\//", "", $rawout);
-
- if ($Config['short_results_path']) {
- $urlbase = $Config['short_results_path'];
- } else {
- $urlbase = $Config['site_domain'] . $Config['site_path'] . '/' . $Config['results_dir'];
- }
- $pasteUrl = $urlbase . $rawout;
-
- $text = preg_replace("/=\{this-paste-url\}/", $pasteUrl, $text);
- $text = preg_replace("/=\{remove-paste-url\}/", $Config['site_domain'] . $Config['site_path'] . "remove.php?p=$pasteID&t=$token", $text);
-
- $fp = fopen($outfile, "w");
- fwrite($fp, $text);
- fclose($fp);
+ return $content;
+ }
+}
- unlink($tmpfname);
+class FileStorage extends StorageEngine
+{
+ private $storage_path;
+
+ function __construct($path) {
+ $this->storage_path = $path;
+ }
+
+ public function setContent($content)
+ {
+ global $config;
+
+ do {
+ $filename = sha1(date('r') . rand(1000));
+ } while (file_exists(realpath($storage_path . '/' , $filename)));
+
+ if ($config['short_results_path']) {
+ $urlbase = $config['short_results_path'];
+ } else {
+ $urlbase = $config['site_domain'] . $config['site_path'] . '/' . $config['results_dir'];
+ }
+ $content = $this->prepareContent($filename, $urlbase, $content);
+ $filename = $filename . ".html";
+
+ // write content
+ $outfile = realpath($storage_path . '/' , $filename);
+ $fp = fopen($outfile, "w");
+ fwrite($fp, $content);
+ fclose($fp);
+
+ return $pasteUrl;
+ }
+}
- return $pasteUrl;
+class MysqlStorage extends StorageEngine
+{
+ private $mysql;
+ private $table;
+
+ function __construct($host, $user, $password, $database, $table) {
+ $this->mysql = new mysqli($host, $user, $password, $database);
+ if ($this->mysql->connect_error) {
+ header('HTTP/1.0 503 Service Unavailable');
+ die('MySQL Connect Error ('
+ . $this->mysql->connect_errno . ') '
+ . $this->mysql->connect_error);
+ }
+
+ $this->table = $table;
+ if (! $this->mysql->query("CREATE TABLE IF NOT EXISTS $table (
+pid VARCHAR(40) PRIMARY KEY,
+content MEDIUMTEXT,
+FULLTEXT INDEX (content)
+);"))
+ {
+ header('HTTP/1.0 503 Service Unavailable');
+ die('MySQL Error (' . $this->mysql->errno . ') ' .
+ $this->mysql->error);
+ }
+ }
+
+ function __destruct()
+ {
+ if ($this->mysql)
+ $this->mysql->close();
+ }
+
+ public function setContent($content)
+ {
+ $name = sha1(date('r') . rand(1000));
+
+ if ($config['short_results_path']) {
+ $urlbase = $config['short_results_path'];
+ } else {
+ $urlbase = $config['site_domain'] . $config['site_path'] . '/' . $config['results_dir'];
+ }
+ $content = $this->prepareContent($name, $urlbase . 'get.php?id=' . $name, $content);
+
+ if ($stmt = $this->mysql->prepare(
+ 'INSERT INTO ? (pid, content) VALUES (?, ?)')) {
+
+ /* bind parameters for markers */
+ $stmt->bind_param("sb", $name, $content);
+
+ /* execute query */
+ $stmt->execute();
+
+ $stmt->close();
+ }
+ }
}
?>
--
cgit v1.2.3