# -*- coding: GBK -*-
|
|
import os
|
from socket import *
|
import struct
|
import binascii
|
import traceback
|
|
#----------------------дÈë
|
def WriteBYTE(buf, value):
|
buf += struct.pack('B', value)
|
return buf
|
|
def WriteWORD(buf, value):
|
buf += struct.pack('H', value)
|
return buf
|
|
def WriteDWORD(buf, value):
|
buf += struct.pack('I', value)
|
return buf
|
|
def WriteFloat(buf, value):
|
buf += struct.pack('f', value)
|
return buf
|
|
def WriteDouble(buf, value):
|
buf += struct.pack('d', value)
|
return buf
|
|
def WriteString(buf, len, value):
|
buf += struct.pack('%ds'%len, value)
|
return buf
|
|
|
#------------------------¶ÁÈ¡
|
def ReadBYTE(buf, pos):
|
curValue = struct.unpack_from('B', buf, pos)
|
pos += 1
|
return curValue[0], pos
|
|
def ReadWORD(buf, pos):
|
curValue = struct.unpack_from('H', buf, pos)
|
pos += 2
|
return curValue[0], pos
|
|
def ReadDWORD(buf, pos):
|
curValue = struct.unpack_from('I', buf, pos)
|
pos += 4
|
return curValue[0], pos
|
|
def ReadFloat(buf, pos):
|
curValue = struct.unpack_from('f', buf, pos)
|
pos += 4
|
return curValue[0], pos
|
|
def ReadDouble(buf, pos):
|
curValue = struct.unpack_from('d', buf, pos)
|
pos += 8
|
return curValue[0], pos
|
|
def ReadString(buf, pos, _len):
|
curValue = struct.unpack_from('%ds'%_len, buf, pos)
|
pos += _len
|
return curValue[0], pos
|
|
def PackParse(data):
|
buf = data
|
recvdata = ''
|
pos = 0
|
try:
|
PackHead, pos = ReadWORD(buf, pos)
|
|
if PackHead != 0xCCFF:
|
print 'PackHead not the 0xCCFF'
|
return data
|
|
packlen, pos = ReadDWORD(buf, pos)
|
if packlen + pos != len(data):
|
print 'PackLen Wrong!'
|
return data
|
|
curPack , pos = ReadDWORD(buf, pos) #SessionID
|
curPack , pos = ReadWORD(buf, pos) #PoolIndex
|
curPack , pos = ReadBYTE(buf, pos) #Type
|
curPack, pos = ReadBYTE(buf, pos) #DestServerType
|
curPack, pos = ReadBYTE(buf, pos) #SrcServerType
|
curPack , pos = ReadString(buf, pos, len(data)-pos)
|
recvdata = str(curPack)
|
|
except:
|
print traceback.format_exc()
|
return data
|
|
return recvdata
|
|
fullfilename = '' #±£´æRoute·µ»ØµÄÏà¹ØÐÅÏ¢µÄÎļþ
|
BUFSIZ = 1024
|
|
def GetServerVersion(sClient, fResult):
|
global fullfilename
|
global BUFSIZ
|
|
data = ''
|
data = WriteDWORD(data,0) #SessionID = 0
|
data = WriteWORD(data,0) #PoolIndex = 0
|
data = WriteBYTE(data,2) #Type = atInner
|
data = WriteBYTE(data,5) #DestServerType = stRoute
|
data = WriteBYTE(data,6) #SrcServerType = stMap
|
data = WriteBYTE(data,37) #TMapServerToBalanceServer = mgServerVersion
|
data = WriteDWORD(data,0)
|
datalen = len(data)
|
head = ''
|
head = WriteDWORD(head,datalen)
|
data = head + data
|
__PackHead = 0xCCFF
|
PackHead = ''
|
PackHead = WriteWORD(PackHead, __PackHead)
|
data = PackHead + data
|
|
recvdata = ''
|
while True:
|
print 'ÇëÇó°æ±¾ÖС£¡£¡£¡£'
|
sClient.send(data)
|
print '½ÓÊÜ°æ±¾ÖС£¡£¡£¡£'
|
dataRecv = sClient.recv(BUFSIZ)
|
if not dataRecv :
|
continue
|
|
resultData = PackParse(dataRecv)
|
#Èç¹ûÊý¾Ý²»ÍêÕû£¬ÔòÖØ·¢ÔÙ½ÓÊÕ¡£
|
if resultData == dataRecv:
|
print 'Êý¾Ý²»ÍêÕû,Öؽӡ£¡£¡£¡£'
|
continue
|
try:
|
#print str(resultData)
|
#print fResult
|
if str(resultData)=="":
|
continue
|
fResult.write(str(resultData))
|
print "ÇëÇó°æ±¾½á¹û[%s]"%str(resultData)
|
except:
|
print "Write %s failed"%fullfilename
|
os.system("pause")
|
return
|
print "Write SoftVersion OK"
|
break
|
|
def GetLineID(sClient, fResult):
|
global fullfilename
|
global BUFSIZ
|
|
data = ''
|
data = WriteDWORD(data,0) #SessionID = 0
|
data = WriteWORD(data,0) #PoolIndex = 0
|
data = WriteBYTE(data,2) #Type = atInner
|
data = WriteBYTE(data,5) #DestServerType = stRoute
|
data = WriteBYTE(data,6) #SrcServerType = stMap
|
data = WriteBYTE(data,41) #TMapServerToBalanceServer = mgLineID
|
data = WriteDWORD(data,0)
|
datalen = len(data)
|
head = ''
|
head = WriteDWORD(head,datalen)
|
data = head + data
|
__PackHead = 0xCCFF
|
PackHead = ''
|
PackHead = WriteWORD(PackHead, __PackHead)
|
data = PackHead + data
|
|
recvdata = ''
|
while True:
|
print 'ÇëÇó·ÖÏßIDÖС£¡£¡£¡£'
|
sClient.send(data)
|
print '½ÓÊÜ·ÖÏßIDÖС£¡£¡£¡£'
|
dataRecv = sClient.recv(BUFSIZ)
|
if not dataRecv :
|
continue
|
|
resultData = PackParse(dataRecv)
|
#Èç¹ûÊý¾Ý²»ÍêÕû£¬ÔòÖØ·¢ÔÙ½ÓÊÕ¡£
|
if resultData == dataRecv:
|
print 'Êý¾Ý²»ÍêÕû,Öؽӡ£¡£¡£¡£'
|
continue
|
try:
|
#print str(resultData)
|
fResult.write(str(resultData))
|
print "ÇëÇó·ÖÏßID½á¹û[%s]"%str(resultData)
|
except:
|
print "Write %s failed"%fullfilename
|
os.system("pause")
|
return
|
print "Write LineID OK"
|
break
|
|
|
|
|
def process(IP,Port):
|
global fullfilename
|
|
print IP,Port
|
|
if not os.path.isdir(os.getcwd()+'//RouterInfoResult'):
|
os.makedirs('RouterInfoResult')
|
|
try:
|
fullfilename = os.getcwd()+'\\RouterInfoResult\\Result.txt'
|
fileSave = open(fullfilename,'w')
|
except:
|
print "Can not open %s in w mode"%fullfilename
|
os.system("pause")
|
return
|
HOST = IP
|
PORT = Port
|
ADDR = (HOST, PORT)
|
|
try:
|
tcpCliSock = socket(AF_INET, SOCK_STREAM)
|
tcpCliSock.connect(ADDR)
|
except:
|
print "Can't Connect To Route Server. IP = %s, Port = %s"%(IP,Port)
|
os.system("pause")
|
return
|
|
GetServerVersion(tcpCliSock, fileSave) #ÇëÇó·þÎñÆ÷°æ±¾ÐÅÏ¢
|
GetLineID(tcpCliSock, fileSave) #ÇëÇó·þÎñÆ÷·ÖÏßID
|
|
fileSave.close()
|
tcpCliSock.close()
|
return
|
|
|
process(os.sys.argv[1], int(os.sys.argv[2]))
|
