feat(edge): PostClient with HMAC+nonce signed event POST
This commit is contained in:
parent
5da838ef0b
commit
67464971bf
2 changed files with 86 additions and 0 deletions
44
edge/alert.py
Normal file
44
edge/alert.py
Normal file
|
|
@ -0,0 +1,44 @@
|
|||
import base64
|
||||
import hashlib
|
||||
import hmac
|
||||
import json
|
||||
|
||||
from edge.events import Event
|
||||
|
||||
|
||||
def sign(secret: str, nonce: str, ts: str, body: bytes) -> str:
|
||||
msg = nonce.encode() + ts.encode() + body
|
||||
return hmac.new(secret.encode(), msg, hashlib.sha256).hexdigest()
|
||||
|
||||
|
||||
class PostClient:
|
||||
def __init__(self, url: str, device_id: str, secret: str,
|
||||
nonce_fn, now_fn, http_post=None, timeout: float = 5.0):
|
||||
self.url = url
|
||||
self.device_id = device_id
|
||||
self.secret = secret
|
||||
self.nonce_fn = nonce_fn
|
||||
self.now_fn = now_fn
|
||||
self.timeout = timeout
|
||||
if http_post is None:
|
||||
import requests
|
||||
http_post = requests.post
|
||||
self._post = http_post
|
||||
|
||||
def send(self, event: Event, snapshots: list[bytes]) -> bool:
|
||||
payload = event.to_payload(self.device_id)
|
||||
payload["snapshots"] = [base64.b64encode(s).decode() for s in snapshots]
|
||||
body = json.dumps(payload, sort_keys=True)
|
||||
nonce = self.nonce_fn()
|
||||
ts = str(self.now_fn())
|
||||
headers = {
|
||||
"Content-Type": "application/json",
|
||||
"X-Nonce": nonce,
|
||||
"X-Timestamp": ts,
|
||||
"X-Signature": sign(self.secret, nonce, ts, body.encode()),
|
||||
}
|
||||
try:
|
||||
resp = self._post(self.url, data=body, headers=headers, timeout=self.timeout)
|
||||
except Exception:
|
||||
return False
|
||||
return 200 <= resp.status_code < 300
|
||||
42
tests/test_alert.py
Normal file
42
tests/test_alert.py
Normal file
|
|
@ -0,0 +1,42 @@
|
|||
import json
|
||||
from edge.events import Event, EventType, Severity
|
||||
from edge.alert import sign, PostClient
|
||||
|
||||
def _event():
|
||||
return Event(EventType.FALL, Severity.HIGH, "cam-01", 7, None, 1000.0, 3.0,
|
||||
{"torso_angle": 78.0}, None)
|
||||
|
||||
class _FakeResp:
|
||||
def __init__(self, code): self.status_code = code
|
||||
|
||||
def test_sign_is_deterministic():
|
||||
s1 = sign("secret", "n1", "1000", b"body")
|
||||
s2 = sign("secret", "n1", "1000", b"body")
|
||||
assert s1 == s2 and len(s1) == 64
|
||||
|
||||
def test_send_posts_signed_payload():
|
||||
captured = {}
|
||||
def fake_post(url, data=None, headers=None, timeout=None):
|
||||
captured["url"] = url
|
||||
captured["headers"] = headers
|
||||
captured["body"] = data
|
||||
return _FakeResp(200)
|
||||
|
||||
client = PostClient("https://x/api/events", "pi-01", "secret",
|
||||
nonce_fn=lambda: "n1", now_fn=lambda: 1000.0,
|
||||
http_post=fake_post)
|
||||
ok = client.send(_event(), [b"\xff\xd8jpeg"])
|
||||
assert ok is True
|
||||
assert captured["headers"]["X-Nonce"] == "n1"
|
||||
assert captured["headers"]["X-Timestamp"] == "1000.0"
|
||||
payload = json.loads(captured["body"])
|
||||
assert payload["device_id"] == "pi-01"
|
||||
assert payload["event_type"] == "fall"
|
||||
assert len(payload["snapshots"]) == 1
|
||||
expected = sign("secret", "n1", "1000.0", captured["body"].encode())
|
||||
assert captured["headers"]["X-Signature"] == expected
|
||||
|
||||
def test_send_returns_false_on_error_code():
|
||||
client = PostClient("https://x", "pi-01", "s", lambda: "n", lambda: 1.0,
|
||||
http_post=lambda *a, **k: _FakeResp(500))
|
||||
assert client.send(_event(), []) is False
|
||||
Loading…
Reference in a new issue