mirror of https://github.com/fatedier/frp
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
99 lines
3.0 KiB
99 lines
3.0 KiB
// Copyright 2017 fatedier, fatedier@gmail.com |
|
// |
|
// Licensed under the Apache License, Version 2.0 (the "License"); |
|
// you may not use this file except in compliance with the License. |
|
// You may obtain a copy of the License at |
|
// |
|
// http://www.apache.org/licenses/LICENSE-2.0 |
|
// |
|
// Unless required by applicable law or agreed to in writing, software |
|
// distributed under the License is distributed on an "AS IS" BASIS, |
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
// See the License for the specific language governing permissions and |
|
// limitations under the License. |
|
|
|
package server |
|
|
|
import ( |
|
"crypto/tls" |
|
"net" |
|
"net/http" |
|
"net/http/pprof" |
|
"time" |
|
|
|
"github.com/gorilla/mux" |
|
"github.com/prometheus/client_golang/prometheus/promhttp" |
|
|
|
"github.com/fatedier/frp/assets" |
|
utilnet "github.com/fatedier/frp/pkg/util/net" |
|
) |
|
|
|
var ( |
|
httpServerReadTimeout = 60 * time.Second |
|
httpServerWriteTimeout = 60 * time.Second |
|
) |
|
|
|
func (svr *Service) RunDashboardServer(address string) (err error) { |
|
// url router |
|
router := mux.NewRouter() |
|
router.HandleFunc("/healthz", svr.Healthz) |
|
|
|
// debug |
|
if svr.cfg.WebServer.PprofEnable { |
|
router.HandleFunc("/debug/pprof/cmdline", pprof.Cmdline) |
|
router.HandleFunc("/debug/pprof/profile", pprof.Profile) |
|
router.HandleFunc("/debug/pprof/symbol", pprof.Symbol) |
|
router.HandleFunc("/debug/pprof/trace", pprof.Trace) |
|
router.PathPrefix("/debug/pprof/").HandlerFunc(pprof.Index) |
|
} |
|
|
|
subRouter := router.NewRoute().Subrouter() |
|
|
|
user, passwd := svr.cfg.WebServer.User, svr.cfg.WebServer.Password |
|
subRouter.Use(utilnet.NewHTTPAuthMiddleware(user, passwd).SetAuthFailDelay(200 * time.Millisecond).Middleware) |
|
|
|
// metrics |
|
if svr.cfg.EnablePrometheus { |
|
subRouter.Handle("/metrics", promhttp.Handler()) |
|
} |
|
|
|
// api, see dashboard_api.go |
|
subRouter.HandleFunc("/api/serverinfo", svr.APIServerInfo).Methods("GET") |
|
subRouter.HandleFunc("/api/proxy/{type}", svr.APIProxyByType).Methods("GET") |
|
subRouter.HandleFunc("/api/proxy/{type}/{name}", svr.APIProxyByTypeAndName).Methods("GET") |
|
subRouter.HandleFunc("/api/traffic/{name}", svr.APIProxyTraffic).Methods("GET") |
|
|
|
// view |
|
subRouter.Handle("/favicon.ico", http.FileServer(assets.FileSystem)).Methods("GET") |
|
subRouter.PathPrefix("/static/").Handler(utilnet.MakeHTTPGzipHandler(http.StripPrefix("/static/", http.FileServer(assets.FileSystem)))).Methods("GET") |
|
|
|
subRouter.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { |
|
http.Redirect(w, r, "/static/", http.StatusMovedPermanently) |
|
}) |
|
|
|
server := &http.Server{ |
|
Addr: address, |
|
Handler: router, |
|
ReadTimeout: httpServerReadTimeout, |
|
WriteTimeout: httpServerWriteTimeout, |
|
} |
|
ln, err := net.Listen("tcp", address) |
|
if err != nil { |
|
return err |
|
} |
|
|
|
if svr.cfg.WebServer.TLS != nil { |
|
cert, err := tls.LoadX509KeyPair(svr.cfg.WebServer.TLS.CertFile, svr.cfg.WebServer.TLS.KeyFile) |
|
if err != nil { |
|
return err |
|
} |
|
tlsCfg := &tls.Config{ |
|
Certificates: []tls.Certificate{cert}, |
|
} |
|
ln = tls.NewListener(ln, tlsCfg) |
|
} |
|
go func() { |
|
_ = server.Serve(ln) |
|
}() |
|
return |
|
}
|
|
|