tests/extmod/tls_sslcontext_ciphers.py


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

# Basic test of tls.SSLContext get_ciphers() and set_ciphers() methods.

try:
    import tls

    tls.SSLContext.get_ciphers
except (ImportError, AttributeError):
    print("SKIP")
    raise SystemExit


ctx = tls.SSLContext(tls.PROTOCOL_TLS_CLIENT)

ciphers = ctx.get_ciphers()

for ci in ciphers:
    # Only print those ciphers know to exist on all ports.
    if ("TLS-ECDHE-ECDSA-WITH-AES" in ci or "TLS-RSA-WITH-AES" in ci) and "CBC" in ci:
        print(ci)

ctx.set_ciphers(ciphers[:1])

# Test error cases.

try:
    ctx.set_ciphers(ciphers[0])
except TypeError as e:
    print(e)

try:
    ctx.set_ciphers(["BAR"])
except OSError as e:
    print(e)