blob: 9c12114eab1f627a11b9b86f8608d2f7458d9c78 (
plain)
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
|
# test that modtls produces a numerical error message when out of heap
import socket, ssl, sys
try:
from micropython import heap_lock, heap_unlock
except:
print("SKIP")
raise SystemExit
try:
from micropython import alloc_emergency_exception_buf
alloc_emergency_exception_buf(256)
except:
pass
# test with heap locked to see it switch to number-only error message
def test(addr):
s = socket.socket()
s.connect(addr)
try:
s.setblocking(False)
s = ssl.wrap_socket(s, do_handshake=False)
heap_lock()
print("heap is locked")
while True:
ret = s.write("foo")
if ret:
break
heap_unlock()
print("wrap: no exception")
except OSError as e:
heap_unlock()
# mbedtls produces "-29184"
# axtls produces "RECORD_OVERFLOW"
ok = "-29184" in str(e) or "RECORD_OVERFLOW" in str(e)
print("wrap:", ok)
if not ok:
print("got exception:", e)
s.close()
if __name__ == "__main__":
# connect to plain HTTP port, oops!
addr = socket.getaddrinfo("micropython.org", 80)[0][-1]
test(addr)
|
