版本封存
This commit is contained in:
37
cmii-uav-watchdog-center/cmd/main.go
Normal file
37
cmii-uav-watchdog-center/cmd/main.go
Normal file
@@ -0,0 +1,37 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"cmii-uav-watchdog-center/config"
|
||||
"cmii-uav-watchdog-center/router"
|
||||
"cmii-uav-watchdog-common/wdd_log"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
func main() {
|
||||
// 获取执行目录
|
||||
execDir, err := filepath.Abs(filepath.Dir(os.Args[0]))
|
||||
if err != nil {
|
||||
wdd_log.Fatal("获取执行目录失败: %v", err)
|
||||
}
|
||||
|
||||
// 配置文件路径
|
||||
configPath := filepath.Join(execDir, "config.json")
|
||||
|
||||
// 加载配置
|
||||
config, err := config.LoadConfig(configPath)
|
||||
if err != nil {
|
||||
wdd_log.Fatal("加载配置失败: %v", err)
|
||||
}
|
||||
|
||||
// 设置路由
|
||||
r := router.SetupRouter()
|
||||
|
||||
// 启动服务器
|
||||
port := config.Server.Port
|
||||
wdd_log.Info("服务器正在监听端口: %d", port)
|
||||
if err := r.Run(":" + strconv.Itoa(port)); err != nil {
|
||||
wdd_log.Error("启动服务器失败: %v", err)
|
||||
}
|
||||
}
|
||||
11
cmii-uav-watchdog-center/config/config.go
Normal file
11
cmii-uav-watchdog-center/config/config.go
Normal file
@@ -0,0 +1,11 @@
|
||||
package config
|
||||
|
||||
// Config 配置模型
|
||||
type Config struct {
|
||||
Server ServerConfig `json:"server"` // 服务器配置
|
||||
}
|
||||
|
||||
// ServerConfig 服务器配置
|
||||
type ServerConfig struct {
|
||||
Port int `json:"port"` // 服务器端口
|
||||
}
|
||||
74
cmii-uav-watchdog-center/config/config_loader.go
Normal file
74
cmii-uav-watchdog-center/config/config_loader.go
Normal file
@@ -0,0 +1,74 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"cmii-uav-watchdog-common/wdd_log"
|
||||
"encoding/json"
|
||||
"os"
|
||||
"sync"
|
||||
)
|
||||
|
||||
var (
|
||||
config *Config
|
||||
configOnce sync.Once
|
||||
)
|
||||
|
||||
// LoadConfig 加载配置
|
||||
func LoadConfig(configPath string) (*Config, error) {
|
||||
configOnce.Do(func() {
|
||||
// 读取配置文件
|
||||
data, err := os.ReadFile(configPath)
|
||||
if err != nil {
|
||||
// 如果配置文件不存在,则创建默认配置
|
||||
if os.IsNotExist(err) {
|
||||
config = createDefaultConfig()
|
||||
// 保存默认配置
|
||||
saveConfig(configPath, config)
|
||||
return
|
||||
}
|
||||
wdd_log.Error("读取配置文件失败: %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
// 解析配置
|
||||
var cfg Config
|
||||
if err := json.Unmarshal(data, &cfg); err != nil {
|
||||
wdd_log.Error("解析配置文件失败: %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
config = &cfg
|
||||
})
|
||||
|
||||
return config, nil
|
||||
}
|
||||
|
||||
// GetConfig 获取配置
|
||||
func GetConfig() *Config {
|
||||
if config == nil {
|
||||
// 如果配置未初始化,则加载默认配置
|
||||
config = createDefaultConfig()
|
||||
}
|
||||
return config
|
||||
}
|
||||
|
||||
// 创建默认配置
|
||||
func createDefaultConfig() *Config {
|
||||
return &Config{
|
||||
Server: ServerConfig{
|
||||
Port: 8080,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
// 保存配置
|
||||
func saveConfig(configPath string, cfg *Config) {
|
||||
data, err := json.MarshalIndent(cfg, "", " ")
|
||||
if err != nil {
|
||||
wdd_log.Error("序列化配置失败: %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
if err := os.WriteFile(configPath, data, 0644); err != nil {
|
||||
wdd_log.Error("保存配置文件失败: %v", err)
|
||||
}
|
||||
}
|
||||
38
cmii-uav-watchdog-center/controllers/auth_controller.go
Normal file
38
cmii-uav-watchdog-center/controllers/auth_controller.go
Normal file
@@ -0,0 +1,38 @@
|
||||
package controllers
|
||||
|
||||
import (
|
||||
"cmii-uav-watchdog-center/services"
|
||||
"cmii-uav-watchdog-common/models"
|
||||
"net/http"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
// ProcessAuthorizationFile 处理授权文件
|
||||
// @Summary 处理授权文件
|
||||
// @Description 接收来自cmii-uav-watchdog的授权文件,解码并生成授权码
|
||||
// @Accept json
|
||||
// @Produce json
|
||||
// @Param authFile body models.AuthorizationFile true "授权文件"
|
||||
// @Success 200 {object} models.AuthorizationCode "授权码"
|
||||
// @Failure 400 {object} map[string]string "请求错误"
|
||||
// @Failure 401 {object} map[string]string "未授权"
|
||||
// @Router /api/v1/auth/process [post]
|
||||
func ProcessAuthorizationFile(c *gin.Context) {
|
||||
var authFile models.AuthorizationFile
|
||||
|
||||
// 解析请求体
|
||||
if err := c.ShouldBindJSON(&authFile); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "无效的请求体"})
|
||||
return
|
||||
}
|
||||
|
||||
authCode, err := services.ProcessAuthorizationFile(authFile)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "生成授权码失败"})
|
||||
return
|
||||
}
|
||||
|
||||
// 返回授权码
|
||||
c.JSON(http.StatusOK, authCode)
|
||||
}
|
||||
45
cmii-uav-watchdog-center/controllers/project_controller.go
Normal file
45
cmii-uav-watchdog-center/controllers/project_controller.go
Normal file
@@ -0,0 +1,45 @@
|
||||
package controllers
|
||||
|
||||
import (
|
||||
"cmii-uav-watchdog-center/services"
|
||||
"cmii-uav-watchdog-common/models"
|
||||
"cmii-uav-watchdog-common/wdd_log"
|
||||
"net/http"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
// GetProjectList 获取项目列表
|
||||
func GetProjectList(c *gin.Context) {
|
||||
|
||||
// 获取项目列表
|
||||
projectList, err := services.GetProjectList()
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, projectList)
|
||||
|
||||
}
|
||||
|
||||
// CreateProject 创建项目
|
||||
func CreateProject(c *gin.Context) {
|
||||
// 绑定 Project
|
||||
var project models.Project
|
||||
if err := c.ShouldBindJSON(&project); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
// 创建项目
|
||||
authFilePath, err := services.CreateProject(&project)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
wdd_log.Info("创建项目成功: %s", authFilePath)
|
||||
|
||||
c.JSON(http.StatusOK, project)
|
||||
}
|
||||
@@ -1,3 +1,40 @@
|
||||
module cmii-uav-watchdog-center
|
||||
|
||||
go 1.23
|
||||
go 1.24
|
||||
|
||||
require (
|
||||
cmii-uav-watchdog-common v0.0.0
|
||||
cmii-uav-watchdog-otp v0.0.0
|
||||
github.com/gin-gonic/gin v1.9.1
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/bytedance/sonic v1.9.1 // indirect
|
||||
github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 // indirect
|
||||
github.com/gabriel-vasile/mimetype v1.4.2 // indirect
|
||||
github.com/gin-contrib/sse v0.1.0 // indirect
|
||||
github.com/go-playground/locales v0.14.1 // indirect
|
||||
github.com/go-playground/universal-translator v0.18.1 // indirect
|
||||
github.com/go-playground/validator/v10 v10.14.0 // indirect
|
||||
github.com/goccy/go-json v0.10.2 // indirect
|
||||
github.com/json-iterator/go v1.1.12 // indirect
|
||||
github.com/klauspost/cpuid/v2 v2.2.4 // indirect
|
||||
github.com/leodido/go-urn v1.2.4 // indirect
|
||||
github.com/mattn/go-isatty v0.0.19 // indirect
|
||||
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
|
||||
github.com/modern-go/reflect2 v1.0.2 // indirect
|
||||
github.com/pelletier/go-toml/v2 v2.0.8 // indirect
|
||||
github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
|
||||
github.com/ugorji/go/codec v1.2.11 // indirect
|
||||
golang.org/x/arch v0.3.0 // indirect
|
||||
golang.org/x/crypto v0.9.0 // indirect
|
||||
golang.org/x/net v0.10.0 // indirect
|
||||
golang.org/x/sys v0.8.0 // indirect
|
||||
golang.org/x/text v0.9.0 // indirect
|
||||
google.golang.org/protobuf v1.30.0 // indirect
|
||||
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||
)
|
||||
|
||||
replace cmii-uav-watchdog-common => ../cmii-uav-watchdog-common
|
||||
|
||||
replace cmii-uav-watchdog-otp => ../cmii-uav-watchdog-otp
|
||||
|
||||
86
cmii-uav-watchdog-center/go.sum
Normal file
86
cmii-uav-watchdog-center/go.sum
Normal file
@@ -0,0 +1,86 @@
|
||||
github.com/bytedance/sonic v1.5.0/go.mod h1:ED5hyg4y6t3/9Ku1R6dU/4KyJ48DZ4jPhfY1O2AihPM=
|
||||
github.com/bytedance/sonic v1.9.1 h1:6iJ6NqdoxCDr6mbY8h18oSO+cShGSMRGCEo7F2h0x8s=
|
||||
github.com/bytedance/sonic v1.9.1/go.mod h1:i736AoUSYt75HyZLoJW9ERYxcy6eaN6h4BZXU064P/U=
|
||||
github.com/chenzhuoyu/base64x v0.0.0-20211019084208-fb5309c8db06/go.mod h1:DH46F32mSOjUmXrMHnKwZdA8wcEefY7UVqBKYGjpdQY=
|
||||
github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 h1:qSGYFH7+jGhDF8vLC+iwCD4WpbV1EBDSzWkJODFLams=
|
||||
github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311/go.mod h1:b583jCggY9gE99b6G5LEC39OIiVsWj+R97kbl5odCEk=
|
||||
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/gabriel-vasile/mimetype v1.4.2 h1:w5qFW6JKBz9Y393Y4q372O9A7cUSequkh1Q7OhCmWKU=
|
||||
github.com/gabriel-vasile/mimetype v1.4.2/go.mod h1:zApsH/mKG4w07erKIaJPFiX0Tsq9BFQgN3qGY5GnNgA=
|
||||
github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE=
|
||||
github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=
|
||||
github.com/gin-gonic/gin v1.9.1 h1:4idEAncQnU5cB7BeOkPtxjfCSye0AAm1R0RVIqJ+Jmg=
|
||||
github.com/gin-gonic/gin v1.9.1/go.mod h1:hPrL7YrpYKXt5YId3A/Tnip5kqbEAP+KLuI3SUcPTeU=
|
||||
github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s=
|
||||
github.com/go-playground/assert/v2 v2.2.0/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4=
|
||||
github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA=
|
||||
github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY=
|
||||
github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY=
|
||||
github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
|
||||
github.com/go-playground/validator/v10 v10.14.0 h1:vgvQWe3XCz3gIeFDm/HnTIbj6UGmg/+t63MyGU2n5js=
|
||||
github.com/go-playground/validator/v10 v10.14.0/go.mod h1:9iXMNT7sEkjXb0I+enO7QXmzG6QCsPWY4zveKFVRSyU=
|
||||
github.com/goccy/go-json v0.10.2 h1:CrxCmQqYDkv1z7lO7Wbh2HN93uovUHgrECaO5ZrCXAU=
|
||||
github.com/goccy/go-json v0.10.2/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I=
|
||||
github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
|
||||
github.com/google/go-cmp v0.5.5 h1:Khx7svrCpmxxtHBq5j2mp/xVjsi8hQMfNLvJFAlrGgU=
|
||||
github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
||||
github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
|
||||
github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
|
||||
github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
|
||||
github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
|
||||
github.com/klauspost/cpuid/v2 v2.2.4 h1:acbojRNwl3o09bUq+yDCtZFc1aiwaAAxtcn8YkZXnvk=
|
||||
github.com/klauspost/cpuid/v2 v2.2.4/go.mod h1:RVVoqg1df56z8g3pUjL/3lE5UfnlrJX8tyFgg4nqhuY=
|
||||
github.com/leodido/go-urn v1.2.4 h1:XlAE/cm/ms7TE/VMVoduSpNBoyc2dOxHs5MZSwAN63Q=
|
||||
github.com/leodido/go-urn v1.2.4/go.mod h1:7ZrI8mTSeBSHl/UaRyKQW1qZeMgak41ANeCNaVckg+4=
|
||||
github.com/mattn/go-isatty v0.0.19 h1:JITubQf0MOLdlGRuRq+jtsDlekdYPia9ZFsB8h/APPA=
|
||||
github.com/mattn/go-isatty v0.0.19/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
|
||||
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg=
|
||||
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
|
||||
github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M=
|
||||
github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk=
|
||||
github.com/pelletier/go-toml/v2 v2.0.8 h1:0ctb6s9mE31h0/lhu+J6OPmVeDxJn+kYnJc2jZR9tGQ=
|
||||
github.com/pelletier/go-toml/v2 v2.0.8/go.mod h1:vuYfssBdrU2XDZ9bYydBu6t+6a6PYNcZljzZR9VXg+4=
|
||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||
github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
|
||||
github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
|
||||
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
|
||||
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||
github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||
github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
|
||||
github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
|
||||
github.com/stretchr/testify v1.8.2/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
|
||||
github.com/stretchr/testify v1.8.3 h1:RP3t2pwF7cMEbC1dqtB6poj3niw/9gnV4Cjg5oW5gtY=
|
||||
github.com/stretchr/testify v1.8.3/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
|
||||
github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=
|
||||
github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08=
|
||||
github.com/ugorji/go/codec v1.2.11 h1:BMaWp1Bb6fHwEtbplGBGJ498wD+LKlNSl25MjdZY4dU=
|
||||
github.com/ugorji/go/codec v1.2.11/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg=
|
||||
golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8=
|
||||
golang.org/x/arch v0.3.0 h1:02VY4/ZcO/gBOH6PUaoiptASxtXU10jazRCP865E97k=
|
||||
golang.org/x/arch v0.3.0/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8=
|
||||
golang.org/x/crypto v0.9.0 h1:LF6fAI+IutBocDJ2OT0Q1g8plpYljMZ4+lty+dsqw3g=
|
||||
golang.org/x/crypto v0.9.0/go.mod h1:yrmDGqONDYtNj3tH8X9dzUun2m2lzPa9ngI6/RUPGR0=
|
||||
golang.org/x/net v0.10.0 h1:X2//UzNDwYmtCLn7To6G58Wr6f5ahEAQgKNzv9Y951M=
|
||||
golang.org/x/net v0.10.0/go.mod h1:0qNGK6F8kojg2nk9dLZ2mShWaEBan6FAoqfSigmmuDg=
|
||||
golang.org/x/sys v0.0.0-20220704084225-05e143d24a9e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.8.0 h1:EBmGv8NaZBZTWvrbjNoL6HVt+IVy3QDQpJs7VRIw3tU=
|
||||
golang.org/x/sys v0.8.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/text v0.9.0 h1:2sjJmO8cDvYveuX97RDLsxlyUxLl+GHoLxBiRdHllBE=
|
||||
golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8=
|
||||
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4=
|
||||
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||
google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=
|
||||
google.golang.org/protobuf v1.30.0 h1:kPPoIgf3TsEvrm0PFe15JQ+570QVxYzEvvHqChK+cng=
|
||||
google.golang.org/protobuf v1.30.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
rsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4=
|
||||
36
cmii-uav-watchdog-center/router/router.go
Normal file
36
cmii-uav-watchdog-center/router/router.go
Normal file
@@ -0,0 +1,36 @@
|
||||
package router
|
||||
|
||||
import (
|
||||
"cmii-uav-watchdog-center/controllers"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
// SetupRouter 设置路由
|
||||
func SetupRouter() *gin.Engine {
|
||||
// 创建一个默认的路由引擎
|
||||
r := gin.Default()
|
||||
|
||||
// API路由组
|
||||
apiV1 := r.Group("/api")
|
||||
{
|
||||
// 授权相关路由
|
||||
auth := apiV1.Group("/auth")
|
||||
{
|
||||
// 处理授权文件
|
||||
auth.POST("/process", controllers.ProcessAuthorizationFile)
|
||||
}
|
||||
|
||||
// 项目相关路由
|
||||
project := apiV1.Group("/project")
|
||||
{
|
||||
// 获取项目列表
|
||||
project.GET("/list", controllers.GetProjectList)
|
||||
// 创建项目
|
||||
project.POST("/create", controllers.CreateProject)
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
return r
|
||||
}
|
||||
64
cmii-uav-watchdog-center/services/auth_service.go
Normal file
64
cmii-uav-watchdog-center/services/auth_service.go
Normal file
@@ -0,0 +1,64 @@
|
||||
package services
|
||||
|
||||
import (
|
||||
"cmii-uav-watchdog-common/models"
|
||||
"cmii-uav-watchdog-common/totp_tier_one"
|
||||
"cmii-uav-watchdog-common/utils"
|
||||
"cmii-uav-watchdog-common/wdd_log"
|
||||
)
|
||||
|
||||
func ProcessAuthorizationFile(authFile models.AuthorizationFile) (models.AuthorizationCode, error) {
|
||||
// 解密项目NameSpace
|
||||
projectNamespace, err := totp_tier_one.Decrypt(authFile.EncryptedNamespace, authFile.ProjectNamespace)
|
||||
if err != nil {
|
||||
wdd_log.Error("解密项目Name失败: %v", err)
|
||||
return models.AuthorizationCode{}, err
|
||||
}
|
||||
|
||||
// 获取项目信息
|
||||
projectInfo, err := GetProjectInfo(projectNamespace)
|
||||
if err != nil {
|
||||
wdd_log.Error("获取项目信息失败: %v", err)
|
||||
return models.AuthorizationCode{}, err
|
||||
}
|
||||
|
||||
// 验证TOTP验证码
|
||||
if !totp_tier_one.VerifyTierOneTOTPCode(authFile.TOTPCode, projectInfo.TierOneSecret) {
|
||||
wdd_log.Warn("TOTP验证失败: %s", authFile.TOTPCode)
|
||||
return models.AuthorizationCode{}, err
|
||||
}
|
||||
|
||||
// 处理加密的主机信息
|
||||
|
||||
// 一级TOTP密钥
|
||||
key := projectInfo.TierOneSecret
|
||||
|
||||
// 注意:由于我们无法解密主机信息(我们只需要验证其有效性),所以直接使用
|
||||
authorizedHostMap := make(map[string]models.HostInfo)
|
||||
for encryptedInfo, hostInfo := range authFile.EncryptedHostMap {
|
||||
// 使用 DecryptHostInfo 验证主机信息是否被篡改
|
||||
isOK, err := totp_tier_one.DecryptHostInfo(encryptedInfo, hostInfo, key)
|
||||
if !isOK || err != nil {
|
||||
wdd_log.Error("主机信息验证失败: %v", err)
|
||||
continue
|
||||
}
|
||||
|
||||
// 验证通过,将主机信息添加到授权列表
|
||||
authorizedHostMap[encryptedInfo] = hostInfo
|
||||
}
|
||||
|
||||
// 生成新的TOTP验证码
|
||||
newTOTPCode, err := totp_tier_one.GenerateTierOneTOTPCode(projectInfo.TierOneSecret)
|
||||
if err != nil {
|
||||
wdd_log.Error("生成TOTP验证码失败: %v", err)
|
||||
return models.AuthorizationCode{}, err
|
||||
}
|
||||
|
||||
return models.AuthorizationCode{
|
||||
TOTPCode: newTOTPCode,
|
||||
CurrentTime: utils.CurentTimeString(),
|
||||
EncryptedHostMap: authorizedHostMap,
|
||||
ProjectNamespace: projectNamespace,
|
||||
EncryptedNamespace: authFile.EncryptedNamespace,
|
||||
}, nil
|
||||
}
|
||||
101
cmii-uav-watchdog-center/services/project_service.go
Normal file
101
cmii-uav-watchdog-center/services/project_service.go
Normal file
@@ -0,0 +1,101 @@
|
||||
package services
|
||||
|
||||
import (
|
||||
"cmii-uav-watchdog-common/models"
|
||||
"cmii-uav-watchdog-common/utils"
|
||||
"cmii-uav-watchdog-common/wdd_log"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"os"
|
||||
)
|
||||
|
||||
const (
|
||||
ProjectFilePath = "C:\\Users\\wddsh\\Documents\\IdeaProjects\\cmii-uav-watchdog-project\\cmii-uav-watchdog-center\\project_file\\"
|
||||
)
|
||||
|
||||
func CreateProject(project *models.Project) (string, error) {
|
||||
|
||||
// 生成TierOneSecret
|
||||
secret, err := GenerateTierOneTOTPSecret()
|
||||
if err != nil {
|
||||
wdd_log.Error("生成TierOneSecret失败: %v", err)
|
||||
return "", err
|
||||
}
|
||||
|
||||
project.TierOneSecret = secret
|
||||
project.CreateTime = utils.CurentTimeString()
|
||||
|
||||
// 存储 project到file中
|
||||
project.AuthFilePath = fmt.Sprintf("%s%s-%s.json", ProjectFilePath, project.Namespace, project.Name)
|
||||
|
||||
// 存储到file中
|
||||
jsonData, err := json.Marshal(project)
|
||||
if err != nil {
|
||||
wdd_log.Error("序列化失败: %v", err)
|
||||
return "", err
|
||||
}
|
||||
|
||||
// 写入文件
|
||||
err = os.WriteFile(project.AuthFilePath, jsonData, 0644)
|
||||
if err != nil {
|
||||
wdd_log.Error("写入文件失败: %v", err)
|
||||
return "", err
|
||||
}
|
||||
|
||||
return project.AuthFilePath, nil
|
||||
}
|
||||
|
||||
// GetProjectList 获取项目列表
|
||||
func GetProjectList() ([]models.Project, error) {
|
||||
|
||||
// 获取项目列表
|
||||
projectFileList, err := os.ReadDir(ProjectFilePath)
|
||||
if err != nil {
|
||||
wdd_log.Error("读取项目列表失败: %v", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
projectList := make([]models.Project, 0)
|
||||
|
||||
// 遍历项目列表
|
||||
for _, projectFilePath := range projectFileList {
|
||||
// 读取项目文件
|
||||
projectFile, err := os.ReadFile(fmt.Sprintf("%s%s", ProjectFilePath, projectFilePath.Name()))
|
||||
if err != nil {
|
||||
wdd_log.Error("读取项目文件失败: %v", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 反序列化
|
||||
var project models.Project
|
||||
err = json.Unmarshal(projectFile, &project)
|
||||
if err != nil {
|
||||
wdd_log.Error("反序列化失败: %v", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 添加到项目列表
|
||||
projectList = append(projectList, project)
|
||||
}
|
||||
|
||||
return projectList, nil
|
||||
}
|
||||
|
||||
// GetProjectInfo 获取项目信息
|
||||
func GetProjectInfo(projectNamespace string) (models.Project, error) {
|
||||
// 获取项目列表
|
||||
projectList, err := GetProjectList()
|
||||
if err != nil {
|
||||
wdd_log.Error("获取项目列表失败: %v", err)
|
||||
return models.Project{}, err
|
||||
}
|
||||
|
||||
// 遍历项目列表
|
||||
for _, project := range projectList {
|
||||
if project.Namespace == projectNamespace {
|
||||
return project, nil
|
||||
}
|
||||
}
|
||||
|
||||
return models.Project{}, fmt.Errorf("项目不存在")
|
||||
}
|
||||
18
cmii-uav-watchdog-center/services/totp_service.go
Normal file
18
cmii-uav-watchdog-center/services/totp_service.go
Normal file
@@ -0,0 +1,18 @@
|
||||
package services
|
||||
|
||||
import (
|
||||
"cmii-uav-watchdog-common/totp_tier_one"
|
||||
"cmii-uav-watchdog-otp/totp"
|
||||
"log"
|
||||
)
|
||||
|
||||
// GenerateTierOneTOTPSecret 生成一级TOTP密钥 只能center调用
|
||||
func GenerateTierOneTOTPSecret() (string, error) {
|
||||
secret, err := totp.Generate(totp_tier_one.TierOneTOTPSecretOpts)
|
||||
if err != nil {
|
||||
log.Printf("生成TOTP密钥失败: %v", err)
|
||||
return "", err
|
||||
}
|
||||
|
||||
return secret.Secret(), nil
|
||||
}
|
||||
Reference in New Issue
Block a user