tests: PASN: Try get_ptksa() from AP a few times

We wait for the PASN auth to complete on the wpas side, but there's no
indication of this on the AP side. So if scheduling ordering is bad, we
can ask the AP for the PTKSA cache before it even received the frame
from the kernel and created the PTKSA entry.

To fix this, try this a few times, to see if it becomes available.

Signed-off-by: Johannes Berg <johannes.berg@intel.com>
This commit is contained in:
Johannes Berg 2023-09-25 09:20:44 +02:00 committed by Jouni Malinen
parent 376bead963
commit 4f4923bb08

View file

@ -52,10 +52,25 @@ def check_pasn_ptk(dev, hapd, cipher, fail_ptk=False, clear_keys=True,
sta_ptksa = dev.get_ptksa(hapd.own_addr(), cipher)
ap_ptksa = hapd.get_ptksa(dev.own_addr(), cipher)
# There's no event indicating hostapd has finished
# processing - so just try once more if it isn't
# yet available, it might become available still.
for i in range(10):
if ap_ptksa:
break
time.sleep(0.1)
ap_ptksa = hapd.get_ptksa(dev.own_addr(), cipher)
if not (sta_ptksa and ap_ptksa):
if fail_ptk:
return
raise Exception("Could not get PTKSA entry")
if not sta_ptksa and not ap_ptksa:
source = 'both'
elif sta_ptksa:
source = 'ap'
else:
source = 'sta'
raise Exception("Could not get PTKSA entry from %s" % source)
logger.info("sta: TK: %s KDK: %s" % (sta_ptksa['tk'], sta_ptksa['kdk']))
logger.info("ap : TK: %s KDK: %s" % (ap_ptksa['tk'], ap_ptksa['kdk']))