mirror of https://github.com/sgoudham/Enso-Bot.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
779 B
Python
41 lines
779 B
Python
5 years ago
|
import ssl
|
||
|
|
import sys
|
||
|
|
|
||
|
|
import idna
|
||
|
|
|
||
|
|
__version__ = '1.1.0'
|
||
|
|
|
||
|
|
real_match_hostname = ssl.match_hostname
|
||
|
|
|
||
|
|
PY_370 = sys.version_info >= (3, 7, 0)
|
||
|
|
|
||
|
|
|
||
|
|
def patched_match_hostname(cert, hostname):
|
||
|
|
try:
|
||
|
|
hostname = idna.encode(hostname, uts46=True).decode('ascii')
|
||
|
|
except UnicodeError:
|
||
|
|
hostname = hostname.encode('idna').decode('ascii')
|
||
|
|
|
||
|
|
return real_match_hostname(cert, hostname)
|
||
|
|
|
||
|
|
|
||
|
|
def patch_match_hostname():
|
||
|
|
if PY_370:
|
||
|
|
return
|
||
|
|
|
||
|
|
if hasattr(ssl.match_hostname, 'patched'):
|
||
|
|
return
|
||
|
|
|
||
|
|
ssl.match_hostname = patched_match_hostname
|
||
|
|
ssl.match_hostname.patched = True
|
||
|
|
|
||
|
|
|
||
|
|
def reset_match_hostname():
|
||
|
|
if PY_370:
|
||
|
|
return
|
||
|
|
|
||
|
|
if not hasattr(ssl.match_hostname, 'patched'):
|
||
|
|
return
|
||
|
|
|
||
|
|
ssl.match_hostname = real_match_hostname
|