login.py

来自「调试程序写的驱动」· Python 代码 · 共 134 行

#!/usr/bin/python

import socket, sys , struct , time

if __name__ == '__main__':

	argv = sys.argv

	if (len(argv)!=4) or (len(argv)==2 and argv[1]=='/?'):

		print '>>>Useage:', argv[0], '<address> <port> <msisdn>'

		sys.exit(0)

HOST = argv[1] #Server IP地址

PORT = int(argv[2]) #Server 端口

msisdn = argv[3]

print '>>>HOST:', HOST

print '>>>PORT:', PORT

print '>>>msisdn:', msisdn


#打开socket

try:

	sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

except socket.error:

	print '>>>Fail to open socket!'

	sys.exit(1)


print '>>>Socket opened...'

#连接

try:

	sock.connect((HOST, PORT))

except socket.error:

	print '>>>Fail to connect!'

	sock.close()

	print '>>>Socket closed...'

	sys.exit(1)


print '>>>Socket connected...'

#compose msg
msgbody = '<Req><msisdn>'+ msisdn +'</msisdn><pwd>111222</pwd><ua/><ver/><metaid/></Req>'''
format = '=HIHII'+str(len(msgbody))+'s'
msg = struct.pack(format , socket.htons(1), socket.htonl(16+len(msgbody)) , socket.htons(1), socket.htonl(1), socket.htonl(1), msgbody)

list = struct.unpack(format, msg)
print list

#发送信息

try:

	sock.send(msg)

except socket.error:

	print '>>>Fail to send message!'
	sys.exit(1)


print '>>>Message sent :', msg


#接收信息
try: 

	msg = sock.recv(1024)

except socket.error:

	print '>>>Fail to receive message!'
	sys.exit(1)


print '>>>Message received:', repr(msg)


while(1):
	format = '=HIHII'
	msg = struct.pack(format , socket.htons(1) , socket.htonl(16) , socket.htons(9), socket.htonl(1), socket.htonl(1))
	try:
		sock.send(msg)

	except socket.error:

		print '>>>Fail to send message!'
		sys.exit(1)


	#print '>>>Message sent :', msg

	#try:
	#	msg = sock.recv(1024)

	#except socket.error:

	#	print '>>>Fail to receive message!'
	#	sys.exit(1)
	#print '>>>Message received:', msg
	time.sleep(5)
	

#关闭socket，退出程序

sock.close()

print '>>>Socket closed...'

print 'Exit client program.'

sys.exit(0)