nextcloud/apps/files_encryption/lib/cryptstream.php

178 lines
5.1 KiB
PHP
Raw Normal View History

2011-10-21 19:02:11 +04:00
<?php
/**
* ownCloud
*
* @author Robin Appelman
* @copyright 2011 Robin Appelman icewind1991@gmail.com
*
* 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/>.
*
*/
/**
* transparently encrypted filestream
2011-11-24 04:44:54 +04:00
*
* you can use it as wrapper around an existing stream by setting
* OC_CryptStream::$sourceStreams['foo']=array('path'=>$path, 'stream'=>$stream)
* and then fopen('crypt://streams/foo');
2011-10-21 19:02:11 +04:00
*/
class OC_CryptStream{
2011-11-24 04:44:54 +04:00
public static $sourceStreams=array();
2011-10-21 19:02:11 +04:00
private $source;
2011-11-24 04:44:54 +04:00
private $path;
private $meta=array();//header/meta for source stream
private $writeCache;
private $size;
2012-06-09 19:33:57 +04:00
private static $rootView;
2011-10-21 19:02:11 +04:00
2012-09-07 17:22:01 +04:00
public function stream_open($path, $mode, $options, &$opened_path) {
if ( ! self::$rootView) {
2012-06-09 19:33:57 +04:00
self::$rootView=new OC_FilesystemView('');
}
2012-10-29 02:58:08 +04:00
$path=str_replace('crypt://', '', $path);
if (dirname($path)=='streams' and isset(self::$sourceStreams[basename($path)])) {
2011-11-24 04:44:54 +04:00
$this->source=self::$sourceStreams[basename($path)]['stream'];
$this->path=self::$sourceStreams[basename($path)]['path'];
$this->size=self::$sourceStreams[basename($path)]['size'];
} else {
2011-11-24 04:44:54 +04:00
$this->path=$path;
if ($mode=='w' or $mode=='w+' or $mode=='wb' or $mode=='wb+') {
$this->size=0;
} else {
2012-10-29 02:58:08 +04:00
$this->size=self::$rootView->filesize($path, $mode);
}
2011-11-24 04:44:54 +04:00
OC_FileProxy::$enabled=false;//disable fileproxies so we can open the source file
2012-10-29 02:58:08 +04:00
$this->source=self::$rootView->fopen($path, $mode);
2011-11-24 04:44:54 +04:00
OC_FileProxy::$enabled=true;
if ( ! is_resource($this->source)) {
2012-10-29 02:58:08 +04:00
OCP\Util::writeLog('files_encryption', 'failed to open '.$path, OCP\Util::ERROR);
2011-11-24 04:44:54 +04:00
}
}
if (is_resource($this->source)) {
2011-11-24 04:44:54 +04:00
$this->meta=stream_get_meta_data($this->source);
2011-10-21 19:02:11 +04:00
}
return is_resource($this->source);
}
2012-08-29 10:42:49 +04:00
2012-09-07 17:22:01 +04:00
public function stream_seek($offset, $whence=SEEK_SET) {
$this->flush();
2012-10-29 02:58:08 +04:00
fseek($this->source, $offset, $whence);
2011-10-21 19:02:11 +04:00
}
2012-08-29 10:42:49 +04:00
2012-09-07 17:22:01 +04:00
public function stream_tell() {
2011-10-21 19:02:11 +04:00
return ftell($this->source);
}
2012-08-29 10:42:49 +04:00
2012-09-07 17:22:01 +04:00
public function stream_read($count) {
//$count will always be 8192 https://bugs.php.net/bug.php?id=21641
//This makes this function a lot simpler but will breake everything the moment it's fixed
$this->writeCache='';
if ($count!=8192) {
OCP\Util::writeLog('files_encryption',
'php bug 21641 no longer holds, decryption will not work',
OCP\Util::FATAL);
die();
2011-11-24 04:44:54 +04:00
}
$pos=ftell($this->source);
2012-10-29 02:58:08 +04:00
$data=fread($this->source, 8192);
if (strlen($data)) {
$result=OC_Crypt::decrypt($data);
} else {
$result='';
2011-11-24 04:44:54 +04:00
}
$length=$this->size-$pos;
if ($length<8192) {
2012-10-29 02:58:08 +04:00
$result=substr($result, 0, $length);
2012-06-16 01:48:39 +04:00
}
return $result;
2011-10-21 19:02:11 +04:00
}
2012-08-29 10:42:49 +04:00
2012-09-07 17:22:01 +04:00
public function stream_write($data) {
2011-10-21 19:02:11 +04:00
$length=strlen($data);
$currentPos=ftell($this->source);
if ($this->writeCache) {
$data=$this->writeCache.$data;
$this->writeCache='';
}
if ($currentPos%8192!=0) {
2011-10-21 19:02:11 +04:00
//make sure we always start on a block start
2012-10-29 02:58:08 +04:00
fseek($this->source, -($currentPos%8192), SEEK_CUR);
2012-11-02 22:53:02 +04:00
$encryptedBlock=fread($this->source, 8192);
2012-10-29 02:58:08 +04:00
fseek($this->source, -($currentPos%8192), SEEK_CUR);
2011-10-21 19:02:11 +04:00
$block=OC_Crypt::decrypt($encryptedBlock);
2012-10-29 02:58:08 +04:00
$data=substr($block, 0, $currentPos%8192).$data;
fseek($this->source, -($currentPos%8192), SEEK_CUR);
2011-10-21 19:02:11 +04:00
}
2012-06-21 20:07:56 +04:00
$currentPos=ftell($this->source);
while ($remainingLength=strlen($data)>0) {
if ($remainingLength<8192) {
$this->writeCache=$data;
$data='';
} else {
2012-10-29 02:58:08 +04:00
$encrypted=OC_Crypt::encrypt(substr($data, 0, 8192));
fwrite($this->source, $encrypted);
$data=substr($data, 8192);
}
2011-10-21 19:02:11 +04:00
}
2012-11-04 14:10:46 +04:00
$this->size=max($this->size, $currentPos+$length);
2011-10-21 19:02:11 +04:00
return $length;
}
2012-11-04 14:10:46 +04:00
public function stream_set_option($option, $arg1, $arg2) {
2012-09-07 17:22:01 +04:00
switch($option) {
2011-10-21 19:02:11 +04:00
case STREAM_OPTION_BLOCKING:
2012-10-29 02:58:08 +04:00
stream_set_blocking($this->source, $arg1);
2011-10-21 19:02:11 +04:00
break;
case STREAM_OPTION_READ_TIMEOUT:
2012-10-29 02:58:08 +04:00
stream_set_timeout($this->source, $arg1, $arg2);
2011-10-21 19:02:11 +04:00
break;
case STREAM_OPTION_WRITE_BUFFER:
2012-10-29 02:58:08 +04:00
stream_set_write_buffer($this->source, $arg1, $arg2);
2011-10-21 19:02:11 +04:00
}
}
2012-09-07 17:22:01 +04:00
public function stream_stat() {
2011-10-21 19:02:11 +04:00
return fstat($this->source);
}
2012-08-29 10:42:49 +04:00
2012-09-07 17:22:01 +04:00
public function stream_lock($mode) {
2012-10-29 02:58:08 +04:00
flock($this->source, $mode);
2011-10-21 19:02:11 +04:00
}
2012-08-29 10:42:49 +04:00
2012-09-07 17:22:01 +04:00
public function stream_flush() {
2011-10-21 19:02:11 +04:00
return fflush($this->source);
}
2012-09-07 17:22:01 +04:00
public function stream_eof() {
2011-10-21 19:02:11 +04:00
return feof($this->source);
}
2012-09-07 17:22:01 +04:00
private function flush() {
if ($this->writeCache) {
$encrypted=OC_Crypt::encrypt($this->writeCache);
2012-10-29 02:58:08 +04:00
fwrite($this->source, $encrypted);
$this->writeCache='';
}
}
2012-09-07 17:22:01 +04:00
public function stream_close() {
$this->flush();
if ($this->meta['mode']!='r' and $this->meta['mode']!='rb') {
2012-10-29 02:58:08 +04:00
OC_FileCache::put($this->path, array('encrypted'=>true, 'size'=>$this->size), '');
2012-04-25 02:10:29 +04:00
}
2011-10-21 19:02:11 +04:00
return fclose($this->source);
}
}