feat: v0.3.8 — full access control stack, GRANTED working, NTAG424 ISO7816 read, PN532 driver
This commit is contained in:
@@ -8,8 +8,9 @@ import utime
|
||||
import ujson
|
||||
import ubinascii
|
||||
|
||||
MAX_ATTEMPTS = 3
|
||||
ALARM_TIMEOUT = 30 # seconds
|
||||
MAX_ATTEMPTS = 3
|
||||
ALARM_TIMEOUT = 30 # seconds
|
||||
DISPLAY_TIMEOUT = 30 # seconds idle before OLED off
|
||||
|
||||
|
||||
# =============================================================================
|
||||
@@ -17,40 +18,12 @@ ALARM_TIMEOUT = 30 # seconds
|
||||
# =============================================================================
|
||||
|
||||
def _draw_nfc_idle(oled, frame=0):
|
||||
"""NFC waves animation — 4 frames."""
|
||||
oled.fill(0)
|
||||
cx, cy = 38, 36
|
||||
|
||||
# N symbol
|
||||
oled.text('N', cx - 4, cy + 6)
|
||||
|
||||
# Waves — expanding arcs approximated with pixels
|
||||
radii = [8, 14, 20]
|
||||
for i, r in enumerate(radii):
|
||||
if i < (frame % 4):
|
||||
# Draw arc (top-right quarter approximation)
|
||||
for angle in range(0, 90, 5):
|
||||
import math
|
||||
x = int(cx + r * math.cos(math.radians(angle)))
|
||||
y = int(cy - r * math.sin(math.radians(angle)))
|
||||
if 0 <= x < 128 and 0 <= y < 64:
|
||||
oled.pixel(x, y, 1)
|
||||
|
||||
oled.text('SCAN BADGE', 30, 56)
|
||||
oled.show()
|
||||
|
||||
|
||||
def _draw_nfc_idle_simple(oled, frame=0):
|
||||
"""Simple NFC animation without math — concentric arcs."""
|
||||
oled.fill(0)
|
||||
|
||||
# Animate waves based on frame
|
||||
waves = frame % 4
|
||||
cx, cy = 40, 28
|
||||
|
||||
waves = frame % 4
|
||||
if waves >= 1:
|
||||
oled.hline(cx, cy - 6, 4, 1)
|
||||
oled.vline(cx + 4, cy - 6, 12, 1)
|
||||
oled.hline(cx, cy - 6, 4, 1)
|
||||
oled.vline(cx + 4, cy - 6, 12, 1)
|
||||
if waves >= 2:
|
||||
oled.hline(cx - 4, cy - 10, 4, 1)
|
||||
oled.vline(cx + 8, cy - 10, 20, 1)
|
||||
@@ -59,18 +32,13 @@ def _draw_nfc_idle_simple(oled, frame=0):
|
||||
oled.hline(cx - 8, cy - 14, 4, 1)
|
||||
oled.vline(cx + 12, cy - 14, 28, 1)
|
||||
oled.hline(cx - 8, cy + 14, 4, 1)
|
||||
|
||||
# NFC dot
|
||||
oled.fill_rect(cx - 2, cy - 2, 5, 5, 1)
|
||||
|
||||
# Bottom text
|
||||
oled.text('SCAN BADGE', 20, 54)
|
||||
oled.show()
|
||||
|
||||
|
||||
def _draw_granted(oled, label=''):
|
||||
oled.fill(0)
|
||||
# Big checkmark
|
||||
oled.line(20, 35, 32, 47, 1)
|
||||
oled.line(32, 47, 56, 23, 1)
|
||||
oled.line(21, 35, 33, 47, 1)
|
||||
@@ -83,7 +51,6 @@ def _draw_granted(oled, label=''):
|
||||
|
||||
def _draw_denied(oled, attempts, max_attempts):
|
||||
oled.fill(0)
|
||||
# X mark
|
||||
oled.line(20, 20, 44, 44, 1)
|
||||
oled.line(44, 20, 20, 44, 1)
|
||||
oled.line(21, 20, 45, 44, 1)
|
||||
@@ -97,8 +64,8 @@ def _draw_alarm(oled, blink=True):
|
||||
oled.fill(1 if blink else 0)
|
||||
fg = 0 if blink else 1
|
||||
oled.text(' !! ALARM !!', 0, 8, fg)
|
||||
oled.text(' ACCESS', 0, 28, fg)
|
||||
oled.text(' DENIED', 0, 40, fg)
|
||||
oled.text(' ACCESS', 0, 28, fg)
|
||||
oled.text(' DENIED', 0, 40, fg)
|
||||
oled.show()
|
||||
|
||||
|
||||
@@ -114,14 +81,12 @@ def run_access_control(cfg, caps):
|
||||
while True:
|
||||
utime.sleep(60)
|
||||
|
||||
# Get hardware refs from boot_manager globals
|
||||
import boot_manager as bm
|
||||
oled = bm._oled
|
||||
nfc = bm._nfc
|
||||
|
||||
key = cfg['comm_key'].encode() if isinstance(cfg['comm_key'], str) else cfg['comm_key']
|
||||
iv = cfg['comm_iv'].encode() if isinstance(cfg['comm_iv'], str) else cfg['comm_iv']
|
||||
relay_pin = cfg.get('pinout', {}).get('relay_pin', 16)
|
||||
|
||||
import urequests
|
||||
from identity_iot_aes import aes_cbc_encrypt, aes_cbc_decrypt
|
||||
@@ -130,7 +95,9 @@ def run_access_control(cfg, caps):
|
||||
tag_url = cfg['identity_tag_url']
|
||||
attempts = 0
|
||||
frame = 0
|
||||
alarm_until = 0
|
||||
alarm_until = 0
|
||||
last_activity = utime.time()
|
||||
display_on = True
|
||||
|
||||
while True:
|
||||
now = utime.time()
|
||||
@@ -144,21 +111,59 @@ def run_access_control(cfg, caps):
|
||||
utime.sleep_ms(500)
|
||||
continue
|
||||
else:
|
||||
attempts = 0
|
||||
attempts = 0
|
||||
alarm_until = 0
|
||||
last_activity = now
|
||||
if oled and not display_on:
|
||||
oled.poweron()
|
||||
display_on = True
|
||||
|
||||
# Display timeout — OLED off after inactivity
|
||||
if display_on and (now - last_activity) > DISPLAY_TIMEOUT:
|
||||
if oled:
|
||||
oled.poweroff()
|
||||
display_on = False
|
||||
|
||||
# Idle animation
|
||||
_draw_nfc_idle_simple(oled, frame)
|
||||
if display_on:
|
||||
_draw_nfc_idle(oled, frame)
|
||||
frame += 1
|
||||
utime.sleep_ms(400)
|
||||
|
||||
# Poll for badge
|
||||
uid = nfc.read_uid(timeout=300)
|
||||
if uid is None:
|
||||
# Poll for badge — read UID + NDEF in one shot
|
||||
try:
|
||||
uid_hex, ndef_text = nfc.read_ndef_iso(timeout=300)
|
||||
except Exception:
|
||||
uid_hex, ndef_text = None, None
|
||||
|
||||
if ndef_text is None:
|
||||
continue
|
||||
if uid_hex:
|
||||
uid_hex = uid_hex.upper()
|
||||
|
||||
# Wake display and reset timeout on any badge detection
|
||||
if not display_on:
|
||||
if oled:
|
||||
oled.poweron()
|
||||
display_on = True
|
||||
last_activity = utime.time()
|
||||
|
||||
# Parse NDEF payload
|
||||
try:
|
||||
payload = ujson.loads(ndef_text)
|
||||
except Exception:
|
||||
print('[access_control] invalid NDEF JSON')
|
||||
continue
|
||||
|
||||
uid_hex = ubinascii.hexlify(uid).decode()
|
||||
print('[access_control] badge uid:', uid_hex)
|
||||
epk = payload.get('epk', '')
|
||||
df2 = payload.get('df2', '')
|
||||
mode = payload.get('mode', 'std')
|
||||
|
||||
if not epk or not df2:
|
||||
print('[access_control] missing epk or df2')
|
||||
continue
|
||||
|
||||
print('[access_control] badge uid:', uid_hex, 'mode:', mode)
|
||||
|
||||
# Get fresh JWT
|
||||
try:
|
||||
@@ -176,15 +181,14 @@ def run_access_control(cfg, caps):
|
||||
utime.sleep(2)
|
||||
continue
|
||||
|
||||
# Challenge badge against identity_tag
|
||||
# Step 1 — Challenge
|
||||
try:
|
||||
payload = ujson.dumps({
|
||||
'request': 'challenge',
|
||||
'badge_uid': uid_hex,
|
||||
'node_id': str(cfg['node_id']),
|
||||
req = ujson.dumps({
|
||||
'request': 'challenge',
|
||||
'tag_uid': uid_hex,
|
||||
'tenant_id': str(cfg['tenant_id']),
|
||||
})
|
||||
cipher = aes_cbc_encrypt(payload, key, iv)
|
||||
cipher = aes_cbc_encrypt(req, key, iv)
|
||||
resp = urequests.post(
|
||||
tag_url,
|
||||
headers={
|
||||
@@ -196,14 +200,94 @@ def run_access_control(cfg, caps):
|
||||
)
|
||||
raw = resp.text
|
||||
resp.close()
|
||||
outer = ujson.loads(raw)
|
||||
outer = ujson.loads(raw)
|
||||
ch = ujson.loads(aes_cbc_decrypt(outer['data'], key, iv))
|
||||
|
||||
if ch.get('status') != 'ok':
|
||||
raise Exception('challenge: ' + str(ch.get('code', 'error')))
|
||||
|
||||
nonce_b64 = ch['challenge']
|
||||
challenge_token = ch['challenge_token']
|
||||
df1_enc = ch.get('df1_encrypted', '')
|
||||
|
||||
except Exception as e:
|
||||
print('[access_control] challenge error:', e)
|
||||
attempts += 1
|
||||
_draw_denied(oled, attempts, MAX_ATTEMPTS)
|
||||
utime.sleep(2)
|
||||
nfc.release_target()
|
||||
continue
|
||||
|
||||
# Step 2 — Reconstruct DEK, decrypt private key, sign nonce
|
||||
try:
|
||||
from identity_iot_aes import aes_cbc_decrypt as aes_dec
|
||||
|
||||
# Decrypt df1 from backend
|
||||
df1 = ubinascii.a2b_base64(aes_dec(df1_enc, key, iv))
|
||||
df2_bytes = ubinascii.a2b_base64(df2)
|
||||
|
||||
# Reconstruct DEK = df1 XOR df2
|
||||
dek = bytes([df1[i] ^ df2_bytes[i] for i in range(32)])
|
||||
|
||||
# Decrypt private key (epk) with DEK — AES-CBC, IV = first 16 bytes of DEK
|
||||
from ucryptolib import aes as _aes
|
||||
import ubinascii as _ub
|
||||
epk_bytes = ubinascii.a2b_base64(epk)
|
||||
cipher_obj = _aes(dek, 2, dek[:16])
|
||||
priv_b64_padded = cipher_obj.decrypt(epk_bytes)
|
||||
# PKCS7 unpad
|
||||
pad = priv_b64_padded[-1]
|
||||
priv_b64 = priv_b64_padded[:-pad].decode('utf-8')
|
||||
seed = ubinascii.a2b_base64(priv_b64)[:32]
|
||||
|
||||
|
||||
|
||||
from identity_iot_ed25519 import ed25519_keypair_from_seed
|
||||
pk, _ = ed25519_keypair_from_seed(seed)
|
||||
|
||||
# Sign nonce
|
||||
from identity_iot_ed25519 import ed25519_sign
|
||||
nonce_bytes = ubinascii.a2b_base64(nonce_b64)
|
||||
signature = ed25519_sign(seed, nonce_bytes)
|
||||
|
||||
sig_b64 = ubinascii.b2a_base64(signature).decode().replace('', '').replace('', '').strip()
|
||||
|
||||
except Exception as e:
|
||||
print('[access_control] sign error:', e)
|
||||
attempts += 1
|
||||
_draw_denied(oled, attempts, MAX_ATTEMPTS)
|
||||
utime.sleep(2)
|
||||
nfc.release_target()
|
||||
continue
|
||||
|
||||
# Step 3 — Verify signature
|
||||
try:
|
||||
req = ujson.dumps({
|
||||
'request': 'verify_signature',
|
||||
'tag_uid': uid_hex,
|
||||
'signature_b64': sig_b64,
|
||||
'challenge_token': challenge_token,
|
||||
'tenant_id': str(cfg['tenant_id']),
|
||||
})
|
||||
cipher = aes_cbc_encrypt(req, key, iv)
|
||||
resp = urequests.post(
|
||||
tag_url,
|
||||
headers={
|
||||
'Content-Type': 'application/json',
|
||||
'Authorization': 'Bearer ' + token.strip(),
|
||||
},
|
||||
data=ujson.dumps({'data': cipher}),
|
||||
timeout=10,
|
||||
)
|
||||
raw = resp.text
|
||||
resp.close()
|
||||
outer = ujson.loads(raw)
|
||||
result = ujson.loads(aes_cbc_decrypt(outer['data'], key, iv))
|
||||
|
||||
if result.get('status') == 'ok':
|
||||
print('[access_control] GRANTED:', uid_hex)
|
||||
attempts = 0
|
||||
_draw_granted(oled, cfg.get('node_label', ''))
|
||||
# Trigger relay
|
||||
if caps.get('has_relay') and bm._relay:
|
||||
bm._relay.value(1)
|
||||
utime.sleep_ms(500)
|
||||
@@ -216,7 +300,7 @@ def run_access_control(cfg, caps):
|
||||
utime.sleep(2)
|
||||
|
||||
except Exception as e:
|
||||
print('[access_control] tag error:', e)
|
||||
print('[access_control] verify error:', e)
|
||||
utime.sleep(1)
|
||||
|
||||
nfc.release_target()
|
||||
Reference in New Issue
Block a user