mirror of
https://gitee.com/IrisVega/frp.git
synced 2024-11-01 22:31:29 +08:00
348 lines
9.4 KiB
Go
348 lines
9.4 KiB
Go
// 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"
|
|
|
|
"github.com/fatedier/frp/g"
|
|
"github.com/fatedier/frp/models/config"
|
|
"github.com/fatedier/frp/models/consts"
|
|
"github.com/fatedier/frp/utils/log"
|
|
"github.com/fatedier/frp/utils/version"
|
|
|
|
"github.com/julienschmidt/httprouter"
|
|
)
|
|
|
|
type GeneralResponse struct {
|
|
Code int64 `json:"code"`
|
|
Msg string `json:"msg"`
|
|
}
|
|
|
|
// api/serverinfo
|
|
type ServerInfoResp struct {
|
|
GeneralResponse
|
|
|
|
Version string `json:"version"`
|
|
BindPort int `json:"bind_port"`
|
|
BindUdpPort int `json:"bind_udp_port"`
|
|
VhostHttpPort int `json:"vhost_http_port"`
|
|
VhostHttpsPort int `json:"vhost_https_port"`
|
|
KcpBindPort int `json:"kcp_bind_port"`
|
|
AuthTimeout int64 `json:"auth_timeout"`
|
|
SubdomainHost string `json:"subdomain_host"`
|
|
MaxPoolCount int64 `json:"max_pool_count"`
|
|
MaxPortsPerClient int64 `json:"max_ports_per_client"`
|
|
HeartBeatTimeout int64 `json:"heart_beat_timeout"`
|
|
|
|
TotalTrafficIn int64 `json:"total_traffic_in"`
|
|
TotalTrafficOut int64 `json:"total_traffic_out"`
|
|
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() {
|
|
log.Info("Http response [/api/serverinfo]: code [%d]", res.Code)
|
|
}()
|
|
|
|
log.Info("Http request: [/api/serverinfo]")
|
|
cfg := &g.GlbServerCfg.ServerCommonConf
|
|
serverStats := StatsGetServer()
|
|
res = ServerInfoResp{
|
|
Version: version.Full(),
|
|
BindPort: cfg.BindPort,
|
|
BindUdpPort: cfg.BindUdpPort,
|
|
VhostHttpPort: cfg.VhostHttpPort,
|
|
VhostHttpsPort: cfg.VhostHttpsPort,
|
|
KcpBindPort: cfg.KcpBindPort,
|
|
AuthTimeout: cfg.AuthTimeout,
|
|
SubdomainHost: cfg.SubDomainHost,
|
|
MaxPoolCount: cfg.MaxPoolCount,
|
|
MaxPortsPerClient: cfg.MaxPortsPerClient,
|
|
HeartBeatTimeout: cfg.HeartBeatTimeout,
|
|
|
|
TotalTrafficIn: serverStats.TotalTrafficIn,
|
|
TotalTrafficOut: serverStats.TotalTrafficOut,
|
|
CurConns: serverStats.CurConns,
|
|
ClientCounts: serverStats.ClientCounts,
|
|
ProxyTypeCounts: serverStats.ProxyTypeCounts,
|
|
}
|
|
|
|
buf, _ = json.Marshal(&res)
|
|
w.Write(buf)
|
|
}
|
|
|
|
// Get proxy info.
|
|
type ProxyStatsInfo struct {
|
|
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"`
|
|
LastStartTime string `json:"last_start_time"`
|
|
LastCloseTime string `json:"last_close_time"`
|
|
Status string `json:"status"`
|
|
}
|
|
|
|
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() {
|
|
log.Info("Http response [/api/proxy/tcp]: code [%d]", res.Code)
|
|
}()
|
|
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
|
|
}
|
|
proxyInfo.Name = ps.Name
|
|
proxyInfo.TodayTrafficIn = ps.TodayTrafficIn
|
|
proxyInfo.TodayTrafficOut = ps.TodayTrafficOut
|
|
proxyInfo.CurConns = ps.CurConns
|
|
proxyInfo.LastStartTime = ps.LastStartTime
|
|
proxyInfo.LastCloseTime = ps.LastCloseTime
|
|
proxyInfos = append(proxyInfos, proxyInfo)
|
|
}
|
|
return
|
|
}
|
|
|
|
// Get proxy info by name.
|
|
type GetProxyStatsResp struct {
|
|
GeneralResponse
|
|
|
|
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"`
|
|
LastStartTime string `json:"last_start_time"`
|
|
LastCloseTime string `json:"last_close_time"`
|
|
Status string `json:"status"`
|
|
}
|
|
|
|
// api/proxy/tcp/:name
|
|
func apiProxyTcpByName(w http.ResponseWriter, r *http.Request, params httprouter.Params) {
|
|
var (
|
|
buf []byte
|
|
res GetProxyStatsResp
|
|
)
|
|
name := params.ByName("name")
|
|
|
|
defer func() {
|
|
log.Info("Http response [/api/proxy/tcp/:name]: code [%d]", res.Code)
|
|
}()
|
|
log.Info("Http request: [/api/proxy/tcp/:name]")
|
|
|
|
res = getProxyStatsByTypeAndName(consts.TcpProxy, name)
|
|
|
|
buf, _ = json.Marshal(&res)
|
|
w.Write(buf)
|
|
}
|
|
|
|
// api/proxy/udp/:name
|
|
func apiProxyUdpByName(w http.ResponseWriter, r *http.Request, params httprouter.Params) {
|
|
var (
|
|
buf []byte
|
|
res GetProxyStatsResp
|
|
)
|
|
name := params.ByName("name")
|
|
|
|
defer func() {
|
|
log.Info("Http response [/api/proxy/udp/:name]: code [%d]", res.Code)
|
|
}()
|
|
log.Info("Http request: [/api/proxy/udp/:name]")
|
|
|
|
res = getProxyStatsByTypeAndName(consts.UdpProxy, name)
|
|
|
|
buf, _ = json.Marshal(&res)
|
|
w.Write(buf)
|
|
}
|
|
|
|
// api/proxy/http/:name
|
|
func apiProxyHttpByName(w http.ResponseWriter, r *http.Request, params httprouter.Params) {
|
|
var (
|
|
buf []byte
|
|
res GetProxyStatsResp
|
|
)
|
|
name := params.ByName("name")
|
|
|
|
defer func() {
|
|
log.Info("Http response [/api/proxy/http/:name]: code [%d]", res.Code)
|
|
}()
|
|
log.Info("Http request: [/api/proxy/http/:name]")
|
|
|
|
res = getProxyStatsByTypeAndName(consts.HttpProxy, name)
|
|
|
|
buf, _ = json.Marshal(&res)
|
|
w.Write(buf)
|
|
}
|
|
|
|
// api/proxy/https/:name
|
|
func apiProxyHttpsByName(w http.ResponseWriter, r *http.Request, params httprouter.Params) {
|
|
var (
|
|
buf []byte
|
|
res GetProxyStatsResp
|
|
)
|
|
name := params.ByName("name")
|
|
|
|
defer func() {
|
|
log.Info("Http response [/api/proxy/https/:name]: code [%d]", res.Code)
|
|
}()
|
|
log.Info("Http request: [/api/proxy/https/:name]")
|
|
|
|
res = getProxyStatsByTypeAndName(consts.HttpsProxy, name)
|
|
|
|
buf, _ = json.Marshal(&res)
|
|
w.Write(buf)
|
|
}
|
|
|
|
func getProxyStatsByTypeAndName(proxyType string, proxyName string) (proxyInfo GetProxyStatsResp) {
|
|
proxyInfo.Name = proxyName
|
|
ps := StatsGetProxiesByTypeAndName(proxyType, proxyName)
|
|
if ps == nil {
|
|
proxyInfo.Code = 1
|
|
proxyInfo.Msg = "no proxy info found"
|
|
} else {
|
|
if pxy, ok := ServerService.pxyManager.GetByName(proxyName); ok {
|
|
proxyInfo.Conf = pxy.GetConf()
|
|
proxyInfo.Status = consts.Online
|
|
} else {
|
|
proxyInfo.Status = consts.Offline
|
|
}
|
|
proxyInfo.TodayTrafficIn = ps.TodayTrafficIn
|
|
proxyInfo.TodayTrafficOut = ps.TodayTrafficOut
|
|
proxyInfo.CurConns = ps.CurConns
|
|
proxyInfo.LastStartTime = ps.LastStartTime
|
|
proxyInfo.LastCloseTime = ps.LastCloseTime
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// api/proxy/traffic/:name
|
|
type GetProxyTrafficResp struct {
|
|
GeneralResponse
|
|
|
|
Name string `json:"name"`
|
|
TrafficIn []int64 `json:"traffic_in"`
|
|
TrafficOut []int64 `json:"traffic_out"`
|
|
}
|
|
|
|
func apiProxyTraffic(w http.ResponseWriter, r *http.Request, params httprouter.Params) {
|
|
var (
|
|
buf []byte
|
|
res GetProxyTrafficResp
|
|
)
|
|
name := params.ByName("name")
|
|
|
|
defer func() {
|
|
log.Info("Http response [/api/proxy/traffic/:name]: code [%d]", res.Code)
|
|
}()
|
|
log.Info("Http request: [/api/proxy/traffic/:name]")
|
|
|
|
res.Name = name
|
|
proxyTrafficInfo := StatsGetProxyTraffic(name)
|
|
if proxyTrafficInfo == nil {
|
|
res.Code = 1
|
|
res.Msg = "no proxy info found"
|
|
} else {
|
|
res.TrafficIn = proxyTrafficInfo.TrafficIn
|
|
res.TrafficOut = proxyTrafficInfo.TrafficOut
|
|
}
|
|
|
|
buf, _ = json.Marshal(&res)
|
|
w.Write(buf)
|
|
}
|