2 # SFA uses two crypto libraries: pyOpenSSL and M2Crypto to implement
3 # the necessary crypto functionality. Ideally just one of these libraries
4 # would be used, but unfortunately each of these libraries is independently
5 # lacking. The pyOpenSSL library is missing many necessary functions, and
6 # the M2Crypto library has crashed inside of some of the functions. The
7 # design decision is to use pyOpenSSL whenever possible as it seems more
8 # stable, and only use M2Crypto for those functions that are not possible
11 # This module exports two classes: Keypair and Certificate.
22 from OpenSSL import crypto
24 from M2Crypto import X509
25 from M2Crypto import EVP
26 from random import randint
28 from sfa.util.faults import *
30 def convert_public_key(key):
31 keyconvert_path = "/usr/bin/keyconvert"
32 if not os.path.isfile(keyconvert_path):
33 raise IOError, "Could not find keyconvert in %s" % keyconvert_path
35 # we can only convert rsa keys
39 (ssh_f, ssh_fn) = tempfile.mkstemp()
40 ssl_fn = tempfile.mktemp()
44 cmd = keyconvert_path + " " + ssh_fn + " " + ssl_fn
47 # this check leaves the temporary file containing the public key so
48 # that it can be expected to see why it failed.
49 # TODO: for production, cleanup the temporary files
50 if not os.path.exists(ssl_fn):
55 k.load_pubkey_from_file(ssl_fn)
60 # remove the temporary files
67 # Public-private key pairs are implemented by the Keypair class.
68 # A Keypair object may represent both a public and private key pair, or it
69 # may represent only a public key (this usage is consistent with OpenSSL).
72 key = None # public/private keypair
73 m2key = None # public key (m2crypto format)
76 # Creates a Keypair object
77 # @param create If create==True, creates a new public/private key and
78 # stores it in the object
79 # @param string If string!=None, load the keypair from the string (PEM)
80 # @param filename If filename!=None, load the keypair from the file
82 def __init__(self, create=False, string=None, filename=None):
86 self.load_from_string(string)
88 self.load_from_file(filename)
91 # Create a RSA public/private key pair and store it inside the keypair object
94 self.key = crypto.PKey()
95 self.key.generate_key(crypto.TYPE_RSA, 1024)
98 # Save the private key to a file
99 # @param filename name of file to store the keypair in
101 def save_to_file(self, filename):
102 open(filename, 'w').write(self.as_pem())
105 # Load the private key from a file. Implicity the private key includes the public key.
107 def load_from_file(self, filename):
108 buffer = open(filename, 'r').read()
109 self.load_from_string(buffer)
112 # Load the private key from a string. Implicitly the private key includes the public key.
114 def load_from_string(self, string):
115 self.key = crypto.load_privatekey(crypto.FILETYPE_PEM, string)
116 self.m2key = M2Crypto.EVP.load_key_string(string)
119 # Load the public key from a string. No private key is loaded.
121 def load_pubkey_from_file(self, filename):
122 # load the m2 public key
123 m2rsakey = M2Crypto.RSA.load_pub_key(filename)
124 self.m2key = M2Crypto.EVP.PKey()
125 self.m2key.assign_rsa(m2rsakey)
127 # create an m2 x509 cert
128 m2name = M2Crypto.X509.X509_Name()
129 m2name.add_entry_by_txt(field="CN", type=0x1001, entry="junk", len=-1, loc=-1, set=0)
130 m2x509 = M2Crypto.X509.X509()
131 m2x509.set_pubkey(self.m2key)
132 m2x509.set_serial_number(0)
133 m2x509.set_issuer_name(m2name)
134 m2x509.set_subject_name(m2name)
135 ASN1 = M2Crypto.ASN1.ASN1_UTCTIME()
137 m2x509.set_not_before(ASN1)
138 m2x509.set_not_after(ASN1)
139 junk_key = Keypair(create=True)
140 m2x509.sign(pkey=junk_key.get_m2_pkey(), md="sha1")
142 # convert the m2 x509 cert to a pyopenssl x509
143 m2pem = m2x509.as_pem()
144 pyx509 = crypto.load_certificate(crypto.FILETYPE_PEM, m2pem)
146 # get the pyopenssl pkey from the pyopenssl x509
147 self.key = pyx509.get_pubkey()
150 # Load the public key from a string. No private key is loaded.
152 def load_pubkey_from_string(self, string):
153 (f, fn) = tempfile.mkstemp()
156 self.load_pubkey_from_file(fn)
160 # Return the private key in PEM format.
163 return crypto.dump_privatekey(crypto.FILETYPE_PEM, self.key)
166 # Return an M2Crypto key object
168 def get_m2_pkey(self):
170 self.m2key = M2Crypto.EVP.load_key_string(self.as_pem())
174 # Returns a string containing the public key represented by this object.
176 def get_pubkey_string(self):
177 m2pkey = self.get_m2_pkey()
178 return base64.b64encode(m2pkey.as_der())
181 # Return an OpenSSL pkey object
183 def get_openssl_pkey(self):
187 # Given another Keypair object, return TRUE if the two keys are the same.
189 def is_same(self, pkey):
190 return self.as_pem() == pkey.as_pem()
192 def sign_string(self, data):
193 k = self.get_m2_pkey()
196 return base64.b64encode(k.sign_final())
198 def verify_string(self, data, sig):
199 k = self.get_m2_pkey()
201 k.verify_update(data)
202 return M2Crypto.m2.verify_final(k.ctx, base64.b64decode(sig), k.pkey)
204 def compute_hash(self, value):
205 return self.sign_string(str(value))
208 # The certificate class implements a general purpose X509 certificate, making
209 # use of the appropriate pyOpenSSL or M2Crypto abstractions. It also adds
210 # several addition features, such as the ability to maintain a chain of
211 # parent certificates, and storage of application-specific data.
213 # Certificates include the ability to maintain a chain of parents. Each
214 # certificate includes a pointer to it's parent certificate. When loaded
215 # from a file or a string, the parent chain will be automatically loaded.
216 # When saving a certificate to a file or a string, the caller can choose
217 # whether to save the parent certificates as well.
227 separator="-----parent-----"
230 # Create a certificate object.
232 # @param create If create==True, then also create a blank X509 certificate.
233 # @param subject If subject!=None, then create a blank certificate and set
235 # @param string If string!=None, load the certficate from the string.
236 # @param filename If filename!=None, load the certficiate from the file.
238 def __init__(self, create=False, subject=None, string=None, filename=None):
240 if create or subject:
243 self.set_subject(subject)
245 self.load_from_string(string)
247 self.load_from_file(filename)
250 # Create a blank X509 certificate and store it in this object.
253 self.cert = crypto.X509()
254 self.cert.set_serial_number(1)
255 self.cert.gmtime_adj_notBefore(0)
256 self.cert.gmtime_adj_notAfter(60*60*24*365*5) # five years
259 # Given a pyOpenSSL X509 object, store that object inside of this
260 # certificate object.
262 def load_from_pyopenssl_x509(self, x509):
266 # Load the certificate from a string
268 def load_from_string(self, string):
269 # if it is a chain of multiple certs, then split off the first one and
270 # load it (support for the ---parent--- tag as well as normal chained certs)
272 string = string.strip()
275 if not string.startswith('-----'):
276 string = '-----BEGIN CERTIFICATE-----\n%s\n-----END CERTIFICATE-----' % string
280 if string.count('-----BEGIN CERTIFICATE-----') > 1 and \
281 string.count(Certificate.separator) == 0:
282 parts = string.split('-----END CERTIFICATE-----',1)
283 parts[0] += '-----END CERTIFICATE-----'
285 parts = string.split(Certificate.separator, 1)
287 self.cert = crypto.load_certificate(crypto.FILETYPE_PEM, parts[0])
289 # if there are more certs, then create a parent and let the parent load
290 # itself from the remainder of the string
291 if len(parts) > 1 and parts[1] != '':
292 self.parent = self.__class__()
293 self.parent.load_from_string(parts[1])
296 # Load the certificate from a file
298 def load_from_file(self, filename):
299 file = open(filename)
301 self.load_from_string(string)
304 # Save the certificate to a string.
306 # @param save_parents If save_parents==True, then also save the parent certificates.
308 def save_to_string(self, save_parents=True):
309 string = crypto.dump_certificate(crypto.FILETYPE_PEM, self.cert)
310 if save_parents and self.parent:
311 string = string + self.parent.save_to_string(save_parents)
315 # Save the certificate to a file.
316 # @param save_parents If save_parents==True, then also save the parent certificates.
318 def save_to_file(self, filename, save_parents=True):
319 string = self.save_to_string(save_parents=save_parents)
320 open(filename, 'w').write(string)
322 # Save the certificate to a random file in /tmp/
323 # @param save_parents If save_parents==True, then also save the parent certificates.
324 def save_to_random_tmp_file(self, save_parents=True):
326 filename = "/tmp/cred_%d" % randint(0,999999999)
327 if not os.path.isfile(filename):
329 self.save_to_file(filename, save_parents)
332 # Sets the issuer private key and name
333 # @param key Keypair object containing the private key of the issuer
334 # @param subject String containing the name of the issuer
335 # @param cert (optional) Certificate object containing the name of the issuer
337 def set_issuer(self, key, subject=None, cert=None):
340 # it's a mistake to use subject and cert params at the same time
342 if isinstance(subject, dict) or isinstance(subject, str):
343 req = crypto.X509Req()
344 reqSubject = req.get_subject()
345 if (isinstance(subject, dict)):
346 for key in reqSubject.keys():
347 setattr(reqSubject, key, name[key])
349 setattr(reqSubject, "CN", subject)
351 # subject is not valid once req is out of scope, so save req
354 # if a cert was supplied, then get the subject from the cert
355 subject = cert.cert.get_issuer()
357 self.issuerSubject = subject
360 # Get the issuer name
362 def get_issuer(self, which="CN"):
363 x = self.cert.get_issuer()
364 return getattr(x, which)
367 # Set the subject name of the certificate
369 def set_subject(self, name):
370 req = crypto.X509Req()
371 subj = req.get_subject()
372 if (isinstance(name, dict)):
373 for key in name.keys():
374 setattr(subj, key, name[key])
376 setattr(subj, "CN", name)
377 self.cert.set_subject(subj)
379 # Get the subject name of the certificate
381 def get_subject(self, which="CN"):
382 x = self.cert.get_subject()
383 return getattr(x, which)
386 # Get the public key of the certificate.
388 # @param key Keypair object containing the public key
390 def set_pubkey(self, key):
391 assert(isinstance(key, Keypair))
392 self.cert.set_pubkey(key.get_openssl_pkey())
395 # Get the public key of the certificate.
396 # It is returned in the form of a Keypair object.
398 def get_pubkey(self):
399 m2x509 = X509.load_cert_string(self.save_to_string())
401 pkey.key = self.cert.get_pubkey()
402 pkey.m2key = m2x509.get_pubkey()
406 # Add an X509 extension to the certificate. Add_extension can only be called
407 # once for a particular extension name, due to limitations in the underlying
410 # @param name string containing name of extension
411 # @param value string containing value of the extension
413 def add_extension(self, name, critical, value):
414 ext = crypto.X509Extension (name, critical, value)
415 self.cert.add_extensions([ext])
418 # Get an X509 extension from the certificate
420 def get_extension(self, name):
421 # pyOpenSSL does not have a way to get extensions
422 m2x509 = X509.load_cert_string(self.save_to_string())
423 value = m2x509.get_ext(name).get_value()
427 # Set_data is a wrapper around add_extension. It stores the parameter str in
428 # the X509 subject_alt_name extension. Set_data can only be called once, due
429 # to limitations in the underlying library.
431 def set_data(self, str, field='subjectAltName'):
432 # pyOpenSSL only allows us to add extensions, so if we try to set the
433 # same extension more than once, it will not work
434 if self.data.has_key(field):
435 raise "cannot set ", field, " more than once"
436 self.data[field] = str
437 self.add_extension(field, 0, str)
440 # Return the data string that was previously set with set_data
442 def get_data(self, field='subjectAltName'):
443 if self.data.has_key(field):
444 return self.data[field]
447 uri = self.get_extension(field)
448 self.data[field] = uri
452 return self.data[field]
455 # Sign the certificate using the issuer private key and issuer subject previous set with set_issuer().
458 assert self.cert != None
459 assert self.issuerSubject != None
460 assert self.issuerKey != None
461 self.cert.set_issuer(self.issuerSubject)
462 self.cert.sign(self.issuerKey.get_openssl_pkey(), self.digest)
465 # Verify the authenticity of a certificate.
466 # @param pkey is a Keypair object representing a public key. If Pkey
467 # did not sign the certificate, then an exception will be thrown.
469 def verify(self, pkey):
470 # pyOpenSSL does not have a way to verify signatures
471 m2x509 = X509.load_cert_string(self.save_to_string())
472 m2pkey = pkey.get_m2_pkey()
474 return m2x509.verify(m2pkey)
476 # XXX alternatively, if openssl has been patched, do the much simpler:
478 # self.cert.verify(pkey.get_openssl_key())
484 # Return True if pkey is identical to the public key that is contained in the certificate.
485 # @param pkey Keypair object
487 def is_pubkey(self, pkey):
488 return self.get_pubkey().is_same(pkey)
491 # Given a certificate cert, verify that this certificate was signed by the
492 # public key contained in cert. Throw an exception otherwise.
494 # @param cert certificate object
496 def is_signed_by_cert(self, cert):
497 k = cert.get_pubkey()
498 result = self.verify(k)
502 # Set the parent certficiate.
504 # @param p certificate object.
506 def set_parent(self, p):
510 # Return the certificate object of the parent of this certificate.
512 def get_parent(self):
516 # Verification examines a chain of certificates to ensure that each parent
517 # signs the child, and that some certificate in the chain is signed by a
518 # trusted certificate.
520 # Verification is a basic recursion: <pre>
521 # if this_certificate was signed by trusted_certs:
524 # return verify_chain(parent, trusted_certs)
527 # At each recursion, the parent is tested to ensure that it did sign the
528 # child. If a parent did not sign a child, then an exception is thrown. If
529 # the bottom of the recursion is reached and the certificate does not match
530 # a trusted root, then an exception is thrown.
532 # @param Trusted_certs is a list of certificates that are trusted.
535 def verify_chain(self, trusted_certs = None):
536 # Verify a chain of certificates. Each certificate must be signed by
537 # the public key contained in it's parent. The chain is recursed
538 # until a certificate is found that is signed by a trusted root.
540 # TODO: verify expiration time
541 #print "====Verify Chain====="
542 # if this cert is signed by a trusted_cert, then we are set
543 for trusted_cert in trusted_certs:
544 #print "***************"
545 # TODO: verify expiration of trusted_cert ?
546 #print "CLIENT CERT", self.dump()
547 #print "TRUSTED CERT", trusted_cert.dump()
548 #print "Client is signed by Trusted?", self.is_signed_by_cert(trusted_cert)
549 if self.is_signed_by_cert(trusted_cert):
550 #print self.get_subject(), "is signed by a root"
553 # if there is no parent, then no way to verify the chain
555 #print self.get_subject(), "has no parent"
556 raise CertMissingParent(self.get_subject())
558 # if it wasn't signed by the parent...
559 if not self.is_signed_by_cert(self.parent):
560 #print self.get_subject(), "is not signed by parent"
561 return CertNotSignedByParent(self.get_subject())
563 # if the parent isn't verified...
564 self.parent.verify_chain(trusted_certs)