StarMirror/py-libp2p
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Raise DecryptionFailedException

Browse Source 
when failed to decrypt read msg
This commit is contained in:
NIC619 2019-10-17 14:30:30 +08:00
parent 5e5c96f1ea
commit 29bf623d2c
No known key found for this signature in database
GPG Key ID: 570C35F5C2D51B17
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
libp2p/security/secio/transport.py
View File

@ -4,6 +4,7 @@ import itertools
from typing import Optional, Tuple from typing import Optional, Tuple
import multihash import multihash
from crypto.authenticated_encryption import InvalidMACException
from libp2p.crypto.authenticated_encryption import ( from libp2p.crypto.authenticated_encryption import (
EncryptionParameters as AuthenticatedEncryptionParameters, EncryptionParameters as AuthenticatedEncryptionParameters,
@ -16,7 +17,7 @@ from libp2p.crypto.ecc import ECCPublicKey
from libp2p.crypto.key_exchange import create_ephemeral_key_pair from libp2p.crypto.key_exchange import create_ephemeral_key_pair
from libp2p.crypto.keys import PrivateKey, PublicKey from libp2p.crypto.keys import PrivateKey, PublicKey
from libp2p.crypto.serialization import deserialize_public_key from libp2p.crypto.serialization import deserialize_public_key
from libp2p.io.exceptions import IOException from libp2p.io.exceptions import DecryptionFailedException, IOException
from libp2p.io.msgio import MsgIOReadWriter from libp2p.io.msgio import MsgIOReadWriter
from libp2p.network.connection.raw_connection_interface import IRawConnection from libp2p.network.connection.raw_connection_interface import IRawConnection
from libp2p.peer.id import ID as PeerID from libp2p.peer.id import ID as PeerID
@ -122,7 +123,11 @@ class SecureSession(BaseSession):
async def read_msg(self) -> bytes: async def read_msg(self) -> bytes:
msg = await self.conn.read_msg() msg = await self.conn.read_msg()
return self.remote_encrypter.decrypt_if_valid(msg) try:
decrypted_msg = self.remote_encrypter.decrypt_if_valid(msg)
except InvalidMACException:
raise DecryptionFailedException
return decrypted_msg
async def write(self, data: bytes) -> int: async def write(self, data: bytes) -> int:
await self.write_msg(data) await self.write_msg(data)