Revision 4d9f2962
Added by Marc Dequènes over 13 years ago
- ID 4d9f2962db882142ce63530d9b8844890bb557be
| lib/cyborghood/cyborg/dsl.rb | ||
|---|---|---|
|
|
||
|
|
||
|
module CyborgHood
|
||
|
module BotnetDSL
|
||
|
module DSL
|
||
|
class BaseDSL < ActiveSupport::BasicObject
|
||
|
def initialize(&block)
|
||
|
self.instance_eval(&block)
|
||
|
_start_dsl
|
||
|
end
|
||
|
|
||
|
reveal logger
|
||
|
reveal :class
|
||
|
reveal :logger
|
||
|
end
|
||
|
|
||
|
class Task < BaseDSL
|
||
|
attr_reader :bot, :errors, :results
|
||
|
attr_reader :bot, :name, :errors, :results
|
||
|
|
||
|
@@task_list = {}
|
||
|
@@task_wip = 0
|
||
|
|
||
|
def self.idle?
|
||
|
@@task_wip == 0
|
||
|
end
|
||
|
|
||
|
def initialize(bot, name = nil, previous_step_data = {}, &block)
|
||
|
@bot = bot
|
||
|
@name = name
|
||
|
|
||
|
@@task_list[@name] = self unless @name.nil?
|
||
|
@@task_wip += 1
|
||
|
|
||
|
@errors = (previous_step_data[:errors] || {}).freeze
|
||
|
@results = (previous_step_data[:results] || {}).freeze
|
||
| ... | ... | |
|
begin
|
||
|
super(&block)
|
||
|
rescue
|
||
|
logger.error "Task '#{@name}': could not initialize DSL: " + $!.to_s
|
||
|
logger.error "Task '#{@name}': error in DSL: " + $!.to_s
|
||
|
logger.debug $!.backtrace.join("\n")
|
||
|
@cancel_run = true
|
||
|
end
|
||
| ... | ... | |
|
|
||
|
def task(name = nil, &block)
|
||
|
_add_subtask("task/#{name}") do |subtask|
|
||
|
self.class.new(name || @name, &block)
|
||
|
self.class.new(@bot, name || @name, &block)
|
||
|
subtask.finish
|
||
|
end
|
||
|
end
|
||
| ... | ... | |
|
attr_reader :name
|
||
|
attr_accessor :results, :errors
|
||
|
|
||
|
def initialize(name, &block)
|
||
|
def initialize(task, name, &block)
|
||
|
@task = task
|
||
|
@name = name
|
||
|
@block = block
|
||
|
|
||
| ... | ... | |
|
def do_it
|
||
|
@block.call self
|
||
|
rescue
|
||
|
msg = "Task '#{@name}': error: " + $!.to_s
|
||
|
msg = "Task '#{@task.name}': error: " + $!.to_s
|
||
|
logger.error msg
|
||
|
@errors << CyberError.new(:unrecoverable, "botnet/client/dsl", msg)
|
||
|
logger.debug $!.backtrace.join("\n")
|
||
|
finish
|
||
|
end
|
||
|
|
||
|
def finish
|
||
|
if @finished
|
||
|
raise CyberError.new(:unrecoverable, "botnet/client/dsl", "Task '#{@name}': subtask should have ended, but it lied")
|
||
|
raise CyberError.new(:unrecoverable, "botnet/client/dsl", "Task '#{@task.name}': subtask should have ended, but it lied")
|
||
|
end
|
||
|
@finished = true
|
||
|
logger.debug "Task '#{@task.name}': subtask '#{@name}' finished"
|
||
|
@task.__send__(:_check_finished)
|
||
|
end
|
||
|
|
||
|
def finished?
|
||
| ... | ... | |
|
end
|
||
|
|
||
|
def _add_subtask(name, cb = nil, &block)
|
||
|
@subtasks << Subtask.new(name, &block)
|
||
|
@subtasks << Subtask.new(self, name, &block)
|
||
|
end
|
||
|
|
||
|
def _subtasks_finished?
|
||
| ... | ... | |
|
return unless _subtasks_finished?
|
||
|
|
||
|
next_step_data = {
|
||
|
:errors = {},
|
||
|
:results = {}
|
||
|
:errors => {},
|
||
|
:results => {}
|
||
|
}
|
||
|
|
||
|
@subtasks.each do |subtask|
|
||
|
next_step_data[:errors][subtask.name] = subtask.errors
|
||
|
next_step_data[:errors][subtask.name] = subtask.errors unless subtask.errors.empty?
|
||
|
next_step_data[:results].merge!(subtask.results)
|
||
|
end
|
||
|
|
||
|
cb = next_step_data[:errors].empty? ? @error_cb : @success_cb
|
||
|
self.class.new(name || @name, next_step_data, &cb) if cb
|
||
|
if cb
|
||
|
self.class.new(@bot, @name, next_step_data, &cb)
|
||
|
else
|
||
|
logger.debug "Task '#{@name}': finished"
|
||
|
end
|
||
|
|
||
|
@@task_wip -= 1
|
||
|
end
|
||
|
|
||
|
def _notification_criterias_match(msg, criterias)
|
||
| ... | ... | |
|
def _start_dsl
|
||
|
return false if @cancel_run
|
||
|
|
||
|
@subtasks.each do |subtask|
|
||
|
# skip broken or canceled subtasks
|
||
|
next if subtask.finished?
|
||
|
subtask.do_it
|
||
|
logger.debug "Task '#{@name}': begining step"
|
||
|
|
||
|
if @subtasks.empty?
|
||
|
_check_finished
|
||
|
else
|
||
|
@subtasks.each do |subtask|
|
||
|
# skip broken or canceled subtasks
|
||
|
next if subtask.finished?
|
||
|
subtask.do_it
|
||
|
end
|
||
|
end
|
||
|
|
||
|
true
|
||
Also available in: Unified diff
[evol] Task-based client DSL §4 (many fixes, helper, bot stop condition, more logging)