class-chunkeddecoderreadstream.php
6 days ago
class-chunkedencoderbytetransformer.php
6 days ago
class-requestreadstream.php
6 days ago
class-seekablerequestreadstream.php
6 days ago
class-chunkeddecoderreadstream.php
98 lines
| 1 | <?php |
| 2 | |
| 3 | namespace WordPress\HttpClient\ByteStream; |
| 4 | |
| 5 | use Exception; |
| 6 | use WordPress\ByteStream\ReadStream\BaseByteReadStream; |
| 7 | |
| 8 | class ChunkedDecoderReadStream extends BaseByteReadStream { |
| 9 | |
| 10 | private $state = self::SCAN_CHUNK_SIZE; |
| 11 | const SCAN_CHUNK_SIZE = 'SCAN_CHUNK_SIZE'; |
| 12 | const SCAN_CHUNK_DATA = 'SCAN_CHUNK_DATA'; |
| 13 | const SCAN_CHUNK_TRAILER = 'SCAN_CHUNK_TRAILER'; |
| 14 | const SCAN_FINAL_CHUNK = 'SCAN_FINAL_CHUNK'; |
| 15 | |
| 16 | private $upstream; |
| 17 | private $chunk_remaining_bytes = 0; |
| 18 | |
| 19 | public function __construct( BaseByteReadStream $upstream ) { |
| 20 | $this->upstream = $upstream; |
| 21 | } |
| 22 | |
| 23 | protected function internal_pull( $n ): string { |
| 24 | if ( self::SCAN_FINAL_CHUNK === $this->state ) { |
| 25 | return ''; |
| 26 | } |
| 27 | |
| 28 | while ( true ) { |
| 29 | if ( self::SCAN_CHUNK_SIZE === $this->state ) { |
| 30 | // Try to peek enough bytes to find chunk size and CRLF. |
| 31 | $this->upstream->pull( 20 ); |
| 32 | $peeked = $this->upstream->peek( 20 ); |
| 33 | |
| 34 | if ( strlen( $peeked ) < 3 ) { // Need at least "0\r\n". |
| 35 | break; |
| 36 | } |
| 37 | |
| 38 | $chunk_bytes_nb = strspn( $peeked, '0123456789abcdefABCDEF' ); |
| 39 | if ( 0 === $chunk_bytes_nb ) { |
| 40 | throw new Exception( 'Invalid chunk size format' ); |
| 41 | } |
| 42 | |
| 43 | $clrf_pos = strpos( $peeked, "\r\n", $chunk_bytes_nb ); |
| 44 | if ( false === $clrf_pos ) { |
| 45 | break; |
| 46 | } |
| 47 | |
| 48 | // Now we can safely consume the chunk header. |
| 49 | $chunk_header = $this->upstream->consume( $clrf_pos + 2 ); |
| 50 | $chunk_size = hexdec( substr( $chunk_header, 0, $chunk_bytes_nb ) ); |
| 51 | |
| 52 | if ( 0 === $chunk_size ) { |
| 53 | $this->state = self::SCAN_FINAL_CHUNK; |
| 54 | |
| 55 | return ''; |
| 56 | } |
| 57 | |
| 58 | $this->chunk_remaining_bytes = $chunk_size; |
| 59 | $this->state = self::SCAN_CHUNK_DATA; |
| 60 | } elseif ( self::SCAN_CHUNK_DATA === $this->state ) { |
| 61 | $bytes_to_read = min( $this->chunk_remaining_bytes, 65536 ); |
| 62 | $available = $this->upstream->pull( $bytes_to_read ); |
| 63 | if ( 0 === $available ) { |
| 64 | break; |
| 65 | } |
| 66 | |
| 67 | $data = $this->upstream->consume( $available ); |
| 68 | $this->chunk_remaining_bytes -= strlen( $data ); |
| 69 | if ( 0 === $this->chunk_remaining_bytes ) { |
| 70 | $this->state = self::SCAN_CHUNK_TRAILER; |
| 71 | } |
| 72 | |
| 73 | return $data; |
| 74 | } elseif ( self::SCAN_CHUNK_TRAILER === $this->state ) { |
| 75 | $this->upstream->pull( 2 ); |
| 76 | $trailer = $this->upstream->peek( 2 ); |
| 77 | |
| 78 | if ( strlen( $trailer ) < 2 ) { |
| 79 | break; |
| 80 | } |
| 81 | |
| 82 | if ( "\r\n" !== $trailer ) { |
| 83 | throw new Exception( 'Expected CRLF after chunk data' ); |
| 84 | } |
| 85 | |
| 86 | $this->upstream->consume( 2 ); |
| 87 | $this->state = self::SCAN_CHUNK_SIZE; |
| 88 | } |
| 89 | } |
| 90 | |
| 91 | return ''; |
| 92 | } |
| 93 | |
| 94 | protected function internal_reached_end_of_data(): bool { |
| 95 | return self::SCAN_FINAL_CHUNK === $this->state; |
| 96 | } |
| 97 | } |
| 98 |