mirror of
https://gitlab2.federez.net/re2o/re2o
synced 2024-12-23 15:33:45 +00:00
Merge branch 'fix_padding_sshfp' into 'dev'
Fix padding sshfp See merge request re2o/re2o!587
This commit is contained in:
commit
643e6f331a
2 changed files with 337 additions and 302 deletions
File diff suppressed because it is too large
Load diff
|
@ -1277,9 +1277,10 @@ class SshFp(RevMixin, AclMixin, models.Model):
|
||||||
|
|
||||||
See RFC: 1 is sha1 , 2 is sha256.
|
See RFC: 1 is sha1 , 2 is sha256.
|
||||||
"""
|
"""
|
||||||
|
pubkey = self.base64_pubkey()
|
||||||
return {
|
return {
|
||||||
"1": hashlib.sha1(base64.b64decode(self.pub_key_entry)).hexdigest(),
|
"1": hashlib.sha1(pubkey).hexdigest(),
|
||||||
"2": hashlib.sha256(base64.b64decode(self.pub_key_entry)).hexdigest(),
|
"2": hashlib.sha256(pubkey).hexdigest(),
|
||||||
}
|
}
|
||||||
|
|
||||||
class Meta:
|
class Meta:
|
||||||
|
@ -1296,6 +1297,31 @@ class SshFp(RevMixin, AclMixin, models.Model):
|
||||||
def can_delete(self, user_request, *args, **kwargs):
|
def can_delete(self, user_request, *args, **kwargs):
|
||||||
return self.machine.can_delete(user_request, *args, **kwargs)
|
return self.machine.can_delete(user_request, *args, **kwargs)
|
||||||
|
|
||||||
|
def base64_pubkey(self):
|
||||||
|
"""Function to decode in base64 the pub key entry
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Base64 decoded value of pub_key_entry
|
||||||
|
|
||||||
|
Because of b64 MUST be divided by 4, we add a "padding" = carracter 3 times.
|
||||||
|
This padding is then ignored if the pubkey is greater than a multiple of 4.
|
||||||
|
More informations on : https://gist.github.com/perrygeo/ee7c65bb1541ff6ac770
|
||||||
|
As said in the thread, this fix is not optimal, however it is very simple as
|
||||||
|
no options on b64decode function exists."""
|
||||||
|
return base64.b64decode(self.pub_key_entry + "===")
|
||||||
|
|
||||||
|
def clean(self, *args, **kwargs):
|
||||||
|
"""Check if the pub_key_entry is a valid base64 entry.
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
ValidationError: the pub key entry is not a valid base64 enty.
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
self.base64_pubkey()
|
||||||
|
except ValueError:
|
||||||
|
raise ValidationError(_("Ssh pub key entry is incorrect base64 entry"))
|
||||||
|
super(SshFp, self).clean(*args, **kwargs)
|
||||||
|
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
return str(self.algo) + " " + str(self.comment)
|
return str(self.algo) + " " + str(self.comment)
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue