shkong/go-cqhttp
1
0
Fork 0
mirror of https://github.com/Mrs4s/go-cqhttp.git synced 2025-05-05 11:33:48 +08:00
Code Issues Projects Releases Wiki Activity

bug fix.

Browse Source
This commit is contained in:
Mrs4s 2020-08-07 16:18:12 +08:00
parent 0564314671
commit 61f103ac1d
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
coolq/cqcode.go
View File

@ -212,9 +212,9 @@ func (bot *CQBot) ToElement(t string, d map[string]string, group bool) (message.
kv := strings.SplitN(line, "=", 2) kv := strings.SplitN(line, "=", 2)
switch kv[0] { switch kv[0] {
case "md5": case "md5":
hash, _ = hex.DecodeString(kv[1]) hash, _ = hex.DecodeString(strings.ReplaceAll(kv[1], "\r", ""))
case "size": case "size":
t, _ := strconv.Atoi(kv[1]) t, _ := strconv.Atoi(strings.ReplaceAll(kv[1], "\r", ""))
size = int32(t) size = int32(t)
} }
} }
@ -223,6 +223,12 @@ func (bot *CQBot) ToElement(t string, d map[string]string, group bool) (message.
hash = r.ReadBytes(16) hash = r.ReadBytes(16)
size = r.ReadInt32() size = r.ReadInt32()
} }
if size == 0 {
return nil, errors.New("img size is 0")
}
if len(hash) != 16 {
return nil, errors.New("invalid hash")
}
if group { if group {
rsp, err := bot.Client.QueryGroupImage(1, hash, size) rsp, err := bot.Client.QueryGroupImage(1, hash, size)
if err != nil { if err != nil {