/** * Test putting a file using chunking * * @dataProvider fopenFailuresProvider */ public function testChunkedPutFails($thrownException, $expectedException, $checkPreviousClass = false) { // setup $storage = $this->getMock('\\OC\\Files\\Storage\\Local', ['fopen'], [['datadir' => \OC::$server->getTempManager()->getTemporaryFolder()]]); \OC\Files\Filesystem::mount($storage, [], $this->user . '/'); $view = $this->getMock('\\OC\\Files\\View', ['getRelativePath', 'resolvePath'], []); $view->expects($this->atLeastOnce())->method('resolvePath')->will($this->returnCallback(function ($path) use($storage) { return [$storage, $path]; })); if ($thrownException !== null) { $storage->expects($this->once())->method('fopen')->will($this->throwException($thrownException)); } else { $storage->expects($this->once())->method('fopen')->will($this->returnValue(false)); } $view->expects($this->any())->method('getRelativePath')->will($this->returnArgument(0)); $_SERVER['HTTP_OC_CHUNKED'] = true; $info = new \OC\Files\FileInfo('/test.txt-chunking-12345-2-0', null, null, ['permissions' => \OCP\Constants::PERMISSION_ALL], null); $file = new \OC\Connector\Sabre\File($view, $info); // put first chunk $this->assertNull($file->put('test data one')); $info = new \OC\Files\FileInfo('/test.txt-chunking-12345-2-1', null, null, ['permissions' => \OCP\Constants::PERMISSION_ALL], null); $file = new \OC\Connector\Sabre\File($view, $info); // action $caughtException = null; try { // last chunk $file->acquireLock(ILockingProvider::LOCK_SHARED); $file->put('test data two'); $file->releaseLock(ILockingProvider::LOCK_SHARED); } catch (\Exception $e) { $caughtException = $e; } $this->assertInstanceOf($expectedException, $caughtException); if ($checkPreviousClass) { $this->assertInstanceOf(get_class($thrownException), $caughtException->getPrevious()); } $this->assertEmpty($this->listPartFiles($view, ''), 'No stray part files'); }
/** * Creates a new file in the directory * * Data will either be supplied as a stream resource, or in certain cases * as a string. Keep in mind that you may have to support either. * * After successful creation of the file, you may choose to return the ETag * of the new file here. * * The returned ETag must be surrounded by double-quotes (The quotes should * be part of the actual string). * * If you cannot accurately determine the ETag, you should not return it. * If you don't store the file exactly as-is (you're transforming it * somehow) you should also not return an ETag. * * This means that if a subsequent GET to this new file does not exactly * return the same contents of what was submitted here, you are strongly * recommended to omit the ETag. * * @param string $name Name of the file * @param resource|string $data Initial payload * @return null|string * @throws Exception\EntityTooLarge * @throws Exception\UnsupportedMediaType * @throws FileLocked * @throws InvalidPath * @throws \Sabre\DAV\Exception * @throws \Sabre\DAV\Exception\BadRequest * @throws \Sabre\DAV\Exception\Forbidden * @throws \Sabre\DAV\Exception\ServiceUnavailable */ public function createFile($name, $data = null) { try { // for chunked upload also updating a existing file is a "createFile" // because we create all the chunks before re-assemble them to the existing file. if (isset($_SERVER['HTTP_OC_CHUNKED'])) { // exit if we can't create a new file and we don't updatable existing file $info = \OC_FileChunking::decodeName($name); if (!$this->fileView->isCreatable($this->path) && !$this->fileView->isUpdatable($this->path . '/' . $info['name'])) { throw new \Sabre\DAV\Exception\Forbidden(); } } else { // For non-chunked upload it is enough to check if we can create a new file if (!$this->fileView->isCreatable($this->path)) { throw new \Sabre\DAV\Exception\Forbidden(); } } $this->fileView->verifyPath($this->path, $name); $path = $this->fileView->getAbsolutePath($this->path) . '/' . $name; // using a dummy FileInfo is acceptable here since it will be refreshed after the put is complete $info = new \OC\Files\FileInfo($path, null, null, array(), null); $node = new \OC\Connector\Sabre\File($this->fileView, $info); $node->acquireLock(ILockingProvider::LOCK_SHARED); return $node->put($data); } catch (\OCP\Files\StorageNotAvailableException $e) { throw new \Sabre\DAV\Exception\ServiceUnavailable($e->getMessage()); } catch (\OCP\Files\InvalidPathException $ex) { throw new InvalidPath($ex->getMessage()); } catch (LockedException $e) { throw new FileLocked($e->getMessage(), $e->getCode(), $e); } }