origami 1.2.5 → 1.2.6

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 400584bfa83ed0b47e3501f85f6ea19733aa365a
+  data.tar.gz: ebe216889afe825f5ea850ac3a330b7b462afb0a
+SHA512:
+  metadata.gz: a5383fd775e5eb11a8a2a1212b4e8523f9751974fae787fd72074206e435b29b6e5d915c05154e25c12617733b317f20e5a3cd3c7f099a275ec995ce55308bb8
+  data.tar.gz: 5dd566b2f6cedca2b1be9b124d809efb46cc8057fc4f6ffbd5a7f22a56c165da1ff0fd1e1ea0e8be8695d98ef8363adbefafeb77fcab256ddea3dad38cbe674a

data/bin/gui/config.rb

@@ -96,10 +96,8 @@ module PDFWalker
       end
       
       def set_verbosity(level)
-        
         @conf["Debug"]['Verbosity'] = level
         save
-        
       end
       
       def verbosity
@@ -113,11 +111,9 @@ module PDFWalker
       private
       
       def set_missing_values
-        
         @conf ||= {}
         
         DEFAULT_CONFIG.each_key do |cat|
-          
           @conf[cat] = {} unless @conf.include?(cat)
           
           DEFAULT_CONFIG[cat].each_pair do |key, value|

data/bin/gui/imgview.rb

@@ -41,11 +41,11 @@ module PDFWalker
       }
     end
 
-    def show_raw_img(data, w, h, bpp, bpr)
+    def show_raw_img(data, w, h, bpc, bpr)
       set_default_size w,h
 
       pixbuf = Gdk::Pixbuf.new data, 
-        Gdk::Pixbuf::ColorSpace::RGB, false, bpp,
+        Gdk::Pixbuf::ColorSpace::RGB, false, bpc,
         w, h,
         bpr
 

data/bin/gui/menu.rb

@@ -153,8 +153,16 @@ module PDFWalker
         :Callback => lambda { |widget, viewer, path|
           stm = viewer.model.get_value(viewer.model.get_iter(path), viewer.class::OBJCOL)
           w,h = stm.Width, stm.Height
+          colors = 
+            case stm.ColorSpace
+              when :DeviceGray.to_o then 1
+              when :DeviceRGB.to_o then 3
+              when :DeviceCMYK.to_o then 4
+            else
+              1
+            end
           bpc = stm.BitsPerComponent || 8
-          bpr = (bpc >> 3) * 3 * w
+          bpr = (w * colors * bpc + 7) >> 3
           data = stm.data
 
           begin
@@ -162,7 +170,7 @@ module PDFWalker
             if stm.Filter == :DCTDecode or (stm.Filter.is_a?(Array) and stm.Filter[0] == :DCTDecode)
               imgview.show_compressed_img data
             else
-              imgview.show_raw_img data, w, h, bpp, bpr
+              imgview.show_raw_img data, w, h, bpc, bpr
             end
           rescue Exception => e
             viewer.parent.error("#{e.class}: #{e.message}")
@@ -179,7 +187,7 @@ module PDFWalker
         if obj.is_a?(Graphics::ImageXObject)
           :Image
         else
-          obj.real_type.to_s.split("::").last.to_sym
+          obj.native_type.to_s.split("::").last.to_sym
         end
       else case obj
         when Origami::PDF

data/bin/gui/treeview.rb

@@ -107,7 +107,7 @@ module PDFWalker
             expand_row(path, false)
           end
           
-          goto(obj.solve) if obj.is_a?(Origami::Reference)
+          goto(obj) if obj.is_a?(Origami::Reference)
         end
       }
       
@@ -136,7 +136,13 @@ module PDFWalker
         if obj.is_a?(Name) and obj.parent.is_a?(Dictionary) and obj.parent.has_key?(obj)
           obj = obj.parent[obj]
         elsif obj.is_a?(Reference)
-          obj = obj.solve
+          obj = 
+            begin 
+              obj.solve
+            rescue InvalidReferenceError
+              @parent.error("Object not found : #{obj}")
+              return
+            end
         end
 
         @treestore.each { |model, path, iter|
@@ -279,7 +285,7 @@ module PDFWalker
       obj = @treestore.append(container)
       @treestore.set_value(obj, OBJCOL, object)
       
-      type = object.real_type.to_s.split('::').last.to_sym
+      type = object.native_type.to_s.split('::').last.to_sym
       
       if name.nil?
         name = 

data/bin/pdfexplode

@@ -0,0 +1,220 @@
+#!/usr/bin/env ruby 
+
+=begin
+
+= Author: 
+  Guillaume Delugré <guillaume/at/security-labs.org>
+
+= Info:
+  Explodes a PDF into separate documents.
+= License:
+	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
+
+begin
+  require 'origami'
+rescue LoadError
+  ORIGAMIDIR = "#{File.dirname(__FILE__)}/../lib"
+  $: << ORIGAMIDIR
+  require 'origami'
+end
+include Origami
+
+require 'optparse'
+require 'rexml/document'
+
+class OptParser
+  BANNER = <<USAGE
+Usage: #{$0} <PDF-file> [-r <range>] [-t pages|rsrc] [-d <output-directory>]
+Explodes a document into separate documents.
+Bug reports or feature requests at: http://origami-pdf.googlecode.com/
+
+Options:
+USAGE
+
+  def self.parser(options)
+    OptionParser.new do |opts|
+      opts.banner = BANNER
+
+      opts.on("-d", "--output-dir DIR", "Output directory.") do |d|
+        options[:output_dir] = d
+      end
+
+      opts.on("-r", "--range PAGES", "Page range (e.g: 2-, 1-3, 5). Default to '-'.") do |r|
+        range =
+          if r.index('-').nil?
+            page = r.to_i
+            Range.new(page-1, page-1)
+          else
+            from, to = r.split('-').map{|bound| bound.to_i}
+            from ||= 1
+            to ||= 0
+            Range.new(from-1, to-1)
+          end
+        options[:page_range] = range
+      end
+
+      opts.on("-t", "--type TYPE", "Split by type. Can be 'pages' or 'rsrc'. Default to 'pages'.") do |t|
+        options[:split_by] = t
+      end
+
+      opts.on_tail("-h", "--help", "Show this message.") do
+        puts opts
+        exit
+      end
+    end
+  end
+
+  def self.parse(args)
+    options = 
+    {
+      :page_range => (0..-1),
+      :split_by => 'pages'
+    }
+
+    self.parser(options).parse!(args)
+
+    options
+  end
+end
+
+begin
+  @options = OptParser.parse(ARGV)
+
+  if ARGV.empty?
+    STDERR.puts "Error: No filename was specified. #{$0} --help for details."
+    exit 1
+  else
+    target = ARGV.shift
+  end
+
+  if @options[:output_dir].nil?
+    @options[:output_dir] = "#{File.join(File.dirname(target), File.basename(target,'.pdf'))}.explode"
+  end
+
+  Origami::OPTIONS[:ignore_bad_references] = true
+  OUTPUT_DIR = @options[:output_dir]
+  Dir::mkdir(OUTPUT_DIR) unless File.directory?(OUTPUT_DIR)
+
+  def split_by_rsrc(n, page, type)
+    all_rsrc = page.resources
+    type_rsrc = page.ls_resources(type)
+    other_rsrc = all_rsrc.keys - type_rsrc.keys
+
+    unless type_rsrc.empty?
+      # Keep only specified resource type.
+      output_file = File.join(OUTPUT_DIR, "page_#{n}_keeponly_#{type}.pdf")
+      PDF.write(output_file) do |pdf|
+        reduced = page.copy
+        # New resource dictionary with only matching resources.
+        reduced.Resources = Resources.new(type => type_rsrc)
+        # Remove mention of other resources.
+        reduced.Contents.data = reduced.Contents.data.lines.to_a.
+          delete_if {|line| other_rsrc.any?{|rsrc| line =~ /#{rsrc}/}}.join
+
+        STDERR.puts "Creating #{output_file}..."
+        pdf.append_page(reduced)
+      end
+
+      # Remove all specified resource type.
+      output_file = File.join(OUTPUT_DIR, "page_#{n}_excluded_#{type}.pdf") 
+      PDF.write(output_file) do |pdf|
+        reduced = page.copy
+        # New resource dictionary with no resource of specified type.
+        reduced.Resources = reduced.Resources.copy 
+        reduced.Resources.delete(type)
+        # Remove mention this resource type.
+        reduced.Contents.data = reduced.Contents.data.lines.to_a.
+          delete_if {|line| type_rsrc.keys.any?{|rsrc| line =~ /#{rsrc}/}}.join
+
+        STDERR.puts "Creating #{output_file}..."
+        pdf.append_page(reduced)
+      end
+
+      # Now treating each resource object separately.
+      type_rsrc.each_pair do |name, rsrc|
+        anyother_rsrc = all_rsrc.keys - [ name ]
+        # Keey only specified resource object.
+        output_file = File.join(OUTPUT_DIR, "page_#{n}_keeponly_#{type}_#{name}.pdf")
+        PDF.write(output_file) do |pdf|
+          reduced = page.copy
+          # New resource dictionary with only specified resource object.
+          reduced.Resources = Resources.new(type => {name => rsrc})
+          # Remove mention of all other resources.
+          reduced.Contents.data = reduced.Contents.data.lines.to_a.
+            delete_if {|line| anyother_rsrc.any?{|rsrc| line =~ /#{rsrc}/}}.join
+
+          STDERR.puts "Creating #{output_file}..."
+          pdf.append_page(reduced)
+        end
+        
+        # Remove only specified resource object.
+        output_file = File.join(OUTPUT_DIR, "page_#{n}_excluded_#{type}_#{name}.pdf")
+        PDF.write(output_file) do |pdf|
+          reduced = page.copy
+          # New resource dictionary with only specified resource object.
+          reduced.Resources = reduced.Resources.copy
+          reduced.Resources[type] = reduced.Resources.send(type).copy
+          reduced.Resources[type].delete(name)
+          # Remove mention of this resource only.
+          reduced.Contents.data = reduced.Contents.data.lines.to_a.
+            delete_if {|line| line =~ /#{name}/}.join
+
+          STDERR.puts "Creating #{output_file}..."
+          pdf.append_page(reduced)
+        end
+      end
+    end
+  end
+
+  params = 
+  {
+    :verbosity => Parser::VERBOSE_QUIET,
+  }
+  pdf = PDF.read(target, params)
+
+  i = @options[:page_range].first + 1
+  pdf.pages[@options[:page_range]].each do |page|
+    case @options[:split_by]
+      when 'pages'
+        output_file = File.join(OUTPUT_DIR, "page_#{i}.pdf")
+        PDF.write(output_file) do |pdf|
+          STDERR.puts "Creating #{output_file}..."
+          pdf.append_page(page)
+        end
+
+      when 'rsrc'
+        [ Resources::EXTGSTATE,
+          Resources::COLORSPACE,
+          Resources::PATTERN,
+          Resources::SHADING,
+          Resources::XOBJECT,
+          Resources::FONT,
+          Resources::PROPERTIES
+        ].each { |type| split_by_rsrc(i, page, type) }
+
+      else
+        raise ArgumentError, "Unknown split option: #{@options[:split_by]}"
+    end
+
+    i += 1
+  end 
+
+rescue SystemExit
+rescue Exception => e
+  STDERR.puts "#{e.class}: #{e.message} #{e.backtrace}"
+  exit 1
+end
+

data/bin/pdfextract

@@ -120,6 +120,9 @@ begin
     @options[:output_dir] = "#{File.basename(target, '.pdf')}.dump"
   end
 
+  # Force data extraction, even for invalid FlateDecode streams.
+  Origami::OPTIONS[:ignore_zlib_errors] = true
+
   OUTPUT_DIR = @options[:output_dir]
   Dir::mkdir(OUTPUT_DIR) unless File.directory?(OUTPUT_DIR)
 

data/lib/origami/acroform.rb

@@ -147,7 +147,7 @@ module Origami
       
     def self.included(receiver) #:nodoc:
 
-      receiver.field   :FT,     :Type => Name, :Default => Type::TEXT, :Required => true
+      receiver.field   :FT,     :Type => Name, :Required => true
       receiver.field   :Parent, :Type => Dictionary
       receiver.field   :Kids,   :Type => Array
       receiver.field   :T,      :Type => String
@@ -254,7 +254,7 @@ module Origami
       end
       
       field   :Type,            :Type => Name, :Default => :SigFieldLock
-      field   :Action,          :Type => Name, :Default => Actions::ALL, :Required => true
+      field   :Action,          :Type => Name, :Required => true
       field   :Fields,          :Type => Array
 
       def pre_build

data/lib/origami/actions.rb

@@ -65,18 +65,20 @@ module Origami
       # Creates a new GoTo Action.
       # _hash_:: A hash of options to set for this jump.
       #
-      def initialize(hash = {})
-        
+      def self.[](hash = {})
         if hash.is_a? Destination
-          super(:S => :GoTo, :D => hash)
+          self.new(:S => :GoTo, :D => hash)
         else
-          super(hash)
+          self.new(hash)
         end
-        
       end
       
     end
     
+    def self.GoTo(hash = {})
+      Action::GoTo[hash]
+    end
+
     #
     # Class representing an action launching an URL.
     #
@@ -91,12 +93,16 @@ module Origami
       # _uri_:: The URI to launch.
       # _ismap_::
       #
-      def initialize(uri, ismap = false)
-        super(:URI => uri, :IsMap => ismap)
+      def self.[](uri, ismap = false)
+        self.new(:URI => uri, :IsMap => ismap)
       end
     
     end
 
+    def self.URI(uri, ismap = false)
+      Action::URI[uri, ismap]
+    end
+
     #
     # Class representing a JavaScript Action.
     #
@@ -109,11 +115,15 @@ module Origami
       # Creates a new JavaScript Action.
       # _script_:: The script to be executed.
       #
-      def initialize(script)
-        super(:JS => script)
+      def self.[](script)
+        self.new(:JS => script)
       end
     end
 
+    def self.JavaScript(script)
+      Action::JavaScript[script]
+    end
+
     #
     # Class representing an Action which run a command on the current system.
     #
@@ -147,19 +157,22 @@ module Origami
     #
     class Named < Action
       
-      NEXTPAGE = :NextPage
-      PREVPAGE = :PrevPage
-      FIRSTPAGE = :FirstPage
-      LASTPAGE = :LastPage
-      PRINT = :Print
-   
       field   :S,         :Type => Name, :Default => :Named, :Required => true
       field   :N,         :Type => Name, :Required => true
 
-      def initialize(type)
-        super(:N => type)
+      def self.[](type)
+        self.new(:N => type)
       end
 
+      NEXTPAGE = self[:NextPage]
+      PREVPAGE = self[:PrevPage]
+      FIRSTPAGE = self[:FirstPage]
+      LASTPAGE = self[:LastPage]
+      PRINT = self[:Print]
+    end
+
+    def self.Named(type)
+      Action::Named[type]
     end
     
     #
@@ -178,11 +191,15 @@ module Origami
       # _dest_:: A Destination in the file.
       # _newwindow_:: Specifies whether the file has to be opened in a new window.
       #
-      def initialize(file, dest = Destination::GlobalFit.new(0), newwindow = false)
-        super(:F => file, :D => dest, :NewWindow => newwindow)
+      def self.[](file, dest = Destination::GlobalFit.new(0), newwindow = false)
+        self.new(:F => file, :D => dest, :NewWindow => newwindow)
       end
       
     end
+
+    def self.GoToR(file, dest = Destination::GlobalFit.new(0), newwindow = false)
+      Action::GoToR[file, dest, newwindow]
+    end
     
     #
     # Class representing a GoTo Action to an embedded pdf file.
@@ -215,11 +232,15 @@ module Origami
         
       end
     
-      def initialize(filename, dest, newwindow = false)
-        super(:T => EmbeddedTarget.new(:R => :C, :N => filename), :D => dest, :NewWindow => newwindow)
+      def self.[](filename, dest, newwindow = false)
+        self.new(:T => EmbeddedTarget.new(:R => :C, :N => filename), :D => dest, :NewWindow => newwindow)
       end
       
     end
+
+    def self.GoToE(filename, dest, newwindow = false)
+      Action::GoToE[filename, dest, newwindow]
+    end
     
     #
     # (PDF 1.2) Send data to a uniform resource locator. p703
@@ -247,31 +268,33 @@ module Origami
       field   :Fields,      :Type => Array
       field   :Flags,       :Type => Integer, :Default => 0
 
-      def initialize(url, fields = [], flags = 0)
-        if not url.is_a? FileSpec
-          url = FileSpec.new(:FS => :URL, :F => url)
-        end
-        
-        super(:F => url, :Fields => fields, :Flags => flags)
+      def self.[](url, fields = [], flags = 0)
+        url = FileSpec.new(:FS => :URL, :F => url) unless url.is_a? FileSpec
+        self.new(:F => url, :Fields => fields, :Flags => flags)
       end
       
     end
 
+    def self.SubmitForm(url, fields = [], flags = 0)
+      Action::SubmitForm[url, fields, flags]
+    end
+
     class ImportData < Action
 
       field   :S,           :Type => Name, :Default => :ImportData, :Required => true
       field   :F,           :Type => Dictionary, :Required => true
 
-      def initialize(file)
-        if not file.is_a? FileSpec
-          file = FileSpec.new(:FS => :File, :F => file)
-        end
-        
-        super(:F => file)
+      def self.[](file)
+        file = FileSpec.new(:FS => :File, :F => file) unless file.is_a? FileSpec
+        self.new(:F => file)
       end
     
     end
 
+    def self.ImportData(file)
+      Action::ImportData[file]
+    end
+
     class RichMediaExecute < Action
 
       field   :S,           :Type => Name, :Default => :RichMediaExecute, :Version => "1.7", :ExtensionLevel => 3, :Required => true
@@ -287,11 +310,15 @@ module Origami
         field   :A,         :Type => Object, :Version => "1.7", :ExtensionLevel => 3
       end
 
-      def initialize(annotation, command, *params)
-        super(:TA => annotation, :CMD => Command.new(:C => command, :A => params))
+      def self.[](annotation, command, *params)
+        self.new(:TA => annotation, :CMD => Command.new(:C => command, :A => params))
       end
     
     end
+
+    def self.RichMediaExecute(annotation, command, *params)
+      Action::RichMediaExecute[annotation, command, *params]
+    end
   
   end
 

data/lib/origami/annotations.rb

@@ -34,7 +34,7 @@ module Origami
     include StandardObject
     
     field   :Type,            :Type => Name, :Default => :Annot
-    field   :Subtype,         :Type => Name, :Default => :Text, :Required => true
+    field   :Subtype,         :Type => Name, :Required => true
     field   :Rect,            :Type => Array, :Default => [ 0 , 0 , 0 , 0 ], :Required => true
     field   :Contents,        :Type => String
     field   :P,               :Type => Dictionary, :Version => "1.3"
@@ -257,7 +257,7 @@ module Origami
 
       include Markup
 
-      field   :Subtype,         :Type => Name, :Default => :Square, :Required => true
+      field   :Subtype,         :Type => Name, :Required => true
       field   :BS,              :Type => Dictionary
       field   :IC,              :Type => Array
       field   :BE,              :Type => Dictionary, :Version => "1.5"
@@ -266,6 +266,7 @@ module Origami
     end
     
     class Square < Shape
+      field   :Subtype,         :Type => Name, :Default => :Square, :Required => true
     end
     
     class Circle < Shape

data/lib/origami/array.rb

@@ -85,7 +85,7 @@ module Origami
       super
     end
     
-    def self.parse(stream) #:nodoc:
+    def self.parse(stream, parser = nil) #:nodoc:
       data = []
       offset = stream.pos
       
@@ -100,7 +100,7 @@ module Origami
           raise InvalidArrayObjectError, "Bad embedded object format"
         end
         
-        value = type.parse(stream)  
+        value = type.parse(stream, parser)
         data << value
       end
     
@@ -155,7 +155,20 @@ module Origami
     
     alias value to_a
 
-    def real_type ; Origami::Array end
+    def copy
+      copy = self.class.new
+      self.each do |obj|
+        copy << obj.copy
+      end
+
+      copy.parent = @parent
+      copy.no, copy.generation = @no, @generation
+      copy.set_indirect(true) if is_indirect?
+      copy.set_pdf(@pdf) if is_indirect?
+      copy
+    end
+
+    def self.native_type ; Origami::Array end
 
   end
 
@@ -167,7 +180,17 @@ module Origami
     class << self
       
       def [](coords)
-        corners = coords.values_at(:llx, :lly, :urx, :ury)
+        corners = 
+          if [ :llx, :lly, :urx, :ury ].all? {|p| coords.include?(p)}
+            coords.values_at(:llx, :lly, :urx, :ury)
+          elsif [ :width, :height ].all? {|p| coords.include?(p)}
+            width, height = coords.values_at(:width, :height)
+            x = coords.values_at(:x).first || 0
+            y = coords.values_at(:y).first || 0
+            [ x, y, x+width, y+height ]
+          else
+            raise ArgumentError, "Bad arguments for #{self.class}: #{coords.inspect}"
+          end
         
         unless corners.all? { |corner| corner.is_a?(Numeric) }
           raise TypeError, "All coords must be numbers"

data/lib/origami/boolean.rb

@@ -59,7 +59,7 @@ module Origami
       super(@value.to_s)
     end
     
-    def self.parse(stream) #:nodoc:
+    def self.parse(stream, parser = nil) #:nodoc:
 
       offset = stream.pos
     
@@ -82,7 +82,7 @@ module Origami
       @value
     end
 
-    def real_type ; Boolean end
+    def self.native_type ; Boolean end
     
     def false?
       @value == false