origami 1.0.2

data/origami/filters/crypt.rb

@@ -0,0 +1,47 @@
+=begin
+
+= File
+	filters/crypt.rb
+
+= Info
+	This file is part of Origami, PDF manipulation framework for Ruby
+	Copyright (C) 2010	Guillaume Delugré <guillaume@security-labs.org>
+	All right reserved.
+	
+  Origami is free software: you can redistribute it and/or modify
+  it under the terms of the GNU Lesser General Public License as published by
+  the Free Software Foundation, either version 3 of the License, or
+  (at your option) any later version.
+
+  Origami is distributed in the hope that it will be useful,
+  but WITHOUT ANY WARRANTY; without even the implied warranty of
+  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+  GNU Lesser General Public License for more details.
+
+  You should have received a copy of the GNU Lesser General Public License
+  along with Origami.  If not, see <http://www.gnu.org/licenses/>.
+
+=end
+
+module Origami
+
+  module Filter
+
+    #
+    # Class representing a Crypt Filter.
+    # TODO.
+    #
+    class Crypt
+      include Filter
+
+      class DecodeParms < Dictionary
+        include StandardObject
+
+        field   :Type,      :Type => Name, :Default => :Crypt
+        field   :Name,      :Type => Name, :Default => :Identity
+      end
+    end
+    
+  end
+end
+

data/origami/filters/dct.rb

@@ -0,0 +1,61 @@
+=begin
+
+= File
+	filters/dct.rb
+
+= Info
+	This file is part of Origami, PDF manipulation framework for Ruby
+	Copyright (C) 2010	Guillaume Delugré <guillaume@security-labs.org>
+	All right reserved.
+	
+  Origami is free software: you can redistribute it and/or modify
+  it under the terms of the GNU Lesser General Public License as published by
+  the Free Software Foundation, either version 3 of the License, or
+  (at your option) any later version.
+
+  Origami is distributed in the hope that it will be useful,
+  but WITHOUT ANY WARRANTY; without even the implied warranty of
+  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+  GNU Lesser General Public License for more details.
+
+  You should have received a copy of the GNU Lesser General Public License
+  along with Origami.  If not, see <http://www.gnu.org/licenses/>.
+
+=end
+
+module Origami
+  
+  module Filter
+
+    #
+    # Class representing a Filter used to encode and decode data with DCT (JPEG) compression algorithm.
+    #
+    class DCT
+      include Filter
+      
+      class DecodeParms < Dictionary
+        include StandardObject
+
+        field   :ColorTransform,    :Type => Integer
+      end
+
+      def initialize(parameters = {})
+        super(DecodeParms.new(parameters))
+      end
+
+      def encode(stream)
+        stream
+      end
+      
+      #
+      # DCTDecode implies that data is a JPEG image container.
+      # Just returns the raw JPEG image as is.
+      #
+      def decode(stream)
+        stream
+      end
+    end
+    
+  end
+end
+

data/origami/filters/flate.rb

@@ -0,0 +1,112 @@
+=begin
+
+= File
+	filters/flate.rb
+
+= Info
+	This file is part of Origami, PDF manipulation framework for Ruby
+	Copyright (C) 2010	Guillaume Delugré <guillaume@security-labs.org>
+	All right reserved.
+	
+  Origami is free software: you can redistribute it and/or modify
+  it under the terms of the GNU Lesser General Public License as published by
+  the Free Software Foundation, either version 3 of the License, or
+  (at your option) any later version.
+
+  Origami is distributed in the hope that it will be useful,
+  but WITHOUT ANY WARRANTY; without even the implied warranty of
+  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+  GNU Lesser General Public License for more details.
+
+  You should have received a copy of the GNU Lesser General Public License
+  along with Origami.  If not, see <http://www.gnu.org/licenses/>.
+
+=end
+
+require 'zlib'
+require 'origami/filters/predictors'
+
+module Origami
+  
+  module Filter
+
+    class InvalidFlateDataError < Exception #:nodoc:
+      attr_reader :zlib_stream, :zlib_except
+
+      def initialize(zlib_stream, zlib_except)
+        super(zlib_except.message)
+
+        @zlib_stream = zlib_stream
+        @zlib_except = zlib_except
+      end
+    end
+    
+    #
+    # Class representing a Filter used to encode and decode data with zlib/Flate compression algorithm.
+    #
+    class Flate
+      include Filter
+      
+      EOD = 257 #:nodoc:
+ 
+      class DecodeParms < Dictionary
+        include StandardObject
+
+        field   :Predictor,         :Type => Integer, :Default => 1
+        field   :Colors,            :Type => Integer, :Default => 1
+        field   :BitsPerComponent,  :Type => Integer, :Default => 8
+        field   :Columns,           :Type => Integer, :Default => 1
+      end
+      
+      #
+      # Create a new Flate Filter.
+      # _parameters_:: A hash of filter options (ignored).
+      #
+      def initialize(parameters = {})
+        super(DecodeParms.new(parameters))
+      end
+      
+      #
+      # Encodes data using zlib/Deflate compression method.
+      # _stream_:: The data to encode.
+      #
+      def encode(stream)
+        if @params.Predictor.is_a?(Integer)
+          colors  = @params.Colors.is_a?(Integer) ? @params.Colors.to_i : 1
+          bpc     = @params.BitsPerComponent.is_a?(Integer) ? @params.BitsPerComponent.to_i : 8
+          columns = @params.Columns.is_a?(Integer) ? @params.Columns.to_i : 1
+
+          stream = Predictor.do_pre_prediction(stream, @params.Predictor.to_i, colors, bpc, columns)
+        end       
+        
+        Zlib::Deflate.deflate(stream, Zlib::BEST_COMPRESSION)
+      end
+      
+      #
+      # Decodes data using zlib/Inflate decompression method.
+      # _stream_:: The data to decode.
+      #
+      def decode(stream)
+        
+        zlib_stream = Zlib::Inflate.new
+        begin
+          uncompressed = zlib_stream.inflate(stream)
+        rescue Zlib::DataError => zlib_except
+          raise InvalidFlateDataError.new(zlib_stream, zlib_except)
+        end
+
+        if @params.Predictor.is_a?(Integer)
+          colors  = @params.Colors.is_a?(Integer) ? @params.Colors.to_i : 1
+          bpc     = @params.BitsPerComponent.is_a?(Integer) ? @params.BitsPerComponent.to_i : 8
+          columns = @params.Columns.is_a?(Integer) ? @params.Columns.to_i : 1
+
+          uncompressed = Predictor.do_post_prediction(uncompressed, @params.Predictor.to_i, colors, bpc, columns)
+        end
+
+        uncompressed
+      end
+
+    end
+  end
+end
+

data/origami/filters/jbig2.rb

@@ -0,0 +1,63 @@
+=begin
+
+= File
+	filters/jbig2.rb
+
+= Info
+	This file is part of Origami, PDF manipulation framework for Ruby
+	Copyright (C) 2010	Guillaume Delugré <guillaume@security-labs.org>
+	All right reserved.
+	
+  Origami is free software: you can redistribute it and/or modify
+  it under the terms of the GNU Lesser General Public License as published by
+  the Free Software Foundation, either version 3 of the License, or
+  (at your option) any later version.
+
+  Origami is distributed in the hope that it will be useful,
+  but WITHOUT ANY WARRANTY; without even the implied warranty of
+  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+  GNU Lesser General Public License for more details.
+
+  You should have received a copy of the GNU Lesser General Public License
+  along with Origami.  If not, see <http://www.gnu.org/licenses/>.
+
+=end
+
+module Origami
+  
+  module Filter
+
+    #
+    # Class representing a Filter used to encode and decode data with JBIG2 compression algorithm.
+    #
+    class JBIG2
+      include Filter
+     
+      class DecodeParms < Dictionary
+        include StandardObject
+
+        field   :JBIG2Globals,    :Type => Stream
+      end
+
+      def initialize(parameters = {})
+        super(DecodeParms.new(parameters))
+      end
+
+      #
+      # Not supported.
+      #
+      def encode(stream)
+        raise NotImplementedError, "#{self.class} is not yet supported"
+      end
+      
+      #
+      # Not supported.
+      #
+      def decode(stream)
+        raise NotImplementedError, "#{self.class} is not yet supported"
+      end
+    end
+    
+  end
+end
+

data/origami/filters/jpx.rb

@@ -0,0 +1,53 @@
+=begin
+
+= File
+	filters/jpx.rb
+
+= Info
+	This file is part of Origami, PDF manipulation framework for Ruby
+	Copyright (C) 2010	Guillaume Delugré <guillaume@security-labs.org>
+	All right reserved.
+	
+  Origami is free software: you can redistribute it and/or modify
+  it under the terms of the GNU Lesser General Public License as published by
+  the Free Software Foundation, either version 3 of the License, or
+  (at your option) any later version.
+
+  Origami is distributed in the hope that it will be useful,
+  but WITHOUT ANY WARRANTY; without even the implied warranty of
+  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+  GNU Lesser General Public License for more details.
+
+  You should have received a copy of the GNU Lesser General Public License
+  along with Origami.  If not, see <http://www.gnu.org/licenses/>.
+
+=end
+
+module Origami
+
+  module Filter
+
+    #
+    # Class representing a Filter used to encode and decode data with JPX compression algorithm.
+    #
+    class JPX
+      include Filter
+      
+      #
+      # Not supported.
+      #
+      def encode(stream)
+        raise NotImplementedError, "#{self.class} is not yet supported"
+      end
+      
+      #
+      # Not supported.
+      #
+      def decode(stream)
+        raise NotImplementedError, "#{self.class} is not yet supported"
+      end
+    end
+
+  end
+end
+

data/origami/filters/lzw.rb

@@ -0,0 +1,195 @@
+=begin
+
+= File
+	filters/lzw.rb
+
+= Info
+	This file is part of Origami, PDF manipulation framework for Ruby
+	Copyright (C) 2010	Guillaume Delugré <guillaume@security-labs.org>
+	All right reserved.
+	
+  Origami is free software: you can redistribute it and/or modify
+  it under the terms of the GNU Lesser General Public License as published by
+  the Free Software Foundation, either version 3 of the License, or
+  (at your option) any later version.
+
+  Origami is distributed in the hope that it will be useful,
+  but WITHOUT ANY WARRANTY; without even the implied warranty of
+  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+  GNU Lesser General Public License for more details.
+
+  You should have received a copy of the GNU Lesser General Public License
+  along with Origami.  If not, see <http://www.gnu.org/licenses/>.
+
+=end
+
+require 'origami/filters/predictors'
+
+module Origami
+
+  module Filter
+
+    class InvalidLZWDataError < Exception #:nodoc:
+    end
+    
+    #
+    # Class representing a filter used to encode and decode data with LZW compression algorithm.
+    #
+    class LZW
+      
+      include Filter
+
+      class DecodeParms < Dictionary
+        include StandardObject
+
+        field   :Predictor,         :Type => Integer, :Default => 1
+        field   :Colors,            :Type => Integer, :Default => 1
+        field   :BitsPerComponent,  :Type => Integer, :Default => 8
+        field   :Columns,           :Type => Integer, :Default => 1
+        field   :EarlyChange,       :Type => Integer, :Default => 1
+      end
+ 
+      EOD = 257 #:nodoc:
+      CLEARTABLE = 256 #:nodoc:
+      
+      #
+      # Creates a new LZW Filter.
+      # _parameters_:: A hash of filter options (ignored).
+      #
+      def initialize(parameters = {})
+        super(DecodeParms.new(parameters))
+      end
+      
+      #
+      # Encodes given data using LZW compression method.
+      # _stream_:: The data to encode.
+      #
+      def encode(string)
+        if @params.Predictor.is_a?(Integer)
+          colors  = @params.Colors.is_a?(Integer) ?  @params.Colors.to_i : 1
+          bpc     = @params.BitsPerComponent.is_a?(Integer) ? @params.BitsPerComponent.to_i : 8
+          columns = @params.Columns.is_a?(Integer) ? @params.Columns.to_i : 1
+
+          string = Predictor.do_pre_prediction(string, @params.Predictor.to_i, colors, bpc, columns)
+        end       
+        
+        codesize = 9
+        result = Utils::BitWriter.new
+        result.write(CLEARTABLE, codesize)
+        table = clear({})
+        
+        s = ''        
+        string.each_byte do |byte|
+          char = byte.chr
+          
+          case table.size
+            when 512 then codesize = 10
+            when 1024 then codesize = 11
+            when 2048 then codesize = 12
+            when 4096
+              result.write(CLEARTABLE, codesize)
+              codesize = 9
+              clear table
+              redo
+          end
+         
+          it = s + char
+          if table.has_key?(it)
+            s = it
+          else
+            result.write(table[s], codesize)
+            table[it] = table.size
+            s = char
+          end
+        end
+         
+        result.write(table[s], codesize)
+        result.write(EOD, codesize)
+        
+        result.final.to_s
+      end
+      
+      #
+      # Decodes given data using LZW compression method.
+      # _stream_:: The data to decode.
+      #
+      def decode(string)
+       
+        result = ""
+        bstring = Utils::BitReader.new(string)
+        codesize = 9
+        table = clear(Hash.new)
+        prevbyte = nil
+
+        until bstring.eod? do
+          byte = bstring.read(codesize)
+
+          case table.size
+            when 510 then codesize = 10
+            when 1022 then codesize = 11
+            when 2046 then codesize = 12
+            when 4095
+              if byte != CLEARTABLE
+              then
+                raise InvalidLZWDataError, 
+                  "LZW table is full and no clear flag was set (codeword #{byte.to_s(2).rjust(codesize,'0')} at bit #{bstring.pos - codesize}/#{bstring.size})"
+              end
+          end
+
+          if byte == CLEARTABLE
+            codesize = 9
+            code = EOD
+            clear table
+            prevbyte = nil
+            redo
+          elsif byte == EOD
+            break
+          else
+            if prevbyte.nil?
+              prevbyte = byte
+              result << table.key(byte)
+              redo
+            else
+              if table.has_value?(byte)
+                entry = table.key(byte)
+              else
+                entry = table.key(prevbyte)
+                entry += entry[0,1]
+              end
+
+              result << entry
+              table[table.key(prevbyte) + entry[0,1]] = table.size
+              prevbyte = byte
+            end
+          end
+        end
+ 
+        if @params.Predictor.is_a?(Integer)
+          colors  = @params.Colors.is_a?(Integer) ?  @params.Colors.to_i : 1
+          bpc     = @params.BitsPerComponent.is_a?(Integer) ? @params.BitsPerComponent.to_i : 8
+          columns = @params.Columns.is_a?(Integer) ? @params.Columns.to_i : 1
+
+          result = Predictor.do_post_prediction(result, @params.Predictor.to_i, colors, bpc, columns)
+        end
+
+        result
+      end
+      
+      private
+
+      def clear(table) #:nodoc:
+        table.clear
+        256.times do |i|
+          table[i.chr] = i
+        end
+        
+        table[CLEARTABLE] = CLEARTABLE
+        table[EOD] = EOD
+        
+        table
+      end
+      
+    end
+  end
+end
+