Initial commit
This commit is contained in:
commit
ccd9753276
3 changed files with 55 additions and 0 deletions
1
.gitignore
vendored
Normal file
1
.gitignore
vendored
Normal file
|
@ -0,0 +1 @@
|
||||||
|
__pycache__
|
16
demo.py
Normal file
16
demo.py
Normal file
|
@ -0,0 +1,16 @@
|
||||||
|
import network, sys, time
|
||||||
|
|
||||||
|
sock = network.Sock()
|
||||||
|
|
||||||
|
port = int(sys.argv[2])
|
||||||
|
|
||||||
|
if sys.argv[1] == "s":
|
||||||
|
sock.listen(("0.0.0.0", port))
|
||||||
|
while True:
|
||||||
|
for (d, a) in sock.poll():
|
||||||
|
print(d, a)
|
||||||
|
time.sleep(0.1)
|
||||||
|
elif sys.argv[1] == "c":
|
||||||
|
while True:
|
||||||
|
sock.send(str(time.time()), (sys.argv[3], port))
|
||||||
|
time.sleep(1)
|
38
network.py
Normal file
38
network.py
Normal file
|
@ -0,0 +1,38 @@
|
||||||
|
import socket
|
||||||
|
from threading import Thread, Lock
|
||||||
|
|
||||||
|
class Sock:
|
||||||
|
def __init__(self):
|
||||||
|
self.sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||||||
|
self.queue = []
|
||||||
|
self.queue_lock = Lock()
|
||||||
|
self.sock_thread = None
|
||||||
|
|
||||||
|
def send(self, message, address):
|
||||||
|
self.sock.sendto(message, address)
|
||||||
|
|
||||||
|
def listen(self, address, length=65535):
|
||||||
|
self.sock_thread = SockThread(self, address, length)
|
||||||
|
self.sock_thread.start()
|
||||||
|
|
||||||
|
def poll(self):
|
||||||
|
if len(self.queue_lock) == 0:
|
||||||
|
return []
|
||||||
|
with self.queue_lock:
|
||||||
|
queue = self.queue
|
||||||
|
self.queue = []
|
||||||
|
return queue
|
||||||
|
|
||||||
|
class SockThread(Thread):
|
||||||
|
def __init__(self, sock, address, length):
|
||||||
|
super.__init__(super)
|
||||||
|
self.sock = sock
|
||||||
|
self.sock.sock.bind(address)
|
||||||
|
self.length = self.length
|
||||||
|
self.sock.sock.bind(address)
|
||||||
|
|
||||||
|
def run(self):
|
||||||
|
while True:
|
||||||
|
r = self.sock.sock.recv(self.length)
|
||||||
|
with self.sock.queue_lock:
|
||||||
|
self.sock.queue.append(r)
|
Loading…
Reference in a new issue