plugins/vector_feed.rb in arachni-0.4.2 vs plugins/vector_feed.rb in arachni-0.4.3

@@ -66,11 +66,11 @@
 
         pages = {}
         page_buffer = []
         print_status "Imported #{feed.size} vectors."
         feed.each do |obj|
-            vector = obj.respond_to?( :value ) ? obj.value : obj
+            vector = (obj.respond_to?( :value ) ? obj.value : obj).symbolize_keys( false )
 
             begin
                 exception_jail{
 
                     if page?( vector )
@@ -102,28 +102,28 @@
             print_bad 'Could not find any usable vectors.'
         end
     end
 
     def page?( vector )
-        vector['type'] == 'page'
+        vector[:type] == 'page'
     end
 
     def page_from_body_vector( vector )
         Page.new(
-            code:             Integer( vector['code'] || 200 ),
-            url:              vector['url'] || framework.opts.url.to_s,
-            body:             vector['body'] || '',
-            response_headers: vector['headers'] || {}
+            code:             Integer( vector[:code] || 200 ),
+            url:              vector[:url]      || framework.opts.url.to_s,
+            body:             vector[:body]     || '',
+            response_headers: vector[:headers]  || {}
         )
     end
 
     def hash_to_element( vector )
         owner  = framework.opts.url.to_s
-        action = vector['action']
-        inputs = vector['inputs']
-        method = vector['method'] || 'get'
-        type   = vector['type'] || 'link'
+        action = vector[:action]
+        inputs = vector[:inputs]
+        method = vector[:method] || 'get'
+        type   = vector[:type]   || 'link'
 
         return if !inputs || inputs.empty?
 
         e = case type
             when Element::LINK
@@ -145,11 +145,11 @@
                 Link.new( owner,
                     action: action,
                     inputs: inputs
                 )
         end
-        (vector['skip'] || []).each { |i| e.immutables << i }
+        (vector[:skip] || []).each { |i| e.immutables << i }
         e
     end
 
     def clean_up
         framework.resume
@@ -212,10 +212,10 @@
   inputs:
     User-Agent: "Blah/2"
 
             },
             author:      'Tasos "Zapotek" Laskos <[e-mail]>',
-            version:     '0.1.2',
+            version:     '0.1.3',
             options:     [
                 Options::Base.new( 'vectors', [false, ' Vector array (for configuration over RPC).'] ),
                 Options::String.new( 'yaml_string', [false, 'A string of YAML serialized vectors (for configuration over RPC).'] ),
                 Options::Path.new( 'yaml_file', [false, 'A file containing the YAML serialized vectors.'] )
             ]