nextcloud/apps/files_encryption/tests/stream.php

86 lines
2.3 KiB
PHP
Raw Normal View History

<?php
/**
* Copyright (c) 2012 Robin Appelman <icewind@owncloud.com>
* This file is licensed under the Affero General Public License version 3 or
* later.
* See the COPYING-README file.
*/
class Test_CryptStream extends UnitTestCase {
private $tmpFiles=array();
2012-08-29 10:42:49 +04:00
2012-09-07 17:22:01 +04:00
function testStream() {
2012-11-04 14:10:46 +04:00
$stream=$this->getStream('test1', 'w', strlen('foobar'));
fwrite($stream, 'foobar');
fclose($stream);
2012-11-04 14:10:46 +04:00
$stream=$this->getStream('test1', 'r', strlen('foobar'));
2012-11-02 22:53:02 +04:00
$data=fread($stream, 6);
fclose($stream);
2012-11-02 22:53:02 +04:00
$this->assertEqual('foobar', $data);
$file=OC::$SERVERROOT.'/3rdparty/MDB2.php';
2012-11-04 14:10:46 +04:00
$source=fopen($file, 'r');
2012-11-02 22:53:02 +04:00
$target=$this->getStream('test2', 'w', 0);
OCP\Files::streamCopy($source, $target);
fclose($target);
fclose($source);
2012-11-04 14:10:46 +04:00
$stream=$this->getStream('test2', 'r', filesize($file));
$data=stream_get_contents($stream);
$original=file_get_contents($file);
2012-10-24 00:53:54 +04:00
$this->assertEqual(strlen($original), strlen($data));
2012-11-02 22:53:02 +04:00
$this->assertEqual($original, $data);
}
/**
* get a cryptstream to a temporary file
* @param string $id
* @param string $mode
2012-06-21 19:37:53 +04:00
* @param int size
* @return resource
*/
2012-11-02 22:53:02 +04:00
function getStream($id, $mode, $size) {
if ($id==='') {
$id=uniqid();
}
if ( ! isset($this->tmpFiles[$id])) {
$file=OCP\Files::tmpFile();
$this->tmpFiles[$id]=$file;
} else {
$file=$this->tmpFiles[$id];
}
2012-11-02 22:53:02 +04:00
$stream=fopen($file, $mode);
2012-11-04 14:10:46 +04:00
OC_CryptStream::$sourceStreams[$id]=array('path'=>'dummy'.$id, 'stream'=>$stream, 'size'=>$size);
2012-11-02 22:53:02 +04:00
return fopen('crypt://streams/'.$id, $mode);
}
2012-06-16 01:48:39 +04:00
2012-09-07 17:22:01 +04:00
function testBinary() {
2012-06-16 01:48:39 +04:00
$file=__DIR__.'/binary';
$source=file_get_contents($file);
2012-11-04 14:10:46 +04:00
$stream=$this->getStream('test', 'w', strlen($source));
2012-11-02 22:53:02 +04:00
fwrite($stream, $source);
2012-06-16 01:48:39 +04:00
fclose($stream);
2012-11-04 14:10:46 +04:00
$stream=$this->getStream('test', 'r', strlen($source));
2012-06-21 19:37:53 +04:00
$data=stream_get_contents($stream);
fclose($stream);
2012-10-24 00:53:54 +04:00
$this->assertEqual(strlen($data), strlen($source));
2012-11-02 22:53:02 +04:00
$this->assertEqual($source, $data);
2012-06-21 19:37:53 +04:00
$file=__DIR__.'/zeros';
$source=file_get_contents($file);
2012-11-04 14:10:46 +04:00
$stream=$this->getStream('test2', 'w', strlen($source));
2012-11-02 22:53:02 +04:00
fwrite($stream, $source);
2012-06-21 19:37:53 +04:00
fclose($stream);
2012-11-04 14:10:46 +04:00
$stream=$this->getStream('test2', 'r', strlen($source));
2012-06-16 01:48:39 +04:00
$data=stream_get_contents($stream);
fclose($stream);
2012-10-24 00:53:54 +04:00
$this->assertEqual(strlen($data), strlen($source));
2012-11-02 22:53:02 +04:00
$this->assertEqual($source, $data);
2012-06-16 01:48:39 +04:00
}
}