moep/dynamicweb
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Added trim function to clean public ssh key before validating and saving it

Browse source
This commit is contained in:
PCoder 2017-09-13 06:07:12 +05:30
commit 662219f524
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
hosting/forms.py
View file

@ -94,13 +94,13 @@ class UserHostingKeyForm(forms.ModelForm):
if 'generate' in self.request.POST: if 'generate' in self.request.POST:
return self.data.get('public_key') return self.data.get('public_key')
KEY_ERROR_MESSAGE = _("Please input a proper SSH key") KEY_ERROR_MESSAGE = _("Please input a proper SSH key")
openssh_pubkey_str = self.data.get('public_key') openssh_pubkey_str = self.data.get('public_key').trim()
with tempfile.NamedTemporaryFile(delete=True) as tmp_public_key_file: with tempfile.NamedTemporaryFile(delete=True) as tmp_public_key_file:
tmp_public_key_file.write(openssh_pubkey_str.encode('utf-8')) tmp_public_key_file.write(openssh_pubkey_str.encode('utf-8'))
tmp_public_key_file.flush() tmp_public_key_file.flush()
try: try:
out = subprocess.check_output( subprocess.check_output(
['ssh-keygen', '-lf', tmp_public_key_file.name]) ['ssh-keygen', '-lf', tmp_public_key_file.name])
except subprocess.CalledProcessError as cpe: except subprocess.CalledProcessError as cpe:
logger.debug( logger.debug(