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.
72 lines
2.1 KiB
72 lines
2.1 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 client |
|
|
|
import ( |
|
"net" |
|
"net/http" |
|
"time" |
|
|
|
"github.com/fatedier/frp/assets" |
|
frpNet "github.com/fatedier/frp/pkg/util/net" |
|
|
|
"github.com/gorilla/mux" |
|
) |
|
|
|
var ( |
|
httpServerReadTimeout = 10 * time.Second |
|
httpServerWriteTimeout = 10 * time.Second |
|
) |
|
|
|
func (svr *Service) RunAdminServer(address string) (err error) { |
|
// url router |
|
router := mux.NewRouter() |
|
|
|
router.HandleFunc("/healthz", svr.healthz) |
|
|
|
subRouter := router.NewRoute().Subrouter() |
|
user, passwd := svr.cfg.AdminUser, svr.cfg.AdminPwd |
|
subRouter.Use(frpNet.NewHTTPAuthMiddleware(user, passwd).Middleware) |
|
|
|
// api, see admin_api.go |
|
subRouter.HandleFunc("/api/reload", svr.apiReload).Methods("GET") |
|
subRouter.HandleFunc("/api/status", svr.apiStatus).Methods("GET") |
|
subRouter.HandleFunc("/api/config", svr.apiGetConfig).Methods("GET") |
|
subRouter.HandleFunc("/api/config", svr.apiPutConfig).Methods("PUT") |
|
|
|
// view |
|
subRouter.Handle("/favicon.ico", http.FileServer(assets.FileSystem)).Methods("GET") |
|
subRouter.PathPrefix("/static/").Handler(frpNet.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, |
|
} |
|
if address == "" { |
|
address = ":http" |
|
} |
|
ln, err := net.Listen("tcp", address) |
|
if err != nil { |
|
return err |
|
} |
|
|
|
go server.Serve(ln) |
|
return |
|
}
|
|
|