frp/server/dashboard_api.go

222 lines
5.9 KiB
Go
Raw Normal View History

2017-03-23 02:01:25 +08:00
// 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 (
"encoding/json"
"net/http"
2017-03-23 02:01:25 +08:00
"github.com/fatedier/frp/models/config"
"github.com/fatedier/frp/models/consts"
"github.com/fatedier/frp/utils/log"
"github.com/fatedier/frp/utils/version"
2017-03-23 02:01:25 +08:00
"github.com/julienschmidt/httprouter"
)
type GeneralResponse struct {
Code int64 `json:"code"`
Msg string `json:"msg"`
}
2017-03-23 02:01:25 +08:00
// api/serverinfo
type ServerInfoResp struct {
GeneralResponse
Version string `json:"version"`
2017-03-23 02:01:25 +08:00
VhostHttpPort int64 `json:"vhost_http_port"`
VhostHttpsPort int64 `json:"vhost_https_port"`
AuthTimeout int64 `json:"auth_timeout"`
SubdomainHost string `json:"subdomain_host"`
MaxPoolCount int64 `json:"max_pool_count"`
HeartBeatTimeout int64 `json:"heart_beat_timeout"`
2017-03-27 01:39:05 +08:00
TotalTrafficIn int64 `json:"total_traffic_in"`
TotalTrafficOut int64 `json:"total_traffic_out"`
2017-03-23 02:01:25 +08:00
CurConns int64 `json:"cur_conns"`
ClientCounts int64 `json:"client_counts"`
ProxyTypeCounts map[string]int64 `json:"proxy_type_count"`
}
func apiServerInfo(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
var (
buf []byte
res ServerInfoResp
)
defer func() {
2017-03-23 02:01:25 +08:00
log.Info("Http response [/api/serverinfo]: code [%d]", res.Code)
}()
2017-03-23 02:01:25 +08:00
log.Info("Http request: [/api/serverinfo]")
cfg := config.ServerCommonCfg
serverStats := StatsGetServer()
res = ServerInfoResp{
Version: version.Full(),
2017-03-23 02:01:25 +08:00
VhostHttpPort: cfg.VhostHttpPort,
VhostHttpsPort: cfg.VhostHttpsPort,
AuthTimeout: cfg.AuthTimeout,
SubdomainHost: cfg.SubDomainHost,
MaxPoolCount: cfg.MaxPoolCount,
HeartBeatTimeout: cfg.HeartBeatTimeout,
2017-03-27 01:39:05 +08:00
TotalTrafficIn: serverStats.TotalTrafficIn,
TotalTrafficOut: serverStats.TotalTrafficOut,
2017-03-23 02:01:25 +08:00
CurConns: serverStats.CurConns,
ClientCounts: serverStats.ClientCounts,
ProxyTypeCounts: serverStats.ProxyTypeCounts,
}
2017-03-23 02:01:25 +08:00
buf, _ = json.Marshal(&res)
w.Write(buf)
}
2017-03-23 02:01:25 +08:00
// Get proxy info.
type ProxyStatsInfo struct {
2017-03-27 01:39:05 +08:00
Name string `json:"name"`
Conf config.ProxyConf `json:"conf"`
TodayTrafficIn int64 `json:"today_traffic_in"`
TodayTrafficOut int64 `json:"today_traffic_out"`
CurConns int64 `json:"cur_conns"`
Status string `json:"status"`
}
2017-03-23 02:01:25 +08:00
type GetProxyInfoResp struct {
GeneralResponse
Proxies []*ProxyStatsInfo `json:"proxies"`
}
// api/proxy/tcp
func apiProxyTcp(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
var (
buf []byte
res GetProxyInfoResp
)
defer func() {
2017-03-23 02:01:25 +08:00
log.Info("Http response [/api/proxy/tcp]: code [%d]", res.Code)
}()
2017-03-23 02:01:25 +08:00
log.Info("Http request: [/api/proxy/tcp]")
res.Proxies = getProxyStatsByType(consts.TcpProxy)
buf, _ = json.Marshal(&res)
w.Write(buf)
}
// api/proxy/udp
func apiProxyUdp(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
var (
buf []byte
res GetProxyInfoResp
)
defer func() {
log.Info("Http response [/api/proxy/udp]: code [%d]", res.Code)
}()
log.Info("Http request: [/api/proxy/udp]")
res.Proxies = getProxyStatsByType(consts.UdpProxy)
buf, _ = json.Marshal(&res)
w.Write(buf)
}
// api/proxy/http
func apiProxyHttp(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
var (
buf []byte
res GetProxyInfoResp
)
defer func() {
log.Info("Http response [/api/proxy/http]: code [%d]", res.Code)
}()
log.Info("Http request: [/api/proxy/http]")
res.Proxies = getProxyStatsByType(consts.HttpProxy)
buf, _ = json.Marshal(&res)
w.Write(buf)
}
// api/proxy/https
func apiProxyHttps(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
var (
buf []byte
res GetProxyInfoResp
)
defer func() {
log.Info("Http response [/api/proxy/https]: code [%d]", res.Code)
}()
log.Info("Http request: [/api/proxy/https]")
res.Proxies = getProxyStatsByType(consts.HttpsProxy)
buf, _ = json.Marshal(&res)
w.Write(buf)
}
func getProxyStatsByType(proxyType string) (proxyInfos []*ProxyStatsInfo) {
proxyStats := StatsGetProxiesByType(proxyType)
proxyInfos = make([]*ProxyStatsInfo, 0, len(proxyStats))
for _, ps := range proxyStats {
proxyInfo := &ProxyStatsInfo{}
if pxy, ok := ServerService.pxyManager.GetByName(ps.Name); ok {
proxyInfo.Conf = pxy.GetConf()
proxyInfo.Status = consts.Online
} else {
proxyInfo.Status = consts.Offline
}
2017-03-27 01:39:05 +08:00
proxyInfo.TodayTrafficIn = ps.TodayTrafficIn
proxyInfo.TodayTrafficOut = ps.TodayTrafficOut
2017-03-23 02:01:25 +08:00
proxyInfo.CurConns = ps.CurConns
2017-03-27 01:39:05 +08:00
proxyInfo.Name = ps.Name
2017-03-23 02:01:25 +08:00
proxyInfos = append(proxyInfos, proxyInfo)
}
return
}
2017-03-27 01:39:05 +08:00
// api/proxy/traffic/:name
type GetProxyTrafficResp struct {
2017-03-23 02:01:25 +08:00
GeneralResponse
2017-03-27 01:39:05 +08:00
Name string `json:"name"`
TrafficIn []int64 `json:"traffic_in"`
TrafficOut []int64 `json:"traffic_out"`
2017-03-23 02:01:25 +08:00
}
2017-03-27 01:39:05 +08:00
func apiProxyTraffic(w http.ResponseWriter, r *http.Request, params httprouter.Params) {
2017-03-23 02:01:25 +08:00
var (
buf []byte
2017-03-27 01:39:05 +08:00
res GetProxyTrafficResp
2017-03-23 02:01:25 +08:00
)
name := params.ByName("name")
defer func() {
2017-03-27 01:39:05 +08:00
log.Info("Http response [/api/proxy/traffic/:name]: code [%d]", res.Code)
2017-03-23 02:01:25 +08:00
}()
2017-03-27 01:39:05 +08:00
log.Info("Http request: [/api/proxy/traffic/:name]")
2017-03-23 02:01:25 +08:00
res.Name = name
2017-03-27 01:39:05 +08:00
proxyTrafficInfo := StatsGetProxyTraffic(name)
if proxyTrafficInfo == nil {
2017-03-23 02:01:25 +08:00
res.Code = 1
res.Msg = "no proxy info found"
} else {
2017-03-27 01:39:05 +08:00
res.TrafficIn = proxyTrafficInfo.TrafficIn
res.TrafficOut = proxyTrafficInfo.TrafficOut
2017-03-23 02:01:25 +08:00
}
2017-03-23 02:01:25 +08:00
buf, _ = json.Marshal(&res)
w.Write(buf)
}