gs105e.py

   1 #!/usr/bin/python
   2 import socket
   3 import time
   4 import binascii
   5 import pprint
   6 import struct
   7 import random
   8
   9 class gs105e:
  10         def __init__(self,host,port):
  11                 self.myhost = host
  12                 self.myport = 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)
  18
  19         def recv(self,recvfunc,maxlen=8192,timeout=0.005):
  20                 self.socket.settimeout(timeout)
  21                 try:
  22                         message, address = self.socket.recvfrom(maxlen)
  23                 except socket.timeout:
  24                         return None
  25                 self.recvfunc(message,address)
  26                 self.recv(recvfunc,maxlen,timeout)
  27
  28         def parse_packet(self,p,hex=0):
  29                 data = {}
  30                 data["seq"]= p[22:24]
  31                 data["cmd"]= p[32:34]
  32                 data["mymac"]= p[8:14]
  33                 data["theirmac"]= p[14:20]
  34                 if hex == 1:
  35                         for a in data:
  36                                 data[a]=binascii.hexlify(data[a])
  37                 return data
  38
  39         def recvfunc(self,m,a):
  40                 hexdata = self.parse_packet(m,1)
  41                 data = self.parse_packet(m)
  42
  43                 if data["seq"] == "\x00\x01":
  44                         pass
  45                 print hexdata["theirmac"]
  46
  47         def send(self,host,port,data):
  48                 self.socket.sendto(data,(host,port))
  49                 self.seq+=1
  50
  51
  52         def baseudp(self,cmd,destmac="\x00\x00\x00\x00\x00\x00"):
  53                 reserved = "\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
  57                 return data
  58
  59         def discover(self):
  60                 self.srcmac = "\x00\x21\x86\x98\xda\x1e"
  61
  62                 reserved = "\x00"
  63                 cmd = "\x00\x01"
  64
  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)
  74 #               time.sleep(1)
  75                 self.send("255.255.255.255",63322, data)
  76                 time.sleep(0.7)
  77
  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)
  86                 self.recv(None)
  87
  88 g = gs105e('',63321)
  89 g.discover()