B wZ[t@sddlZddlZddlmZejZejZejjZejj Z dgZ dddge Z da dadd d Zdd d Zdd dZdddZejdddZdS)N)urllibz www.github.iozdata.astropy.orgzastropy.stsci.eduzwww.astropy.orgFPcCsLytddt||D}Wntjk r<tg}YnX|||S)z_ Obtain all the IPs, including aliases, in a way that supports IPv4/v6 dual stack. cSsg|]}|ddqS)r).0srrAlib/python3.7/site-packages/pytest_remotedata/disable_internet.py !sz%_resolve_host_ips..)setsocketZ getaddrinfoZgaierroradd)hostnameZportZipsrrr_resolve_host_ipss  rcsfdd}|S)z Wraps ``original_function``, which in most cases is assumed to be a `socket.socket` method, to raise an `IOError` for any operations on non-local AF_INET sockets. Allowing Astropy data also automatically allow GitHub data. c slt|dtjrP|djtjtjfkr0||S|dd}d}tdddg}nDt|dtrnt|ddksx||S|dd}d}tddg}rx.new_functionr)rrrrr)rrrrcheck_internet_off,s 0rcCstrdSdad}|rtdtjatji}tj|}tj|tt ||dt _ tt ||dt j _ tt||dt j _t S)a1 Disable internet access via python by preventing connections from being created using the socket module. Presumably this could be worked around by using some other means of accessing the internet, but all default python modules (urllib, requests, etc.) use socket [citation needed]. NTzInternet access disabled)rr) INTERNET_OFFprintrrequestZ build_opener _orig_openerZ ProxyHandlerinstall_openerrsocket_create_connectionr create_connection socket_bindbindsocket_connectconnect)verboserrZ__tracebackhide__Zno_proxy_handleropenerrrrturn_off_internetis(        r-cCs>tsdSda|rtdtjttt_t tj_ t tj_ tS)zL Restore internet access. Not used, but kept in case it is needed. NFzInternet access enabled) r r!rr"r$r#r%r r&r'r(r)r*)r+rrrturn_on_internets r.c cs.t}t|dz dVWd|s(t|dXdS)a7Context manager to temporarily disable internet access (if not already disabled). If it was already disabled before entering the context manager (i.e. `turn_off_internet` was called previously) then this is a no-op and leaves internet access disabled until a manual call to `turn_on_internet`. )r+N)r r-r.)r+Zalready_disabledrrr no_internets   r/)r)FF)FFF)F)F) contextlibr Z six.movesrZsocket_originalr&r%r(r'r*r)rrr r#rrr-r.contextmanagerr/rrrrs&   < )