Philipp Hörist pushed to branch master at gajim / python-nbxmpp
Commits:
-
09cffc03
by Philipp Hörist at 2023-05-10T20:12:53+02:00
2 changed files:
Changes:
| ... | ... | @@ -161,7 +161,7 @@ class OMEMO(BaseModule): |
| 161 | 161 | if not items:
|
| 162 | 162 | yield task.set_result(None)
|
| 163 | 163 | |
| 164 | - yield _parse_bundle(items[0])
|
|
| 164 | + yield _parse_bundle(items[0], device_id)
|
|
| 165 | 165 | |
| 166 | 166 | |
| 167 | 167 | def _parse_omemo_message(stanza):
|
| ... | ... | @@ -235,7 +235,7 @@ def _parse_omemo_message(stanza): |
| 235 | 235 | return OMEMOMessage(sid=sid, iv=iv, keys=keys, payload=payload)
|
| 236 | 236 | |
| 237 | 237 | |
| 238 | -def _parse_bundle(item):
|
|
| 238 | +def _parse_bundle(item, device_id):
|
|
| 239 | 239 | '''
|
| 240 | 240 | <item id='current'>
|
| 241 | 241 | <bundle xmlns='eu.siacs.conversations.axolotl'>
|
| ... | ... | @@ -317,7 +317,9 @@ def _parse_bundle(item): |
| 317 | 317 | |
| 318 | 318 | result['otpks'].append({'key': key, 'id': id_})
|
| 319 | 319 | |
| 320 | - return OMEMOBundle(**result)
|
|
| 320 | + return OMEMOBundle(**result,
|
|
| 321 | + device_id=device_id,
|
|
| 322 | + namespace=Namespace.OMEMO_TEMP)
|
|
| 321 | 323 | |
| 322 | 324 | |
| 323 | 325 | def _make_bundle(bundle):
|
| ... | ... | @@ -732,6 +732,8 @@ class OMEMOBundle(NamedTuple): |
| 732 | 732 | spk_signature: bytes
|
| 733 | 733 | ik: bytes
|
| 734 | 734 | otpks: list[dict[str, str]]
|
| 735 | + device_id: int = -1
|
|
| 736 | + namespace: str = Namespace.OMEMO_TEMP
|
|
| 735 | 737 | |
| 736 | 738 | def pick_prekey(self) -> dict[str, str]:
|
| 737 | 739 | return random.SystemRandom().choice(self.otpks)
|