添加在线调试 api
This commit is contained in:
@@ -1,167 +0,0 @@
|
||||
package handlers
|
||||
|
||||
import (
|
||||
"log/slog"
|
||||
"proxy-server/server/models"
|
||||
"proxy-server/server/pkg/orm"
|
||||
"proxy-server/server/pkg/resp"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/pkg/errors"
|
||||
)
|
||||
|
||||
// region frp 接口
|
||||
|
||||
type FrpData struct {
|
||||
Reject bool
|
||||
RejectReason string
|
||||
Unchange bool
|
||||
}
|
||||
|
||||
func ChanRequest(c *gin.Context) {
|
||||
type Body struct {
|
||||
Content struct {
|
||||
ProxyName string `json:"proxy_name"`
|
||||
ProxyType string `json:"proxy_type"`
|
||||
RemoteAddr string `json:"remote_addr"`
|
||||
User interface{}
|
||||
}
|
||||
}
|
||||
|
||||
op := c.Query("op")
|
||||
if op != "NewUserConn" {
|
||||
_ = c.Error(errors.New("不支持的操作"))
|
||||
return
|
||||
}
|
||||
|
||||
id := c.GetHeader("X-Frp-Reqid")
|
||||
if id == "" {
|
||||
_ = c.Error(errors.New("请求头中缺少 X-Frp-Reqid"))
|
||||
return
|
||||
}
|
||||
|
||||
var body Body
|
||||
err := c.ShouldBindJSON(&body)
|
||||
if err != nil {
|
||||
_ = c.Error(errors.Wrap(err, "解析请求正文失败"))
|
||||
return
|
||||
}
|
||||
content := body.Content
|
||||
|
||||
// 检查此 ip 是否有权限访问目标 node
|
||||
clientIp := strings.Split(content.RemoteAddr, ":")[0]
|
||||
targetNode := content.ProxyName
|
||||
slog.Debug(id + " 用户 " + clientIp + " 请求连接到 " + targetNode)
|
||||
|
||||
var channels []models.Channel
|
||||
err = orm.DB.
|
||||
Joins("INNER JOIN public.nodes n ON channels.node_id = n.id AND n.name = ?", targetNode).
|
||||
Joins("INNER JOIN public.users u ON channels.user_id = u.id").
|
||||
Joins("INNER JOIN public.user_ips ip ON u.id = ip.user_id AND ip.ip_address = ?", clientIp).
|
||||
Find(&channels).Error
|
||||
if err != nil {
|
||||
_ = c.Error(errors.Wrap(err, "查询用户权限失败"))
|
||||
return
|
||||
}
|
||||
|
||||
// 返回响应
|
||||
rsCount := len(channels)
|
||||
if rsCount > 1 {
|
||||
slog.Warn(clientIp + " + " + targetNode + "的组合有多个权限结果，这是不应当存在的")
|
||||
}
|
||||
|
||||
if rsCount == 0 {
|
||||
slog.Debug(id + " 没有权限")
|
||||
reject(c)
|
||||
return
|
||||
}
|
||||
channel := channels[0]
|
||||
if channel.Expiration.Before(time.Now()) {
|
||||
slog.Debug(id + " 权限已过期")
|
||||
reject(c)
|
||||
return
|
||||
}
|
||||
|
||||
slog.Debug(id + " 通过验证")
|
||||
confirm(c)
|
||||
}
|
||||
|
||||
func ChanTest(c *gin.Context) {
|
||||
var body map[string]interface{}
|
||||
err := c.ShouldBindJSON(&body)
|
||||
if err != nil {
|
||||
slog.Error("解析请求正文失败", err)
|
||||
}
|
||||
for k, v := range body {
|
||||
slog.Debug("map", "key: ", k, " value: ", v)
|
||||
}
|
||||
confirm(c)
|
||||
}
|
||||
|
||||
func confirm(c *gin.Context) {
|
||||
c.JSON(200, FrpData{
|
||||
Reject: false,
|
||||
Unchange: true,
|
||||
})
|
||||
}
|
||||
|
||||
func reject(c *gin.Context) {
|
||||
c.JSON(401, FrpData{
|
||||
Reject: true,
|
||||
RejectReason: "客户端没有权限访问该节点",
|
||||
})
|
||||
}
|
||||
|
||||
// endregion
|
||||
|
||||
func ChanAuth(c *gin.Context) {
|
||||
type Body struct {
|
||||
Username string `json:"username"`
|
||||
Password string `json:"password"`
|
||||
}
|
||||
type Data struct {
|
||||
Timeout uint64 `json:"timeout"`
|
||||
}
|
||||
|
||||
var body Body
|
||||
err := c.ShouldBindJSON(&body)
|
||||
if err != nil {
|
||||
_ = c.Error(err)
|
||||
c.JSON(400, resp.Fail("请求参数错误"))
|
||||
return
|
||||
}
|
||||
|
||||
// 查找通道
|
||||
var result *models.Channel
|
||||
orm.DB.
|
||||
Model(&models.Channel{}).
|
||||
Where(&models.Channel{
|
||||
Username: body.Username,
|
||||
Password: body.Password,
|
||||
}).
|
||||
First(&result)
|
||||
if result == nil {
|
||||
_ = c.Error(errors.New("用户信息不存在"))
|
||||
c.JSON(401, resp.Fail("账号密码错误"))
|
||||
return
|
||||
}
|
||||
|
||||
// 验证账号密码 todo 哈希密码验证
|
||||
if result.Username != body.Username || result.Password != body.Password {
|
||||
_ = c.Error(errors.New("账号密码错误"))
|
||||
c.JSON(401, resp.Fail("账号密码错误"))
|
||||
return
|
||||
}
|
||||
|
||||
// 计算到期时间
|
||||
timeout := result.Expiration.Sub(time.Now())
|
||||
|
||||
// todo 保存会话 对于大量短连接的情况，考虑如何保存连接会话信息
|
||||
|
||||
// 返回结果
|
||||
c.JSON(200, resp.Done(Data{
|
||||
Timeout: uint64(timeout.Seconds()),
|
||||
}))
|
||||
}
|
||||
36
server/web/handlers/debug.go
Normal file
36
server/web/handlers/debug.go
Normal file
@@ -0,0 +1,36 @@
|
||||
package handlers
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"proxy-server/server/debug"
|
||||
"slices"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
func GetConsuming(c *gin.Context) {
|
||||
list := debug.ConsumingList()
|
||||
// sort by total time
|
||||
slices.SortFunc(list, func(a debug.Consuming, b debug.Consuming) int {
|
||||
if a.Total < b.Total {
|
||||
return 1
|
||||
} else if a.Total > b.Total {
|
||||
return -1
|
||||
}
|
||||
return 0
|
||||
})
|
||||
// map to string
|
||||
strList := make([]string, len(list))
|
||||
for i := 0; i < len(list); i++ {
|
||||
times := list[i]
|
||||
strList[i] = fmt.Sprintf("Auth: %s, Data: %s, Proxy: %s, Total: %s", times.Auth, times.Data, times.Proxy, times.Total)
|
||||
}
|
||||
c.JSON(200, strList)
|
||||
}
|
||||
|
||||
func RestConsuming(c *gin.Context) {
|
||||
debug.InitConsumingList()
|
||||
c.JSON(200, gin.H{
|
||||
"message": "success",
|
||||
})
|
||||
}
|
||||
@@ -1,116 +0,0 @@
|
||||
package handlers
|
||||
|
||||
import (
|
||||
"os"
|
||||
"proxy-server/server/models"
|
||||
"proxy-server/server/pkg/orm"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/pkg/errors"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type NodeRegisterReq struct {
|
||||
Name string
|
||||
Secret string
|
||||
}
|
||||
|
||||
func NodeRegister(c *gin.Context) {
|
||||
|
||||
// 请求参数
|
||||
var req NodeRegisterReq
|
||||
err := c.ShouldBind(&req)
|
||||
if err != nil {
|
||||
_ = c.Error(errors.Wrap(err, "参数解析错误"))
|
||||
return
|
||||
}
|
||||
|
||||
// 验证 secret
|
||||
secret := os.Getenv("SECRET")
|
||||
if req.Secret != secret {
|
||||
_ = c.Error(errors.New("拒绝连接"))
|
||||
return
|
||||
}
|
||||
|
||||
// 注册节点
|
||||
// todo 查询运营商和地区
|
||||
err = orm.DB.Transaction(func(tx *gorm.DB) error {
|
||||
|
||||
// 查询节点是否已存在
|
||||
var count int64
|
||||
err := orm.DB.Where(&models.Node{
|
||||
Name: req.Name,
|
||||
}).Count(&count).Error
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// 不存在则注册
|
||||
if count == 0 {
|
||||
ipAddress := c.ClientIP()
|
||||
node := models.Node{
|
||||
Name: req.Name,
|
||||
Provider: "",
|
||||
Location: "",
|
||||
IPAddress: ipAddress,
|
||||
}
|
||||
err = orm.DB.Create(&node).Error
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
})
|
||||
if err != nil {
|
||||
_ = c.Error(errors.Wrap(err, "注册节点失败"))
|
||||
return
|
||||
}
|
||||
|
||||
c.Status(200)
|
||||
}
|
||||
|
||||
type NodeReportReq struct {
|
||||
Name string
|
||||
}
|
||||
|
||||
func NodeReport(c *gin.Context) {
|
||||
|
||||
// 请求参数
|
||||
var req NodeReportReq
|
||||
err := c.ShouldBind(&req)
|
||||
if err != nil {
|
||||
_ = c.Error(errors.Wrap(err, "参数解析错误"))
|
||||
return
|
||||
}
|
||||
|
||||
// 上报节点信息
|
||||
err = orm.DB.Transaction(func(tx *gorm.DB) error {
|
||||
|
||||
// 查询节点
|
||||
var node models.Node
|
||||
err = orm.DB.Where(&models.Node{
|
||||
Name: req.Name,
|
||||
}).First(&node).Error
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// 更新节点信息
|
||||
ipAddress := c.ClientIP()
|
||||
if ipAddress != node.IPAddress {
|
||||
err = orm.DB.Model(&node).Update("ip_address", ipAddress).Error
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
})
|
||||
if err != nil {
|
||||
_ = c.Error(errors.Wrap(err, "上报节点信息失败"))
|
||||
return
|
||||
}
|
||||
|
||||
c.Status(200)
|
||||
}
|
||||
@@ -1 +0,0 @@
|
||||
package handlers
|
||||
Reference in New Issue
Block a user