Skip to content

Commit 0e79b87

Browse files
committed
fix: Minor colored output fix to comply with rest of the code
1 parent 03456e5 commit 0e79b87

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

pwncat/modules/linux/implant/authorized_key.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ def remove(self, session: "pwncat.manager.Session"):
4141
user = session.find_user(uid=self.uid)
4242

4343
if current_user.id != self.uid and current_user.id != 0:
44-
raise ModuleFailed(f"must be root or {user.name}")
44+
raise ModuleFailed(f"must be [blue]root[/blue] or [blue]{user.name}[/blue]")
4545

4646
# Ensure the directory exists
4747
homedir = session.platform.Path(user.home)
@@ -93,10 +93,10 @@ def install(self, session: "pwncat.manager.Session", user, key):
9393
yield Status("verifying user permissions")
9494
current_user = session.current_user()
9595
if user != "__pwncat_current__" and current_user.id != 0:
96-
raise ModuleFailed("only root can install implants for other users")
96+
raise ModuleFailed("only [blue]root[/blue] can install implants for other users")
9797

9898
if not os.path.isfile(key):
99-
raise ModuleFailed(f"private key {key} does not exist")
99+
raise ModuleFailed(f"private key [bleu]{key}[/blue] does not exist")
100100

101101
try:
102102
yield Status("reading public key")
@@ -119,7 +119,7 @@ def install(self, session: "pwncat.manager.Session", user, key):
119119
for implant in session.run("enumerate", types=["implant.*"]):
120120
if implant.source == self.name and implant.uid == user_info.uid:
121121
raise ModuleFailed(
122-
f"{self.name} already installed for {user_info.name}"
122+
f"[blue]{self.name}[/blue] already installed for [blue]{user_info.name}[/blue]"
123123
)
124124

125125
# Ensure the directory exists

0 commit comments

Comments
 (0)