origami 1.0.2

data/bin/pdfmetadata

@@ -0,0 +1,123 @@
+#!/usr/bin/env ruby
+
+=begin
+
+= Author: 
+  Guillaume Delugré <guillaume/at/security-labs.org>
+
+= Info:
+  Prints out the metadata contained in a PDF document.
+
+= 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__)}/.."
+  $: << ORIGAMIDIR
+  require 'origami'
+end
+include Origami
+
+require 'optparse'
+
+class OptParser
+  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/
+
+Options:
+USAGE
+
+  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("-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
+    end
+  end
+
+  def self.parse(args)
+    options = 
+    {
+      :output => STDOUT,
+    }
+
+    self.parser(options).parse!(args)
+
+    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?
+      colorprint "[*] Document information dictionary:\n", Colors::MAGENTA
+
+      docinfo = pdf.get_document_info
+      docinfo.each_pair do |name, item|
+        colorprint name.value.to_s.ljust(20, ' '), Colors::GREEN
+        puts ": #{item.solve.value}"
+      end
+      puts
+    end
+  end
+
+  if @options[:doc_stream]
+    if pdf.has_metadata?
+      colorprint "[*] Metadata stream:\n", Colors::MAGENTA
+
+      metadata = pdf.get_metadata
+      metadata.each_pair do |name, item|
+        colorprint name.ljust(20, ' '), Colors::GREEN
+        puts ": #{item}"
+      end
+    end
+  end
+
+rescue SystemExit
+rescue Exception => e
+  STDERR.puts "#{e.class}: #{e.message}"
+  exit 1
+end
+

data/bin/pdfsh

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

data/bin/pdfwalker

@@ -0,0 +1,7 @@
+#!/usr/bin/env ruby
+
+$:.unshift "#{File.dirname(__FILE__)}" if RUBY_VERSION >= '1.9'
+require 'gui/walker'
+
+PDFWalker::Walker.start
+

data/bin/shell/.irbrc

@@ -0,0 +1,104 @@
+begin
+  require 'origami'
+rescue LoadError
+  ORIGAMIDIR = "#{File.dirname(__FILE__)}/../.."
+  $: << ORIGAMIDIR
+  require 'origami'
+end
+include Origami
+
+require 'console.rb'
+require 'readline'
+
+DEFAULT_BANNER = "Welcome to the PDF shell (Origami release #{Origami::VERSION})\n\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 []
+        end
+
+        return space.public_methods
+    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
+
+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
+
+end
+
+Origami.colorprint(DEFAULT_BANNER, Origami::Colors::GREEN)
+#set_completion
+set_prompt
+

data/bin/shell/console.rb

@@ -0,0 +1,136 @@
+=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/>.
+
+=end
+
+require 'hexdump'
+
+if RUBY_VERSION < '1.9'
+  def Kernel.spawn(cmd)
+    fork do
+      exec(cmd)
+    end
+  end
+end
+
+module Origami
+
+  module Object
+    def inspect
+      to_s
+    end
+  end
+
+  unless (RUBY_VERSION < '1.9' and RUBY_PLATFORM =~ /win32/)
+    require 'tempfile'
+
+    class Stream
+      def edit(editor = 'vim')
+        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
+
+    class Page
+      def edit 
+        self.Contents.edit
+      end
+    end
+  end
+  
+  class PDF
+     
+     class Revision
+      def to_s
+        colorprint("----------  Body  ----------\n", Colors::WHITE, true)
+        @body.each_value { |obj|
+          colorprint("#{obj.reference.to_s.rjust(8,' ')}".ljust(10), Colors::MAGENTA)
+          colorprint("#{obj.type}\n", 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
+        #}
+        colorprint("---------- Trailer ---------\n", Colors::WHITE, true) 
+        if not @trailer.dictionary
+          set_fg_color(Colors::BLUE, true) {
+            puts "  [x] No trailer found."
+          }
+        else
+          @trailer.dictionary.each_pair { |entry, value|
+            colorprint("  [*] ", Colors::MAGENTA)
+            colorprint("#{entry.to_s}: ", Colors::YELLOW)
+            colorprint("#{value.to_s}\n", Colors::RED)
+          }
+          colorprint("  [+] ", Colors::MAGENTA)
+          colorprint("startxref: ", Colors::YELLOW)
+          colorprint("#{@trailer.startxref}\n", Colors::RED)
+        end
+      end
+      
+      def inspect
+        to_s
+      end
+     end
+    
+    def to_s
+      puts
+      
+      colorprint("---------- Header ----------\n", Colors::WHITE, true)
+      colorprint("  [+] ", Colors::MAGENTA)
+      colorprint("Major version: ", Colors::YELLOW)
+      colorprint("#{@header.majorversion}\n", Colors::RED)
+      colorprint("  [+] ", Colors::MAGENTA)
+      colorprint("Minor version: ", Colors::YELLOW)
+      colorprint("#{@header.minorversion}\n", Colors::RED)
+      
+      @revisions.each { |revision|
+        revision.to_s
+      }
+      puts
+    end
+    
+    def inspect
+      to_s
+    end
+  end
+end
+

data/bin/shell/hexdump.rb

@@ -0,0 +1,83 @@
+=begin
+
+= File
+	hexdump.rb
+
+= Info
+	This file is part of Origami, PDF manipulation framework for Ruby
+	Copyright (C) 2009	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 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 General Public License for more details.
+
+  You should have received a copy of the GNU General Public License
+  along with Origami.  If not, see <http://www.gnu.org/licenses/>.
+
+=end
+
+if RUBY_VERSION < '1.9'
+  class Fixnum
+    def ord; self; end
+  end
+end
+
+class String #:nodoc:
+
+  def hexdump(bytesperline = 16, upcase = true, offsets = true, delta = 0)
+    
+    dump = ""
+    counter = 0
+    
+    while counter < length
+      
+      offset = sprintf("%010X", counter + delta)
+      
+      linelen = (counter < length - bytesperline) ? bytesperline : (length - counter)
+      bytes = ""
+      linelen.times do |i|
+        
+        byte = self[counter + i].ord.to_s(base=16)
+        if byte.size < 2 then byte.insert(0, "0") end
+        bytes << byte
+        bytes << " " unless i == bytesperline - 1
+        
+      end
+
+      ascii = self[counter, linelen].ascii_print
+      
+      if upcase
+        offset.upcase!
+        bytes.upcase!
+      end
+      
+      if RUBY_PLATFORM =~ /win32/
+        dump << "#{offset if offsets}  #{bytes.to_s.ljust(bytesperline * 3 - 1)}  #{ascii}\n"
+      else
+        dump << "#{Origami.colorize(offset, Colors::YELLOW) if offsets}  #{Origami.colorize(bytes.to_s.ljust(bytesperline * 3 - 1), Colors::BRIGHT_GREY)}  #{ascii}\n"
+      end
+
+      counter += bytesperline
+    end
+
+    dump
+  end
+  
+  def ascii_print
+
+    printable = ""
+    self.each_byte { |c|
+      if c >= ' '[0].ord && c <= '~'[0].ord then printable << c else printable << '.' end
+    }
+
+    return printable
+  end
+  
+end