【Socket】
本文记录了一些socket模块的简单应用，对于具体原理还没来得及深究。
■ 利用socket模块进行端口连接验证和扫描
在linux中常用nc命令来进行远端端口是否开放的验证。但是这个命令并不是系统自带的，所以还比较麻烦。如果利用python自带的socket模块就可以比较*地进行端口验证了。是这样用的：
import socket def port_check(host,port):
sk = socket.socket(socket.AF_INET,socket.SOCK_STREAM) #建立socket对象
sk.settimeout(1) #设置超时，否则可能等很久才会返回连不通端口的信息
try:
sk.connect((host,port)) #进行socket层面的连接，注意参数是一个tuple
except socket.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 import socket 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)) while True:
data = raw_input('> ')
client.send(data)
data = client.recv(4096)
if not data:
break
print data
构造出来的是一个带有字符提示的简单TCP报文发送器。代码很简单，不多说了。
■ 构建tcp服务端
#! /usr/bin/env python
#coding=utf-8 import socket
bind_ip = ""
bind_port = 9999 server = socket.socket(socket.AF_INET,socket.SOCK_STREAM) server.bind((bind_ip,bind_port)) server.listen(5)
try:
while True:
client,add = server.accept()
print "[*]你监听的是：%s:%d" % (add[0],add[1])
while True:
data = client.recv(1024)
if not data:
break
print data
data = raw_input('> ')
client.send(data)
else:
client.close()
except Exception as e:
print e
server.close()
■ 构建UDP服务端
# -*- coding: utf-8 -*-
from socket import *
from time import ctime 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: ',addr except Exception,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 not data:
break
udpCliSock.sendto(data, ADDR) #发送数据
data,ADDR = udpCliSock.recvfrom(BUFSIZE) #接受数据
if not data:
break
print 'Server : ', data except Exception,e:
print 'Error: ',e udpCliSock.close() #关闭客户端