【Python】socket模块应用

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