mirror of
https://github.com/MHSanaei/3x-ui.git
synced 2026-07-08 13:46:08 +00:00
MHSanaei
43500a5470
Catch the panel up to the mtg-multi README (v1.14.0): - Each client can now carry its own 32-hex advertising tag overriding the inbound-level one. The tag lives on the client (settings JSON is the source of truth, clients.ad_tag is the UI projection), is rendered into the fork's [secret-ad-tags] section for active secrets only (mtg rejects a config whose override names an unknown secret), is pushed per entry through PUT /secrets, and is part of the reload fingerprint so a tag edit hot-applies without dropping connections. - The loopback management API can replace the whole secret set, so every mtg process now gets a random per-process api-token; the manager sends it as a bearer token on PUT /secrets and GET /stats and reuses it across config rewrites, because mtg reads the token only at startup. - Malformed tags are rejected at every save path and additionally dropped in InstanceFromInbound: one bad tag would otherwise fail the whole generated config and take every client of the inbound down with it. - SyncInbound never copied a re-keyed mtproto secret into the canonical clients table, so the clients page and subscription links kept serving the old secret, which mtg then rejects. It is now guarded-copied like the other credentials.
70 lines
1.9 KiB
Go
70 lines
1.9 KiB
Go
package mtproto
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"strconv"
|
|
"testing"
|
|
)
|
|
|
|
// serverPort extracts the loopback port a httptest server bound to, so
|
|
// scrapeStats can rebuild the same http://127.0.0.1:<port>/stats URL.
|
|
func serverPort(t *testing.T, srv *httptest.Server) int {
|
|
t.Helper()
|
|
u, err := url.Parse(srv.URL)
|
|
if err != nil {
|
|
t.Fatalf("parse url: %v", err)
|
|
}
|
|
port, err := strconv.Atoi(u.Port())
|
|
if err != nil {
|
|
t.Fatalf("parse port: %v", err)
|
|
}
|
|
return port
|
|
}
|
|
|
|
func TestScrapeStats(t *testing.T) {
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.URL.Path != "/stats" {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
if r.Header.Get("Authorization") != "Bearer sesame" {
|
|
w.WriteHeader(http.StatusUnauthorized)
|
|
return
|
|
}
|
|
_, _ = io.WriteString(w, `{"started_at":"2026-01-01T00:00:00Z","total_connections":2,`+
|
|
`"users":{`+
|
|
`"alice":{"connections":2,"bytes_in":100,"bytes_out":200,"last_seen":"2026-01-01T00:01:00Z"},`+
|
|
`"bob":{"connections":0,"bytes_in":5,"bytes_out":7,"last_seen":null}}}`)
|
|
}))
|
|
defer srv.Close()
|
|
|
|
users, ok := scrapeStats(serverPort(t, srv), "sesame")
|
|
if !ok {
|
|
t.Fatal("scrapeStats should succeed against a valid /stats endpoint")
|
|
}
|
|
if len(users) != 2 {
|
|
t.Fatalf("expected 2 users, got %d: %+v", len(users), users)
|
|
}
|
|
if users["alice"].BytesIn != 100 || users["alice"].BytesOut != 200 || users["alice"].Connections != 2 {
|
|
t.Fatalf("alice stats parsed wrong: %+v", users["alice"])
|
|
}
|
|
if users["bob"].Connections != 0 || users["bob"].BytesIn != 5 {
|
|
t.Fatalf("bob stats parsed wrong: %+v", users["bob"])
|
|
}
|
|
}
|
|
|
|
func TestScrapeStatsUnreachable(t *testing.T) {
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
http.NotFound(w, r)
|
|
}))
|
|
port := serverPort(t, srv)
|
|
srv.Close()
|
|
|
|
if _, ok := scrapeStats(port, ""); ok {
|
|
t.Fatal("scrapeStats must report ok=false when the endpoint is unreachable")
|
|
}
|
|
}
|