diff options
| author | Ada Isopods <me@bathynomus.xyz> | 2025-05-14 02:53:11 -0600 |
|---|---|---|
| committer | Ada Isopods <me@bathynomus.xyz> | 2026-10-21 23:04:18 -0600 |
| commit | 22502a6dc24f546227592a20110b39f3647bca39 (patch) | |
| tree | 18e0297fff4bcd2b7c1f97d886e896e918f6f7fb /cowrieder.py | |
| parent | a943e25a79850dc096a5bd3bc2f751470d84239d (diff) | |
don't include colour data if stdout is a pipe
Diffstat (limited to 'cowrieder.py')
| -rw-r--r-- | cowrieder.py | 28 |
1 files changed, 19 insertions, 9 deletions
diff --git a/cowrieder.py b/cowrieder.py index 756f4c7..0518df6 100644 --- a/cowrieder.py +++ b/cowrieder.py @@ -1,6 +1,7 @@ import json import sys import random +import os if not 0 <= 1 < len(sys.argv): print("Provide a path to a Cowrie JSON log file") @@ -22,19 +23,28 @@ with open(filepath, "r") as logfile: filter_ip = "" if 0 <= 2 < len(sys.argv): filter_ip = sys.argv[2] +cc_session = "" +cc_ip = "" +cc_reset = "" +outpipe = os.isatty(1) for session in sessions: ip = session_ips[session] if(ip == filter_ip or not filter_ip): - random.seed(session) - sr = random.randrange(50, 256) - sg = random.randrange(50, 256) - sb = random.randrange(50, 256) - random.seed(ip) - ir = random.randrange(50, 256) - ig = random.randrange(50, 256) - ib = random.randrange(50, 256) - print("\n--- Session \x1b[38;2;{r};{g};{b}m".format(r=sr, g=sg, b=sb) + session + "\x1b[0m from \x1b[38;2;{r};{g};{b}m".format(r=ir, g=ig, b=ib) + session_ips[session] + "\x1b[0m ---") + if(outpipe): + random.seed(session) + sr = random.randrange(50, 256) + sg = random.randrange(50, 256) + sb = random.randrange(50, 256) + random.seed(ip) + ir = random.randrange(50, 256) + ig = random.randrange(50, 256) + ib = random.randrange(50, 256) + cc_reset = "\x1b[0m" + cc_session = "\x1b[38;2;{r};{g};{b}m".format(r=sr, g=sg, b=sb) + cc_ip = "\x1b[38;2;{r};{g};{b}m".format(r=ir, g=ig, b=ib) + + print("\n--- Session " + cc_session + session + cc_reset + " from " + cc_ip + session_ips[session] + cc_reset +" ---") for entry in entries: if(entry['session'] == session): if(entry['message']): |