bootstrap_visitor.py 2.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364
  1. import string
  2. from primitives_visitor import PrimitivesVisitor, Action
  3. class BootstrapVisitor(PrimitivesVisitor):
  4. def __init__(self, args):
  5. PrimitivesVisitor.__init__(self, args)
  6. self.main = "--main" in args
  7. for f in args:
  8. if "--prepend:" in f:
  9. self.prepend_name = f.split(":", 1)[1]
  10. break
  11. else:
  12. self.prepend_name = "auto"
  13. def rename(self, name):
  14. if name == "initial_IP":
  15. return "%s_%s" % (self.prepend_name, name)
  16. else:
  17. return "%s_%s" % (self.prepend_name, name) if isinstance(name, int) else name
  18. def dump(self):
  19. link_id = 0
  20. if self.main:
  21. call = self.value(Action("call"))
  22. access = self.value(Action("access"))
  23. resolve = self.value(Action("resolve"))
  24. main = self.value("__main")
  25. self.dict(resolve, "var", main)
  26. self.dict(access, "var", resolve)
  27. self.dict(call, "func", access)
  28. self.dict(self.last_instruction, "next", call)
  29. output = []
  30. for t, data in self.output:
  31. if t == "N":
  32. output.append("Node %s_%s()\n" % (self.prepend_name, data))
  33. elif t == "V":
  34. name, value = data
  35. name = name if self.first != name else "initial_IP"
  36. output.append("Node %s_%s(%s)\n" % (self.prepend_name, name, value))
  37. elif t == "D":
  38. source, value, target = data
  39. source = source if self.first != source else "initial_IP"
  40. target = target if self.first != target else "initial_IP"
  41. source = self.rename(source)
  42. target = self.rename(target)
  43. linkname = "%s_%s_%s" % (source, link_id, target)
  44. link_id += 1
  45. output.append("Edge _%s_0(%s, %s)\n" % (linkname, source, target))
  46. output.append("Node _%s_2(%s)\n" % (linkname, value))
  47. output.append("Edge _%s_1(_%s_0, _%s_2)\n" % (linkname, linkname, linkname))
  48. elif t == "E":
  49. name, source, target = data
  50. source = source if self.first != source else "initial_IP"
  51. target = target if self.first != target else "initial_IP"
  52. name = self.rename(name)
  53. source = self.rename(source)
  54. target = self.rename(target)
  55. output.append("Edge _%s(_%s, _%s)\n" % (name, source, target))
  56. return ''.join(output)