ipsilon/tools/certs.py

   1 #!/usr/bin/python
   2 #
   3 # Copyright (C) 2014  Simo Sorce <simo@redhat.com>
   4 #
   5 # see file 'COPYING' for use and warranty information
   6 #
   7 # This program is free software; you can redistribute it and/or modify
   8 # it under the terms of the GNU General Public License as published by
   9 # the Free Software Foundation, either version 3 of the License, or
  10 # (at your option) any later version.
  11 #
  12 # This program is distributed in the hope that it will be useful,
  13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 # GNU General Public License for more details.
  16 #
  17 # You should have received a copy of the GNU General Public License
  18 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19
  20 from subprocess import Popen
  21 import os
  22 import string
  23
  24
  25 class Certificate(object):
  26
  27     def __init__(self, path=None):
  28         self.subject = None
  29         self.key = None
  30         self.cert = None
  31         if path:
  32             self.path = path
  33         else:
  34             self.path = os.getcwd()
  35
  36     def generate(self, prefix, subject):
  37         self.key = os.path.join(self.path, '%s.key' % prefix)
  38         self.cert = os.path.join(self.path, '%s.pem' % prefix)
  39         self.subject = '/CN=%s' % subject
  40         command = ['openssl',
  41                    'req', '-x509', '-batch', '-days', '1825',
  42                    '-newkey', 'rsa:2048', '-nodes', '-subj', self.subject,
  43                    '-keyout', self.key, '-out', self.cert]
  44         proc = Popen(command)
  45         proc.wait()
  46
  47     def get_cert(self):
  48         if not self.cert:
  49             raise ValueError('Certificate unavailable')
  50         with open(self.cert, 'r') as f:
  51             cert = f.readlines()
  52
  53         #poor man stripping of BEGIN/END lines
  54         if cert[0] == '-----BEGIN CERTIFICATE-----\n':
  55             cert = cert[1:]
  56         if cert[-1] == '-----END CERTIFICATE-----\n':
  57             cert = cert[:-1]
  58
  59         return string.join(cert)