Alawn/v2ray-core
1
0
Fork 0
mirror of https://github.com/v2fly/v2ray-core synced 2024-02-27 03:18:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

limit max http header length

Browse Source
This commit is contained in:
Darien Raymond
2016-12-16 17:01:47 +01:00
parent b17f06bb4e
commit 68164d46f9
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
transport/internet/headers/http/http.go
+6 -2
View File
@@ -16,6 +16,8 @@ import (
const (
CRLF = "\r\n"
ENDING = CRLF + CRLF
maxHeaderLength = 8192
)
var (
@@ -47,7 +49,8 @@ type HeaderReader struct {
func (*HeaderReader) Read(reader io.Reader) (*buf.Buffer, error) {
buffer := buf.NewSmall()
for {
totalBytes := 0
for totalBytes < maxHeaderLength {
err := buffer.AppendSupplier(buf.ReadFrom(reader))
if err != nil {
return nil, err
@@ -57,7 +60,8 @@ func (*HeaderReader) Read(reader io.Reader) (*buf.Buffer, error) {
break
}
if buffer.Len() >= len(ENDING) {
leftover := buffer.BytesFrom(buffer.Len() - len(ENDING))
totalBytes += buffer.Len() - len(ENDING)
leftover := buffer.BytesFrom(-len(ENDING))
buffer.Reset(func(b []byte) (int, error) {
return copy(b, leftover), nil
})