summaryrefslogtreecommitdiffstatshomepage
path: root/tests/net_inet/test_tls_sites.py
blob: 67345fd0b9341be6f4eb40008e7c1a291095e4a2 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
try:
    import usocket as _socket
except:
    import _socket
try:
    import ussl as ssl
except:
    import ssl


def test_one(site, opts):
    ai = _socket.getaddrinfo(site, 443)
    addr = ai[0][-1]

    s = _socket.socket()

    try:
        s.connect(addr)

        if "sni" in opts:
            s = ssl.wrap_socket(s, server_hostname=opts["host"])
        else:
            s = ssl.wrap_socket(s)

        s.write(b"GET / HTTP/1.0\r\n\r\n")
        resp = s.read(4096)
#        print(resp)

    finally:
        s.close()


SITES = [
    "google.com",
    "www.google.com",
    "api.telegram.org",
#    "w9rybpfril.execute-api.ap-southeast-2.amazonaws.com",
    {"host": "w9rybpfril.execute-api.ap-southeast-2.amazonaws.com", "sni": True},
]


def main():
    for site in SITES:
        opts = {}
        if isinstance(site, dict):
            opts = site
            site = opts["host"]

        try:
            test_one(site, opts)
            print(site, "ok")
        except Exception as e:
            print(site, repr(e))


main()