mirror of
https://github.com/xvzc/SpoofDPI.git
synced 2024-12-22 14:26:31 +00:00
129 lines
2.3 KiB
Go
129 lines
2.3 KiB
Go
package packet
|
|
|
|
import (
|
|
"strings"
|
|
)
|
|
|
|
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": {},
|
|
}
|
|
|
|
type HttpPacket struct {
|
|
raw []byte
|
|
method string
|
|
domain string
|
|
version string
|
|
}
|
|
|
|
func NewHttpPacket(raw []byte) HttpPacket {
|
|
method, domain, version := parse(raw)
|
|
|
|
return HttpPacket{
|
|
raw: raw,
|
|
method: method,
|
|
domain: domain,
|
|
version: version,
|
|
}
|
|
}
|
|
|
|
func (p *HttpPacket) Raw() []byte {
|
|
return p.raw
|
|
}
|
|
func (p *HttpPacket) Method() string {
|
|
return p.method
|
|
}
|
|
func (p *HttpPacket) Domain() string {
|
|
return p.domain
|
|
}
|
|
func (p *HttpPacket) Version() string {
|
|
return p.version
|
|
}
|
|
|
|
func (p *HttpPacket) IsValidMethod() bool {
|
|
if _, exists := validMethod[p.Method()]; exists {
|
|
return true
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
func (p *HttpPacket) IsConnectMethod() bool {
|
|
return p.Method() == "CONNECT"
|
|
}
|
|
|
|
func (p *HttpPacket) RemoveProxyHeader() {
|
|
s := string(p.raw)
|
|
|
|
lines := strings.Split(s, "\n")
|
|
for i := 0; i < len(lines); i++ {
|
|
if strings.HasPrefix(lines[i], "Proxy-Connection") {
|
|
lines[i] = ""
|
|
}
|
|
}
|
|
|
|
result := ""
|
|
for i := 0; i < len(lines); i++ {
|
|
if lines[i] == "" {
|
|
continue
|
|
}
|
|
|
|
result += lines[i] + "\n"
|
|
}
|
|
|
|
p.raw = []byte(result)
|
|
}
|
|
|
|
func parse(raw []byte) (string, string, string) {
|
|
var firstLine string
|
|
for i := 0; i < len(raw); i++ {
|
|
if (raw)[i] == '\n' {
|
|
firstLine = string((raw)[:i])
|
|
}
|
|
}
|
|
|
|
tokens := strings.Split(firstLine, " ")
|
|
|
|
method := strings.TrimSpace(tokens[0])
|
|
domain := strings.TrimSpace(tokens[1])
|
|
version := strings.TrimSpace(tokens[2])
|
|
|
|
domain = strings.Replace(domain, "http://", "", 1)
|
|
domain = strings.Replace(domain, "https://", "", 1)
|
|
domain = strings.Split(domain, ":")[0]
|
|
domain = strings.Split(domain, "/")[0]
|
|
|
|
return method, domain, version
|
|
}
|