3x-ui/internal/database/shadowsocks_server_key_migration_test.go

package database

import (
	"encoding/base64"
	"encoding/json"
	"path/filepath"
	"testing"

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

// A stored Shadowsocks-2022 server key of the wrong size keeps xray-core from
// starting at all, so startup must replace it and leave every valid key alone.
func TestRepairShadowsocks2022ServerKeys(t *testing.T) {
	dbDir := t.TempDir()
	t.Setenv("XUI_DB_FOLDER", dbDir)
	if err := InitDB(filepath.Join(dbDir, "x-ui.db")); err != nil {
		t.Fatalf("InitDB: %v", err)
	}
	t.Cleanup(func() { _ = CloseDB() })

	key16 := base64.StdEncoding.EncodeToString([]byte("0123456789abcdef"))
	key32 := base64.StdEncoding.EncodeToString([]byte("0123456789abcdef0123456789abcdef"))
	rows := []struct {
		tag, settings, wantKey string
		wantBytes              int
	}{
		{"oversized", `{"method":"2022-blake3-aes-128-gcm","password":"` + key32 + `","clients":[]}`, "", 16},
		{"undersized", `{"method":"2022-blake3-chacha20-poly1305","password":"` + key16 + `","clients":[]}`, "", 32},
		{"valid", `{"method":"2022-blake3-aes-256-gcm","password":"` + key32 + `","clients":[]}`, key32, 32},
		{"raw valid", `{"method":"2022-blake3-aes-128-gcm","password":"0123456789abcd-!","clients":[]}`, "0123456789abcd-!", 0},
		{"legacy cipher", `{"method":"aes-256-gcm","password":"short","clients":[]}`, "short", 0},
	}
	for i, r := range rows {
		ib := model.Inbound{UserId: 1, Port: 31100 + i, Protocol: model.Shadowsocks, Tag: r.tag, Settings: r.settings}
		if err := GetDB().Create(&ib).Error; err != nil {
			t.Fatalf("create %s: %v", r.tag, err)
		}
	}

	if err := repairShadowsocks2022ServerKeys(); err != nil {
		t.Fatalf("repairShadowsocks2022ServerKeys: %v", err)
	}

	for _, r := range rows {
		t.Run(r.tag, func(t *testing.T) {
			var ib model.Inbound
			if err := GetDB().Where("tag = ?", r.tag).First(&ib).Error; err != nil {
				t.Fatal(err)
			}
			var got struct {
				Password string `json:"password"`
			}
			if err := json.Unmarshal([]byte(ib.Settings), &got); err != nil {
				t.Fatal(err)
			}
			if r.wantKey != "" {
				if got.Password != r.wantKey {
					t.Errorf("key changed to %q, want it kept as %q", got.Password, r.wantKey)
				}
				return
			}
			decoded, err := base64.StdEncoding.DecodeString(got.Password)
			if err != nil || len(decoded) != r.wantBytes {
				t.Errorf("repaired key %q: %d bytes (err %v), want %d", got.Password, len(decoded), err, r.wantBytes)
			}
		})
	}
}