lxy/redkale
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

HttpRequest

Browse Source
This commit is contained in:
redkale
2024-09-13 12:33:15 +08:00
parent 0c1f605dbb
commit 334c4920d9
1 changed files with 19 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
src/main/java/org/redkale/net/http/HttpRequest.java
View File

@@ -242,6 +242,7 @@ public class HttpRequest extends Request<HttpContext> {
if (req.getHeaders() != null) { if (req.getHeaders() != null) {
this.headers.setAll(req.getHeaders()); this.headers.setAll(req.getHeaders());
} }
this.headerParsed = true;
this.reqConvertType = req.getReqConvertType(); this.reqConvertType = req.getReqConvertType();
this.reqConvert = this.reqConvert =
req.getReqConvertType() == null ? null : ConvertFactory.findConvert(req.getReqConvertType()); req.getReqConvertType() == null ? null : ConvertFactory.findConvert(req.getReqConvertType());
@@ -456,8 +457,7 @@ public class HttpRequest extends Request<HttpContext> {
final ByteBuffer buffer = buf; final ByteBuffer buffer = buf;
int remain = buffer.remaining(); int remain = buffer.remaining();
if (this.chunkedLength < 0) { // 需要读取length if (this.chunkedLength < 0) { // 需要读取length
ByteArray input = array(); ByteArray input = array().clear();
input.clear();
if (this.chunkedHalfLenBytes != null) { if (this.chunkedHalfLenBytes != null) {
input.put(this.chunkedHalfLenBytes); input.put(this.chunkedHalfLenBytes);
this.chunkedHalfLenBytes = null; this.chunkedHalfLenBytes = null;
@@ -651,7 +651,7 @@ public class HttpRequest extends Request<HttpContext> {
ByteArray bytes = bodyBytes; // body当temp buffer使用 ByteArray bytes = bodyBytes; // body当temp buffer使用
// 读method // 读method
if (this.method == null) { if (this.method == null) {
boolean flag = false; boolean bigger = false;
if (remain >= 5) { if (remain >= 5) {
byte b1 = buffer.get(); byte b1 = buffer.get();
byte b2 = buffer.get(); byte b2 = buffer.get();
@@ -694,14 +694,14 @@ public class HttpRequest extends Request<HttpContext> {
this.getmethod = false; this.getmethod = false;
} }
} else { } else {
flag = true; bigger = true;
bytes.put(b1, b2, b3, b4, b5); bytes.put(b1, b2, b3, b4, b5);
} }
} }
} }
} }
} }
if (flag) { if (bigger) { // method长度大于4
for (; ; ) { for (; ; ) {
if (remain-- < 1) { if (remain-- < 1) {
buffer.clear(); buffer.clear();
@@ -727,7 +727,6 @@ public class HttpRequest extends Request<HttpContext> {
this.getmethod = false; this.getmethod = false;
} }
} else if (size == 4) { } else if (size == 4) {
this.getmethod = false;
if (content[0] == 'P' && content[1] == 'O' && content[2] == 'S' && content[3] == 'T') { if (content[0] == 'P' && content[1] == 'O' && content[2] == 'S' && content[3] == 'T') {
this.method = METHOD_POST; this.method = METHOD_POST;
} else if (content[0] == 'H' && content[1] == 'E' && content[2] == 'A' && content[3] == 'D') { } else if (content[0] == 'H' && content[1] == 'E' && content[2] == 'A' && content[3] == 'D') {
@@ -735,19 +734,17 @@ public class HttpRequest extends Request<HttpContext> {
} else { } else {
this.method = bytes.toString(true, charset); this.method = bytes.toString(true, charset);
} }
} else if (size == 7) {
this.getmethod = false; this.getmethod = false;
if (content[0] == 'O' } else if (size == 7
&& content[1] == 'P' && content[0] == 'O'
&& content[2] == 'T' && content[1] == 'P'
&& content[3] == 'I' && content[2] == 'T'
&& content[4] == 'O' && content[3] == 'I'
&& content[5] == 'N' && content[4] == 'O'
&& content[6] == 'S') { && content[5] == 'N'
this.method = METHOD_OPTIONS; && content[6] == 'S') {
} else { this.method = METHOD_OPTIONS;
this.method = bytes.toString(true, charset); this.getmethod = false;
}
} else { } else {
this.method = bytes.toString(true, charset); this.method = bytes.toString(true, charset);
this.getmethod = false; this.getmethod = false;
@@ -779,6 +776,7 @@ public class HttpRequest extends Request<HttpContext> {
} }
bytes.put(b); bytes.put(b);
} }
size = bytes.length(); size = bytes.length();
if (qst > 0) { // 带?参数 if (qst > 0) { // 带?参数
this.requestPath = decodeable this.requestPath = decodeable
@@ -1092,13 +1090,7 @@ public class HttpRequest extends Request<HttpContext> {
return; return;
} }
headerParsed = true; headerParsed = true;
if (headerBytes == null) { if (headerBytes.length() > 0) {
return;
}
if (bodyBytes.isEmpty()) { // body当temp buffer使用
readHeaderLines(this.headerBytes.wrapByteBuffer(), bodyBytes);
bodyBytes.clear();
} else { // 存有body数据
readHeaderLines(this.headerBytes.wrapByteBuffer(), array().clear()); readHeaderLines(this.headerBytes.wrapByteBuffer(), array().clear());
} }
} }
@@ -1382,7 +1374,7 @@ public class HttpRequest extends Request<HttpContext> {
ByteArrayInputStream in = new ByteArrayInputStream(bodyBytes.content(), 0, bodyBytes.length()); ByteArrayInputStream in = new ByteArrayInputStream(bodyBytes.content(), 0, bodyBytes.length());
GZIPInputStream ungzip = new GZIPInputStream(in); GZIPInputStream ungzip = new GZIPInputStream(in);
int n; int n;
byte[] buffer = array().content(); byte[] buffer = array().clear().content();
while ((n = ungzip.read(buffer)) > 0) { while ((n = ungzip.read(buffer)) > 0) {
out.write(buffer, 0, n); out.write(buffer, 0, n);
} }
@@ -1393,7 +1385,7 @@ public class HttpRequest extends Request<HttpContext> {
infl.setInput(bodyBytes.content(), 0, bodyBytes.length()); infl.setInput(bodyBytes.content(), 0, bodyBytes.length());
ByteArrayOutputStream out = new ByteArrayOutputStream(); ByteArrayOutputStream out = new ByteArrayOutputStream();
int n; int n;
byte[] buffer = array().content(); byte[] buffer = array().clear().content();
while (!infl.finished()) { while (!infl.finished()) {
n = infl.inflate(buffer); n = infl.inflate(buffer);
if (n == 0) { if (n == 0) {