mirror of
https://github.com/MHSanaei/3x-ui.git
synced 2026-10-25 07:15:01 +03:00
35e09191de
AmneziaWG clients accept a randomised PersistentKeepalive range such as 15-25, which makes keepalive timing harder to fingerprint; the panel only stored a single number. A client's keepalive may now be a range, validated the way AmneziaWG parses it. The downloadable .conf and vpn:// profile carry the range, Clash/Mihomo output gets its lower bound, and plain WireGuard stays scalar-only. The range lives in an additive wg_keep_alive_range column, so the integer column and an older panel binary keep working after a rollback.
71 lines
2.4 KiB
Go
71 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestWalkPackagesExportsAllowedStructsWithoutDIInterfaces(t *testing.T) {
|
|
dir := t.TempDir()
|
|
source := `package service
|
|
|
|
type HappLinkGenerator interface {
|
|
Generate()
|
|
}
|
|
|
|
type HappLinkResult struct {
|
|
EncryptedLink string
|
|
}
|
|
`
|
|
if err := os.WriteFile(filepath.Join(dir, "service.go"), []byte(source), 0o644); err != nil {
|
|
t.Fatalf("write service fixture: %v", err)
|
|
}
|
|
|
|
schemas, aliases, err := walkPackages([]packageRequest{{
|
|
Path: dir,
|
|
StructAllow: setOf("HappLinkResult"),
|
|
}})
|
|
if err != nil {
|
|
t.Fatalf("walk packages: %v", err)
|
|
}
|
|
var generated bytes.Buffer
|
|
if err := emitTypes(&generated, schemas, aliases); err != nil {
|
|
t.Fatalf("emit types: %v", err)
|
|
}
|
|
apiSurface := generated.String()
|
|
if !strings.Contains(apiSurface, "export interface HappLinkResult") {
|
|
t.Fatalf("generated types omit allowed response schema:\n%s", apiSurface)
|
|
}
|
|
if strings.Contains(apiSurface, "HappLinkGenerator") {
|
|
t.Fatalf("generated types expose DI interface:\n%s", apiSurface)
|
|
}
|
|
}
|
|
|
|
func TestKeepAliveOverrideEmitsNumberOrString(t *testing.T) {
|
|
dir := t.TempDir()
|
|
source := "package model\ntype Client struct { KeepAlive *int `json:\"keepAlive,omitempty\"` }\n"
|
|
if err := os.WriteFile(filepath.Join(dir, "client.go"), []byte(source), 0o644); err != nil {
|
|
t.Fatalf("write fixture: %v", err)
|
|
}
|
|
union := TypeRef{Kind: KindUnion, Variants: []TypeRef{{Kind: KindInt}, {Kind: KindString}}}
|
|
schemas, _, err := walkPackages([]packageRequest{{Path: dir, StructAllow: setOf("Client"), Overrides: map[string][]walkOverride{"Client": {{Field: "KeepAlive", Type: union}}}}})
|
|
if err != nil {
|
|
t.Fatalf("walk fixture: %v", err)
|
|
}
|
|
var types bytes.Buffer
|
|
if err := emitTypes(&types, schemas, nil); err != nil || !strings.Contains(types.String(), "keepAlive?: number | string;") {
|
|
t.Fatalf("generated TypeScript keepAlive = %s, err = %v", types.String(), err)
|
|
}
|
|
var zod bytes.Buffer
|
|
if err := emitZod(&zod, schemas, nil); err != nil || !strings.Contains(zod.String(), "z.union([z.number().int(), z.string()]).optional()") {
|
|
t.Fatalf("generated Zod keepAlive = %s, err = %v", zod.String(), err)
|
|
}
|
|
var schema bytes.Buffer
|
|
if err := emitJSONSchema(&schema, schemas, nil); err != nil || !strings.Contains(schema.String(), `"oneOf"`) {
|
|
t.Fatalf("generated JSON schema keepAlive = %s, err = %v", schema.String(), err)
|
|
}
|
|
}
|