1
0
mirror of https://github.com/calebstewart/pwncat.git synced 2024-11-24 01:25:37 +01:00

fix: Minor colored output fix to comply with rest of the code

This commit is contained in:
Mitul16 2022-02-16 15:57:18 +05:30
parent 03456e51cb
commit 0e79b87f56
No known key found for this signature in database
GPG Key ID: 5C5BBBE501A9BA25

View File

@ -41,7 +41,7 @@ class AuthorizedKeyImplant(PrivateKey):
user = session.find_user(uid=self.uid) user = session.find_user(uid=self.uid)
if current_user.id != self.uid and current_user.id != 0: if current_user.id != self.uid and current_user.id != 0:
raise ModuleFailed(f"must be root or {user.name}") raise ModuleFailed(f"must be [blue]root[/blue] or [blue]{user.name}[/blue]")
# Ensure the directory exists # Ensure the directory exists
homedir = session.platform.Path(user.home) homedir = session.platform.Path(user.home)
@ -93,10 +93,10 @@ class Module(ImplantModule):
yield Status("verifying user permissions") yield Status("verifying user permissions")
current_user = session.current_user() current_user = session.current_user()
if user != "__pwncat_current__" and current_user.id != 0: if user != "__pwncat_current__" and current_user.id != 0:
raise ModuleFailed("only root can install implants for other users") raise ModuleFailed("only [blue]root[/blue] can install implants for other users")
if not os.path.isfile(key): if not os.path.isfile(key):
raise ModuleFailed(f"private key {key} does not exist") raise ModuleFailed(f"private key [bleu]{key}[/blue] does not exist")
try: try:
yield Status("reading public key") yield Status("reading public key")
@ -119,7 +119,7 @@ class Module(ImplantModule):
for implant in session.run("enumerate", types=["implant.*"]): for implant in session.run("enumerate", types=["implant.*"]):
if implant.source == self.name and implant.uid == user_info.uid: if implant.source == self.name and implant.uid == user_info.uid:
raise ModuleFailed( raise ModuleFailed(
f"{self.name} already installed for {user_info.name}" f"[blue]{self.name}[/blue] already installed for [blue]{user_info.name}[/blue]"
) )
# Ensure the directory exists # Ensure the directory exists