1
0

subService.go 46 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597
  1. package sub
  2. import (
  3. "encoding/base64"
  4. "fmt"
  5. "maps"
  6. "net"
  7. "net/url"
  8. "slices"
  9. "strings"
  10. "time"
  11. "github.com/gin-gonic/gin"
  12. "github.com/goccy/go-json"
  13. "github.com/mhsanaei/3x-ui/v2/database"
  14. "github.com/mhsanaei/3x-ui/v2/database/model"
  15. "github.com/mhsanaei/3x-ui/v2/logger"
  16. "github.com/mhsanaei/3x-ui/v2/util/common"
  17. "github.com/mhsanaei/3x-ui/v2/util/random"
  18. "github.com/mhsanaei/3x-ui/v2/web/service"
  19. "github.com/mhsanaei/3x-ui/v2/xray"
  20. )
  21. // SubService provides business logic for generating subscription links and managing subscription data.
  22. type SubService struct {
  23. address string
  24. showInfo bool
  25. remarkModel string
  26. datepicker string
  27. inboundService service.InboundService
  28. settingService service.SettingService
  29. // nodesByID is populated per request from the Node table so
  30. // resolveInboundAddress can return the node's address for any
  31. // inbound whose NodeID is set. Keeps the per-link host derivation
  32. // O(1) instead of O(N) DB hits.
  33. nodesByID map[int]*model.Node
  34. }
  35. // NewSubService creates a new subscription service with the given configuration.
  36. func NewSubService(showInfo bool, remarkModel string) *SubService {
  37. return &SubService{
  38. showInfo: showInfo,
  39. remarkModel: remarkModel,
  40. }
  41. }
  42. // PrepareForRequest sets per-request state (host + nodes map) on the
  43. // shared SubService. Called by every entry point — GetSubs, GetJson,
  44. // GetClash — so resolveInboundAddress sees the right host and the
  45. // freshly-loaded node map regardless of which sub flavour the client
  46. // hit.
  47. func (s *SubService) PrepareForRequest(host string) {
  48. s.address = host
  49. s.loadNodes()
  50. }
  51. // GetSubs retrieves subscription links for a given subscription ID and host.
  52. func (s *SubService) GetSubs(subId string, host string) ([]string, int64, xray.ClientTraffic, error) {
  53. s.PrepareForRequest(host)
  54. var result []string
  55. var traffic xray.ClientTraffic
  56. var lastOnline int64
  57. var hasEnabledClient bool
  58. var clientTraffics []xray.ClientTraffic
  59. inbounds, err := s.getInboundsBySubId(subId)
  60. if err != nil {
  61. return nil, 0, traffic, err
  62. }
  63. if len(inbounds) == 0 {
  64. return nil, 0, traffic, common.NewError("No inbounds found with ", subId)
  65. }
  66. s.datepicker, err = s.settingService.GetDatepicker()
  67. if err != nil {
  68. s.datepicker = "gregorian"
  69. }
  70. seenEmails := make(map[string]struct{})
  71. for _, inbound := range inbounds {
  72. clients, err := s.inboundService.GetClients(inbound)
  73. if err != nil {
  74. logger.Error("SubService - GetClients: Unable to get clients from inbound")
  75. }
  76. if clients == nil {
  77. continue
  78. }
  79. if len(inbound.Listen) > 0 && inbound.Listen[0] == '@' {
  80. listen, port, streamSettings, err := s.getFallbackMaster(inbound.Listen, inbound.StreamSettings)
  81. if err == nil {
  82. inbound.Listen = listen
  83. inbound.Port = port
  84. inbound.StreamSettings = streamSettings
  85. }
  86. }
  87. for _, client := range clients {
  88. if client.SubID == subId {
  89. if client.Enable {
  90. hasEnabledClient = true
  91. }
  92. result = append(result, s.getLink(inbound, client.Email))
  93. var ct xray.ClientTraffic
  94. ct, clientTraffics = s.appendUniqueTraffic(seenEmails, clientTraffics, inbound.ClientStats, client.Email)
  95. if ct.LastOnline > lastOnline {
  96. lastOnline = ct.LastOnline
  97. }
  98. }
  99. }
  100. }
  101. // Prepare statistics
  102. for index, clientTraffic := range clientTraffics {
  103. if index == 0 {
  104. traffic.Up = clientTraffic.Up
  105. traffic.Down = clientTraffic.Down
  106. traffic.Total = clientTraffic.Total
  107. if clientTraffic.ExpiryTime > 0 {
  108. traffic.ExpiryTime = clientTraffic.ExpiryTime
  109. }
  110. } else {
  111. traffic.Up += clientTraffic.Up
  112. traffic.Down += clientTraffic.Down
  113. if traffic.Total == 0 || clientTraffic.Total == 0 {
  114. traffic.Total = 0
  115. } else {
  116. traffic.Total += clientTraffic.Total
  117. }
  118. if clientTraffic.ExpiryTime != traffic.ExpiryTime {
  119. traffic.ExpiryTime = 0
  120. }
  121. }
  122. }
  123. traffic.Enable = hasEnabledClient
  124. return result, lastOnline, traffic, nil
  125. }
  126. func (s *SubService) getInboundsBySubId(subId string) ([]*model.Inbound, error) {
  127. db := database.GetDB()
  128. var inbounds []*model.Inbound
  129. // allow "hysteria2" so imports stored with the literal v2 protocol
  130. // string still surface here (#4081)
  131. err := db.Model(model.Inbound{}).Preload("ClientStats").Where(`id in (
  132. SELECT DISTINCT inbounds.id
  133. FROM inbounds,
  134. JSON_EACH(JSON_EXTRACT(inbounds.settings, '$.clients')) AS client
  135. WHERE
  136. protocol in ('vmess','vless','trojan','shadowsocks','hysteria','hysteria2')
  137. AND JSON_EXTRACT(client.value, '$.subId') = ? AND enable = ?
  138. )`, subId, true).Find(&inbounds).Error
  139. if err != nil {
  140. return nil, err
  141. }
  142. return inbounds, nil
  143. }
  144. // appendUniqueTraffic resolves the traffic stats for email and appends them
  145. // to acc only the first time email is seen. Shared-email mode lets one
  146. // client_traffics row underpin several inbounds, so without dedupe its
  147. // quota and usage would be counted once per inbound.
  148. func (s *SubService) appendUniqueTraffic(seen map[string]struct{}, acc []xray.ClientTraffic, stats []xray.ClientTraffic, email string) (xray.ClientTraffic, []xray.ClientTraffic) {
  149. ct := s.getClientTraffics(stats, email)
  150. if _, dup := seen[email]; !dup {
  151. seen[email] = struct{}{}
  152. acc = append(acc, ct)
  153. }
  154. return ct, acc
  155. }
  156. func (s *SubService) getClientTraffics(traffics []xray.ClientTraffic, email string) xray.ClientTraffic {
  157. for _, traffic := range traffics {
  158. if traffic.Email == email {
  159. return traffic
  160. }
  161. }
  162. return xray.ClientTraffic{}
  163. }
  164. func (s *SubService) getFallbackMaster(dest string, streamSettings string) (string, int, string, error) {
  165. db := database.GetDB()
  166. var inbound *model.Inbound
  167. err := db.Model(model.Inbound{}).
  168. Where("JSON_TYPE(settings, '$.fallbacks') = 'array'").
  169. Where("EXISTS (SELECT * FROM json_each(settings, '$.fallbacks') WHERE json_extract(value, '$.dest') = ?)", dest).
  170. Find(&inbound).Error
  171. if err != nil {
  172. return "", 0, "", err
  173. }
  174. var stream map[string]any
  175. json.Unmarshal([]byte(streamSettings), &stream)
  176. var masterStream map[string]any
  177. json.Unmarshal([]byte(inbound.StreamSettings), &masterStream)
  178. stream["security"] = masterStream["security"]
  179. stream["tlsSettings"] = masterStream["tlsSettings"]
  180. stream["externalProxy"] = masterStream["externalProxy"]
  181. modifiedStream, _ := json.MarshalIndent(stream, "", " ")
  182. return inbound.Listen, inbound.Port, string(modifiedStream), nil
  183. }
  184. func (s *SubService) getLink(inbound *model.Inbound, email string) string {
  185. switch inbound.Protocol {
  186. case "vmess":
  187. return s.genVmessLink(inbound, email)
  188. case "vless":
  189. return s.genVlessLink(inbound, email)
  190. case "trojan":
  191. return s.genTrojanLink(inbound, email)
  192. case "shadowsocks":
  193. return s.genShadowsocksLink(inbound, email)
  194. case "hysteria", "hysteria2":
  195. return s.genHysteriaLink(inbound, email)
  196. }
  197. return ""
  198. }
  199. // Protocol link generators are intentionally ordered as:
  200. // vmess -> vless -> trojan -> shadowsocks -> hysteria.
  201. func (s *SubService) genVmessLink(inbound *model.Inbound, email string) string {
  202. if inbound.Protocol != model.VMESS {
  203. return ""
  204. }
  205. address := s.resolveInboundAddress(inbound)
  206. obj := map[string]any{
  207. "v": "2",
  208. "add": address,
  209. "port": inbound.Port,
  210. "type": "none",
  211. }
  212. stream := unmarshalStreamSettings(inbound.StreamSettings)
  213. network, _ := stream["network"].(string)
  214. applyVmessNetworkParams(stream, network, obj)
  215. if finalmask, ok := stream["finalmask"].(map[string]any); ok {
  216. applyFinalMaskObj(finalmask, obj)
  217. }
  218. security, _ := stream["security"].(string)
  219. obj["tls"] = security
  220. if security == "tls" {
  221. applyVmessTLSParams(stream, obj)
  222. }
  223. clients, _ := s.inboundService.GetClients(inbound)
  224. clientIndex := findClientIndex(clients, email)
  225. obj["id"] = clients[clientIndex].ID
  226. obj["scy"] = clients[clientIndex].Security
  227. externalProxies, _ := stream["externalProxy"].([]any)
  228. if len(externalProxies) > 0 {
  229. return s.buildVmessExternalProxyLinks(externalProxies, obj, inbound, email)
  230. }
  231. obj["ps"] = s.genRemark(inbound, email, "")
  232. return buildVmessLink(obj)
  233. }
  234. func (s *SubService) genVlessLink(inbound *model.Inbound, email string) string {
  235. if inbound.Protocol != model.VLESS {
  236. return ""
  237. }
  238. address := s.resolveInboundAddress(inbound)
  239. stream := unmarshalStreamSettings(inbound.StreamSettings)
  240. clients, _ := s.inboundService.GetClients(inbound)
  241. clientIndex := findClientIndex(clients, email)
  242. uuid := clients[clientIndex].ID
  243. port := inbound.Port
  244. streamNetwork := stream["network"].(string)
  245. params := make(map[string]string)
  246. params["type"] = streamNetwork
  247. // Add encryption parameter for VLESS from inbound settings
  248. var settings map[string]any
  249. json.Unmarshal([]byte(inbound.Settings), &settings)
  250. if encryption, ok := settings["encryption"].(string); ok {
  251. params["encryption"] = encryption
  252. }
  253. applyShareNetworkParams(stream, streamNetwork, params)
  254. if finalmask, ok := stream["finalmask"].(map[string]any); ok {
  255. applyFinalMaskParams(finalmask, params)
  256. }
  257. security, _ := stream["security"].(string)
  258. switch security {
  259. case "tls":
  260. applyShareTLSParams(stream, params)
  261. if streamNetwork == "tcp" && len(clients[clientIndex].Flow) > 0 {
  262. params["flow"] = clients[clientIndex].Flow
  263. }
  264. case "reality":
  265. applyShareRealityParams(stream, params)
  266. if streamNetwork == "tcp" && len(clients[clientIndex].Flow) > 0 {
  267. params["flow"] = clients[clientIndex].Flow
  268. }
  269. default:
  270. params["security"] = "none"
  271. }
  272. externalProxies, _ := stream["externalProxy"].([]any)
  273. if len(externalProxies) > 0 {
  274. return s.buildExternalProxyURLLinks(
  275. externalProxies,
  276. params,
  277. security,
  278. func(dest string, port int) string {
  279. return fmt.Sprintf("vless://%s@%s:%d", uuid, dest, port)
  280. },
  281. func(ep map[string]any) string {
  282. return s.genRemark(inbound, email, ep["remark"].(string))
  283. },
  284. )
  285. }
  286. link := fmt.Sprintf("vless://%s@%s:%d", uuid, address, port)
  287. return buildLinkWithParams(link, params, s.genRemark(inbound, email, ""))
  288. }
  289. func (s *SubService) genTrojanLink(inbound *model.Inbound, email string) string {
  290. if inbound.Protocol != model.Trojan {
  291. return ""
  292. }
  293. address := s.resolveInboundAddress(inbound)
  294. stream := unmarshalStreamSettings(inbound.StreamSettings)
  295. clients, _ := s.inboundService.GetClients(inbound)
  296. clientIndex := findClientIndex(clients, email)
  297. password := clients[clientIndex].Password
  298. port := inbound.Port
  299. streamNetwork := stream["network"].(string)
  300. params := make(map[string]string)
  301. params["type"] = streamNetwork
  302. applyShareNetworkParams(stream, streamNetwork, params)
  303. if finalmask, ok := stream["finalmask"].(map[string]any); ok {
  304. applyFinalMaskParams(finalmask, params)
  305. }
  306. security, _ := stream["security"].(string)
  307. switch security {
  308. case "tls":
  309. applyShareTLSParams(stream, params)
  310. case "reality":
  311. applyShareRealityParams(stream, params)
  312. if streamNetwork == "tcp" && len(clients[clientIndex].Flow) > 0 {
  313. params["flow"] = clients[clientIndex].Flow
  314. }
  315. default:
  316. params["security"] = "none"
  317. }
  318. externalProxies, _ := stream["externalProxy"].([]any)
  319. if len(externalProxies) > 0 {
  320. return s.buildExternalProxyURLLinks(
  321. externalProxies,
  322. params,
  323. security,
  324. func(dest string, port int) string {
  325. return fmt.Sprintf("trojan://%s@%s:%d", password, dest, port)
  326. },
  327. func(ep map[string]any) string {
  328. return s.genRemark(inbound, email, ep["remark"].(string))
  329. },
  330. )
  331. }
  332. link := fmt.Sprintf("trojan://%s@%s:%d", password, address, port)
  333. return buildLinkWithParams(link, params, s.genRemark(inbound, email, ""))
  334. }
  335. func (s *SubService) genShadowsocksLink(inbound *model.Inbound, email string) string {
  336. if inbound.Protocol != model.Shadowsocks {
  337. return ""
  338. }
  339. address := s.resolveInboundAddress(inbound)
  340. stream := unmarshalStreamSettings(inbound.StreamSettings)
  341. clients, _ := s.inboundService.GetClients(inbound)
  342. var settings map[string]any
  343. json.Unmarshal([]byte(inbound.Settings), &settings)
  344. inboundPassword := settings["password"].(string)
  345. method := settings["method"].(string)
  346. clientIndex := findClientIndex(clients, email)
  347. streamNetwork := stream["network"].(string)
  348. params := make(map[string]string)
  349. params["type"] = streamNetwork
  350. applyShareNetworkParams(stream, streamNetwork, params)
  351. if finalmask, ok := stream["finalmask"].(map[string]any); ok {
  352. applyFinalMaskParams(finalmask, params)
  353. }
  354. security, _ := stream["security"].(string)
  355. if security == "tls" {
  356. applyShareTLSParams(stream, params)
  357. }
  358. encPart := fmt.Sprintf("%s:%s", method, clients[clientIndex].Password)
  359. if method[0] == '2' {
  360. encPart = fmt.Sprintf("%s:%s:%s", method, inboundPassword, clients[clientIndex].Password)
  361. }
  362. externalProxies, _ := stream["externalProxy"].([]any)
  363. if len(externalProxies) > 0 {
  364. proxyParams := cloneStringMap(params)
  365. proxyParams["security"] = security
  366. return s.buildExternalProxyURLLinks(
  367. externalProxies,
  368. proxyParams,
  369. security,
  370. func(dest string, port int) string {
  371. return fmt.Sprintf("ss://%s@%s:%d", base64.StdEncoding.EncodeToString([]byte(encPart)), dest, port)
  372. },
  373. func(ep map[string]any) string {
  374. return s.genRemark(inbound, email, ep["remark"].(string))
  375. },
  376. )
  377. }
  378. link := fmt.Sprintf("ss://%s@%s:%d", base64.StdEncoding.EncodeToString([]byte(encPart)), address, inbound.Port)
  379. return buildLinkWithParams(link, params, s.genRemark(inbound, email, ""))
  380. }
  381. func (s *SubService) genHysteriaLink(inbound *model.Inbound, email string) string {
  382. if !model.IsHysteria(inbound.Protocol) {
  383. return ""
  384. }
  385. var stream map[string]any
  386. json.Unmarshal([]byte(inbound.StreamSettings), &stream)
  387. clients, _ := s.inboundService.GetClients(inbound)
  388. clientIndex := -1
  389. for i, client := range clients {
  390. if client.Email == email {
  391. clientIndex = i
  392. break
  393. }
  394. }
  395. auth := clients[clientIndex].Auth
  396. params := make(map[string]string)
  397. params["security"] = "tls"
  398. tlsSetting, _ := stream["tlsSettings"].(map[string]any)
  399. alpns, _ := tlsSetting["alpn"].([]any)
  400. var alpn []string
  401. for _, a := range alpns {
  402. alpn = append(alpn, a.(string))
  403. }
  404. if len(alpn) > 0 {
  405. params["alpn"] = strings.Join(alpn, ",")
  406. }
  407. if sniValue, ok := searchKey(tlsSetting, "serverName"); ok {
  408. params["sni"], _ = sniValue.(string)
  409. }
  410. tlsSettings, _ := searchKey(tlsSetting, "settings")
  411. if tlsSetting != nil {
  412. if fpValue, ok := searchKey(tlsSettings, "fingerprint"); ok {
  413. params["fp"], _ = fpValue.(string)
  414. }
  415. if insecure, ok := searchKey(tlsSettings, "allowInsecure"); ok {
  416. if insecure.(bool) {
  417. params["insecure"] = "1"
  418. }
  419. }
  420. }
  421. // salamander obfs (Hysteria2). The panel-side link generator already
  422. // emits these; keep the subscription output in sync so a client has
  423. // the obfs password to match the server.
  424. if finalmask, ok := stream["finalmask"].(map[string]any); ok {
  425. applyFinalMaskParams(finalmask, params)
  426. if udpMasks, ok := finalmask["udp"].([]any); ok {
  427. for _, m := range udpMasks {
  428. mask, _ := m.(map[string]any)
  429. if mask == nil || mask["type"] != "salamander" {
  430. continue
  431. }
  432. settings, _ := mask["settings"].(map[string]any)
  433. if pw, ok := settings["password"].(string); ok && pw != "" {
  434. params["obfs"] = "salamander"
  435. params["obfs-password"] = pw
  436. break
  437. }
  438. }
  439. }
  440. }
  441. var settings map[string]any
  442. json.Unmarshal([]byte(inbound.Settings), &settings)
  443. version, _ := settings["version"].(float64)
  444. protocol := "hysteria2"
  445. if int(version) == 1 {
  446. protocol = "hysteria"
  447. }
  448. // Fan out one link per External Proxy entry if any. Previously this
  449. // generator ignored `externalProxy` entirely, so the link kept the
  450. // server's own IP/port even when the admin configured an alternate
  451. // endpoint (e.g. a CDN hostname + port that forwards to the node).
  452. // Matches the behaviour of genVlessLink / genTrojanLink / ….
  453. externalProxies, _ := stream["externalProxy"].([]any)
  454. if len(externalProxies) > 0 {
  455. links := make([]string, 0, len(externalProxies))
  456. for _, externalProxy := range externalProxies {
  457. ep, ok := externalProxy.(map[string]any)
  458. if !ok {
  459. continue
  460. }
  461. dest, _ := ep["dest"].(string)
  462. portF, okPort := ep["port"].(float64)
  463. if dest == "" || !okPort {
  464. continue
  465. }
  466. epRemark, _ := ep["remark"].(string)
  467. link := fmt.Sprintf("%s://%s@%s:%d", protocol, auth, dest, int(portF))
  468. u, _ := url.Parse(link)
  469. q := u.Query()
  470. for k, v := range params {
  471. q.Add(k, v)
  472. }
  473. u.RawQuery = q.Encode()
  474. u.Fragment = s.genRemark(inbound, email, epRemark)
  475. links = append(links, u.String())
  476. }
  477. return strings.Join(links, "\n")
  478. }
  479. // No external proxy configured — fall back to the request host.
  480. link := fmt.Sprintf("%s://%s@%s:%d", protocol, auth, s.address, inbound.Port)
  481. url, _ := url.Parse(link)
  482. q := url.Query()
  483. for k, v := range params {
  484. q.Add(k, v)
  485. }
  486. url.RawQuery = q.Encode()
  487. url.Fragment = s.genRemark(inbound, email, "")
  488. return url.String()
  489. }
  490. // loadNodes refreshes nodesByID from the DB. Called once per request so
  491. // the per-inbound resolveInboundAddress lookups are pure map reads.
  492. // We filter to address != ” so a half-configured node row doesn't
  493. // accidentally produce a useless host like "https://:2053".
  494. func (s *SubService) loadNodes() {
  495. db := database.GetDB()
  496. var nodes []*model.Node
  497. if err := db.Model(&model.Node{}).Where("address != ''").Find(&nodes).Error; err != nil {
  498. logger.Warning("subscription: load nodes failed:", err)
  499. s.nodesByID = nil
  500. return
  501. }
  502. m := make(map[int]*model.Node, len(nodes))
  503. for _, n := range nodes {
  504. m[n.Id] = n
  505. }
  506. s.nodesByID = m
  507. }
  508. // resolveInboundAddress picks the host an external client should
  509. // connect to. Order:
  510. // 1. If the inbound is node-managed and the node has an address, use
  511. // the node's address — central panel's hostname doesn't speak xray
  512. // for that inbound.
  513. // 2. If the inbound binds to a non-wildcard listen address, use it.
  514. // 3. Otherwise fall back to the request's host (whatever the client
  515. // subscribed against).
  516. func (s *SubService) resolveInboundAddress(inbound *model.Inbound) string {
  517. if inbound.NodeID != nil && s.nodesByID != nil {
  518. if n, ok := s.nodesByID[*inbound.NodeID]; ok && n.Address != "" {
  519. return n.Address
  520. }
  521. }
  522. if inbound.Listen == "" || inbound.Listen == "0.0.0.0" || inbound.Listen == "::" || inbound.Listen == "::0" {
  523. return s.address
  524. }
  525. return inbound.Listen
  526. }
  527. func findClientIndex(clients []model.Client, email string) int {
  528. for i, client := range clients {
  529. if client.Email == email {
  530. return i
  531. }
  532. }
  533. return -1
  534. }
  535. func unmarshalStreamSettings(streamSettings string) map[string]any {
  536. var stream map[string]any
  537. json.Unmarshal([]byte(streamSettings), &stream)
  538. return stream
  539. }
  540. func applyPathAndHostParams(settings map[string]any, params map[string]string) {
  541. params["path"] = settings["path"].(string)
  542. if host, ok := settings["host"].(string); ok && len(host) > 0 {
  543. params["host"] = host
  544. } else {
  545. headers, _ := settings["headers"].(map[string]any)
  546. params["host"] = searchHost(headers)
  547. }
  548. }
  549. func applyPathAndHostObj(settings map[string]any, obj map[string]any) {
  550. obj["path"] = settings["path"].(string)
  551. if host, ok := settings["host"].(string); ok && len(host) > 0 {
  552. obj["host"] = host
  553. } else {
  554. headers, _ := settings["headers"].(map[string]any)
  555. obj["host"] = searchHost(headers)
  556. }
  557. }
  558. func applyShareNetworkParams(stream map[string]any, streamNetwork string, params map[string]string) {
  559. switch streamNetwork {
  560. case "tcp":
  561. tcp, _ := stream["tcpSettings"].(map[string]any)
  562. header, _ := tcp["header"].(map[string]any)
  563. typeStr, _ := header["type"].(string)
  564. if typeStr == "http" {
  565. request := header["request"].(map[string]any)
  566. requestPath, _ := request["path"].([]any)
  567. params["path"] = requestPath[0].(string)
  568. headers, _ := request["headers"].(map[string]any)
  569. params["host"] = searchHost(headers)
  570. params["headerType"] = "http"
  571. }
  572. case "kcp":
  573. applyKcpShareParams(stream, params)
  574. case "ws":
  575. ws, _ := stream["wsSettings"].(map[string]any)
  576. applyPathAndHostParams(ws, params)
  577. case "grpc":
  578. grpc, _ := stream["grpcSettings"].(map[string]any)
  579. params["serviceName"] = grpc["serviceName"].(string)
  580. params["authority"], _ = grpc["authority"].(string)
  581. if grpc["multiMode"].(bool) {
  582. params["mode"] = "multi"
  583. }
  584. case "httpupgrade":
  585. httpupgrade, _ := stream["httpupgradeSettings"].(map[string]any)
  586. applyPathAndHostParams(httpupgrade, params)
  587. case "xhttp":
  588. xhttp, _ := stream["xhttpSettings"].(map[string]any)
  589. applyXhttpExtraParams(xhttp, params)
  590. }
  591. }
  592. // applyXhttpExtraObj copies the bidirectional xhttp settings into the
  593. // VMess base64 JSON link object. VMess supports arbitrary keys, so we
  594. // flatten the SplitHTTPConfig "extra" fields directly onto obj.
  595. func applyXhttpExtraObj(xhttp map[string]any, obj map[string]any) {
  596. if xpb, ok := xhttp["xPaddingBytes"].(string); ok && len(xpb) > 0 {
  597. obj["x_padding_bytes"] = xpb
  598. }
  599. maps.Copy(obj, buildXhttpExtra(xhttp))
  600. }
  601. func applyVmessNetworkParams(stream map[string]any, network string, obj map[string]any) {
  602. obj["net"] = network
  603. switch network {
  604. case "tcp":
  605. tcp, _ := stream["tcpSettings"].(map[string]any)
  606. header, _ := tcp["header"].(map[string]any)
  607. typeStr, _ := header["type"].(string)
  608. obj["type"] = typeStr
  609. if typeStr == "http" {
  610. request := header["request"].(map[string]any)
  611. requestPath, _ := request["path"].([]any)
  612. obj["path"] = requestPath[0].(string)
  613. headers, _ := request["headers"].(map[string]any)
  614. obj["host"] = searchHost(headers)
  615. }
  616. case "kcp":
  617. applyKcpShareObj(stream, obj)
  618. case "ws":
  619. ws, _ := stream["wsSettings"].(map[string]any)
  620. applyPathAndHostObj(ws, obj)
  621. case "grpc":
  622. grpc, _ := stream["grpcSettings"].(map[string]any)
  623. obj["path"] = grpc["serviceName"].(string)
  624. obj["authority"] = grpc["authority"].(string)
  625. if grpc["multiMode"].(bool) {
  626. obj["type"] = "multi"
  627. }
  628. case "httpupgrade":
  629. httpupgrade, _ := stream["httpupgradeSettings"].(map[string]any)
  630. applyPathAndHostObj(httpupgrade, obj)
  631. case "xhttp":
  632. xhttp, _ := stream["xhttpSettings"].(map[string]any)
  633. applyPathAndHostObj(xhttp, obj)
  634. if mode, ok := xhttp["mode"].(string); ok {
  635. obj["mode"] = mode
  636. }
  637. applyXhttpExtraObj(xhttp, obj)
  638. }
  639. }
  640. func applyShareTLSParams(stream map[string]any, params map[string]string) {
  641. params["security"] = "tls"
  642. tlsSetting, _ := stream["tlsSettings"].(map[string]any)
  643. alpns, _ := tlsSetting["alpn"].([]any)
  644. var alpn []string
  645. for _, a := range alpns {
  646. alpn = append(alpn, a.(string))
  647. }
  648. if len(alpn) > 0 {
  649. params["alpn"] = strings.Join(alpn, ",")
  650. }
  651. if sniValue, ok := searchKey(tlsSetting, "serverName"); ok {
  652. params["sni"], _ = sniValue.(string)
  653. }
  654. tlsSettings, _ := searchKey(tlsSetting, "settings")
  655. if tlsSetting != nil {
  656. if fpValue, ok := searchKey(tlsSettings, "fingerprint"); ok {
  657. params["fp"], _ = fpValue.(string)
  658. }
  659. }
  660. }
  661. func applyVmessTLSParams(stream map[string]any, obj map[string]any) {
  662. tlsSetting, _ := stream["tlsSettings"].(map[string]any)
  663. alpns, _ := tlsSetting["alpn"].([]any)
  664. if len(alpns) > 0 {
  665. var alpn []string
  666. for _, a := range alpns {
  667. alpn = append(alpn, a.(string))
  668. }
  669. obj["alpn"] = strings.Join(alpn, ",")
  670. }
  671. if sniValue, ok := searchKey(tlsSetting, "serverName"); ok {
  672. obj["sni"], _ = sniValue.(string)
  673. }
  674. tlsSettings, _ := searchKey(tlsSetting, "settings")
  675. if tlsSetting != nil {
  676. if fpValue, ok := searchKey(tlsSettings, "fingerprint"); ok {
  677. obj["fp"], _ = fpValue.(string)
  678. }
  679. }
  680. }
  681. func applyShareRealityParams(stream map[string]any, params map[string]string) {
  682. params["security"] = "reality"
  683. realitySetting, _ := stream["realitySettings"].(map[string]any)
  684. realitySettings, _ := searchKey(realitySetting, "settings")
  685. if realitySetting != nil {
  686. if sniValue, ok := searchKey(realitySetting, "serverNames"); ok {
  687. sNames, _ := sniValue.([]any)
  688. params["sni"] = sNames[random.Num(len(sNames))].(string)
  689. }
  690. if pbkValue, ok := searchKey(realitySettings, "publicKey"); ok {
  691. params["pbk"], _ = pbkValue.(string)
  692. }
  693. if sidValue, ok := searchKey(realitySetting, "shortIds"); ok {
  694. shortIds, _ := sidValue.([]any)
  695. params["sid"] = shortIds[random.Num(len(shortIds))].(string)
  696. }
  697. if fpValue, ok := searchKey(realitySettings, "fingerprint"); ok {
  698. if fp, ok := fpValue.(string); ok && len(fp) > 0 {
  699. params["fp"] = fp
  700. }
  701. }
  702. if pqvValue, ok := searchKey(realitySettings, "mldsa65Verify"); ok {
  703. if pqv, ok := pqvValue.(string); ok && len(pqv) > 0 {
  704. params["pqv"] = pqv
  705. }
  706. }
  707. params["spx"] = "/" + random.Seq(15)
  708. }
  709. }
  710. func buildVmessLink(obj map[string]any) string {
  711. jsonStr, _ := json.MarshalIndent(obj, "", " ")
  712. return "vmess://" + base64.StdEncoding.EncodeToString(jsonStr)
  713. }
  714. func cloneVmessShareObj(baseObj map[string]any, newSecurity string) map[string]any {
  715. newObj := map[string]any{}
  716. for key, value := range baseObj {
  717. if !(newSecurity == "none" && (key == "alpn" || key == "sni" || key == "fp")) {
  718. newObj[key] = value
  719. }
  720. }
  721. return newObj
  722. }
  723. func (s *SubService) buildVmessExternalProxyLinks(externalProxies []any, baseObj map[string]any, inbound *model.Inbound, email string) string {
  724. var links strings.Builder
  725. for index, externalProxy := range externalProxies {
  726. ep, _ := externalProxy.(map[string]any)
  727. newSecurity, _ := ep["forceTls"].(string)
  728. newObj := cloneVmessShareObj(baseObj, newSecurity)
  729. newObj["ps"] = s.genRemark(inbound, email, ep["remark"].(string))
  730. newObj["add"] = ep["dest"].(string)
  731. newObj["port"] = int(ep["port"].(float64))
  732. if newSecurity != "same" {
  733. newObj["tls"] = newSecurity
  734. }
  735. if index > 0 {
  736. links.WriteString("\n")
  737. }
  738. links.WriteString(buildVmessLink(newObj))
  739. }
  740. return links.String()
  741. }
  742. func buildLinkWithParams(link string, params map[string]string, fragment string) string {
  743. parsedURL, _ := url.Parse(link)
  744. q := parsedURL.Query()
  745. for k, v := range params {
  746. q.Add(k, v)
  747. }
  748. parsedURL.RawQuery = q.Encode()
  749. parsedURL.Fragment = fragment
  750. return parsedURL.String()
  751. }
  752. func buildLinkWithParamsAndSecurity(link string, params map[string]string, fragment, security string, omitTLSFields bool) string {
  753. parsedURL, _ := url.Parse(link)
  754. q := parsedURL.Query()
  755. for k, v := range params {
  756. if k == "security" {
  757. v = security
  758. }
  759. if omitTLSFields && (k == "alpn" || k == "sni" || k == "fp") {
  760. continue
  761. }
  762. q.Add(k, v)
  763. }
  764. parsedURL.RawQuery = q.Encode()
  765. parsedURL.Fragment = fragment
  766. return parsedURL.String()
  767. }
  768. func (s *SubService) buildExternalProxyURLLinks(
  769. externalProxies []any,
  770. params map[string]string,
  771. baseSecurity string,
  772. makeLink func(dest string, port int) string,
  773. makeRemark func(ep map[string]any) string,
  774. ) string {
  775. links := make([]string, 0, len(externalProxies))
  776. for _, externalProxy := range externalProxies {
  777. ep, _ := externalProxy.(map[string]any)
  778. newSecurity, _ := ep["forceTls"].(string)
  779. dest, _ := ep["dest"].(string)
  780. port := int(ep["port"].(float64))
  781. securityToApply := baseSecurity
  782. if newSecurity != "same" {
  783. securityToApply = newSecurity
  784. }
  785. links = append(
  786. links,
  787. buildLinkWithParamsAndSecurity(
  788. makeLink(dest, port),
  789. params,
  790. makeRemark(ep),
  791. securityToApply,
  792. newSecurity == "none",
  793. ),
  794. )
  795. }
  796. return strings.Join(links, "\n")
  797. }
  798. func cloneStringMap(source map[string]string) map[string]string {
  799. cloned := make(map[string]string, len(source))
  800. maps.Copy(cloned, source)
  801. return cloned
  802. }
  803. func (s *SubService) genRemark(inbound *model.Inbound, email string, extra string) string {
  804. separationChar := string(s.remarkModel[0])
  805. orderChars := s.remarkModel[1:]
  806. orders := map[byte]string{
  807. 'i': "",
  808. 'e': "",
  809. 'o': "",
  810. }
  811. if len(email) > 0 {
  812. orders['e'] = email
  813. }
  814. if len(inbound.Remark) > 0 {
  815. orders['i'] = inbound.Remark
  816. }
  817. if len(extra) > 0 {
  818. orders['o'] = extra
  819. }
  820. var remark []string
  821. for i := 0; i < len(orderChars); i++ {
  822. char := orderChars[i]
  823. order, exists := orders[char]
  824. if exists && order != "" {
  825. remark = append(remark, order)
  826. }
  827. }
  828. if s.showInfo {
  829. statsExist := false
  830. var stats xray.ClientTraffic
  831. for _, clientStat := range inbound.ClientStats {
  832. if clientStat.Email == email {
  833. stats = clientStat
  834. statsExist = true
  835. break
  836. }
  837. }
  838. // Get remained days
  839. if statsExist {
  840. if !stats.Enable {
  841. return fmt.Sprintf("⛔️N/A%s%s", separationChar, strings.Join(remark, separationChar))
  842. }
  843. if vol := stats.Total - (stats.Up + stats.Down); vol > 0 {
  844. remark = append(remark, fmt.Sprintf("%s%s", common.FormatTraffic(vol), "📊"))
  845. }
  846. now := time.Now().Unix()
  847. switch exp := stats.ExpiryTime / 1000; {
  848. case exp > 0:
  849. remainingSeconds := exp - now
  850. days := remainingSeconds / 86400
  851. hours := (remainingSeconds % 86400) / 3600
  852. minutes := (remainingSeconds % 3600) / 60
  853. if days > 0 {
  854. if hours > 0 {
  855. remark = append(remark, fmt.Sprintf("%dD,%dH⏳", days, hours))
  856. } else {
  857. remark = append(remark, fmt.Sprintf("%dD⏳", days))
  858. }
  859. } else if hours > 0 {
  860. remark = append(remark, fmt.Sprintf("%dH⏳", hours))
  861. } else {
  862. remark = append(remark, fmt.Sprintf("%dM⏳", minutes))
  863. }
  864. case exp < 0:
  865. days := exp / -86400
  866. hours := (exp % -86400) / 3600
  867. minutes := (exp % -3600) / 60
  868. if days > 0 {
  869. if hours > 0 {
  870. remark = append(remark, fmt.Sprintf("%dD,%dH⏳", days, hours))
  871. } else {
  872. remark = append(remark, fmt.Sprintf("%dD⏳", days))
  873. }
  874. } else if hours > 0 {
  875. remark = append(remark, fmt.Sprintf("%dH⏳", hours))
  876. } else {
  877. remark = append(remark, fmt.Sprintf("%dM⏳", minutes))
  878. }
  879. }
  880. }
  881. }
  882. return strings.Join(remark, separationChar)
  883. }
  884. func searchKey(data any, key string) (any, bool) {
  885. switch val := data.(type) {
  886. case map[string]any:
  887. for k, v := range val {
  888. if k == key {
  889. return v, true
  890. }
  891. if result, ok := searchKey(v, key); ok {
  892. return result, true
  893. }
  894. }
  895. case []any:
  896. for _, v := range val {
  897. if result, ok := searchKey(v, key); ok {
  898. return result, true
  899. }
  900. }
  901. }
  902. return nil, false
  903. }
  904. // buildXhttpExtra walks an xhttpSettings map and returns the JSON blob
  905. // that goes into the URL's `extra` param (or, for VMess, the link
  906. // object). Carries ONLY the bidirectional fields from xray-core's
  907. // SplitHTTPConfig — i.e. the ones the server enforces and the client
  908. // must match. Strictly one-sided fields are excluded:
  909. //
  910. // - server-only (noSSEHeader, scMaxBufferedPosts, scStreamUpServerSecs,
  911. // serverMaxHeaderBytes) — client wouldn't read them, so emitting
  912. // them just bloats the URL.
  913. // - client-only (headers, uplinkHTTPMethod, uplinkChunkSize,
  914. // noGRPCHeader, scMinPostsIntervalMs, xmux, downloadSettings) — the
  915. // inbound config doesn't have them; the client configures them
  916. // locally.
  917. //
  918. // Truthy-only guards keep default inbounds emitting the same compact URL
  919. // they did before this helper grew.
  920. func buildXhttpExtra(xhttp map[string]any) map[string]any {
  921. if xhttp == nil {
  922. return nil
  923. }
  924. extra := map[string]any{}
  925. if xpb, ok := xhttp["xPaddingBytes"].(string); ok && len(xpb) > 0 {
  926. extra["xPaddingBytes"] = xpb
  927. }
  928. if obfs, ok := xhttp["xPaddingObfsMode"].(bool); ok && obfs {
  929. extra["xPaddingObfsMode"] = true
  930. for _, field := range []string{"xPaddingKey", "xPaddingHeader", "xPaddingPlacement", "xPaddingMethod"} {
  931. if v, ok := xhttp[field].(string); ok && len(v) > 0 {
  932. extra[field] = v
  933. }
  934. }
  935. }
  936. stringFields := []string{
  937. "sessionPlacement", "sessionKey",
  938. "seqPlacement", "seqKey",
  939. "uplinkDataPlacement", "uplinkDataKey",
  940. "scMaxEachPostBytes",
  941. }
  942. for _, field := range stringFields {
  943. if v, ok := xhttp[field].(string); ok && len(v) > 0 {
  944. extra[field] = v
  945. }
  946. }
  947. // Headers — emitted as the {name: value} map upstream's struct
  948. // expects. The server runtime ignores this field, but the client
  949. // (consuming the share link) honors it. Drop any "host" entry —
  950. // host already wins as a top-level URL param.
  951. if rawHeaders, ok := xhttp["headers"].(map[string]any); ok && len(rawHeaders) > 0 {
  952. out := map[string]any{}
  953. for k, v := range rawHeaders {
  954. if strings.EqualFold(k, "host") {
  955. continue
  956. }
  957. out[k] = v
  958. }
  959. if len(out) > 0 {
  960. extra["headers"] = out
  961. }
  962. }
  963. if len(extra) == 0 {
  964. return nil
  965. }
  966. return extra
  967. }
  968. // applyXhttpExtraParams emits the full xhttp config into the URL query
  969. // params of a vless:// / trojan:// / ss:// link. Sets path/host/mode at
  970. // top level (xray's Build() always lets these win over `extra`) and packs
  971. // everything else into a JSON `extra` param. Also writes the flat
  972. // `x_padding_bytes` param sing-box-family clients understand.
  973. //
  974. // Without this, the admin's custom xPaddingBytes / sessionKey / etc. never
  975. // reach the client and handshakes are silently rejected with
  976. // `invalid padding (...) length: 0` — the client-visible symptom is
  977. // "xhttp doesn't connect" on OpenWRT / sing-box.
  978. //
  979. // Two encodings are written so every popular client can read at least one:
  980. //
  981. // - x_padding_bytes=<range> — flat param, understood by sing-box and its
  982. // derivatives (Podkop, OpenWRT sing-box, Karing, NekoBox, …).
  983. // - extra=<url-encoded-json> — full xhttp settings blob, which is how
  984. // xray-core clients (v2rayNG, Happ, Furious, Exclave, …) pick up the
  985. // bidirectional fields beyond path/host/mode.
  986. func applyXhttpExtraParams(xhttp map[string]any, params map[string]string) {
  987. if xhttp == nil {
  988. return
  989. }
  990. applyPathAndHostParams(xhttp, params)
  991. if mode, ok := xhttp["mode"].(string); ok {
  992. params["mode"] = mode
  993. }
  994. if xpb, ok := xhttp["xPaddingBytes"].(string); ok && len(xpb) > 0 {
  995. params["x_padding_bytes"] = xpb
  996. }
  997. extra := buildXhttpExtra(xhttp)
  998. if extra != nil {
  999. if b, err := json.Marshal(extra); err == nil {
  1000. params["extra"] = string(b)
  1001. }
  1002. }
  1003. }
  1004. var kcpMaskToHeaderType = map[string]string{
  1005. "header-dns": "dns",
  1006. "header-dtls": "dtls",
  1007. "header-srtp": "srtp",
  1008. "header-utp": "utp",
  1009. "header-wechat": "wechat-video",
  1010. "header-wireguard": "wireguard",
  1011. }
  1012. var validFinalMaskUDPTypes = map[string]struct{}{
  1013. "salamander": {},
  1014. "mkcp-aes128gcm": {},
  1015. "header-dns": {},
  1016. "header-dtls": {},
  1017. "header-srtp": {},
  1018. "header-utp": {},
  1019. "header-wechat": {},
  1020. "header-wireguard": {},
  1021. "mkcp-original": {},
  1022. "xdns": {},
  1023. "xicmp": {},
  1024. "noise": {},
  1025. "header-custom": {},
  1026. }
  1027. var validFinalMaskTCPTypes = map[string]struct{}{
  1028. "header-custom": {},
  1029. "fragment": {},
  1030. "sudoku": {},
  1031. }
  1032. // applyKcpShareParams reconstructs legacy KCP share-link fields from either
  1033. // the historical kcpSettings.header/seed shape or the current finalmask model.
  1034. // This keeps subscription output compatible while avoiding panics when older
  1035. // keys are absent from modern inbounds.
  1036. func applyKcpShareParams(stream map[string]any, params map[string]string) {
  1037. extractKcpShareFields(stream).applyToParams(params)
  1038. }
  1039. func applyKcpShareObj(stream map[string]any, obj map[string]any) {
  1040. extractKcpShareFields(stream).applyToObj(obj)
  1041. }
  1042. type kcpShareFields struct {
  1043. headerType string
  1044. seed string
  1045. mtu int
  1046. tti int
  1047. }
  1048. func (f kcpShareFields) applyToParams(params map[string]string) {
  1049. if f.headerType != "" && f.headerType != "none" {
  1050. params["headerType"] = f.headerType
  1051. }
  1052. setStringParam(params, "seed", f.seed)
  1053. setIntParam(params, "mtu", f.mtu)
  1054. setIntParam(params, "tti", f.tti)
  1055. }
  1056. func (f kcpShareFields) applyToObj(obj map[string]any) {
  1057. if f.headerType != "" && f.headerType != "none" {
  1058. obj["type"] = f.headerType
  1059. }
  1060. setStringField(obj, "path", f.seed)
  1061. setIntField(obj, "mtu", f.mtu)
  1062. setIntField(obj, "tti", f.tti)
  1063. }
  1064. func extractKcpShareFields(stream map[string]any) kcpShareFields {
  1065. fields := kcpShareFields{headerType: "none"}
  1066. if kcp, ok := stream["kcpSettings"].(map[string]any); ok {
  1067. if header, ok := kcp["header"].(map[string]any); ok {
  1068. if value, ok := header["type"].(string); ok && value != "" {
  1069. fields.headerType = value
  1070. }
  1071. }
  1072. if value, ok := kcp["seed"].(string); ok && value != "" {
  1073. fields.seed = value
  1074. }
  1075. if value, ok := readPositiveInt(kcp["mtu"]); ok {
  1076. fields.mtu = value
  1077. }
  1078. if value, ok := readPositiveInt(kcp["tti"]); ok {
  1079. fields.tti = value
  1080. }
  1081. }
  1082. for _, rawMask := range normalizedFinalMaskUDPMasks(stream["finalmask"]) {
  1083. mask, _ := rawMask.(map[string]any)
  1084. if mask == nil {
  1085. continue
  1086. }
  1087. maskType, _ := mask["type"].(string)
  1088. if mapped, ok := kcpMaskToHeaderType[maskType]; ok {
  1089. fields.headerType = mapped
  1090. continue
  1091. }
  1092. switch maskType {
  1093. case "mkcp-original":
  1094. fields.seed = ""
  1095. case "mkcp-aes128gcm":
  1096. fields.seed = ""
  1097. settings, _ := mask["settings"].(map[string]any)
  1098. if value, ok := settings["password"].(string); ok && value != "" {
  1099. fields.seed = value
  1100. }
  1101. }
  1102. }
  1103. return fields
  1104. }
  1105. func readPositiveInt(value any) (int, bool) {
  1106. switch number := value.(type) {
  1107. case int:
  1108. return number, number > 0
  1109. case int32:
  1110. return int(number), number > 0
  1111. case int64:
  1112. return int(number), number > 0
  1113. case float32:
  1114. parsed := int(number)
  1115. return parsed, parsed > 0
  1116. case float64:
  1117. parsed := int(number)
  1118. return parsed, parsed > 0
  1119. default:
  1120. return 0, false
  1121. }
  1122. }
  1123. func setStringParam(params map[string]string, key, value string) {
  1124. if value == "" {
  1125. delete(params, key)
  1126. return
  1127. }
  1128. params[key] = value
  1129. }
  1130. func setIntParam(params map[string]string, key string, value int) {
  1131. if value <= 0 {
  1132. delete(params, key)
  1133. return
  1134. }
  1135. params[key] = fmt.Sprintf("%d", value)
  1136. }
  1137. func setStringField(obj map[string]any, key, value string) {
  1138. if value == "" {
  1139. delete(obj, key)
  1140. return
  1141. }
  1142. obj[key] = value
  1143. }
  1144. func setIntField(obj map[string]any, key string, value int) {
  1145. if value <= 0 {
  1146. delete(obj, key)
  1147. return
  1148. }
  1149. obj[key] = value
  1150. }
  1151. // applyFinalMaskParams exports the finalmask payload as the compact
  1152. // `fm=<json>` share-link field used by v2rayN-compatible clients.
  1153. func applyFinalMaskParams(finalmask map[string]any, params map[string]string) {
  1154. if fm, ok := marshalFinalMask(finalmask); ok {
  1155. params["fm"] = fm
  1156. }
  1157. }
  1158. func applyFinalMaskObj(finalmask map[string]any, obj map[string]any) {
  1159. if fm, ok := marshalFinalMask(finalmask); ok {
  1160. obj["fm"] = fm
  1161. }
  1162. }
  1163. func marshalFinalMask(finalmask map[string]any) (string, bool) {
  1164. normalized := normalizeFinalMask(finalmask)
  1165. if !hasFinalMaskContent(normalized) {
  1166. return "", false
  1167. }
  1168. b, err := json.Marshal(normalized)
  1169. if err != nil || len(b) == 0 || string(b) == "null" {
  1170. return "", false
  1171. }
  1172. return string(b), true
  1173. }
  1174. func normalizeFinalMask(finalmask map[string]any) map[string]any {
  1175. tcpMasks := normalizedFinalMaskTCPMasks(finalmask)
  1176. udpMasks := normalizedFinalMaskUDPMasks(finalmask)
  1177. quicParams, hasQuicParams := finalmask["quicParams"].(map[string]any)
  1178. if len(tcpMasks) == 0 && len(udpMasks) == 0 && !hasQuicParams {
  1179. return nil
  1180. }
  1181. result := map[string]any{}
  1182. if len(tcpMasks) > 0 {
  1183. result["tcp"] = tcpMasks
  1184. }
  1185. if len(udpMasks) > 0 {
  1186. result["udp"] = udpMasks
  1187. }
  1188. if hasQuicParams && len(quicParams) > 0 {
  1189. result["quicParams"] = quicParams
  1190. }
  1191. return result
  1192. }
  1193. func normalizedFinalMaskTCPMasks(value any) []any {
  1194. finalmask, _ := value.(map[string]any)
  1195. if finalmask == nil {
  1196. return nil
  1197. }
  1198. rawMasks, _ := finalmask["tcp"].([]any)
  1199. if len(rawMasks) == 0 {
  1200. return nil
  1201. }
  1202. normalized := make([]any, 0, len(rawMasks))
  1203. for _, rawMask := range rawMasks {
  1204. mask, _ := rawMask.(map[string]any)
  1205. if mask == nil {
  1206. continue
  1207. }
  1208. maskType, _ := mask["type"].(string)
  1209. if _, ok := validFinalMaskTCPTypes[maskType]; !ok || maskType == "" {
  1210. continue
  1211. }
  1212. normalizedMask := map[string]any{"type": maskType}
  1213. if settings, ok := mask["settings"].(map[string]any); ok && len(settings) > 0 {
  1214. normalizedMask["settings"] = settings
  1215. }
  1216. normalized = append(normalized, normalizedMask)
  1217. }
  1218. if len(normalized) == 0 {
  1219. return nil
  1220. }
  1221. return normalized
  1222. }
  1223. func normalizedFinalMaskUDPMasks(value any) []any {
  1224. finalmask, _ := value.(map[string]any)
  1225. if finalmask == nil {
  1226. return nil
  1227. }
  1228. rawMasks, _ := finalmask["udp"].([]any)
  1229. if len(rawMasks) == 0 {
  1230. return nil
  1231. }
  1232. normalized := make([]any, 0, len(rawMasks))
  1233. for _, rawMask := range rawMasks {
  1234. mask, _ := rawMask.(map[string]any)
  1235. if mask == nil {
  1236. continue
  1237. }
  1238. maskType, _ := mask["type"].(string)
  1239. if _, ok := validFinalMaskUDPTypes[maskType]; !ok || maskType == "" {
  1240. continue
  1241. }
  1242. normalizedMask := map[string]any{"type": maskType}
  1243. if settings, ok := mask["settings"].(map[string]any); ok && len(settings) > 0 {
  1244. normalizedMask["settings"] = settings
  1245. }
  1246. normalized = append(normalized, normalizedMask)
  1247. }
  1248. if len(normalized) == 0 {
  1249. return nil
  1250. }
  1251. return normalized
  1252. }
  1253. func hasFinalMaskContent(value any) bool {
  1254. switch v := value.(type) {
  1255. case nil:
  1256. return false
  1257. case string:
  1258. return len(v) > 0
  1259. case map[string]any:
  1260. for _, item := range v {
  1261. if hasFinalMaskContent(item) {
  1262. return true
  1263. }
  1264. }
  1265. return false
  1266. case []any:
  1267. return slices.ContainsFunc(v, hasFinalMaskContent)
  1268. default:
  1269. return true
  1270. }
  1271. }
  1272. func searchHost(headers any) string {
  1273. data, _ := headers.(map[string]any)
  1274. for k, v := range data {
  1275. if strings.EqualFold(k, "host") {
  1276. switch v.(type) {
  1277. case []any:
  1278. hosts, _ := v.([]any)
  1279. if len(hosts) > 0 {
  1280. return hosts[0].(string)
  1281. } else {
  1282. return ""
  1283. }
  1284. case any:
  1285. return v.(string)
  1286. }
  1287. }
  1288. }
  1289. return ""
  1290. }
  1291. // PageData is a view model for subpage.html
  1292. // PageData contains data for rendering the subscription information page.
  1293. type PageData struct {
  1294. Host string
  1295. BasePath string
  1296. SId string
  1297. Enabled bool
  1298. Download string
  1299. Upload string
  1300. Total string
  1301. Used string
  1302. Remained string
  1303. Expire int64
  1304. LastOnline int64
  1305. Datepicker string
  1306. DownloadByte int64
  1307. UploadByte int64
  1308. TotalByte int64
  1309. SubUrl string
  1310. SubJsonUrl string
  1311. SubClashUrl string
  1312. Result []string
  1313. }
  1314. // ResolveRequest extracts scheme and host info from request/headers consistently.
  1315. // ResolveRequest extracts scheme, host, and header information from an HTTP request.
  1316. func (s *SubService) ResolveRequest(c *gin.Context) (scheme string, host string, hostWithPort string, hostHeader string) {
  1317. // scheme
  1318. scheme = "http"
  1319. if c.Request.TLS != nil || strings.EqualFold(c.GetHeader("X-Forwarded-Proto"), "https") {
  1320. scheme = "https"
  1321. }
  1322. // base host (no port)
  1323. if h, err := getHostFromXFH(c.GetHeader("X-Forwarded-Host")); err == nil && h != "" {
  1324. host = h
  1325. }
  1326. if host == "" {
  1327. host = c.GetHeader("X-Real-IP")
  1328. }
  1329. if host == "" {
  1330. var err error
  1331. host, _, err = net.SplitHostPort(c.Request.Host)
  1332. if err != nil {
  1333. host = c.Request.Host
  1334. }
  1335. }
  1336. // host:port for URLs
  1337. hostWithPort = c.GetHeader("X-Forwarded-Host")
  1338. if hostWithPort == "" {
  1339. hostWithPort = c.Request.Host
  1340. }
  1341. if hostWithPort == "" {
  1342. hostWithPort = host
  1343. }
  1344. // header display host
  1345. hostHeader = c.GetHeader("X-Forwarded-Host")
  1346. if hostHeader == "" {
  1347. hostHeader = c.GetHeader("X-Real-IP")
  1348. }
  1349. if hostHeader == "" {
  1350. hostHeader = host
  1351. }
  1352. return
  1353. }
  1354. // BuildURLs constructs absolute subscription and JSON subscription URLs for a given subscription ID.
  1355. // It prioritizes configured URIs, then individual settings, and finally falls back to request-derived components.
  1356. func (s *SubService) BuildURLs(scheme, hostWithPort, subPath, subJsonPath, subClashPath, subId string) (subURL, subJsonURL, subClashURL string) {
  1357. if subId == "" {
  1358. return "", "", ""
  1359. }
  1360. configuredSubURI, _ := s.settingService.GetSubURI()
  1361. configuredSubJsonURI, _ := s.settingService.GetSubJsonURI()
  1362. configuredSubClashURI, _ := s.settingService.GetSubClashURI()
  1363. var baseScheme, baseHostWithPort string
  1364. if configuredSubURI == "" || configuredSubJsonURI == "" || configuredSubClashURI == "" {
  1365. baseScheme, baseHostWithPort = s.getBaseSchemeAndHost(scheme, hostWithPort)
  1366. }
  1367. subURL = s.buildSingleURL(configuredSubURI, baseScheme, baseHostWithPort, subPath, subId)
  1368. subJsonURL = s.buildSingleURL(configuredSubJsonURI, baseScheme, baseHostWithPort, subJsonPath, subId)
  1369. subClashURL = s.buildSingleURL(configuredSubClashURI, baseScheme, baseHostWithPort, subClashPath, subId)
  1370. return subURL, subJsonURL, subClashURL
  1371. }
  1372. // getBaseSchemeAndHost determines the base scheme and host from settings or falls back to request values
  1373. func (s *SubService) getBaseSchemeAndHost(requestScheme, requestHostWithPort string) (string, string) {
  1374. subDomain, err := s.settingService.GetSubDomain()
  1375. if err != nil || subDomain == "" {
  1376. return requestScheme, requestHostWithPort
  1377. }
  1378. // Get port and TLS settings
  1379. subPort, _ := s.settingService.GetSubPort()
  1380. subKeyFile, _ := s.settingService.GetSubKeyFile()
  1381. subCertFile, _ := s.settingService.GetSubCertFile()
  1382. // Determine scheme from TLS configuration
  1383. scheme := "http"
  1384. if subKeyFile != "" && subCertFile != "" {
  1385. scheme = "https"
  1386. }
  1387. // Build host:port, always include port for clarity
  1388. hostWithPort := fmt.Sprintf("%s:%d", subDomain, subPort)
  1389. return scheme, hostWithPort
  1390. }
  1391. // buildSingleURL constructs a single URL using configured URI or base components
  1392. func (s *SubService) buildSingleURL(configuredURI, baseScheme, baseHostWithPort, basePath, subId string) string {
  1393. if configuredURI != "" {
  1394. return s.joinPathWithID(configuredURI, subId)
  1395. }
  1396. baseURL := fmt.Sprintf("%s://%s", baseScheme, baseHostWithPort)
  1397. return s.joinPathWithID(baseURL+basePath, subId)
  1398. }
  1399. // joinPathWithID safely joins a base path with a subscription ID
  1400. func (s *SubService) joinPathWithID(basePath, subId string) string {
  1401. if strings.HasSuffix(basePath, "/") {
  1402. return basePath + subId
  1403. }
  1404. return basePath + "/" + subId
  1405. }
  1406. // BuildPageData parses header and prepares the template view model.
  1407. // BuildPageData constructs page data for rendering the subscription information page.
  1408. func (s *SubService) BuildPageData(subId string, hostHeader string, traffic xray.ClientTraffic, lastOnline int64, subs []string, subURL, subJsonURL, subClashURL string, basePath string) PageData {
  1409. download := common.FormatTraffic(traffic.Down)
  1410. upload := common.FormatTraffic(traffic.Up)
  1411. total := "∞"
  1412. used := common.FormatTraffic(traffic.Up + traffic.Down)
  1413. remained := ""
  1414. if traffic.Total > 0 {
  1415. total = common.FormatTraffic(traffic.Total)
  1416. left := max(traffic.Total-(traffic.Up+traffic.Down), 0)
  1417. remained = common.FormatTraffic(left)
  1418. }
  1419. datepicker := s.datepicker
  1420. if datepicker == "" {
  1421. datepicker = "gregorian"
  1422. }
  1423. return PageData{
  1424. Host: hostHeader,
  1425. BasePath: basePath,
  1426. SId: subId,
  1427. Enabled: traffic.Enable,
  1428. Download: download,
  1429. Upload: upload,
  1430. Total: total,
  1431. Used: used,
  1432. Remained: remained,
  1433. Expire: traffic.ExpiryTime / 1000,
  1434. LastOnline: lastOnline,
  1435. Datepicker: datepicker,
  1436. DownloadByte: traffic.Down,
  1437. UploadByte: traffic.Up,
  1438. TotalByte: traffic.Total,
  1439. SubUrl: subURL,
  1440. SubJsonUrl: subJsonURL,
  1441. SubClashUrl: subClashURL,
  1442. Result: subs,
  1443. }
  1444. }
  1445. func getHostFromXFH(s string) (string, error) {
  1446. if strings.Contains(s, ":") {
  1447. realHost, _, err := net.SplitHostPort(s)
  1448. if err != nil {
  1449. return "", err
  1450. }
  1451. return realHost, nil
  1452. }
  1453. return s, nil
  1454. }