python自带模块socket_【Python】socket模块应用

【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()#关闭客户端