diff --git a/pupy/modules/getsystem.py b/pupy/modules/getsystem.py index 855d0911..de9f6a06 100644 --- a/pupy/modules/getsystem.py +++ b/pupy/modules/getsystem.py @@ -10,9 +10,10 @@ class GetSystem(PupyModule): dependencies=["pupwinutils.security"] def init_argparse(self): self.arg_parser = PupyArgumentParser(prog="getsystem", description=self.__doc__) + self.arg_parser.add_argument("--prog", default="cmd.exe", help="Change the default process to create/inject into") def run(self, args): with redirected_stdo(self): - proc_pid=self.client.conn.modules["pupwinutils.security"].getsystem() + proc_pid=self.client.conn.modules["pupwinutils.security"].getsystem(prog=args.prog) migrate(self, proc_pid) self.success("got system !")