tolelom
d597ef2d46
All checks were successful
Server CI/CD / deploy (push) Successful in 1m31s
- 입력 검증 강화 (로그인/체인 핸들러 전체) - boss raid 비관적 잠금으로 동시성 문제 해결 - SSAFY 사용자명 sanitize + 트랜잭션 처리 - constant-time API 키 비교, 보안 헤더, graceful shutdown - 안전하지 않은 기본값 경고 추가 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
84 lines
2.6 KiB
Go
84 lines
2.6 KiB
Go
package download
|
|
|
|
import (
|
|
"log"
|
|
"mime"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
)
|
|
|
|
type Handler struct {
|
|
svc *Service
|
|
baseURL string
|
|
}
|
|
|
|
func NewHandler(svc *Service, baseURL string) *Handler {
|
|
return &Handler{svc: svc, baseURL: baseURL}
|
|
}
|
|
|
|
func (h *Handler) GetInfo(c *fiber.Ctx) error {
|
|
info, err := h.svc.GetInfo()
|
|
if err != nil {
|
|
return c.Status(fiber.StatusNotFound).JSON(fiber.Map{"error": "다운로드 정보가 없습니다"})
|
|
}
|
|
return c.JSON(info)
|
|
}
|
|
|
|
// Upload accepts a raw binary body (application/octet-stream).
|
|
// The filename is passed as a query parameter: ?filename=A301_v1.0.zip
|
|
func (h *Handler) Upload(c *fiber.Ctx) error {
|
|
filename := strings.TrimSpace(c.Query("filename", "game.zip"))
|
|
// 경로 순회 방지: 디렉토리 구분자 제거, 기본 파일명만 사용
|
|
filename = filepath.Base(filename)
|
|
if !strings.HasSuffix(strings.ToLower(filename), ".zip") {
|
|
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "zip 파일만 업로드 가능합니다"})
|
|
}
|
|
if len(filename) > 200 {
|
|
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "파일명이 너무 깁니다"})
|
|
}
|
|
|
|
body := c.Request().BodyStream()
|
|
info, err := h.svc.Upload(filename, body, h.baseURL)
|
|
if err != nil {
|
|
log.Printf("game upload failed: %v", err)
|
|
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "게임 파일 업로드에 실패했습니다"})
|
|
}
|
|
return c.JSON(info)
|
|
}
|
|
|
|
func (h *Handler) ServeFile(c *fiber.Ctx) error {
|
|
path := h.svc.GameFilePath()
|
|
if _, err := os.Stat(path); err != nil {
|
|
return c.Status(fiber.StatusNotFound).JSON(fiber.Map{"error": "파일이 없습니다"})
|
|
}
|
|
info, _ := h.svc.GetInfo()
|
|
filename := "game.zip"
|
|
if info != nil && info.FileName != "" {
|
|
filename = info.FileName
|
|
}
|
|
c.Set("Content-Disposition", mime.FormatMediaType("attachment", map[string]string{"filename": filename}))
|
|
return c.SendFile(path)
|
|
}
|
|
|
|
func (h *Handler) UploadLauncher(c *fiber.Ctx) error {
|
|
body := c.Request().BodyStream()
|
|
info, err := h.svc.UploadLauncher(body, h.baseURL)
|
|
if err != nil {
|
|
log.Printf("launcher upload failed: %v", err)
|
|
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "런처 업로드에 실패했습니다"})
|
|
}
|
|
return c.JSON(info)
|
|
}
|
|
|
|
func (h *Handler) ServeLauncher(c *fiber.Ctx) error {
|
|
path := h.svc.LauncherFilePath()
|
|
if _, err := os.Stat(path); err != nil {
|
|
return c.Status(fiber.StatusNotFound).JSON(fiber.Map{"error": "파일이 없습니다"})
|
|
}
|
|
c.Set("Content-Disposition", `attachment; filename="launcher.exe"`)
|
|
return c.SendFile(path)
|
|
}
|