10 def __init__(self,host,port):
13 self.socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
14 self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
15 self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
16 self.socket.bind((host, port))
17 self.seq = random.randint(100,2000)
19 def recv(self,recvfunc,maxlen=8192,timeout=0.005):
20 self.socket.settimeout(timeout)
22 message, address = self.socket.recvfrom(maxlen)
23 except socket.timeout:
25 self.recvfunc(message,address)
26 self.recv(recvfunc,maxlen,timeout)
28 def parse_packet(self,p,hex=0):
32 data["mymac"]= p[8:14]
33 data["theirmac"]= p[14:20]
36 data[a]=binascii.hexlify(data[a])
39 def recvfunc(self,m,a):
40 hexdata = self.parse_packet(m,1)
41 data = self.parse_packet(m)
43 if data["seq"] == "\x00\x01":
45 print hexdata["theirmac"]
47 def send(self,host,port,data):
48 self.socket.sendto(data,(host,port))
52 def baseudp(self,cmd,destmac="\x00\x00\x00\x00\x00\x00"):
54 data = 2 * "\x01" + 6* reserved + self.srcmac + destmac + 2*reserved
55 data += struct.pack(">h",self.seq)
56 data += "NSDP" + 4 * reserved + cmd
60 self.srcmac = "\x00\x21\x86\x98\xda\x1e"
65 data = self.baseudp(cmd)
66 data += 3 * "\x00" + "\x02"
67 data += 3 * "\x00" + "\x03" + 3 * "\x00" + "\x04" + 3* "\x00" + "\x05"
68 data += 3 * "\x00" + "\x06" + 3 * "\x00" + "\x07" + 3* "\x00" + "\x08"
69 data += 3 * "\x00" + "\x0b" + 3 * "\x00" + "\x0c" + 3* "\x00" + "\x0d"
70 data += 3 * "\x00" + "\x0e" + 3 * "\x00" + "\x0f" + 2 * "\x00"
71 data += "\x74" + 3 * "\x00"
72 data += 2* "\xff" + 2* "\x00"
73 # self.send("255.255.255.255",63322, data)
75 self.send("255.255.255.255",63322, data)
78 data = self.baseudp(cmd)
79 data += 3 * "\x00" + "\x02"
80 data += 3 * "\x00" + "\x03" + 3 * "\x00" + "\x04" + 3* "\x00" + "\x05"
81 data += 3 * "\x00" + "\x06" + 3 * "\x00" + "\x07" + 3* "\x00" + "\x08"
82 data += 3 * "\x00" + "\x0b" + 3 * "\x00" + "\x0c" + 3* "\x00" + "\x0d"
83 data += 3 * "\x00" + "\x0e" + 3 * "\x00" + "\x0f" + 2 * "\x00"
84 data += 2* "\xff" + 2* "\x00"
85 # self.send("255.255.255.255",63322, data)