세션과 쿠키 분석

# ⚠️ 이 코드는 허가된 환경에서만 사용하세요.
import requests

# 단일 요청 — 매번 새 연결, 쿠키 리셋
r = requests.get('https://httpbin.org/cookies/set/visited/yes')
r2 = requests.get('https://httpbin.org/cookies')  # 쿠키 없음

# Session — 쿠키 자동 유지 + 연결 재사용 (성능 ↑)
s = requests.Session()
s.headers.update({'User-Agent': 'WhiteHat/1.0'})

s.get('https://httpbin.org/cookies/set/sessionid/abc123')
r = s.get('https://httpbin.org/cookies')
print(r.json())  # {'cookies': {'sessionid': 'abc123'}}

def analyze_cookies(url: str) -> list[dict]:
    """Set-Cookie 헤더의 보안 플래그를 분석."""
    s = requests.Session()
    r = s.get(url, timeout=5, allow_redirects=False)

    results = []
    for cookie in s.cookies:
        results.append({
            'name': cookie.name,
            'domain': cookie.domain,
            'path': cookie.path,
            'secure': cookie.secure,
            'httponly': cookie._rest.get('HttpOnly') is not None,
            'samesite': cookie._rest.get('SameSite', 'None'),
            'expires': cookie.expires,
        })
    return results

# DVWA(Damn Vulnerable Web App) 같은 학습용 환경에서만!
def auto_login(login_url: str, username: str, password: str) -> requests.Session:
    s = requests.Session()

    # 1. 로그인 페이지 GET → CSRF 토큰 추출
    r = s.get(login_url, timeout=5)
    # 실제로는 BeautifulSoup으로 <input name="csrf"> 찾기 (다음 챕터)

    # 2. 로그인 POST
    r = s.post(login_url, data={
        'username': username,
        'password': password,
        # 'csrf_token': extracted_token,  # 다음 챕터에서
    }, timeout=5, allow_redirects=True)

    # 3. 로그인 성공 여부 판단
    if 'Logout' in r.text or r.url.endswith('/dashboard'):
        return s  # 세션 반환 → 이 세션으로 인증 필요한 페이지 접근
    raise PermissionError('로그인 실패')

# 사용 (DVWA 로컬 실행 가정)
# session = auto_login('http://localhost/login.php', 'admin', 'password')
# r = session.get('http://localhost/admin/users')

# 공격 시나리오
"""
1. 공격자가 세션 ID를 미리 가짐: sessionid=ATTACKER123
2. 피해자에게 그 세션 ID로 로그인하게 만듦
3. 피해자가 로그인하면 → 그 세션이 인증된 상태가 됨
4. 공격자가 같은 세션으로 접근 → 피해자 행세

방어: 로그인 직후 반드시 세션 ID를 새로 발급해야 함
"""

def detect_session_renewal(login_url: str, creds: dict) -> bool:
    """로그인 전후 세션 ID가 바뀌는지 점검."""
    s = requests.Session()
    s.get(login_url)
    before = s.cookies.get('sessionid') or s.cookies.get('PHPSESSID')

    s.post(login_url, data=creds)
    after = s.cookies.get('sessionid') or s.cookies.get('PHPSESSID')

    return before != after  # True여야 안전