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