INNER CODE UNIT · Python

Menu

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

class Menu:

    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()

View source record →