From 61f103ac1dde3d168deae059ceb4ba52f7fd6331 Mon Sep 17 00:00:00 2001 From: Mrs4s <1844812067@qq.com> Date: Fri, 7 Aug 2020 16:18:12 +0800 Subject: [PATCH] bug fix. --- coolq/cqcode.go | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/coolq/cqcode.go b/coolq/cqcode.go index 9e7ceb8..091f924 100644 --- a/coolq/cqcode.go +++ b/coolq/cqcode.go @@ -212,9 +212,9 @@ func (bot *CQBot) ToElement(t string, d map[string]string, group bool) (message. kv := strings.SplitN(line, "=", 2) switch kv[0] { case "md5": - hash, _ = hex.DecodeString(kv[1]) + hash, _ = hex.DecodeString(strings.ReplaceAll(kv[1], "\r", "")) case "size": - t, _ := strconv.Atoi(kv[1]) + t, _ := strconv.Atoi(strings.ReplaceAll(kv[1], "\r", "")) size = int32(t) } } @@ -223,6 +223,12 @@ func (bot *CQBot) ToElement(t string, d map[string]string, group bool) (message. hash = r.ReadBytes(16) 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 { rsp, err := bot.Client.QueryGroupImage(1, hash, size) if err != nil {