Commit 61f103ac authored by Mrs4s's avatar Mrs4s

bug fix.

parent 05643146
...@@ -212,9 +212,9 @@ func (bot *CQBot) ToElement(t string, d map[string]string, group bool) (message. ...@@ -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. ...@@ -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 {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment