class ConstructivisticSMGTest(unittest.TestCase):
  def setUp(self):
    self.smg = ConstructivisticSMG()

  def test_addPointer(self):
    pointer = self.smg.addPointer()
    smgPointer = self.smg.getObject(pointer.getId())
    
    self.assertIs(pointer, smgPointer)
    self.assertTrue(smgPointer.isValid())
    self.assertEqual(self.smg.getPointerSize(), smgPointer.getSize())
    
  def test_createPointerTarget(self):
    data_type = (0,4,0,8)
    data_type_size = (2 * self.smg.getPointerSize()) + sum(data_type)

    pointer = self.smg.addPointer()
    node = self.smg.createPointerTarget(pointer, data_type)
    
    self.assertGreaterEqual(node.getSize(), data_type_size)
    
    hvedges = pointer.getHVEdges(offset=0, datatype=data_type)
    self.assertEqual(1, len(hvedges))
    
    value = hvedges[0].getValue()
    
    ptedge = value.getPTEdge()
    
    self.assertIs(ptedge.getObject(), node)
    self.assertEqual(0, ptedge.getOffset())

  def test_setSomeValue(self):
    obj = self.smg.addObject(16)
    edges = obj.getHVEdges()
    self.assertEqual(0, len(edges))

    value = self.smg.setSomeValue(obj, 8, 4)

    edges = obj.getHVEdges(offset=8)
    self.assertEqual(1, len(edges))
    edge = edges[0]
    self.assertIs(obj, edge.getObject())
    self.assertIs(value, edge.getValue())
    
    edges = obj.getHVEdges(datatype=4)
    self.assertEqual(1, len(edges))
    edge = edges[0]
    self.assertIs(obj, edge.getObject())
    self.assertIs(value, edge.getValue())

  def test_setNullPointer(self):
    obj = self.smg.addObject(16)
    self.smg.setNullPointer(obj, 0)

    edges = obj.getHVEdges(offset=0)
    self.assertEqual(1, len(edges))
    edge = edges[0]
    self.assertEquals(edge.getDataType(), self.smg.getPointerSize())
    self.assertIs(edge.getObject(), obj)
    self.assertIs(edge.getValue(), self.smg.getObject(0))
예제 #2
0
def getSMG():
  data_node = (0,0,4)
  smg = ConstructivisticSMG()
  stack_variable = smg.addPointer()
  tree_root = smg.createPointerTarget(stack_variable, data_node)
  
  tree_lvl2_left = smg.createPointerTarget(tree_root, data_node, offset=0)
  tree_lvl2_right = smg.createPointerTarget(tree_root, data_node, offset=8)
  
  tree_lvl3_ll = smg.createPointerTarget(tree_lvl2_left, data_node, offset=0)
  tree_lvl3_lr = smg.createPointerTarget(tree_lvl2_left, data_node, offset=8)
  
  tree_lvl3_rl = smg.createPointerTarget(tree_lvl2_right, data_node, offset=0)
  tree_lvl3_rr = smg.createPointerTarget(tree_lvl2_right, data_node, offset=8)
  
  for node in ( tree_root, tree_lvl2_left, tree_lvl2_right, tree_lvl3_ll,
                tree_lvl3_lr, tree_lvl3_rl, tree_lvl3_rr ):
    smg.setSomeValue(node, 16, 4)

  for node in ( tree_lvl3_ll, tree_lvl3_lr, tree_lvl3_rl, tree_lvl3_rr ):
    smg.setNullPointer(node, 0)
    smg.setNullPointer(node, 8)
  
  return smg