3x-ui/internal/sub/inactive_external_sub_test.go

package sub

import (
	"fmt"
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"
	"time"

	"github.com/gin-gonic/gin"

	"github.com/mhsanaei/3x-ui/v3/internal/database"
	"github.com/mhsanaei/3x-ui/v3/internal/database/model"
	"github.com/mhsanaei/3x-ui/v3/internal/xray"
)

func seedInactiveExternalOnlySub(t *testing.T, subID, email string, enabled bool, expiry int64) {
	t.Helper()
	db := database.GetDB()
	rec := &model.ClientRecord{Email: email, SubID: subID, UUID: subID + "-uuid", Enable: true, ExpiryTime: expiry}
	if err := db.Create(rec).Error; err != nil {
		t.Fatalf("seed client: %v", err)
	}
	if !enabled {
		if err := db.Model(rec).Update("enable", false).Error; err != nil {
			t.Fatalf("disable client: %v", err)
		}
	}
	if err := db.Create(&xray.ClientTraffic{Email: email, Up: 11, Down: 22, Total: 1024, ExpiryTime: expiry}).Error; err != nil {
		t.Fatalf("seed traffic: %v", err)
	}
	link := "vless://[email protected]:443?type=tcp&security=reality&pbk=abc&sid=12&fp=chrome#external"
	if err := db.Create(&model.ClientExternalLink{ClientId: rec.Id, Kind: model.ExternalLinkKindLink, Value: link, SortIndex: 1}).Error; err != nil {
		t.Fatalf("seed external link: %v", err)
	}
}

func TestInactiveExternalOnlySubRemainsKnownWithoutExposingLinks(t *testing.T) {
	gin.SetMode(gin.TestMode)
	states := []struct {
		name    string
		enabled bool
		expiry  int64
	}{
		{name: "disabled", enabled: false, expiry: time.Now().Add(time.Hour).UnixMilli()},
		{name: "expired", enabled: true, expiry: time.Now().Add(-time.Hour).UnixMilli()},
	}

	for _, state := range states {
		t.Run(state.name, func(t *testing.T) {
			initSubDB(t)
			subID := "external-" + state.name
			email := state.name + "@example.com"
			seedInactiveExternalOnlySub(t, subID, email, state.enabled, state.expiry)

			oldDistFS := distFS
			distFS = testDistFS
			t.Cleanup(func() { distFS = oldDistFS })

			router := gin.New()
			NewSUBController(
				router.Group("/"),
				WithSUBJsonEnabled(true),
				WithSUBClashEnabled(true),
				WithSUBEncryption(false),
			)

			wantHeader := fmt.Sprintf("upload=11; download=22; total=1024; expire=%d", state.expiry/1000)
			for _, path := range []string{"/sub/" + subID, "/json/" + subID + "?view=raw", "/clash/" + subID + "?view=raw"} {
				t.Run(path, func(t *testing.T) {
					if err := database.GetDB().Model(&xray.ClientTraffic{}).Where("email = ?", email).Update("last_sub_fetch", 0).Error; err != nil {
						t.Fatalf("reset last_sub_fetch: %v", err)
					}
					req := httptest.NewRequest(http.MethodGet, path, nil)
					req.Host = "sub.example.com"
					w := httptest.NewRecorder()
					router.ServeHTTP(w, req)

					if w.Code != http.StatusOK {
						t.Fatalf("status = %d, want 200; body=%s", w.Code, w.Body.String())
					}
					if w.Body.Len() != 0 {
						t.Fatalf("inactive external link leaked in body: %s", w.Body.String())
					}
					if got := w.Header().Get("Subscription-Userinfo"); got != wantHeader {
						t.Fatalf("Subscription-Userinfo = %q, want %q", got, wantHeader)
					}
					var traffic xray.ClientTraffic
					if err := database.GetDB().Where("email = ?", email).First(&traffic).Error; err != nil {
						t.Fatalf("load traffic: %v", err)
					}
					if traffic.LastSubFetch == 0 {
						t.Fatal("successful empty response did not update last_sub_fetch")
					}
				})
			}

			req := httptest.NewRequest(http.MethodGet, "/sub/"+subID, nil)
			req.Host = "sub.example.com"
			req.Header.Set("Accept", "text/html")
			w := httptest.NewRecorder()
			router.ServeHTTP(w, req)
			if w.Code != http.StatusOK {
				t.Fatalf("HTML status = %d, want 200; body=%s", w.Code, w.Body.String())
			}
			if strings.Contains(w.Body.String(), "11111111-1111-1111-1111-111111111111") {
				t.Fatalf("HTML page exposed inactive external link: %s", w.Body.String())
			}
			if !strings.Contains(w.Body.String(), `"links":[]`) {
				t.Fatalf("HTML page did not render an empty links list: %s", w.Body.String())
			}
		})
	}
}