1
0

dashboard.go 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475
  1. // Copyright 2017 fatedier, fatedier@gmail.com
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. package server
  15. import (
  16. "fmt"
  17. "net"
  18. "net/http"
  19. "time"
  20. "github.com/fatedier/frp/assets"
  21. "github.com/fatedier/frp/models/config"
  22. frpNet "github.com/fatedier/frp/utils/net"
  23. "github.com/julienschmidt/httprouter"
  24. )
  25. var (
  26. httpServerReadTimeout = 10 * time.Second
  27. httpServerWriteTimeout = 10 * time.Second
  28. )
  29. func RunDashboardServer(addr string, port int) (err error) {
  30. // url router
  31. router := httprouter.New()
  32. user, passwd := config.ServerCommonCfg.DashboardUser, config.ServerCommonCfg.DashboardPwd
  33. // api, see dashboard_api.go
  34. router.GET("/api/serverinfo", frpNet.HttprouterBasicAuth(apiServerInfo, user, passwd))
  35. router.GET("/api/proxy/tcp", frpNet.HttprouterBasicAuth(apiProxyTcp, user, passwd))
  36. router.GET("/api/proxy/udp", frpNet.HttprouterBasicAuth(apiProxyUdp, user, passwd))
  37. router.GET("/api/proxy/http", frpNet.HttprouterBasicAuth(apiProxyHttp, user, passwd))
  38. router.GET("/api/proxy/https", frpNet.HttprouterBasicAuth(apiProxyHttps, user, passwd))
  39. router.GET("/api/proxy/traffic/:name", frpNet.HttprouterBasicAuth(apiProxyTraffic, user, passwd))
  40. // view
  41. router.Handler("GET", "/favicon.ico", http.FileServer(assets.FileSystem))
  42. router.Handler("GET", "/static/*filepath", frpNet.MakeHttpGzipHandler(
  43. frpNet.NewHttpBasicAuthWraper(http.StripPrefix("/static/", http.FileServer(assets.FileSystem)), user, passwd)))
  44. router.HandlerFunc("GET", "/", frpNet.HttpBasicAuth(func(w http.ResponseWriter, r *http.Request) {
  45. http.Redirect(w, r, "/static/", http.StatusMovedPermanently)
  46. }, user, passwd))
  47. address := fmt.Sprintf("%s:%d", addr, port)
  48. server := &http.Server{
  49. Addr: address,
  50. Handler: router,
  51. ReadTimeout: httpServerReadTimeout,
  52. WriteTimeout: httpServerWriteTimeout,
  53. }
  54. if address == "" {
  55. address = ":http"
  56. }
  57. ln, err := net.Listen("tcp", address)
  58. if err != nil {
  59. return err
  60. }
  61. go server.Serve(ln)
  62. return
  63. }