mirror of
https://github.com/Mrs4s/MiraiGo.git
synced 2025-05-04 19:17:38 +08:00
75 lines
2.2 KiB
Go
75 lines
2.2 KiB
Go
package client
|
|
|
|
import (
|
|
"github.com/pkg/errors"
|
|
protobuf "github.com/segmentio/encoding/proto"
|
|
"google.golang.org/protobuf/proto"
|
|
|
|
"github.com/Mrs4s/MiraiGo/internal/packets"
|
|
"github.com/Mrs4s/MiraiGo/internal/protobuf/data/oidb"
|
|
"github.com/Mrs4s/MiraiGo/internal/protobuf/data/oidb/oidb0xbcb"
|
|
)
|
|
|
|
func init() {
|
|
decoders["OidbSvc.0xbcb_0"] = decodeUrlCheckResponse
|
|
}
|
|
|
|
type UrlSecurityLevel int
|
|
|
|
const (
|
|
Safe UrlSecurityLevel = 1
|
|
Unknown UrlSecurityLevel = 2
|
|
Danger UrlSecurityLevel = 3
|
|
)
|
|
|
|
// CheckUrlSafely 通过TX服务器检查URL安全性
|
|
func (c *QQClient) CheckUrlSafely(url string) UrlSecurityLevel {
|
|
i, err := c.sendAndWait(c.buildUrlCheckRequest(url))
|
|
if err != nil {
|
|
return Unknown
|
|
}
|
|
return i.(UrlSecurityLevel)
|
|
}
|
|
|
|
func (c *QQClient) buildUrlCheckRequest(url string) (uint16, []byte) {
|
|
seq := c.nextSeq()
|
|
payload := c.packOIDBPackageProto2(3019, 0, &oidb0xbcb.ReqBody{
|
|
CheckUrlReq: &oidb0xbcb.CheckUrlReq{
|
|
Url: []string{url},
|
|
QqPfTo: proto.String("mqq.group"),
|
|
Type: proto.Uint32(2),
|
|
SendUin: proto.Uint64(uint64(c.Uin)),
|
|
ReqType: proto.String("webview"),
|
|
OriginalUrl: &url,
|
|
IsArk: proto.Bool(false),
|
|
IsFinish: proto.Bool(false),
|
|
SrcUrls: []string{url},
|
|
SrcPlatform: proto.Uint32(1),
|
|
Qua: proto.String("AQQ_2013 4.6/2013 8.4.184945&NA_0/000000&ADR&null18&linux&2017&C2293D02BEE31158&7.1.2&V3"),
|
|
},
|
|
})
|
|
packet := packets.BuildUniPacket(c.Uin, seq, "OidbSvc.0xbcb_0", 1, c.OutGoingPacketSessionId, EmptyBytes, c.sigInfo.d2Key, payload)
|
|
return seq, packet
|
|
}
|
|
|
|
func decodeUrlCheckResponse(_ *QQClient, _ *incomingPacketInfo, payload []byte) (interface{}, error) {
|
|
pkg := &oidb.OIDBSSOPkg{}
|
|
rsp := &oidb0xbcb.RspBody{}
|
|
if err := protobuf.Unmarshal(payload, pkg); err != nil {
|
|
return nil, errors.Wrap(err, "failed to unmarshal protobuf message")
|
|
}
|
|
if err := protobuf.Unmarshal(pkg.Bodybuffer, rsp); err != nil {
|
|
return nil, errors.Wrap(err, "failed to unmarshal protobuf message")
|
|
}
|
|
if rsp.CheckUrlRsp == nil || len(rsp.CheckUrlRsp.Results) == 0 {
|
|
return nil, errors.New("response is empty")
|
|
}
|
|
if rsp.CheckUrlRsp.Results[0].JumpUrl != nil {
|
|
return Danger, nil
|
|
}
|
|
if rsp.CheckUrlRsp.Results[0].GetUmrtype() == 2 {
|
|
return Safe, nil
|
|
}
|
|
return Unknown, nil
|
|
}
|