SpoofDPI/packet/http.go

171 lines
2.8 KiB
Go
Raw Normal View History

2022-01-08 06:35:32 +00:00
package packet
2022-01-07 17:41:02 +00:00
import (
"strings"
)
2022-01-10 19:27:12 +00:00
var validMethod = map[string]struct{}{
"DELETE": {},
"GET": {},
"HEAD": {},
"POST": {},
"PUT": {},
"CONNECT": {},
"OPTIONS": {},
"TRACE": {},
"COPY": {},
"LOCK": {},
"MKCOL": {},
"MOVE": {},
"PROPFIND": {},
"PROPPATCH": {},
"SEARCH": {},
"UNLOCK": {},
"BIND": {},
"REBIND": {},
"UNBIND": {},
"ACL": {},
"REPORT": {},
"MKACTIVITY": {},
"CHECKOUT": {},
"MERGE": {},
"M-SEARCH": {},
"NOTIFY": {},
"SUBSCRIBE": {},
"UNSUBSCRIBE": {},
"PATCH": {},
"PURGE": {},
"MKCALENDAR": {},
"LINK": {},
"UNLINK": {},
}
2022-01-08 15:09:01 +00:00
type HttpPacket struct {
2022-01-11 17:15:45 +00:00
raw []byte
method string
domain string
2022-03-05 02:51:17 +00:00
port string
2022-01-15 16:44:00 +00:00
path string
2022-01-11 17:15:45 +00:00
version string
2022-01-07 17:41:02 +00:00
}
2022-03-05 02:51:17 +00:00
func ParseUrl(raw []byte) {
}
2022-01-10 19:27:12 +00:00
func NewHttpPacket(raw []byte) HttpPacket {
2022-03-05 02:51:17 +00:00
method, domain, port, path, version := parse(raw)
2022-01-11 17:15:45 +00:00
2022-01-08 15:09:01 +00:00
return HttpPacket{
2022-01-11 17:15:45 +00:00
raw: raw,
method: method,
domain: domain,
2022-03-05 02:51:17 +00:00
port: port,
2022-01-15 16:44:00 +00:00
path: path,
2022-01-11 17:15:45 +00:00
version: version,
2022-01-07 18:01:54 +00:00
}
}
2022-01-11 17:15:45 +00:00
func (p *HttpPacket) Raw() []byte {
return p.raw
}
func (p *HttpPacket) Method() string {
return p.method
}
2022-01-15 16:44:00 +00:00
2022-01-11 17:15:45 +00:00
func (p *HttpPacket) Domain() string {
return p.domain
}
2022-01-15 16:44:00 +00:00
2022-03-05 02:51:17 +00:00
func (p *HttpPacket) Port() string {
return p.port
}
2022-01-11 17:15:45 +00:00
func (p *HttpPacket) Version() string {
return p.version
}
func (p *HttpPacket) IsValidMethod() bool {
if _, exists := validMethod[p.Method()]; exists {
2022-01-07 17:41:02 +00:00
return true
}
return false
}
2022-01-11 17:15:45 +00:00
func (p *HttpPacket) IsConnectMethod() bool {
return p.Method() == "CONNECT"
2022-01-07 17:41:02 +00:00
}
2022-01-15 16:44:00 +00:00
func (p *HttpPacket) Tidy() {
2022-01-14 21:40:57 +00:00
s := string(p.raw)
2022-03-04 23:49:28 +00:00
lines := strings.Split(s, "\r\n")
2022-01-15 16:44:00 +00:00
lines[0] = p.method + " " + p.path + " " + p.version
2022-01-14 21:40:57 +00:00
for i := 0; i < len(lines); i++ {
if strings.HasPrefix(lines[i], "Proxy-Connection") {
lines[i] = ""
}
}
result := ""
2022-03-04 23:49:28 +00:00
2022-01-14 21:40:57 +00:00
for i := 0; i < len(lines); i++ {
if lines[i] == "" {
continue
}
2022-03-04 23:49:28 +00:00
result += lines[i] + "\r\n"
2022-01-14 21:40:57 +00:00
}
2022-03-04 23:49:28 +00:00
result += "\r\n"
2022-01-14 21:40:57 +00:00
p.raw = []byte(result)
}
2022-03-05 02:51:17 +00:00
func parse(raw []byte) (string, string, string, string, string) {
2022-01-07 18:34:26 +00:00
var firstLine string
2022-01-10 19:27:12 +00:00
for i := 0; i < len(raw); i++ {
if (raw)[i] == '\n' {
firstLine = string((raw)[:i])
2022-01-15 16:44:00 +00:00
break
2022-01-07 17:41:02 +00:00
}
}
2022-01-07 18:34:26 +00:00
tokens := strings.Split(firstLine, " ")
method := strings.TrimSpace(tokens[0])
2022-01-15 16:44:00 +00:00
url := strings.TrimSpace(tokens[1])
2022-01-07 18:34:26 +00:00
version := strings.TrimSpace(tokens[2])
2022-01-07 17:41:02 +00:00
2022-01-15 16:44:00 +00:00
url = strings.Replace(url, "http://", "", 1)
url = strings.Replace(url, "https://", "", 1)
2022-03-05 02:51:17 +00:00
var domain string
var port string
2022-01-15 16:44:00 +00:00
for i := 0; i < len(url); i++ {
if url[i] == ':' {
domain = url[:i]
2022-03-05 02:51:17 +00:00
port = url[i:]
2022-01-15 16:44:00 +00:00
break
}
if url[i] == '/' {
domain = url[:i]
break
}
}
path := "/"
for i := 0; i < len(url); i++ {
if url[i] == '/' {
path = url[i:]
break
}
}
2022-01-07 17:41:02 +00:00
2022-03-05 02:51:17 +00:00
return method, domain, port, path, version
2022-01-07 17:41:02 +00:00
}