Module: Msf::WebServices::PayloadServlet

Defined in:
lib/msf/core/web_services/servlet/payload_servlet.rb

Class Method Summary collapse

Class Method Details

.api_path ⇒ Object 



3
4
5
 
# File 'lib/msf/core/web_services/servlet/payload_servlet.rb', line 3

def self.api_path
  '/api/v1/payloads'
end

.api_path_with_id ⇒ Object 



7
8
9
 
# File 'lib/msf/core/web_services/servlet/payload_servlet.rb', line 7

def self.api_path_with_id
  "#{self.api_path}/?:id?"
end

.create_payload ⇒ Object 



23
24
25
26
27
28
29
30
31
32
33
34
 
# File 'lib/msf/core/web_services/servlet/payload_servlet.rb', line 23

def self.create_payload
  lambda {
    warden.authenticate!
    begin
      opts = parse_json_request(request, false)
      data = get_db.create_payload(opts)
      set_json_data_response(response: data)
    rescue => e
      print_error_and_create_response(error: e, message: 'There was an error creating the payload:', code: 500)
    end
  }
end

.delete_payload ⇒ Object 



65
66
67
68
69
70
71
72
73
74
75
76
 
# File 'lib/msf/core/web_services/servlet/payload_servlet.rb', line 65

def self.delete_payload
  lambda {
    warden.authenticate!
    begin
      opts = parse_json_request(request, false)
      data = get_db.delete_payload(opts)
      set_json_data_response(response: data)
    rescue => e
      print_error_and_create_response(error: e, message: 'There was an error deleting the payload:', code: 500)
    end
  }
end

.get_payload ⇒ Object 



36
37
38
39
40
41
42
43
44
45
46
47
48
 
# File 'lib/msf/core/web_services/servlet/payload_servlet.rb', line 36

def self.get_payload
  lambda {
    warden.authenticate!
    begin
      sanitized_params = sanitize_params(params, env['rack.request.query_hash'])
      data = get_db.payloads(sanitized_params)
      data = data.first if is_single_object?(data, sanitized_params)
      set_json_data_response(response: data)
    rescue => e
      print_error_and_create_response(error: e, message: 'There was an error getting the payload:', code: 500)
    end
  }
end

.registered(app) ⇒ Object 



11
12
13
14
15
16
17
 
# File 'lib/msf/core/web_services/servlet/payload_servlet.rb', line 11

def self.registered(app)
  app.get self.api_path, &get_payload
  app.get self.api_path_with_id, &get_payload
  app.post self.api_path, &create_payload
  app.put self.api_path_with_id, &update_payload
  app.delete self.api_path, &delete_payload
end

.update_payload ⇒ Object 



50
51
52
53
54
55
56
57
58
59
60
61
62
63
 
# File 'lib/msf/core/web_services/servlet/payload_servlet.rb', line 50

def self.update_payload
  lambda {
    warden.authenticate!
    begin
      opts = parse_json_request(request, false)
      tmp_params = sanitize_params(params)
      opts[:id] = tmp_params[:id] if tmp_params[:id]
      data = get_db.update_payload(opts)
      set_json_data_response(response: data)
    rescue => e
      print_error_and_create_response(error: e, message: 'There was an error updating the payload:', code: 500)
    end
  }
end