chore: refresh gramsrv public release
This commit is contained in:
A
parent
75cebe8dbf
commit
70b6820474
1274 changed files with 378751 additions and 59919 deletions
294
cmd/giftcheck/main.go
Normal file
294
cmd/giftcheck/main.go
Normal file
|
|
@ -0,0 +1,294 @@
|
|||
// Command giftcheck 是 Star 礼物端到端验证工具（开发用，非生产组件）。
|
||||
//
|
||||
// 以用户身份登录本地 telesrv，验证 star gift 全链路对 live 服务端：
|
||||
// 1. payments.getStarGifts —— 打印礼物目录。
|
||||
// 2. 从 dialogs 找一个收礼用户（或 -to 指定）。
|
||||
// 3. payments.getPaymentForm(inputInvoiceStarGift) —— 必须返 paymentFormStarGift（XTR+非空 prices）。
|
||||
// 4. payments.sendStarsForm —— 必须返 paymentResult{updates}（含礼物服务消息 + updateStarsBalance）。
|
||||
// 5. payments.getStarsStatus 前后对比验证扣费。
|
||||
//
|
||||
// 用法： go run ./cmd/giftcheck -phone "+8618800000001"
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"crypto/rand"
|
||||
"flag"
|
||||
"fmt"
|
||||
"net"
|
||||
"os"
|
||||
"os/signal"
|
||||
"strconv"
|
||||
"syscall"
|
||||
|
||||
"go.uber.org/zap"
|
||||
|
||||
"github.com/gotd/log/logzap"
|
||||
"github.com/gotd/td/exchange"
|
||||
"github.com/gotd/td/mtproxy"
|
||||
"github.com/gotd/td/mtproxy/obfuscator"
|
||||
"github.com/gotd/td/proto/codec"
|
||||
"github.com/gotd/td/telegram"
|
||||
"github.com/gotd/td/telegram/dcs"
|
||||
"github.com/gotd/td/tg"
|
||||
"github.com/gotd/td/transport"
|
||||
|
||||
"telesrv/internal/mtprotoedge"
|
||||
)
|
||||
|
||||
type obfuscatedResolver struct {
|
||||
host string
|
||||
port int
|
||||
}
|
||||
|
||||
func (r obfuscatedResolver) dial(ctx context.Context, dc int) (transport.Conn, error) {
|
||||
var d net.Dialer
|
||||
conn, err := d.DialContext(ctx, "tcp", net.JoinHostPort(r.host, strconv.Itoa(r.port)))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
obf := obfuscator.Obfuscated2(rand.Reader, conn)
|
||||
if err := obf.Handshake(codec.IntermediateClientStart, dc, mtproxy.Secret{}); err != nil {
|
||||
_ = conn.Close()
|
||||
return nil, fmt.Errorf("obfuscated2 handshake: %w", err)
|
||||
}
|
||||
proto := transport.NewProtocol(func() transport.Codec { return codec.NoHeader{Codec: codec.Intermediate{}} })
|
||||
tc, err := proto.Handshake(obf)
|
||||
if err != nil {
|
||||
_ = conn.Close()
|
||||
return nil, fmt.Errorf("transport handshake: %w", err)
|
||||
}
|
||||
return tc, nil
|
||||
}
|
||||
|
||||
func (r obfuscatedResolver) Primary(ctx context.Context, dc int, _ dcs.List) (transport.Conn, error) {
|
||||
return r.dial(ctx, dc)
|
||||
}
|
||||
func (r obfuscatedResolver) MediaOnly(ctx context.Context, dc int, _ dcs.List) (transport.Conn, error) {
|
||||
return r.dial(ctx, dc)
|
||||
}
|
||||
func (r obfuscatedResolver) CDN(ctx context.Context, dc int, _ dcs.List) (transport.Conn, error) {
|
||||
return r.dial(ctx, dc)
|
||||
}
|
||||
|
||||
func main() {
|
||||
addr := flag.String("addr", "127.0.0.1:2398", "telesrv MTProto 地址")
|
||||
dcID := flag.Int("dc", 2, "DC id")
|
||||
rsaPath := flag.String("rsa", "data/server_rsa.pem", "server RSA key 路径")
|
||||
apiID := flag.Int("api-id", 1, "api_id")
|
||||
apiHash := flag.String("api-hash", "hash", "api_hash")
|
||||
phone := flag.String("phone", "", "登录手机号")
|
||||
code := flag.String("code", "12345", "开发登录码")
|
||||
toID := flag.Int64("to", 0, "收礼用户 id（0=自动从 dialogs 找）")
|
||||
toHash := flag.Int64("to-hash", 0, "收礼用户 access_hash")
|
||||
flag.Parse()
|
||||
if *phone == "" {
|
||||
fmt.Fprintln(os.Stderr, "缺少 -phone")
|
||||
os.Exit(2)
|
||||
}
|
||||
|
||||
logger, _ := zap.NewDevelopment()
|
||||
defer func() { _ = logger.Sync() }()
|
||||
priv, err := mtprotoedge.LoadOrGenerateRSAKey(*rsaPath)
|
||||
if err != nil {
|
||||
logger.Fatal("加载 RSA key 失败", zap.Error(err))
|
||||
}
|
||||
host, portStr, _ := net.SplitHostPort(*addr)
|
||||
port, _ := strconv.Atoi(portStr)
|
||||
|
||||
client := telegram.NewClient(*apiID, *apiHash, telegram.Options{
|
||||
PublicKeys: []exchange.PublicKey{{RSA: &priv.PublicKey}},
|
||||
Resolver: obfuscatedResolver{host: host, port: port},
|
||||
DCList: dcs.List{Options: []tg.DCOption{{ID: *dcID, IPAddress: host, Port: port, Static: true}}},
|
||||
Logger: logzap.New(logger.Named("client")),
|
||||
})
|
||||
ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
|
||||
defer stop()
|
||||
|
||||
if err := client.Run(ctx, func(ctx context.Context) error {
|
||||
raw := tg.NewClient(client)
|
||||
sent, err := raw.AuthSendCode(ctx, &tg.AuthSendCodeRequest{PhoneNumber: *phone, APIID: *apiID, APIHash: *apiHash, Settings: tg.CodeSettings{}})
|
||||
if err != nil {
|
||||
return fmt.Errorf("sendCode: %w", err)
|
||||
}
|
||||
sc, ok := sent.(*tg.AuthSentCode)
|
||||
if !ok {
|
||||
return fmt.Errorf("sentCode 类型 = %T", sent)
|
||||
}
|
||||
authz, err := raw.AuthSignIn(ctx, &tg.AuthSignInRequest{PhoneNumber: *phone, PhoneCodeHash: sc.PhoneCodeHash, PhoneCode: *code})
|
||||
if err != nil {
|
||||
return fmt.Errorf("signIn: %w", err)
|
||||
}
|
||||
self := authz.(*tg.AuthAuthorization).User.(*tg.User)
|
||||
fmt.Printf("==== 登录: id=%d name=%q ====\n", self.ID, self.FirstName)
|
||||
|
||||
// 1. 目录。
|
||||
gifts := giftCatalog(ctx, raw)
|
||||
if len(gifts) == 0 {
|
||||
return fmt.Errorf("礼物目录为空（animated_emoji 未 seed？）")
|
||||
}
|
||||
gift := gifts[0]
|
||||
fmt.Printf(" 目录 %d 个礼物；选第一个: id=%d stars=%d title=%q sticker=%T\n", len(gifts), gift.ID, gift.Stars, giftTitle(gift), gift.Sticker)
|
||||
|
||||
balBefore := starsBalance(ctx, raw, "扣费前")
|
||||
|
||||
// 2. 收礼用户（-to 指定 id 时从 dialogs 解析其 access_hash；否则取首个 user）。
|
||||
var to *tg.InputPeerUser
|
||||
if *toID != 0 && *toHash != 0 {
|
||||
to = &tg.InputPeerUser{UserID: *toID, AccessHash: *toHash}
|
||||
} else {
|
||||
to = findRecipient(ctx, raw, self.ID, *toID)
|
||||
}
|
||||
if to == nil {
|
||||
return fmt.Errorf("找不到收礼用户（用 -to/-to-hash 指定）")
|
||||
}
|
||||
fmt.Printf(" 收礼用户: id=%d\n", to.UserID)
|
||||
|
||||
inv := &tg.InputInvoiceStarGift{Peer: to, GiftID: gift.ID}
|
||||
|
||||
// 3. getPaymentForm → paymentFormStarGift。
|
||||
formRes, err := raw.PaymentsGetPaymentForm(ctx, &tg.PaymentsGetPaymentFormRequest{Invoice: inv})
|
||||
if err != nil {
|
||||
return fmt.Errorf("getPaymentForm: %w", err)
|
||||
}
|
||||
form, ok := formRes.(*tg.PaymentsPaymentFormStarGift)
|
||||
if !ok {
|
||||
fmt.Printf(" [FAIL] getPaymentForm 返回 %T，want *PaymentsPaymentFormStarGift（TDesktop 单分支 match）\n", formRes)
|
||||
return nil
|
||||
}
|
||||
fmt.Printf(" getPaymentForm: paymentFormStarGift form_id=%d currency=%s prices=%d\n", form.FormID, form.Invoice.Currency, len(form.Invoice.Prices))
|
||||
if form.Invoice.Currency != "XTR" || len(form.Invoice.Prices) == 0 {
|
||||
fmt.Printf(" [FAIL] invoice 须 XTR + 非空 prices\n")
|
||||
return nil
|
||||
}
|
||||
|
||||
// 4. sendStarsForm → paymentResult。
|
||||
payRes, err := raw.PaymentsSendStarsForm(ctx, &tg.PaymentsSendStarsFormRequest{FormID: form.FormID, Invoice: inv})
|
||||
if err != nil {
|
||||
return fmt.Errorf("sendStarsForm: %w", err)
|
||||
}
|
||||
pay, ok := payRes.(*tg.PaymentsPaymentResult)
|
||||
if !ok {
|
||||
fmt.Printf(" [FAIL] sendStarsForm 返回 %T，want *PaymentsPaymentResult（DrKLO 强转）\n", payRes)
|
||||
return nil
|
||||
}
|
||||
inspectGiftUpdates(pay.Updates)
|
||||
|
||||
balAfter := starsBalance(ctx, raw, "扣费后")
|
||||
fmt.Printf("==== 扣费: %d -> %d，差 %d（期望 -%d）====\n", balBefore, balAfter, balBefore-balAfter, gift.Stars)
|
||||
if balBefore-balAfter == gift.Stars {
|
||||
fmt.Println("==== ✅ PASS：star gift 扣费正确 ====")
|
||||
} else {
|
||||
fmt.Println("==== ❌ FAIL：扣费金额不符 ====")
|
||||
}
|
||||
return nil
|
||||
}); err != nil {
|
||||
logger.Fatal("运行失败", zap.Error(err))
|
||||
}
|
||||
}
|
||||
|
||||
func giftCatalog(ctx context.Context, raw *tg.Client) []*tg.StarGift {
|
||||
res, err := raw.PaymentsGetStarGifts(ctx, 0)
|
||||
if err != nil {
|
||||
fmt.Printf(" getStarGifts 失败: %v\n", err)
|
||||
return nil
|
||||
}
|
||||
full, ok := res.(*tg.PaymentsStarGifts)
|
||||
if !ok {
|
||||
fmt.Printf(" getStarGifts 返回 %T\n", res)
|
||||
return nil
|
||||
}
|
||||
out := make([]*tg.StarGift, 0, len(full.Gifts))
|
||||
for _, g := range full.Gifts {
|
||||
if sg, ok := g.(*tg.StarGift); ok {
|
||||
out = append(out, sg)
|
||||
}
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
func giftTitle(g *tg.StarGift) string {
|
||||
t, _ := g.GetTitle()
|
||||
return t
|
||||
}
|
||||
|
||||
func starsBalance(ctx context.Context, raw *tg.Client, label string) int64 {
|
||||
status, err := raw.PaymentsGetStarsStatus(ctx, &tg.PaymentsGetStarsStatusRequest{Peer: &tg.InputPeerSelf{}})
|
||||
if err != nil {
|
||||
fmt.Printf(" [%s] getStarsStatus 失败: %v\n", label, err)
|
||||
return -1
|
||||
}
|
||||
var bal int64
|
||||
if amt, ok := status.Balance.(*tg.StarsAmount); ok {
|
||||
bal = amt.Amount
|
||||
}
|
||||
fmt.Printf(" [%s] 余额=%d stars\n", label, bal)
|
||||
return bal
|
||||
}
|
||||
|
||||
func findRecipient(ctx context.Context, raw *tg.Client, selfID, targetID int64) *tg.InputPeerUser {
|
||||
dlgs, err := raw.MessagesGetDialogs(ctx, &tg.MessagesGetDialogsRequest{OffsetPeer: &tg.InputPeerEmpty{}, Limit: 100})
|
||||
if err != nil {
|
||||
return nil
|
||||
}
|
||||
var users []tg.UserClass
|
||||
switch v := dlgs.(type) {
|
||||
case *tg.MessagesDialogs:
|
||||
users = v.Users
|
||||
case *tg.MessagesDialogsSlice:
|
||||
users = v.Users
|
||||
default:
|
||||
return nil
|
||||
}
|
||||
for _, uc := range users {
|
||||
u, ok := uc.(*tg.User)
|
||||
if !ok || u.ID == selfID || u.Bot || u.Self {
|
||||
continue
|
||||
}
|
||||
if targetID != 0 {
|
||||
if u.ID == targetID {
|
||||
return &tg.InputPeerUser{UserID: u.ID, AccessHash: u.AccessHash}
|
||||
}
|
||||
continue
|
||||
}
|
||||
return &tg.InputPeerUser{UserID: u.ID, AccessHash: u.AccessHash}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func inspectGiftUpdates(res tg.UpdatesClass) {
|
||||
var ups []tg.UpdateClass
|
||||
switch v := res.(type) {
|
||||
case *tg.Updates:
|
||||
ups = v.Updates
|
||||
case *tg.UpdatesCombined:
|
||||
ups = v.Updates
|
||||
case *tg.UpdateShort:
|
||||
ups = []tg.UpdateClass{v.Update}
|
||||
default:
|
||||
fmt.Printf(" [警告] paymentResult.updates 非 Updates 子类型: %T\n", res)
|
||||
return
|
||||
}
|
||||
hasGiftMsg, hasBalance := false, false
|
||||
for _, up := range ups {
|
||||
switch u := up.(type) {
|
||||
case *tg.UpdateNewMessage:
|
||||
if svc, ok := u.Message.(*tg.MessageService); ok {
|
||||
if a, ok := svc.Action.(*tg.MessageActionStarGift); ok {
|
||||
hasGiftMsg = true
|
||||
gid := int64(0)
|
||||
if sg, ok := a.Gift.(*tg.StarGift); ok {
|
||||
gid = sg.ID
|
||||
}
|
||||
fmt.Printf(" messageActionStarGift: msg_id=%d gift_id=%d\n", u.Message.(*tg.MessageService).ID, gid)
|
||||
}
|
||||
}
|
||||
case *tg.UpdateStarsBalance:
|
||||
hasBalance = true
|
||||
if amt, ok := u.Balance.(*tg.StarsAmount); ok {
|
||||
fmt.Printf(" updateStarsBalance: %d\n", amt.Amount)
|
||||
}
|
||||
}
|
||||
}
|
||||
fmt.Printf(" paymentResult: 含礼物服务消息=%v 含 updateStarsBalance=%v\n", hasGiftMsg, hasBalance)
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue