veracode 1.0.0.alpha2 → 1.0.0.alpha7

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: e57a698db0528192fcbd096c38cb4236ef54869f
+  data.tar.gz: f4b1f130125636850f94bddd03356b35ab6678f6
+SHA512:
+  metadata.gz: cbfc1cfc6c98a2dde96540a8f5c2f05baa01f4076e3204c49edcec5db86db752f641bb923b5f998a85e65fabe45cd1b0e05ac5074acdb71658315b1c8459f02c
+  data.tar.gz: 0935f7c827279516a28e56068e9b00c45c214484316f38e85ab618810dfa29811d368498a0822b8705a92bf22322946645e0bf9873e502db7f032fb294ba3665

data/bin/veracode

@@ -16,12 +16,10 @@ require 'veracode'
 require 'veracode/version'
 
 $options =  {
-              :phase1 => false, 
-              :phase2 => false, 
-              :phase3 => true,
               :archive_source => true,
               :include_inherited => false,
               :jruby => false,
+              :environment => false,
             }
 
 subcommand = ARGV.shift
@@ -34,22 +32,13 @@ case subcommand
         $options[:verbose] = true
       end
 
-      opts.on("-a", "--all", "Archive objects at all stages") do
-        $options[:phase1] = true
-        $options[:phase2] = true
-      end
-
-      opts.on("-f", "--file", "Disassemble .rb files") do
-        $options[:disasm] = true
-      end
-
       opts.on("-j", "--jruby", "Force JRuby mode") do
         $options[:jruby] = true
       end
 
-      opts.on("--[no-]source", "[Don't] Include source code in archive") do |s|
-        $options[:archive_source] = s
-      end
+      # opts.on("--[no-]source", "[Don't] Include source code in archive") do |s|
+      #   $options[:archive_source] = s
+      # end
 
       opts.on("-D", "--debug", "Enable debug output") do
         $DEBUG = true
@@ -73,6 +62,23 @@ case subcommand
                     "       #{opts.program_name} help"
     end.parse!
 
+  when "environment", "env"
+    $options[:environment] = true
+
+    OptionParser.new do |opts|
+      opts.banner = "Usage: veracode environment [options]"
+
+      opts.on("-v", "--verbose", "Run verbosely") do
+        $options[:verbose] = true
+      end
+
+      opts.on("-D", "--debug", "Enable debug output") do
+        $DEBUG = true
+      end
+
+    end.parse!
+    Veracode.prepare
+
   else
     $stderr.puts "#{subcommand.dump} is not a valid subcommand"
 

data/lib/veracode.rb

@@ -1,24 +1,35 @@
+require 'pathname'
+require 'set'
 require 'zlib'
-require 'zip/zip'
-require "veracode/version"
+require 'zip'
+require 'veracode/version'
+require 'veracode/schema'
+require 'veracode/gems'
 
 module Veracode
   @run_id = nil
+  @required_libs = Set.new
 
   # Metadata and method disassemblies for all Modules (.txt.gz)
   @disasmlog = nil
-  @disasmlog_filename = "disasm.txt.gz"
+  @disasmlog_filename = 'disasm.txt.gz'
 
   # Error log including capture of STDERR and any errors generated by the gem (.log)
   @errorlog = nil
-  @errorlog_filename = "error.log"
+  @errorlog_filename = 'error.log'
 
   # Index file containing the names of files present in the application directory (.txt)
-  @index_filename = "index.txt"
+  @index_filename = 'index.txt'
 
   # Manifest file containing original names of all files in archive (.txt)
   @manifest = []
-  @manifest_filename = "manifest.txt"
+  @manifest_filename = 'manifest.txt'
+
+  # XML file containing list of gems used by application
+  @gems_filename = 'gems.xml'
+
+  # Archive version file containing archive format version identifier
+  @version_filename = 'version.txt'
 
   # The final archive that will be uploaded to Veracode for analysis (.zip)
   @archive = nil
@@ -29,6 +40,7 @@ module Veracode
   def self.init
     @run_id = Time.now.strftime("%Y%m%d%H%M%S")
     @archive_dirname = File.join("tmp","veracode-#{@run_id}")
+    @required_libs.merge(["pathname", "set", "zlib", "zip/zip", "veracode"])
 
     if !Dir.exists?("tmp")
       begin
@@ -57,9 +69,11 @@ module Veracode
     @disasmlog_filename = File.join(@archive_dirname, @disasmlog_filename)
     @index_filename     = File.join(@archive_dirname, @index_filename)
     @manifest_filename  = File.join(@archive_dirname, @manifest_filename)
+    @gems_filename      = File.join(@archive_dirname, @gems_filename)
+    @version_filename   = File.join(@archive_dirname, @version_filename)
 
     # Try touching each of the files to be written
-    [@disasmlog_filename, @errorlog_filename, @index_filename, @manifest_filename].each {|f|
+    [@disasmlog_filename, @errorlog_filename, @index_filename, @manifest_filename, @gems_filename, @version_filename].each {|f|
       begin
         File.open(f, "wb") {}
       rescue Exception => e
@@ -76,6 +90,7 @@ module Veracode
       log_error "RUBY_DESCRIPTION: #{RUBY_DESCRIPTION}"
       log_error "RAILS_VERSION: " + `rails --version`.chomp
       log_error "GEM_VERSION: #{Veracode::VERSION}"
+      log_error "ARCHIVE_VERSION: #{Veracode::ARCHIVE_VERSION}"
       log_error "PWD: #{Dir.pwd.to_s.dump}"
       log_error "APP_NAME: #{APP_NAME.dump}"
       log_error "RUNID: #{@run_id}"
@@ -86,6 +101,16 @@ module Veracode
       STDERR.reopen(@errorlog)
     end
 
+    begin
+      File.open(@version_filename, "wb") {|version_file|
+        version_file.puts Veracode::ARCHIVE_VERSION
+      }
+    rescue Exception => e
+      log_error "Unable to write to archive version file #{@version_filename}: #{e.message}"
+    end
+
+    list_gems
+
     index_application
 
     @manifest += Dir.glob("*").keep_if {|f| File.file?(f)}
@@ -128,28 +153,47 @@ module Veracode
         }
       }
     rescue Exception => e
-      log_error e.message
-      $stderr.puts "Unable to write manifest file #{@manifest_filename}: #{e.message}"
+      log_error "Unable to write manifest file #{@manifest_filename}: #{e.message}"
+      puts "Unable to write manifest file #{@manifest_filename}: #{e.message}"
     end
 
     @errorlog.flush
 
     begin
-      Zip::ZipFile.open(@archive_filename, Zip::ZipFile::CREATE) { |zf|
-        @manifest.each {|file|
-
-          if file.start_with?(@archive_dirname)
-            name_in_archive = file.sub(/^#{@archive_dirname + File::SEPARATOR}/,"")          
-          else
-            name_in_archive = File.join(APP_NAME, file)
-          end        
-
-          puts "Adding #{file} to archive as #{name_in_archive}" if $options[:verbose]
-          zf.add(name_in_archive, file)
+      if Gem.loaded_specs.keys.include?("zipruby")
+        log_error "zipruby gem detected, using it instead of rubyzip for creating archive"
+        @errorlog.flush
+        Zip::Archive.open(@archive_filename, Zip::CREATE) { |ar|
+          @manifest.each { |file|
+
+            if file.start_with?(@archive_dirname)
+              name_in_archive = file.sub(/^#{@archive_dirname + File::SEPARATOR}/,"")
+            else
+              name_in_archive = File.join(APP_NAME, file)
+            end        
+
+            puts "Adding #{file} to archive as #{name_in_archive}" if $options[:verbose]
+            ar.add_file(name_in_archive, file)
+          }
         }
-      }      
+      else
+        Zip::File.open(@archive_filename, Zip::File::CREATE) { |zf|
+          @manifest.each { |file|
+
+            if file.start_with?(@archive_dirname)
+              name_in_archive = file.sub(/^#{@archive_dirname + File::SEPARATOR}/,"")
+            else
+              name_in_archive = File.join(APP_NAME, file)
+            end        
+
+            puts "Adding #{file} to archive as #{name_in_archive}" if $options[:verbose]
+            zf.add(name_in_archive, file)
+          }
+        }
+      end
     rescue Exception => e
-      $stderr.puts "Unable to create archive #{@manifest_filename}: #{e.message}"
+      log_error "Unable to create archive #{@manifest_filename}: #{e.message}"
+      puts "Unable to create archive #{@manifest_filename}: #{e.message}"
       exit
     end
 
@@ -157,7 +201,7 @@ module Veracode
   end
 
   def self.cleanup
-    [@disasmlog_filename, @errorlog_filename, @index_filename, @manifest_filename].each {|f|
+    [@disasmlog_filename, @errorlog_filename, @index_filename, @manifest_filename, @gems_filename, @version_filename].each {|f|
       begin
         File.delete(f)
       rescue Exception => e
@@ -200,20 +244,33 @@ module Veracode
 
   ##############################################################################
   # Helpers
+  def self.cond_require(lib)
+    if @required_libs.add?(lib)
+      return require lib
+    end
+    return false
+  end
+
   def self.glob_require(files)
+    any_new = false
+    total, count = 0, 0
     Dir.glob(files) do |f|
       print "Requiring #{f.to_s} " if $options[:verbose]
       
       begin
-        require File.expand_path(f)
+        required = require File.expand_path(f)
       rescue Exception => e
         puts "(failed: #{e.message})" if $options[:verbose]
         log_error "Unable to require #{File.expand_path(f).to_s.dump} (#{e.message})"
       else
-        puts "(OK)" if $options[:verbose]
+        puts "(OK: #{(required ? "required" : "already required")})" if $options[:verbose]
       end
-
+      any_new |= required
+      total += 1
+      count += 1 if required
     end
+    puts "#{count}/#{total} files were required" if $options[:verbose]
+    any_new
   end
 
   def self.safe_name(o)
@@ -221,7 +278,11 @@ module Veracode
     when o == ActiveSupport::TimeWithZone
       "ActiveSupport::TimeWithZone"
     when o.is_a?(Module)
-      ( o.name.nil? ? o.to_s : o.name )
+      begin
+        ( o.name.nil? ? o.to_s : o.name.to_s )
+      rescue
+        o.to_s
+      end
     when o.is_a?(Method), o.is_a?(UnboundMethod)
       o.name.to_s
     else
@@ -286,7 +347,13 @@ module Veracode
   def self.prepare_archive
     @disasmlog = Zlib::GzipWriter.new(File.open(@disasmlog_filename, "wb"), nil, nil)
     @disasmlog.puts "#{RUBY_ENGINE}-#{RUBY_VERSION}-p#{RUBY_PATCHLEVEL}"
-    @disasmlog.puts "# " + `rails --version`.chomp
+    if $options[:environment]
+      @disasmlog.puts "# EnvironmentDef %s-%s_rails-%s" % [RUBY_ENGINE, RUBY_VERSION, Rails.version]
+    else
+      @disasmlog.puts "# Environment %s-%s_rails-%s" % [RUBY_ENGINE, RUBY_VERSION, Rails.version]
+    end
+    @disasmlog.puts "# Ruby #{RUBY_ENGINE}-#{RUBY_VERSION}"
+    @disasmlog.puts "# Rails #{Rails.version}"
     @disasmlog.puts
   end
 
@@ -363,7 +430,7 @@ module Veracode
         m.included_modules.map {|m| "include #{m.inspect.dump}\n" }.join : 
         ""
     ) + 
-    ( m.singleton_class.included_modules.count > 0 ? 
+    ( m.respond_to?(:singleton_class) && m.singleton_class.included_modules.count > 0 ? 
         m.singleton_class.included_modules.map {|m| "extend #{m.inspect.dump}\n" }.join : 
         ""
     )
@@ -399,23 +466,26 @@ module Veracode
       end
     end
 
-    if m.respond_to?(:global_variables)
-      m.global_variables.each do |v_symbol|
-        begin
-          v = eval(v_symbol.to_s)
-          formatted_contents += format_variable(v_symbol, v, "global")
-        rescue Exception => e
-          log_error "Error archiving global variable #{v_symbol.to_s.dump}: #{e.message}"
-          formatted_contents += format_variable(v_symbol, :veracode_nil, "global")
-        end
-        
+    begin
+      if m == Kernel
+        m.global_variables.each do |v_symbol|
+          begin
+            v = eval(v_symbol.to_s)
+            formatted_contents += format_variable(v_symbol, v, "global")
+          rescue Exception => e
+            log_error "Error archiving global variable #{v_symbol.to_s.dump}: #{e.message}"
+            formatted_contents += format_variable(v_symbol, :veracode_nil, "global")
+          end
+        end        
       end
+    rescue Exception => e
+      # m.respond_to?(:global_variables) was throwing exceptions
     end
 
     %w[ public protected private ].each {|p|
       get_methods = (p + "_instance_methods").to_sym
-      if m.respond_to?(get_methods) && m.send(get_methods, $options[:include_inherited]).count > 0
-        m.send(get_methods, $options[:include_inherited]).each do |m_symbol|
+      if m.respond_to?(get_methods) && m.__send__(get_methods, $options[:include_inherited]).count > 0
+        m.__send__(get_methods, $options[:include_inherited]).each do |m_symbol|
           begin
             method = m.instance_method(m_symbol)
             formatted_contents += format_method(method, "#{p.to_s}_instance", with_disasm)
@@ -467,7 +537,12 @@ module Veracode
   # Archiving Objects
   def self.archive(objects, with_disasm=true)
 
-    objects = objects - [Veracode]
+    objects = objects - [
+                          Veracode, 
+                          Veracode::ActiveRecord, 
+                          Veracode::ActiveRecord::Model, 
+                          Veracode::ActiveRecord::Schema,
+                         ]
 
     if $options[:verbose]
       puts "Archiving #{objects.count.to_s} objects" + (with_disasm ? " with disassembly" : "")
@@ -493,27 +568,11 @@ module Veracode
   end
 
 
-  def self.archive_schema
-    puts "Archiving schema information" if $options[:verbose]
-    schema_file = File.join("db", "schema.rb")
-    begin
-      add_to_archive  %Q|module "Veracode::Schema"\n| + 
-                      %Q|singleton_method "main" []\n| + 
-                      RubyVM::InstructionSequence.compile_file(schema_file).disasm + 
-                      %Q|== end disasm\n| + 
-                      %Q|endmodule\n\n|
-    rescue Exception => e
-      puts "Unable to retrieve schema information from 'db/schema.rb'. Are your migrations up to date?"
-      log_error "Unable to archive 'db/schema.rb' (#{e.message})"
-    end
-  end
-
-
   def self.compile_templates
 
     begin
-      require 'action_view' unless defined? ActionView
-      require 'action_controller' unless defined? ActionController
+      cond_require 'action_view' unless defined? ActionView
+      cond_require 'action_controller' unless defined? ActionController
     rescue Exception => e
       log_error "Unable to satisfy haml dependencies (#{e.message})"
       return
@@ -536,12 +595,13 @@ module Veracode
     return unless templates.count > 0
 
     puts "Found #{templates.count} templates" if $options[:verbose]
+    log_error "Found #{templates.count} templates"
 
     haml_templates = templates.grep(/\.haml$/)
     if haml_templates.any?
       begin
-        require 'haml' unless defined? Haml
-        require 'haml/template/plugin' unless defined? Haml::Plugin
+        cond_require 'haml' unless defined? Haml
+        cond_require 'haml/template/plugin' unless defined? Haml::Plugin
       rescue Exception => e
         puts "Unable to satisfy haml dependencies"
         log_error "Unable to satisfy haml dependencies (#{e.message})"
@@ -558,13 +618,16 @@ module Veracode
       puts "Compiling template #{template}" if $options[:verbose]
 
       begin
+        # This render will fail, but will trigger compilation of template
         view.render(:file => template)
       rescue Exception => e
         log_error "Compiled template #{template} #{e.message}"
       end
     }
 
-    puts "Compiled #{ActionView::CompiledTemplates.instance_methods.count.to_s} templates " if $options[:verbose]
+    puts "Compiled #{ActionView::CompiledTemplates.instance_methods.count.to_s} templates" if $options[:verbose]
+    log_error "Compiled #{ActionView::CompiledTemplates.instance_methods.count.to_s} templates"
+    log_error "Not all templates were compiled" if ActionView::CompiledTemplates.instance_methods.count < templates.count
   end
 
   def self.compile_erb_templates
@@ -639,9 +702,9 @@ module Veracode
     return unless templates.count > 0
 
     begin
-      require 'action_view'
-      require 'haml'
-      require 'haml/template/plugin'
+      cond_require 'action_view'
+      cond_require 'haml'
+      cond_require 'haml/template/plugin'
     rescue Exception => e
       log_error "Unable to satisfy haml dependencies (#{e.message})"
       return
@@ -680,6 +743,59 @@ module Veracode
 
   end
 
+  def self.require_libs(lib_paths)
+    for lib_path in lib_paths
+      dirsToProcess = [Pathname(lib_path)]
+      until dirsToProcess.count == 0 || !Dir.exists?(dirsToProcess[0])
+        currentDir = dirsToProcess.delete_at(0)
+        for child in currentDir.children
+          if child.directory?
+            dirsToProcess[dirsToProcess.count] = child
+            base = child.to_s.partition("#{lib_path}/")[2]
+            lib = ""
+            for part in base.split('/').reverse
+              lib = "#{part}/#{lib}"
+              lib = lib[0..lib.length-2] if lib[lib.length-1] == '/'
+            begin
+              if cond_require lib
+                puts "requiring #{lib}" if $options[:verbose]
+              end
+            rescue Exception => e
+            end
+          end
+        end
+      end
+    end
+  end
+end
+
+def self.require_rails(gemdir)
+  dirsToProcess = [Pathname(gemdir)]
+  until dirsToProcess.count == 0
+    currentDir = dirsToProcess.delete_at(0)
+    for child in currentDir.children
+      if child.directory?
+        dirsToProcess[dirsToProcess.count] = child
+      end
+      base = child.to_s.partition("#{gemdir}/")[2]
+      if base.index("action_controller") != nil || base.index("action_view") != nil || base.index("active_record") != nil
+        lib = ""
+        for part in base.split('/').reverse
+          lib = "#{part}/#{lib}"
+          lib = lib[0..lib.length-2] if lib[lib.length-1] == '/'
+          lib.chomp!(File.extname(lib))
+          begin
+            if cond_require lib
+              puts "requiring #{lib}" if $options[:verbose]
+            end
+          rescue Exception => e
+          end
+        end
+      end
+    end
+  end
+end
+
 
 
 
@@ -697,16 +813,6 @@ module Veracode
       puts
     end
 
-    if $options[:disasm]
-      rbfiles = File.join("**", "*.rb")
-      Dir[rbfiles].each do |f|
-        puts RubyVM::InstructionSequence.compile_file(f).disasm
-        puts
-      end
-      exit
-    end
-
-    prepare_archive
 
     ################################################################
     ## phase 1 - Create baseline
@@ -715,10 +821,6 @@ module Veracode
     puts "Phase 1 - Initial State" if $options[:verbose]
     self.stats if $options[:verbose]
     
-    if $options[:phase1]
-      puts "Processing and disassembling Ruby standard classes and modules"
-      archive(@modules)
-    end
     ## /phase 1 - Create baseline
     ################################################################
 
@@ -729,10 +831,7 @@ module Veracode
 
     puts "Phase 2 - Load Rails" if $options[:verbose]
     begin
-      require "rails"
-      require 'action_controller'
-      require 'action_view'
-      require 'active_record'
+      require "rails/all"
     rescue Exception => e
       puts "Unable to require rails: #{e.message}"
       log_error "Unable to require rails: #{e.message}"
@@ -741,26 +840,6 @@ module Veracode
       puts "Required rails" if $options[:verbose]
     end
 
-    self.update
-
-    self.stats if $options[:verbose]
-
-    if $options[:phase2]
-      puts "Processing and disassembling Rails classes and modules"
-      archive(@modules)
-    end
-    
-    self.rebaseline
-    ## /phase 2 - Require rails
-    ################################################################
-    
-    
-    
-    ################################################################
-    # phase 3 - require app
-
-    puts "Phase 3 - Imitate Rails" if $options[:verbose]
-
     ## Imitate script/rails
     # APP_PATH = File.expand_path('config/application')
       # APP_PATH is already set in bin/veracode
@@ -774,7 +853,7 @@ module Veracode
       # require 'rails/commands'
 
     ## Imitate rails/commands when console
-    glob_require 'rails/commands/console'
+    cond_require 'rails/commands/console.rb'
     # require APP_PATH # => config/application.rb
 
     glob_require "config/application.rb"
@@ -785,32 +864,66 @@ module Veracode
     
     # Imitate Rails::Console.initialize_console
     # require "pp"
-    glob_require "rails/console/app"
-    glob_require "rails/console/helpers"
+    cond_require "rails/console/app.rb"
+    cond_require "rails/console/helpers.rb"
 
-    glob_require "app/models/**/*.rb"
-    glob_require "app/helpers/**/*.rb"
-    glob_require "app/controllers/application_controller.rb"
-    glob_require "app/controllers/**/*.rb"
+    if $options[:environment]
+      @stdlib = $:
+      @gemdir = Gem.dir
+
+      require_libs(@stdlib)
+      require_rails(@gemdir)
+    end
+
+    self.rebaseline
+
+    self.stats if $options[:verbose]
+
+    ## /phase 2 - Require rails
+    ################################################################
+    
+    
+    
+    ################################################################
+    # phase 3 - require app
+
+    puts "Phase 3 - Imitate Rails" if $options[:verbose]
+
+    any_new = true
+    while any_new
+      any_new = false
+      any_new |= glob_require "lib/**/*.rb"
+      any_new |= glob_require "app/**/*.rb"
+      puts "new successful requires? #{any_new.to_s}" if $options[:verbose]
+    end
 
     compile_templates
 
     self.update
     self.stats if $options[:verbose]
 
-    if $options[:phase3]
-      puts "Processing and disassembling #{APP_NAME} classes and modules"
+    # Ensure compiled templates are fully disassembled in archive
+    @baseline_modules.delete(ActionView::CompiledTemplates)
+
+    if $options[:environment]
+      puts "Processing and disassembling environment"
+      archive(@modules.reject  {|o| safe_name(o) =~ /^#<(Class|Module):0x[0-9a-f]+>/i }
+                      .reject  {|o| safe_name(o) =~ /^Veracode/ }
+                      .reject  {|o| safe_name(o) =~ /^EmptyRails/ }
+                      .reject  {|o| safe_name(o) =~ /^ActionView::CompiledTemplates$/ }, false)
+    else
+      puts "Processing Ruby and Rails classes and modules"
       archive(@baseline_modules, false)
+      add_to_archive "\n# Phase 3 - App disassembly\n"
+      puts "Processing and disassembling #{APP_NAME} classes and modules"
       archive(@modules - @baseline_modules, true)
+      archive_schema
     end
 
-    archive_schema
-
     ## /phase 3 - require app
     ################################################################
 
     finalize_archive
-
     pack_manifest
     cleanup
 

data/lib/veracode/gems.rb

@@ -0,0 +1,49 @@
+module Veracode
+  SupportedGems = %w{
+    actionmailer
+    actionpack
+    activemodel
+    activerecord
+    activeresource
+    activesupport
+    arel
+    builder
+    erubis
+    haml
+    haml-rails
+    rails
+    railties
+    veracode
+  }
+
+  def self.list_gems
+
+    gems = `bundle list`.each_line
+                        .reject {|line| line !~ /^  \* /}
+                        .map {|line| line[4..-1]}
+                        .map {|line| line.split.first}
+
+    begin
+      File.open(@gems_filename, "wb") {|gems_file|
+        gems_file.puts '<messages>'
+        gems.each {|gem|
+          gems_file << <<GEMS_XML
+  <message>
+    <platform>ruby</platform>
+    <name>#{gem}</name>
+    <detailed_message>#{gem}</detailed_message>
+    <token>#{gem}</token>
+    <package>rubygem.#{gem}</package>
+    <errorlevel>#{(SupportedGems.include?(gem) ? "info" : "warn" )}</errorlevel>
+    <type>framework_unsupported</type>
+  </message>
+GEMS_XML
+        }
+        gems_file.puts '</messages>'
+      }
+    rescue Exception => e
+      log_error "Unable to write to gem list to file #{@gems_filename}: #{e.message}"
+    end
+
+  end
+end

data/lib/veracode/schema.rb

@@ -0,0 +1,104 @@
+module Veracode
+  module ActiveRecord
+    class Model
+
+      attr_reader :name, :attributes
+
+      def initialize(name)
+        @name = name
+        @attributes = Array.new
+      end
+
+      %w(
+         binary boolean date datetime decimal float integer primary_key string text time timestamp
+      ).map(&:to_sym).each do |meth|
+        define_method(meth) do |name, *rest|
+          @attributes << [name, meth]
+        end
+      end
+
+    end
+
+    class Schema
+
+      def self.define(info={}, &block)
+        Schema.new.instance_eval(&block)
+      end
+
+      def create_table(name, options={})
+        td = Model.new(name)
+        td.integer('id')
+        yield td if block_given?
+        Veracode.add_to_archive Veracode.format_variable("@@#{td.name}", td.attributes, 'class')
+      end
+
+      # ActiveRecord::ConnectionAdapters::SchemaStatements
+      # http://api.rubyonrails.org/classes/ActiveRecord/ConnectionAdapters/SchemaStatements.html
+      def add_column(table_name, column_name, type, options = {}) ; end
+      def add_index(table_name, column_name, options = {}) ; end
+      def add_index_options(table_name, column_name, options = {}) ; end
+      def add_index_sort_order(option_strings, column_names, options = {}) ; end
+      def add_timestamps(table_name) ; end
+
+      # ActiveRecord::ConnectionAdapters::OracleEnhancedSchemaStatementsExt
+      # http://rubydoc.info/gems/activerecord-oracle_enhanced-adapter/ActiveRecord/ConnectionAdapters/OracleEnhancedSchemaStatementsExt
+      def add_foreign_key(from_table, to_table, options = {})
+        Veracode::log_error "schema.rb: ActiveRecord::ConnectionAdapters::OracleEnhancedSchemaStatementsExt#add_foreign_key called"
+      end
+      def add_primary_key_trigger(table_name, options = {})
+        Veracode::log_error "schema.rb: ActiveRecord::ConnectionAdapters::OracleEnhancedSchemaStatementsExt#add_primary_key_trigger called"
+      end
+      def add_synonym(name, table_name, options = {})
+        Veracode::log_error "schema.rb: ActiveRecord::ConnectionAdapters::OracleEnhancedSchemaStatementsExt#add_synonym called"
+      end
+      # def disable_referential_integrity(&block) ; end
+      # def foreign_key_definition(to_table, options = {}) ; end
+      # def foreign_keys(table_name) ; end
+      # def remove_foreign_key(from_table, options) ; end
+      # def remove_synonym(name) ; end
+      # def supports_foreign_keys? ; end
+      # def synonyms ; end
+
+      # TODO: Return only if real receiver would respond to method
+      # def method_missing(meth, *args, &block)
+      #   if ActiveRecord::ConnectionAdapters::AbstractAdapter.instance_methods.include?(meth)
+      #     log_error "Unhandled method: #{meth} args: #{args.to_s}"
+      #     nil
+      #   else
+      #     super
+      #   end
+      # end
+      # def respond_to_everything(meth)
+      #   log_error "Unhandled respond_to? for: #{meth}" unless really_respond_to?(meth)
+      #   true
+      # end
+      # alias_method :really_respond_to?, :respond_to?
+      # alias_method :respond_to?, :respond_to_everything
+    end
+  end
+
+  def self.archive_schema
+    puts "Evaluating and archiving schema information"
+    schema_file = File.join("db", "schema.rb")
+
+    begin
+      schema = 'Veracode::' + File.read(schema_file).each_line.reject {|l| l =~ /^\s*#/}.join      
+    rescue Exception => e
+      puts "Unable to retrieve schema information from 'db/schema.rb'. Are your migrations up to date?"
+      log_error "Unable to retrieve schema from 'db/schema.rb' (#{e.message})"
+      add_to_archive  %Q|module "Veracode::Schema"\n|
+      add_to_archive  %Q|endmodule\n\n|
+      return
+    end
+
+    add_to_archive  %Q|module "Veracode::Schema"\n|
+    begin
+      eval(schema)
+    rescue Exception => e
+      puts "Unable to evaluate schema information from 'db/schema.rb'. (#{e.message})"
+      log_error "Unable to evaluate 'db/schema.rb' (#{e.message})"
+    end
+    add_to_archive  %Q|endmodule\n\n|
+  end
+
+end

data/lib/veracode/version.rb

@@ -1,3 +1,4 @@
 module Veracode
-  VERSION = "1.0.0.alpha2"
+  VERSION = '1.0.0.alpha7'
+  ARCHIVE_VERSION = '2012-07-04'
 end

metadata

@@ -1,27 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: veracode
 version: !ruby/object:Gem::Version
-  version: 1.0.0.alpha2
-  prerelease: 6
+  version: 1.0.0.alpha7
 platform: ruby
 authors:
 - Veracode
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-06-11 00:00:00.000000000 Z
+date: 2015-12-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rubyzip
-  requirement: &70309839816480 !ruby/object:Gem::Requirement
-    none: false
+  requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ! '>='
+    - - '>='
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '1.0'
   type: :runtime
   prerelease: false
-  version_requirements: *70309839816480
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '1.0'
 description: Prepares your Ruby on Rails app for submission to Veracode.
 email: devcontact@veracode.com
 executables:
@@ -31,30 +33,34 @@ extra_rdoc_files: []
 files:
 - bin/veracode
 - lib/veracode.rb
+- lib/veracode/gems.rb
+- lib/veracode/schema.rb
 - lib/veracode/version.rb
 homepage: http://veracode.com/
 licenses: []
+metadata: {}
 post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
 required_ruby_version: !ruby/object:Gem::Requirement
-  none: false
   requirements:
-  - - ~>
+  - - '>='
     - !ruby/object:Gem::Version
       version: 1.9.3.0
+  - - <
+    - !ruby/object:Gem::Version
+      version: 2.2.0
 required_rubygems_version: !ruby/object:Gem::Requirement
-  none: false
   requirements:
-  - - ! '>'
+  - - '>'
     - !ruby/object:Gem::Version
       version: 1.3.1
 requirements: []
 rubyforge_project: 
-rubygems_version: 1.8.10
+rubygems_version: 2.0.14
 signing_key: 
-specification_version: 3
+specification_version: 4
 summary: Command line tool for preparing your Ruby on Rails app for submission to
   Veracode
 test_files: []