Flik
5a03d9e1f1
- Deleted version comparison logic from `pkg/plugin/sign/version.go`. - Removed unused types and constants from `pkg/plugin/types.go`. - Updated `pkg/protocol/message.go` to remove plugin-related message types. - Enhanced `pkg/proxy/http.go` and `pkg/proxy/socks5.go` to include username/password authentication for HTTP and SOCKS5 proxies. - Modified `pkg/proxy/server.go` to pass authentication parameters to server constructors. - Added new API endpoint to generate installation commands with a token for clients. - Created database functions to manage installation tokens in `internal/server/db/install_token.go`. - Implemented the installation command generation logic in `internal/server/router/handler/install.go`. - Updated web frontend to support installation command generation and display in `web/src/views/ClientsView.vue`.
124 lines
3.1 KiB
Go
124 lines
3.1 KiB
Go
package main
|
||
|
||
// @title GoTunnel API
|
||
// @version 1.0
|
||
// @description GoTunnel 内网穿透服务器 API
|
||
// @host localhost:7500
|
||
// @BasePath /
|
||
// @securityDefinitions.apikey Bearer
|
||
// @in header
|
||
// @name Authorization
|
||
// @description JWT Bearer token
|
||
|
||
import (
|
||
"flag"
|
||
"fmt"
|
||
"log"
|
||
"os"
|
||
"os/signal"
|
||
"syscall"
|
||
"time"
|
||
|
||
_ "github.com/gotunnel/docs" // Swagger docs
|
||
|
||
"github.com/gotunnel/internal/server/app"
|
||
"github.com/gotunnel/internal/server/config"
|
||
"github.com/gotunnel/internal/server/db"
|
||
"github.com/gotunnel/internal/server/tunnel"
|
||
"github.com/gotunnel/pkg/crypto"
|
||
"github.com/gotunnel/pkg/version"
|
||
)
|
||
|
||
// 版本信息（通过 ldflags 注入）
|
||
var Version string
|
||
var BuildTime string
|
||
var GitCommit string
|
||
|
||
func init() {
|
||
version.SetVersion(Version)
|
||
version.SetBuildInfo(GitCommit, BuildTime)
|
||
}
|
||
|
||
func main() {
|
||
configPath := flag.String("c", "server.yaml", "config file path")
|
||
flag.Parse()
|
||
|
||
// 加载 YAML 配置
|
||
cfg, err := config.LoadServerConfig(*configPath)
|
||
if err != nil {
|
||
log.Fatalf("Load config error: %v", err)
|
||
}
|
||
|
||
// 初始化数据库
|
||
clientStore, err := db.NewSQLiteStore(cfg.Server.DBPath)
|
||
if err != nil {
|
||
log.Fatalf("Init database error: %v", err)
|
||
}
|
||
defer clientStore.Close()
|
||
|
||
// 打印 token（便于客户端连接）
|
||
log.Printf("[Server] Token: %s", cfg.Server.Token)
|
||
|
||
// 创建隧道服务
|
||
server := tunnel.NewServer(
|
||
clientStore,
|
||
cfg.Server.BindAddr,
|
||
cfg.Server.BindPort,
|
||
cfg.Server.Token,
|
||
cfg.Server.HeartbeatSec,
|
||
cfg.Server.HeartbeatTimeout,
|
||
)
|
||
|
||
// 配置 TLS（默认启用）
|
||
if !cfg.Server.TLSDisabled {
|
||
tlsConfig, err := crypto.GenerateTLSConfig()
|
||
if err != nil {
|
||
log.Fatalf("Generate TLS config error: %v", err)
|
||
}
|
||
server.SetTLSConfig(tlsConfig)
|
||
log.Printf("[Server] TLS enabled")
|
||
}
|
||
|
||
// 设置流量存储，用于记录流量统计
|
||
server.SetTrafficStore(clientStore)
|
||
|
||
// 启动 Web 控制台
|
||
if cfg.Server.Web.Enabled {
|
||
// 强制生成 Web 凭据（如果未配置）
|
||
if config.GenerateWebCredentials(cfg) {
|
||
log.Printf("[Web] Auto-generated credentials - Username: %s, Password: %s",
|
||
cfg.Server.Web.Username, cfg.Server.Web.Password)
|
||
log.Printf("[Web] Please save these credentials and update your config file")
|
||
// 保存配置以持久化凭据
|
||
if err := config.SaveServerConfig(*configPath, cfg); err != nil {
|
||
log.Printf("[Web] Warning: failed to save config: %v", err)
|
||
}
|
||
}
|
||
|
||
ws := app.NewWebServer(clientStore, server, cfg, *configPath, clientStore)
|
||
addr := fmt.Sprintf("%s:%d", cfg.Server.BindAddr, cfg.Server.Web.BindPort)
|
||
|
||
go func() {
|
||
// 始终使用 JWT 认证
|
||
err := ws.RunWithJWT(addr, cfg.Server.Web.Username, cfg.Server.Web.Password, cfg.Server.Token)
|
||
if err != nil {
|
||
log.Printf("[Web] Server error: %v", err)
|
||
}
|
||
}()
|
||
log.Printf("[Web] Console running at http://%s (authentication required)", addr)
|
||
}
|
||
|
||
// 优雅关闭信号处理
|
||
quit := make(chan os.Signal, 1)
|
||
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
|
||
|
||
go func() {
|
||
<-quit
|
||
log.Printf("[Server] Received shutdown signal")
|
||
server.Shutdown(30 * time.Second)
|
||
os.Exit(0)
|
||
}()
|
||
|
||
log.Fatal(server.Run())
|
||
}
|