openstack/deb-python-pysaml2
Code Issues Proposed changes

Disco service extention in metadata

Browse Source
This commit is contained in:
Roland Hedberg
2010-03-31 16:39:00 +02:00
parent a3c2ecc437
commit 0664c08caa
1 changed files with 20 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
tools/make_metadata.py
View File

@@ -6,7 +6,7 @@ from saml2 import BINDING_SOAP, class_name, make_instance
from saml2.time_util import in_a_while from saml2.time_util import in_a_while
from saml2.utils import parse_attribute_map, args2dict from saml2.utils import parse_attribute_map, args2dict
from saml2.saml import NAME_FORMAT_URI from saml2.saml import NAME_FORMAT_URI
from saml2.sigver import pre_signature_part, sign_statement_using_xmlsec from saml2.sigver import pre_signature_part, SecurityContext
HELP_MESSAGE = """ HELP_MESSAGE = """
Usage: make_metadata [options] 1*configurationfile Usage: make_metadata [options] 1*configurationfile
@@ -153,6 +153,19 @@ def do_sp_sso_descriptor(sp, cert, backward_map):
} }
} }
if "discovery_service" in sp:
desc["extensions"] = {"extension_elements":[
{
"tag":"DiscoveryResponse",
"namespace":md.IDPDISC,
"attributes": {
"index":"1",
"binding": md.IDPDISC,
"location":sp["url"]
}
}
]}
return desc return desc
def do_idp_sso_descriptor(idp, cert): def do_idp_sso_descriptor(idp, cert):
@@ -218,7 +231,7 @@ def entity_descriptor(confd, valid_for):
return ed return ed
def entities_descriptor(eds, valid_for, name, id, sign, xmlsec, keyfile): def entities_descriptor(eds, valid_for, name, id, sign, sc):
d = {"entity_descriptor": eds} d = {"entity_descriptor": eds}
if valid_for: if valid_for:
d["valid_until"] = in_a_while(hours=valid_for) d["valid_until"] = in_a_while(hours=valid_for)
@@ -232,9 +245,8 @@ def entities_descriptor(eds, valid_for, name, id, sign, xmlsec, keyfile):
statement = make_instance(md.EntitiesDescriptor, d) statement = make_instance(md.EntitiesDescriptor, d)
if sign: if sign:
statement = sign_statement_using_xmlsec("%s" % statement, statement = sc.sign_statement_using_xmlsec("%s" % statement,
class_name(statement), class_name(statement))
xmlsec, key_file=keyfile)
return statement return statement
@@ -284,7 +296,9 @@ def main(args):
for conf in args: for conf in args:
confd = eval(open(conf).read()) confd = eval(open(conf).read())
eds.append(entity_descriptor(confd, valid_for)) eds.append(entity_descriptor(confd, valid_for))
print entities_descriptor(eds, valid_for, name, id, sign, xmlsec, keyfile)
sc = SecurityContext(xmlsec, keyfile)
print entities_descriptor(eds, valid_for, name, id, sign, sc)
if __name__ == "__main__": if __name__ == "__main__":
import sys import sys