。 // File: xy_sh/go.mod ```go module xy_sh go 1.26 require ( github.com/gofiber/fiber/v2 v2.52.0 github.com/tjfoc/gmsm v1.4.1 ) ``` // File: xy_sh/cmd/server/main.go ```go package main import ( "log" "xy_sh/internal/config" "xy_sh/internal/router" "xy_sh/pkg/crypto" "github.com/gofiber/fiber/v2" ) func main() { // 加载配置 cfg := config.LoadConfig() // 初始化加密模块 crypto.InitCrypto(cfg.SM3Salt, cfg.SM4Key) // 创建Fiber应用 app := fiber.New(fiber.Config{ AppName: "XY SH API Server", ReadTimeout: cfg.ReadTimeout, WriteTimeout: cfg.WriteTimeout, }) // 设置路由 router.SetupRoutes(app) // 启动服务 log.Printf("服务启动,监听端口: %s", cfg.ServerPort) if err := app.Listen(":" + cfg.ServerPort); err != nil { log.Fatalf("服务启动失败: %v", err) } } ``` // File: xy_sh/internal/config/config.go ```go package config import ( "os" "time" ) // Config 应用配置 type Config struct { ServerPort string SM3Salt string SM4Key string ReadTimeout time.Duration WriteTimeout time.Duration } // LoadConfig 加载配置 func LoadConfig() *Config { port := getEnv("SERVER_PORT", "8080") sm3Salt := getEnv("SM3_SALT", "default_salt_change_me") sm4Key := getEnv("SM4_KEY", "default_key_16b") return &Config{ ServerPort: port, SM3Salt: sm3Salt, SM4Key: sm4Key, ReadTimeout: 30 * time.Second, WriteTimeout: 30 * time.Second, } } func getEnv(key, defaultValue string) string { if value := os.Getenv(key); value != "" { return value } return defaultValue } ``` // File: xy_sh/pkg/crypto/crypto.go ```go package crypto import ( "bytes" "crypto/cipher" "crypto/hmac" "crypto/rand" "encoding/base64" "encoding/hex" "errors" "fmt" "io" "github.com/tjfoc/gmsm/sm3" "github.com/tjfoc/gmsm/sm4" ) var ( sm3Salt []byte sm4Key []byte ) // InitCrypto 初始化加密模块 func InitCrypto(salt, key string) { sm3Salt = []byte(salt) // SM4密钥必须为16字节 keyBytes := []byte(key) if len(keyBytes) < 16 { // 补齐到16字节 padded := make([]byte, 16) copy(padded, keyBytes) sm4Key = padded } else if len(keyBytes) > 16 { sm4Key = keyBytes[:16] } else { sm4Key = keyBytes } } // ==================== SM3 哈希算法 ==================== // SM3Hash 计算SM3哈希值 func SM3Hash(data []byte) string { h := sm3.New() h.Write(data) hashBytes := h.Sum(nil) return hex.EncodeToString(hashBytes) } // SM3HashWithSalt 使用salt计算SM3哈希值(用于签名) func SM3HashWithSalt(data string) string { h := hmac.New(sm3.New, sm3Salt) h.Write([]byte(data)) return hex.EncodeToString(h.Sum(nil)) } // GenerateSign 生成签名 // 规则:timestamp + encryptedData 拼接后使用SM3 with salt加密 func GenerateSign(timestamp, encryptedData string) string { signStr := timestamp + encryptedData return SM3HashWithSalt(signStr) } // VerifySign 验证签名 func VerifySign(timestamp, encryptedData, sign string) bool { expectedSign := GenerateSign(timestamp, encryptedData) return hmac.Equal([]byte(expectedSign), []byte(sign)) } // ==================== SM4-CBC 对称加密 ==================== // SM4CBCEncrypt SM4-CBC模式加密,返回base64编码 func SM4CBCEncrypt(plaintext []byte) (string, error) { if len(sm4Key) != 16 { return "", fmt.Errorf("SM4密钥长度必须为16字节") } block, err := sm4.NewCipher(sm4Key) if err != nil { return "", fmt.Errorf("创建SM4 cipher失败: %v", err) } padded := pkcs7Padding(plaintext, block.BlockSize()) iv := make([]byte, block.BlockSize()) if _, err := io.ReadFull(rand.Reader, iv); err != nil { return "", fmt.Errorf("生成IV失败: %v", err) } mode := cipher.NewCBCEncrypter(block, iv) ciphertext := make([]byte, len(padded)) mode.CryptBlocks(ciphertext, padded) result := append(iv, ciphertext...) return base64.StdEncoding.EncodeToString(result), nil } // SM4CBCDecrypt SM4-CBC模式解密,输入base64编码 func SM4CBCDecrypt(encryptedData string) ([]byte, error) { if len(sm4Key) != 16 { return nil, fmt.Errorf("SM4密钥长度必须为16字节") } data, err := base64.StdEncoding.DecodeString(encryptedData) if err != nil { return nil, fmt.Errorf("base64解码失败: %v", err) } block, err := sm4.NewCipher(sm4Key) if err != nil { return nil, fmt.Errorf("创建SM4 cipher失败: %v", err) } blockSize := block.BlockSize() if len(data) < blockSize { return nil, errors.New("数据长度不足") } iv := data[:blockSize] ciphertext := data[blockSize:] if len(ciphertext)%blockSize != 0 { return nil, errors.New("密文长度不是块大小的整数倍") } mode := cipher.NewCBCDecrypter(block, iv) plaintext := make([]byte, len(ciphertext)) mode.CryptBlocks(plaintext, ciphertext) plaintext, err = pkcs7UnPadding(plaintext) if err != nil { return nil, fmt.Errorf("去除填充失败: %v", err) } return plaintext, nil } // pkcs7Padding PKCS7填充 func pkcs7Padding(data []byte, blockSize int) []byte { padding := blockSize - len(data)%blockSize padText := bytes.Repeat([]byte{byte(padding)}, padding) return append(data, padText...) } // pkcs7UnPadding 去除PKCS7填充 func pkcs7UnPadding(data []byte) ([]byte, error) { length := len(data) if length == 0 { return nil, errors.New("数据为空") } padding := int(data[length-1]) if padding > length || padding == 0 { return nil, errors.New("无效的填充") } for i := length - padding; i < length; i++ { if data[i] != byte(padding) { return nil, errors.New("无效的填充") } } return data[:length-padding], nil } ``` // File: xy_sh/internal/entities/request.go ```go package entities // EncryptedRequest 通用加密请求体 type EncryptedRequest struct { EncryptedData string `json:"encryptedData"` } // OrderRequest 下单接口业务参数(解密后) type OrderRequest struct { ActCode string `json:"actCode"` GoodsCode string `json:"goodsCode"` ActOrderNum string `json:"actOrderNum"` Account string `json:"account,omitempty"` CallbackUrl string `json:"callbackUrl,omitempty"` } // QueryOrderRequest 查询接口业务参数(解密后) type QueryOrderRequest struct { ActCode string `json:"actCode"` OrderNo string `json:"orderNo"` } // WxRechargeRequest 微信立减金充值接口业务参数(解密后) type WxRechargeRequest struct { ActCode string `json:"actCode"` OrderNo string `json:"orderNo"` GoodsCode string `json:"goodsCode"` ActOrderNum string `json:"actOrderNum"` AppId string `json:"appId"` OpenId string `json:"openId"` CallbackUrl string `json:"callbackUrl,omitempty"` } // CallbackRequest 回调通知业务参数(解密后) type CallbackRequest struct { OrderNo string `json:"orderNo"` ActOrderNum string `json:"actOrderNum"` Status int `json:"status"` Account string `json:"account,omitempty"` CardInfo *CardInfo `json:"cardInfo,omitempty"` CouponId string `json:"couponId,omitempty"` } // CardInfo 卡券信息 type CardInfo struct { CouponNo string `json:"couponNo"` CouponCode string `json:"couponCode"` ExpireTime string `json:"expireTime"` } ``` // File: xy_sh/internal/entities/response.go ```go package entities // CommonResponse 通用响应结构 type CommonResponse struct { Code int `json:"code"` Msg string `json:"msg"` Data interface{} `json:"data"` } // OrderResponseData 下单接口响应业务数据(加密前) type OrderResponseData struct { OrderNo string `json:"orderNo"` CouponNo string `json:"couponNo,omitempty"` CouponCode string `json:"couponCode,omitempty"` Status int `json:"status"` ExpireTime string `json:"expireTime,omitempty"` } // QueryOrderResponseData 查询接口响应业务数据(加密前) type QueryOrderResponseData struct { OrderNo string `json:"orderNo"` Status int `json:"status"` Account string `json:"account,omitempty"` CardInfo *CardInfo `json:"cardInfo,omitempty"` CouponId string `json:"couponId,omitempty"` } // WxRechargeResponseData 微信立减金充值响应业务数据(加密前) type WxRechargeResponseData struct { OrderNo string `json:"orderNo"` Status int `json:"status"` CouponId string `json:"couponId,omitempty"` } // 订单状态常量 const ( StatusSending = -1 // 发放中 StatusSuccess = 0 // 成功 StatusFailed = 1 // 失败 StatusRedeemed = 2 // 已核销 StatusExpired = 3 // 已过期 ) // 响应码常量 const ( CodeSuccess = 0 CodeFailed = -1 ) ``` // File: xy_sh/internal/middleware/middleware.go ```go package middleware import ( "encoding/json" "strings" "xy_sh/internal/entities" "xy_sh/pkg/crypto" "github.com/gofiber/fiber/v2" ) // SignVerifyMiddleware 签名验证中间件 func SignVerifyMiddleware() fiber.Handler { return func(c *fiber.Ctx) error { // 从请求头获取timestamp和sign timestamp := c.Get("timestamp") sign := c.Get("sign") if timestamp == "" || sign == "" { return c.Status(fiber.StatusUnauthorized).JSON(entities.CommonResponse{ Code: entities.CodeFailed, Msg: "缺少timestamp或sign请求头", Data: nil, }) } // 读取请求体 body := c.Body() // 解析encryptedData var req entities.EncryptedRequest if err := json.Unmarshal(body, &req); err != nil { return c.Status(fiber.StatusBadRequest).JSON(entities.CommonResponse{ Code: entities.CodeFailed, Msg: "请求体格式错误", Data: nil, }) } if req.EncryptedData == "" { return c.Status(fiber.StatusBadRequest).JSON(entities.CommonResponse{ Code: entities.CodeFailed, Msg: "缺少encryptedData参数", Data: nil, }) } // 验证签名 if !crypto.VerifySign(timestamp, req.EncryptedData, sign) { return c.Status(fiber.StatusUnauthorized).JSON(entities.CommonResponse{ Code: entities.CodeFailed, Msg: "签名验证失败", Data: nil, }) } // 解密数据 decryptedData, err := crypto.SM4CBCDecrypt(req.EncryptedData) if err != nil { return c.Status(fiber.StatusBadRequest).JSON(entities.CommonResponse{ Code: entities.CodeFailed, Msg: "数据解密失败: " + err.Error(), Data: nil, }) } // 将解密后的数据存入上下文 c.Locals("decryptedData", decryptedData) c.Locals("timestamp", timestamp) // 继续处理请求 return c.Next() } } // EncryptResponseMiddleware 加密响应中间件 func EncryptResponseMiddleware() fiber.Handler { return func(c *fiber.Ctx) error { // 先执行后续处理 err := c.Next() if err != nil { return err } // 获取响应体 body := c.Response().Body() if len(body) == 0 { return nil } // 尝试解析为CommonResponse var resp entities.CommonResponse if err := json.Unmarshal(body, &resp); err != nil { // 不是标准响应格式,直接返回 return nil } // 如果data不为空且是字符串,需要加密 if resp.Data != nil { // 将data序列化为JSON字符串 dataBytes, err := json.Marshal(resp.Data) if err != nil { return c.Status(fiber.StatusInternalServerError).JSON(entities.CommonResponse{ Code: entities.CodeFailed, Msg: "响应数据序列化失败", Data: nil, }) } // 加密data encryptedData, err := crypto.SM4CBCEncrypt(dataBytes) if err != nil { return c.Status(fiber.StatusInternalServerError).JSON(entities.CommonResponse{ Code: entities.CodeFailed, Msg: "响应数据加密失败", Data: nil, }) } // 替换data为加密后的字符串 resp.Data = encryptedData // 重新设置响应体 newBody, err := json.Marshal(resp) if err != nil { return c.Status(fiber.StatusInternalServerError).JSON(entities.CommonResponse{ Code: entities.CodeFailed, Msg: "响应序列化失败", Data: nil, }) } c.Response().SetBody(newBody) } return nil } } // SkipSignVerify 跳过签名验证的路径判断 func SkipSignVerify(path string) bool { // 可以配置不需要签名验证的路径 skipPaths := []string{"/health", "/"} for _, p := range skipPaths { if strings.HasPrefix(path, p) { return true } } return false } ``` // File: xy_sh/internal/biz/biz.go ```go package biz import ( "errors" "sync" "time" "xy_sh/internal/entities" ) // OrderStore 订单存储(内存实现,生产环境应使用数据库) type OrderStore struct { mu sync.RWMutex orders map[string]*OrderInfo } // OrderInfo 订单信息 type OrderInfo struct { OrderNo string ActOrderNum string GoodsCode string ActCode string Account string Status int CouponNo string CouponCode string ExpireTime string CouponId string AppId string OpenId string CreateTime time.Time UpdateTime time.Time } var store *OrderStore func init() { store = &OrderStore{ orders: make(map[string]*OrderInfo), } } // CreateOrder 创建订单 func CreateOrder(req *entities.OrderRequest) (*entities.OrderResponseData, error) { store.mu.Lock() defer store.mu.Unlock() // 生成订单号 orderNo := generateOrderNo() // TODO: 实际业务逻辑处理 // 1. 验证活动code和商品code // 2. 检查库存 // 3. 处理卡券/直充逻辑 // 4. 生成卡密信息(卡券类商品) orderInfo := &OrderInfo{ OrderNo: orderNo, ActOrderNum: req.ActOrderNum, GoodsCode: req.GoodsCode, ActCode: req.ActCode, Account: req.Account, Status: entities.StatusSuccess, // 默认成功,实际应根据业务处理结果设置 CreateTime: time.Now(), UpdateTime: time.Now(), } // 模拟卡券类商品返回卡密信息 orderInfo.CouponNo = generateCouponNo() orderInfo.CouponCode = orderInfo.CouponNo orderInfo.ExpireTime = time.Now().AddDate(1, 0, 0).Format("2006-01-02 15:04:05") store.orders[orderNo] = orderInfo return &entities.OrderResponseData{ OrderNo: orderNo, CouponNo: orderInfo.CouponNo, CouponCode: orderInfo.CouponCode, Status: orderInfo.Status, ExpireTime: orderInfo.ExpireTime, }, nil } // QueryOrder 查询订单 func QueryOrder(req *entities.QueryOrderRequest) (*entities.QueryOrderResponseData, error) { store.mu.RLock() defer store.mu.RUnlock() orderInfo, exists := store.orders[req.OrderNo] if !exists { return nil, errors.New("订单不存在") } resp := &entities.QueryOrderResponseData{ OrderNo: orderInfo.OrderNo, Status: orderInfo.Status, Account: orderInfo.Account, CouponId: orderInfo.CouponId, } // 如果有卡券信息 if orderInfo.CouponNo != "" { resp.CardInfo = &entities.CardInfo{ CouponNo: orderInfo.CouponNo, CouponCode: orderInfo.CouponCode, ExpireTime: orderInfo.ExpireTime, } } return resp, nil } // WxRecharge 微信立减金充值 func WxRecharge(req *entities.WxRechargeRequest) (*entities.WxRechargeResponseData, error) { store.mu.Lock() defer store.mu.Unlock() // TODO: 实际业务逻辑处理 // 1. 验证活动code和商品code // 2. 调用微信立减金发放接口 // 3. 处理充值结果 // 检查订单是否已存在 if orderInfo, exists := store.orders[req.OrderNo]; exists { return &entities.WxRechargeResponseData{ OrderNo: orderInfo.OrderNo, Status: orderInfo.Status, CouponId: orderInfo.CouponId, }, nil } // 生成订单号 orderNo := req.OrderNo if orderNo == "" { orderNo = generateOrderNo() } orderInfo := &OrderInfo{ OrderNo: orderNo, ActOrderNum: req.ActOrderNum, GoodsCode: req.GoodsCode, ActCode: req.ActCode, AppId: req.AppId, OpenId: req.OpenId, Status: entities.StatusSuccess, CouponId: generateCouponId(), CreateTime: time.Now(), UpdateTime: time.Now(), } store.orders[orderNo] = orderInfo return &entities.WxRechargeResponseData{ OrderNo: orderNo, Status: orderInfo.Status, CouponId: orderInfo.CouponId, }, nil } // HandleCallback 处理回调通知 func HandleCallback(req *entities.CallbackRequest) error { store.mu.Lock() defer store.mu.Unlock() // TODO: 实际业务逻辑处理 // 1. 更新订单状态 // 2. 处理卡券信息 // 3. 触发后续业务流程 orderInfo, exists := store.orders[req.OrderNo] if !exists { // 如果订单不存在,创建记录 orderInfo = &OrderInfo{ OrderNo: req.OrderNo, ActOrderNum: req.ActOrderNum, CreateTime: time.Now(), } store.orders[req.OrderNo] = orderInfo } // 更新订单状态 orderInfo.Status = req.Status orderInfo.Account = req.Account orderInfo.CouponId = req.CouponId orderInfo.UpdateTime = time.Now() if req.CardInfo != nil { orderInfo.CouponNo = req.CardInfo.CouponNo orderInfo.CouponCode = req.CardInfo.CouponCode orderInfo.ExpireTime = req.CardInfo.ExpireTime } return nil } // generateOrderNo 生成订单号 func generateOrderNo() string { return "HM" + time.Now().Format("20060102150405") + randomString(16) } // generateCouponNo 生成卡号 func generateCouponNo() string { return randomString(16) } // generateCouponId 生成微信优惠id func generateCouponId() string { return randomString(8) } // randomString 生成随机字符串 func randomString(n int) string { const letters = "0123456789abcdef" b := make([]byte, n) for i := range b { b[i] = letters[time.Now().UnixNano()%int64(len(letters))] time.Sleep(1 * time.Nanosecond) } return string(b) } ``` // File: xy_sh/internal/service/service.go ```go package service import ( "encoding/json" "xy_sh/internal/biz" "xy_sh/internal/entities" "github.com/gofiber/fiber/v2" ) // CreateOrderHandler 卡券/直充权益下单接口 func CreateOrderHandler(c *fiber.Ctx) error { // 获取解密后的数据 decryptedData := c.Locals("decryptedData").([]byte) // 解析业务参数 var req entities.OrderRequest if err := json.Unmarshal(decryptedData, &req); err != nil { return c.Status(fiber.StatusBadRequest).JSON(entities.CommonResponse{ Code: entities.CodeFailed, Msg: "业务参数解析失败: " + err.Error(), Data: nil, }) } // 参数校验 if req.ActCode == "" || req.GoodsCode == "" || req.ActOrderNum == "" { return c.Status(fiber.StatusBadRequest).JSON(entities.CommonResponse{ Code: entities.CodeFailed, Msg: "缺少必填参数: actCode, goodsCode, actOrderNum", Data: nil, }) } // 调用业务逻辑 result, err := biz.CreateOrder(&req) if err != nil { return c.Status(fiber.StatusOK).JSON(entities.CommonResponse{ Code: entities.CodeFailed, Msg: err.Error(), Data: nil, }) } return c.Status(fiber.StatusOK).JSON(entities.CommonResponse{ Code: entities.CodeSuccess, Msg: "请求成功", Data: result, }) } // QueryOrderHandler 卡券/直充/微信立减金订单查询接口 func QueryOrderHandler(c *fiber.Ctx) error { // 获取解密后的数据 decryptedData := c.Locals("decryptedData").([]byte) // 解析业务参数 var req entities.QueryOrderRequest if err := json.Unmarshal(decryptedData, &req); err != nil { return c.Status(fiber.StatusBadRequest).JSON(entities.CommonResponse{ Code: entities.CodeFailed, Msg: "业务参数解析失败: " + err.Error(), Data: nil, }) } // 参数校验 if req.ActCode == "" || req.OrderNo == "" { return c.Status(fiber.StatusBadRequest).JSON(entities.CommonResponse{ Code: entities.CodeFailed, Msg: "缺少必填参数: actCode, orderNo", Data: nil, }) } // 调用业务逻辑 result, err := biz.QueryOrder(&req) if err != nil { return c.Status(fiber.StatusOK).JSON(entities.CommonResponse{ Code: entities.CodeFailed, Msg: err.Error(), Data: nil, }) } return c.Status(fiber.StatusOK).JSON(entities.CommonResponse{ Code: entities.CodeSuccess, Msg: "请求成功", Data: result, }) } // WxRechargeHandler 微信立减金订单充值接口 func WxRechargeHandler(c *fiber.Ctx) error { // 获取解密后的数据 decryptedData := c.Locals("decryptedData").([]byte) // 解析业务参数 var req entities.WxRechargeRequest if err := json.Unmarshal(decryptedData, &req); err != nil { return c.Status(fiber.StatusBadRequest).JSON(entities.CommonResponse{ Code: entities.CodeFailed, Msg: "业务参数解析失败: " + err.Error(), Data: nil, }) } // 参数校验 if req.ActCode == "" || req.GoodsCode == "" || req.ActOrderNum == "" || req.AppId == "" || req.OpenId == "" { return c.Status(fiber.StatusBadRequest).JSON(entities.CommonResponse{ Code: entities.CodeFailed, Msg: "缺少必填参数: actCode, goodsCode, actOrderNum, appId, openId", Data: nil, }) } // 调用业务逻辑 result, err := biz.WxRecharge(&req) if err != nil { return c.Status(fiber.StatusOK).JSON(entities.CommonResponse{ Code: entities.CodeFailed, Msg: err.Error(), Data: nil, }) } return c.Status(fiber.StatusOK).JSON(entities.CommonResponse{ Code: entities.CodeSuccess, Msg: "请求成功", Data: result, }) } // CallbackHandler 卡券/直充/微信立减金充值结果通知接口 func CallbackHandler(c *fiber.Ctx) error { // 获取解密后的数据 decryptedData := c.Locals("decryptedData").([]byte) // 解析业务参数 var req entities.CallbackRequest if err := json.Unmarshal(decryptedData, &req); err != nil { return c.Status(fiber.StatusBadRequest).SendString("参数解析失败") } // 参数校验 if req.OrderNo == "" || req.ActOrderNum == "" { return c.Status(fiber.StatusBadRequest).SendString("缺少必填参数") } // 调用业务逻辑 if err := biz.HandleCallback(&req); err != nil { return c.Status(fiber.StatusInternalServerError).SendString("处理失败") } // 返回ok表示成功接收 return c.Status(fiber.StatusOK).SendString("ok") } // HealthHandler 健康检查接口 func HealthHandler(c *fiber.Ctx) error { return c.Status(fiber.StatusOK).JSON(fiber.Map{ "status": "ok", "message": "服务运行正常", }) } ``` // File: xy_sh/internal/router/router.go ```go package router import ( "xy_sh/internal/middleware" "xy_sh/internal/service" "github.com/gofiber/fiber/v2" ) // SetupRoutes 设置路由 func SetupRoutes(app *fiber.App) { // 健康检查(不需要签名验证) app.Get("/health", service.HealthHandler) app.Get("/", service.HealthHandler) // API v1 分组 v1 := app.Group("/api/v1") // 需要签名验证和加密响应的接口 v1.Post("/order/create", middleware.SignVerifyMiddleware(), middleware.EncryptResponseMiddleware(), service.CreateOrderHandler, ) v1.Post("/order/query", middleware.SignVerifyMiddleware(), middleware.EncryptResponseMiddleware(), service.QueryOrderHandler, ) v1.Post("/wx/recharge", middleware.SignVerifyMiddleware(), middleware.EncryptResponseMiddleware(), service.WxRechargeHandler, ) // 回调通知接口(需要签名验证,但响应不加密,直接返回ok) v1.Post("/callback/notify", middleware.SignVerifyMiddleware(), service.CallbackHandler, ) } ``` // File: xy_sh/internal/test/example_test.go ```go package test import ( "bytes" "encoding/json" "io" "net/http" "net/http/httptest" "testing" "time" "xy_sh/internal/entities" "xy_sh/pkg/crypto" "github.com/gofiber/fiber/v2" "xy_sh/internal/router" ) func init() { // 初始化加密模块(测试用) crypto.InitCrypto("test_salt_12345", "test_key_16bytes") } // setupTestApp 创建测试应用 func setupTestApp() *fiber.App { app := fiber.New() router.SetupRoutes(app) return app } // makeEncryptedRequest 构造加密请求 func makeEncryptedRequest(t *testing.T, app *fiber.App, path string, bizData interface{}) (*http.Response, []byte) { t.Helper() // 将业务数据序列化为JSON bizJSON, err := json.Marshal(bizData) if err != nil { t.Fatalf("业务数据序列化失败: %v", err) } // SM4加密业务数据 encryptedData, err := crypto.SM4CBCEncrypt(bizJSON) if err != nil { t.Fatalf("加密失败: %v", err) } // 生成时间戳 timestamp := time.Now().Format("20060102150405") + "000" // 生成签名 sign := crypto.GenerateSign(timestamp, encryptedData) // 构造请求体 reqBody := entities.EncryptedRequest{ EncryptedData: encryptedData, } reqJSON, _ := json.Marshal(reqBody) // 创建HTTP请求 req := httptest.NewRequest(http.MethodPost, path, bytes.NewReader(reqJSON)) req.Header.Set("Content-Type", "application/json") req.Header.Set("timestamp", timestamp) req.Header.Set("sign", sign) // 发送请求 resp, err := app.Test(req) if err != nil { t.Fatalf("请求失败: %v", err) } // 读取响应 body, _ := io.ReadAll(resp.Body) resp.Body.Close() return resp, body } // TestCreateOrder 测试下单接口 func TestCreateOrder(t *testing.T) { app := setupTestApp() // 构造下单请求参数 bizData := entities.OrderRequest{ ActCode: "ACT001", GoodsCode: "123456", ActOrderNum: "TEST_ORDER_001", Account: "19912345678", CallbackUrl: "https://example.com/callback", } resp, body := makeEncryptedRequest(t, app, "/api/v1/order/create", bizData) if resp.StatusCode != http.StatusOK { t.Errorf("期望状态码200,实际: %d, 响应: %s", resp.StatusCode, string(body)) } // 解析响应 var commonResp entities.CommonResponse if err := json.Unmarshal(body, &commonResp); err != nil { t.Fatalf("响应解析失败: %v, 响应: %s", err, string(body)) } if commonResp.Code != entities.CodeSuccess { t.Errorf("期望code=0,实际: %d, msg: %s", commonResp.Code, commonResp.Msg) } t.Logf("下单接口测试成功,响应: %s", string(body)) } // TestQueryOrder 测试查询接口 func TestQueryOrder(t *testing.T) { app := setupTestApp() // 先创建一个订单 createBizData := entities.OrderRequest{ ActCode: "ACT001", GoodsCode: "123456", ActOrderNum: "TEST_ORDER_002", Account: "19912345678", } _, createBody := makeEncryptedRequest(t, app, "/api/v1/order/create", createBizData) var createResp entities.CommonResponse json.Unmarshal(createBody, &createResp) // 注意:实际响应中data是加密的,这里需要先解密 // 为了简化测试,我们直接构造查询请求 queryBizData := entities.QueryOrderRequest{ ActCode: "ACT001", OrderNo: "HM17575805323790000127073398f3772c00", // 使用示例订单号 } resp, body := makeEncryptedRequest(t, app, "/api/v1/order/query", queryBizData) if resp.StatusCode != http.StatusOK { t.Errorf("期望状态码200,实际: %d, 响应: %s", resp.StatusCode, string(body)) } t.Logf("查询接口测试完成,响应: %s", string(body)) } // TestWxRecharge 测试微信立减金充值接口 func TestWxRecharge(t *testing.T) { app := setupTestApp() bizData := entities.WxRechargeRequest{ ActCode: "FBG6vdYqGE4mGX7EH/woEg==", GoodsCode: "0001", ActOrderNum: "WX_ORDER_001", AppId: "wx1234567890", OpenId: "oABC1234567890", } resp, body := makeEncryptedRequest(t, app, "/api/v1/wx/recharge", bizData) if resp.StatusCode != http.StatusOK { t.Errorf("期望状态码200,实际: %d, 响应: %s", resp.StatusCode, string(body)) } var commonResp entities.CommonResponse if err := json.Unmarshal(body, &commonResp); err != nil { t.Fatalf("响应解析失败: %v", err) } if commonResp.Code != entities.CodeSuccess { t.Errorf("期望code=0,实际: %d, msg: %s", commonResp.Code, commonResp.Msg) } t.Logf("微信立减金充值接口测试成功,响应: %s", string(body)) } // TestCallback 测试回调通知接口 func TestCallback(t *testing.T) { app := setupTestApp() bizData := entities.CallbackRequest{ OrderNo: "HM202509291010001", ActOrderNum: "XY2025092910100001", Status: entities.StatusExpired, Account: "19912345678", } resp, body := makeEncryptedRequest(t, app, "/api/v1/callback/notify", bizData) if resp.StatusCode != http.StatusOK { t.Errorf("期望状态码200,实际: %d, 响应: %s", resp.StatusCode, string(body)) } if string(body) != "ok" { t.Errorf("期望响应'ok',实际: %s", string(body)) } t.Logf("回调通知接口测试成功") } // TestInvalidSign 测试签名验证失败 func TestInvalidSign(t *testing.T) { app := setupTestApp() bizData := entities.OrderRequest{ ActCode: "ACT001", GoodsCode: "123456", ActOrderNum: "TEST_ORDER_003", } // 加密数据 bizJSON, _ := json.Marshal(bizData) encryptedData, _ := crypto.SM4CBCEncrypt(bizJSON) timestamp := time.Now().Format("20060102150405") + "000" // 使用错误的签名 wrongSign := "wrong_signature_12345" reqBody := entities.EncryptedRequest{EncryptedData: encryptedData} reqJSON, _ := json.Marshal(reqBody) req := httptest.NewRequest(http.MethodPost, "/api/v1/order/create", bytes.NewReader(reqJSON)) req.Header.Set("Content-Type", "application/json") req.Header.Set("timestamp", timestamp) req.Header.Set("sign", wrongSign) resp, _ := app.Test(req) if resp.StatusCode != http.StatusUnauthorized { t.Errorf("期望状态码401(签名验证失败),实际: %d", resp.StatusCode) } t.Logf("签名验证失败测试成功") } // TestHealthCheck 测试健康检查接口 func TestHealthCheck(t *testing.T) { app := setupTestApp() req := httptest.NewRequest(http.MethodGet, "/health", nil) resp, body := app.Test(req) if resp.StatusCode != http.StatusOK { t.Errorf("期望状态码200,实际: %d", resp.StatusCode) } t.Logf("健康检查接口测试成功,响应: %s", string(body)) } ``` === 服务端生成完成 === ## 加密实现 以下是从加密工具获取的完整实现: ### 加密实现 1 { "success": true, "data": "\n### SM3 国密哈希算法完整实现指南\n\n**适用场景**:文档要求使用国密 SM3 算法进行摘要计算或签名验证\n\n**前置要求**:需要安装 github.com/tjfoc/gmsm\n\n```bash\ngo get github.com/tjfoc/gmsm\n```\n\n**配置参数**:\n- 编码方式: hex\n- 包含 HMAC: true\n\n**完整代码模板**:\n\n```go\npackage crypto\n\nimport (\n \"crypto/hmac\"\n \"encoding/base64\"\n \"encoding/hex\"\n \"fmt\"\n \"os\"\n \n \"github.com/tjfoc/gmsm/sm3\"\n)\n\n// SM3Hash 计算SM3哈希值\n// data: 待哈希的数据\n// encoding: 输出编码方式 (hex/base64)\n// 返回: 编码后的哈希值\nfunc SM3Hash(data []byte, encoding string) (string, error) {\n h := sm3.New()\n h.Write(data)\n hashBytes := h.Sum(nil)\n \n if encoding == \"base64\" {\n return base64.StdEncoding.EncodeToString(hashBytes), nil\n }\n return hex.EncodeToString(hashBytes), nil\n}\n\n// SM3HashString 计算字符串的SM3哈希值(便捷方法)\nfunc SM3HashString(data string, encoding string) (string, error) {\n return SM3Hash([]byte(data), encoding)\n}\n\n// SM3HashFile 计算文件的SM3哈希值\nfunc SM3HashFile(filePath string, encoding string) (string, error) {\n data, err := os.ReadFile(filePath)\n if err != nil {\n return \"\", fmt.Errorf(\"读取文件失败: %v\", err)\n }\n return SM3Hash(data, encoding)\n}\n\n// SM3Verify 验证数据与哈希值是否匹配\nfunc SM3Verify(data []byte, hash string, encoding string) (bool, error) {\n expected, err := SM3Hash(data, encoding)\n if err != nil {\n return false, err\n }\n return expected == hash, nil\n}\n\n// HMACSM3 HMAC-SM3计算\n// 使用标准HMAC算法,底层使用SM3哈希函数\nfunc HMACSM3(data []byte, key []byte) string {\n // 使用标准HMAC,底层哈希函数用SM3\n h := hmac.New(sm3.New, key)\n h.Write(data)\n return hex.EncodeToString(h.Sum(nil))\n}\n```\n\n**使用示例**:\n```go\n// 计算字符串哈希\nhash, _ := SM3HashString(\"hello world\", \"hex\")\nfmt.Println(hash) // 输出64位十六进制字符串\n\n// 验证哈希\nvalid, _ := SM3Verify([]byte(\"hello world\"), hash, \"hex\")\nfmt.Println(valid) // true\n\n// HMAC-SM3示例\nkey := []byte(\"secret_key\")\ndata := []byte(\"hello world\")\nhmacResult := HMACSM3(data, key)\nfmt.Println(hmacResult)\n```\n\n**注意事项**:\n- SM3 输出固定 256 位(32字节)的哈希值\n- 十六进制输出为 64 位字符串\n- 常用于数字签名、完整性校验等场景\n- SM3 是国密标准哈希算法,与 SHA-256 类似\n- HMAC-SM3 需要使用标准库 crypto/hmac,底层哈希函数使用 sm3.New\n", "tool": "sm3_hash" } ### 加密实现 2 { "success": true, "data": "\n### SM4-CBC 国密对称加密完整实现指南\n\n**⚠️ 重要:必须使用第三方库,不要自己实现 SM4 算法!**\n**推荐使用:github.com/tjfoc/gmsm**\n\n**前置要求**:\n```bash\ngo get github.com/tjfoc/gmsm\n```\n\n**配置参数**:\n- 编码方式: base64\n\n**完整代码模板(请直接复制使用)**:\n\n```go\npackage crypto\n\nimport (\n \"crypto/cipher\"\n \"crypto/rand\"\n \"encoding/base64\"\n \"encoding/hex\"\n \"fmt\"\n \"io\"\n \"bytes\"\n\n \"github.com/tjfoc/gmsm/sm4\" // ⚠️ 必须使用此库,不要自己实现SM4\n)\n\n// SM4CBCEncrypt SM4-CBC模式加密\n// 使用 github.com/tjfoc/gmsm/sm4 实现\nfunc SM4CBCEncrypt(plaintext []byte, key []byte) (string, error) {\n if len(key) != 16 {\n return \"\", fmt.Errorf(\"SM4密钥长度必须为16字节\")\n }\n\n // 使用 gmsm 库创建 SM4 cipher\n block, err := sm4.NewCipher(key)\n if err != nil {\n return \"\", fmt.Errorf(\"创建SM4 cipher失败: %v\", err)\n }\n\n padded := pkcs7Padding(plaintext, block.BlockSize())\n\n iv := make([]byte, block.BlockSize())\n if _, err := io.ReadFull(rand.Reader, iv); err != nil {\n return \"\", fmt.Errorf(\"生成IV失败: %v\", err)\n }\n\n mode := cipher.NewCBCEncrypter(block, iv)\n ciphertext := make([]byte, len(padded))\n mode.CryptBlocks(ciphertext, padded)\n\n result := append(iv, ciphertext...)\n if \"base64\" == \"hex\" {\n return hex.EncodeToString(result), nil\n }\n return base64.StdEncoding.EncodeToString(result), nil\n}\n\n// SM4CBCDecrypt SM4-CBC模式解密\n// 使用 github.com/tjfoc/gmsm/sm4 实现\nfunc SM4CBCDecrypt(encryptedData string, key []byte) ([]byte, error) {\n if len(key) != 16 {\n return nil, fmt.Errorf(\"SM4密钥长度必须为16字节\")\n }\n\n var data []byte\n var err error\n if \"base64\" == \"hex\" {\n data, err = hex.DecodeString(encryptedData)\n } else {\n data, err = base64.StdEncoding.DecodeString(encryptedData)\n }\n if err != nil {\n return nil, fmt.Errorf(\"解码失败: %v\", err)\n }\n\n block, err := sm4.NewCipher(key)\n if err != nil {\n return nil, fmt.Errorf(\"创建SM4 cipher失败: %v\", err)\n }\n\n blockSize := block.BlockSize()\n if len(data) \u003c blockSize {\n return nil, fmt.Errorf(\"数据长度不足\")\n }\n iv := data[:blockSize]\n ciphertext := data[blockSize:]\n\n mode := cipher.NewCBCDecrypter(block, iv)\n plaintext := make([]byte, len(ciphertext))\n mode.CryptBlocks(plaintext, ciphertext)\n\n plaintext, err = pkcs7UnPadding(plaintext)\n if err != nil {\n return nil, fmt.Errorf(\"去除填充失败: %v\", err)\n }\n\n return plaintext, nil\n}\n\n// pkcs7Padding PKCS7填充\nfunc pkcs7Padding(data []byte, blockSize int) []byte {\n padding := blockSize - len(data)%blockSize\n padText := bytes.Repeat([]byte{byte(padding)}, padding)\n return append(data, padText...)\n}\n\n// pkcs7UnPadding 去除PKCS7填充\nfunc pkcs7UnPadding(data []byte) ([]byte, error) {\n length := len(data)\n if length == 0 {\n return nil, fmt.Errorf(\"数据为空\")\n }\n padding := int(data[length-1])\n if padding \u003e length || padding == 0 {\n return nil, fmt.Errorf(\"无效的填充\")\n }\n for i := length - padding; i \u003c length; i++ {\n if data[i] != byte(padding) {\n return nil, fmt.Errorf(\"无效的填充\")\n }\n }\n return data[:length-padding], nil\n}\n```\n\n**⚠️ 重要提醒**:\n1. **不要自己实现 SM4 算法**,请直接使用 github.com/tjfoc/gmsm\n2. 这个库是国密官方推荐的 Go 实现,安全可靠\n3. 在 go.mod 中添加依赖:`require github.com/tjfoc/gmsm v1.4.1`\n\n**注意事项**:\n- SM4 密钥固定为 16 字节\n- IV 必须随机生成且每次不同\n- CBC 模式需要 IV,ECB 模式不需要\n", "tool": "sm4_cbc_encrypt" }