origami 1.0.2

data/origami.rb

@@ -0,0 +1,36 @@
+=begin
+
+= File
+	origami.rb
+
+= Info
+	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
+
+
+if RUBY_VERSION < '1.9'
+  class Fixnum
+    def ord; self; end
+  end
+
+  class Hash
+    alias key index
+  end
+end
+
+require 'origami/parsers/pdf'
+require 'origami/parsers/fdf'
+require 'origami/parsers/ppklite'
+

data/origami/3d.rb

@@ -0,0 +1,239 @@
+=begin
+
+= File
+	3d.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
+
+  class Annotation
+    
+    #
+    # 3D Artwork annotation.
+    #
+    class Artwork3D < Annotation
+      field   :Subtype,     :Type => Name, :Default => :"3D", :Version => "1.6", :Required => true
+      field   :"3DD",       :Type => [ Dictionary, Stream ], :Required => true
+      field   :"3DV",       :Type => Object
+      field   :"3DA",       :Type => Dictionary
+      field   :"3DI",       :Type => Boolean, :Default => true
+      field   :"3DB",       :Type => Array
+
+      class Activation < Dictionary
+        include StandardObject
+
+        module Events
+          PAGE_OPEN       = :PO
+          PAGE_CLOSE      = :PC
+          PAGE_VISIBLE    = :PV
+          PAGE_INVISIBLE  = :PI
+          USER_ACTIVATE   = :XA
+          USER_DEACTIVATE = :XD 
+        end
+
+        module State
+          UNINSTANCIATED  = :U
+          INSTANCIATED    = :I
+          LIVE            = :L
+        end
+
+        field   :A,         :Type => Name, :Default => Events::USER_ACTIVATE
+        field   :AIS,       :Type => Name, :Default => State::LIVE
+        field   :D,         :Type => Name, :Default => Events::PAGE_INVISIBLE
+        field   :DIS,       :Type => Name, :Default => State::UNINSTANCIATED
+        field   :TB,        :Type => Boolean, :Version => "1.7", :Default => true
+        field   :NP,        :Type => Boolean, :Version => "1.7", :Default => false
+      end
+    end
+
+  end
+
+  class U3DStream < Stream
+    include StandardObject
+
+    field   :Type,          :Type => Name, :Default => :"3D"
+    field   :Subtype,       :Type => Name, :Default => :U3D, :Required => true
+    field   :VA,            :Type => Dictionary
+    field   :DV,            :Type => Object
+    field   :Resources,     :Type => Dictionary
+    field   :OnInstantiate, :Type => Stream
+    field   :AN,            :Type => Dictionary
+
+    def onInstantiate(action)
+      self[:OnInstantiate] = action
+    end
+  end
+
+  class AnimationStyle3D < Dictionary
+    include StandardObject
+
+    module Styles
+      NONE                = :None
+      LINEAR              = :Linear
+      OSCILLATING         = :Oscillating
+    end
+
+    field   :Type,          :Type => Name, :Default => "3DAnimationStyle"
+    field   :Subtype,       :Type => Name, :Default => Styles::NONE
+    field   :PC,            :Type => Integer, :Default => 0
+    field   :TM,            :Type => Number, :Default => 1
+  end
+
+  class Reference3D < Dictionary
+    include StandardObject
+
+    field   :Type,          :Type => Name, :Default => :"3DRef"
+    field   :"3DD",         :Type => Stream
+  end
+
+  class View3D < Dictionary
+    include StandardObject
+
+    field   :Type,          :Type => Name, :Default => :"3DView"
+    field   :XN,            :Type => ByteString, :Required => true
+    field   :IN,            :Type => ByteString
+    field   :MS,            :Type => Name
+    field   :C2W,           :Type => Array
+    field   :U3DPath,       :Type => [ ByteString, Array ]
+    field   :CO,            :Type => Number
+    field   :P,             :Type => Dictionary
+    field   :O,             :Type => Stream
+    field   :BG,            :Type => Dictionary
+    field   :RM,            :Type => Dictionary, :Version => "1.7"
+    field   :LS,            :Type => Dictionary, :Version => "1.7"
+    field   :SA,            :Type => Array, :Version => "1.7"
+    field   :NA,            :Type => Array, :Version => "1.7"
+    field   :NR,            :Type => Boolean, :Version => "1.7", :Default => false
+  end
+
+  class Projection3D < Dictionary
+    include StandardObject
+
+    ORTHOGRAPHIC          = :O
+    PERSPECTIVE           = :P
+
+    module ClippingStyles
+      EXPLICIT_NEARFAR    = :XNF
+      AUTOMATIC_NEARFAR   = :ANF
+    end
+
+    module Scaling
+      WIDTH               = :W
+      HEIGHT              = :H
+      MINIMUM             = :Min
+      MAXIMUM             = :Max
+      ABSOLUTE            = :Absolute
+    end
+
+    field   :Subtype,       :Type => Name, :Default => ORTHOGRAPHIC
+    field   :CS,            :Type => Name, :Default => ClippingStyles::AUTOMATIC_NEARFAR
+    field   :F,             :Type => Number
+    field   :N,             :Type => Number
+    field   :FOV,           :Type => Number
+    field   :PS,            :Type => [ Number, Name ], :Default => Scaling::WIDTH
+    field   :OS,            :Type => Number, :Default => 1
+    field   :OB,            :Type => Name, :Version => "1.7", :Default => Scaling::ABSOLUTE
+  end
+
+  class Background3D < Dictionary
+    include StandardObject
+
+    field   :Type,          :Type => Name, :Default => :"3DBG"
+    field   :Subtype,       :Type => Name, :Default => :SC
+    field   :CS,            :Type => [ Name, Array ], :Default => Graphics::Color::Space::DEVICE_RGB
+    field   :C,             :Type => Object, :Default => [ 1, 1, 1 ]
+    field   :EA,            :Type => Boolean, :Default => false
+  end
+
+  class RenderMode3D < Dictionary
+    include StandardObject
+
+    module Modes
+      SOLID                           = :Solid
+      SOLID_WIREFRAME                 = :SolidWireFrame
+      TRANSPARENT                     = :Transparent
+      TRANSPARENT_WIREFRAME           = :TransparentWireFrame
+      BOUNDINGBOX                     = :BoundingBox
+      TRANSPARENT_BOUNDINGBOX         = :TransparentBoundingBox
+      TRANSPARENT_BOUNDINGBOX_OUTLINE = :TransparentBoundingBoxOutline
+      WIREFRAME                       = :WireFrame
+      SHADED_WIREFRAME                = :ShadedWireFrame
+      HIDDEN_WIREFRAME                = :HiddenWireFrame
+      VERTICES                        = :Vertices
+      SHADED_VERTICES                 = :ShadedVertices
+      ILLUSTRATION                    = :Illustration
+      SOLID_OUTLINE                   = :SolidOutline
+      SHADED_ILLUSTRATION             = :ShadedIllustration
+    end
+
+    field   :Type,          :Type => Name, :Default => :"3DRenderMode"
+    field   :Subtype,       :Type => Name, :Required => true, :Version => "1.7"
+    field   :AC,            :Type => Array, :Default => [ Graphics::Color::Space::DEVICE_RGB, 0, 0, 0]
+    field   :BG,            :Type => [ Name, Array ], :Default => :BG
+    field   :O,             :Type => Number, :Default => 0.5
+    field   :CV,            :Type => Number, :Default => 45
+  end
+
+  class LightingScheme3D < Dictionary
+    include StandardObject
+
+    module Styles
+      ARTWORK                         = :Artwork
+      NONE                            = :None
+      WHITE                           = :White
+      DAY                             = :Day
+      NIGHT                           = :Night
+      HARD                            = :Hard
+      PRIMARY                         = :Primary
+      BLUE                            = :Blue
+      RED                             = :Red
+      CUBE                            = :Cube
+      CAD                             = :CAD
+      HEADLAMP                        = :HeadLamp
+    end
+
+    field   :Type,          :Type => Name, :Default => :"3DLightingScheme"
+    field   :Subtype,       :Type => Name, :Version => "1.7", :Required => true
+  end
+
+  class CrossSection3D < Dictionary
+    include StandardObject
+
+    field   :Type,          :Type => Name, :Default => :"3DCrossSection"
+    field   :C,             :Type => Array, :Default => [ 0, 0, 0 ]
+    field   :O,             :Type => Array, :Version => "1.7", :Default => [ Null.new, 0, 0 ], :Required => true
+    field   :PO,            :Type => Number, :Default => 0.5
+    field   :PC,            :Type => Array, :Default => [ Graphics::Color::Space::DEVICE_RGB, 1, 1, 1 ]
+    field   :IV,            :Type => Boolean, :Default => false
+    field   :IC,            :Type => Array, :Default => [ Graphics::Color::Space::DEVICE_RGB, 0, 1 ,0]
+  end
+
+  class Node3D < Dictionary
+    include StandardObject
+
+    field   :Type,          :Type => Name, :Default => :"3DNode"
+    field   :N,             :Type => ByteString, :Version => "1.7", :Required => true
+    field   :O,             :Type => Number
+    field   :V,             :Type => Boolean
+    field   :M,             :Type => Array
+  end
+end

data/origami/acroform.rb

@@ -0,0 +1,321 @@
+=begin
+
+= File
+	acroform.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
+
+  class PDF
+
+    #
+    # Returns true if the document contains an acrobat form.
+    #
+    def has_form?
+      (not self.Catalog.nil?) and self.Catalog.has_key? :AcroForm
+    end
+
+    #
+    # Creates a new AcroForm with specified fields.
+    #
+    def create_acroform(*fields)
+      acroform = self.Catalog.AcroForm ||= InteractiveForm.new.set_indirect(true)
+      self.add_fields(*fields)
+
+      acroform
+    end
+
+    #
+    # Add a field to the Acrobat form.
+    # _field_:: The Field to add.
+    #
+    def add_fields(*fields)
+      raise TypeError, "Expected Field arguments" unless fields.all? { |f| f.is_a?(Field) }
+      
+      self.Catalog.AcroForm ||= InteractiveForm.new.set_indirect(true)
+      self.Catalog.AcroForm.Fields ||= []
+      
+      self.Catalog.AcroForm.Fields.concat(fields)
+      fields.each do |field| field.set_indirect(true) end
+      
+      self
+    end
+ 
+    #
+    # Returns an array of Acroform fields.
+    #
+    def fields
+      if self.has_form?
+        if self.Catalog.AcroForm.has_key?(:Fields)
+          self.Catalog.AcroForm[:Fields].map {|field| field.solve}
+        end
+      end
+    end
+
+    #
+    # Iterates over each Acroform Field.
+    #
+    def each_field(&b)
+      if self.has_form?
+        if self.Catalog.AcroForm.has_key?(:Fields)
+          self.Catalog.AcroForm[:Fields].each {|field| b.call(field.solve)}
+        end
+      end
+    end
+    
+    #
+    # Returns the corresponding named Field.
+    #
+    def get_field(name)
+      self.each_field do |field|
+        return field if field[:T].solve == name
+      end
+    end
+  end
+
+  #
+  # Class representing a interactive form Dictionary.
+  #
+  class InteractiveForm < Dictionary
+    
+    include StandardObject
+    
+    #
+    # Flags relative to signature fields.
+    #
+    module SigFlags
+      SIGNATURESEXIST = 1 << 0
+      APPENDONLY = 1 << 1
+    end
+    
+    field   :Fields,            :Type => Array, :Required => true, :Default => []
+    field   :NeedAppearances,   :Type => Boolean, :Default => false
+    field   :SigFlags,          :Type => Integer, :Default => 0
+    field   :CO,                :Type => Array, :Version => "1.3"
+    field   :DR,                :Type => Dictionary
+    field   :DA,                :Type => String
+    field   :Q,                 :Type => Integer
+    field   :XFA,               :Type => [ Stream, Array ]
+    
+  end
+  
+  module Field
+      
+    #
+    # Types of fields.
+    #
+    module Type
+      BUTTON = :Btn
+      TEXT = :Tx
+      CHOICE = :Ch
+      SIGNATURE = :Sig
+    end
+    
+    #
+    # Flags relative to fields.
+    #
+    module Flags
+      READONLY = 1 << 0
+      REQUIRED = 1 << 1
+      NOEXPORT = 1 << 2
+    end
+   
+    module TextAlign
+      LEFT = 0
+      CENTER = 1
+      RIGHT = 2
+    end
+      
+    def self.included(receiver)
+
+      receiver.field   :FT,     :Type => Name, :Default => Type::TEXT, :Required => true
+      receiver.field   :Parent, :Type => Dictionary
+      receiver.field   :Kids,   :Type => Array
+      receiver.field   :T,      :Type => String
+      receiver.field   :TU,     :Type => String, :Version => "1.3"
+      receiver.field   :TM,     :Type => String, :Version => "1.3"
+      receiver.field   :Ff,     :Type => Integer, :Default => 0
+      receiver.field   :V,      :Type => Object
+      receiver.field   :DV,     :Type => Object
+      receiver.field   :AA,     :Type => Dictionary, :Version => "1.2"
+
+      # Variable text fields
+      receiver.field   :DA,     :Type => String, :Default => "/F1 10 Tf 0 g", :Required => true
+      receiver.field   :Q,      :Type => Integer, :Default => TextAlign::LEFT
+      receiver.field   :DS,     :Type => ByteString, :Version => "1.5"
+      receiver.field   :RV,     :Type => [ String, Stream ], :Version => "1.5"
+        
+    end
+   
+    def pre_build
+      
+      if not self.T
+        self.T = "undef#{::Array.new(5) {(0x30 + rand(10)).chr}.join}"
+      end
+      
+      super
+    end
+
+    def onKeyStroke(action)
+    
+      unless action.is_a?(Action)
+        raise TypeError, "An Action object must be passed."
+      end
+      
+      self.AA ||= AdditionalActions.new
+      self.AA.K = action
+        
+    end
+
+    def onFormat(action)
+    
+      unless action.is_a?(Action)
+        raise TypeError, "An Action object must be passed."
+      end
+      
+      self.AA ||= AdditionalActions.new
+      self.AA.F = action
+        
+    end
+
+    def onValidate(action)
+    
+      unless action.is_a?(Action)
+        raise TypeError, "An Action object must be passed."
+      end
+      
+      self.AA ||= AdditionalActions.new
+      self.AA.V = action
+        
+    end
+
+    def onCalculate(action)
+    
+      unless action.is_a?(Action)
+        raise TypeError, "An Action object must be passed."
+      end
+      
+      self.AA ||= AdditionalActions.new
+      self.AA.C = action
+        
+    end
+
+    class Subform < Dictionary
+      include StandardObject
+      include Field
+
+      def add_fields(*fields)
+        self.Kids ||= []
+        self.Kids.concat(fields)
+
+        fields.each do |field| field.Parent = self end
+
+        self
+      end
+    end
+
+    class AdditionalActions < Dictionary
+      
+      include StandardObject
+
+      field   :K,     :Type => Dictionary, :Version => "1.3"
+      field   :F,     :Type => Dictionary, :Version => "1.3"
+      field   :V,     :Type => Dictionary, :Version => "1.3"
+      field   :C,     :Type => Dictionary, :Version => "1.3"
+    end
+    
+    class SignatureLock < Dictionary
+      
+      include StandardObject
+      
+      module Actions
+        ALL = :All
+        INCLUDE = :Include
+        EXCLUDE = :Exclude
+      end
+      
+      field   :Type,            :Type => Name, :Default => :SigFieldLock
+      field   :Action,          :Type => Name, :Default => Actions::ALL, :Required => true
+      field   :Fields,          :Type => Array
+
+      def pre_build
+        
+        if self.Action and self.Action != Actions::ALL
+          self.Fields ||= []
+        end
+        
+        super
+      end
+      
+    end
+    
+    class SignatureSeedValue < Dictionary
+      
+      include StandardObject
+      
+      module Digest
+        SHA1 = :SHA1
+        SHA256 = :SHA256
+        SHA384 = :SHA384
+        SHA512 = :SHA512
+        RIPEMD160 = :RIPEMD160
+      end
+     
+      field   :Type,            :Type => Name, :Default => :SV
+      field   :Filter,          :Type => Name
+      field   :SubFilter,       :Type => Array
+      field   :DigestMethod,    :Type => Array, :Default => Digest::SHA1, :Version => "1.7"
+      field   :V,               :Type => Real, :Default => 1.0
+      field   :Cert,            :Type => Dictionary
+      field   :Reasons,         :Type => Array
+      field   :MDP,             :Type => Dictionary, :Version => "1.6"
+      field   :TimeStamp,       :Type => Dictionary, :Version => "1.6"
+      field   :LegalAttestation,  :Type => Array, :Version => "1.6"
+      field   :AddRevInfo,      :Type => Boolean, :Default => false, :Version => "1.7"
+      field   :Ff,              :Type => Integer, :Default => 0
+     
+    end
+    
+    class CertificateSeedValue < Dictionary
+      
+      include StandardObject
+      
+      module URL
+        BROWSER = :Browser
+        ASSP = :ASSP
+      end
+      
+      field   :Type,            :Type => Name, :Default => :SVCert
+      field   :Subject,         :Type => Array
+      field   :SubjectDN,       :Type => Array, :Version => "1.7"
+      field   :KeyUsage,        :Type => Array, :Version => "1.7"
+      field   :Issuer,          :Type => Array
+      field   :OID,             :Type => Array
+      field   :URL,             :Type => ByteString
+      field   :URLType,         :Type => Name, :Default => URL::BROWSER, :Version => "1.7"
+      field   :Ff,              :Type => Integer, :Default => 0
+
+    end
+    
+  end
+
+end