Merge pull request #22 from xvzc/url

fix url parsing
This commit is contained in:
xvzc 2022-05-08 20:39:51 +09:00 committed by GitHub
commit 20e2065109
2 changed files with 26 additions and 25 deletions

View File

@ -91,7 +91,7 @@ func (conn *Conn) ReadBytes() ([]byte, error) {
return ret, nil
}
func (lConn *Conn) HandleHttp(p packet.HttpPacket) {
func (lConn *Conn) HandleHttp(p *packet.HttpPacket) {
defer lConn.Close()
p.Tidy()
@ -138,7 +138,7 @@ func (lConn *Conn) HandleHttp(p packet.HttpPacket) {
}
func (lConn *Conn) HandleHttps(p packet.HttpPacket) {
func (lConn *Conn) HandleHttps(p *packet.HttpPacket) {
defer lConn.Close()
ip, err := doh.Lookup(p.Domain())

View File

@ -54,20 +54,12 @@ func ParseUrl(raw []byte) {
}
func NewHttpPacket(raw []byte) (HttpPacket, error){
method, domain, port, path, version, err := parse(raw)
if err != nil {
return HttpPacket{}, err
}
func NewHttpPacket(raw []byte) (*HttpPacket, error){
pkt := &HttpPacket{raw: raw}
return HttpPacket{
raw: raw,
method: method,
domain: domain,
port: port,
path: path,
version: version,
}, nil
pkt.parse()
return pkt, nil
}
func (p *HttpPacket) Raw() []byte {
@ -129,11 +121,11 @@ func (p *HttpPacket) Tidy() {
p.raw = []byte(result)
}
func parse(raw []byte) (string, string, string, string, string, error) {
func (p *HttpPacket )parse() error {
var firstLine string
for i := 0; i < len(raw); i++ {
if (raw)[i] == '\r' {
firstLine = string((raw)[:i])
for i := 0; i < len(p.raw); i++ {
if (p.raw)[i] == '\r' {
firstLine = string(p.raw[:i])
break
}
}
@ -141,15 +133,20 @@ func parse(raw []byte) (string, string, string, string, string, error) {
tokens := strings.Split(firstLine, " ")
if (len(tokens) < 3) {
return "", "", "", "", "", errors.New("Unexpected request format")
return errors.New("Error parsing http request")
}
method := tokens[0]
p.method = tokens[0]
url := tokens[1]
version := tokens[2]
p.version = tokens[2]
url = strings.Replace(url, "http://", "", 1)
url = strings.Replace(url, "https://", "", 1)
if strings.HasPrefix(url, "http://") {
url = strings.Replace(url, "http://", "", 1)
}
if strings.HasPrefix(url, "https://") {
url = strings.Replace(url, "https://", "", 1)
}
domain := ""
port := ""
@ -165,6 +162,8 @@ func parse(raw []byte) (string, string, string, string, string, error) {
break
}
}
p.domain = domain
p.port = port
path := "/"
for i := 0; i < len(url); i++ {
@ -174,5 +173,7 @@ func parse(raw []byte) (string, string, string, string, string, error) {
}
}
return method, domain, port, path, version, nil
p.path = path
return nil
}