web.go 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760
  1. // Package web provides the main web server implementation for the 3x-ui panel,
  2. // including HTTP/HTTPS serving, routing, templates, and background job scheduling.
  3. package web
  4. import (
  5. "context"
  6. "crypto/tls"
  7. "embed"
  8. "fmt"
  9. "io"
  10. "io/fs"
  11. "net"
  12. "net/http"
  13. "os"
  14. "strconv"
  15. "strings"
  16. "time"
  17. "github.com/mhsanaei/3x-ui/v3/internal/amneziawgnet"
  18. "github.com/mhsanaei/3x-ui/v3/internal/config"
  19. "github.com/mhsanaei/3x-ui/v3/internal/eventbus"
  20. "github.com/mhsanaei/3x-ui/v3/internal/logger"
  21. "github.com/mhsanaei/3x-ui/v3/internal/mtproto"
  22. "github.com/mhsanaei/3x-ui/v3/internal/tuic"
  23. "github.com/mhsanaei/3x-ui/v3/internal/util/common"
  24. "github.com/mhsanaei/3x-ui/v3/internal/util/sys"
  25. "github.com/mhsanaei/3x-ui/v3/internal/web/controller"
  26. "github.com/mhsanaei/3x-ui/v3/internal/web/job"
  27. "github.com/mhsanaei/3x-ui/v3/internal/web/locale"
  28. "github.com/mhsanaei/3x-ui/v3/internal/web/middleware"
  29. "github.com/mhsanaei/3x-ui/v3/internal/web/network"
  30. "github.com/mhsanaei/3x-ui/v3/internal/web/runtime"
  31. "github.com/mhsanaei/3x-ui/v3/internal/web/service"
  32. "github.com/mhsanaei/3x-ui/v3/internal/web/service/email"
  33. "github.com/mhsanaei/3x-ui/v3/internal/web/service/panel"
  34. "github.com/mhsanaei/3x-ui/v3/internal/web/service/tgbot"
  35. "github.com/mhsanaei/3x-ui/v3/internal/web/websocket"
  36. "github.com/mhsanaei/3x-ui/v3/internal/xray"
  37. "github.com/gin-contrib/gzip"
  38. "github.com/gin-contrib/sessions"
  39. "github.com/gin-contrib/sessions/cookie"
  40. "github.com/gin-gonic/gin"
  41. "github.com/robfig/cron/v3"
  42. )
  43. //go:embed translation/*
  44. var i18nFS embed.FS
  45. // distFS embeds the Vite-built frontend (internal/web/dist/). Every user-facing
  46. // HTML route is served straight out of this FS — the legacy Go
  47. // templates and `web/assets/` tree are gone post-Phase 8.
  48. //go:embed all:dist
  49. var distFS embed.FS
  50. var startTime = time.Now()
  51. // cronPanicLogger adapts the package logger to cron's Printf-style logger so a
  52. // panicking scheduled job is recovered and logged instead of crashing the panel.
  53. type cronPanicLogger struct{}
  54. func (cronPanicLogger) Printf(format string, args ...any) { logger.Errorf(format, args...) }
  55. // wrapDistFS adapts the embedded `dist/` directory so it can be mounted
  56. // as the panel's `/assets/` static route. Vite emits its bundled JS/CSS
  57. // under `dist/assets/`; serving the FS rooted at `dist/assets` makes
  58. // `/assets/<hash>.js` URLs resolve directly.
  59. type wrapDistFS struct {
  60. embed.FS
  61. }
  62. func (f *wrapDistFS) Open(name string) (fs.File, error) {
  63. file, err := f.FS.Open("dist/assets/" + name)
  64. if err != nil {
  65. return nil, err
  66. }
  67. return &wrapAssetsFile{
  68. File: file,
  69. }, nil
  70. }
  71. type wrapAssetsFile struct {
  72. fs.File
  73. }
  74. func (f *wrapAssetsFile) Stat() (fs.FileInfo, error) {
  75. info, err := f.File.Stat()
  76. if err != nil {
  77. return nil, err
  78. }
  79. return &wrapAssetsFileInfo{
  80. FileInfo: info,
  81. }, nil
  82. }
  83. type wrapAssetsFileInfo struct {
  84. fs.FileInfo
  85. }
  86. func (f *wrapAssetsFileInfo) ModTime() time.Time {
  87. return startTime
  88. }
  89. // EmbeddedDist returns the embedded Vite-built frontend filesystem.
  90. // Controllers serve their HTML out of this FS via the dist-page handler
  91. // installed in NewEngine().
  92. func EmbeddedDist() embed.FS {
  93. return distFS
  94. }
  95. // Server represents the main web server for the 3x-ui panel with controllers, services, and scheduled jobs.
  96. type Server struct {
  97. httpServer *http.Server
  98. listener net.Listener
  99. index *controller.IndexController
  100. panel *controller.XUIController
  101. api *controller.APIController
  102. ws *controller.WebSocketController
  103. xrayService service.XrayService
  104. settingService service.SettingService
  105. tgbotService tgbot.Tgbot
  106. wsHub *websocket.Hub
  107. bus *eventbus.Bus
  108. cron *cron.Cron
  109. ctx context.Context
  110. cancel context.CancelFunc
  111. }
  112. // NewServer creates a new web server instance with a cancellable context.
  113. func NewServer() *Server {
  114. ctx, cancel := context.WithCancel(context.Background())
  115. return &Server{
  116. ctx: ctx,
  117. cancel: cancel,
  118. }
  119. }
  120. func (s *Server) isDirectHTTPSConfigured() bool {
  121. certFile, certErr := s.settingService.GetCertFile()
  122. keyFile, keyErr := s.settingService.GetKeyFile()
  123. if certErr != nil || keyErr != nil || certFile == "" || keyFile == "" {
  124. return false
  125. }
  126. _, err := tls.LoadX509KeyPair(certFile, keyFile)
  127. return err == nil
  128. }
  129. // initRouter initializes Gin, registers middleware, templates, static
  130. // assets, controllers and returns the configured engine.
  131. func (s *Server) initRouter() (*gin.Engine, error) {
  132. if config.IsDebug() {
  133. gin.SetMode(gin.DebugMode)
  134. } else {
  135. gin.DefaultWriter = io.Discard
  136. gin.DefaultErrorWriter = io.Discard
  137. gin.SetMode(gin.ReleaseMode)
  138. }
  139. engine := gin.Default()
  140. directHTTPS := s.isDirectHTTPSConfigured()
  141. sendHSTS := directHTTPS && !config.IsSkipHSTS()
  142. engine.Use(middleware.SecurityHeadersMiddleware(sendHSTS))
  143. // Cap request bodies on state-changing requests so a stolen session/API
  144. // token or a buggy client can't force large allocations or long DB
  145. // transactions via bulk create/attach/import endpoints. GET/HEAD/OPTIONS
  146. // carry no body and are left untouched. Database restore legitimately accepts
  147. // large backups and streams them to disk, so only its exact route suffix is
  148. // exempt. Follow-up: make the limit a setting.
  149. const maxRequestBodyBytes = 10 << 20 // 10 MiB
  150. engine.Use(middleware.MaxBodyBytes(maxRequestBodyBytes, "/panel/api/server/importDB"))
  151. webDomain, err := s.settingService.GetWebDomain()
  152. if err != nil {
  153. return nil, err
  154. }
  155. if webDomain != "" {
  156. engine.Use(middleware.DomainValidatorMiddleware(webDomain))
  157. }
  158. secret, err := s.settingService.GetSecret()
  159. if err != nil {
  160. return nil, err
  161. }
  162. basePath, err := s.settingService.GetBasePath()
  163. if err != nil {
  164. return nil, err
  165. }
  166. engine.Use(gzip.Gzip(gzip.DefaultCompression))
  167. assetsBasePath := basePath + "assets/"
  168. store := cookie.NewStore(secret)
  169. // Configure default session cookie options, including expiration (MaxAge)
  170. sessionOptions := sessions.Options{
  171. Path: basePath,
  172. HttpOnly: true,
  173. Secure: directHTTPS,
  174. SameSite: http.SameSiteLaxMode,
  175. }
  176. if sessionMaxAge, err := s.settingService.GetSessionMaxAge(); err == nil && sessionMaxAge > 0 {
  177. sessionOptions.MaxAge = sessionMaxAge * 60 // minutes -> seconds
  178. }
  179. store.Options(sessionOptions)
  180. engine.Use(sessions.Sessions("3x-ui", store))
  181. engine.Use(func(c *gin.Context) {
  182. c.Set("base_path", basePath)
  183. })
  184. engine.Use(func(c *gin.Context) {
  185. uri := c.Request.RequestURI
  186. if strings.HasPrefix(uri, assetsBasePath) {
  187. c.Header("Cache-Control", "max-age=31536000")
  188. }
  189. })
  190. // init i18n — still used by backend strings (errors, log messages,
  191. // SubPage menu entries) even though the Go template engine is gone.
  192. err = locale.InitLocalizer(i18nFS, &s.settingService)
  193. if err != nil {
  194. return nil, err
  195. }
  196. engine.Use(locale.LocalizerMiddleware())
  197. // `/assets/` serves the Vite-built bundle. In dev we pull from disk
  198. // so the Vite watcher's incremental rebuilds show up without
  199. // restarting the binary; in prod we serve the embedded dist FS
  200. // rooted at `dist/assets/`.
  201. if config.IsDebug() {
  202. engine.StaticFS(basePath+"assets", http.FS(os.DirFS("internal/web/dist/assets")))
  203. } else {
  204. engine.StaticFS(basePath+"assets", http.FS(&wrapDistFS{FS: distFS}))
  205. }
  206. // Hand the embedded `dist/` filesystem to the controller package
  207. // before any HTML-serving controller is constructed. Phase 8
  208. // cutover: every HTML route reads from internal/web/dist/ instead of
  209. // rendering a legacy template.
  210. controller.SetDistFS(distFS)
  211. g := engine.Group(basePath)
  212. g.GET("/manifest.webmanifest", controller.ServePWAManifest)
  213. g.GET("/pwa-register.js", controller.ServePWARegister)
  214. g.GET("/service-worker.js", controller.ServePWAServiceWorker)
  215. g.GET("/icons/:name", controller.ServePWAIcon)
  216. s.index = controller.NewIndexController(g)
  217. s.panel = controller.NewXUIController(g)
  218. s.api = controller.NewAPIController(g)
  219. // Initialize WebSocket hub
  220. s.wsHub = websocket.NewHub()
  221. go s.wsHub.Run()
  222. // Initialize WebSocket controller — service owns per-connection pumps,
  223. // controller is HTTP-layer only (auth + upgrade).
  224. s.ws = controller.NewWebSocketController(panel.NewWebSocketService(s.wsHub))
  225. // Register WebSocket route with basePath (g already has basePath prefix)
  226. g.GET("/ws", s.ws.HandleWebSocket)
  227. // Chrome DevTools endpoint for debugging web apps
  228. engine.GET("/.well-known/appspecific/com.chrome.devtools.json", func(c *gin.Context) {
  229. c.JSON(http.StatusOK, gin.H{})
  230. })
  231. // Let unknown panel document routes fall back to the SPA shell, while every
  232. // non-SPA miss still returns a hard 404.
  233. engine.NoRoute(func(c *gin.Context) {
  234. if s.panel.HandleNoRoutePanelSPA(c) {
  235. return
  236. }
  237. c.AbortWithStatus(http.StatusNotFound)
  238. })
  239. return engine, nil
  240. }
  241. // Background-job cadences. Centralized here as the single tuning surface; the
  242. // values are unchanged from the historical hardcoded cron specs. Follow-up:
  243. // make these configurable via settings, add per-tick jitter to de-synchronize
  244. // fleet load, skip expensive jobs when no WebSocket clients are connected or
  245. // node/xray state is unchanged, and export per-job duration/skipped/error
  246. // counters.
  247. const (
  248. cadenceXrayRunning = "@every 1s"
  249. cadenceXrayRestart = "@every 30s"
  250. cadenceXrayTraffic = "@every 5s"
  251. cadenceMtproto = "@every 10s"
  252. cadenceAmneziaWG = "@every 10s"
  253. cadenceTuic = "@every 10s"
  254. cadenceClientIPScan = "@every 10s"
  255. cadenceNodeHeartbeat = "@every 5s"
  256. cadenceNodeTraffic = "@every 5s"
  257. cadenceOutboundSub = "@every 5m"
  258. cadenceReapOrphans = "@every 5m"
  259. cadenceRemoteRouting = "@every 5m"
  260. cadenceXrayLogPrune = "@every 10m"
  261. cadenceCheckHash = "@every 2m"
  262. // cpu.Percent samples over a full minute (blocking), so a finer cadence just
  263. // stacks overlapping samplers; subscribers rate-limit alerts to 1/min anyway.
  264. cadenceCPUAlarm = "@every 1m"
  265. cadenceMemoryAlarm = "@every 1m"
  266. )
  267. // startTask schedules background jobs (Xray checks, traffic jobs, cron
  268. // jobs) which the panel relies on for periodic maintenance and monitoring.
  269. func (s *Server) startTask(restartXray bool, loc *time.Location) {
  270. if restartXray {
  271. err := s.xrayService.RestartXray(true)
  272. if err != nil {
  273. logger.Warning("start xray failed:", err)
  274. }
  275. }
  276. // Check whether xray is running every second
  277. _, _ = s.cron.AddJob(cadenceXrayRunning, job.NewCheckXrayRunningJob())
  278. // Check if xray needs to be restarted every 30 seconds
  279. _, _ = s.cron.AddFunc(cadenceXrayRestart, func() {
  280. s.xrayService.ApplyPendingRestart()
  281. })
  282. go func() {
  283. time.Sleep(time.Second * 5)
  284. _, _ = s.cron.AddJob(cadenceXrayTraffic, job.NewXrayTrafficJob())
  285. }()
  286. // Reconcile mtproto (mtg) sidecars and scrape their traffic
  287. mtJob := job.NewMtprotoJob()
  288. _, _ = s.cron.AddJob(cadenceMtproto, mtJob)
  289. go mtJob.Run()
  290. // Reconcile embedded AmneziaWG interfaces; traffic rides Xray's own stats
  291. awgJob := job.NewAmneziaWGJob()
  292. _, _ = s.cron.AddJob(cadenceAmneziaWG, awgJob)
  293. go awgJob.Run()
  294. tuicJob := job.NewTuicJob()
  295. _, _ = s.cron.AddJob(cadenceTuic, tuicJob)
  296. go tuicJob.Run()
  297. // check client ips from log file every 10 sec
  298. _, _ = s.cron.AddJob(cadenceClientIPScan, job.NewCheckClientIpJob())
  299. _, _ = s.cron.AddJob(cadenceNodeHeartbeat, job.NewNodeHeartbeatJob())
  300. _, _ = s.cron.AddJob(cadenceNodeTraffic, job.NewNodeTrafficSyncJob())
  301. // Outbound subscription auto-refresh (respects per-sub updateInterval)
  302. _, _ = s.cron.AddJob(cadenceOutboundSub, job.NewOutboundSubscriptionJob())
  303. _, _ = s.cron.AddJob(cadenceReapOrphans, job.NewReapSyncOrphansJob())
  304. // Warm permanent routing URLs immediately and refresh them outside the
  305. // latency-sensitive subscription request path.
  306. remoteRoutingJob := job.NewRemoteRoutingJob()
  307. _, _ = s.cron.AddJob(cadenceRemoteRouting, remoteRoutingJob)
  308. common.GoRecover("remote-routing-warm", remoteRoutingJob.Run)
  309. // check client ips from log file every day
  310. _, _ = s.cron.AddJob("@daily", job.NewClearLogsJob())
  311. _, _ = s.cron.AddJob(cadenceXrayLogPrune, job.NewPruneXrayLogsJob())
  312. _, _ = s.cron.AddJob("@hourly", job.NewWarpIpJob())
  313. // Inbound traffic reset jobs
  314. // Run every hour
  315. _, _ = s.cron.AddJob("@hourly", job.NewPeriodicTrafficResetJob("hourly", loc))
  316. // Run once a day, midnight
  317. _, _ = s.cron.AddJob("@daily", job.NewPeriodicTrafficResetJob("daily", loc))
  318. // Run once a week, midnight between Sat/Sun
  319. _, _ = s.cron.AddJob("@weekly", job.NewPeriodicTrafficResetJob("weekly", loc))
  320. // Check monthly reset days at midnight
  321. _, _ = s.cron.AddJob("@daily", job.NewPeriodicTrafficResetJob("monthly", loc))
  322. // LDAP sync scheduling
  323. if ldapEnabled, _ := s.settingService.GetLdapEnable(); ldapEnabled {
  324. runtime, err := s.settingService.GetLdapSyncCron()
  325. if err != nil || runtime == "" {
  326. runtime = "@every 1m"
  327. }
  328. j := job.NewLdapSyncJob()
  329. // job has zero-value services with method receivers that read settings on demand
  330. _, _ = s.cron.AddJob(runtime, j)
  331. }
  332. // Telegram-bot–dependent jobs: periodic stats report + callback-hash cleanup.
  333. isTgbotenabled, err := s.settingService.GetTgbotEnabled()
  334. if (err == nil) && isTgbotenabled {
  335. runtime, err := s.settingService.GetTgbotRuntime()
  336. if err != nil {
  337. logger.Warningf("Add NewStatsNotifyJob: failed to load runtime: %v; using default @daily", err)
  338. runtime = "@daily"
  339. } else if strings.TrimSpace(runtime) == "" {
  340. logger.Warning("Add NewStatsNotifyJob runtime is empty, using default @daily")
  341. runtime = "@daily"
  342. }
  343. logger.Infof("Tg notify enabled,run at %s", runtime)
  344. if _, err = s.cron.AddJob(runtime, job.NewStatsNotifyJob()); err != nil {
  345. logger.Warningf("Add NewStatsNotifyJob: failed to schedule runtime %q: %v", runtime, err)
  346. }
  347. // check for Telegram bot callback query hash storage reset
  348. _, _ = s.cron.AddJob(cadenceCheckHash, job.NewCheckHashStorageJob())
  349. }
  350. // CPU monitor publishes cpu.high events; register it whenever any notifier
  351. // (Telegram or Email) wants them, independent of the Telegram bot being on.
  352. if s.cpuAlarmWanted() {
  353. _, _ = s.cron.AddJob(cadenceCPUAlarm, job.NewCheckCpuJob())
  354. }
  355. // Memory monitor publishes memory.high events; register it whenever any notifier wants them.
  356. if s.memoryAlarmWanted() {
  357. _, _ = s.cron.AddJob(cadenceMemoryAlarm, job.NewCheckMemJob())
  358. }
  359. if mins := sys.MemoryReleaseIntervalMinutes(); mins > 0 {
  360. _, _ = s.cron.AddJob(fmt.Sprintf("@every %dm", mins), job.NewMemoryReleaseJob())
  361. go func() {
  362. time.Sleep(time.Minute)
  363. job.NewMemoryReleaseJob().Run()
  364. }()
  365. }
  366. }
  367. // cpuAlarmWanted reports whether any notifier is configured to receive cpu.high
  368. // alerts, so the minute-long blocking CPU sampler only runs when it's needed.
  369. func (s *Server) cpuAlarmWanted() bool {
  370. wants := func(events string, threshold int) bool {
  371. if threshold <= 0 {
  372. return false
  373. }
  374. for e := range strings.SplitSeq(events, ",") {
  375. if strings.TrimSpace(e) == string(eventbus.EventCPUHigh) {
  376. return true
  377. }
  378. }
  379. return false
  380. }
  381. if on, _ := s.settingService.GetTgbotEnabled(); on {
  382. events, _ := s.settingService.GetTgEnabledEvents()
  383. cpu, _ := s.settingService.GetTgCpu()
  384. if wants(events, cpu) {
  385. return true
  386. }
  387. }
  388. if on, _ := s.settingService.GetSmtpEnable(); on {
  389. events, _ := s.settingService.GetSmtpEnabledEvents()
  390. cpu, _ := s.settingService.GetSmtpCpu()
  391. if wants(events, cpu) {
  392. return true
  393. }
  394. }
  395. return false
  396. }
  397. // memoryAlarmWanted reports whether any notifier is configured to receive memory.high alerts.
  398. func (s *Server) memoryAlarmWanted() bool {
  399. wants := func(events string, threshold int) bool {
  400. if threshold <= 0 {
  401. return false
  402. }
  403. for e := range strings.SplitSeq(events, ",") {
  404. if strings.TrimSpace(e) == string(eventbus.EventMemoryHigh) {
  405. return true
  406. }
  407. }
  408. return false
  409. }
  410. if on, _ := s.settingService.GetTgbotEnabled(); on {
  411. events, _ := s.settingService.GetTgEnabledEvents()
  412. mem, _ := s.settingService.GetTgMemory()
  413. if wants(events, mem) {
  414. return true
  415. }
  416. }
  417. if on, _ := s.settingService.GetSmtpEnable(); on {
  418. events, _ := s.settingService.GetSmtpEnabledEvents()
  419. mem, _ := s.settingService.GetSmtpMemory()
  420. if wants(events, mem) {
  421. return true
  422. }
  423. }
  424. return false
  425. }
  426. // Start initializes and starts the web server with configured settings, routes, and background jobs.
  427. func (s *Server) Start() (err error) {
  428. return s.start(true, true)
  429. }
  430. func (s *Server) StartPanelOnly() (err error) {
  431. return s.start(false, true)
  432. }
  433. func (s *Server) start(restartXray bool, startTgBot bool) (err error) {
  434. // This is an anonymous function, no function name
  435. defer func() {
  436. if err != nil {
  437. _ = s.Stop()
  438. }
  439. }()
  440. loc, err := s.settingService.GetTimeLocation()
  441. if err != nil {
  442. return err
  443. }
  444. service.StartTrafficWriter()
  445. // SkipIfStillRunning stops a slow job (e.g. the 5s traffic poll on a large
  446. // install) from overlapping itself: two concurrent runs of the same job race
  447. // the shared xrayAPI — leaking a grpc connection — and the StatsLastValues
  448. // map, whose concurrent write is a fatal runtime throw cron.Recover can't
  449. // catch. cron.Recover then logs any panic and keeps the scheduler alive.
  450. s.cron = cron.New(
  451. cron.WithLocation(loc),
  452. cron.WithSeconds(),
  453. cron.WithChain(
  454. cron.SkipIfStillRunning(cron.DiscardLogger),
  455. cron.Recover(cron.PrintfLogger(cronPanicLogger{})),
  456. ),
  457. )
  458. s.cron.Start()
  459. // Wire the inbound-runtime manager once so InboundService can route
  460. // add/update/delete to either the local xray or a remote node panel.
  461. // The closures bridge into XrayService (which owns the running xray
  462. // process state) without forcing the runtime package to import service.
  463. runtime.SetManager(runtime.NewManager(runtime.LocalDeps{
  464. APIPort: func() int { return s.xrayService.GetXrayAPIPort() },
  465. SetNeedRestart: func() { s.xrayService.SetToNeedRestart() },
  466. }))
  467. runtime.GetManager().SetNodeEgressResolver(&s.settingService)
  468. // Supply the master client certificate for nodes in mtls mode. Issued lazily
  469. // from the node CA on first use; runtime stays free of a service import.
  470. runtime.SetMasterClientCertProvider(func() (tls.Certificate, error) {
  471. ck, err := s.settingService.EnsureMasterClientCert()
  472. if err != nil {
  473. return tls.Certificate{}, err
  474. }
  475. return tls.X509KeyPair(ck.CertPEM, ck.KeyPEM)
  476. })
  477. engine, err := s.initRouter()
  478. if err != nil {
  479. return err
  480. }
  481. certFile, err := s.settingService.GetCertFile()
  482. if err != nil {
  483. return err
  484. }
  485. keyFile, err := s.settingService.GetKeyFile()
  486. if err != nil {
  487. return err
  488. }
  489. listen, err := s.settingService.GetListen()
  490. if err != nil {
  491. return err
  492. }
  493. port, err := s.settingService.GetPort()
  494. if err != nil {
  495. return err
  496. }
  497. if envPort, configured, envErr := config.GetPortOverride(); configured {
  498. if envErr != nil {
  499. logger.Warning("Ignoring invalid XUI_PORT; using configured web port:", port, envErr)
  500. } else {
  501. port = envPort
  502. logger.Info("Using XUI_PORT override for web panel port:", port)
  503. }
  504. }
  505. listenAddr := net.JoinHostPort(listen, strconv.Itoa(port))
  506. listener, err := (&net.ListenConfig{}).Listen(context.Background(), "tcp", listenAddr)
  507. if err != nil {
  508. return err
  509. }
  510. if certFile != "" || keyFile != "" {
  511. cert, err := tls.LoadX509KeyPair(certFile, keyFile)
  512. if err == nil {
  513. c := &tls.Config{
  514. Certificates: []tls.Certificate{cert},
  515. }
  516. // Opt-in node mTLS: when a trust CA is configured, request and verify
  517. // client certs (VerifyClientCertIfGiven keeps browsers working). With
  518. // no CA the listener is unchanged.
  519. if pool, perr := s.settingService.NodeMtlsClientCAPool(); perr != nil {
  520. logger.Warning("node mTLS: failed to build client CA trust pool:", perr)
  521. } else if pool != nil {
  522. applyNodeMtls(c, pool)
  523. logger.Info("Node mTLS enabled: verifying client certificates for the node API")
  524. }
  525. listener = network.NewAutoHttpsListener(listener)
  526. listener = tls.NewListener(listener, c)
  527. logger.Info("Web server running HTTPS on", listener.Addr())
  528. } else {
  529. logger.Error("Error loading certificates:", err)
  530. logger.Info("Web server running HTTP on", listener.Addr())
  531. }
  532. } else {
  533. logger.Info("Web server running HTTP on", listener.Addr())
  534. }
  535. s.listener = listener
  536. s.httpServer = &http.Server{
  537. Handler: engine,
  538. ReadHeaderTimeout: 5 * time.Second,
  539. ReadTimeout: 30 * time.Second,
  540. WriteTimeout: 30 * time.Second,
  541. IdleTimeout: 120 * time.Second,
  542. }
  543. go network.ServeHTTP(s.httpServer, listener, "Web server")
  544. // Create event bus before startTask so jobs can use it
  545. s.bus = eventbus.New(eventbus.DefaultBufferSize)
  546. service.SetEventBus(s.bus)
  547. job.EventBus = s.bus
  548. tgbot.EventBus = s.bus
  549. // Wire xray crash callback BEFORE startTask so it's ready
  550. xray.OnCrash = func(err error) {
  551. if s.bus != nil {
  552. s.bus.Publish(eventbus.Event{
  553. Type: eventbus.EventXrayCrash,
  554. Data: err.Error(),
  555. })
  556. }
  557. }
  558. // Register email subscriber (always — it checks smtpEnable at runtime)
  559. emailService := email.NewEmailService(s.settingService)
  560. emailSub := email.NewSubscriber(s.settingService, emailService)
  561. s.bus.Subscribe("email-notifier", emailSub.HandleEvent)
  562. // Wire email service to controller for test endpoint
  563. controller.SetEmailService(emailService)
  564. // Wire Telegram test function to controller
  565. controller.SetTestTgFunc(func() error {
  566. if !s.tgbotService.IsRunning() {
  567. return fmt.Errorf("telegram bot is not running (check token and chat ID)")
  568. }
  569. if err := s.tgbotService.TestConnection(); err != nil {
  570. return fmt.Errorf("telegram API test failed: %w", err)
  571. }
  572. s.tgbotService.SendMsgToTgbotAdmins("✅ Test message from 3x-ui")
  573. return nil
  574. })
  575. controller.SetReloadTgbotFunc(func() {
  576. enabled, err := s.settingService.GetTgbotEnabled()
  577. if err != nil || !enabled {
  578. if s.tgbotService.IsRunning() {
  579. s.tgbotService.Stop()
  580. }
  581. if s.bus != nil {
  582. s.bus.Unsubscribe("tg-notifier")
  583. }
  584. return
  585. }
  586. // Start() stops any previous receiver first, so it is safe whether or not the bot is already running.
  587. tgBot := s.tgbotService.NewTgbot()
  588. if startErr := tgBot.Start(i18nFS); startErr != nil {
  589. logger.Warning("reload Telegram bot failed:", startErr)
  590. return
  591. }
  592. if s.bus != nil {
  593. s.bus.Subscribe("tg-notifier", s.tgbotService.HandleEvent)
  594. }
  595. })
  596. s.startTask(restartXray, loc)
  597. if startTgBot {
  598. isTgbotenabled, err := s.settingService.GetTgbotEnabled()
  599. if (err == nil) && isTgbotenabled {
  600. tgBot := s.tgbotService.NewTgbot()
  601. _ = tgBot.Start(i18nFS)
  602. // Subscribe Telegram notifications for event bus
  603. s.bus.Subscribe("tg-notifier", s.tgbotService.HandleEvent)
  604. }
  605. }
  606. return nil
  607. }
  608. // Stop gracefully shuts down the web server, stops Xray, cron jobs, and Telegram bot.
  609. func (s *Server) Stop() error {
  610. return s.stop(true, true)
  611. }
  612. func (s *Server) StopPanelOnly() error {
  613. return s.stop(false, true)
  614. }
  615. func (s *Server) stop(stopXray bool, stopTgBot bool) error {
  616. s.cancel()
  617. if stopXray {
  618. _ = s.xrayService.StopXray()
  619. mtproto.GetManager().StopAll()
  620. amneziawgnet.GetManager().StopAll()
  621. tuic.GetManager().StopAll()
  622. amneziawgnet.GetOutboundManager().StopAll()
  623. }
  624. if s.cron != nil {
  625. s.cron.Stop()
  626. }
  627. if s.bus != nil {
  628. s.bus.Stop()
  629. }
  630. if err := service.PersistSystemMetrics(); err != nil {
  631. logger.Warning("persist system metrics on shutdown failed:", err)
  632. }
  633. if stopXray {
  634. service.StopTrafficWriter()
  635. }
  636. if stopTgBot && s.tgbotService.IsRunning() {
  637. s.tgbotService.Stop()
  638. }
  639. // Gracefully stop WebSocket hub
  640. if s.wsHub != nil {
  641. s.wsHub.Stop()
  642. }
  643. var err1 error
  644. var err2 error
  645. if s.httpServer != nil {
  646. shutdownCtx, shutdownCancel := context.WithTimeout(context.Background(), 10*time.Second)
  647. defer shutdownCancel()
  648. err1 = s.httpServer.Shutdown(shutdownCtx)
  649. }
  650. if s.listener != nil {
  651. err2 = s.listener.Close()
  652. }
  653. return common.Combine(err1, err2)
  654. }
  655. // GetCtx returns the server's context for cancellation and deadline management.
  656. func (s *Server) GetCtx() context.Context {
  657. return s.ctx
  658. }
  659. // GetCron returns the server's cron scheduler instance.
  660. func (s *Server) GetCron() *cron.Cron {
  661. return s.cron
  662. }
  663. // GetWSHub returns the WebSocket hub instance.
  664. func (s *Server) GetWSHub() any {
  665. return s.wsHub
  666. }
  667. func (s *Server) RestartXray() error {
  668. return s.xrayService.RestartXray(true)
  669. }