mirror of https://github.com/v2ray/v2ray-core
101 lines
1.9 KiB
Go
101 lines
1.9 KiB
Go
package http
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"v2ray.com/core/common/dice"
|
|
)
|
|
|
|
func pickString(arr []string) string {
|
|
n := len(arr)
|
|
switch n {
|
|
case 0:
|
|
return ""
|
|
case 1:
|
|
return arr[0]
|
|
default:
|
|
return arr[dice.Roll(n)]
|
|
}
|
|
}
|
|
|
|
func (v *RequestConfig) PickUri() string {
|
|
return pickString(v.Uri)
|
|
}
|
|
|
|
func (v *RequestConfig) PickHeaders() []string {
|
|
n := len(v.Header)
|
|
if n == 0 {
|
|
return nil
|
|
}
|
|
headers := make([]string, n)
|
|
for idx, headerConfig := range v.Header {
|
|
headerName := headerConfig.Name
|
|
headerValue := pickString(headerConfig.Value)
|
|
headers[idx] = headerName + ": " + headerValue
|
|
}
|
|
return headers
|
|
}
|
|
|
|
func (v *RequestConfig) GetVersionValue() string {
|
|
if v == nil || v.Version == nil {
|
|
return "1.1"
|
|
}
|
|
return v.Version.Value
|
|
}
|
|
|
|
func (v *RequestConfig) GetMethodValue() string {
|
|
if v == nil || v.Method == nil {
|
|
return "GET"
|
|
}
|
|
return v.Method.Value
|
|
}
|
|
|
|
func (v *RequestConfig) GetFullVersion() string {
|
|
return "HTTP/" + v.GetVersionValue()
|
|
}
|
|
|
|
func (v *ResponseConfig) HasHeader(header string) bool {
|
|
cHeader := strings.ToLower(header)
|
|
for _, tHeader := range v.Header {
|
|
if strings.ToLower(tHeader.Name) == cHeader {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (v *ResponseConfig) PickHeaders() []string {
|
|
n := len(v.Header)
|
|
if n == 0 {
|
|
return nil
|
|
}
|
|
headers := make([]string, n)
|
|
for idx, headerConfig := range v.Header {
|
|
headerName := headerConfig.Name
|
|
headerValue := pickString(headerConfig.Value)
|
|
headers[idx] = headerName + ": " + headerValue
|
|
}
|
|
return headers
|
|
}
|
|
|
|
func (v *ResponseConfig) GetVersionValue() string {
|
|
if v == nil || v.Version == nil {
|
|
return "1.1"
|
|
}
|
|
return v.Version.Value
|
|
}
|
|
|
|
func (v *ResponseConfig) GetFullVersion() string {
|
|
return "HTTP/" + v.GetVersionValue()
|
|
}
|
|
|
|
func (v *ResponseConfig) GetStatusValue() *Status {
|
|
if v == nil || v.Status == nil {
|
|
return &Status{
|
|
Code: "200",
|
|
Reason: "OK",
|
|
}
|
|
}
|
|
return v.Status
|
|
}
|