zishuo/chatgpt-dingtalk
1
0
Fork 0
mirror of https://github.com/eryajf/chatgpt-dingtalk.git synced 2025-10-08 17:40:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: 解决请求信息通过伪造绕过校验的问题 (#166)

Browse Source
This commit is contained in:
二丫讲梵
2023-04-03 17:17:36 +08:00
committed by GitHub
parent b4ff4d275a
commit 5d648a8ce4
2 changed files with 7 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
pkg/db/chat.go
View File

@@ -21,9 +21,10 @@ type Chat struct {
Content string `gorm:"type:varchar(128);comment:'内容'" json:"content"` // 问题或回答的内容 Content string `gorm:"type:varchar(128);comment:'内容'" json:"content"` // 问题或回答的内容
} }
// 需要考虑下如何处理一个完整对话的问题 type ChatListReq struct {
// 如果是单聊，那么就记录上下两句就好了 Username string `json:"username" form:"username"`
// 如果是串聊，则需要知道哪条是第一条，并依次往下记录 Source string `json:"source" form:"source"`
}
// Add 添加资源 // Add 添加资源
func (c Chat) Add() (uint, error) { func (c Chat) Add() (uint, error) {
@@ -36,11 +37,6 @@ func (c Chat) Find(filter map[string]interface{}, data *Chat) error {
return DB.Where(filter).First(&data).Error return DB.Where(filter).First(&data).Error
} }
type ChatListReq struct {
Username string `json:"username" form:"username"`
Source string `json:"source" form:"source"`
}
// List 获取数据列表 // List 获取数据列表
func (c Chat) List(req ChatListReq) ([]*Chat, error) { func (c Chat) List(req ChatListReq) ([]*Chat, error) {
var list []*Chat var list []*Chat

6
pkg/process/process_request.go
View File

@@ -72,7 +72,7 @@ func ProcessRequest(rmsg *dingbot.ReceiveMsg) error {
logger.Warning(fmt.Errorf("send message error: %v", err)) logger.Warning(fmt.Errorf("send message error: %v", err))
} }
case "查对话": case "查对话":
msg := "使用如下指令进行查询:\n\n---\n\n**#查对话 username:张三**\n\n---\n\n需要注意格式必须严格与上边一致，否则会查询失败\n\n只有钉钉管理员，程序系统管理员，与查自己的情况下，才会被允许" msg := "使用如下指令进行查询:\n\n---\n\n**#查对话 username:张三**\n\n---\n\n需要注意格式必须严格与上边一致，否则将会查询失败\n\n只有程序系统管理员有权限查询，即config.yml中的admin_users指定的人员。"
_, err := rmsg.ReplyToDingtalk(string(dingbot.MARKDOWN), msg) _, err := rmsg.ReplyToDingtalk(string(dingbot.MARKDOWN), msg)
if err != nil { if err != nil {
logger.Warning(fmt.Errorf("send message error: %v", err)) logger.Warning(fmt.Errorf("send message error: %v", err))
@@ -264,8 +264,8 @@ func ImageGenerate(rmsg *dingbot.ReceiveMsg) error {
} }
func SelectHistory(rmsg *dingbot.ReceiveMsg) error { func SelectHistory(rmsg *dingbot.ReceiveMsg) error {
name := strings.TrimSpace(strings.Split(rmsg.Text.Content, ":")[1]) name := strings.TrimSpace(strings.Split(rmsg.Text.Content, ":")[1])
if !rmsg.IsAdmin && name != rmsg.SenderNick && !public.JudgeAdminUsers(rmsg.SenderNick) { if !public.JudgeAdminUsers(rmsg.SenderNick) {
_, err := rmsg.ReplyToDingtalk(string(dingbot.MARKDOWN), "**🤷 抱歉，您没有权限查询其他人的对话记录！**") _, err := rmsg.ReplyToDingtalk(string(dingbot.MARKDOWN), "**🤷 抱歉，您没有查询对话记录的权限，只有程序管理员可以查询！**")
if err != nil { if err != nil {
logger.Error(fmt.Errorf("send message error: %v", err)) logger.Error(fmt.Errorf("send message error: %v", err))
return err return err