chore: refresh gramsrv public release
This commit is contained in:
A
parent
75cebe8dbf
commit
70b6820474
1274 changed files with 378751 additions and 59919 deletions
271
internal/rpc/encrypted_messages_test.go
Normal file
271
internal/rpc/encrypted_messages_test.go
Normal file
|
|
@ -0,0 +1,271 @@
|
|||
package rpc
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/gotd/td/tg"
|
||||
)
|
||||
|
||||
// acceptChat 跑完 request→accept，返回 normal 态密聊 id 与 participant 视角 access_hash。
|
||||
func (f *encryptedFixture) acceptChat(t *testing.T) (chatID int, partAccessHash int64) {
|
||||
t.Helper()
|
||||
res, err := f.router.onMessagesRequestEncryption(f.adminCtx(), &tg.MessagesRequestEncryptionRequest{
|
||||
UserID: &tg.InputUser{UserID: f.participant.ID, AccessHash: f.participant.AccessHash},
|
||||
RandomID: 4242,
|
||||
GA: dhParam(0x55),
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("request: %v", err)
|
||||
}
|
||||
chatID = res.(*tg.EncryptedChatWaiting).ID
|
||||
chat, _, _ := f.store.GetSecretChat(f.ctx, chatID)
|
||||
if _, err := f.router.onMessagesAcceptEncryption(f.participantCtx(), &tg.MessagesAcceptEncryptionRequest{
|
||||
Peer: tg.InputEncryptedChat{ChatID: chatID, AccessHash: chat.ParticipantAccessHash},
|
||||
GB: dhParam(0x66),
|
||||
KeyFingerprint: 7,
|
||||
}); err != nil {
|
||||
t.Fatalf("accept: %v", err)
|
||||
}
|
||||
return chatID, chat.ParticipantAccessHash
|
||||
}
|
||||
|
||||
func encNewMessagePayload(t *testing.T, rec phonePushRecord) *tg.UpdateNewEncryptedMessage {
|
||||
t.Helper()
|
||||
updates, ok := rec.msg.(*tg.Updates)
|
||||
if !ok || len(updates.Updates) != 1 {
|
||||
t.Fatalf("pushed msg = %T, want single-update tg.Updates", rec.msg)
|
||||
}
|
||||
upd, ok := updates.Updates[0].(*tg.UpdateNewEncryptedMessage)
|
||||
if !ok {
|
||||
t.Fatalf("pushed update = %T, want UpdateNewEncryptedMessage", updates.Updates[0])
|
||||
}
|
||||
return upd
|
||||
}
|
||||
|
||||
func TestSendEncryptedRPCFlow(t *testing.T) {
|
||||
f := newEncryptedFixture(t)
|
||||
chatID, _ := f.acceptChat(t)
|
||||
chat, _, _ := f.store.GetSecretChat(f.ctx, chatID)
|
||||
f.sessions.reset()
|
||||
|
||||
// admin 发加密消息 → 投给 participant 设备。
|
||||
data := []byte{0xDE, 0xAD, 0xBE, 0xEF}
|
||||
sent, err := f.router.onMessagesSendEncrypted(f.adminCtx(), &tg.MessagesSendEncryptedRequest{
|
||||
Peer: tg.InputEncryptedChat{ChatID: chatID, AccessHash: chat.AdminAccessHash},
|
||||
RandomID: 88888,
|
||||
Data: data,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("sendEncrypted: %v", err)
|
||||
}
|
||||
sentMsg, ok := sent.(*tg.MessagesSentEncryptedMessage)
|
||||
if !ok || sentMsg.Date == 0 {
|
||||
t.Fatalf("send response = %T %+v, want SentEncryptedMessage{date}", sent, sent)
|
||||
}
|
||||
recs := f.sessions.records()
|
||||
if len(recs) != 1 || recs[0].userID != f.participant.ID {
|
||||
t.Fatalf("send push = %+v, want single push to participant", recs)
|
||||
}
|
||||
upd := encNewMessagePayload(t, recs[0])
|
||||
if upd.Qts != 1 {
|
||||
t.Fatalf("pushed qts = %d, want 1", upd.Qts)
|
||||
}
|
||||
em, ok := upd.Message.(*tg.EncryptedMessage)
|
||||
if !ok || string(em.Bytes) != string(data) || em.ChatID != chatID {
|
||||
t.Fatalf("pushed message = %+v, want EncryptedMessage bytes verbatim", upd.Message)
|
||||
}
|
||||
|
||||
// participant getState：设备 qts = 1。
|
||||
st, err := f.router.onUpdatesGetState(f.participantCtx())
|
||||
if err != nil {
|
||||
t.Fatalf("getState: %v", err)
|
||||
}
|
||||
if st.Qts != 1 {
|
||||
t.Fatalf("participant getState qts = %d, want 1", st.Qts)
|
||||
}
|
||||
|
||||
// participant 离线补差分：从 qts=0 拿回该加密消息。
|
||||
diff, err := f.router.onUpdatesGetDifference(f.participantCtx(), &tg.UpdatesGetDifferenceRequest{Qts: 0})
|
||||
if err != nil {
|
||||
t.Fatalf("getDifference: %v", err)
|
||||
}
|
||||
full, ok := diff.(*tg.UpdatesDifference)
|
||||
if !ok {
|
||||
t.Fatalf("difference = %T, want *tg.UpdatesDifference", diff)
|
||||
}
|
||||
if len(full.NewEncryptedMessages) != 1 || full.State.Qts != 1 {
|
||||
t.Fatalf("difference enc msgs = %d state.qts = %d, want 1/1", len(full.NewEncryptedMessages), full.State.Qts)
|
||||
}
|
||||
gotEM, ok := full.NewEncryptedMessages[0].(*tg.EncryptedMessage)
|
||||
if !ok || string(gotEM.Bytes) != string(data) {
|
||||
t.Fatalf("difference message = %+v, want bytes verbatim", full.NewEncryptedMessages[0])
|
||||
}
|
||||
|
||||
// receivedQueue：确认到 qts=1，返回空 Vector。
|
||||
rq, err := f.router.onMessagesReceivedQueue(f.participantCtx(), 1)
|
||||
if err != nil {
|
||||
t.Fatalf("receivedQueue: %v", err)
|
||||
}
|
||||
if len(rq) != 0 {
|
||||
t.Fatalf("receivedQueue = %v, want empty vector", rq)
|
||||
}
|
||||
|
||||
// 确认后再补差分（qts=1）：已无新消息 → DifferenceEmpty。
|
||||
diff2, err := f.router.onUpdatesGetDifference(f.participantCtx(), &tg.UpdatesGetDifferenceRequest{Qts: 1})
|
||||
if err != nil {
|
||||
t.Fatalf("getDifference 2: %v", err)
|
||||
}
|
||||
if _, ok := diff2.(*tg.UpdatesDifferenceEmpty); !ok {
|
||||
t.Fatalf("difference after ack = %T, want UpdatesDifferenceEmpty", diff2)
|
||||
}
|
||||
|
||||
// 幂等重发同 random_id → 返回首次 date，不产生新 qts。
|
||||
f.sessions.reset()
|
||||
sent2, err := f.router.onMessagesSendEncrypted(f.adminCtx(), &tg.MessagesSendEncryptedRequest{
|
||||
Peer: tg.InputEncryptedChat{ChatID: chatID, AccessHash: chat.AdminAccessHash},
|
||||
RandomID: 88888,
|
||||
Data: data,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("idempotent resend: %v", err)
|
||||
}
|
||||
if sent2.(*tg.MessagesSentEncryptedMessage).Date != sentMsg.Date {
|
||||
t.Fatalf("idempotent resend date = %d, want %d (首次落库 date)", sent2.(*tg.MessagesSentEncryptedMessage).Date, sentMsg.Date)
|
||||
}
|
||||
}
|
||||
|
||||
func encOtherUpdate[T tg.UpdateClass](t *testing.T, diff tg.UpdatesDifferenceClass) T {
|
||||
t.Helper()
|
||||
full, ok := diff.(*tg.UpdatesDifference)
|
||||
if !ok {
|
||||
t.Fatalf("difference = %T, want *tg.UpdatesDifference", diff)
|
||||
}
|
||||
for _, u := range full.OtherUpdates {
|
||||
if got, ok := u.(T); ok {
|
||||
return got
|
||||
}
|
||||
}
|
||||
var zero T
|
||||
t.Fatalf("OtherUpdates %+v missing %T", full.OtherUpdates, zero)
|
||||
return zero
|
||||
}
|
||||
|
||||
// TestEncryptionStateEventOfflineDelivery：participant 在 requestEncryption 时离线，
|
||||
// 重连 getDifference 经 durable 状态事件补回 updateEncryption(requested)，且只补一次。
|
||||
func TestEncryptionStateEventOfflineDelivery(t *testing.T) {
|
||||
f := newEncryptedFixture(t)
|
||||
res, err := f.router.onMessagesRequestEncryption(f.adminCtx(), &tg.MessagesRequestEncryptionRequest{
|
||||
UserID: &tg.InputUser{UserID: f.participant.ID, AccessHash: f.participant.AccessHash},
|
||||
RandomID: 5151,
|
||||
GA: dhParam(0x55),
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("request: %v", err)
|
||||
}
|
||||
chatID := res.(*tg.EncryptedChatWaiting).ID
|
||||
|
||||
// participant 离线补差分：拿到 updateEncryption(encryptedChatRequested, 携 g_a)。
|
||||
diff, err := f.router.onUpdatesGetDifference(f.participantCtx(), &tg.UpdatesGetDifferenceRequest{Qts: 0})
|
||||
if err != nil {
|
||||
t.Fatalf("getDifference: %v", err)
|
||||
}
|
||||
upd := encOtherUpdate[*tg.UpdateEncryption](t, diff)
|
||||
requested, ok := upd.Chat.(*tg.EncryptedChatRequested)
|
||||
if !ok || requested.ID != chatID || len(requested.GA) == 0 {
|
||||
t.Fatalf("offline handshake update = %+v, want EncryptedChatRequested with g_a", upd.Chat)
|
||||
}
|
||||
|
||||
// 再次补差分：已投递 → 不重复（DifferenceEmpty）。
|
||||
diff2, err := f.router.onUpdatesGetDifference(f.participantCtx(), &tg.UpdatesGetDifferenceRequest{Qts: 0})
|
||||
if err != nil {
|
||||
t.Fatalf("getDifference 2: %v", err)
|
||||
}
|
||||
if _, ok := diff2.(*tg.UpdatesDifferenceEmpty); !ok {
|
||||
t.Fatalf("redelivery: difference = %T, want UpdatesDifferenceEmpty", diff2)
|
||||
}
|
||||
}
|
||||
|
||||
// TestDiscardStateEventOfflineDelivery：admin 在 participant 未接受时 discard（账号级），
|
||||
// participant 离线设备 getDifference 补回 encryptedChatDiscarded。
|
||||
func TestDiscardStateEventOfflineDelivery(t *testing.T) {
|
||||
f := newEncryptedFixture(t)
|
||||
res, err := f.router.onMessagesRequestEncryption(f.adminCtx(), &tg.MessagesRequestEncryptionRequest{
|
||||
UserID: &tg.InputUser{UserID: f.participant.ID, AccessHash: f.participant.AccessHash},
|
||||
RandomID: 6262,
|
||||
GA: dhParam(0x55),
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("request: %v", err)
|
||||
}
|
||||
chatID := res.(*tg.EncryptedChatWaiting).ID
|
||||
// admin 撤回邀请。
|
||||
if _, err := f.router.onMessagesDiscardEncryption(f.adminCtx(), &tg.MessagesDiscardEncryptionRequest{ChatID: chatID, DeleteHistory: false}); err != nil {
|
||||
t.Fatalf("discard: %v", err)
|
||||
}
|
||||
// participant 离线补差分：现态是 discarded，重建出 encryptedChatDiscarded。
|
||||
diff, err := f.router.onUpdatesGetDifference(f.participantCtx(), &tg.UpdatesGetDifferenceRequest{Qts: 0})
|
||||
if err != nil {
|
||||
t.Fatalf("getDifference: %v", err)
|
||||
}
|
||||
upd := encOtherUpdate[*tg.UpdateEncryption](t, diff)
|
||||
if _, ok := upd.Chat.(*tg.EncryptedChatDiscarded); !ok {
|
||||
t.Fatalf("offline discard update = %T, want EncryptedChatDiscarded", upd.Chat)
|
||||
}
|
||||
}
|
||||
|
||||
func TestReadEncryptedHistoryPushesPeer(t *testing.T) {
|
||||
f := newEncryptedFixture(t)
|
||||
chatID, _ := f.acceptChat(t)
|
||||
chat, _, _ := f.store.GetSecretChat(f.ctx, chatID)
|
||||
f.sessions.reset()
|
||||
|
||||
ok, err := f.router.onMessagesReadEncryptedHistory(f.adminCtx(), &tg.MessagesReadEncryptedHistoryRequest{
|
||||
Peer: tg.InputEncryptedChat{ChatID: chatID, AccessHash: chat.AdminAccessHash},
|
||||
MaxDate: 1500,
|
||||
})
|
||||
if err != nil || !ok {
|
||||
t.Fatalf("readEncryptedHistory: ok=%v err=%v", ok, err)
|
||||
}
|
||||
recs := f.sessions.records()
|
||||
if len(recs) != 1 || recs[0].userID != f.participant.ID {
|
||||
t.Fatalf("read push = %+v, want single push to participant", recs)
|
||||
}
|
||||
updates := recs[0].msg.(*tg.Updates)
|
||||
rd, ok := updates.Updates[0].(*tg.UpdateEncryptedMessagesRead)
|
||||
if !ok || rd.ChatID != chatID || rd.MaxDate != 1500 {
|
||||
t.Fatalf("read update = %+v, want UpdateEncryptedMessagesRead{chat,max_date=1500}", updates.Updates[0])
|
||||
}
|
||||
}
|
||||
|
||||
func TestReadEncryptedHistoryInvalidMaxDate(t *testing.T) {
|
||||
f := newEncryptedFixture(t)
|
||||
chatID, _ := f.acceptChat(t)
|
||||
chat, _, _ := f.store.GetSecretChat(f.ctx, chatID)
|
||||
_, err := f.router.onMessagesReadEncryptedHistory(f.adminCtx(), &tg.MessagesReadEncryptedHistoryRequest{
|
||||
Peer: tg.InputEncryptedChat{ChatID: chatID, AccessHash: chat.AdminAccessHash},
|
||||
MaxDate: 0,
|
||||
})
|
||||
assertPhoneRPCErr(t, err, "MAX_DATE_INVALID")
|
||||
}
|
||||
|
||||
func TestSetEncryptedTypingPushesPeer(t *testing.T) {
|
||||
f := newEncryptedFixture(t)
|
||||
chatID, _ := f.acceptChat(t)
|
||||
chat, _, _ := f.store.GetSecretChat(f.ctx, chatID)
|
||||
f.sessions.reset()
|
||||
|
||||
ok, err := f.router.onMessagesSetEncryptedTyping(f.adminCtx(), &tg.MessagesSetEncryptedTypingRequest{
|
||||
Peer: tg.InputEncryptedChat{ChatID: chatID, AccessHash: chat.AdminAccessHash},
|
||||
Typing: true,
|
||||
})
|
||||
if err != nil || !ok {
|
||||
t.Fatalf("setEncryptedTyping: ok=%v err=%v", ok, err)
|
||||
}
|
||||
recs := f.sessions.records()
|
||||
if len(recs) != 1 || recs[0].userID != f.participant.ID {
|
||||
t.Fatalf("typing push = %+v, want single push to participant", recs)
|
||||
}
|
||||
if _, ok := recs[0].msg.(*tg.Updates).Updates[0].(*tg.UpdateEncryptedChatTyping); !ok {
|
||||
t.Fatalf("typing update = %T, want UpdateEncryptedChatTyping", recs[0].msg.(*tg.Updates).Updates[0])
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue