1 files changed, 94 insertions, 0 deletions
diff --git a/seek-bzip2/bzip-table.c b/seek-bzip2/bzip-table.c
new file mode 100644
index 0000000..1f4d537
--- /dev/null
+++ b/seek-bzip2/bzip-table.c
@@ -0,0 +1,94 @@
+#include <stdio.h>
+#include <stdlib.h>
+
+#include "micro-bunzip.h"
+
+#define BUF_SIZE 8192
+
+/**
+ * Read a bzip2 file from stdin and print
+ * 1) The block size
+ * 2) The starting offset (in BITS) of each block of compressed data
+ *
+ * This does not completely uncompress the data, so does not do CRC checks,
+ * (gaining 60% or so speedup), bzip2 --test can be used to verify files
+ * first if desired.
+ */
+int main( int argc, char*argv[] )
+{
+    bunzip_data *bd;
+    int status;
+    unsigned long long position;
+    char * c;
+    char buffer[BUF_SIZE];
+    int gotcount;
+    int totalcount;
+
+    /* Attempt to open the bzip2 file, if successfull this consumes the
+     * entire header and moves us to the start of the first block.
+     */
+    if ( ! ( status = start_bunzip( &bd, 0, 0, 0 ) ) )
+    {
+        for ( ; ; )
+        {
+            /* Determine position */
+            position = bd->position;
+            position = position - bd->inbufCount + bd->inbufPos;
+            position = position * 8 - bd->inbufBitCount;
+
+            /* Read one block */
+            status = get_next_block( bd );
+
+            /* Reset the total size counter for each block */
+            totalcount = 0;
+
+            /* Non-zero return value indicates an error, break out */
+            if ( status ) break;
+
+            /* This is really the only other thing init_block does, hrmm */
+            bd->writeCRC = 0xffffffffUL;
+
+            /* Decompress the block and throw away, but keep track of the
+               total size of the decompressed data */
+            for ( ; ; )
+            {
+                gotcount = read_bunzip( bd, buffer, BUF_SIZE );
+                if ( gotcount < 0 )
+                {
+                    status = gotcount;
+                    goto bzip_table_finish;
+                }
+                else if ( gotcount == 0 )
+                {
+                    break;
+                }
+                else
+                {
+                    totalcount += gotcount;
+                }
+            }
+            /* Print the position of the first bit in the block header */
+            fprintf( stdout, "%llu\t%d\n", position, totalcount );
+        }
+    }
+
+bzip_table_finish:
+
+    /* If we reached the last block, do a CRC check */
+    if ( status == RETVAL_LAST_BLOCK && bd->headerCRC == bd->totalCRC )
+    {
+        status = RETVAL_OK;
+    }
+
+    /* Free decompression buffer and bzip_data */
+    if ( bd->dbuf ) free( bd->dbuf );
+    free( bd );
+
+    /* Print error if required */
+    if ( status )
+    {
+        fprintf( stderr, "\n%s\n", bunzip_errors[-status] );
+    }
+
+    return status;
+}