kopia lustrzana https://github.com/micropython/micropython-lib
79 wiersze
2.1 KiB
Python
79 wiersze
2.1 KiB
Python
|
import uerrno
|
||
|
|
import usocket
|
||
|
|
from uasyncio.core import *
|
||
|
|
|
||
|
|
|
||
|
|
DEBUG = 0
|
||
|
|
log = None
|
||
|
|
|
||
|
|
def set_debug(val):
|
||
|
|
global DEBUG, log
|
||
|
|
DEBUG = val
|
||
|
|
if val:
|
||
|
|
import logging
|
||
|
|
log = logging.getLogger("uasyncio.udp")
|
||
|
|
|
||
|
|
|
||
|
|
class UdpSocket:
|
||
|
|
|
||
|
|
def __init__(self, s):
|
||
|
|
self.s = s
|
||
|
|
|
||
|
|
def recv(self, n):
|
||
|
|
try:
|
||
|
|
yield IORead(self.s)
|
||
|
|
return self.s.recv(n)
|
||
|
|
except:
|
||
|
|
#print("recv: exc, cleaning up")
|
||
|
|
#print(uasyncio.core._event_loop.objmap, uasyncio.core._event_loop.poller)
|
||
|
|
#uasyncio.core._event_loop.poller.dump()
|
||
|
|
yield IOReadDone(self.s)
|
||
|
|
#print(uasyncio.core._event_loop.objmap)
|
||
|
|
#uasyncio.core._event_loop.poller.dump()
|
||
|
|
raise
|
||
|
|
|
||
|
|
def recvfrom(self, n):
|
||
|
|
try:
|
||
|
|
yield IORead(self.s)
|
||
|
|
return self.s.recvfrom(n)
|
||
|
|
except:
|
||
|
|
#print("recv: exc, cleaning up")
|
||
|
|
#print(uasyncio.core._event_loop.objmap, uasyncio.core._event_loop.poller)
|
||
|
|
#uasyncio.core._event_loop.poller.dump()
|
||
|
|
yield IOReadDone(self.s)
|
||
|
|
#print(uasyncio.core._event_loop.objmap)
|
||
|
|
#uasyncio.core._event_loop.poller.dump()
|
||
|
|
raise
|
||
|
|
|
||
|
|
def asendto(self, buf, addr=None):
|
||
|
|
while 1:
|
||
|
|
res = self.s.sendto(buf, 0, addr)
|
||
|
|
#print("send res:", res)
|
||
|
|
if res == len(buf):
|
||
|
|
return
|
||
|
|
print("asento: IOWrite")
|
||
|
|
yield IOWrite(self.s)
|
||
|
|
|
||
|
|
def aclose(self):
|
||
|
|
yield IOReadDone(self.s)
|
||
|
|
self.s.close()
|
||
|
|
|
||
|
|
|
||
|
|
def udp_socket(host=None, port=None):
|
||
|
|
if DEBUG and __debug__:
|
||
|
|
log.debug("udp_socket(%s, %s)", host, port)
|
||
|
|
s = usocket.socket(usocket.AF_INET, usocket.SOCK_DGRAM)
|
||
|
|
s.setblocking(False)
|
||
|
|
if host and port:
|
||
|
|
ai = usocket.getaddrinfo(host, port)
|
||
|
|
addr = ai[0][-1]
|
||
|
|
try:
|
||
|
|
s.connect(addr)
|
||
|
|
except OSError as e:
|
||
|
|
if e.args[0] != uerrno.EINPROGRESS:
|
||
|
|
raise
|
||
|
|
if DEBUG and __debug__:
|
||
|
|
log.debug("udp_socket: After connect")
|
||
|
|
return UdpSocket(s)
|
||
|
|
yield
|