mirror of
https://github.com/xvzc/SpoofDPI.git
synced 2024-12-22 14:26:31 +00:00
95 lines
1.8 KiB
Go
95 lines
1.8 KiB
Go
package packet
|
|
|
|
import (
|
|
"strings"
|
|
)
|
|
|
|
type HttpRequest struct {
|
|
Raw *[]byte
|
|
Method string
|
|
Domain string
|
|
Version string
|
|
}
|
|
|
|
func NewHttpRequest(raw *[]byte) HttpRequest {
|
|
method, domain, version := parse(raw)
|
|
return HttpRequest{
|
|
Raw: raw,
|
|
Method: method,
|
|
Domain: domain,
|
|
Version: version,
|
|
}
|
|
}
|
|
|
|
func (r *HttpRequest) IsValidMethod() bool {
|
|
if _, exists := getValidMethods()[r.Method]; exists {
|
|
return true
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
func (r *HttpRequest) IsConnectMethod() bool {
|
|
return r.Method == "CONNECT"
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
func getValidMethods() map[string]struct{} {
|
|
return 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": {},
|
|
}
|
|
}
|