@@ -34,15 +34,18 @@ def on_login(self, context, connection):
3434 if "ssh" in context .protocol :
3535 search_aws_creds_files_payload = f"find { self .search_path_linux } -type f -name credentials -o -name credentials.bk -o -name config.bk -o -name config"
3636 search_aws_creds_files_cmd = f'/bin/bash -c "{ search_aws_creds_files_payload } "'
37- search_aws_creds_files_output = connection .execute (search_aws_creds_files_cmd )
38- context .log .highlight (f"The following files were found: { search_aws_creds_files_output } " )
37+ output = connection .execute (search_aws_creds_files_cmd )
3938 else :
4039 # search for aws_credentials-related files on windows systems
4140 search_aws_creds_files_payload_win = f"Get-ChildItem -Path { self .search_path_win } -Recurse -Force -Include ('credentials','credentials.bk','config','config.bk') -ErrorAction SilentlyContinue | Select FullName -ExpandProperty FullName"
4241 search_aws_creds_files_cmd_win = f'powershell.exe "{ search_aws_creds_files_payload_win } "'
4342 # Somehow wmiexec retrieves bugged output (smb), removing it from the list
4443 if "smb" in context .protocol : # noqa: SIM108
45- search_aws_creds_files_output_win = connection .execute (search_aws_creds_files_cmd_win , True , methods = ["atexec" , "smbexec" , "mmcexec" ])
44+ output = connection .execute (search_aws_creds_files_cmd_win , True , methods = ["wmiexec" , "atexec" , "smbexec" , "mmcexec" ])
4645 else :
47- search_aws_creds_files_output_win = connection .execute (search_aws_creds_files_cmd_win , True )
48- context .log .highlight (f"The following files were found: { search_aws_creds_files_output_win } " )
46+ output = connection .execute (search_aws_creds_files_cmd_win , True )
47+
48+ if output :
49+ context .log .success ("The following files were found:" )
50+ for line in output .splitlines ():
51+ context .log .highlight (line .rstrip ())
0 commit comments