def generatekey(self):
public_key = StringIO()
private_key = StringIO()
public_key_value = None
private_key_value = None
try:
if self.keytype == 'rsa':
key = RSAKey.generate(self.rsabits)
elif self.keytype == 'ecdsa':
key = ECDSAKey.generate(bits=self.ecdsabits)
elif self.keytype == 'dss':
key = DSSKey.generate(bits=self.dssbits)
else:
return None, "sshkey暂时不支持其它类型 %s" % self.keytype
key.write_private_key(private_key)
public_key.write("%s %s %s" %
(key.get_name(), key.get_base64(), self.basename))
public_key_value = public_key.getvalue()
private_key_value = private_key.getvalue()
cache.set("user_%s_private_key" % self.username, private_key_value)
cache.set("user_%s_public_key" % self.username, public_key_value)
except Exception as e:
logger.error(e.args)
return None, e.args
finally:
public_key.close()
private_key.close()
return {
"publickey": public_key_value,
"privatekey": private_key_value
}, None
def test_format_openssh_pubkey():
rsakey = RSAKey.generate(1024)
assert parse_openssh_pubkey(format_openssh_pubkey(rsakey)) == rsakey
dsskey = DSSKey.generate(1024)
assert parse_openssh_pubkey(format_openssh_pubkey(dsskey)) == dsskey
def test_format_openssh_pubkey():
rsakey = RSAKey.generate(1024)
assert parse_openssh_pubkey(format_openssh_pubkey(rsakey)) == rsakey
dsskey = DSSKey.generate(1024)
assert parse_openssh_pubkey(format_openssh_pubkey(dsskey)) == dsskey
