summaryrefslogtreecommitdiff
path: root/paramiko/kex_curve25519.py
diff options
context:
space:
mode:
authorAlex Gaynor <alex.gaynor@gmail.com>2019-02-09 17:10:13 +0000
committerAlex Gaynor <alex.gaynor@gmail.com>2019-02-09 17:10:13 +0000
commit180418b170f608007980ab149e29c76a4a26b029 (patch)
tree6eab60ee570c6370e3d756c37630008d50151388 /paramiko/kex_curve25519.py
parentb2ab3c607b1bd8e47ef717b81fe220db49962ac7 (diff)
downloadparamiko-180418b170f608007980ab149e29c76a4a26b029.tar.gz
blacken
Diffstat (limited to 'paramiko/kex_curve25519.py')
-rw-r--r--paramiko/kex_curve25519.py22
1 files changed, 14 insertions, 8 deletions
diff --git a/paramiko/kex_curve25519.py b/paramiko/kex_curve25519.py
index 4b9f6b00..85448fb6 100644
--- a/paramiko/kex_curve25519.py
+++ b/paramiko/kex_curve25519.py
@@ -4,7 +4,8 @@ import hashlib
from cryptography.exceptions import UnsupportedAlgorithm
from cryptography.hazmat.primitives import constant_time, serialization
from cryptography.hazmat.primitives.asymmetric.x25519 import (
- X25519PrivateKey, X25519PublicKey
+ X25519PrivateKey,
+ X25519PublicKey,
)
from paramiko.message import Message
@@ -15,6 +16,7 @@ from paramiko.ssh_exception import SSHException
_MSG_KEXECDH_INIT, _MSG_KEXECDH_REPLY = range(30, 32)
c_MSG_KEXECDH_INIT, c_MSG_KEXECDH_REPLY = [byte_chr(c) for c in range(30, 32)]
+
class KexCurve25519(object):
def __init__(self, transport):
self.transport = transport
@@ -45,9 +47,11 @@ class KexCurve25519(object):
m = Message()
m.add_byte(c_MSG_KEXECDH_INIT)
- m.add_string(self.key.public_key().public_bytes(
- serialization.Encoding.Raw, serialization.PublicFormat.Raw
- ))
+ m.add_string(
+ self.key.public_key().public_bytes(
+ serialization.Encoding.Raw, serialization.PublicFormat.Raw
+ )
+ )
self.transport._send_message(m)
self.transport._expect_packet(_MSG_KEXECDH_REPLY)
@@ -75,7 +79,7 @@ class KexCurve25519(object):
)
server_key_bytes = self.transport.get_server_key().asbytes()
exchange_key_bytes = self.key.public_key().public_bytes(
- serialization.Encoding.Raw, serialization.PublicFormat.Raw,
+ serialization.Encoding.Raw, serialization.PublicFormat.Raw
)
hm.add_string(server_key_bytes)
hm.add_string(peer_key_bytes)
@@ -111,9 +115,11 @@ class KexCurve25519(object):
self.transport.remote_kex_init,
)
hm.add_string(peer_host_key_bytes)
- hm.add_string(self.key.public_key().public_bytes(
- serialization.Encoding.Raw, serialization.PublicFormat.Raw
- ))
+ hm.add_string(
+ self.key.public_key().public_bytes(
+ serialization.Encoding.Raw, serialization.PublicFormat.Raw
+ )
+ )
hm.add_string(peer_key_bytes)
hm.add_mpint(K)
self.transport._set_K_H(K, hashlib.sha256(hm.asbytes()).digest())