【Socket】
本文记录了一些socket模块的简单应用,对于具体原理还没来得及深究。
■ 利用socket模块进行端口连接验证和扫描
在linux中常用nc命令来进行远端端口是否开放的验证。但是这个命令并不是系统自带的,所以还比较麻烦。如果利用python自带的socket模块就可以比较自由地进行端口验证了。是这样用的:
importsocket
defport_check(host,port):
sk = socket.socket(socket.AF_INET,socket.SOCK_STREAM) #建立socket对象
sk.settimeout(1) #设置超时,否则可能等很久才会返回连不通端口的信息
try:
sk.connect((host,port)) #进行socket层面的连接,注意参数是一个tuple
exceptsocket.error as e:
print "{}:{} is not opened. {}".format(host,port,str(e))
else:
print "{}:{} is opened.".format(host,port)
finally:
sk.close()
if __name__ == '__main__':
port_check('127.0.0.1',22)
如果想要对远程主机进行端口扫描的话,只要在这个函数外面包一层for i in range(1,1024),就可以了,考虑到效率问题还可以引入多线程。更多详细操作,可以看【http://www.cnblogs.com/xuanhun/p/5950433.html】。需要提醒一点的是,在没有得到允许情况下,私自对别人的主机进行端口扫描被认为是不友好的挑衅行为。
■ 构建tcp客户端发送tcp报文
#!/usr/bin/env python#coding=utf-8
importsocket
target_host= "127.0.0.1"target_port= 1234
#建立一个socket对象
client =socket.socket(socket.AF_INET,socket.SOCK_STREAM)#链接客户端
client.connect((target_host,target_port))whileTrue:
data= raw_input('>')
client.send(data)
data= client.recv(4096)if notdata:break
print data
构造出来的是一个带有字符提示的简单TCP报文发送器。代码很简单,不多说了。
■ 构建tcp服务端
#! /usr/bin/env python#coding=utf-8
importsocket
bind_ip= ""bind_port= 9999server=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
server.bind((bind_ip,bind_port))
server.listen(5)try:whileTrue:
client,add=server.accept()print "[*]你监听的是:%s:%d" % (add[0],add[1])whileTrue:
data= client.recv(1024)if notdata:break
printdata
data= raw_input('>')
client.send(data)else:
client.close()exceptException as e:printe
server.close()
■ 构建UDP服务端
#-*- coding: utf-8 -*-
from socket import *
from time importctime
HOST= '' #主机名
PORT = 21567 #端口号
BUFSIZE = 1024 #缓冲区大小1K
ADDR =(HOST,PORT)
udpSerSock=socket(AF_INET, SOCK_DGRAM)
udpSerSock.bind(ADDR)#绑定地址到套接字
while True: #无限循环等待连接到来
try:print 'Waiting for message ....'data, addr= udpSerSock.recvfrom(BUFSIZE) #接受UDP
print 'Get client msg is:', data
udpSerSock.sendto('[%s] %s' %(ctime(),data), addr) #发送UDP
print 'Received from and returned to:',addrexceptException,e:print 'Error:',e
udpSerSock.close()#关闭服务器
■ 构建UDP客户端
#-*- coding: utf-8 -*-
from socket import *HOST= 'localhost' #主机名
PORT = 21567 #端口号 与服务器一致
BUFSIZE = 1024 #缓冲区大小1K
ADDR =(HOST,PORT)
udpCliSock=socket(AF_INET, SOCK_DGRAM)while True: #无限循环等待连接到来
try:
data= raw_input('>')if notdata:breakudpCliSock.sendto(data, ADDR)#发送数据
data,ADDR = udpCliSock.recvfrom(BUFSIZE) #接受数据
if notdata:break
print 'Server :', dataexceptException,e:print 'Error:',e
udpCliSock.close()#关闭客户端