derivepassphrase.git

src/ssh_agent_client/__init__.py             1) # SPDX-FileCopyrightText: 2024 Marco Ricci <m@the13thletter.info>
src/ssh_agent_client/__init__.py             2) #
src/ssh_agent_client/__init__.py             3) # SPDX-License-Identifier: MIT
src/ssh_agent_client/__init__.py             4) 
src/ssh_agent_client/__init__.py             5) """A bare-bones SSH agent client supporting signing and key listing."""
src/ssh_agent_client/__init__.py             6) 
src/ssh_agent_client/__init__.py             7) from __future__ import annotations
src/ssh_agent_client/__init__.py             8) 
src/ssh_agent_client/__init__.py             9) import collections
src/ssh_agent_client/__init__.py            10) import errno
src/ssh_agent_client/__init__.py            11) import os
src/ssh_agent_client/__init__.py            12) import socket

src/ssh_agent_client/__init__.py            13) from typing import TYPE_CHECKING

src/ssh_agent_client/__init__.py            14) 

src/ssh_agent_client/__init__.py            15) from typing_extensions import Self

src/ssh_agent_client/__init__.py            16) 

src/derivepassphrase/ssh_agent/__init__.py  17) from derivepassphrase import _types

src/ssh_agent_client/__init__.py            18) 
src/ssh_agent_client/__init__.py            19) if TYPE_CHECKING:
src/ssh_agent_client/__init__.py            20)     from collections.abc import Sequence

src/ssh_agent_client/__init__.py            21)     from types import TracebackType

src/ssh_agent_client/__init__.py            22) 
src/ssh_agent_client/__init__.py            23) __all__ = ('SSHAgentClient',)

src/ssh_agent_client/__init__.py            24) __author__ = 'Marco Ricci <m@the13thletter.info>'

src/ssh_agent_client/__init__.py            25) 

src/ssh_agent_client/__init__.py            26) # In SSH bytestrings, the "length" of the byte string is stored as
src/ssh_agent_client/__init__.py            27) # a 4-byte/32-bit unsigned integer at the beginning.
src/ssh_agent_client/__init__.py            28) HEAD_LEN = 4
src/ssh_agent_client/__init__.py            29) 

src/ssh_agent_client/__init__.py            30) _socket = socket
src/ssh_agent_client/__init__.py            31) 

src/ssh_agent_client/__init__.py            32) 

src/ssh_agent_client/__init__.py            33) class TrailingDataError(RuntimeError):
src/ssh_agent_client/__init__.py            34)     """The result contained trailing data."""

src/ssh_agent_client/__init__.py            35) 

src/ssh_agent_client/__init__.py            36)     def __init__(self) -> None:

src/ssh_agent_client/__init__.py            37)         super().__init__('Overlong response from SSH agent')

src/ssh_agent_client/__init__.py            38) 

src/ssh_agent_client/__init__.py            39) 

src/derivepassphrase/ssh_agent.py           40) class SSHAgentFailedError(RuntimeError):
src/derivepassphrase/ssh_agent.py           41)     """The SSH agent failed to complete the requested operation."""
src/derivepassphrase/ssh_agent.py           42) 
src/derivepassphrase/ssh_agent.py           43)     def __str__(self) -> str:
src/derivepassphrase/ssh_agent.py           44)         match self.args:
src/derivepassphrase/ssh_agent.py           45)             case (_types.SSH_AGENT.FAILURE.value, b''):  # pragma: no branch
src/derivepassphrase/ssh_agent.py           46)                 return 'The SSH agent failed to complete the request'
src/derivepassphrase/ssh_agent.py           47)             case (_, _msg) if _msg:  # pragma: no cover
src/derivepassphrase/ssh_agent.py           48)                 code = self.args[0]
src/derivepassphrase/ssh_agent.py           49)                 msg = self.args[1].decode('utf-8', 'surrogateescape')
src/derivepassphrase/ssh_agent.py           50)                 return f'[Code {code:d}] {msg:s}'
src/derivepassphrase/ssh_agent.py           51)             case _:  # pragma: no cover
src/derivepassphrase/ssh_agent.py           52)                 return repr(self)
src/derivepassphrase/ssh_agent.py           53) 
src/derivepassphrase/ssh_agent.py           54)     def __repr__(self) -> str:  # pragma: no cover
src/derivepassphrase/ssh_agent.py           55)         return f'{self.__class__.__name__}{self.args!r}'
src/derivepassphrase/ssh_agent.py           56) 
src/derivepassphrase/ssh_agent.py           57) 

src/ssh_agent_client/__init__.py            58) class SSHAgentClient:
src/ssh_agent_client/__init__.py            59)     """A bare-bones SSH agent client supporting signing and key listing.
src/ssh_agent_client/__init__.py            60) 
src/ssh_agent_client/__init__.py            61)     The main use case is requesting the agent sign some data, after
src/ssh_agent_client/__init__.py            62)     checking that the necessary key is already loaded.
src/ssh_agent_client/__init__.py            63) 
src/ssh_agent_client/__init__.py            64)     The main fleshed out methods are `list_keys` and `sign`, which
src/ssh_agent_client/__init__.py            65)     implement the `REQUEST_IDENTITIES` and `SIGN_REQUEST` requests.  If
src/ssh_agent_client/__init__.py            66)     you *really* wanted to, there is enough infrastructure in place to
src/ssh_agent_client/__init__.py            67)     issue other requests as defined in the protocol---it's merely the
src/ssh_agent_client/__init__.py            68)     wrapper functions and the protocol numbers table that are missing.
src/ssh_agent_client/__init__.py            69) 
src/ssh_agent_client/__init__.py            70)     """

src/ssh_agent_client/__init__.py            71) 

src/ssh_agent_client/__init__.py            72)     _connection: socket.socket

src/ssh_agent_client/__init__.py            73) 

src/ssh_agent_client/__init__.py            74)     def __init__(
src/ssh_agent_client/__init__.py            75)         self, /, *, socket: socket.socket | None = None, timeout: int = 125
src/ssh_agent_client/__init__.py            76)     ) -> None:
src/ssh_agent_client/__init__.py            77)         """Initialize the client.
src/ssh_agent_client/__init__.py            78) 
src/ssh_agent_client/__init__.py            79)         Args:
src/ssh_agent_client/__init__.py            80)             socket:
src/ssh_agent_client/__init__.py            81)                 An optional socket, connected to the SSH agent.  If not
src/ssh_agent_client/__init__.py            82)                 given, we query the `SSH_AUTH_SOCK` environment
src/ssh_agent_client/__init__.py            83)                 variable to auto-discover the correct socket address.
src/ssh_agent_client/__init__.py            84)             timeout:
src/ssh_agent_client/__init__.py            85)                 A connection timeout for the SSH agent.  Only used if
src/ssh_agent_client/__init__.py            86)                 the socket is not yet connected.  The default value
src/ssh_agent_client/__init__.py            87)                 gives ample time for agent connections forwarded via
src/ssh_agent_client/__init__.py            88)                 SSH on high-latency networks (e.g. Tor).
src/ssh_agent_client/__init__.py            89) 

src/ssh_agent_client/__init__.py            90)         Raises:
src/ssh_agent_client/__init__.py            91)             KeyError:
src/ssh_agent_client/__init__.py            92)                 The `SSH_AUTH_SOCK` environment was not found.
src/ssh_agent_client/__init__.py            93)             OSError:
src/ssh_agent_client/__init__.py            94)                 There was an error setting up a socket connection to the
src/ssh_agent_client/__init__.py            95)                 agent.
src/ssh_agent_client/__init__.py            96) 

src/ssh_agent_client/__init__.py            97)         """
src/ssh_agent_client/__init__.py            98)         if socket is not None:

src/ssh_agent_client/__init__.py            99)             self._connection = socket

src/ssh_agent_client/__init__.py           100)         else:

src/ssh_agent_client/__init__.py           101)             self._connection = _socket.socket(family=_socket.AF_UNIX)

src/ssh_agent_client/__init__.py           102)         try:

src/ssh_agent_client/__init__.py           103)             # Test whether the socket is connected.
src/ssh_agent_client/__init__.py           104)             self._connection.getpeername()

src/ssh_agent_client/__init__.py           105)         except OSError as e:

src/ssh_agent_client/__init__.py           106)             # This condition is hard to test purposefully, so exclude
src/ssh_agent_client/__init__.py           107)             # from coverage.
src/ssh_agent_client/__init__.py           108)             if e.errno != errno.ENOTCONN:  # pragma: no cover

src/ssh_agent_client/__init__.py           109)                 raise

src/ssh_agent_client/__init__.py           110)             if 'SSH_AUTH_SOCK' not in os.environ:

src/ssh_agent_client/__init__.py           111)                 msg = 'SSH_AUTH_SOCK environment variable'
src/ssh_agent_client/__init__.py           112)                 raise KeyError(msg) from None

src/ssh_agent_client/__init__.py           113)             ssh_auth_sock = os.environ['SSH_AUTH_SOCK']

src/ssh_agent_client/__init__.py           114)             self._connection.settimeout(timeout)

src/ssh_agent_client/__init__.py           115)             self._connection.connect(ssh_auth_sock)

src/ssh_agent_client/__init__.py           116) 
src/ssh_agent_client/__init__.py           117)     def __enter__(self) -> Self:

src/ssh_agent_client/__init__.py           118)         """Close socket connection upon context manager completion."""
src/ssh_agent_client/__init__.py           119)         self._connection.__enter__()

src/ssh_agent_client/__init__.py           120)         return self
src/ssh_agent_client/__init__.py           121) 
src/ssh_agent_client/__init__.py           122)     def __exit__(

src/ssh_agent_client/__init__.py           123)         self,
src/ssh_agent_client/__init__.py           124)         exc_type: type[BaseException] | None,
src/ssh_agent_client/__init__.py           125)         exc_val: BaseException | None,

src/ssh_agent_client/__init__.py           126)         exc_tb: TracebackType | None,

src/ssh_agent_client/__init__.py           127)     ) -> bool:

src/ssh_agent_client/__init__.py           128)         """Close socket connection upon context manager completion."""

src/ssh_agent_client/__init__.py           129)         return bool(

src/ssh_agent_client/__init__.py           130)             self._connection.__exit__(exc_type, exc_val, exc_tb)  # type: ignore[func-returns-value]

src/ssh_agent_client/__init__.py           131)         )
src/ssh_agent_client/__init__.py           132) 
src/ssh_agent_client/__init__.py           133)     @staticmethod

src/ssh_agent_client/__init__.py           134)     def uint32(num: int, /) -> bytes:
src/ssh_agent_client/__init__.py           135)         r"""Format the number as a `uint32`, as per the agent protocol.
src/ssh_agent_client/__init__.py           136) 
src/ssh_agent_client/__init__.py           137)         Args:
src/ssh_agent_client/__init__.py           138)             num: A number.
src/ssh_agent_client/__init__.py           139) 
src/ssh_agent_client/__init__.py           140)         Returns:
src/ssh_agent_client/__init__.py           141)             The number in SSH agent wire protocol format, i.e. as
src/ssh_agent_client/__init__.py           142)             a 32-bit big endian number.
src/ssh_agent_client/__init__.py           143) 
src/ssh_agent_client/__init__.py           144)         Raises:
src/ssh_agent_client/__init__.py           145)             OverflowError:
src/ssh_agent_client/__init__.py           146)                 As per [`int.to_bytes`][].
src/ssh_agent_client/__init__.py           147) 
src/ssh_agent_client/__init__.py           148)         Examples:
src/ssh_agent_client/__init__.py           149)             >>> SSHAgentClient.uint32(16777216)
src/ssh_agent_client/__init__.py           150)             b'\x01\x00\x00\x00'
src/ssh_agent_client/__init__.py           151) 
src/ssh_agent_client/__init__.py           152)         """

src/ssh_agent_client/__init__.py           153)         return int.to_bytes(num, 4, 'big', signed=False)
src/ssh_agent_client/__init__.py           154) 
src/ssh_agent_client/__init__.py           155)     @classmethod
src/ssh_agent_client/__init__.py           156)     def string(cls, payload: bytes | bytearray, /) -> bytes | bytearray:

src/ssh_agent_client/__init__.py           157)         r"""Format the payload as an SSH string, as per the agent protocol.
src/ssh_agent_client/__init__.py           158) 
src/ssh_agent_client/__init__.py           159)         Args:
src/ssh_agent_client/__init__.py           160)             payload: A byte string.
src/ssh_agent_client/__init__.py           161) 
src/ssh_agent_client/__init__.py           162)         Returns:
src/ssh_agent_client/__init__.py           163)             The payload, framed in the SSH agent wire protocol format.
src/ssh_agent_client/__init__.py           164) 
src/ssh_agent_client/__init__.py           165)         Examples:
src/ssh_agent_client/__init__.py           166)             >>> bytes(SSHAgentClient.string(b'ssh-rsa'))
src/ssh_agent_client/__init__.py           167)             b'\x00\x00\x00\x07ssh-rsa'
src/ssh_agent_client/__init__.py           168) 
src/ssh_agent_client/__init__.py           169)         """

src/ssh_agent_client/__init__.py           170)         try:
src/ssh_agent_client/__init__.py           171)             ret = bytearray()
src/ssh_agent_client/__init__.py           172)             ret.extend(cls.uint32(len(payload)))
src/ssh_agent_client/__init__.py           173)             ret.extend(payload)
src/ssh_agent_client/__init__.py           174)         except Exception as e:

src/ssh_agent_client/__init__.py           175)             msg = 'invalid payload type'
src/ssh_agent_client/__init__.py           176)             raise TypeError(msg) from e
src/ssh_agent_client/__init__.py           177)         return ret

src/ssh_agent_client/__init__.py           178) 
src/ssh_agent_client/__init__.py           179)     @classmethod
src/ssh_agent_client/__init__.py           180)     def unstring(cls, bytestring: bytes | bytearray, /) -> bytes | bytearray:

src/ssh_agent_client/__init__.py           181)         r"""Unpack an SSH string.
src/ssh_agent_client/__init__.py           182) 
src/ssh_agent_client/__init__.py           183)         Args:
src/ssh_agent_client/__init__.py           184)             bytestring: A framed byte string.
src/ssh_agent_client/__init__.py           185) 
src/ssh_agent_client/__init__.py           186)         Returns:
src/ssh_agent_client/__init__.py           187)             The unframed byte string, i.e., the payload.
src/ssh_agent_client/__init__.py           188) 
src/ssh_agent_client/__init__.py           189)         Raises:
src/ssh_agent_client/__init__.py           190)             ValueError:

src/ssh_agent_client/__init__.py           191)                 The byte string is not an SSH string.

src/ssh_agent_client/__init__.py           192) 
src/ssh_agent_client/__init__.py           193)         Examples:
src/ssh_agent_client/__init__.py           194)             >>> bytes(SSHAgentClient.unstring(b'\x00\x00\x00\x07ssh-rsa'))
src/ssh_agent_client/__init__.py           195)             b'ssh-rsa'

src/ssh_agent_client/__init__.py           196)             >>> bytes(
src/ssh_agent_client/__init__.py           197)             ...     SSHAgentClient.unstring(SSHAgentClient.string(b'ssh-ed25519'))
src/ssh_agent_client/__init__.py           198)             ... )

src/ssh_agent_client/__init__.py           199)             b'ssh-ed25519'
src/ssh_agent_client/__init__.py           200) 

src/ssh_agent_client/__init__.py           201)         """  # noqa: E501

src/ssh_agent_client/__init__.py           202)         n = len(bytestring)

src/ssh_agent_client/__init__.py           203)         msg = 'malformed SSH byte string'

src/ssh_agent_client/__init__.py           204)         if n < HEAD_LEN or n != HEAD_LEN + int.from_bytes(
src/ssh_agent_client/__init__.py           205)             bytestring[:HEAD_LEN], 'big', signed=False

src/ssh_agent_client/__init__.py           206)         ):
src/ssh_agent_client/__init__.py           207)             raise ValueError(msg)
src/ssh_agent_client/__init__.py           208)         return bytestring[HEAD_LEN:]

src/ssh_agent_client/__init__.py           209) 

src/ssh_agent_client/__init__.py           210)     @classmethod
src/ssh_agent_client/__init__.py           211)     def unstring_prefix(
src/ssh_agent_client/__init__.py           212)         cls, bytestring: bytes | bytearray, /
src/ssh_agent_client/__init__.py           213)     ) -> tuple[bytes | bytearray, bytes | bytearray]:
src/ssh_agent_client/__init__.py           214)         r"""Unpack an SSH string at the beginning of the byte string.
src/ssh_agent_client/__init__.py           215) 
src/ssh_agent_client/__init__.py           216)         Args:
src/ssh_agent_client/__init__.py           217)             bytestring:
src/ssh_agent_client/__init__.py           218)                 A (general) byte string, beginning with a framed/SSH
src/ssh_agent_client/__init__.py           219)                 byte string.
src/ssh_agent_client/__init__.py           220) 
src/ssh_agent_client/__init__.py           221)         Returns:
src/ssh_agent_client/__init__.py           222)             A 2-tuple `(a, b)`, where `a` is the unframed byte
src/ssh_agent_client/__init__.py           223)             string/payload at the beginning of input byte string, and
src/ssh_agent_client/__init__.py           224)             `b` is the remainder of the input byte string.
src/ssh_agent_client/__init__.py           225) 
src/ssh_agent_client/__init__.py           226)         Raises:
src/ssh_agent_client/__init__.py           227)             ValueError:
src/ssh_agent_client/__init__.py           228)                 The byte string does not begin with an SSH string.
src/ssh_agent_client/__init__.py           229) 
src/ssh_agent_client/__init__.py           230)         Examples:
src/ssh_agent_client/__init__.py           231)             >>> a, b = SSHAgentClient.unstring_prefix(

src/ssh_agent_client/__init__.py           232)             ...     b'\x00\x00\x00\x07ssh-rsa____trailing data'
src/ssh_agent_client/__init__.py           233)             ... )

src/ssh_agent_client/__init__.py           234)             >>> (bytes(a), bytes(b))
src/ssh_agent_client/__init__.py           235)             (b'ssh-rsa', b'____trailing data')
src/ssh_agent_client/__init__.py           236)             >>> a, b = SSHAgentClient.unstring_prefix(

src/ssh_agent_client/__init__.py           237)             ...     SSHAgentClient.string(b'ssh-ed25519')
src/ssh_agent_client/__init__.py           238)             ... )

src/ssh_agent_client/__init__.py           239)             >>> (bytes(a), bytes(b))
src/ssh_agent_client/__init__.py           240)             (b'ssh-ed25519', b'')
src/ssh_agent_client/__init__.py           241) 
src/ssh_agent_client/__init__.py           242)         """
src/ssh_agent_client/__init__.py           243)         n = len(bytestring)

src/ssh_agent_client/__init__.py           244)         msg = 'malformed SSH byte string'
src/ssh_agent_client/__init__.py           245)         if n < HEAD_LEN:
src/ssh_agent_client/__init__.py           246)             raise ValueError(msg)
src/ssh_agent_client/__init__.py           247)         m = int.from_bytes(bytestring[:HEAD_LEN], 'big', signed=False)
src/ssh_agent_client/__init__.py           248)         if m + HEAD_LEN > n:
src/ssh_agent_client/__init__.py           249)             raise ValueError(msg)

src/ssh_agent_client/__init__.py           250)         return (
src/ssh_agent_client/__init__.py           251)             bytestring[HEAD_LEN : m + HEAD_LEN],
src/ssh_agent_client/__init__.py           252)             bytestring[m + HEAD_LEN :],
src/ssh_agent_client/__init__.py           253)         )

src/ssh_agent_client/__init__.py           254) 

src/ssh_agent_client/__init__.py           255)     def request(
src/ssh_agent_client/__init__.py           256)         self, code: int, payload: bytes | bytearray, /
src/ssh_agent_client/__init__.py           257)     ) -> tuple[int, bytes | bytearray]:
src/ssh_agent_client/__init__.py           258)         """Issue a generic request to the SSH agent.
src/ssh_agent_client/__init__.py           259) 
src/ssh_agent_client/__init__.py           260)         Args:
src/ssh_agent_client/__init__.py           261)             code:
src/ssh_agent_client/__init__.py           262)                 The request code.  See the SSH agent protocol for
src/ssh_agent_client/__init__.py           263)                 protocol numbers to use here (and which protocol numbers
src/ssh_agent_client/__init__.py           264)                 to expect in a response).
src/ssh_agent_client/__init__.py           265)             payload:
src/ssh_agent_client/__init__.py           266)                 A byte string containing the payload, or "contents", of
src/ssh_agent_client/__init__.py           267)                 the request.  Request-specific.  `request` will add any
src/ssh_agent_client/__init__.py           268)                 necessary wire framing around the request code and the
src/ssh_agent_client/__init__.py           269)                 payload.
src/ssh_agent_client/__init__.py           270) 
src/ssh_agent_client/__init__.py           271)         Returns:
src/ssh_agent_client/__init__.py           272)             A 2-tuple consisting of the response code and the payload,
src/ssh_agent_client/__init__.py           273)             with all wire framing removed.
src/ssh_agent_client/__init__.py           274) 
src/ssh_agent_client/__init__.py           275)         Raises:
src/ssh_agent_client/__init__.py           276)             EOFError:
src/ssh_agent_client/__init__.py           277)                 The response from the SSH agent is truncated or missing.
src/ssh_agent_client/__init__.py           278) 
src/ssh_agent_client/__init__.py           279)         """
src/ssh_agent_client/__init__.py           280)         request_message = bytearray([code])
src/ssh_agent_client/__init__.py           281)         request_message.extend(payload)

src/ssh_agent_client/__init__.py           282)         self._connection.sendall(self.string(request_message))

src/ssh_agent_client/__init__.py           283)         chunk = self._connection.recv(HEAD_LEN)
src/ssh_agent_client/__init__.py           284)         if len(chunk) < HEAD_LEN:
src/ssh_agent_client/__init__.py           285)             msg = 'cannot read response length'
src/ssh_agent_client/__init__.py           286)             raise EOFError(msg)

src/ssh_agent_client/__init__.py           287)         response_length = int.from_bytes(chunk, 'big', signed=False)

src/ssh_agent_client/__init__.py           288)         response = self._connection.recv(response_length)

src/ssh_agent_client/__init__.py           289)         if len(response) < response_length:

src/ssh_agent_client/__init__.py           290)             msg = 'truncated response from SSH agent'
src/ssh_agent_client/__init__.py           291)             raise EOFError(msg)

src/ssh_agent_client/__init__.py           292)         return response[0], response[1:]
src/ssh_agent_client/__init__.py           293) 

src/derivepassphrase/ssh_agent/__init__.py 294)     def list_keys(self) -> Sequence[_types.KeyCommentPair]:

src/ssh_agent_client/__init__.py           295)         """Request a list of keys known to the SSH agent.
src/ssh_agent_client/__init__.py           296) 
src/ssh_agent_client/__init__.py           297)         Returns:
src/ssh_agent_client/__init__.py           298)             A read-only sequence of key/comment pairs.
src/ssh_agent_client/__init__.py           299) 
src/ssh_agent_client/__init__.py           300)         Raises:
src/ssh_agent_client/__init__.py           301)             EOFError:
src/ssh_agent_client/__init__.py           302)                 The response from the SSH agent is truncated or missing.

src/ssh_agent_client/__init__.py           303)             TrailingDataError:

src/ssh_agent_client/__init__.py           304)                 The response from the SSH agent is too long.

src/derivepassphrase/ssh_agent.py          305)             SSHAgentFailedError:

src/ssh_agent_client/__init__.py           306)                 The agent failed to complete the request.

src/ssh_agent_client/__init__.py           307) 
src/ssh_agent_client/__init__.py           308)         """
src/ssh_agent_client/__init__.py           309)         response_code, response = self.request(

src/derivepassphrase/ssh_agent/__init__.py 310)             _types.SSH_AGENTC.REQUEST_IDENTITIES.value, b''

src/ssh_agent_client/__init__.py           311)         )

src/derivepassphrase/ssh_agent/__init__.py 312)         if response_code != _types.SSH_AGENT.IDENTITIES_ANSWER.value:

src/derivepassphrase/ssh_agent.py          313)             raise SSHAgentFailedError(response_code, response)

src/ssh_agent_client/__init__.py           314)         response_stream = collections.deque(response)

src/ssh_agent_client/__init__.py           315) 

src/ssh_agent_client/__init__.py           316)         def shift(num: int) -> bytes:

src/ssh_agent_client/__init__.py           317)             buf = collections.deque(b'')
src/ssh_agent_client/__init__.py           318)             for _ in range(num):

src/ssh_agent_client/__init__.py           319)                 try:
src/ssh_agent_client/__init__.py           320)                     val = response_stream.popleft()
src/ssh_agent_client/__init__.py           321)                 except IndexError:
src/ssh_agent_client/__init__.py           322)                     response_stream.extendleft(reversed(buf))

src/ssh_agent_client/__init__.py           323)                     msg = 'truncated response from SSH agent'
src/ssh_agent_client/__init__.py           324)                     raise EOFError(msg) from None

src/ssh_agent_client/__init__.py           325)                 buf.append(val)
src/ssh_agent_client/__init__.py           326)             return bytes(buf)

src/ssh_agent_client/__init__.py           327) 

src/ssh_agent_client/__init__.py           328)         key_count = int.from_bytes(shift(4), 'big')

src/derivepassphrase/ssh_agent/__init__.py 329)         keys: collections.deque[_types.KeyCommentPair]

src/ssh_agent_client/__init__.py           330)         keys = collections.deque()

src/ssh_agent_client/__init__.py           331)         for _ in range(key_count):

src/ssh_agent_client/__init__.py           332)             key_size = int.from_bytes(shift(4), 'big')
src/ssh_agent_client/__init__.py           333)             key = shift(key_size)
src/ssh_agent_client/__init__.py           334)             comment_size = int.from_bytes(shift(4), 'big')
src/ssh_agent_client/__init__.py           335)             comment = shift(comment_size)
src/ssh_agent_client/__init__.py           336)             # Both `key` and `comment` are not wrapped as SSH strings.

src/derivepassphrase/ssh_agent/__init__.py 337)             keys.append(_types.KeyCommentPair(key, comment))

src/ssh_agent_client/__init__.py           338)         if response_stream:

src/ssh_agent_client/__init__.py           339)             raise TrailingDataError

src/ssh_agent_client/__init__.py           340)         return keys
src/ssh_agent_client/__init__.py           341) 
src/ssh_agent_client/__init__.py           342)     def sign(

src/ssh_agent_client/__init__.py           343)         self,
src/ssh_agent_client/__init__.py           344)         /,
src/ssh_agent_client/__init__.py           345)         key: bytes | bytearray,
src/ssh_agent_client/__init__.py           346)         payload: bytes | bytearray,
src/ssh_agent_client/__init__.py           347)         *,
src/ssh_agent_client/__init__.py           348)         flags: int = 0,
src/ssh_agent_client/__init__.py           349)         check_if_key_loaded: bool = False,

src/ssh_agent_client/__init__.py           350)     ) -> bytes | bytearray:
src/ssh_agent_client/__init__.py           351)         """Request the SSH agent sign the payload with the key.
src/ssh_agent_client/__init__.py           352) 
src/ssh_agent_client/__init__.py           353)         Args:
src/ssh_agent_client/__init__.py           354)             key:
src/ssh_agent_client/__init__.py           355)                 The public SSH key to sign the payload with, in the same
src/ssh_agent_client/__init__.py           356)                 format as returned by, e.g., the `list_keys` method.
src/ssh_agent_client/__init__.py           357)                 The corresponding private key must have previously been
src/ssh_agent_client/__init__.py           358)                 loaded into the agent to successfully issue a signature.
src/ssh_agent_client/__init__.py           359)             payload:
src/ssh_agent_client/__init__.py           360)                 A byte string of data to sign.
src/ssh_agent_client/__init__.py           361)             flags:
src/ssh_agent_client/__init__.py           362)                 Optional flags for the signing request.  Currently
src/ssh_agent_client/__init__.py           363)                 passed on as-is to the agent.  In real-world usage, this
src/ssh_agent_client/__init__.py           364)                 could be used, e.g., to request more modern hash
src/ssh_agent_client/__init__.py           365)                 algorithms when signing with RSA keys.  (No such
src/ssh_agent_client/__init__.py           366)                 real-world usage is currently implemented.)
src/ssh_agent_client/__init__.py           367)             check_if_key_loaded:
src/ssh_agent_client/__init__.py           368)                 If true, check beforehand (via `list_keys`) if the
src/ssh_agent_client/__init__.py           369)                 corresponding key has been loaded into the agent.
src/ssh_agent_client/__init__.py           370) 
src/ssh_agent_client/__init__.py           371)         Returns:
src/ssh_agent_client/__init__.py           372)             The binary signature of the payload under the given key.
src/ssh_agent_client/__init__.py           373) 
src/ssh_agent_client/__init__.py           374)         Raises:
src/ssh_agent_client/__init__.py           375)             EOFError:
src/ssh_agent_client/__init__.py           376)                 The response from the SSH agent is truncated or missing.

src/ssh_agent_client/__init__.py           377)             TrailingDataError:

src/ssh_agent_client/__init__.py           378)                 The response from the SSH agent is too long.

src/derivepassphrase/ssh_agent.py          379)             SSHAgentFailedError:

src/ssh_agent_client/__init__.py           380)                 The agent failed to complete the request.

src/ssh_agent_client/__init__.py           381)             KeyError:

src/ssh_agent_client/__init__.py           382)                 `check_if_key_loaded` is true, and the `key` was not
src/ssh_agent_client/__init__.py           383)                 loaded into the agent.
src/ssh_agent_client/__init__.py           384) 
src/ssh_agent_client/__init__.py           385)         """
src/ssh_agent_client/__init__.py           386)         if check_if_key_loaded:
src/ssh_agent_client/__init__.py           387)             loaded_keys = frozenset({pair.key for pair in self.list_keys()})
src/ssh_agent_client/__init__.py           388)             if bytes(key) not in loaded_keys:

src/ssh_agent_client/__init__.py           389)                 msg = 'target SSH key not loaded into agent'
src/ssh_agent_client/__init__.py           390)                 raise KeyError(msg)

src/ssh_agent_client/__init__.py           391)         request_data = bytearray(self.string(key))
src/ssh_agent_client/__init__.py           392)         request_data.extend(self.string(payload))
src/ssh_agent_client/__init__.py           393)         request_data.extend(self.uint32(flags))
src/ssh_agent_client/__init__.py           394)         response_code, response = self.request(

src/derivepassphrase/ssh_agent/__init__.py 395)             _types.SSH_AGENTC.SIGN_REQUEST.value, request_data

src/ssh_agent_client/__init__.py           396)         )

src/derivepassphrase/ssh_agent/__init__.py 397)         if response_code != _types.SSH_AGENT.SIGN_RESPONSE.value:

src/derivepassphrase/ssh_agent.py          398)             raise SSHAgentFailedError(response_code, response)