# coding=utf-8

import argparse
import re
import itertools
import sys
import time

import pandas
from urllib3.connectionpool import xrange
import HackRequests

import numpy as np
import pandas as pd
import html5lib

class BFhttp(object):
def __init__(self, args):
self.args = args
self.payloads = []
self.payloadsQueue = HackRequests.queue.Queue()  # FIFO 队列中，先添加的任务先取回
self.hack = HackRequests.hackRequests()
self.result = {}
self.result2 = []
self.pwdlength = 0
self.number = 0

def parseRaw(self):
with open(self.args.reqfile) as f:
self.raw = f.read()

def parsePayloads(self):
for i in range(self.payloadLenght):
f = open(self.args.payloads[i], 'r')
for p in f:
self.payloads.append(p.rstrip())

def getOrgRes(self):
html = self.hack.httpraw(self.raw).text()
self.orgData = html

def nowTime(self):
return time.time()

def _callback(self, r: HackRequests.response,ywlstr):
if self.orgData == r.text():
bruteSate = "Same"
else:
bruteSate = "Different"

# payloads = "\"" + str(ywlstr) + "\""
# id = "\"" + str(self.number) + "\""
payloads = str(ywlstr)
id = str(self.number)

self.result.setdefault('id', []).append(id)
self.result.setdefault('payloads', []).append(payloads)
self.result.setdefault('bruteSate', []).append(bruteSate)
self.result.setdefault('status_code', []).append(r.status_code)
self.result.setdefault('text_len', []).append(len(r.text()))
self.number = self.number + 1

def saveresult(self):
data = self.result
ser1 = pd.DataFrame(data, index=pd.Series(range(self.number)))
pandas.set_option('display.width', 1000)  # 设置字符显示宽度
pandas.set_option('display.max_rows', None)  # 设置显示最大行
pandas.set_option('display.max_columns', None)  # 设置显示最大例

if self.args.outputmode == "csv":
if self.args.outfile:
f = open(self.args.outfile, 'w')
f.close()
ser1.to_csv(self.args.outfile, index=False)
message = self.args.outfile

else:
f = open('excel.xlsx', 'w')
f.close()
ser1.to_excel('example.csv', sheet_name='Sheet1')
message = 'example.csv'

if self.args.outputmode == "excel":
if self.args.outfile:
f = open(self.args.outfile, 'w')
f.close()
ser1.to_excel(self.args.outfile, sheet_name='Sheet1')
message = self.args.outfile

else:
f = open('excel.xlsx', 'w')
f.close()
ser1.to_excel('excel.xlsx', sheet_name='Sheet1')
message = 'excel.xlsx'

if self.args.outputmode == "html":
self.saveResult()
message = "Please use json.html+json.js+json.txt"

if self.args.outputmode:
print("=====================\n")
print("created:" + message + "\n")
print("=====================\n")

print(ser1)

def run(self):
self.payloadLenght = len(self.args.payloads)
self.parsePayloads()
self.payloads = list(itertools.combinations(self.payloads, self.payloadLenght))
self.payloads = list(set(self.payloads))  # 去重
self.parseRaw()
self.getOrgRes()

for p in xrange(len(self.payloads)):
self.payloadsQueue.put(self.payloads[p])  # 将 item 放入队列。

self.start()

def parseRawb(self, count, payloads):

keyWord = 'pybf' + str(count + 1)
try:
Content_Length = re.findall('Content-Length: (.*?)\n', self._raw)[0].rstrip()

if len(keyWord) > len(payloads[count - 1]):
sub = len(keyWord) - len(payloads[count - 1])
NewContent_Length = int(Content_Length) - sub
self._raw = self._raw.replace(re.findall('Content-Length: (.*?)\n', self._raw)[0].rstrip(),
str(NewContent_Length))

elif len(keyWord) < len(payloads[count - 1]):
sub = len(payloads[count - 1]) - len(keyWord)
NewContent_Length = int(Content_Length) + sub
self._raw = self._raw.replace(re.findall('Content-Length: (.*?)\n', self._raw)[0].rstrip(),
str(NewContent_Length))
except Exception as e:
pass
finally:
self._raw = self._raw.replace(keyWord, payloads[count - 1])

def start(self):
self._queue = self.payloadsQueue
self.lenght = self.payloadLenght
self._raw = self.raw

threadpool = HackRequests.threadpool(threadnum=self.args.thread, callback=self._callback, timeout=args.timeout)

while not self._queue.empty():  # .empty()如果队列为空，返回 True ，否则返回 False
payloads = self._queue.get_nowait()  # 相当于 get(False) ，从队列中移除并返回一个项目

for i in range(self.lenght):
self.parseRawb(i, payloads)
threadpool.httpraw(raw=self._raw,ywlstr=str(payloads))
self._raw = self.raw

threadpool.run()
self.saveresult()

def saveResult(self):
data = str(self.result)

if self.payloadLenght == 1:
data = data.replace('\'', '\"')
data = data.replace('"("', '\"')
data = data.replace('",)"', '\"')

else:
data = data.replace('\'', '\"')
data = data.replace('"(', '[')
data = data.replace(')"', ']')

f = open('data\data.txt', 'w')
f.write(data)
f.close()

if __name__ == '__main__':
print('''author:ywl
_   _
_ __  _   _| |_| |__   ___  _ __
| '_ \| | | | __| '_ \ / _ \| '_ \\
| |_) | |_| | |_| | | | (_) | | | |
| .__/ \__, |\__|_| |_|\___/|_| |_|
|_|    |___/
''')
parser = argparse.ArgumentParser(description="httpBruteForce Tool Ver:1.0")
parser.add_argument("-n", "--thread", metavar="", type=int, default=10, help="Thread Count default is 10")
parser.add_argument("-t", "--timeout", metavar="", type=int, default=2, help="timeout default is 2")
parser.add_argument("-r", "--reqfile", metavar="", help="raqFile")
parser.add_argument("-p", "--payloads", nargs="*", help="Payloads,pybfn...pybf2,pybf1")
parser.add_argument("-om", "--outputmode", type=str, help="outputmode: csv,excel,html")
parser.add_argument("-of", "--outfile", type=str, default="",
help="output file address,default is current location")
args = parser.parse_args()

var xmlhttp = new XMLHttpRequest();
xmlhttp.onreadystatechange = function() {
if (this.readyState == 4 && this.status == 200) {
data = JSON.parse(this.responseText);
var sort = 1;
getdata(sort);
}
};
xmlhttp.open("GET", "data.txt", true);
xmlhttp.send();

function getdata(sort,size=20){
var ids = data.id;
var payloads = data.payloads;
var bruteSates = data.bruteSate;
var status_codes = data.status_code;
var text_lens = data.text_len;
var tbody = '';

if(sort == 5){
for (var i=0; i<text_lens.length-1;i++) {
for (var j=0; j<text_lens.length-1-i;j++) {
if(text_lens[j] < text_lens[j + 1]){
var temp = text_lens[j];
text_lens[j] = text_lens[j+1];
text_lens[j+1] = temp;

var temp = ids[j];
ids[j] = ids[j+1];
ids[j+1] = temp;

var temp = payloads[j];
payloads[j] = payloads[j+1];
payloads[j+1] = temp;

var temp = bruteSates[j];
bruteSates[j] = bruteSates[j+1];
bruteSates[j+1] = temp;

flag +=1;
}
}
}else if(sort == 3){
var flag = 0;
for (var i=0; i<text_lens.length;i++) {
if(bruteSates[i] == "Different"){
var temp2 = bruteSates[i];
bruteSates[i] = bruteSates[flag];
bruteSates[flag] = temp2;

var temp2 = ids[i];
ids[i] = ids[flag];
ids[flag] = temp2;

var temp2 = payloads[i];
payloads[i] = payloads[flag];
payloads[flag] = temp2;

var temp2 = status_codes[i];
status_codes[i] = status_codes[flag];
status_codes[flag] = temp2;

var temp2 = text_lens[i];
text_lens[i] = text_lens[flag];
text_lens[flag] = temp2;

flag +=1;
}
}
}

for (i in data.id) {
var id = ids[i];
var payload = payloads[i];
var bruteSate = bruteSates[i];
var status_code = status_codes[i];
var text_len = text_lens[i];
tbody += '<tr class="active">'+
'<td><p align="center"><span style="font-size:'+size+'px" class="badge">'+id+'</span></p></td>' +
'<td><p align="center"><span style="font-size:'+size+'px" class="badge">'+payload+'</span></p></td>' +
'<td><p align="center"><span style="font-size:'+size+'px" class="badge">'+bruteSate+'</span></p></td>' +
'<td><p align="center"><span style="font-size:'+size+'px" class="badge">'+status_code+'</span></p></td>' +
'<td><p align="center"><span style="font-size:'+size+'px" class="badge">'+text_len+'</span></p></td>' +
'</tr>';
}

var html = document.getElementById("tbody");
html.innerHTML = tbody;
}

$(document).ready(function(){
$("#bt1").click(function(){
$("button").attr("class","glyphicon glyphicon-chevron-down btn-xs");
$("#bt1").attr("class","glyphicon glyphicon-chevron-down btn-xs btn btn-default active");
history.go(0);
});

$("#bt3").click(function(){
$("button").attr("class","glyphicon glyphicon-chevron-down btn-xs");
$("#bt3").attr("class","glyphicon glyphicon-chevron-down btn-xs btn btn-default active");
getdata(3);
});

$("#bt4").click(function(){
$("button").attr("class","glyphicon glyphicon-chevron-down btn-xs");
$("#bt4").attr("class","glyphicon glyphicon-chevron-down btn-xs btn btn-default active");
getdata(4);
});

$("#bt5").click(function(){
$("button").attr("class","glyphicon glyphicon-chevron-down btn-xs");
$("#bt5").attr("class","glyphicon glyphicon-chevron-down btn-xs btn btn-default active");
getdata(5);
});
});