tolelom
cc751653c4
All checks were successful
Server CI/CD / deploy (push) Successful in 1m26s
보안: - RPC 응답 HTTP 상태코드 검증 (chain/client) - SSAFY OAuth 에러 응답 내부 로깅으로 변경 (제3자 상세 노출 제거) - resolveUsername에서 username 노출 제거 - LIKE 쿼리 특수문자 이스케이프 (bossraid/repository) - 파일명 경로 순회 방지 + 길이 제한 (download/handler) - ServerAuth 실패 로깅 추가 안정성: - AutoMigrate 에러 시 서버 종료 - GetLatest() 에러 시 nil 반환 (초기화 안 된 포인터 방지) - 멱등성 캐시 저장 시 새 context 사용 - SSAFY HTTP 클라이언트 타임아웃 10s - io.ReadAll/rand.Read 에러 처리 - Login에서 DB 에러/Not Found 구분 검증 강화: - 중복 플레이어 검증 (bossraid/service) - username 길이 제한 50자 (auth/handler, bossraid/handler) - 역할 변경 시 세션 무효화 - 지갑 복호화 실패 로깅 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
71 lines
1.8 KiB
Go
71 lines
1.8 KiB
Go
package middleware
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"log"
|
|
"time"
|
|
|
|
"a301_server/pkg/database"
|
|
"github.com/gofiber/fiber/v2"
|
|
)
|
|
|
|
const idempotencyTTL = 10 * time.Minute
|
|
const redisTimeout = 5 * time.Second
|
|
|
|
type cachedResponse struct {
|
|
StatusCode int `json:"s"`
|
|
Body json.RawMessage `json:"b"`
|
|
}
|
|
|
|
// Idempotency checks the Idempotency-Key header to prevent duplicate transactions.
|
|
// If the same key is seen again within the TTL, the cached response is returned.
|
|
func Idempotency(c *fiber.Ctx) error {
|
|
key := c.Get("Idempotency-Key")
|
|
if key == "" {
|
|
return c.Next()
|
|
}
|
|
|
|
// userID가 있으면 키에 포함하여 사용자 간 캐시 충돌 방지
|
|
redisKey := "idempotency:"
|
|
if uid, ok := c.Locals("userID").(uint); ok {
|
|
redisKey += fmt.Sprintf("u%d:", uid)
|
|
}
|
|
redisKey += key
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), redisTimeout)
|
|
defer cancel()
|
|
|
|
// Check if this key was already processed
|
|
cached, err := database.RDB.Get(ctx, redisKey).Bytes()
|
|
if err == nil && len(cached) > 0 {
|
|
var cr cachedResponse
|
|
if json.Unmarshal(cached, &cr) == nil {
|
|
c.Set("Content-Type", "application/json")
|
|
c.Set("X-Idempotent-Replay", "true")
|
|
return c.Status(cr.StatusCode).Send(cr.Body)
|
|
}
|
|
}
|
|
|
|
// Process the request
|
|
if err := c.Next(); err != nil {
|
|
return err
|
|
}
|
|
|
|
// Cache successful responses (2xx)
|
|
status := c.Response().StatusCode()
|
|
if status >= 200 && status < 300 {
|
|
cr := cachedResponse{StatusCode: status, Body: c.Response().Body()}
|
|
if data, err := json.Marshal(cr); err == nil {
|
|
writeCtx, writeCancel := context.WithTimeout(context.Background(), redisTimeout)
|
|
defer writeCancel()
|
|
if err := database.RDB.Set(writeCtx, redisKey, data, idempotencyTTL).Err(); err != nil {
|
|
log.Printf("WARNING: idempotency cache write failed (key=%s): %v", key, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|