Add nessus_scan_export and nessus_scan_export_status methods

Author: root

lib/nessus/nessus-xmlrpc.rb

@@ -52,7 +52,7 @@ def get_server_properties
       return resp
     end
 
-    def user_add(username,password,permissions,type)
+    def user_add(username, password, permissions, type)
       payload = {
         :username => username, 
         :password => password, 
@@ -62,15 +62,15 @@ def user_add(username,password,permissions,type)
       }
       request = Net::HTTP::Post.new("/users")
       request.set_form_data(payload)
-      request.add_field("X-Cookie",@token)
+      request.add_field("X-Cookie", @token)
       resp = @connection.request(request)
       resp = JSON.parse(resp.body)
       return resp
     end
 
     def user_delete(user_id)
       request = Net::HTTP::Delete.new("/users/#{user_id}")
-      request.add_field("X-Cookie",@token)
+      request.add_field("X-Cookie", @token)
       resp = @connection.request(request)
       return resp.code
     end
@@ -82,77 +82,77 @@ def user_chpasswd(user_id, password)
       }
       request = Net::HTTP::Put.new("/users/#{user_id}/chpasswd")
       request.set_form_data(payload)
-      request.add_field("X-Cookie",@token)
+      request.add_field("X-Cookie", @token)
       resp = @connection.request(request)
       return resp.code
     end
 
     def user_logout
       request = Net::HTTP::Delete.new("/session")
-      request.add_field("X-Cookie",@token)
+      request.add_field("X-Cookie", @token)
       resp = @connection.request(request)
       return resp.code
     end
 
     def list_policies
       request = Net::HTTP::Get.new("/policies")
-      request.add_field("X-Cookie",@token)
+      request.add_field("X-Cookie", @token)
       resp = @connection.request(request)
       resp = JSON.parse(resp.body)
       return resp
     end
 
     def list_users
       request = Net::HTTP::Get.new("/users")
-      request.add_field("X-Cookie",@token)
+      request.add_field("X-Cookie", @token)
       resp = @connection.request(request)
       resp = JSON.parse(resp.body)
       return resp
     end
 
     def list_folders
       request = Net::HTTP::Get.new("/folders")
-      request.add_field("X-Cookie",@token)
+      request.add_field("X-Cookie", @token)
       resp = @connection.request(request)
       resp = JSON.parse(resp.body)
       return resp
     end
 
     def list_scanners
       request = Net::HTTP::Get.new("/scanners")
-      request.add_field("X-Cookie",@token)
+      request.add_field("X-Cookie", @token)
       resp = @connection.request(request)
       resp = JSON.parse(resp.body)
       return resp
     end
 
     def list_families
       request = Net::HTTP::Get.new("/plugins/families")
-      request.add_field("X-Cookie",@token)
+      request.add_field("X-Cookie", @token)
       resp = @connection.request(request)
       resp = JSON.parse(resp.body)
       return resp
     end
 
     def list_plugins(family_id)
       request = Net::HTTP::Get.new("/plugins/families/#{family_id}")
-      request.add_field("X-Cookie",@token)
+      request.add_field("X-Cookie", @token)
       resp = @connection.request(request)
       resp = JSON.parse(resp.body)
       return resp
     end
 
     def plugin_details(plugin_id)
       request = Net::HTTP::Get.new("/plugins/plugin/#{plugin_id}")
-      request.add_field("X-Cookie",@token)
+      request.add_field("X-Cookie", @token)
       resp = @connection.request(request)
       resp = JSON.parse(resp.body)
       return resp
     end
 
     def is_admin
       request = Net::HTTP::Get.new("/session")
-      request.add_field("X-Cookie",@token)
+      request.add_field("X-Cookie", @token)
       resp = @connection.request(request)
       resp = JSON.parse(resp.body)
       if resp["permissions"] == 128
@@ -164,7 +164,7 @@ def is_admin
 
     def server_properties
       request = Net::HTTP::Get.new("/server/properties")
-      request.add_field("X-Cookie",@token)
+      request.add_field("X-Cookie", @token)
       resp = @connection.request(request)
       resp = JSON.parse(resp.body)
       return resp
@@ -182,7 +182,7 @@ def scan_create(uuid, name, description, targets)
       }
       request = Net::HTTP::Post.new("/scans")
       request.body = payload.to_json
-      request.add_field("X-Cookie",@token)
+      request.add_field("X-Cookie", @token)
       request["Content-Type"] = "application/json"
       resp = @connection.request(request)
       resp = JSON.parse(resp.body)
@@ -191,52 +191,85 @@ def scan_create(uuid, name, description, targets)
 
     def scan_launch(scan_id)
       request = Net::HTTP::Post.new("/scans/#{scan_id}/launch")
-      request.add_field("X-Cookie",@token)
+      request.add_field("X-Cookie", @token)
       resp = @connection.request(request)
       resp = JSON.parse(resp.body)
       return resp
     end
 
     def server_status
       request = Net::HTTP::Get.new("/server/status")
-      request.add_field("X-Cookie",@token)
+      request.add_field("X-Cookie", @token)
       resp = @connection.request(request)
       resp = JSON.parse(resp.body)
       return resp
     end
 
     def scan_list
       request = Net::HTTP::Get.new("/scans")
-      request.add_field("X-Cookie",@token)
+      request.add_field("X-Cookie", @token)
+      resp = @connection.request(request)
+      resp = JSON.parse(resp.body)
+      return resp
+    end
+
+    def scan_details(scan_id)
+      request = Net::HTTP::Get.new("/scans/#{scan_id}")
+      request.add_field("X-Cookie", @token)
       resp = @connection.request(request)
       resp = JSON.parse(resp.body)
       return resp
     end
 
     def scan_pause(scan_id)
       request = Net::HTTP::Post.new("/scans/#{scan_id}/pause")
-      request.add_field("X-Cookie",@token)
+      request.add_field("X-Cookie", @token)
       resp = @connection.request(request)
       resp = JSON.parse(resp.body)
       return resp
     end
 
     def scan_resume(scan_id)
       request = Net::HTTP::Post.new("/scans/#{scan_id}/resume")
-      request.add_field("X-Cookie",@token)
+      request.add_field("X-Cookie", @token)
       resp = @connection.request(request)
       resp = JSON.parse(resp.body)
       return resp
     end
 
     def scan_stop(scan_id)
       request = Net::HTTP::Post.new("/scans/#{scan_id}/stop")
-      request.add_field("X-Cookie",@token)
+      request.add_field("X-Cookie", @token)
       resp = @connection.request(request)
       resp = JSON.parse(resp.body)
       return resp
     end
 
+    def scan_export(scan_id, format)
+      payload = {
+        :format => format
+      }
+      request = Net::HTTP::Post.new("/scans/#{scan_id}/export")
+      request.body = payload.to_json
+      request["Content-Type"] = "application/json"
+      request.add_field("X-Cookie", @token)
+      resp = @connection.request(request)
+      resp = JSON.parse(resp.body)
+      return resp
+    end
+
+    def scan_export_status(scan_id, file_id)
+      request = Net::HTTP::Get.new("/scans/#{scan_id}/export/#{file_id}/status")
+      request.add_field("X-Cookie", @token)
+      resp = @connection.request(request)
+      if resp.code == "200"
+        return "ready"
+      else
+        resp = JSON.parse(resp.body)
+        return resp
+      end
+    end
+
     def policy_delete(policy_id)
       request = Net::HTTP::Delete.new("/policies/#{policy_id}")
       request.add_field("X-Cookie",@token)