asciinema!

utils/asciinema.py

@@ -0,0 +1,123 @@
+#!/usr/bin/env python
+
+import getopt
+import json
+import os
+import sys
+import struct
+
+OP_OPEN, OP_CLOSE, OP_WRITE, OP_EXEC = 1, 2, 3, 4
+TYPE_INPUT, TYPE_OUTPUT, TYPE_INTERACT = 1, 2, 3
+
+COLOR_INTERACT = '\033[36m'
+COLOR_INPUT = '\033[33m'
+COLOR_RESET = '\033[0m'
+
+def playlog(fd, settings):
+
+    thelog = {}
+    thelog['version'] = 1
+    thelog['width'] = 80
+    thelog['height'] = 24
+    thelog['duration'] = 0.0
+    thelog['command'] = "/bin/bash"
+    thelog['title'] = "Cowrie Recording"
+    theenv = {}
+    theenv['TERM'] = "xterm256-color"   
+    theenv['SHELL'] = "/bin/bash"
+    thelog["env"] = theenv
+    stdout = []
+    thelog["stdout"] = stdout
+ 
+    ssize = struct.calcsize('<iLiiLL')
+
+    currtty, prevtime, prefdir = 0, 0, 0
+    sleeptime = 0.0
+
+    color = None
+
+    while 1:
+        try:
+            (op, tty, length, dir, sec, usec) = \
+                struct.unpack('<iLiiLL', fd.read(ssize))
+            data = fd.read(length)
+        except struct.error:
+            break
+
+        if currtty == 0: currtty = tty
+
+        if str(tty) == str(currtty) and op == OP_WRITE:
+            # the first stream seen is considered 'output'
+            if prefdir == 0:
+                prefdir = dir
+            if dir == TYPE_INTERACT:
+                color = COLOR_INTERACT
+            elif dir == TYPE_INPUT:
+                color = COLOR_INPUT
+            if dir == prefdir:
+                curtime = float(sec) + float(usec) / 1000000
+                if prevtime != 0:
+                    sleeptime = curtime - prevtime
+                prevtime = curtime
+                if settings['colorify'] and color:
+                    sys.stdout.write(color)
+
+                thedata = [sleeptime, data]
+                thelog['duration'] = curtime
+                stdout.append(thedata)
+                
+                if settings['colorify'] and color:
+                    sys.stdout.write(COLOR_RESET)
+                    color = None
+
+        elif str(tty) == str(currtty) and op == OP_CLOSE:
+            break
+
+    if settings['output'] == "":
+        json.dump(thelog, sys.stdout, indent=4)
+    else:
+        with open(settings['output'], "w") as outfp:
+            json.dump(thelog, outfp, indent=4)
+
+
+def help(verbose=False):
+
+    print 'usage: %s [-c] [-o output] <tty-log-file> <tty-log-file>...' % \
+        os.path.basename(sys.argv[0])
+
+    if verbose:
+        print '  -c             colorify the output based on what streams are being received'
+        print '  -h             display this help'
+        print '  -o             write to the specified output file'
+
+
+if __name__ == '__main__':
+
+    settings = {
+        'colorify':     0,
+        'output': ""
+        }
+
+    try:
+        optlist, args = getopt.getopt(sys.argv[1:], 'hco:' )
+    except getopt.GetoptError, error:
+        sys.stderr.write( '{}: {}\n'.format(sys.argv[0], error))
+        help()
+        sys.exit(1)
+
+    for o, a in optlist:
+        if o == '-h': help()
+        if o == '-c': settings['colorify'] = True
+        if o == '-o': settings['output'] = a
+
+    if len(args)<1:
+        help()
+        sys.exit(1)
+
+    for logfile in args:
+        try:
+            logfd = open(logfile, 'rb')
+            playlog(logfd, settings)
+        except IOError as e:
+            sys.stderr.write( "{}: {}\n".format(sys.argv[0], e))
+