Add agent option ndp

nunomcruz · Feb 1, 2017 · 7e4430f · 7e4430f
1 parent 56a9d7f
commit 7e4430f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/bettercap/network/network.rb b/lib/bettercap/network/network.rb
@@ -139,8 +139,12 @@ def ip2name( address )
   # complete their job.
   # If +address+ is not nil only that ip will be probed.
   def start_agents( ctx, address = nil )
-    [ 'Icmp', 'Udp', 'Arp' ].each do |name|
-      BetterCap::Loader.load("BetterCap::Discovery::Agents::#{name}").new(ctx, address)
+    if ctx.options.core.use_ipv6
+      BetterCap::Loader.load("BetterCap::Discovery::Agents::Ndp").new(ctx, address)
+    else
+      [ 'Icmp', 'Udp', 'Arp' ].each do |name|
+        BetterCap::Loader.load("BetterCap::Discovery::Agents::#{name}").new(ctx, address)
+      end
     end
     ctx.packets.wait_empty( ctx.timeout )
   end