fix: google geolocation recon script

rapid7 · Nov 2, 2018 · ea0ba6b · ea0ba6b
1 parent 57dbe95
commit ea0ba6b
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/tools/recon/google_geolocate_bssid.rb b/tools/recon/google_geolocate_bssid.rb
@@ -18,18 +18,19 @@
 require 'rex/google/geolocation'
 require 'optparse'
 
-if ARGV.empty?
-  $stderr.puts("Usage: #{$PROGRAM_NAME} <mac> [mac] ...")
+if ARGV.length < 2
+  $stderr.puts("Usage: #{$PROGRAM_NAME} <api_key> <mac> [mac] ...")
   $stderr.puts("Ask Google for the location of the given set of BSSIDs")
   $stderr.puts
-  $stderr.puts("Example: iwlist sc 2>/dev/null|awk '/Address/{print $5}'|xargs #{$PROGRAM_NAME}")
-  $stderr.puts("Example: /System/Library/PrivateFrameworks/Apple80211.framework/Versions/A/Resources/airport -I|awk '/BSSID/{print $2}'|xargs #{$PROGRAM_NAME}")
+  $stderr.puts("Example: iwlist sc 2>/dev/null|awk '/Address/{print $5}'|xargs #{$PROGRAM_NAME} <api_key>")
+  $stderr.puts("Example: /System/Library/PrivateFrameworks/Apple80211.framework/Versions/A/Resources/airport -I|awk '/BSSID/{print $2}'|xargs #{$PROGRAM_NAME} <api_key>")
   exit(1)
 end
 
 g = Rex::Google::Geolocation.new
-ARGV.each do |mac|
-  g.add_wlan(mac, nil, -83)
+g.set_api_key(ARGV[0])
+ARGV.drop(1).each do |mac|
+  g.add_wlan(mac, -83)
 end
 
 g.fetch!