Python程序设计——安全渗透测试的常见模块

一、Socket 模块

  Socket模块的主要目的是帮助在网络上的两个程序之间建立信息通道。
  Python中两个基本的Socket模块：

   - 服务端Socket
   - 客户端Socket

1.实例化 Socket 类

   格式为：

   - socket(family,type[,protocal])
实例化的三个参数：

   1. 地址族（family）

   常用的协议族有AF_INET、AF_INET6、AF_LOCAL（或称AF_UNIX、UNIX域Socket）、AF_ROUTE等。

   默认值为socket.AF_INET，通常使用这个默认值即可。

   2. 流（type）

   SOCK_STREAM，这是TCP类型，保证数据顺序及可靠性；（默认值）

   SOCK_DGRAM，用于UDP类型

   SOCK_RAW，这是原始类型，允许对底层协议如IP或ICMP进行直接访问，基本不会用到

   3. 使用的协议

   参数是可选的。通常赋值“0”，由系统自动选择

   TCP类型的Socket格式为：

   - s = socket.socket() ——> 相当于
socket.socket(socket.AF_INET,socket.SOCK_STREAM)。

   UDP类型的Socket格式为：

   - s=socket.socket(socket.AF_INET,socket.SOCK_DGRAM)

2.Socket常用的函数—服务端函数

bind()：

   这个函数由服务端Socket调用，会将之前创建Socket与指定的IP地址和端口进行绑定。如果之前使用了AF_INET初始化Socket，那么这里可以使用元组（host，port）的形式表示地址。

• s.bind((‘127.0.0.1’,2345))
  

listen()：

   这个函数用于在使用TCP的服务端 开启 监听模式。可以使用一个参数来指定可以挂起的最大连接数量。

• s.listen(5)
  

accept()：

   这个函数用于在使用TCP的 服务端 接收连接，一般是阻塞态。接受TCP连接并返回（conn，address）

   其中，conn是新的套接字对象，可以用来接收和发送数据；address是连接客户端的地址。

3.客户端的函数

send()：

   用于在使用 TCP 时发送数据，利用这个函数可以将string代表的数据发送到已经连接的Socket，返回值是发送字节的数量。
但是可能未将指定的内容全部发送。

• send(string[,flag])
  

sendall()：

   与send()相类似，也是用于在使用TCP时发送数据，与send()的区别是完整发送TCP数据。将string中的数据发送到连接的套接字，但在返回之前会尝试发送所有数据。成功返回None，失败则抛出异常。

• sendall(string[,flag])
  

recv()：

   这个函数由服务端Socket调用，会将之前创建Socket与指定的IP地址和端口进行绑定。如果之前使用了AF_INET初始化Socket，那么这里可以使用元组（host，port）的形式表示地址。

• recv(bufsize[,flag])
  
• bind()：

   这个函数TCP时接收数据，接收Socket的数据。数据以字符串形式返回，bufsize指定最多可以接收的数量，flag这个参数一般不会使用。

• s.bind((‘127.0.0.1’,2345))
  

sendto()：

   这个函数用于在使用UDP时发送数据，返回值是发送的字节数。address是形式为（ipaddr，port）的元组，指定远程地址。

• sendto(string[,flag],address)
  

recvfrom()：

   UDP专用，接收数据，返回数据远端的IP地址和端口，但返回值是（data,address）。其中，data是包含接收数据的字符串，address是发送数据的套接字地址。

• close()：
     关闭socket。

4.使用Socket编写一个简单的服务端和客户端

   可以将程序保存为server.py，然后在编辑器中执行。但是程序简单，也可以直接在Python的命令行中执行。

import socket     
s1 = socket.socket()     
s1.bind(("127.0.0.1",2345))     
s1.listen(5)     
while 1：         
	conn,address = s1.accept()        
	print "a new connect from", address         
	conn.sendall("Hello world")         
	conn.close()

使用connect()函数来连接到目标服务端：

import socket
s2 = socket.socket()
s2.connect(("127.0.0.1",2345))
data = s2.recv(1024)
s2.close()
print 'Received', repr(data)

二、python-nmap模块

Nmap的功能：

1. 主机发现功能
2. 端口扫描
3. 服务及版本检测
4. 操作系统检测

1.python-nmap

为Nmap功能的 Python 模块文件。

   注意：在Python中正常使用python-nmap模块，必须先在系统中安装Nmap；Windows操作系统下直接下载安装即可。

安装过程

1. Linux操作系统下的命令：
   sudo apt-get install nmap
   

2. 安装python-nmap
   sudo pip install python-nmap
   

3. 安装成功后，打开终端，启动Python，然后导入Nmap

python-nmap模块的核心：

   PortScanner、PortScannerAsync、PortScannerError、PortScannerHostDict、PortScannerYield等5个类，其中最为重要的是PortScanner类。

2.python-nmap模块类的实例化

PortScanner类

nmap.PortScanner()

PortScannerAsync类

PortScannerAsync类和PortScanner类的功能相似，可以实现异步扫描
nmap.PortScannerAsync()