Enable ipv6 gateway options

nunomcruz · Jan 27, 2017 · 1fe242c · 1fe242c
1 parent a786350
commit 1fe242c
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 5 deletions.
diff --git a/lib/bettercap/context.rb b/lib/bettercap/context.rb
@@ -80,11 +80,20 @@ def initialize
 
   # Update the Context state parsing network related informations.
   def update!
-    gw = @options.core.gateway || Network.get_gateway
-    raise BetterCap::Error, "Could not detect the gateway address for interface #{@options.core.iface}, "\
-                            'make sure you\'ve specified the correct network interface to use and to have the '\
-                            'correct network configuration, this could also happen if bettercap '\
-                            'is launched from a virtual environment.' unless Network::Validator.is_ip?(gw)
+    if @options.core.use_ipv6
+      gw = @options.core.gateway || Network.get_ipv6_gateway
+      raise BetterCap::Error, "Could not detect the gateway address for interface #{@options.core.iface}, "\
+                              'make sure you\'ve specified the correct network interface to use and to have the '\
+                              'correct network configuration, this could also happen if bettercap '\
+                              'is launched from a virtual environment.' unless Network::Validator.is_ipv6?(gw)
+
+    else
+      gw = @options.core.gateway || Network.get_gateway
+      raise BetterCap::Error, "Could not detect the gateway address for interface #{@options.core.iface}, "\
+                              'make sure you\'ve specified the correct network interface to use and to have the '\
+                              'correct network configuration, this could also happen if bettercap '\
+                              'is launched from a virtual environment.' unless Network::Validator.is_ip?(gw)
+    end
 
     unless @options.core.use_mac.nil?
       cfg = PacketFu::Utils.ifconfig @options.core.iface

diff --git a/lib/bettercap/options/core_options.rb b/lib/bettercap/options/core_options.rb
@@ -84,6 +84,11 @@ def parse!( ctx, opts )
       raise BetterCap::Error, "The specified gateway '#{v}' is not a valid IPv4 address." unless Network::Validator.is_ip?(v)
     end
 
+    opts.on( '--gateway6 ADDRESS', 'Manually specify the IPv6 gateway address, if not specified the current gateway will be retrieved and used. ' ) do |v|
+      @gateway = v
+      raise BetterCap::Error, "The specified gateway '#{v}' is not a valid IPv6 address." unless Network::Validator.is_ipv6?(v)
+    end
+
     opts.on( '-T', '--target ADDRESS1,ADDRESS2', 'Target IP addresses, if not specified the whole subnet will be targeted.' ) do |v|
       self.targets = v
     end