1
0

client_inbound_apply.go 37 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290
  1. package service
  2. import (
  3. "context"
  4. "encoding/json"
  5. "fmt"
  6. "maps"
  7. "strings"
  8. "time"
  9. "github.com/mhsanaei/3x-ui/v3/internal/database"
  10. "github.com/mhsanaei/3x-ui/v3/internal/database/model"
  11. "github.com/mhsanaei/3x-ui/v3/internal/logger"
  12. "github.com/mhsanaei/3x-ui/v3/internal/util/common"
  13. "github.com/mhsanaei/3x-ui/v3/internal/util/random"
  14. "github.com/mhsanaei/3x-ui/v3/internal/web/runtime"
  15. "github.com/mhsanaei/3x-ui/v3/internal/xray"
  16. "gorm.io/gorm"
  17. )
  18. func sameClientConfigExceptUpdatedAt(a, b map[string]any) bool {
  19. aa := maps.Clone(a)
  20. bb := maps.Clone(b)
  21. delete(aa, "updated_at")
  22. delete(bb, "updated_at")
  23. an, aerr := json.Marshal(aa)
  24. bn, berr := json.Marshal(bb)
  25. return aerr == nil && berr == nil && string(an) == string(bn)
  26. }
  27. // advancePushedInbound advances the node's reconcile-skip fingerprint from the
  28. // pre-edit settings to the saved ones after every per-client push succeeded.
  29. func advancePushedInbound(rt runtime.Runtime, prevSettings string, ib *model.Inbound) {
  30. rem, ok := rt.(*runtime.Remote)
  31. if !ok {
  32. return
  33. }
  34. prev := *ib
  35. prev.Settings = prevSettings
  36. rem.AdvancePushedInbound(&prev, ib)
  37. }
  38. // delInboundClients removes several clients from a single inbound in one pass:
  39. // one settings rewrite, one runtime sweep, one Save and one SyncInbound for the
  40. // whole batch, instead of repeating the full per-client cycle. It mirrors the
  41. // semantics of DelInboundClientByEmail for each removed client. needRestart is
  42. // the OR across all removals.
  43. func (s *ClientService) delInboundClients(inboundSvc *InboundService, inboundId int, recs []*model.ClientRecord, keepTraffic bool) (bool, error) {
  44. if len(recs) == 0 {
  45. return false, nil
  46. }
  47. defer lockInbound(inboundId).Unlock()
  48. oldInbound, err := inboundSvc.GetInbound(inboundId)
  49. if err != nil {
  50. logger.Error("Load Old Data Error")
  51. return false, err
  52. }
  53. var settings map[string]any
  54. if err := json.Unmarshal([]byte(oldInbound.Settings), &settings); err != nil {
  55. return false, err
  56. }
  57. // Match by email — the client's stable identity (see Delete). Removes every
  58. // entry carrying a wanted email, independent of credential drift.
  59. wanted := make(map[string]struct{}, len(recs))
  60. for _, rec := range recs {
  61. if rec.Email != "" {
  62. wanted[rec.Email] = struct{}{}
  63. }
  64. }
  65. interfaceClients, ok := settings["clients"].([]any)
  66. if !ok {
  67. return false, common.NewError("invalid clients format in inbound settings")
  68. }
  69. type removedClient struct {
  70. email string
  71. needApiDel bool
  72. }
  73. removed := make([]removedClient, 0, len(wanted))
  74. newClients := make([]any, 0, len(interfaceClients))
  75. for _, client := range interfaceClients {
  76. c, ok := client.(map[string]any)
  77. if !ok {
  78. newClients = append(newClients, client)
  79. continue
  80. }
  81. email, _ := c["email"].(string)
  82. if _, hit := wanted[email]; hit && email != "" {
  83. enable, _ := c["enable"].(bool)
  84. removed = append(removed, removedClient{email: email, needApiDel: enable})
  85. continue
  86. }
  87. newClients = append(newClients, client)
  88. }
  89. if len(removed) == 0 {
  90. return false, nil
  91. }
  92. db := database.GetDB()
  93. newClients = compactOrphans(db, newClients)
  94. if newClients == nil {
  95. newClients = []any{}
  96. }
  97. settings["clients"] = newClients
  98. newSettings, err := json.MarshalIndent(settings, "", " ")
  99. if err != nil {
  100. return false, err
  101. }
  102. prevSettings := oldInbound.Settings
  103. oldInbound.Settings = string(newSettings)
  104. var sharedSet map[string]bool
  105. if !keepTraffic {
  106. removedEmails := make([]string, 0, len(removed))
  107. for _, r := range removed {
  108. if r.email != "" {
  109. removedEmails = append(removedEmails, r.email)
  110. }
  111. }
  112. var sharedErr error
  113. sharedSet, sharedErr = inboundSvc.emailsUsedByOtherInbounds(removedEmails, inboundId)
  114. if sharedErr != nil {
  115. return false, sharedErr
  116. }
  117. }
  118. needRestart := false
  119. // Read each client's live state before the DB write (DelClientStat would
  120. // erase the enable flag we need to decide on a runtime removal).
  121. type delTarget struct {
  122. email string
  123. emailShared bool
  124. notDepleted bool
  125. needApiDel bool
  126. }
  127. targets := make([]delTarget, 0, len(removed))
  128. for _, r := range removed {
  129. email := r.email
  130. emailShared := sharedSet[strings.ToLower(strings.TrimSpace(email))]
  131. notDepleted := false
  132. if len(email) > 0 {
  133. var enables []bool
  134. if err := db.Model(xray.ClientTraffic{}).Where("email = ?", email).Limit(1).Pluck("enable", &enables).Error; err != nil {
  135. logger.Error("Get stats error")
  136. return needRestart, err
  137. }
  138. notDepleted = len(enables) > 0 && enables[0]
  139. }
  140. targets = append(targets, delTarget{email: email, emailShared: emailShared, notDepleted: notDepleted, needApiDel: r.needApiDel})
  141. }
  142. // Persist the batch deletion atomically, serialized against the traffic poll
  143. // to avoid the cross-transaction lock-order deadlock (runSerializedTx).
  144. if txErr := runSerializedTx(func(tx *gorm.DB) error {
  145. for _, t := range targets {
  146. if t.emailShared || keepTraffic {
  147. continue
  148. }
  149. if e := inboundSvc.DelClientIPs(tx, t.email); e != nil {
  150. logger.Error("Error in delete client IPs")
  151. return e
  152. }
  153. if len(t.email) > 0 {
  154. if e := inboundSvc.DelClientStat(tx, t.email); e != nil {
  155. logger.Error("Delete stats Data Error")
  156. return e
  157. }
  158. }
  159. }
  160. if e := tx.Save(oldInbound).Error; e != nil {
  161. return e
  162. }
  163. finalClients, gcErr := inboundSvc.GetClients(oldInbound)
  164. if gcErr != nil {
  165. return gcErr
  166. }
  167. if err := s.SyncInbound(tx, inboundId, finalClients); err != nil {
  168. return err
  169. }
  170. if oldInbound.NodeID != nil {
  171. return (&NodeService{}).MarkNodeDirtyTx(tx, *oldInbound.NodeID)
  172. }
  173. return nil
  174. }); txErr != nil {
  175. return needRestart, txErr
  176. }
  177. // Resolve the node push plan once for the whole batch instead of per email.
  178. var nodeRt runtime.Runtime
  179. nodePush := false
  180. if oldInbound.NodeID != nil {
  181. rt, push, _, perr := inboundSvc.nodePushPlan(oldInbound)
  182. if perr != nil {
  183. return needRestart, perr
  184. }
  185. nodeRt, nodePush = rt, push
  186. // Large batches collapse into one reconcile push rather than M deletes.
  187. if nodePush && len(targets) > nodeBulkPushThreshold {
  188. nodePush = false
  189. }
  190. }
  191. // Apply runtime deletes after commit — outside the serialized writer so a
  192. // slow node call can't stall traffic accounting.
  193. nodePushFailed := false
  194. for _, t := range targets {
  195. if len(t.email) == 0 {
  196. continue
  197. }
  198. if oldInbound.NodeID == nil {
  199. if t.needApiDel && t.notDepleted {
  200. rt, rterr := inboundSvc.runtimeFor(oldInbound)
  201. if rterr != nil {
  202. needRestart = true
  203. } else if err1 := rt.RemoveUser(context.Background(), oldInbound, t.email); err1 != nil {
  204. if !strings.Contains(err1.Error(), fmt.Sprintf("User %s not found.", t.email)) {
  205. needRestart = true
  206. }
  207. }
  208. }
  209. } else if nodePush {
  210. if err1 := nodeRt.DeleteUser(context.Background(), oldInbound, t.email); err1 != nil {
  211. logger.Warning("Error in deleting client on", nodeRt.Name(), ":", err1)
  212. nodePushFailed = true
  213. }
  214. }
  215. }
  216. if nodePush && !nodePushFailed {
  217. advancePushedInbound(nodeRt, prevSettings, oldInbound)
  218. }
  219. return needRestart, nil
  220. }
  221. func (s *ClientService) checkEmailsExistForClients(inboundSvc *InboundService, clients []model.Client, emailSubIDs map[string]string) (string, error) {
  222. if emailSubIDs == nil {
  223. var err error
  224. emailSubIDs, err = inboundSvc.getAllEmailSubIDs()
  225. if err != nil {
  226. return "", err
  227. }
  228. }
  229. seen := make(map[string]string, len(clients))
  230. for _, client := range clients {
  231. if client.Email == "" {
  232. continue
  233. }
  234. key := strings.ToLower(client.Email)
  235. if prev, ok := seen[key]; ok {
  236. if prev != client.SubID || client.SubID == "" {
  237. return client.Email, nil
  238. }
  239. continue
  240. }
  241. seen[key] = client.SubID
  242. if existingSub, ok := emailSubIDs[key]; ok {
  243. if client.SubID == "" || existingSub == "" || existingSub != client.SubID {
  244. return client.Email, nil
  245. }
  246. }
  247. }
  248. return "", nil
  249. }
  250. func (s *ClientService) AddInboundClient(inboundSvc *InboundService, data *model.Inbound) (bool, error) {
  251. return s.addInboundClient(inboundSvc, data, nil)
  252. }
  253. // addInboundClient is AddInboundClient with an optional precomputed email→subId
  254. // map. Bulk callers pass a single snapshot so the global getAllEmailSubIDs scan
  255. // runs once for the whole batch instead of once per target inbound; a nil map
  256. // makes it compute its own (the single-add path).
  257. func (s *ClientService) addInboundClient(inboundSvc *InboundService, data *model.Inbound, emailSubIDs map[string]string) (bool, error) {
  258. defer lockInbound(data.Id).Unlock()
  259. clients, err := inboundSvc.GetClients(data)
  260. if err != nil {
  261. return false, err
  262. }
  263. var settings map[string]any
  264. err = json.Unmarshal([]byte(data.Settings), &settings)
  265. if err != nil {
  266. return false, err
  267. }
  268. interfaceClients := settings["clients"].([]any)
  269. nowTs := time.Now().Unix() * 1000
  270. for i := range interfaceClients {
  271. if cm, ok := interfaceClients[i].(map[string]any); ok {
  272. if _, ok2 := cm["created_at"]; !ok2 {
  273. cm["created_at"] = nowTs
  274. }
  275. cm["updated_at"] = nowTs
  276. existingSub, _ := cm["subId"].(string)
  277. if strings.TrimSpace(existingSub) == "" {
  278. cm["subId"] = random.NumLower(16)
  279. }
  280. interfaceClients[i] = cm
  281. }
  282. }
  283. existEmail, err := s.checkEmailsExistForClients(inboundSvc, clients, emailSubIDs)
  284. if err != nil {
  285. return false, err
  286. }
  287. if existEmail != "" {
  288. return false, common.NewError("Duplicate email:", existEmail)
  289. }
  290. oldInbound, err := inboundSvc.GetInbound(data.Id)
  291. if err != nil {
  292. return false, err
  293. }
  294. existingClients, err := inboundSvc.GetClients(oldInbound)
  295. if err != nil {
  296. return false, err
  297. }
  298. // A client already on this inbound is skipped instead of appended again:
  299. // checkEmailsExistForClients exempts a matching subId so one identity can
  300. // live on several inbounds, which let retried or raced adds duplicate the
  301. // same email inside a single settings array (#5770). clients and
  302. // interfaceClients are parsed from the same data.Settings array, so they
  303. // stay index-aligned while filtering.
  304. if len(existingClients) > 0 && len(clients) > 0 {
  305. existingEmails := make(map[string]struct{}, len(existingClients))
  306. for _, c := range existingClients {
  307. if c.Email != "" {
  308. existingEmails[strings.ToLower(c.Email)] = struct{}{}
  309. }
  310. }
  311. keptClients := make([]model.Client, 0, len(clients))
  312. keptWire := make([]any, 0, len(interfaceClients))
  313. for i, c := range clients {
  314. if c.Email != "" {
  315. if _, dup := existingEmails[strings.ToLower(c.Email)]; dup {
  316. continue
  317. }
  318. }
  319. keptClients = append(keptClients, c)
  320. if i < len(interfaceClients) {
  321. keptWire = append(keptWire, interfaceClients[i])
  322. }
  323. }
  324. if len(keptClients) == 0 {
  325. return false, nil
  326. }
  327. clients = keptClients
  328. interfaceClients = keptWire
  329. }
  330. if oldInbound.Protocol == model.WireGuard {
  331. if dErr := defaultWireguardClients(existingClients, clients, interfaceClients); dErr != nil {
  332. return false, dErr
  333. }
  334. }
  335. for _, client := range clients {
  336. if strings.TrimSpace(client.Email) == "" {
  337. return false, common.NewError("client email is required")
  338. }
  339. switch oldInbound.Protocol {
  340. case "trojan":
  341. if client.Password == "" {
  342. return false, common.NewError("empty client ID")
  343. }
  344. case "shadowsocks":
  345. if client.Email == "" {
  346. return false, common.NewError("empty client ID")
  347. }
  348. case "hysteria":
  349. if client.Auth == "" {
  350. return false, common.NewError("empty client ID")
  351. }
  352. case "wireguard":
  353. if client.PublicKey == "" {
  354. return false, common.NewError("wireguard client requires a key")
  355. }
  356. case "mtproto":
  357. if client.Secret == "" {
  358. return false, common.NewError("mtproto client requires a secret")
  359. }
  360. default:
  361. if client.ID == "" {
  362. return false, common.NewError("empty client ID")
  363. }
  364. }
  365. }
  366. var oldSettings map[string]any
  367. err = json.Unmarshal([]byte(oldInbound.Settings), &oldSettings)
  368. if err != nil {
  369. return false, err
  370. }
  371. if oldInbound.Protocol == model.Shadowsocks {
  372. applyShadowsocksClientMethod(interfaceClients, oldSettings)
  373. }
  374. oldClients, _ := oldSettings["clients"].([]any)
  375. oldClients = compactOrphans(database.GetDB(), oldClients)
  376. oldClients = append(oldClients, interfaceClients...)
  377. oldSettings["clients"] = oldClients
  378. newSettings, err := json.MarshalIndent(oldSettings, "", " ")
  379. if err != nil {
  380. return false, err
  381. }
  382. prevSettings := oldInbound.Settings
  383. oldInbound.Settings = string(newSettings)
  384. needRestart := false
  385. rt, push, _, perr := inboundSvc.nodePushPlan(oldInbound)
  386. if perr != nil {
  387. return false, perr
  388. }
  389. // Persist client stats + inbound atomically, serialized against the traffic
  390. // poll to avoid the cross-transaction lock-order deadlock (runSerializedTx).
  391. if txErr := runSerializedTx(func(tx *gorm.DB) error {
  392. for i := range clients {
  393. if len(clients[i].Email) == 0 {
  394. continue
  395. }
  396. if e := inboundSvc.AddClientStat(tx, data.Id, &clients[i]); e != nil {
  397. return e
  398. }
  399. }
  400. if e := tx.Save(oldInbound).Error; e != nil {
  401. return e
  402. }
  403. finalClients, gcErr := inboundSvc.GetClients(oldInbound)
  404. if gcErr != nil {
  405. return gcErr
  406. }
  407. if err := s.SyncInbound(tx, oldInbound.Id, finalClients); err != nil {
  408. return err
  409. }
  410. if oldInbound.NodeID != nil {
  411. return (&NodeService{}).MarkNodeDirtyTx(tx, *oldInbound.NodeID)
  412. }
  413. return nil
  414. }); txErr != nil {
  415. return false, txErr
  416. }
  417. // Apply to the running runtime after commit — outside the serialized writer
  418. // so a slow node call can't stall traffic accounting.
  419. if oldInbound.NodeID == nil {
  420. if !push {
  421. needRestart = true
  422. } else {
  423. for _, client := range clients {
  424. if len(client.Email) == 0 {
  425. needRestart = true
  426. continue
  427. }
  428. if !client.Enable {
  429. continue
  430. }
  431. cipher := ""
  432. if oldInbound.Protocol == "shadowsocks" {
  433. cipher = oldSettings["method"].(string)
  434. }
  435. err1 := rt.AddUser(context.Background(), oldInbound, map[string]any{
  436. "email": client.Email,
  437. "id": client.ID,
  438. "auth": client.Auth,
  439. "security": client.Security,
  440. "flow": client.Flow,
  441. "password": client.Password,
  442. "cipher": cipher,
  443. "publicKey": client.PublicKey,
  444. "allowedIPs": client.AllowedIPs,
  445. "preSharedKey": client.PreSharedKey,
  446. "keepAlive": keepAliveStr(client.KeepAlive),
  447. })
  448. if err1 == nil {
  449. logger.Debug("Client added on", rt.Name(), ":", client.Email)
  450. } else {
  451. logger.Debug("Error in adding client on", rt.Name(), ":", err1)
  452. needRestart = true
  453. }
  454. }
  455. }
  456. } else {
  457. // Large batches would be M sequential per-client RPCs; the inbound's saved
  458. // settings already hold the final set, so mark dirty and let one reconcile
  459. // push converge the node instead.
  460. if push && len(clients) > nodeBulkPushThreshold {
  461. push = false
  462. }
  463. for _, client := range clients {
  464. if push {
  465. if err1 := rt.AddClient(context.Background(), oldInbound, client); err1 != nil {
  466. logger.Warning("Error in adding client on", rt.Name(), ":", err1)
  467. push = false
  468. }
  469. }
  470. }
  471. if push {
  472. advancePushedInbound(rt, prevSettings, oldInbound)
  473. }
  474. }
  475. return needRestart, nil
  476. }
  477. func (s *ClientService) UpdateInboundClient(inboundSvc *InboundService, data *model.Inbound, oldEmail string) (bool, error) {
  478. defer lockInbound(data.Id).Unlock()
  479. clients, err := inboundSvc.GetClients(data)
  480. if err != nil {
  481. return false, err
  482. }
  483. var settings map[string]any
  484. err = json.Unmarshal([]byte(data.Settings), &settings)
  485. if err != nil {
  486. return false, err
  487. }
  488. interfaceClients := settings["clients"].([]any)
  489. oldInbound, err := inboundSvc.GetInbound(data.Id)
  490. if err != nil {
  491. return false, err
  492. }
  493. oldClients, err := inboundSvc.GetClients(oldInbound)
  494. if err != nil {
  495. return false, err
  496. }
  497. newClientId := ""
  498. switch oldInbound.Protocol {
  499. case "trojan":
  500. newClientId = clients[0].Password
  501. case "shadowsocks":
  502. newClientId = clients[0].Email
  503. case "hysteria":
  504. newClientId = clients[0].Auth
  505. case "wireguard":
  506. newClientId = clients[0].Email
  507. case "mtproto":
  508. newClientId = clients[0].Email
  509. default:
  510. newClientId = clients[0].ID
  511. }
  512. // Locate the client to replace by email — the client's stable identity.
  513. // Credentials (uuid/password/auth) can drift from the inbound JSON, so they
  514. // are never used for matching.
  515. clientIndex := -1
  516. for index, oldClient := range oldClients {
  517. if strings.EqualFold(oldClient.Email, oldEmail) {
  518. oldEmail = oldClient.Email
  519. clientIndex = index
  520. break
  521. }
  522. }
  523. if newClientId == "" || clientIndex == -1 {
  524. return false, common.NewError("empty client ID")
  525. }
  526. if strings.TrimSpace(clients[0].Email) == "" {
  527. return false, common.NewError("client email is required")
  528. }
  529. if clients[0].Email != oldEmail {
  530. existEmail, err := s.checkEmailsExistForClients(inboundSvc, clients, nil)
  531. if err != nil {
  532. return false, err
  533. }
  534. if existEmail != "" {
  535. return false, common.NewError("Duplicate email:", existEmail)
  536. }
  537. }
  538. // WireGuard keys are never rotated by an edit: when the incoming payload omits
  539. // them (a metadata-only change), carry the stored credentials forward so the
  540. // settings JSON and the running peer keep the client's identity.
  541. if oldInbound.Protocol == model.WireGuard && clientIndex >= 0 && clientIndex < len(oldClients) {
  542. old := oldClients[clientIndex]
  543. if clients[0].PrivateKey == "" {
  544. clients[0].PrivateKey = old.PrivateKey
  545. }
  546. if clients[0].PublicKey == "" {
  547. clients[0].PublicKey = old.PublicKey
  548. }
  549. if len(clients[0].AllowedIPs) == 0 {
  550. clients[0].AllowedIPs = old.AllowedIPs
  551. } else {
  552. normalized, nErr := normalizeWireguardAllowedIPs(clients[0].AllowedIPs)
  553. if nErr != nil {
  554. return false, nErr
  555. }
  556. if len(normalized) == 0 {
  557. clients[0].AllowedIPs = old.AllowedIPs
  558. } else {
  559. peers := make([]string, 0, len(oldClients))
  560. for i := range oldClients {
  561. if i == clientIndex {
  562. continue
  563. }
  564. peers = append(peers, oldClients[i].AllowedIPs...)
  565. }
  566. if hit := wireguardAllowedIPsCollision(normalized, peers); hit != "" {
  567. return false, common.NewError("wireguard: allowedIPs entry already used by another client:", hit)
  568. }
  569. clients[0].AllowedIPs = normalized
  570. }
  571. }
  572. if clients[0].PreSharedKey == "" {
  573. clients[0].PreSharedKey = old.PreSharedKey
  574. }
  575. if clients[0].KeepAlive == 0 {
  576. clients[0].KeepAlive = old.KeepAlive
  577. }
  578. }
  579. var oldSettings map[string]any
  580. err = json.Unmarshal([]byte(oldInbound.Settings), &oldSettings)
  581. if err != nil {
  582. return false, err
  583. }
  584. settingsClients, _ := oldSettings["clients"].([]any)
  585. var preservedCreated any
  586. var preservedSubID string
  587. var oldClientMap map[string]any
  588. if clientIndex >= 0 && clientIndex < len(settingsClients) {
  589. if oldMap, ok := settingsClients[clientIndex].(map[string]any); ok {
  590. oldClientMap = oldMap
  591. if v, ok2 := oldMap["created_at"]; ok2 {
  592. preservedCreated = v
  593. }
  594. preservedSubID, _ = oldMap["subId"].(string)
  595. }
  596. }
  597. if oldInbound.Protocol == model.Shadowsocks {
  598. applyShadowsocksClientMethod(interfaceClients, oldSettings)
  599. }
  600. if len(interfaceClients) > 0 {
  601. if newMap, ok := interfaceClients[0].(map[string]any); ok {
  602. if preservedCreated == nil {
  603. preservedCreated = time.Now().Unix() * 1000
  604. }
  605. newMap["created_at"] = preservedCreated
  606. newSub, _ := newMap["subId"].(string)
  607. if strings.TrimSpace(newSub) == "" {
  608. if strings.TrimSpace(preservedSubID) != "" {
  609. newMap["subId"] = preservedSubID
  610. } else {
  611. newMap["subId"] = random.NumLower(16)
  612. }
  613. }
  614. if v, ok2 := newMap["subId"].(string); ok2 {
  615. clients[0].SubID = v
  616. }
  617. if oldInbound.Protocol == model.WireGuard {
  618. newMap["privateKey"] = clients[0].PrivateKey
  619. newMap["publicKey"] = clients[0].PublicKey
  620. newMap["allowedIPs"] = clients[0].AllowedIPs
  621. if clients[0].PreSharedKey != "" {
  622. newMap["preSharedKey"] = clients[0].PreSharedKey
  623. }
  624. if clients[0].KeepAlive > 0 {
  625. newMap["keepAlive"] = clients[0].KeepAlive
  626. }
  627. }
  628. if oldClientMap != nil && sameClientConfigExceptUpdatedAt(oldClientMap, newMap) {
  629. if v, ok2 := oldClientMap["updated_at"]; ok2 {
  630. newMap["updated_at"] = v
  631. } else {
  632. delete(newMap, "updated_at")
  633. }
  634. } else {
  635. newMap["updated_at"] = time.Now().Unix() * 1000
  636. }
  637. interfaceClients[0] = newMap
  638. }
  639. }
  640. settingsClients[clientIndex] = interfaceClients[0]
  641. oldSettings["clients"] = settingsClients
  642. if oldInbound.Protocol == model.VLESS {
  643. hasVisionFlow := false
  644. for _, c := range settingsClients {
  645. cm, ok := c.(map[string]any)
  646. if !ok {
  647. continue
  648. }
  649. if flow, _ := cm["flow"].(string); flow == "xtls-rprx-vision" {
  650. hasVisionFlow = true
  651. break
  652. }
  653. }
  654. if !hasVisionFlow {
  655. delete(oldSettings, "testseed")
  656. }
  657. }
  658. newSettings, err := json.MarshalIndent(oldSettings, "", " ")
  659. if err != nil {
  660. return false, err
  661. }
  662. if string(newSettings) == oldInbound.Settings {
  663. return false, nil
  664. }
  665. prevSettings := oldInbound.Settings
  666. oldInbound.Settings = string(newSettings)
  667. needRestart := false
  668. // Resolve the push plan before the DB write so a node-state lookup failure
  669. // still aborts the whole update without committing anything (it used to roll
  670. // the transaction back). nodePushPlan only reads, so order doesn't matter.
  671. var rt runtime.Runtime
  672. var push bool
  673. if len(oldEmail) > 0 {
  674. var perr error
  675. rt, push, _, perr = inboundSvc.nodePushPlan(oldInbound)
  676. if perr != nil {
  677. return false, perr
  678. }
  679. }
  680. // Persist client stats + inbound atomically, serialized against the traffic
  681. // poll to avoid the cross-transaction lock-order deadlock (runSerializedTx).
  682. if txErr := runSerializedTx(func(tx *gorm.DB) error {
  683. if len(clients[0].Email) > 0 {
  684. if len(oldEmail) > 0 {
  685. emailUnchanged := strings.EqualFold(oldEmail, clients[0].Email)
  686. targetExists := int64(0)
  687. if !emailUnchanged {
  688. if e := tx.Model(xray.ClientTraffic{}).Where("email = ?", clients[0].Email).Count(&targetExists).Error; e != nil {
  689. return e
  690. }
  691. }
  692. if emailUnchanged || targetExists == 0 {
  693. if e := inboundSvc.UpdateClientStat(tx, oldEmail, &clients[0]); e != nil {
  694. return e
  695. }
  696. if e := inboundSvc.UpdateClientIPs(tx, oldEmail, clients[0].Email); e != nil {
  697. return e
  698. }
  699. } else {
  700. stillUsed, sErr := inboundSvc.emailUsedByOtherInbounds(oldEmail, data.Id)
  701. if sErr != nil {
  702. return sErr
  703. }
  704. if !stillUsed {
  705. if e := inboundSvc.DelClientStat(tx, oldEmail); e != nil {
  706. return e
  707. }
  708. if e := inboundSvc.DelClientIPs(tx, oldEmail); e != nil {
  709. return e
  710. }
  711. }
  712. if e := inboundSvc.UpdateClientStat(tx, clients[0].Email, &clients[0]); e != nil {
  713. return e
  714. }
  715. }
  716. } else {
  717. if e := inboundSvc.AddClientStat(tx, data.Id, &clients[0]); e != nil {
  718. return e
  719. }
  720. }
  721. } else {
  722. stillUsed, sErr := inboundSvc.emailUsedByOtherInbounds(oldEmail, data.Id)
  723. if sErr != nil {
  724. return sErr
  725. }
  726. if !stillUsed {
  727. if e := inboundSvc.DelClientStat(tx, oldEmail); e != nil {
  728. return e
  729. }
  730. if e := inboundSvc.DelClientIPs(tx, oldEmail); e != nil {
  731. return e
  732. }
  733. }
  734. }
  735. if e := tx.Save(oldInbound).Error; e != nil {
  736. return e
  737. }
  738. finalClients, gcErr := inboundSvc.GetClients(oldInbound)
  739. if gcErr != nil {
  740. return gcErr
  741. }
  742. if err := s.SyncInbound(tx, oldInbound.Id, finalClients); err != nil {
  743. return err
  744. }
  745. if oldInbound.NodeID != nil {
  746. return (&NodeService{}).MarkNodeDirtyTx(tx, *oldInbound.NodeID)
  747. }
  748. return nil
  749. }); txErr != nil {
  750. return false, txErr
  751. }
  752. // Apply to the running runtime after the DB is committed — outside the
  753. // serialized writer so a slow node call can't stall traffic accounting.
  754. if len(oldEmail) > 0 {
  755. if oldInbound.NodeID == nil {
  756. if !push {
  757. needRestart = true
  758. } else {
  759. if oldClients[clientIndex].Enable {
  760. err1 := rt.RemoveUser(context.Background(), oldInbound, oldEmail)
  761. if err1 == nil {
  762. logger.Debug("Old client deleted on", rt.Name(), ":", oldEmail)
  763. } else if strings.Contains(err1.Error(), fmt.Sprintf("User %s not found.", oldEmail)) {
  764. logger.Debug("User is already deleted. Nothing to do more...")
  765. } else {
  766. logger.Debug("Error in deleting client on", rt.Name(), ":", err1)
  767. needRestart = true
  768. }
  769. }
  770. if clients[0].Enable {
  771. cipher := ""
  772. if oldInbound.Protocol == "shadowsocks" {
  773. cipher = oldSettings["method"].(string)
  774. }
  775. err1 := rt.AddUser(context.Background(), oldInbound, map[string]any{
  776. "email": clients[0].Email,
  777. "id": clients[0].ID,
  778. "security": clients[0].Security,
  779. "flow": clients[0].Flow,
  780. "auth": clients[0].Auth,
  781. "password": clients[0].Password,
  782. "cipher": cipher,
  783. "publicKey": clients[0].PublicKey,
  784. "allowedIPs": clients[0].AllowedIPs,
  785. "preSharedKey": clients[0].PreSharedKey,
  786. "keepAlive": keepAliveStr(clients[0].KeepAlive),
  787. })
  788. if err1 == nil {
  789. logger.Debug("Client edited on", rt.Name(), ":", clients[0].Email)
  790. } else {
  791. logger.Debug("Error in adding client on", rt.Name(), ":", err1)
  792. needRestart = true
  793. }
  794. }
  795. }
  796. } else if push {
  797. if err1 := rt.UpdateUser(context.Background(), oldInbound, oldEmail, clients[0]); err1 != nil {
  798. logger.Warning("Error in updating client on", rt.Name(), ":", err1)
  799. } else {
  800. advancePushedInbound(rt, prevSettings, oldInbound)
  801. }
  802. }
  803. } else {
  804. logger.Debug("Client old email not found")
  805. needRestart = true
  806. }
  807. return needRestart, nil
  808. }
  809. func (s *ClientService) DelInboundClientByEmail(inboundSvc *InboundService, inboundId int, email string, keepTraffic bool, fullDelete bool) (bool, error) {
  810. defer lockInbound(inboundId).Unlock()
  811. oldInbound, err := inboundSvc.GetInbound(inboundId)
  812. if err != nil {
  813. logger.Error("Load Old Data Error")
  814. return false, err
  815. }
  816. var settings map[string]any
  817. if err := json.Unmarshal([]byte(oldInbound.Settings), &settings); err != nil {
  818. return false, err
  819. }
  820. interfaceClients, ok := settings["clients"].([]any)
  821. if !ok {
  822. return false, common.NewError("invalid clients format in inbound settings")
  823. }
  824. var newClients []any
  825. needApiDel := false
  826. found := false
  827. for _, client := range interfaceClients {
  828. c, ok := client.(map[string]any)
  829. if !ok {
  830. continue
  831. }
  832. if cEmail, ok := c["email"].(string); ok && cEmail == email {
  833. found = true
  834. needApiDel, _ = c["enable"].(bool)
  835. } else {
  836. newClients = append(newClients, client)
  837. }
  838. }
  839. if !found {
  840. return false, fmt.Errorf("%w for email: %s", ErrClientNotInInbound, email)
  841. }
  842. db := database.GetDB()
  843. newClients = compactOrphans(db, newClients)
  844. if newClients == nil {
  845. newClients = []any{}
  846. }
  847. settings["clients"] = newClients
  848. newSettings, err := json.MarshalIndent(settings, "", " ")
  849. if err != nil {
  850. return false, err
  851. }
  852. prevSettings := oldInbound.Settings
  853. oldInbound.Settings = string(newSettings)
  854. emailShared, err := inboundSvc.emailUsedByOtherInbounds(email, inboundId)
  855. if err != nil {
  856. return false, err
  857. }
  858. needRestart := false
  859. // Decide what to delete and the push plan before the serialized DB write —
  860. // these are reads, and nodePushPlan failing should abort before committing.
  861. delStat := false
  862. if len(email) > 0 && !emailShared && !keepTraffic {
  863. traffic, tErr := inboundSvc.GetClientTrafficByEmail(email)
  864. if tErr != nil {
  865. return false, tErr
  866. }
  867. delStat = traffic != nil
  868. }
  869. // The runtime user is scoped to this inbound's tag + email, so the push plan
  870. // is resolved independently of emailShared — a sibling inbound still carrying
  871. // the email must not suppress removing the user from this inbound's Xray.
  872. var rt runtime.Runtime
  873. var push bool
  874. if len(email) > 0 && (oldInbound.NodeID != nil || needApiDel) {
  875. r, p, _, perr := inboundSvc.nodePushPlan(oldInbound)
  876. if perr != nil {
  877. return false, perr
  878. }
  879. rt, push = r, p
  880. }
  881. // Persist the deletion atomically, serialized against the traffic poll to
  882. // avoid the cross-transaction lock-order deadlock (runSerializedTx).
  883. if txErr := runSerializedTx(func(tx *gorm.DB) error {
  884. if !emailShared && !keepTraffic {
  885. if e := inboundSvc.DelClientIPs(tx, email); e != nil {
  886. logger.Error("Error in delete client IPs")
  887. return e
  888. }
  889. }
  890. if delStat {
  891. if e := inboundSvc.DelClientStat(tx, email); e != nil {
  892. logger.Error("Delete stats Data Error")
  893. return e
  894. }
  895. }
  896. if e := tx.Save(oldInbound).Error; e != nil {
  897. return e
  898. }
  899. finalClients, gcErr := inboundSvc.GetClients(oldInbound)
  900. if gcErr != nil {
  901. return gcErr
  902. }
  903. if err := s.SyncInbound(tx, inboundId, finalClients); err != nil {
  904. return err
  905. }
  906. if oldInbound.NodeID != nil {
  907. return (&NodeService{}).MarkNodeDirtyTx(tx, *oldInbound.NodeID)
  908. }
  909. return nil
  910. }); txErr != nil {
  911. return false, txErr
  912. }
  913. // Apply the runtime delete after commit — outside the serialized writer so a
  914. // slow node call can't stall traffic accounting. Independent of emailShared:
  915. // Xray users are keyed by inbound tag, so the user must be removed from this
  916. // inbound's runtime even when the same email survives in another inbound.
  917. if len(email) > 0 {
  918. if oldInbound.NodeID == nil {
  919. // Local inbound: a disabled client isn't in the running Xray, so only
  920. // a live one (needApiDel) needs an API removal.
  921. if needApiDel {
  922. if !push {
  923. needRestart = true
  924. } else if err1 := rt.RemoveUser(context.Background(), oldInbound, email); err1 == nil {
  925. logger.Debug("Client deleted on", rt.Name(), ":", email)
  926. needRestart = false
  927. } else if strings.Contains(err1.Error(), fmt.Sprintf("User %s not found.", email)) {
  928. logger.Debug("User is already deleted. Nothing to do more...")
  929. } else {
  930. logger.Debug("Error in deleting client on", rt.Name(), ":", email)
  931. needRestart = true
  932. }
  933. }
  934. } else {
  935. // Node inbound: propagate the delete regardless of the enable flag —
  936. // the node's own DB still carries a disabled client and would
  937. // resurrect it on the next snapshot otherwise. A full client delete
  938. // must remove the node's client record too, not just detach it from
  939. // this inbound (#5797).
  940. if push {
  941. var err1 error
  942. if fullDelete {
  943. err1 = rt.DeleteClient(context.Background(), email)
  944. } else {
  945. err1 = rt.DeleteUser(context.Background(), oldInbound, email)
  946. }
  947. if err1 != nil {
  948. logger.Warning("Error in deleting client on", rt.Name(), ":", err1)
  949. } else {
  950. advancePushedInbound(rt, prevSettings, oldInbound)
  951. }
  952. }
  953. }
  954. }
  955. return needRestart, nil
  956. }
  957. func (s *ClientService) SetClientTelegramUserID(inboundSvc *InboundService, trafficId int, tgId int64) (bool, error) {
  958. traffic, inbound, err := inboundSvc.GetClientInboundByTrafficID(trafficId)
  959. if err != nil {
  960. return false, err
  961. }
  962. if inbound == nil {
  963. return false, common.NewError("Inbound Not Found For Traffic ID:", trafficId)
  964. }
  965. clientEmail := traffic.Email
  966. oldClients, err := inboundSvc.GetClients(inbound)
  967. if err != nil {
  968. return false, err
  969. }
  970. found := false
  971. for _, oldClient := range oldClients {
  972. if oldClient.Email == clientEmail {
  973. found = true
  974. break
  975. }
  976. }
  977. if !found {
  978. return false, common.NewError("Client Not Found For Email:", clientEmail)
  979. }
  980. var settings map[string]any
  981. err = json.Unmarshal([]byte(inbound.Settings), &settings)
  982. if err != nil {
  983. return false, err
  984. }
  985. clients := settings["clients"].([]any)
  986. var newClients []any
  987. for client_index := range clients {
  988. c := clients[client_index].(map[string]any)
  989. if c["email"] == clientEmail {
  990. c["tgId"] = tgId
  991. c["updated_at"] = time.Now().Unix() * 1000
  992. newClients = append(newClients, any(c))
  993. }
  994. }
  995. settings["clients"] = newClients
  996. modifiedSettings, err := json.MarshalIndent(settings, "", " ")
  997. if err != nil {
  998. return false, err
  999. }
  1000. inbound.Settings = string(modifiedSettings)
  1001. needRestart, err := s.UpdateInboundClient(inboundSvc, inbound, clientEmail)
  1002. return needRestart, err
  1003. }
  1004. func (s *ClientService) CheckIsEnabledByEmail(inboundSvc *InboundService, clientEmail string) (bool, error) {
  1005. _, inbound, err := inboundSvc.GetClientInboundByEmail(clientEmail)
  1006. if err != nil {
  1007. return false, err
  1008. }
  1009. if inbound == nil {
  1010. return false, common.NewError("Inbound Not Found For Email:", clientEmail)
  1011. }
  1012. clients, err := inboundSvc.GetClients(inbound)
  1013. if err != nil {
  1014. return false, err
  1015. }
  1016. isEnable := false
  1017. for _, client := range clients {
  1018. if client.Email == clientEmail {
  1019. isEnable = client.Enable
  1020. break
  1021. }
  1022. }
  1023. return isEnable, err
  1024. }
  1025. func (s *ClientService) ToggleClientEnableByEmail(inboundSvc *InboundService, clientEmail string) (bool, bool, error) {
  1026. _, inbound, err := inboundSvc.GetClientInboundByEmail(clientEmail)
  1027. if err != nil {
  1028. return false, false, err
  1029. }
  1030. if inbound == nil {
  1031. return false, false, common.NewError("Inbound Not Found For Email:", clientEmail)
  1032. }
  1033. oldClients, err := inboundSvc.GetClients(inbound)
  1034. if err != nil {
  1035. return false, false, err
  1036. }
  1037. found := false
  1038. clientOldEnabled := false
  1039. for _, oldClient := range oldClients {
  1040. if oldClient.Email == clientEmail {
  1041. found = true
  1042. clientOldEnabled = oldClient.Enable
  1043. break
  1044. }
  1045. }
  1046. if !found {
  1047. return false, false, common.NewError("Client Not Found For Email:", clientEmail)
  1048. }
  1049. var settings map[string]any
  1050. err = json.Unmarshal([]byte(inbound.Settings), &settings)
  1051. if err != nil {
  1052. return false, false, err
  1053. }
  1054. clients := settings["clients"].([]any)
  1055. var newClients []any
  1056. for client_index := range clients {
  1057. c := clients[client_index].(map[string]any)
  1058. if c["email"] == clientEmail {
  1059. c["enable"] = !clientOldEnabled
  1060. c["updated_at"] = time.Now().Unix() * 1000
  1061. newClients = append(newClients, any(c))
  1062. }
  1063. }
  1064. settings["clients"] = newClients
  1065. modifiedSettings, err := json.MarshalIndent(settings, "", " ")
  1066. if err != nil {
  1067. return false, false, err
  1068. }
  1069. inbound.Settings = string(modifiedSettings)
  1070. needRestart, err := s.UpdateInboundClient(inboundSvc, inbound, clientEmail)
  1071. if err != nil {
  1072. return false, needRestart, err
  1073. }
  1074. return !clientOldEnabled, needRestart, nil
  1075. }
  1076. func (s *ClientService) SetClientEnableByEmail(inboundSvc *InboundService, clientEmail string, enable bool) (bool, bool, error) {
  1077. current, err := s.CheckIsEnabledByEmail(inboundSvc, clientEmail)
  1078. if err != nil {
  1079. return false, false, err
  1080. }
  1081. if current == enable {
  1082. return false, false, nil
  1083. }
  1084. newEnabled, needRestart, err := s.ToggleClientEnableByEmail(inboundSvc, clientEmail)
  1085. if err != nil {
  1086. return false, needRestart, err
  1087. }
  1088. return newEnabled == enable, needRestart, nil
  1089. }
  1090. // applyClientFieldByEmail loads the inbound currently hosting clientEmail,
  1091. // confirms the client exists, applies mutate to the matching client (plus a
  1092. // refreshed updated_at), and hands a single-client update payload to
  1093. // UpdateInboundClient. The rebuilt clients array intentionally contains only
  1094. // the matched client — that is the input contract UpdateInboundClient expects
  1095. // (clients[0] is the new data; clientEmail locates the row to replace). It
  1096. // backs the single-field by-email setters below.
  1097. // applyClientFieldByEmail mutates a client field on every inbound the email is
  1098. // attached to. A multi-inbound client is one logical identity: patching only
  1099. // the first inbound's JSON would leave the siblings stale, and the next
  1100. // SyncInbound over a stale sibling would revert the edit in the normalized
  1101. // records (#5039).
  1102. func (s *ClientService) applyClientFieldByEmail(inboundSvc *InboundService, clientEmail string, mutate func(c map[string]any)) (bool, error) {
  1103. inboundIds, err := s.GetInboundIdsForEmail(database.GetDB(), clientEmail)
  1104. if err != nil {
  1105. return false, err
  1106. }
  1107. if len(inboundIds) == 0 {
  1108. // Legacy fallback for clients that only live in the inbound JSON and
  1109. // were never normalized into client_inbounds.
  1110. _, inbound, gErr := inboundSvc.GetClientInboundByEmail(clientEmail)
  1111. if gErr != nil {
  1112. return false, gErr
  1113. }
  1114. if inbound == nil {
  1115. return false, common.NewError("Inbound Not Found For Email:", clientEmail)
  1116. }
  1117. inboundIds = []int{inbound.Id}
  1118. }
  1119. needRestart := false
  1120. found := false
  1121. for _, ibId := range inboundIds {
  1122. inbound, gErr := inboundSvc.GetInbound(ibId)
  1123. if gErr != nil {
  1124. return needRestart, gErr
  1125. }
  1126. var settings map[string]any
  1127. if uErr := json.Unmarshal([]byte(inbound.Settings), &settings); uErr != nil {
  1128. return needRestart, uErr
  1129. }
  1130. clients, _ := settings["clients"].([]any)
  1131. // UpdateInboundClient expects a single-client payload, so keep only the
  1132. // matching entry in the scratch copy; it splices the result back into
  1133. // the inbound's full client list itself.
  1134. var newClients []any
  1135. for client_index := range clients {
  1136. c, ok := clients[client_index].(map[string]any)
  1137. if !ok {
  1138. continue
  1139. }
  1140. if c["email"] == clientEmail {
  1141. mutate(c)
  1142. c["updated_at"] = time.Now().Unix() * 1000
  1143. newClients = append(newClients, any(c))
  1144. }
  1145. }
  1146. if len(newClients) == 0 {
  1147. continue
  1148. }
  1149. found = true
  1150. settings["clients"] = newClients
  1151. modifiedSettings, mErr := json.MarshalIndent(settings, "", " ")
  1152. if mErr != nil {
  1153. return needRestart, mErr
  1154. }
  1155. inbound.Settings = string(modifiedSettings)
  1156. nr, uErr := s.UpdateInboundClient(inboundSvc, inbound, clientEmail)
  1157. if uErr != nil {
  1158. return needRestart, uErr
  1159. }
  1160. needRestart = needRestart || nr
  1161. }
  1162. if !found {
  1163. return needRestart, common.NewError("Client Not Found For Email:", clientEmail)
  1164. }
  1165. return needRestart, nil
  1166. }
  1167. func (s *ClientService) ResetClientIpLimitByEmail(inboundSvc *InboundService, clientEmail string, count int) (bool, error) {
  1168. return s.applyClientFieldByEmail(inboundSvc, clientEmail, func(c map[string]any) {
  1169. c["limitIp"] = count
  1170. })
  1171. }
  1172. func (s *ClientService) ResetClientExpiryTimeByEmail(inboundSvc *InboundService, clientEmail string, expiry_time int64) (bool, error) {
  1173. return s.applyClientFieldByEmail(inboundSvc, clientEmail, func(c map[string]any) {
  1174. c["expiryTime"] = expiry_time
  1175. })
  1176. }
  1177. func (s *ClientService) ResetClientTrafficLimitByEmail(inboundSvc *InboundService, clientEmail string, totalGB int) (bool, error) {
  1178. if totalGB < 0 {
  1179. return false, common.NewError("totalGB must be >= 0")
  1180. }
  1181. return s.applyClientFieldByEmail(inboundSvc, clientEmail, func(c map[string]any) {
  1182. c["totalGB"] = totalGB * 1024 * 1024 * 1024
  1183. })
  1184. }