From dfb2f28f903bbc63dae61619b167b3e02ab3b1b5 Mon Sep 17 00:00:00 2001 From: Mitul16 Date: Mon, 26 Jul 2021 11:18:42 +0530 Subject: [PATCH] Added 'PlatformError' to PrivateKey implant 'trigger' method --- pwncat/facts/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pwncat/facts/__init__.py b/pwncat/facts/__init__.py index 7601294..1cf7e9e 100644 --- a/pwncat/facts/__init__.py +++ b/pwncat/facts/__init__.py @@ -17,6 +17,7 @@ import pwncat from pwncat.db import Fact from pwncat.channel import ChannelError from pwncat.modules import ModuleFailed +from pwncat.platform import PlatformError from pwncat.facts.tamper import ( # noqa: F401 Tamper, CreatedFile, @@ -369,7 +370,7 @@ class PrivateKey(Implant): user=user.name, identity=filp.name, ) - except ChannelError as exc: + except (ChannelError, PlatformError) as exc: manager.log( f"[yellow]warning[/yellow]: {self.source} implant failed; removing implant types." )