-
Notifications
You must be signed in to change notification settings - Fork 0
/
header.go
100 lines (93 loc) · 2.4 KB
/
header.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
// Copyright 2013 Liviu G. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package goscgi
import (
"errors"
"net"
"net/http"
"strconv"
"time"
)
var (
InvalidHeaderErr = errors.New("Invalid header")
InvalidContentErr = errors.New("Invalid content size")
UnexpectedEndErr = errors.New("Unexpected end of stream")
)
// http://www.python.ca/scgi/protocol.txt
func ReadHeader(conn net.Conn, settings *Settings) (http.Header, error) {
var err error
const buffSize = 8 // first we read only 8 bytes from which we determine the headerSize
var buff [buffSize]byte
var alreadyRead, readCnt int
for alreadyRead < buffSize {
conn.SetReadDeadline(time.Now().Add(settings.ReadTimeout))
if readCnt, err = conn.Read(buff[alreadyRead:]); err != nil {
return nil, err
}
alreadyRead += readCnt
}
var idx int
var headerSize int
var headerSizeStr string
for idx = 0; idx < readCnt; idx++ {
if buff[idx] == ':' {
headerSizeStr = string(buff[:idx])
idx++ // skip ':'
break
}
}
if len(headerSizeStr) == 0 {
return nil, InvalidHeaderErr
}
if int64hs, err := strconv.ParseInt(headerSizeStr, 10, 0); err != nil {
return nil, err
} else {
headerSize = int(int64hs)
}
if headerSize <= 0 || headerSize > settings.MaxHeaderSize {
return nil, InvalidHeaderErr
}
headerSize++ // add the final ','
headerBuff := make([]byte, headerSize)
alreadyRead = readCnt - idx // alreadyRead := size of data read after ':'
if alreadyRead > 0 {
// copy alreadyRead data from the initial buffer to the headerBuff
copy(headerBuff[:alreadyRead], buff[idx:readCnt])
}
for alreadyRead < headerSize {
conn.SetReadDeadline(time.Now().Add(settings.ReadTimeout))
if readCnt, err = conn.Read(headerBuff[alreadyRead:]); err != nil {
return nil, err
}
alreadyRead += readCnt
}
header := http.Header{}
var name string
nameExpected := true
idx = 0
for {
if idx >= headerSize {
return nil, UnexpectedEndErr
}
if nameExpected && headerBuff[idx] == ',' {
break // end of header reached
}
baseIdx := idx
for idx < headerSize && headerBuff[idx] != 0 {
idx++
}
if idx >= headerSize {
return nil, UnexpectedEndErr
}
if str := string(headerBuff[baseIdx:idx]); nameExpected {
name = str
nameExpected = false
} else {
header.Add(name, str)
nameExpected = true
}
idx++ // skip 0
}
return header, nil
}