mirror of
https://gitee.com/IrisVega/frp.git
synced 2024-11-01 22:31:29 +08:00
666f122a72
The ClientCommonConf, configuration file path, and server UDP port are now passed around as arguments instead of being shared between components as global variables. This allows for multiple clients to exist in the same process, and allows client.Session to be used as a library more easily.
326 lines
8.3 KiB
Go
326 lines
8.3 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 client
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"sort"
|
|
"strings"
|
|
|
|
"github.com/fatedier/frp/client/proxy"
|
|
"github.com/fatedier/frp/models/config"
|
|
"github.com/fatedier/frp/utils/log"
|
|
)
|
|
|
|
type GeneralResponse struct {
|
|
Code int
|
|
Msg string
|
|
}
|
|
|
|
// GET api/reload
|
|
|
|
func (svr *Service) apiReload(w http.ResponseWriter, r *http.Request) {
|
|
res := GeneralResponse{Code: 200}
|
|
|
|
log.Info("Http request [/api/reload]")
|
|
defer func() {
|
|
log.Info("Http response [/api/reload], code [%d]", res.Code)
|
|
w.WriteHeader(res.Code)
|
|
if len(res.Msg) > 0 {
|
|
w.Write([]byte(res.Msg))
|
|
}
|
|
}()
|
|
|
|
content, err := config.GetRenderedConfFromFile(svr.cfgFile)
|
|
if err != nil {
|
|
res.Code = 400
|
|
res.Msg = err.Error()
|
|
log.Warn("reload frpc config file error: %s", res.Msg)
|
|
return
|
|
}
|
|
|
|
newCommonCfg, err := config.UnmarshalClientConfFromIni(content)
|
|
if err != nil {
|
|
res.Code = 400
|
|
res.Msg = err.Error()
|
|
log.Warn("reload frpc common section error: %s", res.Msg)
|
|
return
|
|
}
|
|
|
|
pxyCfgs, visitorCfgs, err := config.LoadAllConfFromIni(svr.cfg.User, content, newCommonCfg.Start)
|
|
if err != nil {
|
|
res.Code = 400
|
|
res.Msg = err.Error()
|
|
log.Warn("reload frpc proxy config error: %s", res.Msg)
|
|
return
|
|
}
|
|
|
|
err = svr.ReloadConf(pxyCfgs, visitorCfgs)
|
|
if err != nil {
|
|
res.Code = 500
|
|
res.Msg = err.Error()
|
|
log.Warn("reload frpc proxy config error: %s", res.Msg)
|
|
return
|
|
}
|
|
log.Info("success reload conf")
|
|
return
|
|
}
|
|
|
|
type StatusResp struct {
|
|
Tcp []ProxyStatusResp `json:"tcp"`
|
|
Udp []ProxyStatusResp `json:"udp"`
|
|
Http []ProxyStatusResp `json:"http"`
|
|
Https []ProxyStatusResp `json:"https"`
|
|
Stcp []ProxyStatusResp `json:"stcp"`
|
|
Xtcp []ProxyStatusResp `json:"xtcp"`
|
|
}
|
|
|
|
type ProxyStatusResp struct {
|
|
Name string `json:"name"`
|
|
Type string `json:"type"`
|
|
Status string `json:"status"`
|
|
Err string `json:"err"`
|
|
LocalAddr string `json:"local_addr"`
|
|
Plugin string `json:"plugin"`
|
|
RemoteAddr string `json:"remote_addr"`
|
|
}
|
|
|
|
type ByProxyStatusResp []ProxyStatusResp
|
|
|
|
func (a ByProxyStatusResp) Len() int { return len(a) }
|
|
func (a ByProxyStatusResp) Swap(i, j int) { a[i], a[j] = a[j], a[i] }
|
|
func (a ByProxyStatusResp) Less(i, j int) bool { return strings.Compare(a[i].Name, a[j].Name) < 0 }
|
|
|
|
func NewProxyStatusResp(status *proxy.ProxyStatus, serverAddr string) ProxyStatusResp {
|
|
psr := ProxyStatusResp{
|
|
Name: status.Name,
|
|
Type: status.Type,
|
|
Status: status.Status,
|
|
Err: status.Err,
|
|
}
|
|
switch cfg := status.Cfg.(type) {
|
|
case *config.TcpProxyConf:
|
|
if cfg.LocalPort != 0 {
|
|
psr.LocalAddr = fmt.Sprintf("%s:%d", cfg.LocalIp, cfg.LocalPort)
|
|
}
|
|
psr.Plugin = cfg.Plugin
|
|
if status.Err != "" {
|
|
psr.RemoteAddr = fmt.Sprintf("%s:%d", serverAddr, cfg.RemotePort)
|
|
} else {
|
|
psr.RemoteAddr = serverAddr + status.RemoteAddr
|
|
}
|
|
case *config.UdpProxyConf:
|
|
if cfg.LocalPort != 0 {
|
|
psr.LocalAddr = fmt.Sprintf("%s:%d", cfg.LocalIp, cfg.LocalPort)
|
|
}
|
|
if status.Err != "" {
|
|
psr.RemoteAddr = fmt.Sprintf("%s:%d", serverAddr, cfg.RemotePort)
|
|
} else {
|
|
psr.RemoteAddr = serverAddr + status.RemoteAddr
|
|
}
|
|
case *config.HttpProxyConf:
|
|
if cfg.LocalPort != 0 {
|
|
psr.LocalAddr = fmt.Sprintf("%s:%d", cfg.LocalIp, cfg.LocalPort)
|
|
}
|
|
psr.Plugin = cfg.Plugin
|
|
psr.RemoteAddr = status.RemoteAddr
|
|
case *config.HttpsProxyConf:
|
|
if cfg.LocalPort != 0 {
|
|
psr.LocalAddr = fmt.Sprintf("%s:%d", cfg.LocalIp, cfg.LocalPort)
|
|
}
|
|
psr.Plugin = cfg.Plugin
|
|
psr.RemoteAddr = status.RemoteAddr
|
|
case *config.StcpProxyConf:
|
|
if cfg.LocalPort != 0 {
|
|
psr.LocalAddr = fmt.Sprintf("%s:%d", cfg.LocalIp, cfg.LocalPort)
|
|
}
|
|
psr.Plugin = cfg.Plugin
|
|
case *config.XtcpProxyConf:
|
|
if cfg.LocalPort != 0 {
|
|
psr.LocalAddr = fmt.Sprintf("%s:%d", cfg.LocalIp, cfg.LocalPort)
|
|
}
|
|
psr.Plugin = cfg.Plugin
|
|
}
|
|
return psr
|
|
}
|
|
|
|
// GET api/status
|
|
func (svr *Service) apiStatus(w http.ResponseWriter, r *http.Request) {
|
|
var (
|
|
buf []byte
|
|
res StatusResp
|
|
)
|
|
res.Tcp = make([]ProxyStatusResp, 0)
|
|
res.Udp = make([]ProxyStatusResp, 0)
|
|
res.Http = make([]ProxyStatusResp, 0)
|
|
res.Https = make([]ProxyStatusResp, 0)
|
|
res.Stcp = make([]ProxyStatusResp, 0)
|
|
res.Xtcp = make([]ProxyStatusResp, 0)
|
|
|
|
log.Info("Http request [/api/status]")
|
|
defer func() {
|
|
log.Info("Http response [/api/status]")
|
|
buf, _ = json.Marshal(&res)
|
|
w.Write(buf)
|
|
}()
|
|
|
|
ps := svr.ctl.pm.GetAllProxyStatus()
|
|
for _, status := range ps {
|
|
switch status.Type {
|
|
case "tcp":
|
|
res.Tcp = append(res.Tcp, NewProxyStatusResp(status, svr.cfg.ServerAddr))
|
|
case "udp":
|
|
res.Udp = append(res.Udp, NewProxyStatusResp(status, svr.cfg.ServerAddr))
|
|
case "http":
|
|
res.Http = append(res.Http, NewProxyStatusResp(status, svr.cfg.ServerAddr))
|
|
case "https":
|
|
res.Https = append(res.Https, NewProxyStatusResp(status, svr.cfg.ServerAddr))
|
|
case "stcp":
|
|
res.Stcp = append(res.Stcp, NewProxyStatusResp(status, svr.cfg.ServerAddr))
|
|
case "xtcp":
|
|
res.Xtcp = append(res.Xtcp, NewProxyStatusResp(status, svr.cfg.ServerAddr))
|
|
}
|
|
}
|
|
sort.Sort(ByProxyStatusResp(res.Tcp))
|
|
sort.Sort(ByProxyStatusResp(res.Udp))
|
|
sort.Sort(ByProxyStatusResp(res.Http))
|
|
sort.Sort(ByProxyStatusResp(res.Https))
|
|
sort.Sort(ByProxyStatusResp(res.Stcp))
|
|
sort.Sort(ByProxyStatusResp(res.Xtcp))
|
|
return
|
|
}
|
|
|
|
// GET api/config
|
|
func (svr *Service) apiGetConfig(w http.ResponseWriter, r *http.Request) {
|
|
res := GeneralResponse{Code: 200}
|
|
|
|
log.Info("Http get request [/api/config]")
|
|
defer func() {
|
|
log.Info("Http get response [/api/config], code [%d]", res.Code)
|
|
w.WriteHeader(res.Code)
|
|
if len(res.Msg) > 0 {
|
|
w.Write([]byte(res.Msg))
|
|
}
|
|
}()
|
|
|
|
if svr.cfgFile == "" {
|
|
res.Code = 400
|
|
res.Msg = "frpc has no config file path"
|
|
log.Warn("%s", res.Msg)
|
|
return
|
|
}
|
|
|
|
content, err := config.GetRenderedConfFromFile(svr.cfgFile)
|
|
if err != nil {
|
|
res.Code = 400
|
|
res.Msg = err.Error()
|
|
log.Warn("load frpc config file error: %s", res.Msg)
|
|
return
|
|
}
|
|
|
|
rows := strings.Split(content, "\n")
|
|
newRows := make([]string, 0, len(rows))
|
|
for _, row := range rows {
|
|
row = strings.TrimSpace(row)
|
|
if strings.HasPrefix(row, "token") {
|
|
continue
|
|
}
|
|
newRows = append(newRows, row)
|
|
}
|
|
res.Msg = strings.Join(newRows, "\n")
|
|
}
|
|
|
|
// PUT api/config
|
|
func (svr *Service) apiPutConfig(w http.ResponseWriter, r *http.Request) {
|
|
res := GeneralResponse{Code: 200}
|
|
|
|
log.Info("Http put request [/api/config]")
|
|
defer func() {
|
|
log.Info("Http put response [/api/config], code [%d]", res.Code)
|
|
w.WriteHeader(res.Code)
|
|
if len(res.Msg) > 0 {
|
|
w.Write([]byte(res.Msg))
|
|
}
|
|
}()
|
|
|
|
// get new config content
|
|
body, err := ioutil.ReadAll(r.Body)
|
|
if err != nil {
|
|
res.Code = 400
|
|
res.Msg = fmt.Sprintf("read request body error: %v", err)
|
|
log.Warn("%s", res.Msg)
|
|
return
|
|
}
|
|
|
|
if len(body) == 0 {
|
|
res.Code = 400
|
|
res.Msg = "body can't be empty"
|
|
log.Warn("%s", res.Msg)
|
|
return
|
|
}
|
|
|
|
// get token from origin content
|
|
token := ""
|
|
b, err := ioutil.ReadFile(svr.cfgFile)
|
|
if err != nil {
|
|
res.Code = 400
|
|
res.Msg = err.Error()
|
|
log.Warn("load frpc config file error: %s", res.Msg)
|
|
return
|
|
}
|
|
content := string(b)
|
|
|
|
for _, row := range strings.Split(content, "\n") {
|
|
row = strings.TrimSpace(row)
|
|
if strings.HasPrefix(row, "token") {
|
|
token = row
|
|
break
|
|
}
|
|
}
|
|
|
|
tmpRows := make([]string, 0)
|
|
for _, row := range strings.Split(string(body), "\n") {
|
|
row = strings.TrimSpace(row)
|
|
if strings.HasPrefix(row, "token") {
|
|
continue
|
|
}
|
|
tmpRows = append(tmpRows, row)
|
|
}
|
|
|
|
newRows := make([]string, 0)
|
|
if token != "" {
|
|
for _, row := range tmpRows {
|
|
newRows = append(newRows, row)
|
|
if strings.HasPrefix(row, "[common]") {
|
|
newRows = append(newRows, token)
|
|
}
|
|
}
|
|
} else {
|
|
newRows = tmpRows
|
|
}
|
|
content = strings.Join(newRows, "\n")
|
|
|
|
err = ioutil.WriteFile(svr.cfgFile, []byte(content), 0644)
|
|
if err != nil {
|
|
res.Code = 500
|
|
res.Msg = fmt.Sprintf("write content to frpc config file error: %v", err)
|
|
log.Warn("%s", res.Msg)
|
|
return
|
|
}
|
|
}
|