root/gost_software
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add reverse proxy option for ws, mws, wss and mwss.

Browse Source
This commit is contained in:
chor 2020-04-25 23:27:28 +08:00
parent 6e46ac03c7
commit b34f9839f1
2 changed files with 39 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cmd/gost/route.go
View File

@ -134,6 +134,7 @@ func parseChainNode(ns string) (nodes []gost.Node, err error) {
wsOpts.WriteBufferSize = node.GetInt("wbuf")
wsOpts.UserAgent = node.Get("agent")
wsOpts.Path = node.Get("path")
wsOpts.Reverse = node.Get("reverse")
timeout := node.GetDuration("timeout")
@ -352,6 +353,7 @@ func (r *route) GenRouters() ([]router, error) {
wsOpts.ReadBufferSize = node.GetInt("rbuf")
wsOpts.WriteBufferSize = node.GetInt("wbuf")
wsOpts.Path = node.Get("path")
wsOpts.Reverse = node.Get("reverse")
ttl := node.GetDuration("ttl")
timeout := node.GetDuration("timeout")

37
ws.go
View File

@ -9,6 +9,7 @@ import (
"net"
"net/http"
"net/http/httputil"
"strings"
"sync"
"time"
@ -31,6 +32,7 @@ type WSOptions struct {
EnableCompression bool
UserAgent string
Path string
Reverse string
}
type wsTransporter struct {
@ -66,6 +68,29 @@ func (tr *wsTransporter) Handshake(conn net.Conn, options ...HandshakeOption) (n
return websocketClientConn(url.String(), conn, nil, wsOptions)
}
func forword(opts *WSOptions) func(w http.ResponseWriter, req *http.Request) {
return func(w http.ResponseWriter, req *http.Request) {
urlStr := opts.Reverse
if !strings.HasPrefix(urlStr, "http://") && !strings.HasPrefix(urlStr, "https://") {
urlStr = "http://" + urlStr
}
reqURL, err := url.ParseRequestURI(urlStr)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
}
req.URL.Host = reqURL.Host
req.URL.Scheme = reqURL.Scheme
req.Host = reqURL.Host
httputil.NewSingleHostReverseProxy(reqURL).ServeHTTP(w, req)
}
}
type mwsTransporter struct {
tcpTransporter
options *WSOptions
@ -381,6 +406,9 @@ func WSListener(addr string, options *WSOptions) (Listener, error) {
}
mux := http.NewServeMux()
mux.Handle(path, http.HandlerFunc(l.upgrade))
if options.Reverse != "" {
mux.HandleFunc("/", forword(options))
}
l.srv = &http.Server{
Addr: addr,
Handler: mux,
@ -479,6 +507,9 @@ func MWSListener(addr string, options *WSOptions) (Listener, error) {
mux := http.NewServeMux()
mux.Handle(path, http.HandlerFunc(l.upgrade))
if options.Reverse != "" {
mux.HandleFunc("/", forword(options))
}
l.srv = &http.Server{
Addr: addr,
Handler: mux,
@ -604,6 +635,9 @@ func WSSListener(addr string, tlsConfig *tls.Config, options *WSOptions) (Listen
mux := http.NewServeMux()
mux.Handle(path, http.HandlerFunc(l.upgrade))
if options.Reverse != "" {
mux.HandleFunc("/", forword(options))
}
l.srv = &http.Server{
Addr: addr,
TLSConfig: tlsConfig,
@ -670,6 +704,9 @@ func MWSSListener(addr string, tlsConfig *tls.Config, options *WSOptions) (Liste
mux := http.NewServeMux()
mux.Handle(path, http.HandlerFunc(l.upgrade))
if options.Reverse != "" {
mux.HandleFunc("/", forword(options))
}
l.srv = &http.Server{
Addr: addr,
TLSConfig: tlsConfig,