|
|
@ -146,157 +146,3 @@ class HostSSHRequest(SignRequest, object): |
|
|
|
return [ |
|
|
|
("Hostname", self.host_name) |
|
|
|
] |
|
|
|
|
|
|
|
class Authority(object): |
|
|
|
ca_type = None |
|
|
|
request_allowed = [] |
|
|
|
|
|
|
|
def __init__(self, ca_id, name, ca_dir): |
|
|
|
self.ca_id = ca_id |
|
|
|
self.name = name |
|
|
|
self.ca_dir = ca_dir |
|
|
|
|
|
|
|
@property |
|
|
|
def path(self): |
|
|
|
return os.path.join(self.ca_dir, self.ca_id) |
|
|
|
|
|
|
|
def generate(self): |
|
|
|
raise NotImplementedError() |
|
|
|
|
|
|
|
def sign(self, request): |
|
|
|
raise NotImplementedError() |
|
|
|
|
|
|
|
def __repr__(self): |
|
|
|
return ( "%s %s" % ( self.__class__.__name__, self.ca_type ) ) |
|
|
|
|
|
|
|
class SSHAuthority(Authority): |
|
|
|
ca_type = 'ssh' |
|
|
|
|
|
|
|
request_allowed = [ UserSSHRequest, HostSSHRequest, ] |
|
|
|
|
|
|
|
key_algorithm = 'ed25519' |
|
|
|
|
|
|
|
user_validity = '+52w' |
|
|
|
host_validity = '+52w' |
|
|
|
|
|
|
|
def generate(self): |
|
|
|
if os.path.exists(self.path): |
|
|
|
raise ValueError("A CA with the same id and type already exists") |
|
|
|
|
|
|
|
subprocess.check_output(['ssh-keygen', |
|
|
|
'-f', self.path, |
|
|
|
'-t', self.key_algorithm, |
|
|
|
'-C', self.name]) |
|
|
|
|
|
|
|
with open(self.path + '.serial', 'w') as stream: |
|
|
|
stream.write(str(0)) |
|
|
|
|
|
|
|
|
|
|
|
def sign(self, request): |
|
|
|
|
|
|
|
assert type(request) in self.request_allowed |
|
|
|
|
|
|
|
pub_key_path = os.path.join(OUTPUT_PATH, request.req_id + '.pub') |
|
|
|
cert_path = os.path.join(OUTPUT_PATH, request.req_id + '-cert.pub') |
|
|
|
|
|
|
|
with open(self.path + '.serial', 'r') as stream: |
|
|
|
next_serial = int(stream.read()) |
|
|
|
with open(self.path + '.serial', 'w') as stream: |
|
|
|
stream.write(str(next_serial + 1)) |
|
|
|
|
|
|
|
with open(pub_key_path, 'w') as stream: |
|
|
|
stream.write(request.key_data) |
|
|
|
|
|
|
|
ca_private_key = self.path |
|
|
|
|
|
|
|
if type(request) == UserSSHRequest: |
|
|
|
login_names = [ request.user_name, ] |
|
|
|
if request.root_requested: |
|
|
|
login_names.append('root') |
|
|
|
|
|
|
|
subprocess.check_output(['ssh-keygen', |
|
|
|
'-s', ca_private_key, |
|
|
|
'-I', 'user_%s' % request.user_name, |
|
|
|
'-n', ','.join(login_names), |
|
|
|
'-V', self.user_validity, |
|
|
|
'-z', str(next_serial), |
|
|
|
pub_key_path]) |
|
|
|
elif type(request) == HostSSHRequest: |
|
|
|
subprocess.check_output(['ssh-keygen', |
|
|
|
'-s', ca_private_key, |
|
|
|
'-I', 'host_%s' % request.host_name.replace('.', '_'), |
|
|
|
'-h', |
|
|
|
'-n', request.host_name, |
|
|
|
'-V', self.host_validity, |
|
|
|
'-z', str(next_serial), |
|
|
|
pub_key_path]) |
|
|
|
|
|
|
|
return cert_path |
|
|
|
|
|
|
|
class SSLAuthority(Authority): |
|
|
|
ca_type = 'ssl' |
|
|
|
request_allowed = [ HostSSLRequest, ] |
|
|
|
|
|
|
|
ca_key_algorithm = 'des3' |
|
|
|
key_length = '4096' |
|
|
|
|
|
|
|
key_algorithm = 'sha256' |
|
|
|
ca_validity = '365' |
|
|
|
cert_validity = '365' |
|
|
|
|
|
|
|
def generate(self): |
|
|
|
if os.path.exists(self.path): |
|
|
|
raise ValueError("A CA with the same id and type already exists") |
|
|
|
|
|
|
|
subprocess.check_output(['openssl', |
|
|
|
'genrsa', |
|
|
|
'-%s'%self.ca_key_algorithm, |
|
|
|
'-out', '%s'%(self.path), |
|
|
|
self.key_length]) |
|
|
|
|
|
|
|
subprocess.check_output(['openssl', |
|
|
|
'req', |
|
|
|
'-new', |
|
|
|
'-x509', |
|
|
|
'-days', self.ca_validity, |
|
|
|
'-key', self.path, |
|
|
|
# '-extensions', 'v3_ca' |
|
|
|
'-out', "%s.pub"%self.path, |
|
|
|
# '-config', "%s.conf"%self.path |
|
|
|
]) |
|
|
|
|
|
|
|
with open(self.path + '.serial', 'w') as stream: |
|
|
|
stream.write(str(0)) |
|
|
|
|
|
|
|
|
|
|
|
def sign(self, request): |
|
|
|
OUTPUT_PATH |
|
|
|
|
|
|
|
assert type(request) in [HostSSLRequest] |
|
|
|
|
|
|
|
pub_key_path = os.path.join(OUTPUT_PATH, request.req_id + '.pub') |
|
|
|
cert_path = os.path.join(OUTPUT_PATH, request.req_id + '-cert.pub') |
|
|
|
|
|
|
|
with open(self.path + '.serial', 'r') as stream: |
|
|
|
next_serial = int(stream.read()) |
|
|
|
with open(self.path + '.serial', 'w') as stream: |
|
|
|
stream.write(str(next_serial + 1)) |
|
|
|
|
|
|
|
with open(pub_key_path, 'w') as stream: |
|
|
|
stream.write(request.key_data) |
|
|
|
|
|
|
|
ca_private_key = self.path |
|
|
|
|
|
|
|
# openssl x509 -req -days 360 -in server.csr -CA ca.crt -CAkey ca.key -CAcreateserial -out server.crt # print() |
|
|
|
subprocess.check_output(['openssl', |
|
|
|
'x509', |
|
|
|
'-req', |
|
|
|
'-days', self.ca_validity, |
|
|
|
'-in', pub_key_path, |
|
|
|
'-CA', "%s.pub"%self.path, |
|
|
|
'-CAkey', self.path, |
|
|
|
'-CAcreateserial', |
|
|
|
'-out', cert_path, |
|
|
|
'-%s'%self.key_algorithm]) |
|
|
|
|
|
|
|
return cert_path |