|
|
@@ -1,11 +1,14 @@
|
|
|
package sub
|
|
|
|
|
|
import (
|
|
|
+ "encoding/base64"
|
|
|
+ "net/url"
|
|
|
"strings"
|
|
|
"testing"
|
|
|
|
|
|
"github.com/mhsanaei/3x-ui/v3/internal/database"
|
|
|
"github.com/mhsanaei/3x-ui/v3/internal/database/model"
|
|
|
+ wgutil "github.com/mhsanaei/3x-ui/v3/internal/util/wireguard"
|
|
|
)
|
|
|
|
|
|
// inboundLinks (the "Export all inbound links" path) must render the remark
|
|
|
@@ -23,6 +26,16 @@ func TestInboundLinks_RemarkTemplateClientTokens(t *testing.T) {
|
|
|
if err := db.Create(ib).Error; err != nil {
|
|
|
t.Fatalf("seed inbound: %v", err)
|
|
|
}
|
|
|
+ client := &model.ClientRecord{
|
|
|
+ Email: "john@e", SubID: "subABC", UUID: "11111111-2222-4333-8444-000000000001",
|
|
|
+ Enable: true, Comment: "vip", TgID: 777,
|
|
|
+ }
|
|
|
+ if err := db.Create(client).Error; err != nil {
|
|
|
+ t.Fatalf("seed client: %v", err)
|
|
|
+ }
|
|
|
+ if err := db.Create(&model.ClientInbound{ClientId: client.Id, InboundId: ib.Id}).Error; err != nil {
|
|
|
+ t.Fatalf("seed client_inbound: %v", err)
|
|
|
+ }
|
|
|
|
|
|
svc := NewSubService("{{INBOUND}}-{{EMAIL}}-{{COMMENT}}-{{SUB_ID}}-{{TELEGRAM_ID}}-{{SHORT_ID}}|📊{{TRAFFIC_LEFT}}|⏳{{DAYS_LEFT}}D")
|
|
|
svc.PrepareForRequest("req.example.com")
|
|
|
@@ -41,3 +54,170 @@ func TestInboundLinks_RemarkTemplateClientTokens(t *testing.T) {
|
|
|
t.Fatalf("display mode must drop the traffic/expiry segments: %s", frag)
|
|
|
}
|
|
|
}
|
|
|
+
|
|
|
+// inboundLinks must use the clients-table UUID when the inbound settings JSON
|
|
|
+// still embeds a stale id (#6436).
|
|
|
+func TestInboundLinks_UsesClientsTableUUIDWhenSettingsStale(t *testing.T) {
|
|
|
+ seedSubDB(t)
|
|
|
+ db := database.GetDB()
|
|
|
+ stale := "11111111-1111-1111-1111-111111111111"
|
|
|
+ fresh := "22222222-2222-2222-2222-222222222222"
|
|
|
+ settings := `{"clients":[{"id":"` + stale + `","email":"stale@e","subId":"subStale","enable":true}],"decryption":"none"}`
|
|
|
+ ib := &model.Inbound{
|
|
|
+ UserId: 1, Tag: "stale-uuid", Enable: true, Listen: "203.0.113.5", Port: 4432,
|
|
|
+ Protocol: model.VLESS, Remark: "Stale", Settings: settings,
|
|
|
+ StreamSettings: `{"network":"tcp","security":"none","tcpSettings":{"header":{"type":"none"}}}`,
|
|
|
+ }
|
|
|
+ if err := db.Create(ib).Error; err != nil {
|
|
|
+ t.Fatalf("seed inbound: %v", err)
|
|
|
+ }
|
|
|
+ client := &model.ClientRecord{Email: "stale@e", SubID: "subStale", UUID: fresh, Enable: true}
|
|
|
+ if err := db.Create(client).Error; err != nil {
|
|
|
+ t.Fatalf("seed client: %v", err)
|
|
|
+ }
|
|
|
+ if err := db.Create(&model.ClientInbound{ClientId: client.Id, InboundId: ib.Id}).Error; err != nil {
|
|
|
+ t.Fatalf("seed client_inbound: %v", err)
|
|
|
+ }
|
|
|
+
|
|
|
+ svc := NewSubService("{{EMAIL}}")
|
|
|
+ svc.PrepareForRequest("req.example.com")
|
|
|
+ links := svc.inboundLinks(ib)
|
|
|
+ if len(links) != 1 {
|
|
|
+ t.Fatalf("links = %d, want 1: %v", len(links), links)
|
|
|
+ }
|
|
|
+ if !strings.Contains(links[0], fresh) {
|
|
|
+ t.Fatalf("link missing fresh UUID %q: %s", fresh, links[0])
|
|
|
+ }
|
|
|
+ if strings.Contains(links[0], stale) {
|
|
|
+ t.Fatalf("link still carries stale settings UUID %q: %s", stale, links[0])
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+// inboundLinks must still emit a VLESS link when clients exist only in the
|
|
|
+// inbound settings JSON (no clients / client_inbounds rows) — ListClientsForInbound
|
|
|
+// returns empty and clientsForLinkExport falls back to GetClients (#6458).
|
|
|
+func TestInboundLinks_SettingsOnlyVLESSProducesLink(t *testing.T) {
|
|
|
+ seedSubDB(t)
|
|
|
+ db := database.GetDB()
|
|
|
+ uuid := "33333333-3333-3333-3333-333333333333"
|
|
|
+ settings := `{"clients":[{"id":"` + uuid + `","email":"settings@e","subId":"subSettings","enable":true}],"decryption":"none"}`
|
|
|
+ ib := &model.Inbound{
|
|
|
+ UserId: 1, Tag: "settings-only", Enable: true, Listen: "203.0.113.5", Port: 4434,
|
|
|
+ Protocol: model.VLESS, Remark: "SettingsOnly", Settings: settings,
|
|
|
+ StreamSettings: `{"network":"tcp","security":"none","tcpSettings":{"header":{"type":"none"}}}`,
|
|
|
+ }
|
|
|
+ if err := db.Create(ib).Error; err != nil {
|
|
|
+ t.Fatalf("seed inbound: %v", err)
|
|
|
+ }
|
|
|
+
|
|
|
+ svc := NewSubService("{{EMAIL}}")
|
|
|
+ svc.PrepareForRequest("req.example.com")
|
|
|
+ links := svc.inboundLinks(ib)
|
|
|
+ if len(links) != 1 {
|
|
|
+ t.Fatalf("links = %d, want 1: %v", len(links), links)
|
|
|
+ }
|
|
|
+ if !strings.Contains(links[0], "vless://") {
|
|
|
+ t.Fatalf("link = %q, want vless:// prefix", links[0])
|
|
|
+ }
|
|
|
+ if !strings.Contains(links[0], uuid) {
|
|
|
+ t.Fatalf("link missing settings UUID %q: %s", uuid, links[0])
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+// inboundLinks must keep each WireGuard/AmneziaWG inbound's own tunnel address
|
|
|
+// and private key when the same email is attached to both — those fields live
|
|
|
+// only in the per-inbound settings JSON, not the shared clients.wg_* columns.
|
|
|
+func TestInboundLinks_PreservesPerInboundWireGuardIdentity(t *testing.T) {
|
|
|
+ seedSubDB(t)
|
|
|
+ db := database.GetDB()
|
|
|
+
|
|
|
+ serverPriv, serverPub, err := wgutil.GenerateWireguardKeypair()
|
|
|
+ if err != nil {
|
|
|
+ t.Fatalf("server keypair: %v", err)
|
|
|
+ }
|
|
|
+ wgPriv, _, err := wgutil.GenerateWireguardKeypair()
|
|
|
+ if err != nil {
|
|
|
+ t.Fatalf("wg client keypair: %v", err)
|
|
|
+ }
|
|
|
+ awgPriv, _, err := wgutil.GenerateWireguardKeypair()
|
|
|
+ if err != nil {
|
|
|
+ t.Fatalf("awg client keypair: %v", err)
|
|
|
+ }
|
|
|
+ // Shared clients row deliberately holds the *other* tunnel's key/address
|
|
|
+ // (last sync wins) — the failure mode ListClientsForInbound alone would export.
|
|
|
+ mergedPriv, _, err := wgutil.GenerateWireguardKeypair()
|
|
|
+ if err != nil {
|
|
|
+ t.Fatalf("merged keypair: %v", err)
|
|
|
+ }
|
|
|
+
|
|
|
+ email := "dual@e"
|
|
|
+ wgSettings := `{"secretKey":"` + serverPriv + `","clients":[{"email":"` + email + `","privateKey":"` + wgPriv + `","allowedIPs":["10.0.0.5/32"],"enable":true}]}`
|
|
|
+ awgSettings := `{"server":{"privateKey":"` + serverPriv + `","publicKey":"` + serverPub + `","mtu":1420},` +
|
|
|
+ `"clients":[{"email":"` + email + `","privateKey":"` + awgPriv + `","allowedIPs":["10.8.1.5/32"],"enable":true}]}`
|
|
|
+
|
|
|
+ wgIb := &model.Inbound{
|
|
|
+ UserId: 1, Tag: "wg-dual", Enable: true, Listen: "203.0.113.7", Port: 51820,
|
|
|
+ Protocol: model.WireGuard, Remark: "WG", Settings: wgSettings,
|
|
|
+ }
|
|
|
+ awgIb := &model.Inbound{
|
|
|
+ UserId: 1, Tag: "awg-dual", Enable: true, Listen: "203.0.113.8", Port: 443,
|
|
|
+ Protocol: model.AmneziaWG, Remark: "AWG", Settings: awgSettings,
|
|
|
+ }
|
|
|
+ for _, ib := range []*model.Inbound{wgIb, awgIb} {
|
|
|
+ if err := db.Create(ib).Error; err != nil {
|
|
|
+ t.Fatalf("create inbound %s: %v", ib.Tag, err)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ rec := &model.ClientRecord{
|
|
|
+ Email: email, SubID: "subDual", Enable: true,
|
|
|
+ PrivateKey: mergedPriv, AllowedIPs: "10.9.9.9/32",
|
|
|
+ }
|
|
|
+ if err := db.Create(rec).Error; err != nil {
|
|
|
+ t.Fatalf("create client: %v", err)
|
|
|
+ }
|
|
|
+ for _, ib := range []*model.Inbound{wgIb, awgIb} {
|
|
|
+ if err := db.Create(&model.ClientInbound{ClientId: rec.Id, InboundId: ib.Id}).Error; err != nil {
|
|
|
+ t.Fatalf("create client_inbound %s: %v", ib.Tag, err)
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ svc := NewSubService("{{EMAIL}}")
|
|
|
+ svc.PrepareForRequest("req.example.com")
|
|
|
+
|
|
|
+ wgLinks := svc.inboundLinks(wgIb)
|
|
|
+ if len(wgLinks) != 1 {
|
|
|
+ t.Fatalf("wg links = %d, want 1: %v", len(wgLinks), wgLinks)
|
|
|
+ }
|
|
|
+ wu, err := url.Parse(wgLinks[0])
|
|
|
+ if err != nil {
|
|
|
+ t.Fatalf("wg link parse: %v (%s)", err, wgLinks[0])
|
|
|
+ }
|
|
|
+ if wu.User.Username() != wgPriv {
|
|
|
+ t.Fatalf("wg private key = %q, want inbound settings key %q (not merged %q)", wu.User.Username(), wgPriv, mergedPriv)
|
|
|
+ }
|
|
|
+ if got := wu.Query().Get("address"); got != "10.0.0.5/32" {
|
|
|
+ t.Fatalf("wg address = %q, want 10.0.0.5/32 (not merged 10.9.9.9/32)", got)
|
|
|
+ }
|
|
|
+
|
|
|
+ awgLinks := svc.inboundLinks(awgIb)
|
|
|
+ if len(awgLinks) != 1 {
|
|
|
+ t.Fatalf("awg links = %d, want 1: %v", len(awgLinks), awgLinks)
|
|
|
+ }
|
|
|
+ if !strings.HasPrefix(awgLinks[0], "vpn://") {
|
|
|
+ t.Fatalf("awg link = %q, want vpn:// prefix", awgLinks[0])
|
|
|
+ }
|
|
|
+ raw, err := base64.RawURLEncoding.DecodeString(strings.TrimPrefix(awgLinks[0], "vpn://"))
|
|
|
+ if err != nil {
|
|
|
+ t.Fatalf("awg link decode: %v (%s)", err, awgLinks[0])
|
|
|
+ }
|
|
|
+ textCfg := string(raw)
|
|
|
+ if !strings.Contains(textCfg, "PrivateKey = "+awgPriv) {
|
|
|
+ t.Fatalf("awg config missing inbound private key %q:\n%s", awgPriv, textCfg)
|
|
|
+ }
|
|
|
+ if !strings.Contains(textCfg, "Address = 10.8.1.5/32") {
|
|
|
+ t.Fatalf("awg config missing inbound address 10.8.1.5/32:\n%s", textCfg)
|
|
|
+ }
|
|
|
+ if strings.Contains(textCfg, mergedPriv) || strings.Contains(textCfg, "10.9.9.9/32") {
|
|
|
+ t.Fatalf("awg config leaked merged clients-table tunnel identity:\n%s", textCfg)
|
|
|
+ }
|
|
|
+}
|