137 lines
3.0 KiB
Python
137 lines
3.0 KiB
Python
import os
|
|
import time
|
|
import threading
|
|
import sys
|
|
import socket
|
|
|
|
def get_free_local_addr():
|
|
with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s:
|
|
s.connect(("8.8.8.8", 80))
|
|
return s.getsockname()
|
|
# return (ip,port)
|
|
|
|
class background(threading.Thread) :
|
|
def __init__(self,func,kw) :
|
|
threading.Thread.__init__(self)
|
|
self.func = func
|
|
self.kw = kw
|
|
|
|
def run(self) :
|
|
if self.func!=None :
|
|
self.func(**self.kw)
|
|
return
|
|
|
|
def BackgroundCall(func,datas) :
|
|
b=background(func,datas)
|
|
b.start()
|
|
return
|
|
|
|
class SocketServerError(Exception) :
|
|
pass
|
|
|
|
class SocketClientError(Exception) :
|
|
pass
|
|
|
|
class SocketServer(threading.Thread) :
|
|
|
|
def __init__(self,host,port,max_connect=10,callee=None) :
|
|
threading.Thread.__init__(self, name = 'SocketServer')
|
|
self.setDaemon(False)
|
|
self.host = host
|
|
self.port = int(port)
|
|
self.max_c = max_connect
|
|
self.ready = False
|
|
self.keep_running = 0
|
|
self.callee = callee
|
|
self.setSocketServer()
|
|
|
|
def setSocketServer(self) :
|
|
try :
|
|
self.sock = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
|
|
self.sock.bind((self.host,self.port))
|
|
self.sock.listen(self.max_c)
|
|
self.ready = True
|
|
except Exception as e:
|
|
print('setSocketServer() Error:%s\nhost=%s,port=%d' % (e,self.host,self.port))
|
|
pass
|
|
|
|
def run(self) :
|
|
if not self.ready :
|
|
raise SocketServerError('not ready')
|
|
callee = self.callee
|
|
if self.callee!=None :
|
|
callee = self.callee
|
|
self.keep_running = 1
|
|
while self.keep_running :
|
|
conn,addr = self.sock.accept()
|
|
BackgroundCall(callee,{'conn':conn,'addr':addr})
|
|
# conn.close()
|
|
|
|
def stop(self) :
|
|
self.keep_running = 0
|
|
|
|
def callee(self,conn,addr) :
|
|
while 1 :
|
|
d = conn.recv(1024)
|
|
if d==None :
|
|
break
|
|
conn.send(d)
|
|
con.close()
|
|
|
|
class SocketClient :
|
|
|
|
def __init__(self,host,port) :
|
|
self.host = host
|
|
self.port = port
|
|
self.ready = False
|
|
self.connect()
|
|
|
|
# if tim ==0 not blocking
|
|
def timeout(self,tim) :
|
|
if self.ready :
|
|
self.sock.setblocking(tim>0)
|
|
if tim>0 :
|
|
self.sock.settimeout(tim)
|
|
|
|
def connect(self) :
|
|
try :
|
|
self.sock = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
|
|
self.sock.connect((self.host,self.port))
|
|
self.ready = True
|
|
except Exception as e:
|
|
self.ready = False
|
|
print('Socket connect error,%s\nhost=%s,port=%s' % (e,self.host,self.port))
|
|
raise SocketClientError('connect error')
|
|
|
|
def read(self,size) :
|
|
try :
|
|
data = self.sock.recv(size)
|
|
return data
|
|
except Exception as e:
|
|
print('recv error,%s' % e)
|
|
raise SocketClientError('recv error')
|
|
|
|
def write(self,data) :
|
|
try :
|
|
self.sock.send(data)
|
|
except Exception as e:
|
|
print('recv error,%s' % e)
|
|
raise SocketClientError('send error')
|
|
|
|
def close(self) :
|
|
self.sock.close()
|
|
self.ready = False
|
|
|
|
if __name__ == '__main__' :
|
|
s = SocketServer('localhost',12232)
|
|
s.start()
|
|
time.sleep(5)
|
|
while 1 :
|
|
c = SocketClient('localhost',12232)
|
|
msg = 'msg1'
|
|
print("send:",msg)
|
|
c.write(msg)
|
|
d = c.read(1024)
|
|
print("get:",d)
|
|
time.sleep(1)
|