scifijunkie/pwnagotchi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Tweak Wigle plugin

Browse Source 
Adds a new 'donate' option to allow Wigle to include data for commercial use.

Also changes data header to reflect this is a pwnagotchi plugin.

Signed-off-by: HeroCC <herocc@herocc.com>
This commit is contained in:
HeroCC 2020-10-03 17:01:42 +02:00
parent 713a14c504
commit 3bd9cd4f18
2 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pwnagotchi/defaults.toml
View File

@ -52,6 +52,7 @@ main.plugins.wpa-sec.whitelist = []
main.plugins.wigle.enabled = false main.plugins.wigle.enabled = false
main.plugins.wigle.api_key = "" main.plugins.wigle.api_key = ""
main.plugins.wigle.whitelist = [] main.plugins.wigle.whitelist = []
main.plugins.wigle.donate = true
main.plugins.bt-tether.enabled = false main.plugins.bt-tether.enabled = false

16
pwnagotchi/plugins/default/wigle.py
View File

@ -9,6 +9,7 @@ from datetime import datetime
from pwnagotchi.utils import WifiInfo, FieldNotFoundError, extract_from_pcap, StatusFile, remove_whitelisted from pwnagotchi.utils import WifiInfo, FieldNotFoundError, extract_from_pcap, StatusFile, remove_whitelisted
from threading import Lock from threading import Lock
from pwnagotchi import plugins from pwnagotchi import plugins
from pwnagotchi._version import __version__ as __pwnagotchi_version__
def _extract_gps_data(path): def _extract_gps_data(path):
@ -34,14 +35,14 @@ def _format_auth(data):
return out return out
def _transform_wigle_entry(gps_data, pcap_data): def _transform_wigle_entry(gps_data, pcap_data, plugin_version):
""" """
Transform to wigle entry in file Transform to wigle entry in file
""" """
dummy = StringIO() dummy = StringIO()
# write kismet header # write kismet header
dummy.write( dummy.write(
"WigleWifi-1.4,appRelease=20190201,model=Kismet,release=2019.02.01.{},device=kismet,display=kismet,board=kismet,brand=kismet\n") "WigleWifi-1.4,appRelease={},model=pwnagotchi,release={},device=pwnagotchi,display=kismet,board=kismet,brand=pwnagotchi\n".format(plugin_version, __pwnagotchi_version__))
dummy.write( dummy.write(
"MAC,SSID,AuthMode,FirstSeen,Channel,RSSI,CurrentLatitude,CurrentLongitude,AltitudeMeters,AccuracyMeters,Type") "MAC,SSID,AuthMode,FirstSeen,Channel,RSSI,CurrentLatitude,CurrentLongitude,AltitudeMeters,AccuracyMeters,Type")
@ -62,7 +63,7 @@ def _transform_wigle_entry(gps_data, pcap_data):
return dummy.getvalue() return dummy.getvalue()
def _send_to_wigle(lines, api_key, timeout=30): def _send_to_wigle(lines, api_key, donate=True, timeout=30):
""" """
Uploads the file to wigle-net Uploads the file to wigle-net
""" """
@ -76,7 +77,7 @@ def _send_to_wigle(lines, api_key, timeout=30):
headers = {'Authorization': f"Basic {api_key}", headers = {'Authorization': f"Basic {api_key}",
'Accept': 'application/json'} 'Accept': 'application/json'}
data = {'donate': 'false'} data = {'donate': 'on' if donate else 'false'}
payload = {'file': dummy, 'type': 'text/csv'} payload = {'file': dummy, 'type': 'text/csv'}
try: try:
@ -112,6 +113,9 @@ class Wigle(plugins.Plugin):
if not 'whitelist' in self.options: if not 'whitelist' in self.options:
self.options['whitelist'] = list() self.options['whitelist'] = list()
if not 'donate' in self.options:
self.options['donate'] = True
self.ready = True self.ready = True
def on_internet_available(self, agent): def on_internet_available(self, agent):
@ -172,14 +176,14 @@ class Wigle(plugins.Plugin):
logging.debug("WIGLE: %s", sc_e) logging.debug("WIGLE: %s", sc_e)
self.skip.append(gps_file) self.skip.append(gps_file)
continue continue
new_entry = _transform_wigle_entry(gps_data, pcap_data) new_entry = _transform_wigle_entry(gps_data, pcap_data, self.__version__)
csv_entries.append(new_entry) csv_entries.append(new_entry)
no_err_entries.append(gps_file) no_err_entries.append(gps_file)
if csv_entries: if csv_entries:
display.set('status', "Uploading gps-data to wigle.net ...") display.set('status', "Uploading gps-data to wigle.net ...")
display.update(force=True) display.update(force=True)
try: try:
_send_to_wigle(csv_entries, self.options['api_key']) _send_to_wigle(csv_entries, self.options['api_key'], donate=self.options['donate'])
reported += no_err_entries reported += no_err_entries
self.report.update(data={'reported': reported}) self.report.update(data={'reported': reported})
logging.info("WIGLE: Successfully uploaded %d files", len(no_err_entries)) logging.info("WIGLE: Successfully uploaded %d files", len(no_err_entries))