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