Brandon Perry
51 lines
1.3 KiB
Ruby
51 lines
1.3 KiB
Ruby
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
|