[박인성] 웹서버 7단계 완료 - 남은 Figma 요구사항 구현 & 취약점 방어 로직 구현 / 26.01.15.

[Hexeong]

완료 작업 목록

• 기존은 regitstration 이후 자동으로 로그인 되도록 구현되어 있으나, regitstration 이후 로그인 페이지로 이동하도록 수정해야 한다.
• 지원하지 않는 타입을 업로드하는 걸 막아야 한다.
• 업로드한 파일에 대해서 Payload 크기에 대한 제한으로 방어로직 구현
• /user/info 요청에 대해 html에서 둘의 값이 같지 않으면 전송하지 않도록 바꿔야 함.
• [Readme.md](http://Readme.md) 정리
• JavaDoc으로 Github Pages 만들기

구현하며 고민했던 내용

• 지원하지 않는 파일을 올리는 사용자에 대해서

  • 당연히 서비스 로직 상에서 이미지 외의 업로드 파일을 염두해두지 않고 만들었기에 서버에서 지원하는 이미지 파일이 아닌 경우, 사용자에게 올리지 말라고 해야 합니다.

  • 때문에 해당 요청을 거부하는 로직을 어디서 구현할지 고민했습니다.

  • 원래 MultipartFile에 대해서 body를 해석하는 부분을 ArgumentResolver에서 구현했으니, 해석이 끝나고 지원하는 타입이 아니라면 예외를 발생시키면 되겠다 싶었습니다.

  • 하지만, 구현할수록 ArgumentResolver의 책임이 너무 크다는 느낌이 들더군요..

  • 그래서 이걸 저는 새로 Interceptor 구현체를 만들어 요청 차단을 하기로 했습니다.

    • 구현 방식으로는 정규표현식으로 filename=\"(.*?)\"" 을 써서 group안에 들어있는 텍스트를 검사하는 방식으로 구현했습니다.

    public class FileUploadSafetyInterceptor implements Interceptor{
    
        private static final String COOKIE_HEADER_KEY = "Set-Cookie";
        private static final String HEADER_LOCATION = "Location";
        private static final Pattern FILENAME_PATTERN = Pattern.compile("filename=\"(.*?)\"");
        private static final String MY_DOMAIN = "http://localhost:8080";
    
        @Override
        public boolean preHandle(HttpRequest req, HttpResponse res, Object handler) throws CustomException {
            if (handler instanceof HandlerMethod hm) {
                boolean flag = false;
                for (Parameter params : hm.getHandlerMethod().getParameters()) {
                    if (params.isAnnotationPresent(MultipartFormdata.class)) {
                        flag = true;
                        break;
                    }
                }
                if (flag) {
                    Matcher matcher = FILENAME_PATTERN.matcher(req.bodyText());
                    while (matcher.find()) {
                        String originalFilename = matcher.group(1);
                        if (originalFilename == null || originalFilename.isBlank()) {
                            continue;
                        }
    
                        if (!UploadableFileType.isAllowed(originalFilename)) {
                            String redirectPathOnFail = req.headers().getOrDefault("referer", "/");
                            if (!redirectPathOnFail.startsWith(MY_DOMAIN))
                                redirectPathOnFail = "/";
    
                            res.setStatus(HttpStatus.FOUND);
                            res.addHeader(HEADER_LOCATION, redirectPathOnFail);
    
                            String encodedMsg = URLEncoder.encode("업로드를 지원하지 않는 이미지 파일 형식입니다.", StandardCharsets.UTF_8);
                            res.addHeader(COOKIE_HEADER_KEY, "alertMessage=" + encodedMsg + "; Path=/");
                            throw new CustomException(ErrorCode.NOT_SUPPORTED_FILE_TYPE);
                        }
                    }
                }
            }
            return true;
        };
    }

  • 다음과 같이 구현하여 성공적으로 지원하는 업로드 이미지 파일 형식이 아니라면 차단하는 로직을 완성했습니다.

    • 구현하고 나니 모든 요청에 대해 Interceptor의 대상인지를 검사하는 로직이 많이 비효율적으로 보이네요.. HandlerExecutionChain을 넘겨줄 때, 지원하면 넘겨주는 방식으로 구현해도 될 것 같다는 생각입니다. 추후 시간이 나면 한번 작업해보겠습니다!
    • 또한, 현재 구조가 Http 메시지를 파싱할 때, body 전체를 String 타입으로 저장하다 보니, 영상 같은 body가 무거워지는 경우에 대해서는 메모리 상에서 문제가 될 것 같은 느낌이 드네요…

AI 도움 받은 부분

• 버그를 찾는 과정과 HTML 코드 수정하는 과정에서 AI의 도움을 받았습니다!