(23.01.16.)Spring 프레임워크 : 이메일 인증 보완 - 다른 기기에서 인증시 세션이 달라 인증 오류 고치기

이메일 인증 보완

-인증을 신청한 기기 이외의 기기에서도 인증 답변 메일 확인

-이메일에 인증용 암호 문자열을 전달

-이메일 수신자가 메일을 열고 링크 클릭 > 암호문자열을 다시 서버로 보내준다.

-이메일을 보낼때의 서버측 메소드와 암호문자열을 다시 받을 때의 메소드는 다르다.

-이용자의 세션에 암호문자열을 기억시켜놓고 돌아왔을 때의 암호 문자열을 비교한다.

-위의 절차가 모두 동일 시스템에서 실행된다면 아무런 문제가 없다.

-하지만 다른 기기에서 인증에 응답을 누르면 세션이 다른 세션으로 바뀌기 때문에 인증이 작동하지 않는다.

 

 

따라서 인증 링크를 누를 때 세션 id까지 같이 줘서 해당 세션 id에 있는 문자열과 함께 비교하게끔 해야한다.

 

 

 

 

먼저 메인 어플리케이션에 @ServletComponentScan을 선언해 줘야한다.

 

 

핸들러 클래스를 만들어서 세션이 새로 생성될 때 마다 리스너를 통해서 그 사실을 알 수 있다.

 

HttpSessionHandler.java

import java.util.*;
import jakarta.servlet.annotation.WebListener;
import jakarta.servlet.http.HttpSession;
import jakarta.servlet.http.HttpSessionEvent;
import jakarta.servlet.http.HttpSessionListener;
 
@WebListener
public class HttpSessionHandler implements HttpSessionListener{
    
    public static Map<String,HttpSession> map = new HashMap<>();
    
    public void sessionCreated(HttpSessionEvent se) 
    {
        HttpSession s = se.getSession();
        map.put(s.getId(), s);
    }
    public void sessionDestroyed(HttpSessionEvent se) 
    {
        HttpSession s = se.getSession();
        map.remove(s.getId());
    }
}
 

 

 

 

EmailController.java

 
import java.util.Map;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import com.ezen.spring.web.service.EmailService;
import jakarta.servlet.http.HttpSession;
import lombok.extern.slf4j.Slf4j;
 
@Slf4j
@Controller
@RequestMapping("/mail")
public class EmailController {
 
    @Autowired
    private EmailService svc;
 
    @GetMapping("/test")
    @ResponseBody
    public String sendTestMail() {
        boolean isSent = svc.sendSimpleText();
        // boolean isSent = svc.sendAttachMail();
        return isSent ? "메일 보내기 성공" : "메일 보내기 실패";
    }
 
    @GetMapping("/auth/{code}/{sid}") // 보낸 메일에서 이용자가 인증 링크를 클릭했을 때
    @ResponseBody
    public String index(@PathVariable("sid") String sid ,@PathVariable("code") String code) {
        Map<String, HttpSession> map = HttpSessionHandler.map;
        HttpSession originalSession = map.get(sid);
        if(originalSession.getAttribute("code").equals(code)) {
            return"인증완료";
        }
        return "인증실패";
    }
}

 

 

 

EmailService.java

import java.io.File;
import java.util.*;
import jakarta.activation.DataHandler;
import jakarta.activation.FileDataSource;
import jakarta.mail.BodyPart;
import jakarta.mail.Message;
import jakarta.mail.MessagingException;
import jakarta.mail.Multipart;
import jakarta.mail.internet.InternetAddress;
import jakarta.mail.internet.MimeBodyPart;
import jakarta.mail.internet.MimeMessage;
import jakarta.mail.internet.MimeMultipart;
import jakarta.servlet.http.HttpSession;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.mail.SimpleMailMessage;
import org.springframework.mail.javamail.JavaMailSender;
import org.springframework.stereotype.Service;
import lombok.extern.slf4j.Slf4j;
 
@Slf4j
@Service
public class EmailService
{
   @Autowired
   private JavaMailSender sender;
   
   private HttpSession session;
   
   public void setSession(HttpSession session) {
        this.session = session;
    }
    
   
   public boolean sendSimpleText()
   {
      List<String> receivers = new ArrayList<>();
      //메일 보낼 수신자를 db에서 써내서 리스트에 담아서 반복문을 돌린다.
      receivers.add("parkjh080@gmail.com");
 
      String[] arrReceiver = (String[])receivers.toArray(new String[receivers.size()]);
                                                          //배열의 저장공간 확보(메모리 크기지정)
      
      SimpleMailMessage simpleMailMessage = new SimpleMailMessage();
      //심플의 뜻은 일반 텍스트만 보내겠다는 뜻
      //메세지 종류:심플메일메세지 , 마임메세지
      
      simpleMailMessage.setTo(arrReceiver);
      simpleMailMessage.setSubject("Spring Boot Mail Test");
      simpleMailMessage.setText("스프링에서 메일 보내기 테스트");
      //SimpleMailMessage를 사용하여 html 을 전달하더라도 수신자의 화면에는 html이 해석되지 않음
      simpleMailMessage.setText("<a href='/mail/auth/"+ createRandomStr()+"'>인증</a>");
                                                          //유일한 인증문자열을 생성하는 자바 메소드
      sender.send(simpleMailMessage);
      return true;
   }
   
   private String createRandomStr()
   {
      UUID randomUUID = UUID.randomUUID();
      return randomUUID.toString().replaceAll("-", "");
                                      //"-"를 없앰
   }
   public boolean sendMimeMessage()
   {
      MimeMessage mimeMessage = sender.createMimeMessage();
 
      try {
         InternetAddress[] addressTo = new InternetAddress[1];
         addressTo[0] = new InternetAddress("메일 주소 입력");
 
         mimeMessage.setRecipients(Message.RecipientType.TO, addressTo);
 
         mimeMessage.setSubject("마임 메시지(Text) 테스트");
         
         mimeMessage.setContent("This is mimemessage", "text/plain;charset=utf-8");
                                                     //text/html or text/plain
         
         sender.send(mimeMessage);
         return true;
      } catch (MessagingException e) {
         log.error("에러={}", e);
      }
      return false;
   }
   
   public Map<String, Object> sendHTMLMessage(String sid)
   {
      MimeMessage mimeMessage = sender.createMimeMessage();
      String code = createRandomStr();
      Map<String, Object> map = new HashMap<>();
 
      try {
         InternetAddress[] addressTo = new InternetAddress[1];
         addressTo[0] = new InternetAddress("메일 주소 입력");
 
         mimeMessage.setRecipients(Message.RecipientType.TO, addressTo);
 
         mimeMessage.setSubject("마임 메시지(HTML) 테스트");
         
         mimeMessage.setContent("<a href='http://localhost/mail/auth/"+code+'/'+sid+"'>메일주소 인증</a>", "text/html;charset=utf-8");
         
         sender.send(mimeMessage);
         
         map.put("code", code);
         map.put("sent", true);
         return map;
      } catch (MessagingException e) {
         log.error("에러={}", e);
      }
      map.put("sent", false);
      return map;
   }
   
   public boolean sendAttachMail()
   {
      MimeMessage mimeMessage = sender.createMimeMessage();
      
      Multipart multipart = new MimeMultipart();
 
      try {
         InternetAddress[] addressTo = new InternetAddress[1];
         addressTo[0] = new InternetAddress("메일 주소 입력");
 
         mimeMessage.setRecipients(Message.RecipientType.TO, addressTo);
 
         mimeMessage.setSubject("마임 메시지(첨부파일) 테스트");
         
         // Fill the message 바디 파트를 만들어야한다.
         BodyPart messageBodyPart = new MimeBodyPart();
 
         messageBodyPart.setContent("<a href='http://localhost/mail/auth/abc123'>메일주소 인증</a>", "text/html;charset=utf-8");
         
         multipart.addBodyPart(messageBodyPart);
          
         // Part two is attachment
         messageBodyPart = new MimeBodyPart();
         File file = new File("C:/test/첨부.txt");
         FileDataSource fds = new FileDataSource(file);
         messageBodyPart.setDataHandler(new DataHandler(fds));
         
         String fileName = fds.getName();
         messageBodyPart.setFileName(fileName);
         
         multipart.addBodyPart(messageBodyPart);
          
         // Put parts in message
         mimeMessage.setContent(multipart);
         
         sender.send(mimeMessage);
         
         return true;
      }catch(Exception ex) {
         log.error("에러={}", ex);
      }
      return false;
   }
   public Map<String,Object> checkMessage(HttpSession session)
   {
      MimeMessage mimeMessage = sender.createMimeMessage();
 
      try {
         InternetAddress[] addressTo = new InternetAddress[1];
         addressTo[0] = new InternetAddress("메일 주소 입력");
 
         mimeMessage.setRecipients(Message.RecipientType.TO, addressTo);
 
         mimeMessage.setSubject("마임 메시지(HTML) 테스트");
         String str = createRandomStr();//유일한 인증문자열을 생성하는 자바 메소드
         
         mimeMessage.setContent("<a href='http://localhost/mail/auth/"+str+"'>메일주소 인증</a>", "text/html;charset=utf-8");
         
         sender.send(mimeMessage);
         session.setAttribute("code", str);
         Map<String,Object> map = new HashMap<>();
         map.put("result", true);
         map.put("code",str);
         return map;
      } catch (MessagingException e) {
         log.error("에러={}", e);
      }
 
      return null;
   }
   public boolean checkMessageResult(String code,HttpSession session)
   {
       log.info("코드"+session.getAttribute("code"));
       if(session.getAttribute("code").equals(code)) {
           
           return true;
       }
       return false;
   }
}