Merge pull request #10 from clee-r7/add-http-put-request

Add HTTP PUT request method

Author: jbarnett-r7

lib/metasploit/framework/data_service/remote/http/core.rb

@@ -19,6 +19,7 @@ class RemoteHTTPDataService
   GET_REQUEST = 'GET'
   POST_REQUEST = 'POST'
   DELETE_REQUEST = 'DELETE'
+  PUT_REQUEST = 'PUT'
 
   #
   # @param [String] endpoint A valid http or https URL. Cannot be nil
@@ -82,6 +83,19 @@ def delete_data(path, data_hash, query = nil)
     make_request(DELETE_REQUEST, path, data_hash, query)
   end
 
+  #
+  # Send PUT request to store data for the specified resource at the HTTP endpoint
+  #
+  # @param path - The URI path to send the request
+  # @param data_hash - A hash representation of the object to be stored. Cannot be nil or empty.
+  # @param query - A hash representation of the URI query data. Key-value pairs will be URL-encoded.
+  #
+  # @return A wrapped response (ResponseWrapper), see below.
+  #
+  def put_data(path, data_hash, query = nil)
+    make_request(PUT_REQUEST, path, data_hash, query)
+  end
+
   #
   # Make the specified request_type
   #
@@ -106,6 +120,8 @@ def make_request(request_type, path, data_hash = nil, query = nil)
           request = Net::HTTP::Post.new(uri.request_uri)
         when DELETE_REQUEST
           request = Net::HTTP::Delete.new(uri.request_uri)
+        when PUT_REQUEST
+          request = Net::HTTP::Put.new(uri.request_uri)
         else
           raise Exception, 'A request_type must be specified'
       end