xy_sh-20260724171749/generate.md at 26404d69c70d9a71675d181d6b484a44edebb049

32 KiB

Raw Blame History

。

现在开始生成所有文件。

// File: xy_sh/go.mod

module xy_sh

go 1.26

require (
	github.com/gofiber/fiber/v2 v2.52.6
	github.com/tjfoc/gmsm v1.4.1
)

require (
	github.com/andybalholm/brotli v1.1.0 // indirect
	github.com/google/uuid v1.6.0 // indirect
	github.com/klauspost/compress v1.17.9 // indirect
	github.com/mattn/go-colorable v0.1.13 // indirect
	github.com/mattn/go-isatty v0.0.20 // indirect
	github.com/russross/blackfriday/v2 v2.1.0 // indirect
	github.com/valyala/bytebufferpool v1.0.0 // indirect
	github.com/valyala/fasthttp v1.52.0 // indirect
	github.com/valyala/tcplisten v1.0.0 // indirect
	golang.org/x/sys v0.28.0 // indirect
)

// File: xy_sh/cmd/server/main.go

package main

import (
	"log"
	"xy_sh/internal/config"
	"xy_sh/internal/router"
)

func main() {
	cfg := config.LoadConfig()

	app := router.SetupRouter(cfg)

	log.Printf("服务启动，监听端口: %s", cfg.ServerPort)
	if err := app.Listen(":" + cfg.ServerPort); err != nil {
		log.Fatalf("服务启动失败: %v", err)
	}
}

// File: xy_sh/internal/config/config.go

package config

import "os"

// Config 应用配置
type Config struct {
	ServerPort string // 服务监听端口
	SM4Key     string // SM4 加密密钥
	SM3Salt    string // SM3 盐值
}

// LoadConfig 加载配置
func LoadConfig() *Config {
	return &Config{
		ServerPort: getEnv("SERVER_PORT", "8080"),
		SM4Key:     getEnv("SM4_KEY", ""),
		SM3Salt:    getEnv("SM3_SALT", ""),
	}
}

func getEnv(key, defaultVal string) string {
	if val := os.Getenv(key); val != "" {
		return val
	}
	return defaultVal
}

// File: xy_sh/internal/entities/request.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"`
}

// QueryRequest 订单查询业务参数
type QueryRequest struct {
	ActCode string `json:"actCode"`
	OrderNo string `json:"orderNo"`
}

// WechatRechargeRequest 微信立减金充值业务参数
type WechatRechargeRequest 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

package entities

// ApiResponse 通用API响应
type ApiResponse struct {
	Code int         `json:"code"`
	Msg  string      `json:"msg"`
	Data interface{} `json:"data"`
}

// OrderResponse 下单响应业务数据
type OrderResponse struct {
	OrderNo    string `json:"orderNo"`
	CouponNo   string `json:"couponNo,omitempty"`
	CouponCode string `json:"couponCode,omitempty"`
	Status     int    `json:"status"`
	ExpireTime string `json:"expireTime,omitempty"`
}

// QueryResponse 查询响应业务数据
type QueryResponse struct {
	OrderNo  string     `json:"orderNo"`
	Status   int        `json:"status"`
	Account  string     `json:"account,omitempty"`
	CardInfo *CardInfo  `json:"cardInfo,omitempty"`
	CouponID string     `json:"couponId,omitempty"`
}

// WechatRechargeResponse 微信立减金充值响应业务数据
type WechatRechargeResponse struct {
	OrderNo  string `json:"orderNo"`
	Status   int    `json:"status"`
	CouponID string `json:"couponId,omitempty"`
}

// CallbackResponse 回调通知响应
type CallbackResponse struct {
	Code int         `json:"code"`
	Msg  string      `json:"msg"`
	Data interface{} `json:"data"`
}

// File: xy_sh/internal/middleware/middleware.go

package middleware

import (
	"encoding/json"
	"strings"

	"github.com/gofiber/fiber/v2"
	"xy_sh/internal/config"
	"xy_sh/pkg/crypto"
)

// VerifySign 验证签名中间件
func VerifySign(cfg *config.Config) fiber.Handler {
	return func(c *fiber.Ctx) error {
		timestamp := c.Get("timestamp")
		sign := c.Get("sign")

		if timestamp == "" || sign == "" {
			return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
				"code": -1,
				"msg":  "缺少timestamp或sign请求头",
			})
		}

		// 解析请求体获取 encryptedData
		var req struct {
			EncryptedData string `json:"encryptedData"`
		}
		if err := c.BodyParser(&req); err != nil {
			return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
				"code": -1,
				"msg":  "请求体解析失败",
			})
		}

		// 签名规则：sm3WithSalt(timestamp + encryptedData)
		rawStr := timestamp + req.EncryptedData
		expectedSign, err := crypto.SM3WithSalt([]byte(rawStr), []byte(cfg.SM3Salt))
		if err != nil {
			return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
				"code": -1,
				"msg":  "签名计算失败",
			})
		}

		if !strings.EqualFold(sign, expectedSign) {
			return c.Status(fiber.StatusUnauthorized).JSON(fiber.Map{
				"code": -1,
				"msg":  "签名验证失败",
			})
		}

		return c.Next()
	}
}

// ParseEncryptedBody 解密请求体中间件，将解密后的数据存入c.Locals
func ParseEncryptedBody(cfg *config.Config) fiber.Handler {
	return func(c *fiber.Ctx) error {
		var req struct {
			EncryptedData string `json:"encryptedData"`
		}
		if err := c.BodyParser(&req); err != nil {
			return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
				"code": -1,
				"msg":  "请求体解析失败",
			})
		}

		if req.EncryptedData == "" {
			return c.Next()
		}

		key := []byte(cfg.SM4Key)
		plaintext, err := crypto.SM4CBCDecrypt(req.EncryptedData, key)
		if err != nil {
			return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
				"code": -1,
				"msg":  "解密失败: " + err.Error(),
			})
		}

		c.Locals("decryptedBody", string(plaintext))
		return c.Next()
	}
}

// EncryptResponse 加密响应数据中间件
func EncryptResponse(cfg *config.Config) fiber.Handler {
	return func(c *fiber.Ctx) error {
		// 先执行后续handler
		err := c.Next()
		if err != nil {
			return err
		}

		// 获取响应体
		body := c.Response().Body()
		if len(body) == 0 {
			return nil
		}

		var resp struct {
			Code int             `json:"code"`
			Msg  string          `json:"msg"`
			Data json.RawMessage `json:"data,omitempty"`
		}
		if err := json.Unmarshal(body, &resp); err != nil {
			return nil
		}

		// 如果data不为空，需要加密
		if len(resp.Data) > 0 && string(resp.Data) != "null" {
			key := []byte(cfg.SM4Key)
			encrypted, err := crypto.SM4CBCEncrypt(resp.Data, key)
			if err != nil {
				return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
					"code": -1,
					"msg":  "响应加密失败",
				})
			}

			resp.Data = json.RawMessage(`"` + encrypted + `"`)
			encryptedBody, _ := json.Marshal(resp)
			c.Response().SetBody(encryptedBody)
		}

		return nil
	}
}

// File: xy_sh/internal/biz/biz.go

package biz

import (
	"encoding/json"
	"fmt"
	"xy_sh/internal/config"
	"xy_sh/internal/entities"
	"xy_sh/pkg/crypto"
)

// Biz 业务逻辑层
type Biz struct {
	cfg *config.Config
}

// NewBiz 创建业务逻辑实例
func NewBiz(cfg *config.Config) *Biz {
	return &Biz{cfg: cfg}
}

// DecryptAndUnmarshal 解密并反序列化请求数据
func (b *Biz) DecryptAndUnmarshal(encryptedData string, target interface{}) error {
	key := []byte(b.cfg.SM4Key)
	plaintext, err := crypto.SM4CBCDecrypt(encryptedData, key)
	if err != nil {
		return fmt.Errorf("解密失败: %v", err)
	}
	if err := json.Unmarshal(plaintext, target); err != nil {
		return fmt.Errorf("JSON解析失败: %v", err)
	}
	return nil
}

// EncryptAndMarshal 序列化并加密响应数据
func (b *Biz) EncryptAndMarshal(data interface{}) (string, error) {
	jsonBytes, err := json.Marshal(data)
	if err != nil {
		return "", fmt.Errorf("JSON序列化失败: %v", err)
	}
	key := []byte(b.cfg.SM4Key)
	encrypted, err := crypto.SM4CBCEncrypt(jsonBytes, key)
	if err != nil {
		return "", fmt.Errorf("加密失败: %v", err)
	}
	return encrypted, nil
}

// CreateOrder 卡券/直充权益下单
func (b *Biz) CreateOrder(req *entities.OrderRequest) (*entities.OrderResponse, error) {
	// TODO: 实现下单业务逻辑
	// 1. 校验参数
	// 2. 调用供应商接口
	// 3. 返回结果
	return &entities.OrderResponse{
		OrderNo:    "HM" + fmt.Sprintf("%d", 0),
		Status:     0,
		ExpireTime: "2025-12-31 23:59:59",
	}, nil
}

// QueryOrder 查询订单
func (b *Biz) QueryOrder(req *entities.QueryRequest) (*entities.QueryResponse, error) {
	// TODO: 实现查询业务逻辑
	return &entities.QueryResponse{
		OrderNo: req.OrderNo,
		Status:  0,
	}, nil
}

// WechatRecharge 微信立减金充值
func (b *Biz) WechatRecharge(req *entities.WechatRechargeRequest) (*entities.WechatRechargeResponse, error) {
	// TODO: 实现微信立减金充值业务逻辑
	return &entities.WechatRechargeResponse{
		OrderNo:  req.OrderNo,
		Status:   0,
		CouponID: "123456",
	}, nil
}

// HandleCallback 处理回调通知
func (b *Biz) HandleCallback(req *entities.CallbackRequest) error {
	// TODO: 实现回调处理逻辑
	// 1. 更新订单状态
	// 2. 通知相关系统
	return nil
}

// File: xy_sh/internal/service/service.go

package service

import (
	"encoding/json"

	"github.com/gofiber/fiber/v2"
	"xy_sh/internal/biz"
	"xy_sh/internal/config"
	"xy_sh/internal/entities"
)

// Service HTTP服务层
type Service struct {
	biz *biz.Biz
	cfg *config.Config
}

// NewService 创建服务实例
func NewService(cfg *config.Config) *Service {
	return &Service{
		biz: biz.NewBiz(cfg),
		cfg: cfg,
	}
}

// CreateOrder 卡券/直充权益下单
func (s *Service) CreateOrder(c *fiber.Ctx) error {
	// 获取解密后的请求体
	decryptedBody, ok := c.Locals("decryptedBody").(string)
	if !ok || decryptedBody == "" {
		return c.JSON(entities.ApiResponse{
			Code: -1,
			Msg:  "解密后的请求体为空",
		})
	}

	var req entities.OrderRequest
	if err := json.Unmarshal([]byte(decryptedBody), &req); err != nil {
		return c.JSON(entities.ApiResponse{
			Code: -1,
			Msg:  "请求参数解析失败: " + err.Error(),
		})
	}

	// 调用业务逻辑
	resp, err := s.biz.CreateOrder(&req)
	if err != nil {
		return c.JSON(entities.ApiResponse{
			Code: -1,
			Msg:  err.Error(),
		})
	}

	// 加密响应数据
	encryptedData, err := s.biz.EncryptAndMarshal(resp)
	if err != nil {
		return c.JSON(entities.ApiResponse{
			Code: -1,
			Msg:  "响应加密失败",
		})
	}

	return c.JSON(entities.ApiResponse{
		Code: 0,
		Msg:  "请求成功",
		Data: encryptedData,
	})
}

// QueryOrder 卡券/直充/微信立减金订单查询
func (s *Service) QueryOrder(c *fiber.Ctx) error {
	decryptedBody, ok := c.Locals("decryptedBody").(string)
	if !ok || decryptedBody == "" {
		return c.JSON(entities.ApiResponse{
			Code: -1,
			Msg:  "解密后的请求体为空",
		})
	}

	var req entities.QueryRequest
	if err := json.Unmarshal([]byte(decryptedBody), &req); err != nil {
		return c.JSON(entities.ApiResponse{
			Code: -1,
			Msg:  "请求参数解析失败: " + err.Error(),
		})
	}

	resp, err := s.biz.QueryOrder(&req)
	if err != nil {
		return c.JSON(entities.ApiResponse{
			Code: -1,
			Msg:  err.Error(),
		})
	}

	encryptedData, err := s.biz.EncryptAndMarshal(resp)
	if err != nil {
		return c.JSON(entities.ApiResponse{
			Code: -1,
			Msg:  "响应加密失败",
		})
	}

	return c.JSON(entities.ApiResponse{
		Code: 0,
		Msg:  "请求成功",
		Data: encryptedData,
	})
}

// WechatRecharge 微信立减金订单充值
func (s *Service) WechatRecharge(c *fiber.Ctx) error {
	decryptedBody, ok := c.Locals("decryptedBody").(string)
	if !ok || decryptedBody == "" {
		return c.JSON(entities.ApiResponse{
			Code: -1,
			Msg:  "解密后的请求体为空",
		})
	}

	var req entities.WechatRechargeRequest
	if err := json.Unmarshal([]byte(decryptedBody), &req); err != nil {
		return c.JSON(entities.ApiResponse{
			Code: -1,
			Msg:  "请求参数解析失败: " + err.Error(),
		})
	}

	resp, err := s.biz.WechatRecharge(&req)
	if err != nil {
		return c.JSON(entities.ApiResponse{
			Code: -1,
			Msg:  err.Error(),
		})
	}

	encryptedData, err := s.biz.EncryptAndMarshal(resp)
	if err != nil {
		return c.JSON(entities.ApiResponse{
			Code: -1,
			Msg:  "响应加密失败",
		})
	}

	return c.JSON(entities.ApiResponse{
		Code: 0,
		Msg:  "请求成功",
		Data: encryptedData,
	})
}

// HandleCallback 处理供应商回调通知
func (s *Service) HandleCallback(c *fiber.Ctx) error {
	decryptedBody, ok := c.Locals("decryptedBody").(string)
	if !ok || decryptedBody == "" {
		return c.Status(fiber.StatusBadRequest).SendString("invalid request")
	}

	var req entities.CallbackRequest
	if err := json.Unmarshal([]byte(decryptedBody), &req); err != nil {
		return c.Status(fiber.StatusBadRequest).SendString("invalid request")
	}

	if err := s.biz.HandleCallback(&req); err != nil {
		// 处理失败，返回非200状态码触发重试
		return c.Status(fiber.StatusInternalServerError).SendString("process failed")
	}

	// 处理成功，返回HTTP 200且内容为ok
	return c.Status(fiber.StatusOK).SendString("ok")
}

// File: xy_sh/internal/router/router.go

package router

import (
	"github.com/gofiber/fiber/v2"
	"xy_sh/internal/config"
	"xy_sh/internal/middleware"
	"xy_sh/internal/service"
)

// SetupRouter 设置路由
func SetupRouter(cfg *config.Config) *fiber.App {
	app := fiber.New()

	svc := service.NewService(cfg)

	// API路由组，应用签名验证和解密中间件
	api := app.Group("/api", middleware.VerifySign(cfg), middleware.ParseEncryptedBody(cfg))

	// 接口1：卡券/直充权益下单
	api.Post("/order/create", svc.CreateOrder)

	// 接口2：卡券/直充/微信立减金订单查询
	api.Post("/order/query", svc.QueryOrder)

	// 接口3：微信立减金订单充值
	api.Post("/order/wechat-recharge", svc.WechatRecharge)

	// 接口4：卡券/直充/微信立减金充值结果通知（回调接口，同样需要验签和解密）
	api.Post("/callback/notice", svc.HandleCallback)

	return app
}

// File: xy_sh/internal/test/example_test.go

package test

import (
	"bytes"
	"encoding/json"
	"fmt"
	"io"
	"net/http"
	"testing"
	"time"

	"xy_sh/internal/config"
	"xy_sh/internal/entities"
	"xy_sh/pkg/crypto"
)

// TestCreateOrder 模拟卡券/直充权益下单请求
func TestCreateOrder(t *testing.T) {
	cfg := &config.Config{
		ServerPort: "8080",
		SM4Key:     "1234567890abcdef", // 16字节密钥
		SM3Salt:    "test_salt",
	}

	// 构造业务参数
	req := entities.OrderRequest{
		ActCode:     "ACT001",
		GoodsCode:   "123456",
		ActOrderNum: "00001",
		Account:     "19912345678",
		CallbackURL: "https://xxx/notice",
	}

	// 加密业务参数
	reqJSON, _ := json.Marshal(req)
	encryptedData, err := crypto.SM4CBCEncrypt(reqJSON, []byte(cfg.SM4Key))
	if err != nil {
		t.Fatalf("加密失败: %v", err)
	}

	// 生成签名
	timestamp := fmt.Sprintf("%d", time.Now().UnixMilli())
	rawStr := timestamp + encryptedData
	sign, err := crypto.SM3WithSalt([]byte(rawStr), []byte(cfg.SM3Salt))
	if err != nil {
		t.Fatalf("签名失败: %v", err)
	}

	// 构造请求体
	body := map[string]string{
		"encryptedData": encryptedData,
	}
	bodyJSON, _ := json.Marshal(body)

	// 发送请求
	httpReq, _ := http.NewRequest("POST", "http://localhost:"+cfg.ServerPort+"/api/order/create", bytes.NewReader(bodyJSON))
	httpReq.Header.Set("Content-Type", "application/json")
	httpReq.Header.Set("timestamp", timestamp)
	httpReq.Header.Set("sign", sign)

	client := &http.Client{}
	resp, err := client.Do(httpReq)
	if err != nil {
		t.Fatalf("请求失败: %v", err)
	}
	defer resp.Body.Close()

	respBody, _ := io.ReadAll(resp.Body)
	t.Logf("响应: %s", string(respBody))
}

// TestQueryOrder 模拟订单查询请求
func TestQueryOrder(t *testing.T) {
	cfg := &config.Config{
		ServerPort: "8080",
		SM4Key:     "1234567890abcdef",
		SM3Salt:    "test_salt",
	}

	req := entities.QueryRequest{
		ActCode: "FBG6vdYqGE4mGX7EH/woEg==",
		OrderNo: "HM1757046717684000102707339840b23222",
	}

	reqJSON, _ := json.Marshal(req)
	encryptedData, err := crypto.SM4CBCEncrypt(reqJSON, []byte(cfg.SM4Key))
	if err != nil {
		t.Fatalf("加密失败: %v", err)
	}

	timestamp := fmt.Sprintf("%d", time.Now().UnixMilli())
	rawStr := timestamp + encryptedData
	sign, _ := crypto.SM3WithSalt([]byte(rawStr), []byte(cfg.SM3Salt))

	body := map[string]string{"encryptedData": encryptedData}
	bodyJSON, _ := json.Marshal(body)

	httpReq, _ := http.NewRequest("POST", "http://localhost:"+cfg.ServerPort+"/api/order/query", bytes.NewReader(bodyJSON))
	httpReq.Header.Set("Content-Type", "application/json")
	httpReq.Header.Set("timestamp", timestamp)
	httpReq.Header.Set("sign", sign)

	client := &http.Client{}
	resp, err := client.Do(httpReq)
	if err != nil {
		t.Fatalf("请求失败: %v", err)
	}
	defer resp.Body.Close()

	respBody, _ := io.ReadAll(resp.Body)
	t.Logf("响应: %s", string(respBody))
}

// TestWechatRecharge 模拟微信立减金充值请求
func TestWechatRecharge(t *testing.T) {
	cfg := &config.Config{
		ServerPort: "8080",
		SM4Key:     "1234567890abcdef",
		SM3Salt:    "test_salt",
	}

	req := entities.WechatRechargeRequest{
		ActCode:     "FBG6vdYqGE4mGX7EH/woEg==",
		GoodsCode:   "0001",
		ActOrderNum: "0001",
		OpenID:      "0001",
		AppID:       "00001",
	}

	reqJSON, _ := json.Marshal(req)
	encryptedData, err := crypto.SM4CBCEncrypt(reqJSON, []byte(cfg.SM4Key))
	if err != nil {
		t.Fatalf("加密失败: %v", err)
	}

	timestamp := fmt.Sprintf("%d", time.Now().UnixMilli())
	rawStr := timestamp + encryptedData
	sign, _ := crypto.SM3WithSalt([]byte(rawStr), []byte(cfg.SM3Salt))

	body := map[string]string{"encryptedData": encryptedData}
	bodyJSON, _ := json.Marshal(body)

	httpReq, _ := http.NewRequest("POST", "http://localhost:"+cfg.ServerPort+"/api/order/wechat-recharge", bytes.NewReader(bodyJSON))
	httpReq.Header.Set("Content-Type", "application/json")
	httpReq.Header.Set("timestamp", timestamp)
	httpReq.Header.Set("sign", sign)

	client := &http.Client{}
	resp, err := client.Do(httpReq)
	if err != nil {
		t.Fatalf("请求失败: %v", err)
	}
	defer resp.Body.Close()

	respBody, _ := io.ReadAll(resp.Body)
	t.Logf("响应: %s", string(respBody))
}

// TestCallback 模拟回调通知请求
func TestCallback(t *testing.T) {
	cfg := &config.Config{
		ServerPort: "8080",
		SM4Key:     "1234567890abcdef",
		SM3Salt:    "test_salt",
	}

	req := entities.CallbackRequest{
		OrderNo:     "HM202509291010001",
		ActOrderNum: "XY2025092910100001",
		Status:      3,
		Account:     "19912345678",
	}

	reqJSON, _ := json.Marshal(req)
	encryptedData, err := crypto.SM4CBCEncrypt(reqJSON, []byte(cfg.SM4Key))
	if err != nil {
		t.Fatalf("加密失败: %v", err)
	}

	timestamp := fmt.Sprintf("%d", time.Now().UnixMilli())
	rawStr := timestamp + encryptedData
	sign, _ := crypto.SM3WithSalt([]byte(rawStr), []byte(cfg.SM3Salt))

	body := map[string]string{"encryptedData": encryptedData}
	bodyJSON, _ := json.Marshal(body)

	httpReq, _ := http.NewRequest("POST", "http://localhost:"+cfg.ServerPort+"/api/callback/notice", bytes.NewReader(bodyJSON))
	httpReq.Header.Set("Content-Type", "application/json")
	httpReq.Header.Set("timestamp", timestamp)
	httpReq.Header.Set("sign", sign)

	client := &http.Client{}
	resp, err := client.Do(httpReq)
	if err != nil {
		t.Fatalf("请求失败: %v", err)
	}
	defer resp.Body.Close()

	respBody, _ := io.ReadAll(resp.Body)
	t.Logf("响应: %s", string(respBody))
}

// File: xy_sh/pkg/crypto/crypto.go

package crypto

import (
	"crypto/cipher"
	"crypto/hmac"
	"crypto/rand"
	"encoding/base64"
	"encoding/hex"
	"fmt"
	"io"
	"bytes"
	"math/big"
	"time"

	"github.com/tjfoc/gmsm/sm3"
	"github.com/tjfoc/gmsm/sm4"
)

// ==================== SM4-CBC 加密/解密 ====================

// SM4CBCEncrypt SM4-CBC模式加密
func SM4CBCEncrypt(plaintext []byte, key []byte) (string, error) {
	if len(key) != 16 {
		return "", fmt.Errorf("SM4密钥长度必须为16字节")
	}

	block, err := sm4.NewCipher(key)
	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模式解密
func SM4CBCDecrypt(encryptedData string, key []byte) ([]byte, error) {
	if len(key) != 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(key)
	if err != nil {
		return nil, fmt.Errorf("创建SM4 cipher失败: %v", err)
	}

	blockSize := block.BlockSize()
	if len(data) < blockSize {
		return nil, fmt.Errorf("数据长度不足")
	}
	iv := data[:blockSize]
	ciphertext := data[blockSize:]

	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, fmt.Errorf("数据为空")
	}
	padding := int(data[length-1])
	if padding > length || padding == 0 {
		return nil, fmt.Errorf("无效的填充")
	}
	for i := length - padding; i < length; i++ {
		if data[i] != byte(padding) {
			return nil, fmt.Errorf("无效的填充")
		}
	}
	return data[:length-padding], nil
}

// ==================== SM3 哈希 ====================

// SM3Hash 计算SM3哈希值
func SM3Hash(data []byte, encoding string) (string, error) {
	h := sm3.New()
	h.Write(data)
	hashBytes := h.Sum(nil)

	if encoding == "base64" {
		return base64.StdEncoding.EncodeToString(hashBytes), nil
	}
	return hex.EncodeToString(hashBytes), nil
}

// SM3HashString 计算字符串的SM3哈希值
func SM3HashString(data string, encoding string) (string, error) {
	return SM3Hash([]byte(data), encoding)
}

// SM3Verify 验证数据与哈希值是否匹配
func SM3Verify(data []byte, hash string, encoding string) (bool, error) {
	expected, err := SM3Hash(data, encoding)
	if err != nil {
		return false, err
	}
	return expected == hash, nil
}

// HMACSM3 HMAC-SM3计算
func HMACSM3(data []byte, key []byte) string {
	h := hmac.New(sm3.New, key)
	h.Write(data)
	return hex.EncodeToString(h.Sum(nil))
}

// SM3WithSalt SM3加盐哈希（对应Java的SmUtil.sm3WithSalt）
// 实现方式：使用HMAC-SM3，将salt作为HMAC密钥
func SM3WithSalt(data []byte, salt []byte) (string, error) {
	h := hmac.New(sm3.New, salt)
	h.Write(data)
	return hex.EncodeToString(h.Sum(nil)), nil
}

// ==================== 时间戳和随机数 ====================

// GenerateTimestamp 生成秒级时间戳
func GenerateTimestamp() string {
	return fmt.Sprintf("%d", time.Now().Unix())
}

// GenerateTimestampMillis 生成毫秒级时间戳
func GenerateTimestampMillis() string {
	return fmt.Sprintf("%d", time.Now().UnixMilli())
}

// GenerateNonce 生成指定长度的随机字符串
func GenerateNonce(length int) (string, error) {
	const charset = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
	b := make([]byte, length)
	for i := range b {
		num, err := rand.Int(rand.Reader, big.NewInt(int64(len(charset))))
		if err != nil {
			return "", err
		}
		b[i] = charset[num.Int64()]
	}
	return string(b), nil
}

=== 服务端生成完成 ===

加密实现

以下是从加密工具获取的完整实现：

加密实现 1

{ "success": true, "data": "\n### SM3 国密哈希算法完整实现指南\n\n适用场景：文档要求使用国密 SM3 算法进行摘要计算或签名验证\n\n前置要求：需要安装 github.com/tjfoc/gmsm\n\nbash\ngo get github.com/tjfoc/gmsm\n\n\n配置参数：\n- 编码方式: hex\n- 包含 HMAC: true\n\n完整代码模板：\n\ngo\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使用示例：\ngo\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前置要求：\nbash\ngo get github.com/tjfoc/gmsm\n\n\n配置参数：\n- 编码方式: base64\n\n完整代码模板（请直接复制使用）：\n\ngo\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" }

加密实现 3

{ "success": true, "data": "\n### 时间戳和随机数生成规范\n\n代码模板：\n\ngo\npackage crypto\n\nimport (\n \"crypto/rand\"\n \"fmt\"\n \"math/big\"\n \"time\"\n)\n\n// GenerateTimestamp 生成秒级时间戳\nfunc GenerateTimestamp() string {\n return fmt.Sprintf(\"%d\", time.Now().Unix())\n}\n\n// GenerateTimestampMillis 生成毫秒级时间戳\nfunc GenerateTimestampMillis() string {\n return fmt.Sprintf(\"%d\", time.Now().UnixMilli())\n}\n\n// GenerateNonce 生成指定长度的随机字符串（加密安全）\nfunc GenerateNonce(length int) (string, error) {\n const charset = \"abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789\"\n b := make([]byte, length)\n for i := range b {\n num, err := rand.Int(rand.Reader, big.NewInt(int64(len(charset))))\n if err != nil {\n return \"\", err\n }\n b[i] = charset[num.Int64()]\n }\n return string(b), nil\n}\n\n\n注意事项：\n- 确认文档要求的是秒级还是毫秒级时间戳\n- 确认nonce的长度要求（通常16-32位）\n- 生产环境建议使用加密安全的随机数生成器\n", "tool": "nonce_timestamp" }

32 KiB Raw Blame History Unescape Escape

加密实现

加密实现 1

加密实现 2

加密实现 3

32 KiB

Raw Blame History