Merge pull request #66 from Venafi/ssh_certificates

Fixes to SSH key pair encoding
Venafi · Aug 17, 2021 · b6f2c96 · b6f2c96
2 parents 4aad5e3 + ab58f62
commit b6f2c96
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/vcert/ssh_utils.py b/vcert/ssh_utils.py
@@ -20,7 +20,7 @@
 from cryptography.hazmat.primitives.asymmetric import rsa
 
 
-from errors import ClientBadData
+from .errors import ClientBadData
 
 PATH_SEPARATOR = "\\"
 CA_ROOT_PATH = PATH_SEPARATOR + "VED" + PATH_SEPARATOR + "Certificate Authority" + PATH_SEPARATOR + "SSH" \
@@ -175,7 +175,7 @@ def generate(self, key_size=DEFAULT_SSH_KEY_SIZE, passphrase=None):
         :return:
         """
         if passphrase:
-            encryption = serialization.BestAvailableEncryption(passphrase)
+            encryption = serialization.BestAvailableEncryption(passphrase.encode())
         else:
             encryption = serialization.NoEncryption()
 
@@ -192,8 +192,8 @@ def generate(self, key_size=DEFAULT_SSH_KEY_SIZE, passphrase=None):
             encoding=serialization.Encoding.OpenSSH,
             format=serialization.PublicFormat.OpenSSH
         )
-        self._private_key = private_key
-        self._public_key = public_key
+        self._private_key = private_key.decode()
+        self._public_key = public_key.decode()
 
     def private_key(self):
         return self._private_key