-#!/usr/bin/env python
+#!/usr/bin/env python3
"""
This program parses the output from pcap_compile() to visualize the CFG after
import sys, os
import string
-import subprocess
-import json
html_template = string.Template("""
<html>
<head>
- <title>BPF compiler optimization phases for $expr </title>
+ <title>BPF compiler optimization phases for "${expr_html}"</title>
<style type="text/css">
.hc {
/* half width container */
<script type="text/javascript" src="https://ajax.googleapis.com/ajax/libs/jquery/1.10.2/jquery.min.js"/></script>
<!--script type="text/javascript" src="./jquery.min.js"/></script-->
<script type="text/javascript">
- var expr = '$expr';
+ var expr = '${expr_json}';
var exprid = 1;
- var gcount = $gcount;
- var logs = JSON.parse('$logs');
+ var gcount = ${gcount};
+ var logs = JSON.parse('${logs}');
logs[gcount] = "";
var leftsvg = null;
</head>
<body style="width: 96%">
<div>
- <h1>$expr</h1>
+ <h1>${expr_html}</h1>
<div style="text-align: center;">
<button id="backward" type="button"><<</button>
""")
def write_html(expr, gcount, logs):
- logs = map(lambda s: s.strip().replace("\n", "<br/>"), logs)
+ import html
+ import json
- global html_template
- html = html_template.safe_substitute(expr=expr.encode("string-escape"), gcount=gcount, logs=json.dumps(logs).encode("string-escape"))
- with file("expr1.html", "wt") as f:
- f.write(html)
+ # In the Python 2.7 version this used to be str.encode('string-escape'),
+ # which was a normal string, but in Python 3 the "string_escape" encoding
+ # no longer exists and even with the "unicode_escape" encoding encode()
+ # always returns a binary string. So let's just escape the single quotes
+ # here and hope the result is a valid JavaScript string literal.
+ def encode(s):
+ return s.replace("'", "\'")
+
+ mapping = {
+ 'expr_html': html.escape(expr),
+ 'expr_json': encode(expr),
+ 'gcount': gcount,
+ 'logs': encode(json.dumps([s.strip().replace("\n", "<br/>") for s in logs])),
+ }
+ with open("expr1.html", "wt") as f:
+ f.write(html_template.safe_substitute(mapping))
def render_on_html(infile):
+ import subprocess
+
expr = None
gid = 1
log = ""
if indot == 2:
try:
- p = subprocess.Popen(['dot', '-Tsvg'], stdin=subprocess.PIPE, stdout=subprocess.PIPE)
+ svg=subprocess.check_output(['dot', '-Tsvg'], input=dot, universal_newlines=True)
except OSError as ose:
- print "Failed to run 'dot':", ose
- print "(Is Graphviz installed?)"
- exit(1)
+ print("Failed to run 'dot':", ose)
+ print("(Is Graphviz installed?)")
+ return False
+ except subprocess.CalledProcessError as cpe:
+ print("Got an error from the 'dot' process: ", cpe)
+ return False
- svg = p.communicate(dot)[0]
- with file("expr1_g%03d.svg" % (gid), "wt") as f:
+ with open("expr1_g%03d.svg" % gid, "wt") as f:
f.write(svg)
logs.append(log)
return True
def run_httpd():
- import SimpleHTTPServer
- import SocketServer
+ import http.server
- class MySocketServer(SocketServer.TCPServer):
- allow_reuse_address = True
- Handler = SimpleHTTPServer.SimpleHTTPRequestHandler
- httpd = MySocketServer(("localhost", 0), Handler)
- print "open this link: http://localhost:%d/expr1.html" % (httpd.server_address[1])
+ httpd = http.server.HTTPServer(("localhost", 0), http.server.SimpleHTTPRequestHandler)
+ print("open this link: http://localhost:%d/expr1.html" % httpd.server_port)
try:
httpd.serve_forever()
except KeyboardInterrupt as e:
import shutil
os.chdir(tempfile.mkdtemp(prefix="visopts-"))
atexit.register(shutil.rmtree, os.getcwd())
- print "generated files under directory: %s" % os.getcwd()
- print " the directory will be removed when this program has finished."
+ print("generated files under directory: %s" % os.getcwd())
+ print(" the directory will be removed when this program has finished.")
if not render_on_html(sys.stdin):
return 1
if __name__ == "__main__":
if '-h' in sys.argv or '--help' in sys.argv:
- print __doc__
+ print(__doc__)
exit(0)
exit(main())
projects / libpcap / commitdiff
