def run(self):
self.im = Image.open(self.args.file)
if self.file_type == "png":
self.find_png_type()
print(f"{colored('[+]', 'green')} Starting LStegB program...")
if self.args.all:
print(f"{colored('[+]', 'green')} Starting basic process...")
basic = Basic(self.im, self.bits_quantity, self.channels,
self.steps)
basic.bruteforce()
print(f"{colored('[-]', 'red')} Ending basic process...")
print(f"{colored('[+]', 'green')} Starting pit process...")
if self.im.mode != "RGB":
print(colored("PIT only supports RGB files", "red"))
else:
pit = Pit(self.im, self.steps)
pit.bruteforce()
print(f"{colored('[-]', 'red')} Ending pit process...")
else:
if self.args.basic:
print(f"{colored('[+]', 'green')} Starting basic process...")
basic = Basic(self.im, self.bits_quantity, self.channels,
self.steps)
basic.bruteforce()
print(f"{colored('[-]', 'red')} Ending basic process...")
if self.args.pit:
print(f"{colored('[+]', 'green')} Starting pit process...")
if self.im.mode != "RGB":
print(colored("PIT only supports RGB files", "red"))
else:
pit = Pit(self.im, self.steps)
pit.bruteforce()
print(f"{colored('[-]', 'red')} Ending pit process...")
print(f"{colored('[-]', 'red')} Ending LStegB program...")
