varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2025-12-31 20:36:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Noise: encrypt and decrypt in NoiseConnection

Browse Source 
TODO: Add a buffer to read only `n` bytes in `read(n)`
This commit is contained in:
mhchia
2020-02-17 16:30:44 +08:00
parent 6ea96e9313
commit f8739268e2
3 changed files with 34 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
libp2p/security/noise/io.py
View File

@ -88,7 +88,6 @@ def decode_msg_body(noise_msg: bytes) -> bytes:
]
# TODO: Add comments
class NoiseHandshakeReadWriter(MsgReadWriter):
read_writer: MsgReadWriter
noise_state: NoiseState
@ -106,3 +105,22 @@ class NoiseHandshakeReadWriter(MsgReadWriter):
noise_msg_encrypted = await self.read_writer.read_msg()
noise_msg = self.noise_state.read_message(noise_msg_encrypted)
return decode_msg_body(noise_msg)
class NoiseTransportReadWriter(MsgReadWriter):
read_writer: MsgReadWriter
noise_state: NoiseState
def __init__(self, conn: IRawConnection, noise_state: NoiseState) -> None:
self.read_writer = NoisePacketReadWriter(cast(ReadWriter, conn))
self.noise_state = noise_state
async def write_msg(self, data: bytes) -> None:
noise_msg = encode_msg_body(data)
data_encrypted = self.noise_state.encrypt(noise_msg)
await self.read_writer.write_msg(data_encrypted)
async def read_msg(self) -> bytes:
noise_msg_encrypted = await self.read_writer.read_msg()
noise_msg = self.noise_state.decrypt(noise_msg_encrypted)
return decode_msg_body(noise_msg)