rbzip2 0.1.0

data/.gitignore

@@ -0,0 +1,3 @@
+.yardoc/
+doc/
+pkg/

data/.travis.yml

@@ -0,0 +1,9 @@
+rvm:
+  - 1.8.7
+  - 1.9.2
+  - 1.9.3
+  - jruby
+  - rbx
+  - rbx-2.0
+  - ree
+  - ruby-head

data/Gemfile

@@ -0,0 +1,3 @@
+source :rubygems
+
+gemspec

data/Gemfile.lock

@@ -0,0 +1,28 @@
+PATH
+  remote: .
+  specs:
+    rbzip2 (0.1.0)
+
+GEM
+  remote: http://rubygems.org/
+  specs:
+    diff-lcs (1.1.3)
+    metaclass (0.0.1)
+    mocha (0.10.0)
+      metaclass (~> 0.0.1)
+    rake (0.9.2.2)
+    rspec-core (2.7.1)
+    rspec-expectations (2.7.0)
+      diff-lcs (~> 1.1.2)
+    yard (0.7.3)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  mocha (~> 0.10.0)
+  rake (~> 0.9.2)
+  rbzip2!
+  rspec-core (~> 2.7.1)
+  rspec-expectations (~> 2.7.0)
+  yard (~> 0.7.3)

data/LICENSE

@@ -0,0 +1,25 @@
+Copyright (c) 2011, Sebastian Staudt
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without modification,
+are permitted provided that the following conditions are met:
+
+* Redistributions of source code must retain the above copyright notice,
+  this list of conditions and the following disclaimer.
+* Redistributions in binary form must reproduce the above copyright notice,
+  this list of conditions and the following disclaimer in the documentation
+  and/or other materials provided with the distribution.
+* Neither the name of the author nor the names of its contributors
+  may be used to endorse or promote products derived from this software
+  without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
+ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
+ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
+ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

data/README.md

@@ -0,0 +1,59 @@
+RBzip2
+======
+
+RBzip2 is a gem providing a pure Ruby implementation of the [bzip2][1]
+algorithm used for compression and decompression.
+
+It is based on the code of the [Apache Commons Compress][2] project and adds
+a straight Ruby-like API. There are no external dependencies like other gems or
+libraries. Therefore it will run on any Ruby implementation and the respective
+operating systems supported by those implementations.
+
+## Features
+
+ * Decompression of bzip2 compressed IOs (like `File` or `StringIO`)
+
+## Usage
+
+    require 'rbzip2'
+
+    file = File.new 'somefile.bz2'  # open a compressed file
+    io   = RBzip2::IO.new file      # wrap the file into RBzip2's IO
+    data = io.read                  # read data into a string
+
+## Future plans
+
+ * Simple decompression of strings
+ * Compression of raw data
+ * Simple creation of compressed files
+ * Two-way compressed IO that will (de)compress as you read/write
+
+## Installation
+
+To install RBzip2 as a Ruby gem use the following command:
+
+    gem install rbzip2
+
+To use it as a dependency managed by Bundler add the following to your
+`Gemfile`:
+
+    gem 'rbzip2'
+
+## License
+
+This code is free software; you can redistribute it and/or modify it under the
+terms of the new BSD License. A copy of this license can be found in the
+included LICENSE file.
+
+## Credits
+
+* Sebastian Staudt -- koraktor(at)gmail.com
+
+## See Also
+
+* [Documentation](http://rubydoc.info/gems/rbzip2)
+* [GitHub project page](https://github.com/koraktor/rbzip2)
+* [bzip2 project page][1]
+
+ [1]: http://bzip.org
+ [2]: http://commons.apache.org/compress

data/Rakefile

@@ -0,0 +1,39 @@
+# -*- encoding: utf-8 -*-
+
+# This code is free software; you can redistribute it and/or modify it under
+# the terms of the new BSD License.
+#
+# Copyright (c) 2011, Sebastian Staudt
+
+require 'rspec/core/rake_task'
+require 'rubygems/package_task'
+
+task :default => :spec
+
+spec = Gem::Specification.load 'rbzip2.gemspec'
+Gem::PackageTask.new(spec) do |pkg|
+end
+
+RSpec::Core::RakeTask.new('spec') do |t|
+end
+
+begin
+  require 'yard'
+
+  YARD::Rake::YardocTask.new do |yardoc|
+    yardoc.name    = 'doc'
+    yardoc.files   = [ 'lib/**/*.rb', 'LICENSE', 'README.md' ]
+    yardoc.options = [ '--private', '--title', 'RBzip2 — API Documentation' ]
+  end
+rescue LoadError
+  desc 'Generate YARD Documentation (not available)'
+  task :doc do
+    $stderr.puts 'You need YARD to build the documentation. Install it using `gem install yard`.'
+  end
+end
+
+desc 'Clean documentation and package directories'
+task :clean do
+  FileUtils.rm_rf 'doc'
+  FileUtils.rm_rf 'pkg'
+end

data/lib/rbzip2/constants.rb

@@ -0,0 +1,66 @@
+# This code is free software; you can redistribute it and/or modify it under
+# the terms of the new BSD License.
+#
+# Copyright (c) 2011, Sebastian Staudt
+
+module RBzip2
+
+  BASEBLOCKSIZE = 100000
+  MAX_ALPHA_SIZE = 258
+  MAX_CODE_LEN = 23
+  RUNA = 0
+  RUNB = 1
+  N_GROUPS = 6
+  G_SIZE = 50
+  N_ITERS = 4
+  MAX_SELECTORS = (2 + (900000 / G_SIZE))
+  NUM_OVERSHOOT_BYTES = 20
+
+  EOF = 0
+  START_BLOCK_STATE = 1
+  RAND_PART_A_STATE = 2
+  RAND_PART_B_STATE = 3
+  RAND_PART_C_STATE = 4
+  NO_RAND_PART_A_STATE = 5
+  NO_RAND_PART_B_STATE = 6
+  NO_RAND_PART_C_STATE = 7
+  
+  RNUMS = [
+    619, 720, 127, 481, 931, 816, 813, 233, 566, 247, 985, 724, 205, 454, 863,
+    491, 741, 242, 949, 214, 733, 859, 335, 708, 621, 574,  73, 654, 730, 472,
+    419, 436, 278, 496, 867, 210, 399, 680, 480,  51, 878, 465, 811, 169, 869,
+    675, 611, 697, 867, 561, 862, 687, 507, 283, 482, 129, 807, 591, 733, 623,
+    150, 238,  59, 379, 684, 877, 625, 169, 643, 105, 170, 607, 520, 932, 727,
+    476, 693, 425, 174, 647,  73, 122, 335, 530, 442, 853, 695, 249, 445, 515,
+    909, 545, 703, 919, 874, 474, 882, 500, 594, 612, 641, 801, 220, 162, 819,
+    984, 589, 513, 495, 799, 161, 604, 958, 533, 221, 400, 386, 867, 600, 782,
+    382, 596, 414, 171, 516, 375, 682, 485, 911, 276,  98, 553, 163, 354, 666,
+    933, 424, 341, 533, 870, 227, 730, 475, 186, 263, 647, 537, 686, 600, 224,
+    469,  68, 770, 919, 190, 373, 294, 822, 808, 206, 184, 943, 795, 384, 383,
+    461, 404, 758, 839, 887, 715,  67, 618, 276, 204, 918, 873, 777, 604, 560,
+    951, 160, 578, 722,  79, 804,  96, 409, 713, 940, 652, 934, 970, 447, 318,
+    353, 859, 672, 112, 785, 645, 863, 803, 350, 139,  93, 354,  99, 820, 908,
+    609, 772, 154, 274, 580, 184,  79, 626, 630, 742, 653, 282, 762, 623, 680,
+     81, 927, 626, 789, 125, 411, 521, 938, 300, 821,  78, 343, 175, 128, 250,
+    170, 774, 972, 275, 999, 639, 495,  78, 352, 126, 857, 956, 358, 619, 580,
+    124, 737, 594, 701, 612, 669, 112, 134, 694, 363, 992, 809, 743, 168, 974,
+    944, 375, 748,  52, 600, 747, 642, 182, 862,  81, 344, 805, 988, 739, 511,
+    655, 814, 334, 249, 515, 897, 955, 664, 981, 649, 113, 974, 459, 893, 228,
+    433, 837, 553, 268, 926, 240, 102, 654, 459, 51,  686, 754, 806, 760, 493,
+    403, 415, 394, 687, 700, 946, 670, 656, 610, 738, 392, 760, 799, 887, 653,
+    978, 321, 576, 617, 626, 502, 894, 679, 243, 440, 680, 879, 194, 572, 640,
+    724, 926,  56, 204, 700, 707, 151, 457, 449, 797, 195, 791, 558, 945, 679,
+    297,  59,  87, 824, 713, 663, 412, 693, 342, 606, 134, 108, 571, 364, 631,
+    212, 174, 643, 304, 329, 343,  97, 430, 751, 497, 314, 983, 374, 822, 928,
+    140, 206,  73, 263, 980, 736, 876, 478, 430, 305, 170, 514, 364, 692, 829,
+     82, 855, 953, 676, 246, 369, 970, 294, 750, 807, 827, 150, 790, 288, 923,
+    804, 378, 215, 828, 592, 281, 565, 555, 710,  82, 896, 831, 547, 261, 524,
+    462, 293, 465, 502,  56, 661, 821, 976, 991, 658, 869, 905, 758, 745, 193,
+    768, 550, 608, 933, 378, 286, 215, 979, 792, 961,  61, 688, 793, 644, 986,
+    403, 106, 366, 905, 644, 372, 567, 466, 434, 645, 210, 389, 550, 919, 135,
+    780, 773, 635, 389, 707, 100, 626, 958, 165, 504, 920, 176, 193, 713, 857,
+    265, 203,  50, 668, 108, 645, 990, 626, 197, 510, 357, 358, 850, 858, 364,
+    936, 638
+  ]
+  
+end

data/lib/rbzip2/crc.rb

@@ -0,0 +1,105 @@
+# This code is free software; you can redistribute it and/or modify it under
+# the terms of the new BSD License.
+#
+# Copyright (c) 2011, Sebastian Staudt
+
+class RBzip2::CRC
+
+  CRC32_TABLE = [
+    0x00000000, 0x04c11db7, 0x09823b6e, 0x0d4326d9,
+    0x130476dc, 0x17c56b6b, 0x1a864db2, 0x1e475005,
+    0x2608edb8, 0x22c9f00f, 0x2f8ad6d6, 0x2b4bcb61,
+    0x350c9b64, 0x31cd86d3, 0x3c8ea00a, 0x384fbdbd,
+    0x4c11db70, 0x48d0c6c7, 0x4593e01e, 0x4152fda9,
+    0x5f15adac, 0x5bd4b01b, 0x569796c2, 0x52568b75,
+    0x6a1936c8, 0x6ed82b7f, 0x639b0da6, 0x675a1011,
+    0x791d4014, 0x7ddc5da3, 0x709f7b7a, 0x745e66cd,
+    0x9823b6e0, 0x9ce2ab57, 0x91a18d8e, 0x95609039,
+    0x8b27c03c, 0x8fe6dd8b, 0x82a5fb52, 0x8664e6e5,
+    0xbe2b5b58, 0xbaea46ef, 0xb7a96036, 0xb3687d81,
+    0xad2f2d84, 0xa9ee3033, 0xa4ad16ea, 0xa06c0b5d,
+    0xd4326d90, 0xd0f37027, 0xddb056fe, 0xd9714b49,
+    0xc7361b4c, 0xc3f706fb, 0xceb42022, 0xca753d95,
+    0xf23a8028, 0xf6fb9d9f, 0xfbb8bb46, 0xff79a6f1,
+    0xe13ef6f4, 0xe5ffeb43, 0xe8bccd9a, 0xec7dd02d,
+    0x34867077, 0x30476dc0, 0x3d044b19, 0x39c556ae,
+    0x278206ab, 0x23431b1c, 0x2e003dc5, 0x2ac12072,
+    0x128e9dcf, 0x164f8078, 0x1b0ca6a1, 0x1fcdbb16,
+    0x018aeb13, 0x054bf6a4, 0x0808d07d, 0x0cc9cdca,
+    0x7897ab07, 0x7c56b6b0, 0x71159069, 0x75d48dde,
+    0x6b93dddb, 0x6f52c06c, 0x6211e6b5, 0x66d0fb02,
+    0x5e9f46bf, 0x5a5e5b08, 0x571d7dd1, 0x53dc6066,
+    0x4d9b3063, 0x495a2dd4, 0x44190b0d, 0x40d816ba,
+    0xaca5c697, 0xa864db20, 0xa527fdf9, 0xa1e6e04e,
+    0xbfa1b04b, 0xbb60adfc, 0xb6238b25, 0xb2e29692,
+    0x8aad2b2f, 0x8e6c3698, 0x832f1041, 0x87ee0df6,
+    0x99a95df3, 0x9d684044, 0x902b669d, 0x94ea7b2a,
+    0xe0b41de7, 0xe4750050, 0xe9362689, 0xedf73b3e,
+    0xf3b06b3b, 0xf771768c, 0xfa325055, 0xfef34de2,
+    0xc6bcf05f, 0xc27dede8, 0xcf3ecb31, 0xcbffd686,
+    0xd5b88683, 0xd1799b34, 0xdc3abded, 0xd8fba05a,
+    0x690ce0ee, 0x6dcdfd59, 0x608edb80, 0x644fc637,
+    0x7a089632, 0x7ec98b85, 0x738aad5c, 0x774bb0eb,
+    0x4f040d56, 0x4bc510e1, 0x46863638, 0x42472b8f,
+    0x5c007b8a, 0x58c1663d, 0x558240e4, 0x51435d53,
+    0x251d3b9e, 0x21dc2629, 0x2c9f00f0, 0x285e1d47,
+    0x36194d42, 0x32d850f5, 0x3f9b762c, 0x3b5a6b9b,
+    0x0315d626, 0x07d4cb91, 0x0a97ed48, 0x0e56f0ff,
+    0x1011a0fa, 0x14d0bd4d, 0x19939b94, 0x1d528623,
+    0xf12f560e, 0xf5ee4bb9, 0xf8ad6d60, 0xfc6c70d7,
+    0xe22b20d2, 0xe6ea3d65, 0xeba91bbc, 0xef68060b,
+    0xd727bbb6, 0xd3e6a601, 0xdea580d8, 0xda649d6f,
+    0xc423cd6a, 0xc0e2d0dd, 0xcda1f604, 0xc960ebb3,
+    0xbd3e8d7e, 0xb9ff90c9, 0xb4bcb610, 0xb07daba7,
+    0xae3afba2, 0xaafbe615, 0xa7b8c0cc, 0xa379dd7b,
+    0x9b3660c6, 0x9ff77d71, 0x92b45ba8, 0x9675461f,
+    0x8832161a, 0x8cf30bad, 0x81b02d74, 0x857130c3,
+    0x5d8a9099, 0x594b8d2e, 0x5408abf7, 0x50c9b640,
+    0x4e8ee645, 0x4a4ffbf2, 0x470cdd2b, 0x43cdc09c,
+    0x7b827d21, 0x7f436096, 0x7200464f, 0x76c15bf8,
+    0x68860bfd, 0x6c47164a, 0x61043093, 0x65c52d24,
+    0x119b4be9, 0x155a565e, 0x18197087, 0x1cd86d30,
+    0x029f3d35, 0x065e2082, 0x0b1d065b, 0x0fdc1bec,
+    0x3793a651, 0x3352bbe6, 0x3e119d3f, 0x3ad08088,
+    0x2497d08d, 0x2056cd3a, 0x2d15ebe3, 0x29d4f654,
+    0xc5a92679, 0xc1683bce, 0xcc2b1d17, 0xc8ea00a0,
+    0xd6ad50a5, 0xd26c4d12, 0xdf2f6bcb, 0xdbee767c,
+    0xe3a1cbc1, 0xe760d676, 0xea23f0af, 0xeee2ed18,
+    0xf0a5bd1d, 0xf464a0aa, 0xf9278673, 0xfde69bc4,
+    0x89b8fd09, 0x8d79e0be, 0x803ac667, 0x84fbdbd0,
+    0x9abc8bd5, 0x9e7d9662, 0x933eb0bb, 0x97ffad0c,
+    0xafb010b1, 0xab710d06, 0xa6322bdf, 0xa2f33668,
+    0xbcb4666d, 0xb8757bda, 0xb5365d03, 0xb1f740b4
+  ]
+
+  attr_accessor :global_crc
+
+  def initialize
+    initialize_crc
+  end
+
+  def initialize_crc
+    @global_crc = 0xffffffff
+  end
+
+  def final_crc
+    @global_crc ^ 0xffffffff
+  end
+
+  def update_crc(in_ch, repeat = -1)
+    if repeat == -1
+      temp = ((@global_crc >> 24) ^ in_ch) % 256
+      temp = 256 + temp if temp < 0
+      @global_crc = ((@global_crc << 8) ^ CRC32_TABLE[temp]) % 2**32
+    else
+      global_crc_shadow = @global_crc
+      repeat.downto(0) do
+        temp = ((global_crc_shadow >> 24) ^ in_ch) % 256
+        temp = 256 + temp if temp < 0
+        global_crc_shadow = (global_crc_shadow << 8) ^ CRC32_TABLE[temp]
+      end
+      @global_crc = global_crc_shadow % 2**32
+    end
+  end
+
+end

data/lib/rbzip2/data.rb

@@ -0,0 +1,51 @@
+# This code is free software; you can redistribute it and/or modify it under
+# the terms of the new BSD License.
+#
+# Copyright (c) 2011, Sebastian Staudt
+
+require 'rbzip2/constants'
+
+class RBzip2::Data
+
+  attr_reader :base, :cftab, :get_and_move_to_front_decode_yy, :in_use,
+              :limit, :ll8, :min_lens, :perm, :receive_decoding_tables_pos,
+              :selector, :selector_mtf, :seq_to_unseq, :temp_char_array_2d,
+              :unzftab, :tt
+
+  def initialize(block_size)
+    @in_use = Array.new 256, false
+
+    @seq_to_unseq = Array.new 256, 0
+    @selector = Array.new RBzip2::MAX_SELECTORS, 0
+    @selector_mtf = Array.new RBzip2::MAX_SELECTORS, 0
+
+    @unzftab = Array.new 256, 0
+
+    @base = Array.new RBzip2::N_GROUPS
+    RBzip2::N_GROUPS.times { |i| @base[i] = Array.new(RBzip2::MAX_ALPHA_SIZE, 0) }
+    @limit = Array.new RBzip2::N_GROUPS
+    RBzip2::N_GROUPS.times { |i| @limit[i] = Array.new(RBzip2::MAX_ALPHA_SIZE, 0) }
+    @perm = Array.new RBzip2::N_GROUPS
+    RBzip2:: N_GROUPS.times { |i| @perm[i] = Array.new(RBzip2::MAX_ALPHA_SIZE, 0) }
+    @min_lens = Array.new RBzip2::N_GROUPS, 0
+
+    @cftab = Array.new 257, 0
+    @get_and_move_to_front_decode_yy = Array.new 256
+    @temp_char_array_2d = Array.new RBzip2::N_GROUPS
+    RBzip2::N_GROUPS.times { |i| @temp_char_array_2d[i] = Array.new(RBzip2::MAX_ALPHA_SIZE, 0) }
+    @receive_decoding_tables_pos = Array.new RBzip2::N_GROUPS, 0
+
+    @ll8 = Array.new block_size * RBzip2::BASEBLOCKSIZE
+  end
+
+  def init_tt(size)
+    tt_shadow = @tt
+
+    if tt_shadow.nil? || tt_shadow.size < size
+      @tt = tt_shadow = Array.new(size)
+    end
+
+    tt_shadow
+  end
+
+end

data/lib/rbzip2/decompressor.rb

@@ -0,0 +1,684 @@
+# This code is free software; you can redistribute it and/or modify it under
+# the terms of the new BSD License.
+#
+# Copyright (c) 2011, Sebastian Staudt
+
+require 'stringio'
+
+require 'rbzip2/crc'
+require 'rbzip2/constants'
+require 'rbzip2/data'
+
+module RBzip2::Decompressor
+
+  def count(read)
+    @bytes_read += read if read != -1
+  end
+
+  def read(length = uncompressed)
+    raise 'stream closed' if @io.nil?
+
+    if length == 1
+      r = read0
+      count (r < 0 ? -1 : 1)
+      r
+    elsif length > 0
+      r = StringIO.new
+      length.times do
+        b = read0
+        break if b < 0
+        r.write b.chr
+      end
+      count r.size
+      r.string
+    end
+  end
+
+  def read0
+    ret_char = @current_char
+
+    case @current_state
+      when RBzip2::EOF
+        return -1
+
+      when RBzip2::RAND_PART_B_STATE
+        setup_rand_part_b
+
+      when RBzip2::RAND_PART_C_STATE
+        setup_rand_part_c
+
+      when RBzip2::NO_RAND_PART_B_STATE
+        setup_no_rand_part_b
+
+      when RBzip2::NO_RAND_PART_C_STATE
+        setup_no_rand_part_c
+
+      when RBzip2::START_BLOCK_STATE
+      when RBzip2::RAND_PART_A_STATE
+      when RBzip2::NO_RAND_PART_A_STATE
+      else
+        raise 'illegal state'
+    end
+
+    ret_char
+  end
+
+  def make_maps
+    in_use = @data.in_use
+    seq_to_unseq = @data.seq_to_unseq;
+
+    n_in_use_shadow = 0;
+
+    256.times do |i|
+      if in_use[i]
+        seq_to_unseq[n_in_use_shadow] = i
+        n_in_use_shadow += 1
+      end
+    end
+
+    @n_in_use = n_in_use_shadow;
+  end
+
+  def init
+    check_magic
+
+    block_size = @io.read(1).to_i
+    raise 'Illegal block size.' if block_size < 1 || block_size > 9
+    @block_size = block_size
+
+    init_block
+    setup_block
+  end
+
+  def check_magic
+    raise 'Magic number does not match "BZh".' unless @io.read(3) == 'BZh'
+  end
+
+  def init_block
+    magic = [ubyte, ubyte, ubyte, ubyte, ubyte, ubyte]
+
+    if magic == [0x17, 0x72, 0x45, 0x38, 0x50, 0x90]
+      complete
+    elsif magic != [0x31, 0x41, 0x59, 0x26, 0x53, 0x59]
+      @current_state = RBzip2::EOF
+
+      raise 'Bad block header.'
+    else
+      @stored_block_crc = int
+      @block_randomised = bit
+
+      @data = RBzip2::Data.new @block_size if @data.nil?
+
+      get_and_move_to_front_decode
+
+      @crc.initialize_crc
+      @current_state = RBzip2::START_BLOCK_STATE
+    end
+  end
+
+  def end_block
+    @computed_block_crc = @crc.final_crc
+
+    if @stored_block_crc != @computed_block_crc
+      @computed_combined_crc = (@stored_combined_crc << 1) | (@stored_combined_crc >> 31)
+      @computed_combined_crc ^= @stored_block_crc
+
+      raise 'BZip2 CRC error'
+    end
+
+    @computed_combined_crc = (@computed_combined_crc << 1) | (@computed_combined_crc >> 31)
+    @computed_combined_crc ^= @computed_block_crc
+  end
+
+  def complete
+    @stored_combined_crc = int
+    @current_state = RBzip2::EOF
+    @data = nil
+
+    raise 'BZip2 CRC error' if @stored_combined_crc != @computed_combined_crc
+  end
+
+  def close
+    if @io != $stdin
+      @io = nil
+      @data = nil
+    end
+  end
+
+  def r(n)
+    live_shadow = @live
+    buff_shadow = @buff
+
+    if live_shadow < n
+      begin
+        thech = @io.read(1)[0].ord
+
+        raise 'unexpected end of stream' if thech < 0
+
+        buff_shadow = (buff_shadow << 8) | thech
+        live_shadow += 8
+      end while live_shadow < n
+
+      @buff = buff_shadow
+    end
+
+    @live = live_shadow - n
+
+    (buff_shadow >> (live_shadow - n)) & ((1 << n) - 1)
+  end
+
+  def bit
+    r(1) != 0
+  end
+
+  def ubyte
+    r 8
+  end
+
+  def int
+    (((((r(8) << 8) | r(8)) << 8) | r(8)) << 8) | r(8)
+  end
+
+  def create_decode_tables(limit, base, perm, length, min_len, max_len, alpha_size)
+    pp = 0
+    (min_len..max_len).each do |i|
+      alpha_size.times do |j|
+        if length[j] == i
+          perm[pp] = j
+          pp += 1
+        end
+      end
+    end
+
+    RBzip2::MAX_CODE_LEN.downto 1 do |i|
+      base[i] = 0
+      limit[i] = 0
+    end
+
+    alpha_size.times do |i|
+      base[length[i] + 1] += 1
+    end
+
+    b = 0
+    1.upto(RBzip2::MAX_CODE_LEN - 1) do |i|
+      b += base[i]
+      base[i] = b
+    end
+
+    vec = 0
+    min_len.upto(max_len) do |i|
+      b = base[i]
+      nb = base[i + 1]
+      vec += nb - b
+      b = nb
+      limit[i] = vec -1
+      vec = vec << 1
+    end
+
+    (min_len + 1).upto(max_len) do |i|
+      base[i] = ((limit[i - 1] + 1) << 1) - base[i]
+    end
+  end
+
+  def receive_decoding_tables
+    in_use = @data.in_use
+    pos = @data.receive_decoding_tables_pos
+    selector = @data.selector
+    selector_mtf = @data.selector_mtf
+
+    in_use16 = 0
+
+    16.times do |i|
+      in_use16 |= 1 << i if bit
+    end
+
+    255.downto(0) do |i|
+      in_use[i] = false
+    end
+
+    16.times do |i|
+       if (in_use16 & (1 << i)) != 0
+        i16 = i << 4
+        16.times do |j|
+          in_use[i16 + j] = true if bit
+        end
+      end
+    end
+
+    make_maps
+    alpha_size = @n_in_use + 2
+
+    groups = r 3
+    selectors = r 15
+
+    selectors.times do |i|
+      j = 0
+      while bit
+        j += 1
+      end
+      selector_mtf[i] = j
+    end
+
+    groups.downto(0) do |v|
+      pos[v] = v
+    end
+
+    selectors.times do |i|
+      v = selector_mtf[i] & 0xff
+      tmp = pos[v]
+
+      while v > 0 do
+        pos[v] = pos[v -= 1]
+      end
+
+      pos[0] = tmp
+      selector[i] = tmp
+    end
+
+    len = @data.temp_char_array_2d
+
+    groups.times do |t|
+      curr = r 5
+      len_t = len[t]
+      alpha_size.times do |i|
+        while bit
+          curr += bit ? -1 : 1
+        end
+        len_t[i] = curr
+      end
+      @data.temp_char_array_2d[t] = len_t
+    end
+
+    create_huffman_decoding_tables alpha_size, groups
+  end
+
+  def create_huffman_decoding_tables(alpha_size, groups)
+    len = @data.temp_char_array_2d
+    min_lens = @data.min_lens
+    limit = @data.limit
+    base = @data.base
+    perm = @data.perm
+
+    groups.times do |t|
+      min_len = 32
+      max_len = 0
+      len_t = len[t]
+
+      (alpha_size - 1).downto 0 do |i|
+        lent = len_t[i]
+        max_len = lent if lent > max_len
+        min_len = lent if lent < min_len
+      end
+
+      create_decode_tables limit[t], base[t], perm[t], len[t], min_len, max_len, alpha_size
+      min_lens[t] = min_len
+    end
+  end
+
+  def get_and_move_to_front_decode
+    @orig_ptr = r 24
+    receive_decoding_tables
+
+    ll8 = @data.ll8
+    unzftab = @data.unzftab
+    selector = @data.selector
+    seq_to_unseq = @data.seq_to_unseq
+    yy = @data.get_and_move_to_front_decode_yy
+    min_lens = @data.min_lens
+    limit = @data.limit
+    base = @data.base
+    perm = @data.perm
+    limit_last = @block_size * RBzip2::BASEBLOCKSIZE
+
+    256.downto(0) do |i|
+      yy[i] = i
+      unzftab[i] = 0
+    end
+
+    group_no = 0
+    group_pos = RBzip2::G_SIZE - 1
+    eob = @n_in_use + 1
+    next_sym = get_and_move_to_front_decode0 0
+    buff_shadow = @buff
+    live_shadow = @live
+    last_shadow = -1
+    zt = selector[group_no] & 0xff
+    base_zt = base[zt]
+    limit_zt = limit[zt]
+    perm_zt = perm[zt]
+    min_lens_zt = min_lens[zt]
+
+    while next_sym != eob
+      if (next_sym == RBzip2::RUNA) || (next_sym == RBzip2::RUNB)
+        s = -1
+
+        n = 1
+        loop do
+          if next_sym == RBzip2::RUNA
+            s += n
+          elsif next_sym == RBzip2::RUNB
+            s += n << 1
+          else
+            break
+          end
+
+          if group_pos == 0
+            group_pos = RBzip2::G_SIZE - 1
+            group_no += 1
+            zt = selector[group_no] & 0xff
+            base_zt = base[zt]
+            limit_zt = limit[zt]
+            perm_zt = perm[zt]
+            min_lens_zt = min_lens[zt]
+          else
+            group_pos -= 1
+          end
+
+          zn = min_lens_zt
+
+          while live_shadow < zn
+            thech = @io.read(1)[0].ord
+
+            raise 'unexpected end of stream' if thech < 0
+
+            buff_shadow = (buff_shadow << 8) | thech
+            live_shadow += 8
+          end
+
+          zvec = (buff_shadow >> (live_shadow - zn)) & ((1 << zn) - 1)
+          live_shadow -= zn
+
+          while zvec > limit_zt[zn]
+            zn += 1
+
+            while live_shadow < 1
+              thech = @io.read(1)[0].ord
+
+              raise 'unexpected end of stream' if thech < 0
+
+              buff_shadow = (buff_shadow << 8) | thech
+              live_shadow += 8
+            end
+
+            live_shadow -= 1
+            zvec = (zvec << 1) | ((buff_shadow >> live_shadow) & 1)
+          end
+
+          next_sym = perm_zt[zvec - base_zt[zn]]
+
+          n = n << 1
+        end
+
+        ch = seq_to_unseq[yy[0]]
+        unzftab[ch & 0xff] += s + 1
+
+        while s >= 0
+          last_shadow += 1
+          ll8[last_shadow] = ch
+          s -= 1
+        end
+
+        raise 'block overrun' if last_shadow >= limit_last
+      else
+        last_shadow += 1
+        raise 'block overrun' if last_shadow >= limit_last
+
+        tmp = yy[next_sym - 1]
+        unzftab[seq_to_unseq[tmp] & 0xff] += 1
+        ll8[last_shadow] = seq_to_unseq[tmp]
+
+        yy[1, next_sym - 1] = yy[0, next_sym - 1]
+        yy[0] = tmp
+
+        if group_pos == 0
+          group_pos = RBzip2::G_SIZE - 1
+          group_no += 1
+          zt = selector[group_no] & 0xff
+          base_zt = base[zt]
+          limit_zt = limit[zt]
+          perm_zt = perm[zt]
+          min_lens_zt = min_lens[zt]
+        else
+          group_pos -= 1
+        end
+
+        zn = min_lens_zt
+
+        while live_shadow < zn
+          thech = @io.read(1)[0].ord
+
+          raise 'unexpected end of stream' if thech < 0
+
+          buff_shadow = (buff_shadow << 8) | thech
+          live_shadow += 8
+        end
+        zvec = (buff_shadow >> (live_shadow - zn)) & ((1 << zn) - 1)
+        live_shadow -= zn
+
+        while zvec > limit_zt[zn]
+          zn += 1
+          while live_shadow < 1
+            thech = @io.read(1)[0].ord
+
+            raise 'unexpected end of stream' if thech < 0
+
+            buff_shadow = (buff_shadow << 8) | thech
+            live_shadow += 8
+          end
+          live_shadow -= 1
+          zvec = (zvec << 1) | ((buff_shadow >> live_shadow) & 1)
+        end
+
+        next_sym = perm_zt[zvec - base_zt[zn]]
+      end
+    end
+
+    @last = last_shadow
+    @live = live_shadow
+    @buff = buff_shadow
+  end
+
+  def get_and_move_to_front_decode0(group_no)
+    zt = @data.selector[group_no] & 0xff
+    limit_zt = @data.limit[zt]
+    zn = @data.min_lens[zt]
+    zvec = r zn
+    live_shadow = @live
+    buff_shadow = @buff
+
+    while zvec > limit_zt[zn]
+      zn += 1
+
+      while live_shadow < 1
+        thech = @io.read(1)[0].ord
+
+        raise 'unexpected end of stream' if thech < 0
+
+        buff_shadow = (buff_shadow << 8) | thech
+        live_shadow += 8
+      end
+
+      live_shadow -=1
+      zvec = (zvec << 1) | ((buff_shadow >> live_shadow) & 1)
+    end
+
+    @live = live_shadow
+    @buff = buff_shadow
+
+    @data.perm[zt][zvec - @data.base[zt][zn]]
+  end
+
+  def setup_block
+    return if @data.nil?
+
+    cftab = @data.cftab
+    tt = @data.init_tt @last + 1
+    ll8 = @data.ll8
+    cftab[0] = 0
+    cftab[1, 256] = @data.unzftab[0, 256]
+
+    c = cftab[0]
+    1.upto(256) do |i|
+      c += cftab[i]
+      cftab[i] = c
+    end
+
+    last_shadow = @last
+    (last_shadow + 1).times do |i|
+      cftab_i = ll8[i] & 0xff
+      tt[cftab[cftab_i]] = i
+      cftab[cftab_i] += 1
+    end
+
+    raise 'stream corrupted' if @orig_ptr < 0 || @orig_ptr >= tt.size
+
+    @su_t_pos = tt[@orig_ptr]
+    @su_count = 0
+    @su_i2 = 0
+    @su_ch2 = 256
+
+    if @block_randomised
+      @su_r_n_to_go = 0
+      @su_r_t_pos = 0
+
+      setup_rand_part_a
+    else
+      setup_no_rand_part_a
+    end
+  end
+
+  def setup_rand_part_a
+    if @su_i2 <= @last
+      @su_ch_prev = @su_ch2
+      su_ch2_shadow = @data.ll8[@su_t_pos] & 0xff
+      @su_t_pos = @data.tt[@su_t_pos]
+
+      if @su_r_n_to_go == 0
+        @su_r_n_to_go = RBzip2::RNUMS[@su_r_t_pos] - 1
+        @su_r_t_pos += 1
+        @su_r_t_pos = 0 if @su_r_t_pos == 512
+      else
+        @su_r_n_to_go -= 1
+      end
+
+      @su_ch2 = su_ch2_shadow ^= (@su_r_n_to_go == 1) ? 1 : 0
+      @su_i2 += 1
+      @current_char = su_ch2_shadow
+      @current_state = RBzip2::RAND_PART_B_STATE
+      @crc.update_crc su_ch2_shadow
+    else
+      end_block
+      init_block
+      setup_block
+    end
+  end
+
+  def setup_no_rand_part_a
+    if @su_i2 <= @last
+      @su_ch_prev = @su_ch2
+      su_ch2_shadow = @data.ll8[@su_t_pos] & 0xff
+      @su_ch2 = su_ch2_shadow
+      @su_t_pos = @data.tt[@su_t_pos]
+      @su_i2 += 1
+      @current_char = su_ch2_shadow
+      @current_state = RBzip2::NO_RAND_PART_B_STATE
+      @crc.update_crc su_ch2_shadow
+    else
+      @current_state = RBzip2::NO_RAND_PART_A_STATE
+      end_block
+      init_block
+      setup_block
+    end
+  end
+
+  def setup_rand_part_b
+    if @su_ch2 != @su_ch_prev
+      @current_state = RBzip2::RAND_PART_A_STATE
+      @su_count = 1
+      setup_rand_part_a
+    else
+      @su_count += 1
+      if @su_count >= 4
+        @su_z = @data.ll8[@su_t_pos] & 0xff
+        @su_t_pos = @data.tt[@su_t_pos]
+
+        if @su_r_n_to_go == 0
+          @su_r_n_to_go = RBzip2::RNUMS[@su_r_t_pos] - 1
+          @su_r_t_pos += 1
+          @su_r_t_pos = 0 if @su_r_t_pos == 512
+        else
+          @su_r_n_to_go -= 1
+        end
+
+        @su_j2 = 0
+        @current_state = RBzip2::RAND_PART_C_STATE
+        @su_z ^= 1 if @su_r_n_to_go == 1
+        setup_rand_part_c
+      else
+        @current_state = RBzip2::RAND_PART_A_STATE
+        setup_rand_part_a
+      end
+    end
+  end
+
+  def setup_rand_part_c
+    if @su_j2 < @su_z
+      @current_char = @su_ch2
+      @crc.update_crc @su_ch2
+      @su_j2 += 1
+    else
+      @current_state = RBzip2::RAND_PART_A_STATE
+      @su_i2 += 1
+      @su_count = 0
+      setup_rand_part_a
+    end
+  end
+
+  def setup_no_rand_part_b
+    if @su_ch2 != @su_ch_prev
+      @su_count = 1
+      setup_no_rand_part_a
+    else
+      @su_count += 1
+      if @su_count >= 4
+        @su_z = @data.ll8[@su_t_pos] & 0xff
+        @su_t_pos = @data.tt[@su_t_pos]
+        @su_j2 = 0
+        setup_no_rand_part_c
+      else
+        setup_no_rand_part_a
+      end
+    end
+  end
+
+  def setup_no_rand_part_c
+    if @su_j2 < @su_z
+      su_ch2_shadow = @su_ch2
+      @current_char = su_ch2_shadow
+      @crc.update_crc su_ch2_shadow
+      @su_j2 += 1
+      @current_state = RBzip2::NO_RAND_PART_C_STATE
+    else
+      @su_i2 += 1
+      @su_count = 0
+      setup_no_rand_part_a
+    end
+  end
+
+  def size
+    if @io.is_a? StringIO
+      @io.size
+    elsif @io.is_a? File
+      @io.stat.size
+    end
+  end
+
+  def uncompressed
+    @last + 1
+  end
+
+  def inspect
+    "#<#{self.class}: @io=#{@io.inspect} size=#{size} uncompressed=#{uncompressed}>"
+  end
+
+end

data/lib/rbzip2/io.rb

@@ -0,0 +1,25 @@
+# This code is free software; you can redistribute it and/or modify it under
+# the terms of the new BSD License.
+#
+# Copyright (c) 2011, Sebastian Staudt
+
+require 'rbzip2/decompressor'
+
+class RBzip2::IO
+
+  include RBzip2::Decompressor
+
+  def initialize(io)
+    @buff = 0
+    @bytes_read = 0
+    @computed_combined_crc = 0
+    @crc = RBzip2::CRC.new
+    @current_char = -1
+    @io = io
+    @live = 0
+    @stored_combined_crc = 0
+    @su_t_pos = 0
+    init
+  end
+
+end

data/lib/rbzip2/version.rb

@@ -0,0 +1,10 @@
+# This code is free software; you can redistribute it and/or modify it under
+# the terms of the new BSD License.
+#
+# Copyright (c) 2011, Sebastian Staudt
+
+module RBzip2
+
+  VERSION = '0.1.0' unless const_defined? :VERSION
+
+end

data/lib/rbzip2.rb

@@ -0,0 +1,14 @@
+# This code is free software; you can redistribute it and/or modify it under
+# the terms of the new BSD License.
+#
+# Copyright (c) 2011, Sebastian Staudt
+
+module RBzip2
+
+  autoload :CRC,          'rbzip2/crc'
+  autoload :Data,         'rbzip2/data'
+  autoload :Decompressor, 'rbzip2/decompressor'
+  autoload :IO,           'rbzip2/io'
+  autoload :VERSION,      'rbzip2/version'
+
+end

data/rbzip2.gemspec

@@ -0,0 +1,27 @@
+# This code is free software; you can redistribute it and/or modify it under
+# the terms of the new BSD License.
+#
+# Copyright (c) 2011, Sebastian Staudt
+
+require File.expand_path(File.dirname(__FILE__) + '/lib/rbzip2/version')
+
+Gem::Specification.new do |s|
+  s.name        = 'rbzip2'
+  s.version     = RBzip2::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = [ 'Sebastian Staudt' ]
+  s.email       = [ 'koraktor@gmail.com' ]
+  s.homepage    = 'https://github.com/koraktor/rbzip2'
+  s.summary     = 'Pure Ruby impementation of bzip2'
+  s.description = 'A pure Ruby implementation of the bzip2 compression algorithm.'
+
+  s.add_development_dependency 'mocha', '~> 0.10.0'
+  s.add_development_dependency 'rake', '~> 0.9.2'
+  s.add_development_dependency 'rspec-core', '~> 2.7.1'
+  s.add_development_dependency 'rspec-expectations', '~> 2.7.0'
+  s.add_development_dependency 'yard', '~> 0.7.3'
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- spec/*`.split("\n")
+  s.require_paths = [ 'lib' ]
+end

data/spec/fixtures/test.txt

@@ -0,0 +1,11 @@
+This is a test fixture for RBzip2.
+Its contents will be compressed and decompressed to test the functionality.
+
+Lorem ipsum dolor sit amet, consetetur sadipscing elitr, sed diam nonumy eirmod
+tempor invidunt ut labore et dolore magna aliquyam erat, sed diam voluptua. At
+vero eos et accusam et justo duo dolores et ea rebum. Stet clita kasd
+gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet. Lorem ipsum
+dolor sit amet, consetetur sadipscing elitr, sed diam nonumy eirmod tempor
+invidunt ut labore et dolore magna aliquyam erat, sed diam voluptua. At vero
+eos et accusam et justo duo dolores et ea rebum. Stet clita kasd gubergren, no
+sea takimata sanctus est Lorem ipsum dolor sit amet.

data/spec/helper.rb

@@ -0,0 +1,12 @@
+require 'rspec/core'
+require 'rspec/expectations'
+
+require 'rbzip2'
+
+include RBzip2
+
+RSpec.configure do |config|
+  config.color_enabled = true
+  config.extend RBzip2
+  config.mock_framework = :mocha
+end

data/spec/io_spec.rb

@@ -0,0 +1,37 @@
+# This code is free software; you can redistribute it and/or modify it under
+# the terms of the new BSD License.
+#
+# Copyright (c) 2011, Sebastian Staudt
+
+require 'helper'
+
+describe RBzip2::IO do
+
+  before do
+    @txt_file = File.new File.join(File.dirname(__FILE__), 'fixtures/test.txt')
+    bz2_file  = File.new File.join(File.dirname(__FILE__), 'fixtures/test.bz2')
+    @bz2_io   = RBzip2::IO.new bz2_file
+  end
+
+  it 'allows decompressing data' do
+    RBzip2::IO.should include(Decompressor)
+  end
+
+  it 'acts like a standard IO' do
+    methods = RBzip2::IO.instance_methods.map { |m| m.to_sym }
+    methods.should include(:read, :close)
+  end
+
+  it 'knows its size' do
+    @bz2_io.size.should be(375)
+  end
+
+  it 'knows the size of the uncompressed data' do
+    @bz2_io.uncompressed.should be(704)
+  end
+
+  it 'should be able to decompress compressed data' do
+    @bz2_io.read.should eq(@txt_file.read)
+  end
+
+end

metadata

@@ -0,0 +1,166 @@
+--- !ruby/object:Gem::Specification 
+name: rbzip2
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: 
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Sebastian Staudt
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-11-09 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  type: :development
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 55
+        segments: 
+        - 0
+        - 10
+        - 0
+        version: 0.10.0
+  prerelease: false
+  name: mocha
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  type: :development
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 63
+        segments: 
+        - 0
+        - 9
+        - 2
+        version: 0.9.2
+  prerelease: false
+  name: rake
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  type: :development
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 17
+        segments: 
+        - 2
+        - 7
+        - 1
+        version: 2.7.1
+  prerelease: false
+  name: rspec-core
+  version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  type: :development
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 19
+        segments: 
+        - 2
+        - 7
+        - 0
+        version: 2.7.0
+  prerelease: false
+  name: rspec-expectations
+  version_requirements: *id004
+- !ruby/object:Gem::Dependency 
+  type: :development
+  requirement: &id005 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 5
+        segments: 
+        - 0
+        - 7
+        - 3
+        version: 0.7.3
+  prerelease: false
+  name: yard
+  version_requirements: *id005
+description: A pure Ruby implementation of the bzip2 compression algorithm.
+email: 
+- koraktor@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- .travis.yml
+- Gemfile
+- Gemfile.lock
+- LICENSE
+- README.md
+- Rakefile
+- lib/rbzip2.rb
+- lib/rbzip2/constants.rb
+- lib/rbzip2/crc.rb
+- lib/rbzip2/data.rb
+- lib/rbzip2/decompressor.rb
+- lib/rbzip2/io.rb
+- lib/rbzip2/version.rb
+- rbzip2.gemspec
+- spec/fixtures/test.bz2
+- spec/fixtures/test.txt
+- spec/helper.rb
+- spec/io_spec.rb
+homepage: https://github.com/koraktor/rbzip2
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.8.11
+signing_key: 
+specification_version: 3
+summary: Pure Ruby impementation of bzip2
+test_files: 
+- spec/fixtures/test.bz2
+- spec/fixtures/test.txt
+- spec/helper.rb
+- spec/io_spec.rb