#define MARKER_CHAR 0xff
#define STUFF_MARKER 0x00
-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)
{
memset(source, 0, sizeof(*source));
source->bytes = (uint8_t*)malloc(BYTERESERVOIR_SIZE);
source->byte_read_ptr = source->bytes;
source->input_func = input_func;
+ source->padding_bytes_available = padding_bytes;
source->userdata = userdata;
}
exit(1);
}
if (bytes_read == 0) {
- fprintf(stderr, "Premature EOF\n");
- exit(1);
+ source->source_eof = true;
+ if (source->padding_bytes_available > 0) {
+ unsigned padding_to_add = source->padding_bytes_available;
+ if (padding_to_add > bytes_to_read) {
+ padding_to_add = bytes_to_read;
+ }
+ memset(source->bytes + source->bytes_available, 0, padding_to_add);
+ source->padding_bytes_available -= padding_to_add;
+ source->bytes_available += padding_to_add;
+ } else {
+ fprintf(stderr, "Premature EOF\n");
+ exit(1);
+ }
+ } else {
+ source->bytes_available += bytes_read;
}
-
- source->bytes_available += bytes_read;
}
// Fill the bit reservoir one by one byte until we have enough.