X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=bitsource.h;h=60744b32e12bdc7809f8b0d7cf47d3271a8b8295;hb=a733460846675aab7117c5cf1213d1b62785793d;hp=6cabbf004245139d6f3f5a8f5f5fac5cfdb646e6;hpb=352c52d38c212a2aa93a9970731c9d7a2f8d4dac;p=fjl

diff --git a/bitsource.h b/bitsource.h
index 6cabbf0..60744b3 100644
--- a/bitsource.h
+++ b/bitsource.h
@@ -2,6 +2,7 @@
 #define _BITSOURCE_H 1
 
 #include <assert.h>
+#include <stdbool.h>
 #include <stddef.h>
 #include <stdint.h>
 #include <sys/types.h>
@@ -34,14 +35,22 @@ struct bit_source {
 	// When this is empty, it needs to be refilled from the input
 	// stream.
 	uint8_t* bytes;
+	uint8_t* byte_read_ptr;
 	unsigned bytes_available;
 
+	// Some clients will purposedly read a bit ahead of the stream, causing
+	// problems at EOF. Thus, the client is allowed to request that we pad
+	// the end stream with a few bytes after the source reports EOF.
+	int padding_bytes_available;
+
 	// Data source.
 	input_func_t* input_func;
 	void* userdata;
+	bool source_eof;
 };
 
-void init_bit_source(struct bit_source* source, input_func_t* input_func, void* userdata);
+void init_bit_source(struct bit_source* source, input_func_t* input_func,
+                     unsigned padding_bytes, void* userdata);
 
 // Internal function. Do not use.
 void possibly_refill_slow_path(struct bit_source* source, unsigned num_bits);
@@ -62,8 +71,8 @@ static inline void possibly_refill(struct bit_source* source, unsigned num_bits)
 	// Slower path (~99% of remaining invocations?)
 	assert(source->bits_available + BITRESERVOIR_FILL_SIZE < BITRESERVOIR_SIZE);
 	if (source->bytes_available >= sizeof(bitreservoir_fill_t)) {
-		bitreservoir_fill_t fill = read_bitreservoir_fill(source->bytes);
-		source->bytes += sizeof(bitreservoir_fill_t);
+		bitreservoir_fill_t fill = read_bitreservoir_fill(source->byte_read_ptr);
+		source->byte_read_ptr += sizeof(bitreservoir_fill_t);
 		source->bytes_available -= sizeof(bitreservoir_fill_t);
 		source->bits |= (bitreservoir_t)fill << (BITRESERVOIR_SIZE - BITRESERVOIR_FILL_SIZE - source->bits_available);
 		source->bits_available += BITRESERVOIR_FILL_SIZE;