remove file locking - code will continue to live in it's own app

This commit is contained in:
Thomas Müller 2014-06-04 20:11:54 +02:00
parent 7a20d22daa
commit ec7225da66
4 changed files with 0 additions and 564 deletions

View File

@ -1,309 +0,0 @@
<?php
/**
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU AFFERO GENERAL PUBLIC LICENSE
* License as published by the Free Software Foundation; either
* version 3 of the License, or any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU AFFERO GENERAL PUBLIC LICENSE for more details.
*
* You should have received a copy of the GNU Affero General Public
* License along with this library. If not, see <http://www.gnu.org/licenses/>.
*
*/
namespace OC\Files;
use OCP\Config;
use OC\Files\Filesystem;
use OCP\Files\LockNotAcquiredException;
/**
* Class Lock
* @package OC\Files
*/
class Lock implements \OCP\Files\Lock {
/** @var int $retries Number of lock retries to attempt */
public static $retries = 40;
/** @var int $retryInterval Milliseconds between retries */
public static $retryInterval = 50;
/** @var string $path Filename of the file as represented in storage */
protected $path;
/** @var array $stack A stack of lock data */
protected $stack = array();
/** @var resource $handle A file handle used to maintain a lock */
protected $handle;
/** @var string $lockFile Filename of the lock file */
protected $lockFile;
/** @var resource $lockFileHandle The file handle used to maintain a lock on the lock file */
protected $lockFileHandle;
/**
* Constructor for the lock instance
* @param string $path Absolute pathname for a local file on which to obtain a lock
*/
public function __construct($path) {
$this->path = Filesystem::normalizePath($path);
}
protected function obtainReadLock($existingHandle = null) {
\OC_Log::write('lock', sprintf('INFO: Read lock requested for %s', $this->path), \OC_Log::DEBUG);
$timeout = Lock::$retries;
// Re-use an existing handle or get a new one
if(empty($existingHandle)) {
$handle = fopen($this->path, 'r');
}
else {
$handle = $existingHandle;
}
do {
if($this->isLockFileLocked($this->getLockFile($this->path))) {
\OC_Log::write('lock', sprintf('INFO: Read lock has locked lock file %s for %s', $this->getLockFile($this->path), $this->path), \OC_Log::DEBUG);
do {
usleep(Lock::$retryInterval);
$timeout--;
} while($this->isLockFileLocked($this->getLockFile($this->path)) && $timeout > 0);
\OC_Log::write('lock', sprintf('INFO: Lock file %s has become unlocked for %s', $this->getLockFile($this->path), $this->path), \OC_Log::DEBUG);
}
} while ((!$lockReturn = flock($handle, LOCK_SH | LOCK_NB, $wouldBlock)) && $timeout > 0);
if ($wouldBlock == true || $lockReturn == false || $timeout <= 0) {
\OC_Log::write('lock', sprintf('FAIL: Failed to acquire read lock for %s', $this->path), \OC_Log::DEBUG);
return false;
}
$this->handle = $handle;
\OC_Log::write('lock', sprintf('PASS: Acquired read lock for %s', $this->path), \OC_Log::DEBUG);
return true;
}
protected function obtainWriteLock($existingHandle = null) {
\OC_Log::write('lock', sprintf('INFO: Write lock requested for %s', $this->path), \OC_Log::DEBUG);
// Re-use an existing handle or get a new one
if (empty($existingHandle)) {
$handle = fopen($this->path, 'c');
}
else {
$handle = $existingHandle;
}
// If the file doesn't exist, but we can create a lock for it
if (!file_exists($this->path) && $this->lockLockFile($this->path)) {
$lockReturn = flock($handle, LOCK_EX | LOCK_NB, $wouldBlock);
if ($lockReturn == false || $wouldBlock == true) {
\OC_Log::write('lock', sprintf('FAIL: Write lock failed, unable to exclusively lock new file %s', $this->path), \OC_Log::DEBUG);
return false;
}
$this->handle = $handle;
return true;
}
// Since this file does exist, wait for locks to release to get an exclusive lock
$timeout = Lock::$retries;
$haveBlock = false;
while ((!$lockReturn = flock($handle, LOCK_EX | LOCK_NB, $wouldBlock)) && $timeout > 0) {
// We don't have a lock on the original file, try to get a lock on its lock file
if ($haveBlock || $haveBlock = $this->lockLockFile($this->lockFile)) {
usleep(Lock::$retryInterval);
}
else {
\OC_Log::write('lock', sprintf('FAIL: Write lock failed, unable to lock original %s or lock file', $this->path), \OC_Log::DEBUG);
return false;
}
$timeout--;
}
if ($wouldBlock == true || $lockReturn == false) {
\OC_Log::write('lock', sprintf('FAIL: Write lock failed due to timeout on %s', $this->path), \OC_Log::DEBUG);
return false;
}
\OC_Log::write('lock', sprintf('PASS: Write lock succeeded on %s', $this->path), \OC_Log::DEBUG);
return true;
}
/**
* Create a lock file and lock it
* Sets $this->lockFile to the specified lock file, indicating that the lock file is IN USE for this lock instance
* Also sets $this->lockFileHandle to a file handle of the lock file
* @param string $filename The name of the file to lock
* @return bool False if lock can't be acquired, true if it can.
*/
protected function lockLockFile ( $filename ) {
$lockFile = $this->getLockFile($filename);
\OC_Log::write('lock', sprintf('INFO: Locking lock file %s for %s', $lockFile, $filename), \OC_Log::DEBUG);
// If we already manage the lockfile, success
if(!empty($this->lockFile)) {
\OC_Log::write('lock', sprintf('PASS: Lock file %s was locked by this request for %s', $lockFile, $filename), \OC_Log::DEBUG);
return true;
}
// Check if the lockfile exists, and if not, try to create it
\OC_Log::write('lock', sprintf('INFO: Does lock file %s already exist? %s', $lockFile, file_exists($lockFile) ? 'yes' : 'no'), \OC_Log::DEBUG);
$handle = fopen($lockFile, 'c');
if(!$handle) {
\OC_Log::write('lock', sprintf('FAIL: Could not create lock file %s', $lockFile), \OC_Log::DEBUG);
return false;
}
// Attempt to acquire lock on lock file
$wouldBlock = false;
$timeout = self::$retries;
// Wait for lock over timeout
while((!$lockReturn = flock($handle, LOCK_EX | LOCK_NB, $wouldBlock)) && $timeout > 0) {
\OC_Log::write('lock', sprintf('FAIL: Could not acquire lock on lock file %s, %s timeout increments remain.', $lockFile, $timeout), \OC_Log::DEBUG);
usleep(self::$retryInterval);
$timeout--;
}
if ($wouldBlock == true || $lockReturn == false) {
\OC_Log::write('lock', sprintf('FAIL: Could not acquire lock on lock file %s', $lockFile), \OC_Log::DEBUG);
return false;
}
fwrite($handle, $filename);
\OC_Log::write('lock', sprintf('PASS: Wrote filename to lock lock file %s', $lockFile), \OC_Log::DEBUG);
$this->lockFile = $lockFile;
$this->lockFileHandle = $handle;
return true;
}
/**
* Add a lock of a specific type to the stack
* @param integer $lockType A constant representing the type of lock to queue
* @param null|resource $existingHandle An existing file handle from an fopen()
* @throws LockNotAcquiredException
*/
public function addLock($lockType, $existingHandle = null) {
if(!isset($this->stack[$lockType])) {
switch($lockType) {
case Lock::READ:
$result = $this->obtainReadLock($existingHandle);
break;
case Lock::WRITE:
$result = $this->obtainWriteLock($existingHandle);
break;
default:
$result = false;
break;
}
if($result) {
$this->stack[$lockType] = 0;
}
else {
throw new LockNotAcquiredException($this->path, $lockType);
}
}
\OC_Log::write('lock', sprintf('INFO: Incrementing lock type %d count for %s', $lockType, $this->path), \OC_Log::DEBUG);
$this->stack[$lockType]++;
}
/**
* Release locks on handles and files
*/
public function release($lockType) {
if(isset($this->stack[$lockType])) {
$this->stack[$lockType]--;
if($this->stack[$lockType] <= 0) {
unset($this->stack[$lockType]);
}
}
if(count($this->stack) == 0) {
// No more locks needed on this file, release the handle and/or lockfile
$this->releaseAll();
}
return true;
}
/**
* Get the lock file associated to a file
* @param string $filename The filename of the file to create a lock file for
* @return string The filename of the lock file
*/
public static function getLockFile($filename) {
static $locksDir = false;
if(!$locksDir) {
$dataDir = Config::getSystemValue('datadirectory');
$locksDir = $dataDir . '/.locks';
if(!file_exists($locksDir)) {
mkdir($locksDir);
}
}
$filename = Filesystem::normalizePath($filename);
return $locksDir . '/' . sha1($filename) . '.lock';
}
/**
* Determine if a file has an associated and flocked lock file
* @param string $lockFile The filename of the lock file to check
* @return bool True if the lock file is flocked
*/
protected function isLockFileLocked($lockFile) {
if(file_exists($lockFile)) {
if($handle = fopen($lockFile, 'c')) {
if($lock = flock($handle, LOCK_EX | LOCK_NB)) {
// Got lock, not blocking, release and unlink
unlink($lockFile);
fclose($handle);
flock($lock, LOCK_UN);
return false;
}
else {
return true;
}
}
else {
return true;
}
}
return false;
}
/**
* Release all queued locks on the file
* @return bool
*/
public function releaseAll() {
$this->stack = array();
\OC_Log::write('lock', sprintf('INFO: Releasing locks on %s', $this->path), \OC_Log::DEBUG);
if (!empty($this->handle) && is_resource($this->handle)) {
flock($this->handle, LOCK_UN);
\OC_Log::write('lock', sprintf('INFO: Released lock handle %s on %s', $this->handle, $this->path), \OC_Log::DEBUG);
$this->handle = null;
}
if (!empty($this->lockFile) && file_exists($this->lockFile)) {
unlink($this->lockFile);
\OC_Log::write('lock', sprintf('INFO: Released lock file %s on %s', $this->lockFile, $this->path), \OC_Log::DEBUG);
$this->lockFile = null;
}
\OC_Log::write('lock', sprintf('FREE: Released locks on %s', $this->path), \OC_Log::DEBUG);
return true;
}
public function __destruct() {
// Only releaseAll if we have locks to release
if(!empty($this->handle) || !empty($this->lockFile)) {
\OC_Log::write('lock', sprintf('INFO: Destroying locks on %s', $this->path), \OC_Log::DEBUG);
$this->releaseAll();
}
}
}

View File

@ -1,182 +0,0 @@
<?php
/**
* Copyright (c) 2013 ownCloud, Inc.
* This file is licensed under the Affero General Public License version 3 or
* later.
* See the COPYING-README file.
*/
namespace OC\Files\Storage\Wrapper;
use OC\Files\Filesystem;
use OC\Files\Lock;
/**
* Class LockingWrapper
* A Storage Wrapper used to lock files at the system level
* @package OC\Files\Storage\Wrapper
*
* Notes: Does the $locks array need to be global to all LockingWrapper instances, such as in the case of two paths
* that point to the same physical file? Otherwise accessing the file from a different path the second time would show
* the file as locked, even though this process is the one locking it.
*/
class LockingWrapper extends Wrapper {
/** @var \OCP\Files\Lock[] $locks Holds an array of lock instances indexed by path for this storage */
protected $locks = array();
/**
* Acquire a lock on a file
* @param string $path Path to file, relative to this storage
* @param integer $lockType A Lock class constant, Lock::READ/Lock::WRITE
* @param null|resource $existingHandle An existing file handle from an fopen()
* @return bool|\OCP\Files\Lock Lock instance on success, false on failure
*/
protected function getLock($path, $lockType, $existingHandle = null){
$path = Filesystem::normalizePath($this->storage->getLocalFile($path));
if(!isset($this->locks[$path])) {
$this->locks[$path] = new Lock($path);
}
$this->locks[$path]->addLock($lockType, $existingHandle);
return $this->locks[$path];
}
/**
* Release an existing lock
* @param string $path Path to file, relative to this storage
* @param integer $lockType The type of lock to release
* @param bool $releaseAll If true, release all outstanding locks
* @return bool true on success, false on failure
*/
protected function releaseLock($path, $lockType, $releaseAll = false){
$path = Filesystem::normalizePath($this->storage->getLocalFile($path));
if(isset($this->locks[$path])) {
if($releaseAll) {
return $this->locks[$path]->releaseAll();
}
else {
return $this->locks[$path]->release($lockType);
}
}
return true;
}
/**
* see http://php.net/manual/en/function.file_get_contents.php
* @param string $path
* @return string
* @throws \Exception
*/
public function file_get_contents($path) {
try {
$this->getLock($path, Lock::READ);
$result = $this->storage->file_get_contents($path);
}
catch(\Exception $originalException) {
// Need to release the lock before more operations happen in upstream exception handlers
$this->releaseLock($path, Lock::READ);
throw $originalException;
}
$this->releaseLock($path, Lock::READ);
return $result;
}
public function file_put_contents($path, $data) {
try {
$this->getLock($path, Lock::WRITE);
$result = $this->storage->file_put_contents($path, $data);
}
catch(\Exception $originalException) {
// Release lock, throw original exception
$this->releaseLock($path, Lock::WRITE);
throw $originalException;
}
$this->releaseLock($path, Lock::WRITE);
return $result;
}
public function copy($path1, $path2) {
try {
$this->getLock($path1, Lock::READ);
$this->getLock($path2, Lock::WRITE);
$result = $this->storage->copy($path1, $path2);
}
catch(\Exception $originalException) {
// Release locks, throw original exception
$this->releaseLock($path1, Lock::READ);
$this->releaseLock($path2, Lock::WRITE);
throw $originalException;
}
$this->releaseLock($path1, Lock::READ);
$this->releaseLock($path2, Lock::WRITE);
return $result;
}
public function rename($path1, $path2) {
try {
$this->getLock($path1, Lock::READ);
$this->getLock($path2, Lock::WRITE);
$result = $this->storage->rename($path1, $path2);
}
catch(\Exception $originalException) {
// Release locks, throw original exception
$this->releaseLock($path1, Lock::READ);
$this->releaseLock($path2, Lock::WRITE);
throw $originalException;
}
$this->releaseLock($path1, Lock::READ);
$this->releaseLock($path2, Lock::WRITE);
return $result;
}
public function fopen($path, $mode) {
$lockType = Lock::READ;
switch ($mode) {
case 'r+':
case 'rb+':
case 'w+':
case 'wb+':
case 'x+':
case 'xb+':
case 'a+':
case 'ab+':
case 'c+':
case 'w':
case 'wb':
case 'x':
case 'xb':
case 'a':
case 'ab':
case 'c':
$lockType = Lock::WRITE;
break;
}
// The handle for $this->fopen() is used outside of this class, so the handle/lock can't be closed
// Instead, it will be closed when the request goes out of scope
// Storage doesn't have an fclose()
if($result = $this->storage->fopen($path, $mode)) {
$this->getLock($path, $lockType, $result);
}
return $result;
}
public function unlink($path) {
try {
if (\OC\Files\Filesystem::is_file($path)) {
$this->getLock($path, Lock::WRITE);
}
$result = $this->storage->unlink($path);
}
catch(\Exception $originalException) {
// Need to release the lock before more operations happen in upstream exception handlers
$this->releaseLock($path, Lock::WRITE);
throw $originalException;
}
$this->releaseLock($path, Lock::WRITE);
return $result;
}
}

View File

@ -71,16 +71,6 @@ class OC_Util {
return $storage; return $storage;
}); });
// Set up flock
\OC\Files\Filesystem::addStorageWrapper('oc_flock', function($mountPoint, /** @var \OC\Files\Storage\Storage|null $storage */ $storage){
// lock files on all local storage
if ($storage instanceof \OC\Files\Storage\Storage && $storage->isLocal()) {
return new \OC\Files\Storage\Wrapper\LockingWrapper(array('storage' => $storage));
} else {
return $storage;
}
});
$userDir = '/'.$user.'/files'; $userDir = '/'.$user.'/files';
$userRoot = OC_User::getHome($user); $userRoot = OC_User::getHome($user);
$userDirectory = $userRoot . '/files'; $userDirectory = $userRoot . '/files';

View File

@ -1,63 +0,0 @@
<?php
/**
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU AFFERO GENERAL PUBLIC LICENSE
* License as published by the Free Software Foundation; either
* version 3 of the License, or any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU AFFERO GENERAL PUBLIC LICENSE for more details.
*
* You should have received a copy of the GNU Affero General Public
* License along with this library. If not, see <http://www.gnu.org/licenses/>.
*
*/
namespace OCP\Files;
/**
* Class Lock
* @package OC\Files
*/
interface Lock {
const READ = 1;
const WRITE = 2;
/**
* Constructor for the lock instance
* @param string $path Absolute pathname for a local file on which to obtain a lock
*/
public function __construct($path);
/**
* Add a lock of a specific type to the stack
* @param integer $lockType A constant representing the type of lock to queue
* @param null|resource $existingHandle An existing file handle from an fopen()
* @throws LockNotAcquiredException
*/
public function addLock($lockType, $existingHandle = null);
/**
* Release locks on handles and files
*/
public function release($lockType);
/**
* Get the lock file associated to a file
* @param string $filename The filename of the file to create a lock file for
* @return string The filename of the lock file
*/
public static function getLockFile($filename);
/**
* Release all queued locks on the file
* @return bool
*/
public function releaseAll();
}