Method: WASP::Wasp#_attack

Defined in:

lib/wasp/wasp.rb

#_attack(wasps, num, conn, urls, time = nil) ⇒ Object

# File 'lib/wasp/wasp.rb', line 415

def _attack (wasps, num, conn, urls, time=nil)
  coop = conn / wasps.count
  num = num / wasps.count if not num.nil?
  
  print "Get communication channel for wasps..\n"
  threads = []
  wasps.each do |b|
    threads << Thread.new(b) { |wasp|
      begin
        wasp[:ssh] = Net::SSH.start(wasp[:instance_name], wasp[:login], 
                     :keys => ENV['HOME'] + '/.ssh/' + wasp[:key_name] + '.pem')
        print "Wasp " + "#{wasp[:instance_id]}".yellow + " is joining the wasp.\n"                       
      rescue => ex
        puts "[Error] ".red + "#{ex.message}"
        puts "Cannot reach to " + "#{wasp[:instance_id]}".yellow
      end
    }
  end      
  threads.each { |aThread| aThread.join }
  
  threads = []
  i = 0
  wasps.each do |b|
    url = urls[i%urls.count]
    i += 1
    threads << Thread.new(b) { |wasp|
      begin
      weapon = get_weapon
      if weapon.check(wasp[:ssh]) then 
        print "Wasp " + "#{wasp[:instance_id]}".yellow + " is firing its stings. Ping ping!\n"
        
          wasp[:rawreport] = weapon.fire(num, coop, url, wasp[:instance_id], @report, @keepalive, @without_cookie, time, @header) 
          
          print "Wasp " + "#{wasp[:instance_id]}".yellow + " is out of ammo.\n"
        else
          print "Wasp " + "#{wasp[:instance_id]}".yellow + " has no weapon.\n"
        end
      rescue => ex
        print "[WARN]".yellow + " #{ex.message}.\n"
      end
    }
  end
  
  threads.each { |aThread| aThread.join }
  
  wasps.each do |wasp|
    begin
      wasp[:ssh].close
    rescue => ex
      puts "[WARN]".yellow + "#{ex.message}"
    end
  end
  
  wasps
end