xy_sh-20260723184208/xy_sh/valid.md

// File: xy_sh/go.mod

module xy_sh

go 1.23

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

// File: xy_sh/go.sum

// File: xy_sh/Dockerfile

FROM golang:1.23-alpine AS builder

WORKDIR /app
COPY go.mod go.sum ./
RUN go mod download
COPY . .
RUN CGO_ENABLED=0 GOOS=linux go build -o /server ./cmd/server

FROM alpine:3.19
RUN apk --no-cache add ca-certificates tzdata
WORKDIR /app
COPY --from=builder /server .
EXPOSE 8080
CMD ["./server"]

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

package main

import (
	"log"
	"xy_sh/internal/config"
	"xy_sh/internal/middleware"
	"xy_sh/internal/service"

	"github.com/gofiber/fiber/v2"
)

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

	app := fiber.New(fiber.Config{
		ReadTimeout:  cfg.ReadTimeout,
		WriteTimeout: cfg.WriteTimeout,
	})

	// 全局中间件
	app.Use(middleware.CORS())
	app.Use(middleware.RequestLogger())

	// 注册路由
	service.RegisterRoutes(app, 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"
	"time"
)

// Config 应用配置
type Config struct {
	ServerPort   string
	ReadTimeout  time.Duration
	WriteTimeout time.Duration
	SM3Salt      string
	SM4Key       []byte
	CallbackURL  string
}

// Load 加载配置，优先从环境变量读取
func Load() *Config {
	sm4KeyStr := getEnv("SM4_KEY", "1234567890abcdef")
	return &Config{
		ServerPort:   getEnv("SERVER_PORT", "8080"),
		ReadTimeout:  10 * time.Second,
		WriteTimeout: 10 * time.Second,
		SM3Salt:      getEnv("SM3_SALT", "default_salt"),
		SM4Key:       []byte(sm4KeyStr),
		CallbackURL:  getEnv("CALLBACK_URL", ""),
	}
}

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

// File: xy_sh/internal/entities/request.go

package entities

// OrderCreateBizReq 卡券/直充权益下单请求业务参数
type OrderCreateBizReq struct {
	ActCode     string `json:"actCode"`
	GoodsCode   string `json:"goodsCode"`
	ActOrderNum string `json:"actOrderNum"`
	Account     string `json:"account,omitempty"`
	CallbackURL string `json:"callbackUrl,omitempty"`
}

// OrderQueryBizReq 卡券/直充/微信立减金订单查询请求业务参数
type OrderQueryBizReq struct {
	ActCode string `json:"actCode"`
	OrderNo string `json:"orderNo"`
}

// WechatRechargeBizReq 微信立减金订单充值请求业务参数
type WechatRechargeBizReq 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"`
}

// CallbackNotifyBizReq 回调通知请求业务参数
type CallbackNotifyBizReq 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"`
}

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

// OrderQueryBizResp 订单查询响应业务参数
type OrderQueryBizResp struct {
	OrderNo  string     `json:"orderNo"`
	Status   int        `json:"status"`
	Account  string     `json:"account,omitempty"`
	CardInfo *CardInfo  `json:"cardInfo,omitempty"`
	CouponId string     `json:"couponId,omitempty"`
}

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

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

package middleware

import (
	"log"
	"time"

	"github.com/gofiber/fiber/v2"
	"github.com/gofiber/fiber/v2/middleware/cors"
)

// CORS 跨域中间件
func CORS() fiber.Handler {
	return cors.New(cors.Config{
		AllowOrigins: "*",
		AllowMethods: "GET,POST,PUT,DELETE,OPTIONS",
		AllowHeaders: "Origin,Content-Type,Accept,timestamp,sign",
	})
}

// RequestLogger 请求日志中间件
func RequestLogger() fiber.Handler {
	return func(c *fiber.Ctx) error {
		start := time.Now()
		err := c.Next()
		duration := time.Since(start)
		log.Printf("[%s] %s %s - %d (%v)",
			c.Method(),
			c.Path(),
			c.IP(),
			c.Response().StatusCode(),
			duration,
		)
		return err
	}
}

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

package biz

import (
	"xy_sh/internal/entities"
)

// Biz 业务逻辑接口
type Biz interface {
	// CreateOrder 卡券/直充权益下单
	CreateOrder(req *entities.OrderCreateBizReq) (*entities.OrderCreateBizResp, error)
	// QueryOrder 卡券/直充/微信立减金订单查询
	QueryOrder(req *entities.OrderQueryBizReq) (*entities.OrderQueryBizResp, error)
	// WechatRecharge 微信立减金订单充值
	WechatRecharge(req *entities.WechatRechargeBizReq) (*entities.WechatRechargeBizResp, error)
	// ProcessCallback 处理回调通知
	ProcessCallback(req *entities.CallbackNotifyBizReq) error
}

type bizImpl struct{}

// NewBiz 创建业务逻辑实例
func NewBiz() Biz {
	return &bizImpl{}
}

func (b *bizImpl) CreateOrder(req *entities.OrderCreateBizReq) (*entities.OrderCreateBizResp, error) {
	// TODO: 实现下单业务逻辑
	return &entities.OrderCreateBizResp{
		OrderNo: "HM" + generateOrderNo(),
		Status:  -1,
	}, nil
}

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

func (b *bizImpl) WechatRecharge(req *entities.WechatRechargeBizReq) (*entities.WechatRechargeBizResp, error) {
	// TODO: 实现微信立减金充值业务逻辑
	return &entities.WechatRechargeBizResp{
		OrderNo: req.OrderNo,
		Status:  -1,
	}, nil
}

func (b *bizImpl) ProcessCallback(req *entities.CallbackNotifyBizReq) error {
	// TODO: 处理回调通知业务逻辑
	return nil
}

func generateOrderNo() string {
	return "ORDERNO"
}

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

package service

import (
	"encoding/json"
	"log"

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

	"github.com/gofiber/fiber/v2"
)

// RegisterRoutes 注册所有路由
func RegisterRoutes(app *fiber.App, cfg *config.Config) {
	b := biz.NewBiz()

	// 接口1：卡券/直充权益下单
	app.Post("/api/order/create", handleWithDecrypt(cfg, b, func(reqBytes []byte) (interface{}, error) {
		var bizReq entities.OrderCreateBizReq
		if err := json.Unmarshal(reqBytes, &bizReq); err != nil {
			return nil, err
		}
		return b.CreateOrder(&bizReq)
	}))

	// 接口2：卡券/直充/微信立减金订单查询
	app.Post("/api/order/query", handleWithDecrypt(cfg, b, func(reqBytes []byte) (interface{}, error) {
		var bizReq entities.OrderQueryBizReq
		if err := json.Unmarshal(reqBytes, &bizReq); err != nil {
			return nil, err
		}
		return b.QueryOrder(&bizReq)
	}))

	// 接口3：微信立减金订单充值
	app.Post("/api/order/recharge", handleWithDecrypt(cfg, b, func(reqBytes []byte) (interface{}, error) {
		var bizReq entities.WechatRechargeBizReq
		if err := json.Unmarshal(reqBytes, &bizReq); err != nil {
			return nil, err
		}
		return b.WechatRecharge(&bizReq)
	}))

	// 接口4：卡券/直充/微信立减金充值结果通知接口（接收供应商回调）
	app.Post("/api/callback/notify", handleCallback(cfg, b))
}

// handleWithDecrypt 通用处理：验证签名 -> 解密 -> 执行业务 -> 加密响应
func handleWithDecrypt(cfg *config.Config, b biz.Biz, handler func([]byte) (interface{}, error)) fiber.Handler {
	return func(c *fiber.Ctx) error {
		// 1. 验证签名
		timestamp := c.Get("timestamp")
		sign := c.Get("sign")
		if timestamp == "" || sign == "" {
			return c.JSON(entities.ApiResponse{
				Code: -1,
				Msg:  "缺少timestamp或sign",
			})
		}

		// 2. 获取加密数据
		var reqBody struct {
			EncryptedData string `json:"encryptedData"`
		}
		if err := c.BodyParser(&reqBody); err != nil {
			return c.JSON(entities.ApiResponse{
				Code: -1,
				Msg:  "请求体解析失败",
			})
		}

		// 3. 验证SM3签名：sign = SM3(salt + timestamp + encryptedData)
		expectedSign := crypto.SM3WithSalt(cfg.SM3Salt, timestamp+reqBody.EncryptedData)
		if expectedSign != sign {
			log.Printf("签名验证失败: expected=%s, got=%s", expectedSign, sign)
			return c.JSON(entities.ApiResponse{
				Code: -1,
				Msg:  "签名验证失败",
			})
		}

		// 4. SM4解密业务数据
		bizData, err := crypto.SM4ECBDecrypt(reqBody.EncryptedData, cfg.SM4Key)
		if err != nil {
			log.Printf("SM4解密失败: %v", err)
			return c.JSON(entities.ApiResponse{
				Code: -1,
				Msg:  "数据解密失败",
			})
		}

		// 5. 执行业务逻辑
		result, err := handler(bizData)
		if err != nil {
			log.Printf("业务处理失败: %v", err)
			return c.JSON(entities.ApiResponse{
				Code: -1,
				Msg:  err.Error(),
			})
		}

		// 6. 加密响应数据
		respBytes, err := json.Marshal(result)
		if err != nil {
			return c.JSON(entities.ApiResponse{
				Code: -1,
				Msg:  "响应序列化失败",
			})
		}

		encryptedResp, err := crypto.SM4ECBEncrypt(respBytes, cfg.SM4Key)
		if err != nil {
			return c.JSON(entities.ApiResponse{
				Code: -1,
				Msg:  "响应加密失败",
			})
		}

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

// handleCallback 处理供应商回调通知
func handleCallback(cfg *config.Config, b biz.Biz) fiber.Handler {
	return func(c *fiber.Ctx) error {
		// 1. 验证签名
		timestamp := c.Get("timestamp")
		sign := c.Get("sign")
		if timestamp == "" || sign == "" {
			return c.Status(400).SendString("missing auth headers")
		}

		// 2. 获取加密数据
		var reqBody struct {
			EncryptedData string `json:"encryptedData"`
		}
		if err := c.BodyParser(&reqBody); err != nil {
			return c.Status(400).SendString("invalid body")
		}

		// 3. 验证SM3签名
		expectedSign := crypto.SM3WithSalt(cfg.SM3Salt, timestamp+reqBody.EncryptedData)
		if expectedSign != sign {
			log.Printf("回调签名验证失败: expected=%s, got=%s", expectedSign, sign)
			return c.Status(400).SendString("sign verification failed")
		}

		// 4. SM4解密业务数据
		bizData, err := crypto.SM4ECBDecrypt(reqBody.EncryptedData, cfg.SM4Key)
		if err != nil {
			log.Printf("回调数据SM4解密失败: %v", err)
			return c.Status(400).SendString("decrypt failed")
		}

		// 5. 解析回调业务参数
		var callbackReq entities.CallbackNotifyBizReq
		if err := json.Unmarshal(bizData, &callbackReq); err != nil {
			log.Printf("回调数据解析失败: %v", err)
			return c.Status(400).SendString("parse failed")
		}

		// 6. 处理回调业务
		if err := b.ProcessCallback(&callbackReq); err != nil {
			log.Printf("回调处理失败: %v", err)
			return c.Status(500).SendString("process failed")
		}

		// 7. 返回成功（必须返回200且内容为ok）
		return c.Status(200).SendString("ok")
	}
}

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

package crypto

import (
	"encoding/base64"
	"encoding/hex"
	"fmt"
	"time"

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

// ==================== SM3 with Salt ====================

// SM3WithSalt 使用SM3加盐哈希
// 对应Java hutool的 SmUtil.sm3WithSalt(salt).digestHex(data)
// hutool实现为：SM3(salt + data)
func SM3WithSalt(salt string, data string) string {
	h := sm3.New()
	h.Write([]byte(salt))
	h.Write([]byte(data))
	return hex.EncodeToString(h.Sum(nil))
}

// ==================== SM4-ECB 加密/解密 ====================

// SM4ECBEncrypt SM4-ECB模式加密，返回base64编码字符串
func SM4ECBEncrypt(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())
	ciphertext := make([]byte, len(padded))
	for i := 0; i < len(padded); i += block.BlockSize() {
		block.Encrypt(ciphertext[i:i+block.BlockSize()], padded[i:i+block.BlockSize()])
	}

	return base64.StdEncoding.EncodeToString(ciphertext), nil
}

// SM4ECBDecrypt SM4-ECB模式解密，输入base64编码字符串，返回明文
func SM4ECBDecrypt(encryptedData string, key []byte) ([]byte, error) {
	if len(key) != 16 {
		return nil, fmt.Errorf("SM4密钥长度必须为16字节")
	}

	ciphertext, 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(ciphertext)%blockSize != 0 {
		return nil, fmt.Errorf("密文长度不是块大小的整数倍")
	}

	plaintext := make([]byte, len(ciphertext))
	for i := 0; i < len(ciphertext); i += blockSize {
		block.Decrypt(plaintext[i:i+blockSize], ciphertext[i:i+blockSize])
	}

	plaintext, err = pkcs7UnPadding(plaintext)
	if err != nil {
		return nil, fmt.Errorf("去除填充失败: %v", err)
	}

	return plaintext, nil
}

// ==================== PKCS7 填充 ====================

func pkcs7Padding(data []byte, blockSize int) []byte {
	padding := blockSize - len(data)%blockSize
	padText := make([]byte, padding)
	for i := range padText {
		padText[i] = byte(padding)
	}
	return append(data, padText...)
}

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

// ==================== 时间戳工具 ====================

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