165 lines
6.4 KiB
Python
165 lines
6.4 KiB
Python
|
from pyasn1.type import univ, constraint, char, namedtype, tag
|
||
|
from pyasn1.codec.der.decoder import decode
|
||
|
from common.helpers import prep, InvalidFormat
|
||
|
|
||
|
MAX = 64
|
||
|
oid_xmppaddr = '(1, 3, 6, 1, 5, 5, 7, 8, 5)'
|
||
|
oid_dnssrv = '(1, 3, 6, 1, 5, 5, 7, 8, 7)'
|
||
|
|
||
|
|
||
|
|
||
|
class DirectoryString(univ.Choice):
|
||
|
componentType = namedtype.NamedTypes(
|
||
|
namedtype.NamedType(
|
||
|
'teletexString', char.TeletexString().subtype(
|
||
|
subtypeSpec=constraint.ValueSizeConstraint(1, MAX))),
|
||
|
namedtype.NamedType(
|
||
|
'printableString', char.PrintableString().subtype(
|
||
|
subtypeSpec=constraint.ValueSizeConstraint(1, MAX))),
|
||
|
namedtype.NamedType(
|
||
|
'universalString', char.UniversalString().subtype(
|
||
|
subtypeSpec=constraint.ValueSizeConstraint(1, MAX))),
|
||
|
namedtype.NamedType(
|
||
|
'utf8String', char.UTF8String().subtype(
|
||
|
subtypeSpec=constraint.ValueSizeConstraint(1, MAX))),
|
||
|
namedtype.NamedType(
|
||
|
'bmpString', char.BMPString().subtype(
|
||
|
subtypeSpec=constraint.ValueSizeConstraint(1, MAX))),
|
||
|
namedtype.NamedType(
|
||
|
'ia5String', char.IA5String().subtype(
|
||
|
subtypeSpec=constraint.ValueSizeConstraint(1, MAX))),
|
||
|
namedtype.NamedType(
|
||
|
'gString', univ.OctetString().subtype(
|
||
|
subtypeSpec=constraint.ValueSizeConstraint(1, MAX))),
|
||
|
)
|
||
|
|
||
|
class AttributeValue(DirectoryString):
|
||
|
pass
|
||
|
|
||
|
class AttributeType(univ.ObjectIdentifier):
|
||
|
pass
|
||
|
|
||
|
class AttributeTypeAndValue(univ.Sequence):
|
||
|
componentType = namedtype.NamedTypes(
|
||
|
namedtype.NamedType('type', AttributeType()),
|
||
|
namedtype.NamedType('value', AttributeValue()),
|
||
|
)
|
||
|
|
||
|
class RelativeDistinguishedName(univ.SetOf):
|
||
|
componentType = AttributeTypeAndValue()
|
||
|
|
||
|
class RDNSequence(univ.SequenceOf):
|
||
|
componentType = RelativeDistinguishedName()
|
||
|
|
||
|
class Name(univ.Choice):
|
||
|
componentType = namedtype.NamedTypes(
|
||
|
namedtype.NamedType('', RDNSequence()),
|
||
|
)
|
||
|
|
||
|
class GeneralName(univ.Choice):
|
||
|
componentType = namedtype.NamedTypes(
|
||
|
namedtype.NamedType('otherName', univ.Sequence().subtype(
|
||
|
implicitTag=tag.Tag(tag.tagClassContext,
|
||
|
tag.tagFormatConstructed, 0x0))),
|
||
|
namedtype.NamedType('rfc822Name', char.IA5String().subtype(
|
||
|
implicitTag=tag.Tag(tag.tagClassContext,
|
||
|
tag.tagFormatSimple, 1))),
|
||
|
namedtype.NamedType('dNSName', char.IA5String().subtype(
|
||
|
implicitTag=tag.Tag(tag.tagClassContext,
|
||
|
tag.tagFormatSimple, 2))),
|
||
|
namedtype.NamedType('x400Address', univ.Sequence().subtype(
|
||
|
implicitTag=tag.Tag(tag.tagClassContext,
|
||
|
tag.tagFormatConstructed, 0x3))),
|
||
|
namedtype.NamedType('directoryName', Name().subtype(
|
||
|
implicitTag=tag.Tag(tag.tagClassContext,
|
||
|
tag.tagFormatConstructed, 0x4))),
|
||
|
namedtype.NamedType('ediPartyName', univ.Sequence().subtype(
|
||
|
implicitTag=tag.Tag(tag.tagClassContext,
|
||
|
tag.tagFormatConstructed, 0x5))),
|
||
|
namedtype.NamedType('uniformResourceIdentifier',
|
||
|
char.IA5String().subtype(
|
||
|
implicitTag=tag.Tag(tag.tagClassContext,
|
||
|
tag.tagFormatSimple, 6))),
|
||
|
namedtype.NamedType('iPAddress', univ.OctetString().subtype(
|
||
|
implicitTag=tag.Tag(tag.tagClassContext,
|
||
|
tag.tagFormatSimple, 7))),
|
||
|
namedtype.NamedType('registeredID', univ.ObjectIdentifier().subtype(
|
||
|
implicitTag=tag.Tag(tag.tagClassContext,
|
||
|
tag.tagFormatSimple, 8))),
|
||
|
)
|
||
|
|
||
|
class GeneralNames(univ.SequenceOf):
|
||
|
componentType = GeneralName()
|
||
|
sizeSpec = univ.SequenceOf.sizeSpec + constraint.ValueSizeConstraint(1, MAX)
|
||
|
|
||
|
|
||
|
#s = '0\x1a\x82\rwww.gajim.org\x82\tgajim.org'
|
||
|
s = '0\x81\x86\x82\x0c*.jabber.org\x82\njabber.org\xa0\x1a\x06\x08+\x06\x01\x05\x05\x07\x08\x05\xa0\x0e\x0c\x0c*.jabber.org\xa0\x1a\x06\x08+\x06\x01\x05\x05\x07\x08\x07\xa0\x0e\x16\x0c*.jabber.org\xa0\x18\x06\x08+\x06\x01\x05\x05\x07\x08\x05\xa0\x0c\x0c\njabber.org\xa0\x18\x06\x08+\x06\x01\x05\x05\x07\x08\x07\xa0\x0c\x16\njabber.org'
|
||
|
|
||
|
def _parse_asn1(asn1):
|
||
|
obj = decode(asn1, asn1Spec=GeneralNames())[0]
|
||
|
r = {}
|
||
|
for o in obj:
|
||
|
name = o.getName()
|
||
|
if name == 'dNSName':
|
||
|
if name not in r:
|
||
|
r[name] = []
|
||
|
r[name].append(str(o.getComponent()))
|
||
|
if name == 'otherName':
|
||
|
if name not in r:
|
||
|
r[name] = {}
|
||
|
tag = str(tuple(o.getComponent())[0])
|
||
|
val = str(tuple(o.getComponent())[1])
|
||
|
if tag not in r[name]:
|
||
|
r[name][tag] = []
|
||
|
r[name][tag].append(val)
|
||
|
if name == 'uniformResourceIdentifier':
|
||
|
r['uniformResourceIdentifier'] = True
|
||
|
return r
|
||
|
|
||
|
def check_certificate(cert, domain):
|
||
|
cnt = cert.get_extension_count()
|
||
|
if '.' in domain:
|
||
|
compared_domain = domain.split('.', 1)[1]
|
||
|
else:
|
||
|
compared_domain = ''
|
||
|
srv_domain = '_xmpp-client.' + domain
|
||
|
compared_srv_domain = '_xmpp-client.' + compared_domain
|
||
|
for i in range(0, cnt):
|
||
|
ext = cert.get_extension(i)
|
||
|
if ext.get_short_name() == 'subjectAltName':
|
||
|
r = _parse_asn1(ext.get_data())
|
||
|
if 'otherName' in r:
|
||
|
if oid_xmppaddr in r['otherName']:
|
||
|
for host in r['otherName'][oid_xmppaddr]:
|
||
|
try:
|
||
|
host = prep(None, host, None)
|
||
|
except InvalidFormat:
|
||
|
continue
|
||
|
if host == domain:
|
||
|
return True
|
||
|
if oid_dnssrv in r['otherName']:
|
||
|
for host in r['otherName'][oid_dnssrv]:
|
||
|
if host.startswith('_xmpp-client.*.'):
|
||
|
if host.replace('*.', '', 1) == compared_srv_domain:
|
||
|
return True
|
||
|
continue
|
||
|
if host == srv_domain:
|
||
|
return True
|
||
|
if 'dNSName' in r:
|
||
|
for host in r['dNSName']:
|
||
|
if host.startswith('*.'):
|
||
|
if host[2:] == compared_domain:
|
||
|
return True
|
||
|
continue
|
||
|
if host == domain:
|
||
|
return True
|
||
|
if r:
|
||
|
return False
|
||
|
break
|
||
|
|
||
|
subject = cert.get_subject()
|
||
|
if subject.commonName == domain:
|
||
|
return True
|
||
|
return False
|