origami 1.2.7 → 2.0.0

data/bin/pdfmetadata

@@ -2,122 +2,123 @@
 
 =begin
 
-= Author: 
-  Guillaume Delugré <guillaume/at/security-labs.org>
+= Info
+    Prints out the metadata contained in a PDF document.
 
-= Info:
-  Prints out the metadata contained in a PDF document.
+= License
+    Copyright (C) 2016  Guillaume Delugré.
 
-= 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 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.
+    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/>.
+    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'
+    require 'origami'
 rescue LoadError
-  ORIGAMIDIR = "#{File.dirname(__FILE__)}/../lib"
-  $: << ORIGAMIDIR
-  require 'origami'
+    $: << File.join(__dir__, '../lib')
+    require 'origami'
 end
 include Origami
 
+require 'colorize'
 require 'optparse'
 
 class OptParser
-  BANNER = <<USAGE
+    BANNER = <<USAGE
 Usage: #{$0} [<PDF-file>] [-i] [-x]
 Prints out the metadata contained in a PDF document.
-Bug reports or feature requests at: http://origami-pdf.googlecode.com/
+Bug reports or feature requests at: http://github.com/gdelugre/origami
 
 Options:
 USAGE
 
-  def self.parser(options)
-    OptionParser.new do |opts|
-      opts.banner = BANNER
+    def self.parser(options)
+        OptionParser.new do |opts|
+            opts.banner = BANNER
 
-      opts.on("-i", "Extracts document info metadata") do |i|
-        options[:doc_info] = true
-      end
+            opts.on("-i", "Extracts document info metadata") do |i|
+                options[:doc_info] = true
+            end
 
-      opts.on("-x", "Extracts XMP document metadata stream") do |i|
-        options[:doc_stream] = true
-      end
+            opts.on("-x", "Extracts XMP document metadata stream") do |i|
+                options[:doc_stream] = true
+            end
 
-      opts.on_tail("-h", "--help", "Show this message") do
-        puts opts
-        exit
-      end
+            opts.on("-n", "Turn off colorized output.") do
+                options[:disable_colors] = true
+            end
+
+            opts.on_tail("-h", "--help", "Show this message") do
+                puts opts
+                exit
+            end
+        end
     end
-  end
 
-  def self.parse(args)
-    options = 
-    {
-      :output => STDOUT,
-    }
+    def self.parse(args)
+        options =
+        {
+            disable_colors: false
+        }
 
-    self.parser(options).parse!(args)
+        self.parser(options).parse!(args)
 
-    options
-  end
+        options
+    end
 end
 
 begin
-  @options = OptParser.parse(ARGV)
-
-  unless @options[:doc_info] or @options[:doc_stream]
-    @options[:doc_info] = @options[:doc_stream] = true
-  end
-
-  target = (ARGV.empty?) ? STDIN : ARGV.shift
-  params = 
-  {
-    :verbosity => Parser::VERBOSE_QUIET,
-  }
-
-  pdf = PDF.read(target, params)
-
-  if @options[:doc_info]
-    if pdf.has_document_info?
-      Console.colorprint "[*] Document information dictionary:\n", Console::Colors::MAGENTA
-
-      docinfo = pdf.get_document_info
-      docinfo.each_pair do |name, item|
-        Console.colorprint name.value.to_s.ljust(20, ' '), Console::Colors::GREEN
-        puts ": #{item.solve.value}"
-      end
-      puts
+    @options = OptParser.parse(ARGV)
+
+    unless @options[:doc_info] or @options[:doc_stream]
+        @options[:doc_info] = @options[:doc_stream] = true
     end
-  end
 
-  if @options[:doc_stream]
-    if pdf.has_metadata?
-      Console.colorprint "[*] Metadata stream:\n", Console::Colors::MAGENTA
+    String.disable_colorization @options[:disable_colors]
+
+    target = (ARGV.empty?) ? STDIN : ARGV.shift
+    params =
+    {
+        verbosity: Parser::VERBOSE_QUIET,
+        lazy: true
+    }
 
-      metadata = pdf.get_metadata
-      metadata.each_pair do |name, item|
-        Console.colorprint name.ljust(20, ' '), Console::Colors::GREEN
-        puts ": #{item}"
-      end
+    pdf = PDF.read(target, params)
+
+    if @options[:doc_info] and pdf.document_info?
+        puts "[*] Document information dictionary:".magenta
+
+        docinfo = pdf.document_info
+        docinfo.each_pair do |name, item|
+            obj = item.solve
+
+            print name.value.to_s.ljust(20, ' ').green
+            puts ": #{obj.respond_to?(:to_utf8) ? obj.to_utf8 : obj.value}"
+        end
+        puts
     end
-  end
 
-rescue SystemExit
-rescue Exception => e
-  STDERR.puts "#{e.class}: #{e.message}"
-  exit 1
-end
+    if @options[:doc_stream] and pdf.metadata?
+        puts "[*] Metadata stream:".magenta
 
+        metadata = pdf.metadata
+        metadata.each_pair do |name, item|
+            print name.ljust(20, ' ').green
+            puts ": #{item}"
+        end
+    end
+
+rescue
+    abort "#{$!.class}: #{$!.message}"
+end

data/bin/pdfsh

@@ -1,13 +1,12 @@
 #!/usr/bin/env ruby
 
 begin
-  require 'irb'
+    require 'irb'
 rescue LoadError
-  abort "Error: you need to install irb to run this application."
+    abort "Error: you need to install irb to run this application."
 end
 
-$:.unshift "#{File.dirname(__FILE__)}/shell"
-ENV["IRBRC"] = "#{File.dirname(__FILE__)}/shell/.irbrc"
+$:.unshift File.join(__dir__, 'shell')
+ENV["IRBRC"] = File.join(__dir__, "shell", ".irbrc")
 
 IRB.start
-

data/bin/pdfwalker

@@ -1,7 +1,6 @@
 #!/usr/bin/env ruby
 
-$:.unshift "#{File.dirname(__FILE__)}"
+$:.unshift(__dir__)
 require 'gui/walker'
 
 PDFWalker::Walker.start(ARGV[0])
-

data/bin/shell/.irbrc

@@ -1,9 +1,8 @@
 begin
-  require 'origami'
+    require 'origami'
 rescue LoadError
-  ORIGAMIDIR = "#{File.dirname(__FILE__)}/../../lib"
-  $: << ORIGAMIDIR
-  require 'origami'
+    $: << File.join(__dir__, '../../lib')
+    require 'origami'
 end
 include Origami
 
@@ -12,95 +11,59 @@ require 'readline'
 
 OPENSSL_SUPPORT = (defined?(OpenSSL).nil?) ? 'no' : 'yes'
 JAVASCRIPT_SUPPORT = (defined?(PDF::JavaScript::Engine).nil?) ? 'no' : 'yes'
-DEFAULT_BANNER = "Welcome to the PDF shell (Origami release #{Origami::VERSION}) [OpenSSL: #{OPENSSL_SUPPORT}, JavaScript: #{JAVASCRIPT_SUPPORT}]\n\n"
+DEFAULT_BANNER = "Welcome to the PDF shell (Origami release #{Origami::VERSION}) [OpenSSL: #{OPENSSL_SUPPORT}, JavaScript: #{JAVASCRIPT_SUPPORT}]\n"
 
 def set_completion
-
-  completionProc = proc { |input|
-    bind = IRB.conf[:MAIN_CONTEXT].workspace.binding
-    
-    validClasses = Origami.constants.reject do |name| 
-      obj = Origami.const_get(name)
-      (not obj.is_a?(Module) and not obj.is_a?(Class)) or obj <= Exception
-    end
-
-    case input
-
-#    # Classes
-#    when /^([A-Z][^:\.\(]*)$/
-#      classname = $1
-#      candidates = validClasses
-#      return candidates.grep(/^#{classname}/)
-#
-#    # Methods
-#    when /^([^:.\(]*)\.([^:.]*)$/
-#      classname = $1
-#      method = Regexp.quote($2)
-#      candidates = []
-#      if validClasses.include? $1
-#        begin
-#          candidates = eval("Origami::#{classname}.methods", bind)
-#        rescue Exception
-#          candidates = []
-#        end
-#        return candidates.grep(/^#{method}/).collect{|e| classname + "." + e}
-#      else
-#        begin
-#          var = $1.dup
-#          classname = eval("#{classname}.class", bind).to_s
-#          if validClasses.include?(classname.split("::").last)
-#            candidates = eval("#{classname}.public_instance_methods", bind)
-#          end
-#        rescue Exception => e
-#          candidates = []
-#        end
-#        return candidates.grep(/^#{method}/).collect{|e| var + "." + e}
-#      end
-#
-      # Mod/class
-      when /^(.*)::$/
-        begin
-          space = eval("Origami::#{$1}", bind)
-        rescue Exception
-          return []
-        end
-
-        return space.constants.reject{|const| space.const_get(const) <= Exception}
-
-      when /^(.*).$/
-        begin 
-          space = eval($1, bind)
-        rescue
-          return []
+    completionProc = proc { |input|
+        bind = IRB.conf[:MAIN_CONTEXT].workspace.binding
+
+        case input
+        when /^(.*)::$/
+            begin
+                space = eval("Origami::#{$1}", bind)
+            rescue Exception
+                return []
+            end
+
+            return space.constants.reject{|const| space.const_get(const) <= Exception}
+
+        when /^(.*).$/
+            begin
+                space = eval($1, bind)
+            rescue
+                return []
+            end
+
+            return space.public_methods
         end
+    }
 
-        return space.public_methods
+    if Readline.respond_to?("basic_word_break_characters=")
+        Readline.basic_word_break_characters= " \t\n\"\\'`><=;|&{("
     end
-  }
-
-  if Readline.respond_to?("basic_word_break_characters=")
-    Readline.basic_word_break_characters= " \t\n\"\\'`><=;|&{("
-  end
-  Readline.completion_append_character = nil
-  Readline.completion_proc = completionProc
 
+    Readline.completion_append_character = nil
+    Readline.completion_proc = completionProc
 end
 
 def set_prompt
+    IRB.conf[:PROMPT][:PDFSH] = {
+        PROMPT_C: "?>> ",
+        RETURN: "%s\n",
+        PROMPT_I: ">>> ",
+        PROMPT_N: ">>> ",
+        PROMPT_S: nil
+    }
+
+    IRB.conf[:PROMPT_MODE] = :PDFSH
+    IRB.conf[:AUTO_INDENT] = true
+end
 
-  IRB.conf[:PROMPT][:PDFSH] = {
-    :PROMPT_C => "?>> ",
-    :RETURN => "%s\n",
-    :PROMPT_I => ">>> ",
-    :PROMPT_N => ">>> ",
-    :PROMPT_S => nil
-  }
-
-  IRB.conf[:PROMPT_MODE] = :PDFSH
+# Print the shell banner.
+puts DEFAULT_BANNER.green
 
-end
+# Import the type conversion helper routines.
+TOPLEVEL_BINDING.eval("using Origami::TypeConversion")
 
-Console.colorprint(DEFAULT_BANNER, Console::Colors::GREEN)
 #set_completion
 set_prompt
-

data/bin/shell/console.rb

@@ -1,150 +1,125 @@
 =begin
 
-= File
-	console.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/>.
+    This file is part of Origami, PDF manipulation framework for Ruby
+    Copyright (C) 2016	Guillaume Delugré.
+
+    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 'tempfile'
 require 'hexdump'
+require 'colorize'
 
-if RUBY_VERSION < '1.9'
-  def Kernel.spawn(cmd)
-    fork do
-      exec(cmd)
-    end
-  end
-end
+String.disable_colorization(false)
 
 module Origami
-
-  module Object
-    def inspect
-      to_s
+    module Object
+        def inspect
+            to_s
+        end
     end
-  end
-
-  unless RUBY_VERSION < '1.9'
-    require 'tempfile'
 
     class Stream
-      def edit(editor = ENV['EDITOR'])
-        tmpfile = Tempfile.new("origami")
-        tmpfile.write(self.data)
-        tmpfile.close
-
-        Process.wait Kernel.spawn "#{editor} #{tmpfile.path}"
-        
-        self.data = File.read(tmpfile.path)
-        tmpfile.unlink
-
-        true
-      end
-
-      def inspect
-        self.data.hexdump
-      end
-    end
+        def edit(editor = ENV['EDITOR'])
+            Tempfile.open("origami") do |tmpfile|
+                tmpfile.write(self.data)
+                tmpfile.flush
 
-    class Page
-      def edit 
-        self.Contents.edit
-      end
-    end
-  end
-  
-  class PDF
-
-    if defined?(PDF::JavaScript::Engine)
-      class JavaScript::Engine
-        def shell
-          while (print 'js> '; line = gets)
-            begin
-              puts exec(line)
-            rescue V8::JSError => e
-              puts "Error: #{e.message}"
+                Process.wait Kernel.spawn "#{editor} #{tmpfile.path}"
+
+                self.data = File.read(tmpfile.path)
+                tmpfile.unlink
             end
-          end
+
+            true
+        end
+
+        def inspect
+            self.data.hexdump
         end
-      end
     end
-     
-     class Revision
-      def to_s
-        Console.colorprint("----------  Body  ----------\n", Console::Colors::WHITE, true)
-        @body.each_value { |obj|
-          Console.colorprint("#{obj.reference.to_s.rjust(8,' ')}".ljust(10), Console::Colors::MAGENTA)
-          Console.colorprint("#{obj.type}\n", Console::Colors::YELLOW)
-        }
-        #colorprint("---------- Xrefs -----------\n", Colors::BRIGHT_WHITE, true)
-        #set_fg_color(Colors::BLUE, true) {
-        #  if not @xreftable
-        #    puts "  [x] No xref table found."
-        #  else
-        #    @xreftable.to_s.each_line { |line|
-        #      puts "  " + line
-        #    }
-        #  end
-        #}
-        Console.colorprint("---------- Trailer ---------\n", Console::Colors::WHITE, true) 
-        if not @trailer.dictionary
-          Console.set_fg_color(Console::Colors::BLUE, true) {
-            puts "  [x] No trailer found."
-          }
-        else
-          @trailer.dictionary.each_pair { |entry, value|
-            Console.colorprint("  [*] ", Console::Colors::MAGENTA)
-            Console.colorprint("#{entry.to_s}: ", Console::Colors::YELLOW)
-            Console.colorprint("#{value.to_s}\n", Console::Colors::RED)
-          }
-          Console.colorprint("  [+] ", Console::Colors::MAGENTA)
-          Console.colorprint("startxref: ", Console::Colors::YELLOW)
-          Console.colorprint("#{@trailer.startxref}\n", Console::Colors::RED)
+
+    class Page < Dictionary
+        def edit
+            each_content_stream do |stream|
+                stream.edit
+            end
         end
-      end
-      
-      def inspect
-        to_s
-      end
-     end
-    
-    def to_s
-      puts
-      
-      Console.colorprint("---------- Header ----------\n", Console::Colors::WHITE, true)
-      Console.colorprint("  [+] ", Console::Colors::MAGENTA)
-      Console.colorprint("Major version: ", Console::Colors::YELLOW)
-      Console.colorprint("#{@header.majorversion}\n", Console::Colors::RED)
-      Console.colorprint("  [+] ", Console::Colors::MAGENTA)
-      Console.colorprint("Minor version: ", Console::Colors::YELLOW)
-      Console.colorprint("#{@header.minorversion}\n", Console::Colors::RED)
-      
-      @revisions.each { |revision|
-        revision.to_s
-      }
-      puts
     end
-    
-    def inspect
-      to_s
+
+    class PDF
+        if defined?(PDF::JavaScript::Engine)
+            class JavaScript::Engine
+                def shell
+                    while (print 'js> '.magenta; line = gets)
+                        begin
+                            puts exec(line)
+                        rescue V8::JSError => e
+                            puts "Error: #{e.message}"
+                        end
+                    end
+                end
+            end
+        end
+
+        class Revision
+            def to_s
+                puts "----------  Body  ----------".white.bold
+                @body.each_value { |obj|
+                    print "#{obj.reference.to_s.rjust(8,' ')}".ljust(10).magenta
+                    puts "#{obj.type}".yellow
+                }
+
+                puts "---------- Trailer ---------".white.bold
+                if not @trailer.dictionary
+                    puts "  [x] No trailer found.".blue
+                else
+                    @trailer.dictionary.each_pair { |entry, value|
+                        print "  [*] ".magenta
+                        print "#{entry.to_s}: ".yellow
+                        puts "#{value.to_s}".red
+                    }
+
+                    print "  [+] ".magenta
+                    print "startxref: ".yellow
+                    puts "#{@trailer.startxref}".red
+                end
+            end
+
+            def inspect
+                to_s
+            end
+        end
+
+        def to_s
+            puts
+
+            puts "---------- Header ----------".white.bold
+            print "  [+] ".magenta
+            print "Version: ".yellow
+            puts "#{@header.major_version}.#{@header.minor_version}".red
+
+            @revisions.each { |revision|
+                revision.to_s
+            }
+            puts
+        end
+
+        def inspect
+            to_s
+        end
     end
-  end
-end
 
+end