BaseMsgReadWriter

- Change `BaseMsgReadWriter` to encode/decode messages with abstract
method, which can be implemented by the subclasses. This allows us to
create subclasses `FixedSizeLenMsgReadWriter` and
`VarIntLenMsgReadWriter`.

libp2p/security/insecure/transport.py

@@ -3,7 +3,7 @@ from libp2p.crypto.keys import PrivateKey, PublicKey
 from libp2p.crypto.pb import crypto_pb2
 from libp2p.crypto.serialization import deserialize_public_key
 from libp2p.io.abc import ReadWriteCloser
-from libp2p.io.msgio import BaseMsgReadWriter
+from libp2p.io.msgio import FixedSizeLenMsgReadWriter
 from libp2p.network.connection.exceptions import RawConnError
 from libp2p.network.connection.raw_connection_interface import IRawConnection
 from libp2p.peer.id import ID
@@ -23,7 +23,7 @@ PLAINTEXT_PROTOCOL_ID = TProtocol("/plaintext/2.0.0")
 SIZE_PLAINTEXT_LEN_BYTES = 4
 
 
-class PlaintextHandshakeReadWriter(BaseMsgReadWriter):
+class PlaintextHandshakeReadWriter(FixedSizeLenMsgReadWriter):
     size_len_bytes = SIZE_PLAINTEXT_LEN_BYTES
 
 

libp2p/security/noise/io.py

@@ -3,7 +3,7 @@ from typing import cast
 from noise.connection import NoiseConnection as NoiseState
 
 from libp2p.io.abc import EncryptedMsgReadWriter, MsgReadWriteCloser, ReadWriteCloser
-from libp2p.io.msgio import BaseMsgReadWriter, encode_msg_with_length
+from libp2p.io.msgio import FixedSizeLenMsgReadWriter, encode_msg_with_length
 from libp2p.network.connection.raw_connection_interface import IRawConnection
 
 SIZE_NOISE_MESSAGE_LEN = 2
@@ -19,7 +19,7 @@ BYTE_ORDER = "big"
 #                   <-2 bytes-><-       max=65533 bytes        ->
 
 
-class NoisePacketReadWriter(BaseMsgReadWriter):
+class NoisePacketReadWriter(FixedSizeLenMsgReadWriter):
     size_len_bytes = SIZE_NOISE_MESSAGE_LEN
 
 

libp2p/security/secio/transport.py

@@ -19,7 +19,7 @@ from libp2p.crypto.keys import PrivateKey, PublicKey
 from libp2p.crypto.serialization import deserialize_public_key
 from libp2p.io.abc import EncryptedMsgReadWriter
 from libp2p.io.exceptions import DecryptionFailedException, IOException
-from libp2p.io.msgio import BaseMsgReadWriter
+from libp2p.io.msgio import FixedSizeLenMsgReadWriter
 from libp2p.network.connection.raw_connection_interface import IRawConnection
 from libp2p.peer.id import ID as PeerID
 from libp2p.security.base_transport import BaseSecureTransport
@@ -50,18 +50,18 @@ DEFAULT_SUPPORTED_CIPHERS = "AES-128"
 DEFAULT_SUPPORTED_HASHES = "SHA256"
 
 
-class MsgIOReadWriter(BaseMsgReadWriter):
+class SecioPacketReadWriter(FixedSizeLenMsgReadWriter):
     size_len_bytes = SIZE_SECIO_LEN_BYTES
 
 
 class SecioMsgReadWriter(EncryptedMsgReadWriter):
-    read_writer: MsgIOReadWriter
+    read_writer: SecioPacketReadWriter
 
     def __init__(
         self,
         local_encryption_parameters: AuthenticatedEncryptionParameters,
         remote_encryption_parameters: AuthenticatedEncryptionParameters,
-        read_writer: MsgIOReadWriter,
+        read_writer: SecioPacketReadWriter,
     ) -> None:
         self.local_encryption_parameters = local_encryption_parameters
         self.remote_encryption_parameters = remote_encryption_parameters
@@ -170,7 +170,7 @@ class SessionParameters:
         pass
 
 
-async def _response_to_msg(read_writer: MsgIOReadWriter, msg: bytes) -> bytes:
+async def _response_to_msg(read_writer: SecioPacketReadWriter, msg: bytes) -> bytes:
     await read_writer.write_msg(msg)
     return await read_writer.read_msg()
 
@@ -234,7 +234,7 @@ async def _establish_session_parameters(
     local_peer: PeerID,
     local_private_key: PrivateKey,
     remote_peer: Optional[PeerID],
-    conn: MsgIOReadWriter,
+    conn: SecioPacketReadWriter,
     nonce: bytes,
 ) -> Tuple[SessionParameters, bytes]:
     # establish shared encryption parameters
@@ -326,7 +326,7 @@ async def _establish_session_parameters(
 def _mk_session_from(
     local_private_key: PrivateKey,
     session_parameters: SessionParameters,
-    conn: MsgIOReadWriter,
+    conn: SecioPacketReadWriter,
     is_initiator: bool,
 ) -> SecureSession:
     key_set1, key_set2 = initialize_pair_for_encryption(
@@ -371,7 +371,7 @@ async def create_secure_session(
     to the ``remote_peer``. Raise `SecioException` when `conn` closed.
     Raise `InconsistentNonce` when handshake failed
     """
-    msg_io = MsgIOReadWriter(conn)
+    msg_io = SecioPacketReadWriter(conn)
     try:
         session_parameters, remote_nonce = await _establish_session_parameters(
             local_peer, local_private_key, remote_peer, msg_io, local_nonce