97fd18a7e2
This is a workaround for the case where threading is enabled without a GIL. In such a configuration, creating a new global variable is not atomic and threads have race conditions resizing/accessing the global dict. Signed-off-by: Damien George <damien@micropython.org>
66 lines
1.9 KiB
Python
66 lines
1.9 KiB
Python
# Test interruption of select.poll by EINTR signal, when
|
|
# MICROPY_PY_SELECT_POSIX_OPTIMISATIONS is enabled.
|
|
|
|
try:
|
|
import time, gc, select, socket, _thread
|
|
|
|
time.time_ns # Check for time_ns on MicroPython
|
|
select.poll # Raises AttributeError for CPython implementations without poll()
|
|
except (ImportError, AttributeError):
|
|
print("SKIP")
|
|
raise SystemExit
|
|
|
|
# Use a new UDP socket for tests, which should be writable but not readable.
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
|
localhost_addr_info = socket.getaddrinfo("127.0.0.1", 8000)
|
|
try:
|
|
s.bind(localhost_addr_info[0][-1])
|
|
except OSError:
|
|
# Target can't bind to localhost.
|
|
# Most likely it doesn't have a NIC and the test cannot be run.
|
|
s.close()
|
|
print("SKIP")
|
|
raise SystemExit
|
|
|
|
|
|
def thread_main():
|
|
lock.acquire()
|
|
time.sleep(0.2)
|
|
print("thread gc start")
|
|
# The unix gc.collect() implementation will raise EINTR on other threads.
|
|
# Could possibly use _thread._interrupt_main() instead if MicroPython had it.
|
|
gc.collect()
|
|
print("thread gc end")
|
|
|
|
|
|
# Pre-allocate global variables here so the global dict is not resized by the main
|
|
# thread while the secondary thread runs. This is a workaround for the bug described
|
|
# in https://github.com/micropython/micropython/pull/11604
|
|
poller = None
|
|
t0 = None
|
|
result = None
|
|
dt_ms = None
|
|
|
|
# Start a thread to interrupt the main thread during its call to poll.
|
|
lock = _thread.allocate_lock()
|
|
lock.acquire()
|
|
_thread.start_new_thread(thread_main, ())
|
|
|
|
# Create the poller object.
|
|
poller = select.poll()
|
|
poller.register(s, select.POLLIN)
|
|
|
|
# Poll on the UDP socket for a set timeout, which should be reached.
|
|
print("poll")
|
|
lock.release()
|
|
t0 = time.time_ns()
|
|
result = poller.poll(400)
|
|
dt_ms = (time.time_ns() - t0) / 1e6
|
|
print("result:", result)
|
|
if 380 <= dt_ms <= 600:
|
|
print("dt in range")
|
|
else:
|
|
print("dt not in range:", dt_ms)
|
|
|
|
# Clean up.
|
|
s.close()
|