mirrors/Xray-core
1
0
Fork 0
mirror of https://github.com/XTLS/Xray-core.git synced 2024-12-23 22:19:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add separate host config for websocket

Browse Source
This commit is contained in:
yuhan6665 2024-03-29 02:27:02 -04:00 committed by RPRX
parent e2302b421c
commit 7e3a8d3a04
3 changed files with 19 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
infra/conf/transport_internet.go
View File

@ -146,6 +146,7 @@ func (c *TCPConfig) Build() (proto.Message, error) {
}
type WebSocketConfig struct {
Host string `json:"host"`
Path string `json:"path"`
Headers map[string]string `json:"headers"`
AcceptProxyProtocol bool `json:"acceptProxyProtocol"`
@ -154,10 +155,6 @@ type WebSocketConfig struct {
// Build implements Buildable.
func (c *WebSocketConfig) Build() (proto.Message, error) {
path := c.Path
header := make(map[string]string);
for key, value := range c.Headers {
header[key] = value;
}
var ed uint32
if u, err := url.Parse(path); err == nil {
if q := u.Query(); q.Get("ed") != "" {
@ -168,9 +165,18 @@ func (c *WebSocketConfig) Build() (proto.Message, error) {
path = u.String()
}
}
// If http host is not set in the Host field, but in headers field, we add it to Host Field here.
// If we don't do that, http host will be overwritten as address.
// Host priority: Host field > headers field > address.
if c.Host == "" && c.Headers["host"] != "" {
c.Host = c.Headers["host"]
} else if c.Host == "" && c.Headers["Host"] != "" {
c.Host = c.Headers["Host"]
}
config := &websocket.Config{
Path: path,
Header: header,
Host: c.Host,
Header: c.Headers,
AcceptProxyProtocol: c.AcceptProxyProtocol,
Ed: ed,
}
@ -178,8 +184,8 @@ func (c *WebSocketConfig) Build() (proto.Message, error) {
}
type HttpUpgradeConfig struct {
Path string `json:"path"`
Host string `json:"host"`
Path string `json:"path"`
Headers map[string]string `json:"headers"`
AcceptProxyProtocol bool `json:"acceptProxyProtocol"`
}

1
transport/internet/websocket/config.go
View File

@ -25,6 +25,7 @@ func (c *Config) GetRequestHeader() http.Header {
for k, v := range c.Header {
header.Add(k, v)
}
header.Set("Host", c.Host)
return header
}

6
transport/internet/websocket/hub.go
View File

@ -21,6 +21,7 @@ import (
)
type requestHandler struct {
host string
path string
ln *Listener
}
@ -37,6 +38,10 @@ var upgrader = &websocket.Upgrader{
}
func (h *requestHandler) ServeHTTP(writer http.ResponseWriter, request *http.Request) {
if len(h.host) > 0 && request.Host != h.host {
writer.WriteHeader(http.StatusNotFound)
return
}
if request.URL.Path != h.path {
writer.WriteHeader(http.StatusNotFound)
return
@ -125,6 +130,7 @@ func ListenWS(ctx context.Context, address net.Address, port net.Port, streamSet
l.server = http.Server{
Handler: &requestHandler{
host: wsSettings.Host,
path: wsSettings.GetNormalizedPath(),
ln: l,
},