astra/owpengram-server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

perf: sync protocol and core hardening updates

Browse source
This commit is contained in:
A 2026-07-11 19:48:26 +08:00
parent 152fed3b87
commit 4390ebf5a9
283 changed files with 29231 additions and 2295 deletions
Download patch file Download diff file Expand all files Collapse all files

94
internal/rpc/account_themes_compat_test.go
View file

@ -6,31 +6,39 @@ import (
"github.com/gotd/td/bin"
"github.com/gotd/td/tg"
"github.com/gotd/td/tgerr"
"go.uber.org/zap"
"go.uber.org/zap/zaptest/observer"
"telesrv/internal/domain"
)
// TestLegacyThemeWireDecode 验证按 DrKLO 12.8.1 的 theme 构造器(比 gotd schema 新)
// 手写解码后能正确复用现有 handler。直接构造 DrKLO 的 wire 字节喂给 fallback compat 层。
func TestLegacyThemeWireDecode(t *testing.T) {
// TestLegacyThemeWireDispatch 验证 DrKLO theme 构造器经过 Router.Dispatch 的完整链路：
// layerwire 结构预检 -> gotd dispatcher fallback -> compat 解码 -> 现有 handler。
// 不能只直调 tryLegacyThemeRPC，否则会掩盖预检早于 fallback 的路由回归。
func TestLegacyThemeWireDispatch(t *testing.T) {
const userID = 1000010
ctx := WithUserID(context.Background(), userID)
var authKeyID [8]byte
authKeyID[0] = 1
const sessionID = 99
files := &fakeFiles{docs: map[int64]domain.Document{
777: {ID: 777, AccessHash: 7, DCID: 2, MimeType: "application/x-tgtheme-android", Size: 4096},
}}
r := newThemeRouter(t, files)
// createTheme 0x8432c21f:flags(=4,document) + slug + title + InputDocument。
// createTheme 0x8432c21f:flags(document+single settings) + slug + title。
var cb bin.Buffer
cb.PutID(legacyCreateThemeID)
cb.PutInt32(1 << 2) // document present
cb.PutString("") // empty slug → auto
cb.PutInt32((1 << 2) | (1 << 3))
cb.PutString("") // empty slug → auto
cb.PutString("Legacy Theme")
(&tg.InputDocument{ID: 777, AccessHash: 7}).Encode(&cb)
(&tg.InputThemeSettings{BaseTheme: &tg.BaseThemeDay{}, AccentColor: 0x3997d3}).Encode(&cb)
enc, handled, err := r.tryLegacyThemeRPC(ctx, &cb)
if !handled || err != nil {
t.Fatalf("createTheme legacy = handled %v err %v", handled, err)
enc, err := r.Dispatch(ctx, authKeyID, sessionID, &cb)
if err != nil {
t.Fatalf("createTheme legacy dispatch: %v", err)
}
th, ok := enc.(*tg.Theme)
if !ok {
@ -44,9 +52,29 @@ func TestLegacyThemeWireDecode(t *testing.T) {
} else if d, _ := doc.(*tg.Document); d == nil || d.ID != 777 {
t.Fatalf("created theme document = %#v, want id 777", doc)
}
if settings, ok := th.GetSettings(); !ok || len(settings) != 1 || settings[0].AccentColor != 0x3997d3 {
t.Fatalf("created theme settings = %#v ok=%v, want one legacy setting", settings, ok)
}
mustEncodeTheme(t, th)
slug := th.Slug
// updateTheme 0x5cb367d5:flags(=2,title) + format + InputTheme + title。
var ub bin.Buffer
ub.PutID(legacyUpdateThemeID)
ub.PutInt32(1 << 1)
ub.PutString("android")
(&tg.InputTheme{ID: th.ID, AccessHash: th.AccessHash}).Encode(&ub)
ub.PutString("Legacy Theme Updated")
enc, err = r.Dispatch(ctx, authKeyID, sessionID, &ub)
if err != nil {
t.Fatalf("updateTheme legacy dispatch: %v", err)
}
updated, ok := enc.(*tg.Theme)
if !ok || updated.Title != "Legacy Theme Updated" {
t.Fatalf("updateTheme legacy result = %#v, want updated title", enc)
}
// getTheme 0x8d9d742b:format + InputThemeSlug + document_id(被忽略)。
var gb bin.Buffer
gb.PutID(legacyGetThemeID)
@ -54,9 +82,9 @@ func TestLegacyThemeWireDecode(t *testing.T) {
(&tg.InputThemeSlug{Slug: slug}).Encode(&gb)
gb.PutLong(12345) // document_id ignored
enc, handled, err = r.tryLegacyThemeRPC(ctx, &gb)
if !handled || err != nil {
t.Fatalf("getTheme legacy = handled %v err %v", handled, err)
enc, err = r.Dispatch(ctx, authKeyID, sessionID, &gb)
if err != nil {
t.Fatalf("getTheme legacy dispatch: %v", err)
}
got, ok := enc.(*tg.Theme)
if !ok || got.Slug != slug {
@ -73,18 +101,44 @@ func TestLegacyThemeWireDecode(t *testing.T) {
ib.PutString("android")
(&tg.InputThemeSlug{Slug: slug}).Encode(&ib)
enc, handled, err = r.tryLegacyThemeRPC(ctx, &ib)
if !handled || err != nil {
t.Fatalf("installTheme legacy = handled %v err %v", handled, err)
enc, err = r.Dispatch(ctx, authKeyID, sessionID, &ib)
if err != nil {
t.Fatalf("installTheme legacy dispatch: %v", err)
}
if _, ok := enc.(*tg.BoolTrue); !ok {
t.Fatalf("installTheme legacy result = %T, want *tg.BoolTrue", enc)
}
// 非 theme 构造器 → 不处理。
var ob bin.Buffer
ob.PutID(0x12345678)
if _, handled, _ := r.tryLegacyThemeRPC(ctx, &ob); handled {
t.Fatalf("unrelated ctor should not be handled")
// 已声明 legacy 方法仍必须精确消费完整结构，截断字段不能到手写 decoder。
var malformed bin.Buffer
malformed.PutID(legacyCreateThemeID)
malformed.PutInt32(0)
malformed.PutString("slug") // missing title
if _, err := r.Dispatch(ctx, authKeyID, sessionID, &malformed); !tgerr.Is(err, "INPUT_REQUEST_INVALID") {
t.Fatalf("malformed legacy theme err = %v, want INPUT_REQUEST_INVALID", err)
}
}
func TestUnknownRPCReachesCompatibilityTraceAfterOpaquePreflight(t *testing.T) {
const unknownID = uint32(0x12345678)
r := newThemeRouter(t, &fakeFiles{})
r.deps.Auth = &captureAuthService{}
core, logs := observer.New(zap.WarnLevel)
r.log = zap.New(core)
var b bin.Buffer
b.PutID(unknownID)
// Deliberately resembles a forged vector count. Because the constructor is unknown, the
// body remains opaque and is never decoded or allocated from; total frame/RPC budgets bound it.
b.PutUint32(0xffffffff)
if _, err := r.Dispatch(context.Background(), [8]byte{1}, 101, &b); !tgerr.Is(err, "NOT_IMPLEMENTED") {
t.Fatalf("unknown dispatch err = %v, want NOT_IMPLEMENTED", err)
}
entries := logs.FilterMessage("Unhandled RPC (compatibility trace)").All()
if len(entries) != 1 {
t.Fatalf("compatibility trace entries = %d, want 1", len(entries))
}
if got, ok := entries[0].ContextMap()["type_id"]; !ok || got != "0x12345678" {
t.Fatalf("trace type_id = %#v, want %#x", got, unknownID)
}
}