[이펙티브 자바] Item08 완벽공략. Finalizer 공격

 

item08. finalizer와 cleaner 사용을 피하라.  

" p42. Finalizer 공격"

 

 

https://hyejin.tistory.com/1011

[이펙티브 자바] Item8. finalizer와 cleaner 사용을 피하라.

여기서 finalizer를 사용하면 안되는 이유 중 하나로 "finalizer를 사용한 클래스는 finalizer 공격에 노출되어 심각한 보안 문제를 일으킬 수 있다" 라고 했는데 이때 finalizer 공격은 생성자나 직렬화 과정에서 예외가 발생하면, 이 생성되다 만 객체에서 악의적인 하위 클래스의 finalizer가 수행될 수 있게 된다고 한다. 

 

public class Account {
    private String accountId;

    public Account(String accountId) {
        this.accountId = accountId;

        if (accountId.equals("aaa")) {
            throw new IllegalArgumentException("aaa의 계정을 막습니다.");
        }
    }

    public void transfer(BigDecimal amount, String to) {
        System.out.printf("transfer %f from %s to %s", amount, accountId, to);
    }
 }

이렇게 aaa의 계정을 막는 Account 클래스가 있고, aaa는 다른 사람에게 계좌 이체를 못하게 한다. 

 

 

class AccountTest {

    @Test
    void 일반_계정() {
        Account account = new Account("hyejin");
        account.transfer(BigDecimal.valueOf(10.4), "hello");
    }

    @Test
    void aaa_계정() {
        Account account = new Account("aaa");
        account.transfer(BigDecimal.valueOf(10.4), "hello~");
    }
  }

따라서 일반 계정은 테스트 성공하는 반면, aaa 계정은 실패한다. 

 

 

이렇게 보면 aaa는 계정이 막혀 계좌 이체를 할 수 없는 것 같아 보이지만 

finalizer 공격을 사용하면 aaa가 계좌 이체할 수 있게 할 수 있다. 

 

public class BrokenAccount extends Account {

    public BrokenAccount(String accountId) {
        super(accountId);
    }

    @Override
    protected void finalize() throws Throwable {
        this.transfer(BigDecimal.valueOf(100), "hyejin");
    }
}

Account를 상속받는 클래스를 하나 만들고. finalize를 오버라이딩 해서 여기서 transfer 메서드를 호출한다. 

 

@Test
void broken_aaa_계정() throws InterruptedException {
    Account account = null;
    try {
        account = new BrokenAccount("aaa");
    } catch (Exception e) {
        System.out.println("이러면???");
    }
    System.gc();
    Thread.sleep(3000L); // gc하기 위한 시간
}

그리고 우선 null로 만들고, aaa 계정은 IllegalArgumentException을 발생시키므로 try-catch로 잡아주고, gc를 시작시키면 

finalizer가 동작하면서 transer 를 실행한다..! 

 

이런 finalizer 공격을 막기 위해서는 그냥 Account 클래스 자체를 final로 하여 상속받지 못하게 할 수도 있으며, 

근데 이 클래스가 상속 구조를 가져야한다면 

public class Account {
    private String accountId;

    public Account(String accountId) {
        this.accountId = accountId;

        if (accountId.equals("aaa")) {
            throw new IllegalArgumentException("aaa의 계정을 막습니다.");
        }
    }

    public void transfer(BigDecimal amount, String to) {
        System.out.printf("transfer %f from %s to %s", amount, accountId, to);
    }

    @Override
    protected  final void finalize() throws Throwable {
        // 오버라이딩을 허용하지 않도록 막아둠
    }
}

이렇게 Account 클래스에서 finalize를 오버라이딩 못하게 final로 선언해서 막아두면 된다. 

그리고 다시 테스트를 해보면 이제 aaa가 계좌 이체를 하지 못하는 것을 확인할 수 있다.