374 lines
11 KiB
PHP
374 lines
11 KiB
PHP
<?php
|
|
/**
|
|
* PrivateBin
|
|
*
|
|
* a zero-knowledge paste bin
|
|
*
|
|
* @link https://github.com/PrivateBin/PrivateBin
|
|
* @copyright 2012 Sébastien SAUVAGE (sebsauvage.net)
|
|
* @license https://www.opensource.org/licenses/zlib-license.php The zlib/libpng License
|
|
* @version 0.22
|
|
*/
|
|
|
|
namespace PrivateBin\data;
|
|
|
|
/**
|
|
* privatebin_data
|
|
*
|
|
* Model for data access, implemented as a singleton.
|
|
*/
|
|
class data extends AbstractData
|
|
{
|
|
/**
|
|
* directory where data is stored
|
|
*
|
|
* @access private
|
|
* @static
|
|
* @var string
|
|
*/
|
|
private static $_dir = 'data/';
|
|
|
|
/**
|
|
* get instance of singleton
|
|
*
|
|
* @access public
|
|
* @static
|
|
* @param array $options
|
|
* @return privatebin_data
|
|
*/
|
|
public static function getInstance($options = null)
|
|
{
|
|
// if given update the data directory
|
|
if (
|
|
is_array($options) &&
|
|
array_key_exists('dir', $options)
|
|
) {
|
|
self::$_dir = $options['dir'] . DIRECTORY_SEPARATOR;
|
|
}
|
|
// if needed initialize the singleton
|
|
if (!(self::$_instance instanceof privatebin_data)) {
|
|
self::$_instance = new self;
|
|
self::_init();
|
|
}
|
|
return self::$_instance;
|
|
}
|
|
|
|
/**
|
|
* Create a paste.
|
|
*
|
|
* @access public
|
|
* @param string $pasteid
|
|
* @param array $paste
|
|
* @return bool
|
|
*/
|
|
public function create($pasteid, $paste)
|
|
{
|
|
$storagedir = self::_dataid2path($pasteid);
|
|
if (is_file($storagedir . $pasteid)) {
|
|
return false;
|
|
}
|
|
if (!is_dir($storagedir)) {
|
|
mkdir($storagedir, 0705, true);
|
|
}
|
|
return (bool) @file_put_contents($storagedir . $pasteid, json_encode($paste));
|
|
}
|
|
|
|
/**
|
|
* Read a paste.
|
|
*
|
|
* @access public
|
|
* @param string $pasteid
|
|
* @return stdClass|false
|
|
*/
|
|
public function read($pasteid)
|
|
{
|
|
if (!$this->exists($pasteid)) {
|
|
return false;
|
|
}
|
|
$paste = json_decode(
|
|
file_get_contents(self::_dataid2path($pasteid) . $pasteid)
|
|
);
|
|
if (property_exists($paste->meta, 'attachment')) {
|
|
$paste->attachment = $paste->meta->attachment;
|
|
unset($paste->meta->attachment);
|
|
if (property_exists($paste->meta, 'attachmentname')) {
|
|
$paste->attachmentname = $paste->meta->attachmentname;
|
|
unset($paste->meta->attachmentname);
|
|
}
|
|
}
|
|
return $paste;
|
|
}
|
|
|
|
/**
|
|
* Delete a paste and its discussion.
|
|
*
|
|
* @access public
|
|
* @param string $pasteid
|
|
* @return void
|
|
*/
|
|
public function delete($pasteid)
|
|
{
|
|
// Delete the paste itself.
|
|
@unlink(self::_dataid2path($pasteid) . $pasteid);
|
|
|
|
// Delete discussion if it exists.
|
|
$discdir = self::_dataid2discussionpath($pasteid);
|
|
if (is_dir($discdir)) {
|
|
// Delete all files in discussion directory
|
|
$dir = dir($discdir);
|
|
while (false !== ($filename = $dir->read())) {
|
|
if (is_file($discdir . $filename)) {
|
|
@unlink($discdir . $filename);
|
|
}
|
|
}
|
|
$dir->close();
|
|
|
|
// Delete the discussion directory.
|
|
@rmdir($discdir);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Test if a paste exists.
|
|
*
|
|
* @access public
|
|
* @param string $pasteid
|
|
* @return void
|
|
*/
|
|
public function exists($pasteid)
|
|
{
|
|
return is_file(self::_dataid2path($pasteid) . $pasteid);
|
|
}
|
|
|
|
/**
|
|
* Create a comment in a paste.
|
|
*
|
|
* @access public
|
|
* @param string $pasteid
|
|
* @param string $parentid
|
|
* @param string $commentid
|
|
* @param array $comment
|
|
* @return bool
|
|
*/
|
|
public function createComment($pasteid, $parentid, $commentid, $comment)
|
|
{
|
|
$storagedir = self::_dataid2discussionpath($pasteid);
|
|
$filename = $pasteid . '.' . $commentid . '.' . $parentid;
|
|
if (is_file($storagedir . $filename)) {
|
|
return false;
|
|
}
|
|
if (!is_dir($storagedir)) {
|
|
mkdir($storagedir, 0705, true);
|
|
}
|
|
return (bool) @file_put_contents($storagedir . $filename, json_encode($comment));
|
|
}
|
|
|
|
/**
|
|
* Read all comments of paste.
|
|
*
|
|
* @access public
|
|
* @param string $pasteid
|
|
* @return array
|
|
*/
|
|
public function readComments($pasteid)
|
|
{
|
|
$comments = array();
|
|
$discdir = self::_dataid2discussionpath($pasteid);
|
|
if (is_dir($discdir)) {
|
|
// Delete all files in discussion directory
|
|
$dir = dir($discdir);
|
|
while (false !== ($filename = $dir->read())) {
|
|
// Filename is in the form pasteid.commentid.parentid:
|
|
// - pasteid is the paste this reply belongs to.
|
|
// - commentid is the comment identifier itself.
|
|
// - parentid is the comment this comment replies to (It can be pasteid)
|
|
if (is_file($discdir . $filename)) {
|
|
$comment = json_decode(file_get_contents($discdir . $filename));
|
|
$items = explode('.', $filename);
|
|
// Add some meta information not contained in file.
|
|
$comment->id = $items[1];
|
|
$comment->parentid = $items[2];
|
|
|
|
// Store in array
|
|
$key = $this->getOpenSlot($comments, (int) $comment->meta->postdate);
|
|
$comments[$key] = $comment;
|
|
}
|
|
}
|
|
$dir->close();
|
|
|
|
// Sort comments by date, oldest first.
|
|
ksort($comments);
|
|
}
|
|
return $comments;
|
|
}
|
|
|
|
/**
|
|
* Test if a comment exists.
|
|
*
|
|
* @access public
|
|
* @param string $pasteid
|
|
* @param string $parentid
|
|
* @param string $commentid
|
|
* @return void
|
|
*/
|
|
public function existsComment($pasteid, $parentid, $commentid)
|
|
{
|
|
return is_file(
|
|
self::_dataid2discussionpath($pasteid) .
|
|
$pasteid . '.' . $commentid . '.' . $parentid
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Returns up to batch size number of paste ids that have expired
|
|
*
|
|
* @access private
|
|
* @param int $batchsize
|
|
* @return array
|
|
*/
|
|
protected function _getExpiredPastes($batchsize)
|
|
{
|
|
$pastes = array();
|
|
$firstLevel = array_filter(
|
|
scandir(self::$_dir),
|
|
array('self', '_isFirstLevelDir')
|
|
);
|
|
if (count($firstLevel) > 0) {
|
|
// try at most 10 times the $batchsize pastes before giving up
|
|
for ($i = 0, $max = $batchsize * 10; $i < $max; ++$i) {
|
|
$firstKey = array_rand($firstLevel);
|
|
$secondLevel = array_filter(
|
|
scandir(self::$_dir . $firstLevel[$firstKey]),
|
|
array('self', '_isSecondLevelDir')
|
|
);
|
|
|
|
// skip this folder in the next checks if it is empty
|
|
if (count($secondLevel) == 0) {
|
|
unset($firstLevel[$firstKey]);
|
|
continue;
|
|
}
|
|
|
|
$secondKey = array_rand($secondLevel);
|
|
$path = self::$_dir . $firstLevel[$firstKey] .
|
|
DIRECTORY_SEPARATOR . $secondLevel[$secondKey];
|
|
if (!is_dir($path)) {
|
|
continue;
|
|
}
|
|
$thirdLevel = array_filter(
|
|
scandir($path),
|
|
array('PrivateBin\\model\\paste', 'isValidId')
|
|
);
|
|
if (count($thirdLevel) == 0) {
|
|
continue;
|
|
}
|
|
$thirdKey = array_rand($thirdLevel);
|
|
$pasteid = $thirdLevel[$thirdKey];
|
|
if (in_array($pasteid, $pastes)) {
|
|
continue;
|
|
}
|
|
|
|
if ($this->exists($pasteid)) {
|
|
$data = $this->read($pasteid);
|
|
if (
|
|
property_exists($data->meta, 'expire_date') &&
|
|
$data->meta->expire_date < time()
|
|
) {
|
|
$pastes[] = $pasteid;
|
|
if (count($pastes) >= $batchsize) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return $pastes;
|
|
}
|
|
|
|
/**
|
|
* initialize privatebin
|
|
*
|
|
* @access private
|
|
* @static
|
|
* @return void
|
|
*/
|
|
private static function _init()
|
|
{
|
|
// Create storage directory if it does not exist.
|
|
if (!is_dir(self::$_dir)) {
|
|
mkdir(self::$_dir, 0705);
|
|
}
|
|
// Create .htaccess file if it does not exist.
|
|
if (!is_file(self::$_dir . '.htaccess')) {
|
|
file_put_contents(
|
|
self::$_dir . '.htaccess',
|
|
'Allow from none' . PHP_EOL .
|
|
'Deny from all' . PHP_EOL
|
|
);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Convert paste id to storage path.
|
|
*
|
|
* The idea is to creates subdirectories in order to limit the number of files per directory.
|
|
* (A high number of files in a single directory can slow things down.)
|
|
* eg. "f468483c313401e8" will be stored in "data/f4/68/f468483c313401e8"
|
|
* High-trafic websites may want to deepen the directory structure (like Squid does).
|
|
*
|
|
* eg. input 'e3570978f9e4aa90' --> output 'data/e3/57/'
|
|
*
|
|
* @access private
|
|
* @static
|
|
* @param string $dataid
|
|
* @return void
|
|
*/
|
|
private static function _dataid2path($dataid)
|
|
{
|
|
return self::$_dir . substr($dataid, 0, 2) . DIRECTORY_SEPARATOR .
|
|
substr($dataid, 2, 2) . DIRECTORY_SEPARATOR;
|
|
}
|
|
|
|
/**
|
|
* Convert paste id to discussion storage path.
|
|
*
|
|
* eg. input 'e3570978f9e4aa90' --> output 'data/e3/57/e3570978f9e4aa90.discussion/'
|
|
*
|
|
* @access private
|
|
* @static
|
|
* @param string $dataid
|
|
* @return void
|
|
*/
|
|
private static function _dataid2discussionpath($dataid)
|
|
{
|
|
return self::_dataid2path($dataid) . $dataid .
|
|
'.discussion' . DIRECTORY_SEPARATOR;
|
|
}
|
|
|
|
/**
|
|
* Check that the given element is a valid first level directory.
|
|
*
|
|
* @access private
|
|
* @static
|
|
* @param string $element
|
|
* @return bool
|
|
*/
|
|
private static function _isFirstLevelDir($element)
|
|
{
|
|
return self::_isSecondLevelDir($element) &&
|
|
is_dir(self::$_dir . DIRECTORY_SEPARATOR . $element);
|
|
}
|
|
|
|
/**
|
|
* Check that the given element is a valid second level directory.
|
|
*
|
|
* @access private
|
|
* @static
|
|
* @param string $element
|
|
* @return bool
|
|
*/
|
|
private static function _isSecondLevelDir($element)
|
|
{
|
|
return (bool) preg_match('/^[a-f0-9]{2}$/', $element);
|
|
}
|
|
}
|