upstream/nextcloud
1
0
Fork 0
mirror of https://github.com/nextcloud/server.git synced 2026-06-13 18:50:47 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

fix: apply quota with writeStream

Browse source 
Signed-off-by: Robin Appelman <robin@icewind.nl>
This commit is contained in:
Robin Appelman 2026-04-29 19:14:57 +02:00
parent f3a5bc8c17
commit 9f5535d01b
3 changed files with 57 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
lib/private/Files/Storage/Wrapper/Quota.php
View file

@ -11,6 +11,8 @@ use OC\Files\Filesystem;
use OC\SystemConfig;
use OCP\Files\Cache\ICacheEntry;
use OCP\Files\FileInfo;
use OCP\Files\GenericFileException;
use OCP\Files\NotEnoughSpaceException;
use OCP\Files\Storage\IStorage;
class Quota extends Wrapper {
@ -214,4 +216,31 @@ class Quota extends Wrapper {
public function enableQuota(bool $enabled): void {
$this->enabled = $enabled;
}
#[\Override]
public function writeStream(string $path, $stream, ?int $size = null): int {
if (!$this->hasQuota()) {
return parent::writeStream($path, $stream, $size);
}
$free = $this->free_space($path);
if ($this->shouldApplyQuota($path) && $free == 0) {
throw new NotEnoughSpaceException();
}
if ($size !== null) {
if ($size < $free) {
return parent::writeStream($path, $stream, $size);
} else {
throw new NotEnoughSpaceException();
}
} else {
// force fallback through `fopen` to handle the quota
try {
return parent::writeStreamFallback($path, $stream);
} catch (GenericFileException) {
throw new NotEnoughSpaceException();
}
}
}
}

7
lib/private/Files/Storage/Wrapper/Wrapper.php
View file

@ -392,6 +392,13 @@ class Wrapper implements Storage, ILockingStorage, IWriteStreamStorage {
return $storage->writeStream($path, $stream, $size);
}
return $this->writeStreamFallback($path, $stream);
}
/**
* @param resource $stream
*/
protected function writeStreamFallback(string $path, $stream): int {
$target = $this->fopen($path, 'w');
if ($target === false) {
throw new GenericFileException('Failed to open ' . $path);

21
tests/lib/Files/Storage/Wrapper/QuotaTest.php
View file

@ -235,4 +235,25 @@ class QuotaTest extends \Test\Files\Storage\Storage {
$fh = $instance->fopen('files/test.txt', 'w');
$this->assertFalse($fh);
}
public function testNoWriteStreamQuota(): void {
$instance = $this->getLimitedStorage(5.0);
$stream = fopen('php://temp', 'w+');
fwrite($stream, 'foo');
rewind($stream);
$instance->writeStream('files/test.txt', $stream);
$stream = fopen('php://temp', 'w+');
fwrite($stream, 'foobar');
rewind($stream);
$this->expectException(Files\NotEnoughSpaceException::class);
$instance->writeStream('files/test.txt', $stream);
}
public function testNoWriteStreamQuotaZero(): void {
$instance = $this->getLimitedStorage(0.0);
$stream = fopen('php://temp', 'w+');
$this->expectException(Files\NotEnoughSpaceException::class);
$instance->writeStream('files/test.txt', $stream);
}
}