106 lines
3.2 KiB
Go
106 lines
3.2 KiB
Go
package common
|
||
|
||
import (
|
||
"errors"
|
||
"github.com/gogf/gf/v2/frame/g"
|
||
"github.com/gogf/gf/v2/util/gconv"
|
||
"net/http"
|
||
"sort"
|
||
"strings"
|
||
)
|
||
|
||
// FormatPublicKey 将原始公钥字符串格式化为标准PEM格式的公钥
|
||
// 功能：为原始公钥添加PEM头部和尾部，并按64字符长度拆分换行，符合PKCS#8标准格式要求
|
||
// 参数 publicKey: 原始未格式化的公钥字符串（通常为Base64编码且无换行）
|
||
// 返回值: 格式化后的PEM格式公钥字符串
|
||
func FormatPublicKey(publicKey string) (pKey string) {
|
||
var buffer strings.Builder
|
||
// 写入PEM格式头部
|
||
buffer.WriteString("-----BEGIN PUBLIC KEY-----\n")
|
||
|
||
// 定义每行公钥的标准长度（PEM格式要求64字符/行）
|
||
rawLen := 64
|
||
keyLen := len(publicKey)
|
||
// 计算需要拆分的总行数（向上取整）
|
||
raws := keyLen / rawLen
|
||
temp := keyLen % rawLen
|
||
if temp > 0 {
|
||
raws++ // 若有余数则增加一行
|
||
}
|
||
|
||
// 按行拆分并写入公钥内容
|
||
start := 0
|
||
end := start + rawLen
|
||
for i := 0; i < raws; i++ {
|
||
if i == raws-1 {
|
||
// 最后一行取剩余所有字符（处理不足64字符的情况）
|
||
buffer.WriteString(publicKey[start:])
|
||
} else {
|
||
// 非最后行取固定64字符
|
||
buffer.WriteString(publicKey[start:end])
|
||
}
|
||
buffer.WriteByte('\n') // 每行结束添加换行符
|
||
start += rawLen
|
||
end = start + rawLen
|
||
}
|
||
|
||
// 写入PEM格式尾部
|
||
buffer.WriteString("-----END PUBLIC KEY-----\n")
|
||
pKey = buffer.String()
|
||
return
|
||
}
|
||
|
||
// ParseNotifyToBodyMap 将HTTP请求中的表单数据解析为键值对映射
|
||
// 功能：解析请求表单数据，提取单值字段并转换为map[string]interface{}格式
|
||
// 参数 req: 包含表单数据的HTTP请求对象
|
||
// 返回值: 解析后的键值对映射(bm)和可能的错误(err)
|
||
func ParseNotifyToBodyMap(req *http.Request) (bm map[string]interface{}, err error) {
|
||
// 解析请求表单数据，若失败则返回错误
|
||
if err = req.ParseForm(); err != nil {
|
||
return nil, err
|
||
}
|
||
// 获取解析后的表单数据（key为字段名，value为字符串切片形式的字段值）
|
||
var form map[string][]string = req.Form
|
||
// 初始化结果映射，预分配容量（表单字段数+1，预留扩展空间）
|
||
bm = make(map[string]interface{}, len(form)+1)
|
||
// 遍历表单字段，仅保留单值字段（忽略多值字段）
|
||
for k, v := range form {
|
||
if len(v) == 1 {
|
||
bm[k] = v[0]
|
||
}
|
||
}
|
||
return
|
||
}
|
||
|
||
// BuildSignStr 根据传入的g.Map构建签名字符串
|
||
// 规则：对所有非空值的键进行字母排序后，按"key=value&"格式拼接，最后去除末尾的"&"
|
||
// 参数 bm: 包含键值对的g.Map
|
||
// 返回值: 构建好的签名字符串和可能的错误
|
||
func BuildSignStr(bm g.Map) (string, error) {
|
||
var (
|
||
buf strings.Builder
|
||
keyList []string
|
||
)
|
||
// 收集所有键名
|
||
for k := range bm {
|
||
keyList = append(keyList, k)
|
||
}
|
||
// 对键名进行字母排序
|
||
sort.Strings(keyList)
|
||
// 遍历排序后的键，拼接非空值的键值对
|
||
for _, k := range keyList {
|
||
if v := bm[k]; v != "" {
|
||
buf.WriteString(k)
|
||
buf.WriteByte('=')
|
||
buf.WriteString(gconv.String(v))
|
||
buf.WriteByte('&')
|
||
// 去除末尾多余的'&'字符
|
||
// 检查是否有有效的键值对被拼接
|
||
}
|
||
}
|
||
if buf.Len() <= 0 {
|
||
return "", errors.New("length is error")
|
||
}
|
||
return buf.String()[:buf.Len()-1], nil
|
||
}
|