66 lines
1.7 KiB
PHP
66 lines
1.7 KiB
PHP
<?php
|
|
|
|
namespace Amp\ByteStream\Base64;
|
|
|
|
use Amp\ByteStream\InputStream;
|
|
use Amp\ByteStream\StreamException;
|
|
use Amp\Promise;
|
|
use function Amp\call;
|
|
|
|
final class Base64DecodingInputStream implements InputStream
|
|
{
|
|
/** @var InputStream|null */
|
|
private $source;
|
|
|
|
/** @var string|null */
|
|
private $buffer = '';
|
|
|
|
public function __construct(InputStream $source)
|
|
{
|
|
$this->source = $source;
|
|
}
|
|
|
|
public function read(): Promise
|
|
{
|
|
return call(function () {
|
|
if ($this->source === null) {
|
|
throw new StreamException('Failed to read stream chunk due to invalid base64 data');
|
|
}
|
|
|
|
$chunk = yield $this->source->read();
|
|
if ($chunk === null) {
|
|
if ($this->buffer === null) {
|
|
return null;
|
|
}
|
|
|
|
$chunk = \base64_decode($this->buffer, true);
|
|
if ($chunk === false) {
|
|
$this->source = null;
|
|
$this->buffer = null;
|
|
|
|
throw new StreamException('Failed to read stream chunk due to invalid base64 data');
|
|
}
|
|
|
|
$this->buffer = null;
|
|
|
|
return $chunk;
|
|
}
|
|
|
|
$this->buffer .= $chunk;
|
|
|
|
$length = \strlen($this->buffer);
|
|
$chunk = \base64_decode(\substr($this->buffer, 0, $length - $length % 4), true);
|
|
if ($chunk === false) {
|
|
$this->source = null;
|
|
$this->buffer = null;
|
|
|
|
throw new StreamException('Failed to read stream chunk due to invalid base64 data');
|
|
}
|
|
|
|
$this->buffer = \substr($this->buffer, $length - $length % 4);
|
|
|
|
return $chunk;
|
|
});
|
|
}
|
|
}
|