Skip to content

Commit f08d6d4

Browse files
Xiao Zhuc00kiemon5ter
authored andcommitted
add timeout to config and httpbase
1 parent a924435 commit f08d6d4

File tree

4 files changed

+7
-4
lines changed

4 files changed

+7
-4
lines changed

src/saml2/config.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,7 @@
7676
"name_id_format",
7777
"signing_algorithm",
7878
"digest_algorithm",
79+
"timeout",
7980
]
8081

8182
SP_ARGS = [
@@ -228,6 +229,7 @@ def __init__(self, homedir="."):
228229
self.delete_tmpfiles = True
229230
self.signing_algorithm = None
230231
self.digest_algorithm = None
232+
self.timeout = None
231233

232234
def setattr(self, context, attr, val):
233235
if context == "":

src/saml2/entity.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -175,7 +175,7 @@ def __init__(self, entity_type, config=None, config_file="",
175175

176176
HTTPBase.__init__(self, self.config.verify_ssl_cert,
177177
self.config.ca_certs, self.config.key_file,
178-
self.config.cert_file)
178+
self.config.cert_file, self.config.timeout)
179179

180180
if self.config.vorg:
181181
for vo in self.config.vorg.values():

src/saml2/httpbase.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,7 @@ def dict2set_list(dic):
100100

101101
class HTTPBase(object):
102102
def __init__(self, verify=True, ca_bundle=None, key_file=None,
103-
cert_file=None):
103+
cert_file=None, timeout=None):
104104
self.request_args = {"allow_redirects": False}
105105
#self.cookies = {}
106106
self.cookiejar = http_cookiejar.CookieJar()
@@ -111,6 +111,7 @@ def __init__(self, verify=True, ca_bundle=None, key_file=None,
111111
self.request_args["verify"] = ca_bundle
112112
if key_file:
113113
self.request_args["cert"] = (cert_file, key_file)
114+
self.request_args["timeout"] = timeout
114115

115116
self.sec = None
116117
self.user = None

src/saml2/mdstore.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1032,9 +1032,9 @@ def __init__(self, attrc, config, ca_certs=None,
10321032
MetaData.__init__(self, attrc, check_validity=check_validity)
10331033

10341034
if disable_ssl_certificate_validation:
1035-
self.http = HTTPBase(verify=False, ca_bundle=ca_certs)
1035+
self.http = HTTPBase(verify=False, ca_bundle=ca_certs, timeout=config.timeout)
10361036
else:
1037-
self.http = HTTPBase(verify=True, ca_bundle=ca_certs)
1037+
self.http = HTTPBase(verify=True, ca_bundle=ca_certs, timeout=config.timeout)
10381038

10391039
self.security = security_context(config)
10401040
self.ii = 0

0 commit comments

Comments
 (0)