From 14e91f8ed283b16a2e32d8ca223dd39aebe8047e Mon Sep 17 00:00:00 2001 From: Christophe Mehay Date: Mon, 29 Apr 2019 09:14:51 +0200 Subject: [PATCH] Better error on v2 load bad key --- pytor/onion.py | 4 ++++ setup.py | 2 +- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/pytor/onion.py b/pytor/onion.py index 5a45a75..57c171d 100644 --- a/pytor/onion.py +++ b/pytor/onion.py @@ -180,6 +180,10 @@ class OnionV2(Onion): def set_private_key(self, key: bytes) -> None: 'Add private key' + if not key.startswith(b'-----BEGIN RSA PRIVATE KEY-----'): + raise Exception( + 'Private key does not seems to be a valid RSA PEM key' + ) self._save_keypair(RSA.importKey(key.strip())) def _get_private_key_has_native(self) -> bytes: diff --git a/setup.py b/setup.py index c2e5cec..9c27d12 100644 --- a/setup.py +++ b/setup.py @@ -6,7 +6,7 @@ from setuptools import setup setup( name='pytor', - version='0.1.3', + version='0.1.4', packages=find_packages(),