Method ServeHTTP

wsserver.go:148–191

(w http.ResponseWriter, r *http.Request)

}

func (s *WSServer) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	conn, err := upgrader.Upgrade(w, r, nil)
	if err != nil {
		return
	}
	c := conn.UnderlyingConn()
	defer c.Close()
	from := c.RemoteAddr().String()
	if s.XForwardedFor && r.Header.Get("X-Forwarded-For") != "" {
		s1 := strings.Split(r.Header.Get("X-Forwarded-For"), ", ")[0]
		h, _, err := net.SplitHostPort(s1)
		if err != nil {
			h = s1
		}
		if net.ParseIP(h) != nil {
			_, p, err := net.SplitHostPort(from)
			if err == nil {
				from = net.JoinHostPort(h, p)
			}
		}
	}
	var ss Exchanger
	if !s.WithoutBrook {
		ss, err = NewStreamServer(s.Password, from, c, s.TCPTimeout, s.UDPTimeout)
	}
	if s.WithoutBrook {
		ss, err = NewSimpleStreamServer(s.Password, from, c, s.TCPTimeout, s.UDPTimeout)
	}
	if err != nil {
		Log(Error{"from": from, "error": err.Error()})
		return
	}
	defer ss.Clean()
	if ss.Network() == "tcp" {
		if err := s.TCPHandle(ss); err != nil {
			Log(Error{"from": from, "dst": ss.Dst(), "error": err.Error()})
		}
	}
	if ss.Network() == "udp" {
		if err := s.UDPHandle(ss); err != nil {
			Log(Error{"from": from, "dst": ss.Dst(), "error": err.Error()})
		}
	}
}

func (s *WSServer) TCPHandle(ss Exchanger) error {
	rc, err := DialTCP("tcp", "", ss.Dst())