ProxyServer::Connection: Use Arachni::HTTP::Client.headers

Arachni · Jul 6, 2017 · 5676fa4 · 5676fa4
1 parent 016b159
commit 5676fa4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/lib/arachni/http/proxy_server/connection.rb b/lib/arachni/http/proxy_server/connection.rb
@@ -84,7 +84,7 @@ def initialize( options = {} )
                                 url:     sanitize_url( @parser.request_url, headers ),
                                 method:  method,
                                 body:    @body,
-                                headers: headers
+                                headers: Arachni::HTTP::Client.headers.to_h.merge( headers )
                             )
                         )
 
@@ -296,7 +296,7 @@ def cleanup_request_headers( headers )
             headers.delete name
         end
 
-        headers
+        headers.to_h
     end
 
     def cleanup_response_headers( headers )

diff --git a/spec/arachni/http/proxy_server_spec.rb b/spec/arachni/http/proxy_server_spec.rb
@@ -208,6 +208,7 @@ def test_proxy( proxy )
                     "Accept: */*\r\n" <<
                     "Content-Length: 7\r\n" <<
                     "Accept-Language: en-US,en;q=0.8,he;q=0.6\r\n" <<
+                    "#{Arachni::HTTP::Client::SEED_HEADER_NAME}: #{Arachni::Utilities.random_seed}\r\n" <<
                     "Content-Type: application/x-www-form-urlencoded\r\n\r\n").split( "\r\n" ).sort
                 )