diff --git a/qa/Makefile.am b/qa/Makefile.am index 5cfd58bded..b299e41959 100644 --- a/qa/Makefile.am +++ b/qa/Makefile.am @@ -1,2 +1,2 @@ SUBDIRS = coccinelle -EXTRA_DIST = wirefuzz.pl +EXTRA_DIST = wirefuzz.pl sock_to_gzip_file.py diff --git a/qa/sock_to_gzip_file.py b/qa/sock_to_gzip_file.py new file mode 100755 index 0000000000..4c51782e08 --- /dev/null +++ b/qa/sock_to_gzip_file.py @@ -0,0 +1,57 @@ +#!/usr/bin/python +#I love the python Power Glove. It's so bad! +#Usage: sudo -u suricata ./sock_to_gzip_file.py --output-file="http.log.gz" --listen-sock="http.log.sock" + +import socket,os +import gzip +import sys +from optparse import OptionParser + +if __name__ == "__main__": + parser = OptionParser() + #Path to the socket + parser.add_option("--listen-sock", dest="lsock", type="string", help="Path to the socket we will listen on.") + #Path to gzip file we will write + parser.add_option("--output-file", dest="output", type="string", help="Path to file name to output gzip file we will write to.") + + #parse the opts + (options, args) = parser.parse_args() + + options.usage = "example: sudo -u suricata ./sock_to_gzip_file.py --output-file=\"http.log.gz\" --listen-sock=\"http.log.sock\"\n" + #Open the output file + if options.output: + try: + f = gzip.open(options.output, 'wb') + except Exception,e: + print("Error: could not open output file %s:\n%s\n", options.output, e) + sys.exit(-1) + else: + print("Error: --output-file option required and was not specified\n%s" % (options.usage)) + sys.exit(-1) + + #Open our socket and bind + if options.lsock: + if os.path.exists(options.lsock): + try: + os.remove(options.lsock) + except OSError: + pass + try: + s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) + s.bind(options.lsock) + s.listen(1) + conn, addr = s.accept() + except Exception,e: + print("Error: Failed to bind socket %s\n%s\n", options.lsock, e) + sys.exit(-1) + else: + print("Error: --listen-sock option required and was not specified\n%s" % (options.usage)) + sys.exit(-1) + + #Read data from the socket and write to the file + while 1: + data = conn.recv(1024) + if not data: break + f.write(data) + conn.close() + f.close()