#*****************************************************************************#
#*                                                                           *#
#*  Copyright (c) 2002, Peter Shannon                                        *#
#*  All rights reserved.                                                     *#
#*                                                                           *#
#*  Redistribution and use in source and binary forms, with or without       *#
#*  modification, are permitted provided that the following conditions       *#
#*  are met:                                                                 *#
#*                                                                           *#
#*      * Redistributions of source code must retain the above               *#
#*        copyright notice, this list of conditions and the following        *#
#*        disclaimer.                                                        *#
#*                                                                           *#
#*      * Redistributions in binary form must reproduce the above            *#
#*        copyright notice, this list of conditions and the following        *#
#*        disclaimer in the documentation and/or other materials             *#
#*        provided with the distribution.                                    *#
#*                                                                           *#
#*      * The name of the contributors may be used to endorse or promote     *#
#*        products derived from this software without specific prior         *#
#*        written permission.                                                *#
#*                                                                           *#
#*  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS      *#
#*  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT        *#
#*  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS        *#
#*  FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS   *#
#*  OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,          *#
#*  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT         *#
#*  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,    *#
#*  DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY    *#
#*  THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT      *#
#*  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE    *#
#*  OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.     *#
#*                                                                           *#
#*****************************************************************************#

import types, time, pprint, cStringIO, _der
from _simpledb import OidData as _OidData 
from _der import *

DEBUG = 0

_oidData = _OidData()
obj2oid = _oidData.obj2oid
oid2obj = _oidData.oid2obj

_fragments = []

def _docset():
   return _der._docset() + _fragments

#---------- crypto driver ----------#

class CryptoDriver(object):
   """Dispatcher for crypto calls.

   This module has very minimal dependencies on crypto code, as it's
   almost entirely about ASN.1 encoding and decoding.  Rather than
   wiring in the handful of crypto calls, we dispatch them through
   this driver.  The default driver uses POW, but you can replace it
   with any crypto package you like.

   This is a virtual class.  You will have to subtype it.
   """

   def getOID(self, digestType):
      """Convert a digest identifier into an OID.

      If the identifier we get is a tuple, we assume it's already an
      OID and just return it.  If the identifier is in the driver
      identifier mapping table, we use that to return an OID.
      Otherwise, we try mapping it via the name-to-OID database.
      """
      if isinstance(digestType, tuple):
         return digestType
      if digestType in self.driver2OID:
         return self.driver2OID[digestType]
      return obj2oid(digestType)
         
   def sign(self, key, oid, plaintext):
      """Sign something with an RSA key and a given digest algorithm."""
      raise NotImplementedError

   def verify(self, key, oid, plaintext, signature):
      """Verify a signature."""
      raise NotImplementedError

   def toPublicDER(self, key):
      """Get the DER representation of an RSA key."""
      raise NotImplementedError

   def fromPublicDER(self, der):
      """Set the driver representation of an RSA key from DER."""
      raise NotImplementedError

class POWCryptoDriver(CryptoDriver):
   """Dispatcher for crypto calls using POW package."""

   def __init__(self):
      global POW
      import POW
      self.driver2OID = {
         POW.MD2_DIGEST       :  (1, 2, 840, 113549, 1, 1, 2),    # md2WithRSAEncryption
         POW.MD5_DIGEST       :  (1, 2, 840, 113549, 1, 1, 4),    # md5WithRSAEncryption
         POW.SHA_DIGEST       :  (1, 3, 14, 3, 2, 15),            # shaWithRSAEncryption
         POW.SHA1_DIGEST      :  (1, 2, 840, 113549, 1, 1, 5),    # sha1withRSAEncryption
         POW.RIPEMD160_DIGEST :  (1, 2, 840, 113549, 1, 1, 6),    # ripemd160WithRSAEncryption
         POW.SHA256_DIGEST    :  (1, 2, 840, 113549, 1, 1, 11),   # sha256WithRSAEncryption
         POW.SHA384_DIGEST    :  (1, 2, 840, 113549, 1, 1, 12),   # sha384WithRSAEncryption
         POW.SHA512_DIGEST    :  (1, 2, 840, 113549, 1, 1, 13),   # sha512WithRSAEncryption
         }
      self.OID2driver = dict((v,k) for k,v in self.driver2OID.items())
         
   def _digest(self, oid, plaintext):
      digest = POW.Digest(self.OID2driver[oid])
      digest.update(plaintext)
      return digest.digest()

   def sign(self, key, oid, plaintext):
      return key.sign(self._digest(oid, plaintext), self.OID2driver[oid])

   def verify(self, key, oid, plaintext, signature):
      return key.verify(signature, self._digest(oid, plaintext), self.OID2driver[oid])

   def toPublicDER(self, key):
      return key.derWrite(POW.RSA_PUBLIC_KEY)

   def fromPublicDER(self, der):
      return POW.derRead(POW.RSA_PUBLIC_KEY, der)

_cryptoDriver = None                    # Don't touch this directly

def setCryptoDriver(driver):
   """Set crypto driver.

   The driver should be an instance of CryptoDriver.
   """
   assert isinstance(driver, CryptoDriver)
   global _cryptoDriver
   _cryptoDriver = driver

def getCryptoDriver():
   """Return the currently selected CryptoDriver instance.

   If no driver has been selected, instantiate the default POW driver.
   """
   global _cryptoDriver
   if _cryptoDriver is None:
      setCryptoDriver(POWCryptoDriver())
   return _cryptoDriver

#---------- crypto driver ----------#

def _addFragment(frag):
   global _fragments
   _fragments.append(frag)

_addFragment('''
<modulefunction>
   <header>
      <name>utc2time</name>
      <parameter>time</parameter>
   </header>
   <body>
      <para>
         This is a helper function for turning a UTCTime string into an
         integer.  It isn't built into the encoder since the various
         functions which are used to manipulate the tm structure are
         notoriously unreliable.
      </para>
   </body>
</modulefunction>
''')
def utc2time(val):
   'der encoded value not including tag or length'
   if not isinstance(val, types.StringType):
      raise DerError, 'argument should be a string'
   t = time.strptime(val, '%y%m%d%H%M%SZ')
   return int(time.mktime(t))

_addFragment('''
<modulefunction>
   <header>
      <name>time2utc</name>
      <parameter>time</parameter>
   </header>
   <body>
      <para>
         This is a helper function for turning an integer into a
         UTCTime string.  It isn't built into the encoder since the
         various functions which are used to manipulate the tm structure
         are notoriously unreliable.
      </para>
   </body>
</modulefunction>
''')
def time2utc(val):
   'numerical time value like time_t'
   val = int(val)
   t = time.gmtime(val)
   return time.strftime('%y%m%d%H%M%SZ', t)

_addFragment('''
<modulefunction>
   <header>
      <name>gen2time</name>
      <parameter>time</parameter>
   </header>
   <body>
      <para>
         This is a helper function for turning a GeneralizedTime string into an
         integer.  It isn't built into the encoder since the various
         functions which are used to manipulate the tm structure are
         notoriously unreliable.
      </para>
   </body>
</modulefunction>
''')
def gen2Time(val):
   'der encoded value not including tag or length'
   if not isinstance(val, types.StringType):
      raise DerError, 'argument should be a string'
   t = time.strptime(val, '%Y%m%d%H%M%SZ')
   return int(time.mktime(t))

_addFragment('''
<modulefunction>
   <header>
      <name>time2gen</name>
      <parameter>time</parameter>
   </header>
   <body>
      <para>
         This is a helper function for turning an integer into a
         GeneralizedTime string.  It isn't built into the encoder since the
         various functions which are used to manipulate the tm structure
         are notoriously unreliable.
      </para>
   </body>
</modulefunction>
''')
def time2gen(val):
   'numerical time value like time_t'
   val = int(val)
   t = time.gmtime(val)
   return time.strftime('%Y%m%d%H%M%SZ', t)

_addFragment('''
<method>
   <header>
      <name>ip42oct</name>
      <parameter>ip</parameter>
   </header>
   <body>
      <para>
         <parameter>ip</parameter> should be a list or tuple of integers,
         from 0 to 256.
      </para>
      <example>
         <title>Setting <classname>IpAddress</classname></title>
         <programlisting>
            ip = IpAddress()
            ip.set( ip42oct(192, 168, 0, 231) )
         </programlisting>
      </example>
   </body>
</method>
''')
def ip42oct(val0, val1, val2, val3):
   return chr(val0) + chr(val1) + chr(val2) + chr(val3)

_addFragment('''
<method>
   <header>
      <name>oct2ip4</name>
      <parameter>val</parameter>
   </header>
   <body>
      <para>
         Returns a tuple of 4 integers, from 0 to 256.
      </para>
   </body>
</method>
''')
def oct2ip4(val):
   if not isinstance(val, types.StringType) or len(val) != 4:
      raise DerError, 'parameter should be string of 4 characters'
   return ( ord(val[0]), ord(val[1]), ord(val[2]), ord(val[3]) ) 

#---------- certificate support ----------#
class TbsCertificate(Sequence):
   def __init__(self, optional=0, default=''):

      self.version = Integer()
      self.explicitVersion = Explicit( CLASS_CONTEXT, FORM_CONSTRUCTED, 0, self.version, 0, 'oAMCAQA=\n' )

      self.serial = Integer()
      self.signature = AlgorithmIdentifier()
      self.issuer = Name()
      self.subject = Name()
      self.subjectPublicKeyInfo = SubjectPublicKeyInfo()

      self.validity = Validity()

      self.issuerUniqueID = BitString(1)
      self.issuerUniqueID.implied( CLASS_CONTEXT, FORM_PRIMITIVE, 1 )
      self.subjectUniqueID = BitString(1)
      self.subjectUniqueID.implied( CLASS_CONTEXT, FORM_PRIMITIVE, 2 )

      self.extensions = Extensions()
      self.explicitExtensions = Explicit( CLASS_CONTEXT, FORM_CONSTRUCTED, 3, self.extensions, 1 )

      contents =  [   
                     self.explicitVersion, 
                     self.serial, 
                     self.signature,
                     self.issuer,
                     self.validity,
                     self.subject,
                     self.subjectPublicKeyInfo,
                     self.issuerUniqueID,
                     self.subjectUniqueID,
                     self.explicitExtensions
                  ]

      Sequence.__init__(self, contents, optional, default)

class Validity(Sequence):
   def __init__(self, optional=0, default=''):
      Time = lambda : Choice({ 'generalTime' : GeneralizedTime(), 'utcTime' : UtcTime() })
      self.notBefore = Time()
      self.notAfter = Time()
      contents = [self.notBefore, self.notAfter]
      Sequence.__init__(self, contents, optional, default)

# IA5String should not be allowed in DirectoryString, but old
# implementations (deprecated but not quite outlawed by RFC 3280)
# sometimes use it for EmailAddress attributes in subject names, which
# triggers decode failures here unless we violate RFC 3280 by allowing
# IA5String.  Do not use, do not use, do not use.

class DirectoryString(Choice):
   def __init__(self, optional=0, default=''):
      choices =   {  'teletexString'            :  T61String(),
                     'printableString'          :  PrintableString(),
                     'universalString'          :  UniversalString(),
                     'bmpString'                :  BmpString(),
                     'utf8String'               :  Utf8String(),
                     'ia5String'                :  IA5String()    } 

      Choice.__init__(self, choices, optional, default)

class AttributeTypeAndValue(Sequence):
   def __init__(self, optional=0, default=''):
      self.type = Oid()
      self.dirstr = DirectoryString()
      contents = [ self.type, self.dirstr ] 
      Sequence.__init__(self, contents, optional, default)

class RelativeDistinguishedName(SetOf):
   def __init__(self, optional=0, default=''):
      SetOf.__init__(self, AttributeTypeAndValue, optional, default)

class Name(SequenceOf):
   def __init__(self, optional=0, default=''):
      SequenceOf.__init__(self, RelativeDistinguishedName, optional, default)

class AlgorithmIdentifier(Sequence):
   def __init__(self, optional=0, default=''):
      self.algorithm = Oid()
      self.parameters = Null()
      contents = [self.algorithm, self.parameters]
      Sequence.__init__(self, contents, optional, default)

class SubjectPublicKeyInfo(Sequence):
   def __init__(self, optional=0, default=''):
      self.algorithmId = AlgorithmIdentifier()
      self.subjectPublicKey = AltBitString()
      contents =  [ self.algorithmId, self.subjectPublicKey ]
      Sequence.__init__(self, contents, optional, default)

class Extensions(SequenceOf):
   def __init__(self, optional=0, default=''):
      SequenceOf.__init__(self, Extension, optional, default)

_addFragment('''
<class>
   <header>
      <name>Certificate</name>
      <super>Sequence</super>
   </header>
   <body>
      <example>
         <title>Setting <classname>Certificate</classname></title>
         <programlisting>
      rsa = POW.Asymmetric()
      cert = POW.pkix.Certificate()
      cert.setVersion(1)
      cert.setSerial(5)

      name = ( (( o2i('countryName'), ('printableString', 'GB') ),), 
               (( o2i('stateOrProvinceName'), ('printableString', 'Hertfordshire') ),), 
               (( o2i('organizationName'), ('printableString', 'The House') ),),
               (( o2i('commonName'), ('printableString', 'Client') ),) ) 

      cert.setIssuer(name)
      cert.setSubject(name)

      now = POW.pkix.time2gen( time.time() )
      then = POW.pkix.time2gen(time.time() + 60*60*24*365*12)
      cert.setNotBefore( ('generalTime',  now) )
      cert.setNotAfter( ( 'generalTime',  then) )
      cert.setIssuerUniqueID((1,0,1,0))
      cert.setSubjectUniqueID((1,0,0,1))
      cert.sign(rsa, POW.MD5_DIGEST)
         </programlisting>
      </example>
   </body>
</class>
''')

class Certificate(Sequence):

   _addFragment('''
   <constructor>
      <header>
         <memberof>Certificate</memberof>
         <parameter>optional=0</parameter>
         <parameter>default=''</parameter>
      </header>
   </constructor>
   ''')
   def __init__(self, optional=0, default=''):
      self.tbs = TbsCertificate()
      self.signatureAlgorithm = AlgorithmIdentifier()
      self.signatureValue = AltBitString()
      contents = [ self.tbs, self.signatureAlgorithm, self.signatureValue ] 
      Sequence.__init__(self, contents, optional, default)

   _addFragment('''
   <method>
      <header>
         <memberof>Certificate</memberof>
         <name>setVersion</name>
         <parameter>version</parameter>
      </header>
      <body>
         <para>
            This function sets an <classname>Integer</classname> object.  0
            indicates a version 1 certificate, 1 a version 2 certificate and 2 a
            version 3 certificate.
         </para>
      </body>
   </method>
   ''')
   def setVersion(self, version):
      self.tbs.version.set(version)

   _addFragment('''
   <method>
      <header>
         <memberof>Certificate</memberof>
         <name>getVersion</name>
      </header>
      <body>
         <para>
            This function returns whatever the version object is set to,
            this should be 0, 1 or 2. 
         </para>
      </body>
   </method>
   ''')
   def getVersion(self):
      return self.tbs.version.get()

   _addFragment('''
   <method>
      <header>
         <memberof>Certificate</memberof>
         <name>setSerial</name>
         <parameter>serial</parameter>
      </header>
      <body>
         <para>
            This function sets an <classname>Integer</classname> object.
            No two certificates issued should ever have the same serial
            number.  
         </para>
      </body>
   </method>
   ''')
   def setSerial(self, serial):
      self.tbs.serial.set(serial)

   _addFragment('''
   <method>
      <header>
         <memberof>Certificate</memberof>
         <name>getVersion</name>
      </header>
      <body>
         <para>
            This function returns whatever the serial object is set to.
         </para>
      </body>
   </method>
   ''')
   def getSerial(self):
      return self.tbs.serial.get()

   _addFragment('''
   <method>
      <header>
         <memberof>Certificate</memberof>
         <name>setIssuer</name>
         <parameter>names</parameter>
      </header>
      <body>
         <para>
            This function sets an <classname>Name</classname> object.
            See <classname>Certificate</classname> class for an example.
         </para>
     </body>
   </method>
   ''')
   def setIssuer(self, issuer):
      self.tbs.issuer.set(issuer)

   _addFragment('''
   <method>
      <header>
         <memberof>Certificate</memberof>
         <name>getIssuer</name>
      </header>
      <body>
         <para>
            This function returns a complex tuple containing other tuples.
         </para>
      </body>
   </method>
   ''')
   def getIssuer(self):
      return self.tbs.issuer.get()

   _addFragment('''
   <method>
      <header>
         <memberof>Certificate</memberof>
         <name>setSubject</name>
         <parameter>names</parameter>
      </header>
      <body>
         <para>
            This function sets an <classname>Name</classname> object.
            See <classname>Certificate</classname> class for an example.
         </para>
      </body>
   </method>
   ''')
   def setSubject(self, subject):
      self.tbs.subject.set(subject)

   _addFragment('''
   <method>
      <header>
         <memberof>Certificate</memberof>
         <name>getSubject</name>
      </header>
      <body>
         <para>
            This function returns a complex tuple containing other tuples.
         </para>
      </body>
   </method>
   ''')
   def getSubject(self):
      return self.tbs.subject.get()

   _addFragment('''
   <method>
      <header>
         <memberof>Certificate</memberof>
         <name>setNotBefore</name>
         <parameter>time</parameter>
      </header>
      <body>
         <para>
            This function sets a <classname>Choice</classname> object.
            It can be either a <classname>GeneralTime</classname> or
            <classname>UTCTime</classname> object.  The functions
            <function>gen2time</function>, <function>utc2time</function>, 
            <function>time2gen</function> and <function>time2utc</function>
            can be used to convert to and from integer times and their
            string representation.
         </para>
         <example>
            <title><function>setNotBefore</function> method usage</title>
            <programlisting>
      cert = POW.pkix.Certificate()
      now = POW.pkix.time2gen( time.time() )
      cert.setNotBefore( ('generalTime',  now) )
            </programlisting>
         </example>
      </body>
   </method>
   ''')
   def setNotBefore(self, nb):
      self.tbs.validity.notBefore.set(nb)

   _addFragment('''
   <method>
      <header>
         <memberof>Certificate</memberof>
         <name>getNotBefore</name>
      </header>
      <body>
         <para>
            This function returns a tuple indicating which type of time was
            stored and its value.  See <function>setNotBefore</function> for details.
         </para>
      </body>
   </method>
   ''')
   def getNotBefore(self):
      return self.tbs.validity.notBefore.get()

   _addFragment('''
   <method>
      <header>
         <memberof>Certificate</memberof>
         <name>setNotAfter</name>
         <parameter>time</parameter>
      </header>
      <body>
         <para>
            This function sets a <classname>Choice</classname> object.
            See <function>setNotBefore</function> for details.
         </para>
      </body>
   </method>
   ''')
   def setNotAfter(self, na):
      self.tbs.validity.notAfter.set(na)

   _addFragment('''
   <method>
      <header>
         <memberof>Certificate</memberof>
         <name>getNotAfter</name>
      </header>
      <body>
         <para>
            This function returns a tuple indicating which type of time was
            stored and its value.  See <function>setNotBefore</function> for details.
         </para>
      </body>
   </method>
   ''')
   def getNotAfter(self):
      return self.tbs.validity.notAfter.get()

   _addFragment('''
   <method>
      <header>
         <memberof>Certificate</memberof>
         <name>setIssuerUniqueID</name>
         <parameter>id</parameter>
      </header>
      <body>
         <para>
            This function sets a <classname>BitString</classname> object.
            This is part of the X509v2 standard and is quite poorly
            regarded in general, its use is not recommended.  It is set
            using the normal <classname>BitString</classname> method, that
            is with a sequence of true/false objects.
         </para>
      </body>
   </method>
   ''')
   def setIssuerUniqueID(self, id):
      self.tbs.issuerUniqueID.set(id)

   _addFragment('''
   <method>
      <header>
         <memberof>Certificate</memberof>
         <name>getIssuerUniqueID</name>
      </header>
      <body>
         <para>
            This function returns a tuple of integers, 1 or 0.
         </para>
      </body>
   </method>
   ''')
   def getIssuerUniqueID(self):
      return self.tbs.issuerUniqueID.get()

   _addFragment('''
   <method>
      <header>
         <memberof>Certificate</memberof>
         <name>setSubjectUniqueID</name>
         <parameter>id</parameter>
      </header>
      <body>
         <para>
            This function sets a <classname>BitString</classname> object.
            This is part of the X509v2 standard and is quite poorly
            regarded in general, its use is not recommended.  It is set
            using the normal <classname>BitString</classname> method, that
            is with a sequence of true/false objects.
         </para>
      </body>
   </method>
   ''')
   def setSubjectUniqueID(self, id):
      self.tbs.subjectUniqueID.set(id)

   _addFragment('''
   <method>
      <header>
         <memberof>Certificate</memberof>
         <name>getSubjectUniqueID</name>
      </header>
      <body>
         <para>
            This function returns a tuple of integers, 1 or 0.
         </para>
      </body>
   </method>
   ''')
   def getSubjectUniqueID(self):
      return self.tbs.subjectUniqueID.get()

   _addFragment('''
   <method>
      <header>
         <memberof>Certificate</memberof>
         <name>setExtensions</name>
         <parameter>extns</parameter>
      </header>
      <body>
         <para>
            This method sets an <classname>Extensions</classname> object,
            defined as SEQUENCE OF Extension.  The parameter
            <parameter>extns</parameter> should consist of a list or tuple
            of values suitable to set an extension.  See the extension
            class for details.
         </para>
      </body>
   </method>
   ''')
   def setExtensions(self, extns):
      self.tbs.extensions.set(extns)

   _addFragment('''
   <method>
      <header>
         <memberof>Certificate</memberof>
         <name>getExtensions</name>
      </header>
      <body>
         <para>
            This function returns a tuple of
            <classname>Extension</classname> values.  See
            <classname>Extension</classname> for details.
         </para>
      </body>
   </method>
   ''')
   def getExtensions(self):
      return self.tbs.extensions.get()

   def getExtension(self, oid):
      for x in self.getExtensions():
         if x[0] == oid:
            return x
      return None

   _addFragment('''
   <method>
      <header>
         <memberof>Certificate</memberof>
         <name>sign</name>
         <parameter>rsa</parameter>
         <parameter>digestType</parameter>
      </header>
      <body>
         <para>
            This function updates structured of the
            <classname>Certificate</classname> and
            <constant>tbs</constant> as appropriate and performs the
            specified digest on the <constant>tbs</constant> and set
            <constant>signedText</constant> to signed the digest.  
         </para>
      </body>
   </method>
   ''')
   def sign(self, rsa, digestType):
      driver = getCryptoDriver()
      oid = driver.getOID(digestType)
      self.tbs.signature.set([oid, None])
      signedText = driver.sign(rsa, oid, self.tbs.toString())
      self.signatureAlgorithm.set([oid, None])
      self.signatureValue.set(signedText)

   _addFragment('''
   <method>
      <header>
         <memberof>Certificate</memberof>
         <name>verify</name>
         <parameter>rsa</parameter>
      </header>
      <body>
         <para>
            This function works out what kind of digest was used to
            during signing, calculates the digest of   
            <constant>tbs</constant> and verifies the envelope using the
            key.  
         </para>
      </body>
   </method>
   ''')
   def verify(self, rsa):
      driver = getCryptoDriver()
      oid = self.signatureAlgorithm.get()[0]
      return driver.verify(rsa, oid, self.tbs.toString(), self.signatureValue.get())

#---------- certificate support ----------#
#---------- CRL ----------#

class RevokedCertificate(Sequence):
   def __init__(self, optional=0, default=''):
      self.userCertificate = Integer()
      self.revocationDate = Choice( { 'generalTime' : GeneralizedTime(), 'utcTime' : UtcTime() } )
      self.crlEntryExtensions = Extensions(1)
      contents = [ self.userCertificate, self.revocationDate, self.crlEntryExtensions ] 
      Sequence.__init__(self, contents, optional, default)

class RevokedCertificates(SequenceOf):
   def __init__(self, optional=0, default=''):
      SequenceOf.__init__(self, RevokedCertificate, optional, default)

class TbsCertList(Sequence):
   def __init__(self, optional=0, default=''):
      self.version = Integer(1)
      self.signature = AlgorithmIdentifier()
      self.issuer = Name()
      self.thisUpdate = Choice( { 'generalTime' : GeneralizedTime(), 'utcTime' : UtcTime() } )
      self.nextUpdate = Choice( { 'generalTime' : GeneralizedTime(), 'utcTime' : UtcTime() }, 1 )
      self.revokedCertificates = RevokedCertificates(1)
      self.crlExtensions = Extensions()
      self.explicitCrlExtensions = Explicit( CLASS_CONTEXT, FORM_CONSTRUCTED, 0, self.crlExtensions, 1 )
      contents = [   self.version, 
                     self.signature,
                     self.issuer,
                     self.thisUpdate, 
                     self.nextUpdate,
                     self.revokedCertificates,
                     self.explicitCrlExtensions    ] 
      Sequence.__init__(self, contents, optional, default)

_addFragment('''
<class>
   <header>
      <name>CertificateList</name>
      <super>Sequence</super>
   </header>
   <body>
      <example>
         <title>Setting <classname>CertificateList</classname></title>
         <programlisting>
      now = POW.pkix.time2gen( time.time() )
      then = POW.pkix.time2gen(time.time() + 60*60*24*365*12)
      rsa = POW.Asymmetric()

      crl = POW.pkix.CertificateList()
      crl.setThisUpdate( ('generalTime', now ) )

      name = ( (( o2i('countryName'), ('printableString', 'GB') ),), 
               (( o2i('stateOrProvinceName'), ('printableString', 'Hertfordshire') ),), 
               (( o2i('organizationName'), ('printableString', 'The House') ),),
               (( o2i('commonName'), ('printableString', 'Client') ),) ) 

      myRevocations = (
                        (1, ('generalTime', now), ()),
                        (2, ('generalTime', now), ()),
                        (3, ('generalTime', now), (( o2i('cRLReason'), 0, 1),)) 
                      ) 

      crl.setIssuer(name)
      crl.setRevokedCertificates( myRevocations )

      crl.sign(rsa, POW.MD5_DIGEST)
         </programlisting>
      </example>
   </body>
</class>
''')
class CertificateList(Sequence):
   _addFragment('''
   <constructor>
      <header>
         <memberof>CertificateList</memberof>
         <parameter>optional=0</parameter>
         <parameter>default=''</parameter>
      </header>
   </constructor>
   ''')
   def __init__(self, optional=0, default=''):
      self.tbs = TbsCertList()
      self.signatureAlgorithm = AlgorithmIdentifier()
      self.signature = AltBitString()
      contents = [self.tbs, self.signatureAlgorithm, self.signature]
      Sequence.__init__(self, contents, optional, default)

   _addFragment('''
   <method>
      <header>
         <memberof>CertificateList</memberof>
         <name>setVersion</name>
         <parameter>version</parameter>
      </header>
      <body>
         <para>
            This function sets an <classname>Integer</classname> object.  0
            indicates a version 1 CRL, and 1 a version 2 CRL.
         </para>
      </body>
   </method>
   ''')
   def setVersion(self, version):
      self.tbs.version.set(version)

   _addFragment('''
   <method>
      <header>
         <memberof>CertificateList</memberof>
         <name>getVersion</name>
      </header>
      <body>
         <para>
            This function returns whatever the version object is set to,
            this should be 0, 1 or 2. 
         </para>
      </body>
   </method>
   ''')
   def getVersion(self):
      return self.tbs.version.get()

   _addFragment('''
   <method>
      <header>
         <memberof>CertificateList</memberof>
         <name>setIssuer</name>
         <parameter>names</parameter>
      </header>
      <body>
         <para>
            This function sets an <classname>Name</classname> object.
         </para>
     </body>
   </method>
   ''')
   def setIssuer(self, issuer):
      self.tbs.issuer.set(issuer)

   _addFragment('''
   <method>
      <header>
         <memberof>CertificateList</memberof>
         <name>getIssuer</name>
      </header>
      <body>
         <para>
            This function returns a complex tuple containing other tuples.
         </para>
      </body>
   </method>
   ''')
   def getIssuer(self):
      return self.tbs.issuer.get()

   _addFragment('''
   <method>
      <header>
         <memberof>setThisUpdate</memberof>
         <name>setNotBefore</name>
         <parameter>time</parameter>
      </header>
      <body>
         <para>
            This function sets a <classname>Choice</classname> object.
            It can be either a <classname>GeneralTime</classname> or
            <classname>UTCTime</classname> object.  The functions
            <function>gen2time</function>, <function>utc2time</function>, 
            <function>time2gen</function> and <function>time2utc</function>
            can be used to convert to and from integer times and their
            string representation.
         </para>
         <example>
            <title><function>setNotBefore</function> method usage</title>
            <programlisting>
      crl = POW.pkix.CertificateList()
      now = POW.pkix.time2gen( time.time() )
      crl.setNotBefore( ('generalTime',  now) )
            </programlisting>
         </example>
      </body>
   </method>
   ''')
   def setThisUpdate(self, nu):
      self.tbs.thisUpdate.set(nu)

   _addFragment('''
   <method>
      <header>
         <memberof>CertificateList</memberof>
         <name>getThisUpdate</name>
      </header>
      <body>
         <para>
            This function returns a tuple containing two strings. The first
            is either 'utcTime' or 'generalTime' and the second is the time
            value as a string.
         </para>
      </body>
   </method>
   ''')
   def getThisUpdate(self):
      return self.tbs.thisUpdate.get()

   _addFragment('''
   <method>
      <header>
         <memberof>CertificateList</memberof>
         <name>setNextUpdate</name>
      </header>
      <body>
         <para>
            See set <function>setThisUpdate</function>.
         </para>
      </body>
   </method>
   ''')
   def setNextUpdate(self, nu):
      self.tbs.nextUpdate.set(nu)

   _addFragment('''
   <method>
      <header>
         <memberof>CertificateList</memberof>
         <name>getNextUpdate</name>
      </header>
      <body>
         <para>
            See set <function>getThisUpdate</function>.
         </para>
      </body>
   </method>
   ''')
   def getNextUpdate(self):
      return self.tbs.nextUpdate.get()

   _addFragment('''
   <method>
      <header>
         <memberof>CertificateList</memberof>
         <name>setExtensions</name>
         <parameter>extns</parameter>
      </header>
      <body>
         <para>
            This method sets an <classname>Extensions</classname> object,
            defined as SEQUENCE OF Extension.  The parameter
            <parameter>extns</parameter> should consist of a list or tuple
            of values suitable to set an extension.  See the extension
            class for details.
         </para>
      </body>
   </method>
   ''')
   def setExtensions(self, extns):
      self.tbs.crlExtensions.set(extns)

   _addFragment('''
   <method>
      <header>
         <memberof>CertificateList</memberof>
         <name>getExtensions</name>
      </header>
      <body>
         <para>
            This function returns a tuple of
            <classname>Extension</classname> values.  See
            <classname>Extension</classname> for details.
         </para>
      </body>
   </method>
   ''')
   def getExtensions(self):
      return self.tbs.crlExtensions.get()

   def getExtension(self, oid):
      for x in self.getExtensions():
         if x[0] == oid:
            return x
      return None

   _addFragment('''
   <method>
      <header>
         <memberof>CertificateList</memberof>
         <name>setRevokedCertificates</name>
      </header>
      <body>
         <para>
            This function sets a sequence of
            <classname>revokedCertificate</classname> objects.
            This object is optional.  See
            <classname>CertificateList</classname> for an example of its
            use.
         </para>
      </body>
   </method>
   ''')
   def setRevokedCertificates(self, rc):
      self.tbs.revokedCertificates.set(rc)

   _addFragment('''
   <method>
      <header>
         <memberof>CertificateList</memberof>
         <name>getRevokedCertificates</name>
      </header>
      <body>
         <para>
            This function return a sequence of
            <classname>revokedCertificate</classname> objects or None.
         </para>
      </body>
   </method>
   ''')
   def getRevokedCertificates(self):
      return self.tbs.revokedCertificates.get()

   _addFragment('''
   <method>
      <header>
         <memberof>Certificate</memberof>
         <name>sign</name>
      </header>
      <body>
         <para>
            This function updates structured of the
            <classname>certificateList</classname> and
            <classname>tBSCertList</classname> as appropriate, performs the
            specified digest on the <classname>tBSCertList</classname> and sets
            <constant>signedValue</constant> to signed the digest.  
         </para>
      </body>
   </method>
   ''')
   def sign(self, rsa, digestType):
      driver = getCryptoDriver()
      oid = driver.getOID(digestType)
      self.tbs.signature.set([oid, None])
      signedText = driver.sign(rsa, oid, self.tbs.toString())
      self.signatureAlgorithm.set([oid, None])
      self.signature.set(signedText)

   _addFragment('''
   <method>
      <header>
         <memberof>CertificateList</memberof>
         <name>verify</name>
      </header>
      <body>
         <para>
            This function works out what kind of digest was used to during
            signing, calculates the digest of
            <classname>tBSCertList</classname> and verifies the
            <constant>signedText</constant> using the key.  
         </para>
      </body>
   </method>
   ''')
   def verify(self, rsa):
      driver = getCryptoDriver()
      oid = self.signatureAlgorithm.get()[0]
      return driver.verify(rsa, oid, self.tbs.toString(), self.signature.get())

#---------- CRL ----------#
#---------- PKCS10 ----------#

# My ASN.1-foo (and perhaps this ASN.1 implementation) isn't quite up
# to X.501 or PKCS #10, so this is partly based on a dump of what
# OpenSSL generates, and doesn't handle attributes other than X.509v3
# extensions.

class PKCS10AttributeSet(SetOf):
   def __init__(self, optional=0, default=''):
      SetOf.__init__(self, Extensions, optional, default)

class PKCS10AttributeChoice(Choice):
   def __init__(self, optional=0, default=''):
      choices = { 'single' : Extensions(),
                  'set'    : PKCS10AttributeSet() }
      Choice.__init__(self, choices, optional, default)

class PKCS10Attributes(Sequence):
   def __init__(self, optional=1, default=''):
      self.oid = Oid()
      self.val = PKCS10AttributeChoice()
      contents = [ self.oid, self.val ]
      Sequence.__init__(self, contents, optional, default)

class CertificationRequestInfo(Sequence):
   def __init__(self, optional=0, default=''):
      self.version = Integer()
      self.subject = Name()
      self.subjectPublicKeyInfo = SubjectPublicKeyInfo()
      self.attributes = PKCS10Attributes()
      self.explicitAttributes = Explicit(CLASS_CONTEXT, FORM_CONSTRUCTED, 0, self.attributes)
      contents = [ self.version, self.subject, self.subjectPublicKeyInfo, self.explicitAttributes ]
      Sequence.__init__(self, contents, optional, default)

class CertificationRequest(Sequence):
   def __init__(self, optional=0, default=''):
      self.certificationRequestInfo = CertificationRequestInfo()
      self.signatureAlgorithm = AlgorithmIdentifier()
      self.signatureValue = AltBitString()
      contents = [ self.certificationRequestInfo, self.signatureAlgorithm, self.signatureValue ] 
      Sequence.__init__(self, contents, optional, default)

   def sign(self, rsa, digestType):
      driver = getCryptoDriver()
      oid = driver.getOID(digestType)
      self.certificationRequestInfo.subjectPublicKeyInfo.fromString(driver.toPublicDER(rsa))
      signedText = driver.sign(rsa, oid, self.certificationRequestInfo.toString())
      self.signatureAlgorithm.set([oid, None])
      self.signatureValue.set(signedText)

   def verify(self):
      driver = getCryptoDriver()
      oid = self.signatureAlgorithm.get()[0]
      rsa = driver.fromPublicDER(self.certificationRequestInfo.subjectPublicKeyInfo.toString())
      return driver.verify(rsa, oid, self.certificationRequestInfo.toString(), self.signatureValue.get())

   def getExtensions(self):
      oid = self.certificationRequestInfo.attributes.oid.get()
      if oid is None:
         return ()
      if oid != (1, 2, 840, 113549, 1, 9, 14) or \
         self.certificationRequestInfo.attributes.val.choice != "set" or \
         len(self.certificationRequestInfo.attributes.val.choices["set"]) > 1:
         raise DerError, "failed to understand X.501 Attribute encoding, sorry: %s" % self.get()
      return self.certificationRequestInfo.attributes.val.choices["set"][0].get()

   def getExtension(self, oid):
      for x in self.getExtensions():
         if x[0] == oid:
            return x
      return None

   def setExtensions(self, exts):
      self.certificationRequestInfo.attributes.oid.set((1, 2, 840, 113549, 1, 9, 14))
      self.certificationRequestInfo.attributes.val.set(("set", [exts]))

#---------- PKCS10 ----------#
#---------- GeneralNames object support ----------#
class OtherName(Sequence):
   def __init__(self, optional=0, default=''):
      self.typeId = Oid()
      self.any = Any()
      contents = [self.typeId, self.any]
      Sequence.__init__(self, contents, optional, default)

class EdiPartyName(Sequence):
   def __init__(self, optional=0, default=''):
      self.nameAssigner = DirectoryString()
      self.partyName = DirectoryString()
      self.explicitNameAssigner = Explicit( CLASS_CONTEXT, FORM_CONSTRUCTED, 0, self.nameAssigner, 1 )
      self.explicitPartyName = Explicit( CLASS_CONTEXT, FORM_CONSTRUCTED, 1, self.partyName )
      contents = [ self.explicitNameAssigner, self.explicitPartyName ]
      Sequence.__init__(self, contents, optional, default)

class IpAddress(OctetString):
   pass

class GeneralName(Choice):
   def __init__(self, optional=0, default=''):

      otherName = OtherName()
      otherName.implied( CLASS_CONTEXT, FORM_CONSTRUCTED, 0 )
      rfc822Name = IA5String()
      rfc822Name.implied( CLASS_CONTEXT, FORM_PRIMITIVE, 1 )
      dnsName = IA5String()
      dnsName.implied( CLASS_CONTEXT, FORM_PRIMITIVE, 2 )
      directoryName = Name()
      explicitDirectoryName = Explicit( CLASS_CONTEXT, FORM_CONSTRUCTED, 4, directoryName)
      ediPartyName = EdiPartyName()
      ediPartyName.implied( CLASS_CONTEXT, FORM_CONSTRUCTED, 5 )
      uri = IA5String()
      uri.implied( CLASS_CONTEXT, FORM_PRIMITIVE, 6 )
      ipAddress = IpAddress()
      ipAddress.implied( CLASS_CONTEXT, FORM_PRIMITIVE, 7 )
      registeredId = Oid()
      registeredId.implied( CLASS_CONTEXT, FORM_PRIMITIVE, 8 )

      choices =   {  'otherName'                :  otherName ,
                     'rfc822Name'               :  rfc822Name ,
                     'dNSName'                  :  dnsName ,
                     'directoryName'            :  explicitDirectoryName ,
                     'ediPartyName'             :  ediPartyName ,
                     'uri'                      :  uri ,
                     'iPAddress'                :  ipAddress ,
                     'registeredId'             :  registeredId  }

      Choice.__init__(self, choices, optional, default)

class GeneralNames(SequenceOf):
   def __init__(self, optional=0, default=''):
      SequenceOf.__init__(self, GeneralName, optional, default)

#---------- GeneralNames object support ----------#
#---------- X509v3 extensions ----------#

_addFragment('''
<class>
   <header>
      <name>BasicConstraints</name>
      <super>Sequence</super>
   </header>
   <body>
      <para>
         This little extension has recently caused plenty of problems for
         several large organisations.  It consist of a
         <classname>Boolean</classname> and an
         <classname>Integer</classname>.  The first indicates if the owner
         is a CA, the second indicates how long a chain of CAs you should
         trust which the subject of this certificate trusts.  
      </para>
      <example>
         <title>Setting <classname>BasicConstraints</classname></title>
         <programlisting>
            bc = BasicConstraints()
            bc.set( (1, 1) )
         </programlisting>
      </example>
   </body>
</class>
''')
class BasicConstraints(Sequence):
   _addFragment('''
   <constructor>
      <header>
         <memberof>BasicConstraints</memberof>
         <parameter>optional=0</parameter>
         <parameter>default=''</parameter>
      </header>
   </constructor>
   ''') 
   def __init__(self, optional=0, default=''):
      self.ca = Boolean(0, 'AQEA\n')
      self.pathLenConstraint = Integer(1)
      contents = [self.ca, self.pathLenConstraint]
      Sequence.__init__(self, contents, optional, default)

_addFragment('''
<class>
   <header>
      <name>KeyUsage</name>
      <super>BitString</super>
   </header>
</class>
''')
class KeyUsage(BitString):
   pass

_addFragment('''
<class>
   <header>
      <name>SubjectAltName</name>
      <super>GeneralNames</super>
   </header>
</class>
''')
class SubjectAltName(GeneralNames):
   pass

_addFragment('''
<class>
   <header>
      <name>IssuerAltName</name>
      <super>GeneralNames</super>
   </header>
</class>
''')
class IssuerAltName(GeneralNames):
   pass

_addFragment('''
<class>
   <header>
      <name>SubjectKeyIdentifier</name>
      <super>OctetString</super>
   </header>
</class>
''')
class SubjectKeyIdentifier(OctetString):
   pass

_addFragment('''
<class>
   <header>
      <name>AuthorityKeyIdentifier</name>
      <super>Sequence</super>
   </header>
   <body>
      <para>
      </para>
      <example>
         <title>Setting <classname>AuthorityKeyIdentifier</classname></title>
         <programlisting>
      id = AuthorityKeyIdentifier()
      authdigest = POW.Digest( POW.SHA1_DIGEST )
      authdigest.update(rsa.derWrite(POW.RSA_PUBLIC_KEY))
      keyHash = authdigest.digest()
      id.set( (keyHash, None, None) )
         </programlisting>
      </example>
   </body>

</class>
''')
class AuthorityKeyIdentifier(Sequence):
   _addFragment('''
   <constructor>
      <header>
         <memberof>AuthorityKeyIdentifier</memberof>
         <parameter>optional=0</parameter>
         <parameter>default=''</parameter>
      </header>
   </constructor>
   ''') 
   def __init__(self, optional=0, default=''):
      self.keyIdentifier = OctetString(1)
      self.keyIdentifier.implied( CLASS_CONTEXT, FORM_PRIMITIVE, 0 )
      self.authorityCertIssuer = GeneralNames(1)
      self.authorityCertIssuer.implied( CLASS_CONTEXT, FORM_CONSTRUCTED, 1 )
      self.authorityCertSerialNumber = Integer(1)
      self.authorityCertSerialNumber.implied( CLASS_CONTEXT, FORM_PRIMITIVE, 2 )
      contents = [self.keyIdentifier, self.authorityCertIssuer, self.authorityCertSerialNumber]
      Sequence.__init__(self, contents, optional, default)

_addFragment('''
<class>
   <header>
      <name>PrivateKeyUsagePeriod</name>
      <super>Sequence</super>
   </header>
   <body>
      <example>
         <title>Setting <classname>PrivateKeyUsagePeriod</classname></title>
         <programlisting>
      period = PrivateKeyUsagePeriod()
      period.set( ( time2gen( time.time() ), None) )
         </programlisting>
      </example>
   </body>
</class>
''')
class PrivateKeyUsagePeriod(Sequence):
   _addFragment('''
   <constructor>
      <header>
         <memberof>PrivateKeyUsagePeriod</memberof>
         <parameter>optional=0</parameter>
         <parameter>default=''</parameter>
      </header>
   </constructor>
   ''') 
   def __init__(self, optional=0, default=''):
      self.notBefore = GeneralizedTime()
      self.notBefore.implied( CLASS_CONTEXT, FORM_PRIMITIVE, 0 )
      self.notAfter = GeneralizedTime()
      self.notAfter.implied( CLASS_CONTEXT, FORM_PRIMITIVE, 1 )
      contents =  [self.notBefore, self.notAfter]
      Sequence.__init__(self, contents, optional, default)

class DisplayText(Choice):
   def __init__(self, optional=0, default=''):
      choices =   {  'visibleString'            :  VisibleString(),
                     'bmpString'                :  BmpString(),
                     'utf8String'               :  Utf8String()   } 

      Choice.__init__(self, choices, optional, default)

class NoticeNumbers(SequenceOf):
   def __init__(self, optional=0, default=''):
      SequenceOf.__init__(self, Integer, optional, default)

class NoticeReference(Sequence):
   def __init__(self, optional=0, default=''):
      self.organization = DisplayText()
      self.noticeNumbers = NoticeNumbers()
      contents =  [self.organization, self.noticeNumbers]
      Sequence.__init__(self, contents, optional, default)

class UserNotice(Sequence):
   def __init__(self, optional=0, default=''):
      self.noticeRef = NoticeReference(1)
      self.explicitText = DisplayText(1)
      contents =  [self.noticeRef, self.explicitText]
      Sequence.__init__(self, contents, optional, default)

class Qualifier(Choice):
   def __init__(self, optional=0, default=''):
      choices =   {  'cPSuri'       :  IA5String(),
                     'userNotice'   :  UserNotice()  }

      Choice.__init__(self, choices, optional, default)

class PolicyQualifierInfo(Sequence):
   def __init__(self,  optional=0, default=''):
      self.policyQualifierId = Oid()
      self.qualifier = Qualifier()
      contents =  [self.policyQualifierId, self.qualifier]
      Sequence.__init__(self, contents, optional, default)

class PolicyQualifiers(SequenceOf):
   def __init__(self, optional=0, default=''):
      SequenceOf.__init__(self, PolicyQualifierInfo, optional, default)

class PolicyInformation(Sequence):
   def __init__(self, optional=0, default=''):
      self.policyIdentifier = Oid()
      self.policyQualifiers = PolicyQualifiers(1)
      contents =  [self.policyIdentifier, self.policyQualifiers]
      Sequence.__init__(self, contents, optional, default)

_addFragment('''
<class>
   <header>
      <name>CertificatePolicies</name>
      <super>SequenceOf</super>
   </header>
   <body>
      <example>
         <title>Setting <classname>CertificatePolicies</classname></title>
         <programlisting>
       data = ( 
                  ( o2i('id-cti-ets-proofOfReceipt'), (
                     (o2i('cps'),     ('cPSuri', 'http://www.p-s.org.uk/policies/policy1')), 
                     (o2i('unotice'), (   'userNotice', 
                                          ((('visibleString', 'The House'),(1,2,3)), 
                                             ('visibleString', 'We guarentee nothing')))),
                  )),
                  ( o2i('id-cti-ets-proofOfOrigin'), (
                     (o2i('cps'), ('cPSuri', 'http://www.p-s.org.uk/policies/policy2')), 
                  ))
                )
      policies = CertificatePolicies()
      policies.set( data )
         </programlisting>
      </example>
   </body>
</class>
''')
class CertificatePolicies(SequenceOf):
   _addFragment('''
   <constructor>
      <header>
         <memberof>CertificatePolicies</memberof>
         <parameter>optional=0</parameter>
         <parameter>default=''</parameter>
      </header>
   </constructor>
   ''') 
   def __init__(self, optional=0, default=''):
      SequenceOf.__init__(self, PolicyInformation, optional, default)

class DistributionPointName(Choice):
   def __init__(self, optional=0, default=''):
      fullName = GeneralNames()
      fullName.implied( CLASS_CONTEXT, FORM_CONSTRUCTED, 0 )
      nameRelativeToCRLIssuer = RelativeDistinguishedName()
      nameRelativeToCRLIssuer.implied( CLASS_CONTEXT, FORM_CONSTRUCTED, 1 )
   
      choices =   {  'fullName'                 :  fullName,
                     'nameRelativeToCRLIssuer ' :  nameRelativeToCRLIssuer  }

      Choice.__init__(self, choices, optional, default)

class DistributionPoint(Sequence):
   def __init__(self, optional=0, default=''):
      self.distributionPoint = DistributionPointName(1)
      self.explicitDistributionPoint = Explicit(CLASS_CONTEXT, FORM_CONSTRUCTED, 0, self.distributionPoint)
      self.reasons = BitString(1)
      self.reasons.implied( CLASS_CONTEXT, FORM_PRIMITIVE, 1 )
      self.cRLIssuer = GeneralNames(1)
      self.cRLIssuer.implied( CLASS_CONTEXT, FORM_CONSTRUCTED, 2 )
      contents =  [self.explicitDistributionPoint, self.reasons, self.cRLIssuer]
      Sequence.__init__(self, contents, optional, default)

_addFragment('''
<class>
   <header>
      <name>CRLDistrobutionPoints</name>
      <super>SequenceOf</super>
   </header>
   <body>
      <example>
         <title>Setting <classname>CRLDistrobutionPoints</classname></title>
         <programlisting>
      n1 =  ('directoryName',  
               (  (( o2i('countryName'), ('printableString', 'UK') ),), 
                  (( o2i('stateOrProvinceName'), ('printableString', 'Herts') ),), 
                  (( o2i('organizationName'), ('printableString', 'The House') ),),
                  (( o2i('commonName'), ('printableString', 'Shannon Works') ),) ) ) 

      n2 = ('iPAddress', POW.pkix.ip42oct(192,168,100,51)) 

      data = ( ( ('fullName',(n1, n2)), (1,1,1,1,1), (n1,) ), )
      points = CRLDistrobutionPoints()
      points.set( data )
         </programlisting>
      </example>
   </body>
</class>
''')
class CRLDistributionPoints(SequenceOf):
   _addFragment('''
   <constructor>
      <header>
         <memberof>CRLDistrobutionPoints</memberof>
         <parameter>optional=0</parameter>
         <parameter>default=''</parameter>
      </header>
   </constructor>
   ''') 
   def __init__(self, optional=0, default=''):
      SequenceOf.__init__(self, DistributionPoint, optional, default)

_addFragment('''
<class>
   <header>
      <name>CrlNumber</name>
      <super>Integer</super>
   </header>
</class>
''')
class CrlNumber(Integer):
   pass

_addFragment('''
<class>
   <header>
      <name>DeltaCrlIndicator</name>
      <super>Integer</super>
   </header>
</class>
''')
class DeltaCrlIndicator(Integer):
   pass

_addFragment('''
<class>
   <header>
      <name>InvalidityDate</name>
      <super>GeneralizedTime</super>
   </header>
</class>
''')
class InvalidityDate(GeneralizedTime):
   pass

_addFragment('''
<class>
   <header>
      <name>CrlReason</name>
      <super>Enum</super>
   </header>
</class>
''')
class CrlReason(Enum):
   pass

_addFragment('''
<class>
   <header>
      <name>IPAddressRange</name>
      <super>Sequence</super>
   </header>
</class>
''')
class IPAddressRange(Sequence):
   def __init__(self, optional=0, default=''):
      self.min = BitString()
      self.max = BitString()
      contents = [ self.min, self.max ]
      Sequence.__init__(self, contents, optional, default)

_addFragment('''
<class>
   <header>
      <name>IPAddressOrRange</name>
      <super>Choice</super>
   </header>
</class>
''')
class IPAddressOrRange(Choice):
   def __init__(self, optional=0, default=''):
      choices = { 'addressPrefix' : BitString(),
                  'addressRange'  : IPAddressRange() }
      Choice.__init__(self, choices, optional, default)

_addFragment('''
<class>
   <header>
      <name>IPAddressesOrRanges</name>
      <super>SequenceOf</super>
   </header>
</class>
''')
class IPAddressesOrRanges(SequenceOf):
   def __init__(self, optional=0, default=''):
      SequenceOf.__init__(self, IPAddressOrRange, optional, default)

_addFragment('''
<class>
   <header>
      <name>IPAddressChoice</name>
      <super>Choice</super>
   </header>
</class>
''')
class IPAddressChoice(Choice):
   def __init__(self, optional=0, default=''):
      choices = { 'inherit'             : Null(),
                  'addressesOrRanges'   : IPAddressesOrRanges() }
      Choice.__init__(self, choices, optional, default)

_addFragment('''
<class>
   <header>
      <name>IPAddressFamily</name>
      <super>Sequence</super>
   </header>
</class>
''')
class IPAddressFamily(Sequence):
   def __init__(self, optional=0, default=''):
      self.addressFamily = OctetString()
      self.ipAddressChoice = IPAddressChoice()
      contents = [ self.addressFamily, self.ipAddressChoice ]
      Sequence.__init__(self, contents, optional, default)

_addFragment('''
<class>
   <header>
      <name>IPAddrBlocks</name>
      <super>SequenceOf</super>
   </header>
   <body>
      <para>
         Implementation of RFC 3779 section 2.2.3.
      </para>
   </body>
</class>
''')
class IPAddrBlocks(SequenceOf):
   def __init__(self, optional=0, default=''):
      SequenceOf.__init__(self, IPAddressFamily, optional, default)

_addFragment('''
<class>
   <header>
      <name>ASRange</name>
      <super>Sequence</super>
   </header>
</class>
''')
class ASRange(Sequence):
   def __init__(self, optional=0, default=''):
      self.min = Integer()
      self.max = Integer()
      contents = [ self.min, self.max ]
      Sequence.__init__(self, contents, optional, default)

_addFragment('''
<class>
   <header>
      <name>ASIdOrRange</name>
      <super>Choice</super>
   </header>
</class>
''')
class ASIdOrRange(Choice):
   def __init__(self, optional=0, default=''):
      choices = { 'id'    : Integer(),
                  'range' : ASRange() }
      Choice.__init__(self, choices, optional, default)

_addFragment('''
<class>
   <header>
      <name>ASIdsOrRanges</name>
      <super>SequenceOf</super>
   </header>
</class>
''')
class ASIdsOrRanges(SequenceOf):
   def __init__(self, optional=0, default=''):
      SequenceOf.__init__(self, ASIdOrRange, optional, default)

_addFragment('''
<class>
   <header>
      <name>ASIdentifierChoice</name>
      <super>Choice</super>
   </header>
</class>
''')
class ASIdentifierChoice(Choice):
   def __init__(self, optional=0, default=''):
      choices = { 'inherit'       : Null(),
                  'asIdsOrRanges' : ASIdsOrRanges() }
      Choice.__init__(self, choices, optional, default)

_addFragment('''
<class>
   <header>
      <name>ASIdentifiers</name>
      <super>Sequence</super>
   </header>
   <body>
      <para>
         Implementation of RFC 3779 section 3.2.3.
      </para>
   </body>
</class>
''')
class ASIdentifiers(Sequence):
   def __init__(self, optional=0, default=''):
      #
      # This is what we -should- be doing
      #self.asnum = ASIdentifierChoice()
      #self.rdi   = ASIdentifierChoice()
      #self.explicitAsnum = Explicit(CLASS_CONTEXT, FORM_CONSTRUCTED, 0, self.asnum, 1)
      #self.explictRdi    = Explicit(CLASS_CONTEXT, FORM_CONSTRUCTED, 1, self.rdi,   1)
      #contents = [ self.explicitAsnum, self.explictRdi ]
      #
      # ...but it generates a spurious empty RDI clause, so try this instead
      # since we know that we never use RDI anyway.
      self.asnum = ASIdentifierChoice()
      self.explicitAsnum = Explicit(CLASS_CONTEXT, FORM_CONSTRUCTED, 0, self.asnum, 1)
      contents = [ self.explicitAsnum ]
      #
      Sequence.__init__(self, contents, optional, default)

   def set(self, values):
      assert len(values) == 1 or (len(values) == 2 and values[1] is None)
      Sequence.set(self, (values[0],))

_addFragment('''
<class>
   <header>
      <name>AccessDescription</name>
      <super>Sequence</super>
   </header>
</class>
''')
class AccessDescription(Sequence):
   def __init__(self, optional=0, default=''):
      self.accessMethod = Oid()
      self.accessLocation = GeneralName()
      contents = [ self.accessMethod, self.accessLocation ]
      Sequence.__init__(self, contents, optional, default)

_addFragment('''
<class>
   <header>
      <name>AuthorityInfoAccess</name>
      <super>SequenceOf</super>
   </header>
   <body>
      <para>
         Implementation of RFC 3280 section 4.2.2.1.
      </para>
   </body>
</class>
''')
class AuthorityInfoAccess(SequenceOf):
   def __init__(self, optional=0, default=''):
      SequenceOf.__init__(self, AccessDescription, optional, default)

_addFragment('''
<class>
   <header>
      <name>SubjectInfoAccess</name>
      <super>SequenceOf</super>
   </header>
   <body>
      <para>
         Implementation of RFC 3280 section 4.2.2.2.
      </para>
   </body>
</class>
''')
class SubjectInfoAccess(SequenceOf):
   def __init__(self, optional=0, default=''):
      SequenceOf.__init__(self, AccessDescription, optional, default)

#---------- X509v3 extensions ----------#

_addFragment('''
<class>
   <header>
      <name>Extension</name>
      <super>Sequence</super>
   </header>
   <body>
      <para>
         This class is a useful little object.  It is set by passing three
         values: an oid, an integer(a boolean really) and a value.  The
         boolean indicates if this extension is critical.  The value is
         used to set the extension once it has been created.  The oid
         is used to create the correct object which, to be fully supported it must
         be one of these:
         <simplelist>
            <member><classname>basicConstraints</classname></member>
            <member><classname>subjectAltName</classname></member>
            <member><classname>issuerAltName</classname></member>
            <member><classname>authorityKeyIdentifier</classname></member>
            <member><classname>privateKeyUsagePeriod</classname></member>
            <member><classname>certificatePolicies</classname></member>
            <member><classname>cRLDistributionPoints</classname></member>
            <member><classname>subjectKeyIdentifier</classname></member>
            <member><classname>keyUsage</classname></member>
            <member><classname>crlNumber</classname></member>
            <member><classname>deltaCrlIndicator</classname></member>
            <member><classname>invalidityDate</classname></member>
            <member><classname>crlReason</classname></member>
         </simplelist>
      </para>
      <example>
         <title>Setting <classname>Extension</classname></title>
         <programlisting>
            extn = Extension()
            email = ('rfc822Name', 'peter_shannon@yahoo.com')
            extn.set( (obj2oid('subjectAltName'),1, (email,)) )
         </programlisting>
      </example>
   </body>
</class>
''')
class Extension(Sequence):

   classMap =     {  
                      (2, 5, 29, 19)  :  BasicConstraints,             
                      (2, 5, 29, 17)  :  SubjectAltName,
                      (2, 5, 29, 18)  :  IssuerAltName,
                      (2, 5, 29, 35)  :  AuthorityKeyIdentifier,
                      (2, 5, 29, 16)  :  PrivateKeyUsagePeriod,
                      (2, 5, 29, 32)  :  CertificatePolicies,
                      (2, 5, 29, 31)  :  CRLDistributionPoints,
                      (2, 5, 29, 14)  :  SubjectKeyIdentifier,
                      (2, 5, 29, 15)  :  KeyUsage,
                      (2, 5, 29, 20)  :  CrlNumber,
                      (2, 5, 29, 27)  :  DeltaCrlIndicator,
                      (2, 5, 29, 24)  :  InvalidityDate,
                      (2, 5, 29, 21)  :  CrlReason,
                      (1, 3, 6, 1, 5, 5, 7, 1, 1)  : AuthorityInfoAccess,
                      (1, 3, 6, 1, 5, 5, 7, 1, 7)  : IPAddrBlocks,
                      (1, 3, 6, 1, 5, 5, 7, 1, 8)  : ASIdentifiers,
                      (1, 3, 6, 1, 5, 5, 7, 1, 11) : SubjectInfoAccess,
                  }
#   Missing -- fix later
#                                         extendedKeyUsage  
#                                         privateKeyUsagePeriod 
#                                         policyMappings 
#                                         nameConstraints 
#                                         policyConstraints 
#                                         subjectDirectoryAttributes 
#                                         instructionCode
#                                         issuingDistrobutionPoint

   def __init__(self, optional=0, default=''):
      self.extnID = Oid()
      self.critical = Boolean(0, 'AQEA')
      self.extnValue = OctetString()
      contents = [self.extnID, self.critical, self.extnValue]
      Sequence.__init__(self, contents, optional, default)

   _addFragment('''
   <method>
      <header>
         <memberof>Extension</memberof>
         <name>set</name>
         <parameter>values</parameter>
      </header>
      <body>
         <para>
            <parameter>values</parameter> should be a sequence of three
            values, the oid, critical marker and a value to set the
            extension.  If an unknown oid is passed to this function it
            will raise an exception.  <parameter>critical</parameter> is a
            boolean.  <parameter>value</parameter> will be used to set the
            extension after it has been created.
         </para>
      </body>
   </method>
   ''')
   def set(self, (oid, critical, val) ):
      self.extnID.set( oid )
      self.critical.set( critical )

      extnObj = None
      if self.classMap.has_key(oid):
         extnObj = self.classMap[oid]()
      else:
         if not (isinstance(oid, types.TupleType) or isinstance(oid, types.ListType)):
            raise DerError, 'the oid should be specified as a sequence of integers'
         else:
            raise DerError, 'unknown object extension %s' % oid

      try:
         extnObj.set( val )
         self.extnValue.set( extnObj.toString() )
      except DerError, e:
         raise DerError, 'failed to set %s, with:\n\t%s\nresulting in:\n\t%s' % (oid, val, `e`)

   _addFragment('''
   <method>
      <header>
         <memberof>Extension</memberof>
         <name>get</name>
      </header>
      <body>
         <para>
            There are several ways this function might fail to decode an
            extension.  Firstly if the extension was marked critical but if
            the oid cannot be mapped to a class or If a failure occurs decoding the
            <constant>extnValue</constant>, an exception will be raised.
            If a failure occurred and the extension was not marked critical it
            will return a tuple like this: <constant>(oid, critical,
            ())</constant>.  If no failures occur a tuple will be returned,
            containg the oid, critical and extension values.
         </para>
      </body>
   </method>
   ''')
   def get(self):
      oid = self.extnID.get()
      critical = self.critical.get()

      if self.classMap.has_key(oid):
         extnObj = self.classMap[oid]()
      else:
         if critical:
            raise DerError, 'failed to read critical extension %s' % str(oid)
         else:
            return (oid, critical, ())

      try:
         extnObj = self.classMap[oid]()
         extnObj.fromString(self.extnValue.get())
         value = extnObj.get()
      except:
         if critical:
            raise DerError, 'failed to read critical extension %s' % str(oid)
         else:
            return (oid, critical, ())

      return (oid, critical, value)