3 from vstf.common import constants
4 from vstf.common import message
7 class UdpServer(object):
9 super(UdpServer, self).__init__()
11 os.unlink(constants.sockaddr)
13 if os.path.exists(constants.sockaddr):
14 raise Exception("socket not found %s" % constants.sockaddr)
15 self.conn=socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
17 def listen(self,backlog=5):
18 self.conn.listen(backlog)
21 return self.conn.accept()
23 def bind(self, addr=constants.sockaddr):
24 return self.conn.bind(addr)
26 # def send(self, data, addr):
27 # return message.sendto(self.conn.sendto, data, addr)
29 # def recv(self, size=constants.buff_size):
30 # return message.recv(self.conn.recvfrom)
36 class UdpClient(object):
38 super(UdpClient, self).__init__()
39 if not os.path.exists(constants.sockaddr):
40 raise Exception("socket not found %s" % constants.sockaddr)
41 self.conn=socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
43 def connect(self, addr=constants.sockaddr):
44 return self.conn.connect(addr)
47 message.send(self.conn.send, data)
50 return message.recv(self.conn.recv)