From e703a1606471901008938c72900367770afee8f3 Mon Sep 17 00:00:00 2001
From: makefu <root@pigstarter.de>
Date: Mon, 22 Apr 2013 15:17:15 +0200
Subject: update Reaktor - add color translator

---
 Reaktor/IRC/asybot.py               |  8 ++++++--
 Reaktor/IRC/index                   |  2 +-
 Reaktor/IRC/translate_colors.py     | 31 +++++++++++++++++++++++++++++++
 Reaktor/commands/whatweb            |  6 +++---
 Reaktor/repos/consolidate_dns/index |  5 +++--
 Reaktor/repos/whatweb               |  2 +-
 6 files changed, 45 insertions(+), 9 deletions(-)
 create mode 100644 Reaktor/IRC/translate_colors.py

(limited to 'Reaktor')

diff --git a/Reaktor/IRC/asybot.py b/Reaktor/IRC/asybot.py
index df758ed6..2cb533ea 100755
--- a/Reaktor/IRC/asybot.py
+++ b/Reaktor/IRC/asybot.py
@@ -2,7 +2,7 @@
 #
 # //Reaktor/IRC/asybot.py
 #
-
+from translate_colors import translate_colors
 def is_executable(x):
   import os
   return os.path.exists(x) and os.access(x, os.X_OK)
@@ -24,6 +24,9 @@ formatter = logging.Formatter( '%(filename)s: %(levelname)s: %(message)s')
 hdlr.setFormatter(formatter)
 log.addHandler(hdlr)
 
+# s/\x1B\[([0-9]{1,2}(;[0-9]{1,2})?)?[m|K]//g -- removes color codes
+
+
 class asybot(asychat):
   def __init__(self, server, port, nickname, targets, **kwargs):
     asychat.__init__(self)
@@ -110,6 +113,7 @@ class asybot(asychat):
     def PRIVMSG(text):
       for line in self.wrapper.wrap(text):
         msg = 'PRIVMSG %s :%s' % (','.join(params), line)
+        log.info(msg)
         self.push(msg)
         sleep(1)
 
@@ -151,7 +155,7 @@ class asybot(asychat):
           return
         pid = p.pid
         for line in iter(p.stdout.readline,""):
-          PRIVMSG(line)
+          PRIVMSG(translate_colors(line))
           log.debug('%s stdout: %s' % (pid, line)) 
         p.wait()
         elapsed = time() - start
diff --git a/Reaktor/IRC/index b/Reaktor/IRC/index
index 50022ec9..cc2652fe 100755
--- a/Reaktor/IRC/index
+++ b/Reaktor/IRC/index
@@ -3,4 +3,4 @@ set -xeuf
 
 # cd //Reaktor
 cd $(dirname $(readlink -f $0))/..
-host=irc.freenode.net target='#krebsco' python IRC/asybot.py "$@"
+host=irc.freenode.net target='#krebs' python IRC/asybot.py "$@"
diff --git a/Reaktor/IRC/translate_colors.py b/Reaktor/IRC/translate_colors.py
new file mode 100644
index 00000000..bd716618
--- /dev/null
+++ b/Reaktor/IRC/translate_colors.py
@@ -0,0 +1,31 @@
+
+
+COLOR_MAP = {
+    "\x1b[0m" :  "\x0F", # reset
+    "\x1b[37m" : "\x0300",
+    "\x1b[30m" : "\x0301",
+    "\x1b[34m" : "\x0302",
+    "\x1b[32m" : "\x0303",
+    "\x1b[31m" : "\x0304",
+    "\x1b[33m" : "\x0305",
+    "\x1b[35m" : "\x0306",
+    "\x1b[33m" : "\x0307",
+    "\x1b[33m" : "\x0308",
+    "\x1b[32m" : "\x0309",
+    "\x1b[36m" : "\x0310",
+    "\x1b[36m" : "\x0311",
+    "\x1b[34m" : "\x0312",
+    "\x1b[31m" : "\x0313",
+    "\x1b[30m" : "\x0314",
+    "\x1b[37m" : "\x0315",
+    "\x1b[1m" :  "\x02", # bold on
+    "\x1b[22m" : "\x02" # bold off
+  }
+def translate_colors (line):
+  for color,replace in COLOR_MAP.items():
+    line = line.replace(color,replace)
+  return line
+
+if __name__ == "__main__":
+  import sys
+  print (translate_colors(sys.stdin.read()))
diff --git a/Reaktor/commands/whatweb b/Reaktor/commands/whatweb
index 84130d5c..afe20360 100755
--- a/Reaktor/commands/whatweb
+++ b/Reaktor/commands/whatweb
@@ -2,6 +2,6 @@
 #wrapper for WhatWeb
 here=$(dirname `readlink -f $0`)
 whatweb_bin="$here/../repos/whatweb/whatweb"
-[ -e "$whatweb_bin" ] || ( echo "!! Whatweb app does not exist" && exit 1 )
-[ "balls$1" = "balls" ] && ( echo "!! no host given" && exit 1)
-exec $whatweb_bin -a 3 "$1"
+[ ! -e "$whatweb_bin" ] &&  echo "!! Whatweb app does not exist" && exit 1 
+[ -z "${1:-}" ] &&  echo "!! no host given" && exit 1
+exec $whatweb_bin -a 3 "$1" 2>&1
diff --git a/Reaktor/repos/consolidate_dns/index b/Reaktor/repos/consolidate_dns/index
index 1a0dd81e..3dd42fbd 100755
--- a/Reaktor/repos/consolidate_dns/index
+++ b/Reaktor/repos/consolidate_dns/index
@@ -9,7 +9,7 @@ import tempfile
 os.chdir (os.path.dirname (os.path.realpath (sys.argv[0])))
 dnsrecon_enabled = False
 DNSRECON = "../dnsrecon/dnsrecon.py"
-dnsrecon_wordlist="namelist.txt"
+dnsrecon_wordlist="../dnsrecon/namelist.txt"
 silent=open("/dev/null","w")
 gxfr_enabled = False
 GXFR = "../gxfr/gxfr.py"
@@ -42,7 +42,8 @@ else:
 if dnsrecon_enabled:
   dnsrecon_tmp = tempfile.NamedTemporaryFile(delete=False).name
   print ("Starting dnsrecon, this may take some time")
-  p = Popen([DNSRECON,"-d",DOMAIN,"--csv",dnsrecon_tmp,'-D',dnsrecon_wordlist,"-t","brt,srv,axfr","--skip"],stdout=silent,stderr=silent)
+  #print (" ".join([DNSRECON,"-d",DOMAIN,"--csv",dnsrecon_tmp,'-D',dnsrecon_wordlist,"-t","brt,srv,axfr","--skip"]))
+  p = Popen([DNSRECON,"-d",DOMAIN,"--csv",dnsrecon_tmp,'-D',dnsrecon_wordlist,"-t","brt,srv,axfr"] ,stdout=silent,stderr=silent)
   p.wait()
   reader = csv.reader(open(dnsrecon_tmp))
   for row in reader:
diff --git a/Reaktor/repos/whatweb b/Reaktor/repos/whatweb
index daab5f21..0918a0d9 160000
--- a/Reaktor/repos/whatweb
+++ b/Reaktor/repos/whatweb
@@ -1 +1 @@
-Subproject commit daab5f21f13024ee8ec47e88f668c5308d6b59da
+Subproject commit 0918a0d9b75df77f9c3e9eb360b6b22824582a20
-- 
cgit v1.2.3