Aaron Soto
54 lines
1.4 KiB
Ruby
54 lines
1.4 KiB
Ruby
module Msf::DBManager::Payload
|
|
|
|
def create_payload(opts)
|
|
if opts[:uuid] && !opts[:uuid].to_s.empty?
|
|
if Mdm::Payload.find_by(uuid: opts[:uuid])
|
|
raise ArgumentError.new("A payload with this uuid already exists.")
|
|
end
|
|
end
|
|
|
|
Mdm::Payload.create!(opts)
|
|
end
|
|
|
|
def payloads(opts)
|
|
::ActiveRecord::Base.connection_pool.with_connection do
|
|
if opts[:id] && !opts[:id].to_s.empty?
|
|
return Array.wrap(Mdm::Payload.find(opts[:id]))
|
|
end
|
|
|
|
wspace = Msf::Util::DBManager.process_opts_workspace(opts, framework)
|
|
return wspace.payloads.where(opts)
|
|
end
|
|
end
|
|
|
|
def update_payload(opts)
|
|
::ActiveRecord::Base.connection_pool.with_connection do
|
|
wspace = Msf::Util::DBManager.process_opts_workspace(opts, framework, false)
|
|
opts[:workspace] = wspace if wspace
|
|
|
|
id = opts.delete(:id)
|
|
Mdm::Payload.update(id, opts)
|
|
end
|
|
end
|
|
|
|
def delete_payload(opts)
|
|
raise ArgumentError.new("The following options are required: :ids") if opts[:ids].nil?
|
|
|
|
::ActiveRecord::Base.connection_pool.with_connection do
|
|
deleted = []
|
|
opts[:ids].each do |payload_id|
|
|
payload = Mdm::Payload.find(payload_id)
|
|
begin
|
|
deleted << payload.destroy
|
|
rescue
|
|
elog("Forcibly deleting #{payload}")
|
|
deleted << payload.delete
|
|
end
|
|
end
|
|
|
|
return deleted
|
|
end
|
|
end
|
|
|
|
end
|