fix incorrect information in the filecache when using encryption

This commit is contained in:
Robin Appelman 2012-02-21 20:48:14 +01:00
parent d9c7e4c333
commit c20319d377
6 changed files with 62 additions and 24 deletions

View File

@ -90,7 +90,6 @@ class OC_CryptStream{
} }
public function stream_write($data){ public function stream_write($data){
error_log('write to '. $this->path);
$length=strlen($data); $length=strlen($data);
$written=0; $written=0;
$currentPos=ftell($this->source); $currentPos=ftell($this->source);
@ -148,9 +147,7 @@ class OC_CryptStream{
} }
public function stream_close(){ public function stream_close(){
if(OC_FileCache::inCache($this->path)){ OC_FileCache::put($this->path,array('encrypted'=>true));
OC_FileCache::put($this->path,array('encrypted'=>true));
}
return fclose($this->source); return fclose($this->source);
} }
} }

View File

@ -38,13 +38,7 @@ class OC_FileProxy_Encryption extends OC_FileProxy{
if(is_null(self::$blackList)){ if(is_null(self::$blackList)){
self::$blackList=explode(',',OC_Appconfig::getValue('files_encryption','type_blacklist','jpg,png,jpeg,avi,mpg,mpeg,mkv,mp3,oga,ogv,ogg')); self::$blackList=explode(',',OC_Appconfig::getValue('files_encryption','type_blacklist','jpg,png,jpeg,avi,mpg,mpeg,mkv,mp3,oga,ogv,ogg'));
} }
if(isset(self::$metaData[$path])){ if(self::isEncrypted($path)){
$metadata=self::$metaData[$path];
}else{
$metadata=OC_FileCache::get($path);
self::$metaData[$path]=$metadata;
}
if($metadata['encrypted']){
return true; return true;
} }
$extention=substr($path,strrpos($path,'.')+1); $extention=substr($path,strrpos($path,'.')+1);
@ -62,7 +56,7 @@ class OC_FileProxy_Encryption extends OC_FileProxy{
if(isset(self::$metaData[$path])){ if(isset(self::$metaData[$path])){
$metadata=self::$metaData[$path]; $metadata=self::$metaData[$path];
}else{ }else{
$metadata=OC_FileCache::get($path); $metadata=OC_FileCache::getCached($path);
self::$metaData[$path]=$metadata; self::$metaData[$path]=$metadata;
} }
return (bool)$metadata['encrypted']; return (bool)$metadata['encrypted'];
@ -92,14 +86,19 @@ class OC_FileProxy_Encryption extends OC_FileProxy{
fclose($result); fclose($result);
$result=fopen('crypt://'.$path,$meta['mode']); $result=fopen('crypt://'.$path,$meta['mode']);
}elseif(self::shouldEncrypt($path) and $meta['mode']!='r'){ }elseif(self::shouldEncrypt($path) and $meta['mode']!='r'){
if(OC_Filesystem::file_exists($path)){ if(OC_Filesystem::file_exists($path) and OC_Filesystem::filesize($path)>0){
//first encrypt the target file so we don't end up with a half encrypted file //first encrypt the target file so we don't end up with a half encrypted file
OC_Log::write('files_encryption','Decrypting '.$path.' before writing',OC_Log::DEBUG); OC_Log::write('files_encryption','Decrypting '.$path.' before writing',OC_Log::DEBUG);
if($result){ $tmp=fopen('php://temp');
fclose($result); while(!feof($result)){
$chunk=fread($result,8192);
if($chunk){
fwrite($tmp,$chunk);
}
} }
$tmpFile=OC_Filesystem::toTmpFile($path); fclose($result);
OC_Filesystem::fromTmpFile($tmpFile,$path); OC_Filesystem::file_put_contents($path,$tmp);
fclose($tmp);
} }
$result=fopen('crypt://'.$path,$meta['mode']); $result=fopen('crypt://'.$path,$meta['mode']);
} }
@ -117,6 +116,10 @@ class OC_FileProxy_Encryption extends OC_FileProxy{
} }
public function postGetMimeType($path,$mime){ public function postGetMimeType($path,$mime){
return OC_Helper::getMimeType('crypt://'.$path,'w'); if((!OC_FileCache::inCache($path) and self::shouldEncrypt($path)) or self::isEncrypted($path)){
return OC_Helper::getMimeType('crypt://'.$path,'w');
}else{
return $mime;
}
} }
} }

View File

@ -56,6 +56,7 @@ class OC_MEDIA{
*/ */
public static function updateFile($params){ public static function updateFile($params){
$path=$params['path']; $path=$params['path'];
if(!$path) return;
require_once 'lib_scanner.php'; require_once 'lib_scanner.php';
require_once 'lib_collection.php'; require_once 'lib_collection.php';
//fix a bug where there were multiply '/' in front of the path, it should only be one //fix a bug where there were multiply '/' in front of the path, it should only be one

View File

@ -48,7 +48,8 @@ if(strpos($dir,'..') === false){
for($i=0;$i<$fileCount;$i++){ for($i=0;$i<$fileCount;$i++){
$target=stripslashes($dir) . $files['name'][$i]; $target=stripslashes($dir) . $files['name'][$i];
if(is_uploaded_file($files['tmp_name'][$i]) and OC_Filesystem::fromTmpFile($files['tmp_name'][$i],$target)){ if(is_uploaded_file($files['tmp_name'][$i]) and OC_Filesystem::fromTmpFile($files['tmp_name'][$i],$target)){
$result[]=array( "status" => "success", 'mime'=>OC_Filesystem::getMimeType($target),'size'=>OC_Filesystem::filesize($target),'name'=>$files['name'][$i]); $meta=OC_FileCache::getCached($target);
$result[]=array( "status" => "success", 'mime'=>$meta['mimetype'],'size'=>$meta['size'],'name'=>$files['name'][$i]);
} }
} }
OC_JSON::encodedPrint($result); OC_JSON::encodedPrint($result);

View File

@ -28,6 +28,8 @@
* It will try to keep the data up to date but changes from outside ownCloud can invalidate the cache * It will try to keep the data up to date but changes from outside ownCloud can invalidate the cache
*/ */
class OC_FileCache{ class OC_FileCache{
private static $savedData=array();
/** /**
* get the filesystem info from the cache * get the filesystem info from the cache
* @param string path * @param string path
@ -93,6 +95,14 @@ class OC_FileCache{
self::update($id,$data); self::update($id,$data);
return; return;
} }
if(isset(self::$savedData[$path])){
$data=array_merge($data,self::$savedData[$path]);
unset(self::$savedData[$path]);
}
if(!isset($data['size']) or !isset($data['mtime'])){//save incomplete data for the next time we write it
self::$savedData[$path]=$data;
return;
}
if(!isset($data['encrypted'])){ if(!isset($data['encrypted'])){
$data['encrypted']=false; $data['encrypted']=false;
} }
@ -101,9 +111,8 @@ class OC_FileCache{
} }
$mimePart=dirname($data['mimetype']); $mimePart=dirname($data['mimetype']);
$user=OC_User::getUser(); $user=OC_User::getUser();
$query=OC_DB::prepare('INSERT INTO *PREFIX*fscache(parent, name, path, size, mtime, ctime, mimetype, mimepart,user,writable) VALUES(?,?,?,?,?,?,?,?,?,?)'); $query=OC_DB::prepare('INSERT INTO *PREFIX*fscache(parent, name, path, size, mtime, ctime, mimetype, mimepart,user,writable,encrypted,versioned) VALUES(?,?,?,?,?,?,?,?,?,?,?,?)');
$query->execute(array($parent,basename($path),$path,$data['size'],$data['mtime'],$data['ctime'],$data['mimetype'],$mimePart,$user,$data['writable'])); $query->execute(array($parent,basename($path),$path,$data['size'],$data['mtime'],$data['ctime'],$data['mimetype'],$mimePart,$user,$data['writable'],$data['encrypted'],$data['versioned']));
} }
/** /**
@ -324,6 +333,28 @@ class OC_FileCache{
self::increaseSize(dirname($fullPath),$size-$cachedSize); self::increaseSize(dirname($fullPath),$size-$cachedSize);
} }
public static function getCached($path,$root=''){
if(!$root){
$root=OC_Filesystem::getRoot();
}else{
if($root=='/'){
$root='';
}
}
$path=$root.$path;
$query=OC_DB::prepare('SELECT ctime,mtime,mimetype,size,encrypted,versioned,writable FROM *PREFIX*fscache WHERE path=?');
$result=$query->execute(array($path))->fetchRow();
if(is_array($result)){
if(isset(self::$savedData[$path])){
$result=array_merge($result,self::$savedData[$path]);
}
return $result;
}else{
OC_Log::write('get(): file not found in cache ('.$path.')','core',OC_Log::DEBUG);
return false;
}
}
private static function getCachedSize($path,$root){ private static function getCachedSize($path,$root){
if(!$root){ if(!$root){
$root=OC_Filesystem::getRoot(); $root=OC_Filesystem::getRoot();
@ -332,6 +363,7 @@ class OC_FileCache{
$root=''; $root='';
} }
} }
$path=$root.$path;
$query=OC_DB::prepare('SELECT size FROM *PREFIX*fscache WHERE path=?'); $query=OC_DB::prepare('SELECT size FROM *PREFIX*fscache WHERE path=?');
$result=$query->execute(array($path)); $result=$query->execute(array($path));
if($row=$result->fetchRow()){ if($row=$result->fetchRow()){

View File

@ -171,6 +171,7 @@ class OC_FilesystemView {
} }
fclose($target); fclose($target);
fclose($data); fclose($data);
OC_Hook::emit( OC_Filesystem::CLASSNAME, OC_Filesystem::signal_post_write, array( OC_Filesystem::signal_param_path => $path));
return true; return true;
}else{ }else{
return false; return false;
@ -278,6 +279,9 @@ class OC_FilesystemView {
} }
public function fromTmpFile($tmpFile,$path){ public function fromTmpFile($tmpFile,$path){
if(OC_Filesystem::isValidPath($path)){ if(OC_Filesystem::isValidPath($path)){
if(!$tmpFile){
debug_print_backtrace();
}
$source=fopen($tmpFile,'r'); $source=fopen($tmpFile,'r');
if($source){ if($source){
$this->file_put_contents($path,$source); $this->file_put_contents($path,$source);
@ -286,7 +290,7 @@ class OC_FilesystemView {
}else{ }else{
} }
}else{ }else{
error_log('invalid path'); return false;
} }
} }