TheKingOfDuck/burpFakeIP
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix bug

Browse Source
This commit is contained in:
鸭王
2021-09-24 20:24:39 +08:00
committed by GitHub
parent e0f6cc657c
commit a294abf590
1 changed files with 13 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
src/main/java/burp/Utils.java
View File

@@ -37,7 +37,7 @@ public class Utils {
} }
//更新header //更新header
byte[] newMessage = BurpExtender.helpers.buildHttpMessage(headers, getHttpRequestBody(currentRequest).getBytes()); byte[] newMessage = BurpExtender.helpers.buildHttpMessage(headers, getHttpRequestBody(currentRequest));
currentRequest.setRequest(newMessage); currentRequest.setRequest(newMessage);
} }
@@ -45,38 +45,29 @@ public class Utils {
public static void addfakeip(IHttpRequestResponse iHttpRequestResponse, String ip) { public static void addfakeip(IHttpRequestResponse iHttpRequestResponse, String ip) {
//获取原请求信息
IRequestInfo requestInfo = BurpExtender.helpers.analyzeRequest(iHttpRequestResponse);
List<String> headers = requestInfo.getHeaders();
byte contentType = BurpExtender.helpers.analyzeRequest(iHttpRequestResponse).getContentType(); //为每个请求添加一个Header
headers = headers.stream().filter(key -> !key.equals(Config.AUTOXFF_KEY)).collect(Collectors.toList());
headers.add(String.format("%s: %s", Config.AUTOXFF_KEY, ip));
if (contentType != CONTENT_TYPE_MULTIPART) { //更新header
//获取原请求信息 byte[] newMessage = BurpExtender.helpers.buildHttpMessage(headers, getHttpRequestBody(iHttpRequestResponse));
IRequestInfo requestInfo = BurpExtender.helpers.analyzeRequest(iHttpRequestResponse); iHttpRequestResponse.setRequest(newMessage);
List<String> headers = requestInfo.getHeaders();
//为每个请求添加一个Header
headers = headers.stream().filter(key -> !key.equals(Config.AUTOXFF_KEY)).collect(Collectors.toList());
headers.add(String.format("%s: %s", Config.AUTOXFF_KEY, ip));
//更新header
byte[] newMessage = BurpExtender.helpers.buildHttpMessage(headers, getHttpRequestBody(iHttpRequestResponse).getBytes());
iHttpRequestResponse.setRequest(newMessage);
}
} }
private static String getHttpRequestBody(IHttpRequestResponse httpRequestResponse) { private static byte[] getHttpRequestBody(IHttpRequestResponse httpRequestResponse) {
byte[] request = httpRequestResponse.getRequest(); byte[] request = httpRequestResponse.getRequest();
IRequestInfo requestInfo = BurpExtender.helpers.analyzeRequest(request); IRequestInfo requestInfo = BurpExtender.helpers.analyzeRequest(request);
int httpBodyOffset = requestInfo.getBodyOffset(); int httpBodyOffset = requestInfo.getBodyOffset();
int httpBodyLength = request.length - httpBodyOffset; int httpBodyLength = request.length - httpBodyOffset;
String httpBody = null; byte[] httpBody = new byte[httpBodyLength];
try { System.arraycopy(request,httpBodyOffset,httpBody,0,httpBodyLength);
httpBody = new String(request, httpBodyOffset, httpBodyLength, "UTF-8");
} catch (UnsupportedEncodingException e) {
throw new RuntimeException(e);
}
return httpBody; return httpBody;
} }