metasploit-framework/lib/sqlmap/sqlmap_manager.rb

require 'json'

module Sqlmap
  class Manager
    def initialize(session)
      @session = session
    end

    def new_task
      res = @session.get('/task/new')
      return JSON.parse(res.body)
    end

    def delete_task(task_id)
      res = @session.get('/task/' + task_id + '/delete')
      return JSON.parse(res.body)
    end

    def set_option(task_id, key, value)
      post = { key => value }
      res = @session.post('/option/' + task_id + '/set', nil, post.to_json, {'ctype' => 'application/json'})
      return JSON.parse(res.body)
    end

    def get_options(task_id)
      res = @session.get('/option/' + task_id + '/list')
      return JSON.parse(res.body)
    end

    def start_task(task_id, options = {})
      res = @session.post('/scan/' + task_id + '/start' , nil, options.to_json, {'ctype' => 'application/json'})
      return JSON.parse(res.body)
    end

    def get_task_status(task_id)
      res = @session.get('/scan/' + task_id + '/status')
      return JSON.parse(res.body)
    end

    def get_task_log(task_id)
      res = @session.get('/scan/' + task_id + '/log')
      return JSON.parse(res.body)
    end

    def get_task_data(task_id)
      res = @session.get('/scan/' + task_id + '/data')
      return JSON.parse(res.body)
    end
  end
end