chrissewell/pwnagotchi
1
0
Fork 0
mirror of https://github.com/jayofelony/pwnagotchi.git synced 2025-07-01 18:37:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

Update wifi channels

Browse Source
This commit is contained in:
Jeroen Oudshoorn
2024-01-05 21:28:28 +01:00
parent 017ffd3792
commit 28e2ce9ff7
4 changed files with 36 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
pwnagotchi/ai/featurizer.py
View File

@ -5,14 +5,9 @@ import pwnagotchi.mesh.wifi as wifi
MAX_EPOCH_DURATION = 1024
def describe(extended=False):
if not extended:
histogram_size = wifi.NumChannels
else:
# see https://github.com/evilsocket/pwnagotchi/issues/583
histogram_size = wifi.NumChannelsExt
histogram_size = wifi.NumChannels
return histogram_size, (1,
shape = (1,
# aps per channel
histogram_size +
# clients per channel

11
pwnagotchi/ai/gym.py
View File

@ -35,14 +35,11 @@ class Environment(gym.Env):
self._epoch_num = 0
self._last_render = None
# see https://github.com/evilsocket/pwnagotchi/issues/583
self._supported_channels = agent.supported_channels()
self._extended_spectrum = any(ch > 150 for ch in self._supported_channels)
self._histogram_size, self._observation_shape = featurizer.describe(self._extended_spectrum)
channels = agent.supported_channels()
Environment.params += [
Parameter('_channel_%d' % ch, min_value=0, max_value=1, meta=ch + 1) for ch in
range(self._histogram_size) if ch + 1 in self._supported_channels
range(featurizer.histogram_size) if ch + 1 in channels
]
self.last = {
@ -55,7 +52,7 @@ class Environment(gym.Env):
}
self.action_space = spaces.MultiDiscrete([p.space_size() for p in Environment.params if p.trainable])
self.observation_space = spaces.Box(low=0, high=1, shape=self._observation_shape, dtype=np.float32)
self.observation_space = spaces.Box(low=0, high=1, shape=featurizer.shape, dtype=np.float32)
self.reward_range = reward.range
@staticmethod
@ -123,7 +120,7 @@ class Environment(gym.Env):
return self.last['state_v']
def _render_histogram(self, hist):
for ch in range(self._histogram_size):
for ch in range(featurizer.histogram_size):
if hist[ch]:
logging.info(" CH %d: %s" % (ch + 1, hist[ch]))

3
pwnagotchi/mesh/wifi.py
View File

@ -1,5 +1,4 @@
NumChannels = 140
NumChannelsExt = 165 # see https://github.com/evilsocket/pwnagotchi/issues/583
NumChannels = 177
def freq_to_channel(freq):

10
pwnagotchi/utils.py
View File

@ -130,10 +130,10 @@ def merge_config(user, default):
def keys_to_str(data):
if isinstance(data,list):
if isinstance(data, list):
converted_list = list()
for item in data:
if isinstance(item,list) or isinstance(item,dict):
if isinstance(item, list) or isinstance(item, dict):
converted_list.append(keys_to_str(item))
else:
converted_list.append(item)
@ -277,7 +277,8 @@ def load_config(args):
elif config['ui']['display']['type'] in ('ws_154inch', 'ws154inch', 'waveshare_154inch', 'waveshare154inch'):
config['ui']['display']['type'] = 'waveshare154inch'
elif config['ui']['display']['type'] in ('waveshare144lcd', 'ws_144inch', 'ws144inch', 'waveshare_144inch', 'waveshare144inch'):
elif config['ui']['display']['type'] in (
'waveshare144lcd', 'ws_144inch', 'ws144inch', 'waveshare_144inch', 'waveshare144inch'):
config['ui']['display']['type'] = 'waveshare144lcd'
elif config['ui']['display']['type'] in ('ws_213d', 'ws213d', 'waveshare_213d', 'waveshare213d'):
@ -318,7 +319,8 @@ def total_unique_handshakes(path):
def iface_channels(ifname):
channels = []
output = subprocess.getoutput("/sbin/iwlist %s freq" % ifname)
phy = subprocess.getoutput("/sbin/iw %s info | grep wiphy | cut -d ' ' -f 2" % ifname)
output = subprocess.getoutput("/sbin/iw phy%s channels | grep ' MHz' | sed 's/^.*\[//g' | sed s/\].*\$//g" % phy)
for line in output.split("\n"):
line = line.strip()
if line.startswith("Channel "):