XiaoMo/frp
1
0
Fork 0
mirror of https://github.com/fatedier/frp.git synced 2025-07-27 07:35:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

vhost: use new readClientHello function (#2504)

Browse Source
This commit is contained in:
fatedier
2021-08-03 22:58:03 +08:00
committed by GitHub
parent 09f39de74e
commit 86b2e686a5
3 changed files with 80 additions and 155 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/e2e/mock/server/httpserver/server.go
View File

@@ -11,7 +11,7 @@ import (
type Server struct {
bindAddr string
bindPort int
hanlder http.Handler
handler http.Handler
l net.Listener
tlsConfig *tls.Config
@@ -54,14 +54,14 @@ func WithTlsConfig(tlsConfig *tls.Config) Option {
func WithHandler(h http.Handler) Option {
return func(s *Server) *Server {
s.hanlder = h
s.handler = h
return s
}
}
func WithResponse(resp []byte) Option {
return func(s *Server) *Server {
s.hanlder = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
s.handler = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.Write(resp)
})
return s
@@ -76,7 +76,7 @@ func (s *Server) Run() error {
addr := net.JoinHostPort(s.bindAddr, strconv.Itoa(s.bindPort))
hs := &http.Server{
Addr: addr,
Handler: s.hanlder,
Handler: s.handler,
TLSConfig: s.tlsConfig,
}