src/gui/widgets/FunctionWidget.cxx

   1 #include "FunctionWidget.hxx"
   2
   3 #include "core/Function.hxx"
   4 #include "core/BasicBlock.hxx"
   5 #include "core/InformationManager.hxx"
   6
   7 #include "gui/widgets/CFGScene.hxx"
   8 #include "gui/Mainwindow.hxx"
   9
  10 namespace {
  11         BasicBlockWidget *
  12         local__add_basic_block(BasicBlock * block,
  13                                Mainwindow * mainwindow, InformationManager * manager,
  14                                std::map<uint64_t, BasicBlockWidget*>& known_blocks,
  15                                CFGScene * scene, uint64_t starty, uint64_t startx);
  16 }
  17
  18 FunctionWidget::FunctionWidget(Function* function, Mainwindow* mainwindow)
  19         : function(function)
  20         , mainwindow(mainwindow)
  21         , logger(log4cxx::Logger::getLogger("gui.Mainwindow")) {
  22
  23         // CFG
  24         CFGScene * scene = new CFGScene;
  25         InformationManager* manager = function->getManager();
  26
  27         BasicBlock * block = manager->getBasicBlock(function->getStartAddress());
  28         LOG4CXX_DEBUG(logger, "Building widget for functionction " << function->getName());
  29         for (auto i : function->blocks()) {
  30                 LOG4CXX_DEBUG(logger, "Functionction contains Block " << i.second->getName());
  31         }
  32
  33         uint64_t start_address(std::numeric_limits<uint64_t>::max());
  34         for (auto b : function->blocks()) {
  35                 if (b.first < start_address)
  36                         start_address = b.first;
  37         }
  38
  39         std::map<uint64_t, BasicBlockWidget*> _blocks;
  40         local__add_basic_block(block, mainwindow,
  41                                manager, _blocks, scene, start_address, 100);
  42
  43         QGraphicsView * view = new QGraphicsView(scene);
  44         addTab(view, "CFG");
  45
  46         // Listing
  47         QTableWidget * t = new QTableWidget();
  48         t->setColumnCount(3);
  49         t->horizontalHeader()->setSectionResizeMode(QHeaderView::ResizeToContents);
  50
  51         addTab(t, "Listing");
  52 }
  53
  54
  55
  56
  57 namespace {
  58         BasicBlockWidget *
  59         local__add_basic_block(BasicBlock * block,
  60                                Mainwindow * mainwindow, InformationManager * manager,
  61                                std::map<uint64_t, BasicBlockWidget*>& known_blocks,
  62                                CFGScene * scene, uint64_t starty, uint64_t startx) {
  63
  64                 decltype(known_blocks.begin()) old;
  65                 if (!block) return NULL;
  66                 if ((old = known_blocks.find(block->getStartAddress())) != known_blocks.end())
  67                         return old->second;
  68
  69                 std::stringstream s;
  70                 s << "BLOCK_" << std::hex << block->getStartAddress()
  71                   << "_" << block->getEndAddress();
  72                 BasicBlockWidget * widget = new BasicBlockWidget(s.str().c_str(),
  73                                                                  block, mainwindow);
  74
  75                 known_blocks.insert(std::make_pair(block->getStartAddress(), widget));
  76
  77                 scene->addItem(widget);
  78                 widget->setFlag(QGraphicsItem::ItemIsMovable, true);
  79                 widget->moveBy(100*startx, block->getStartAddress() - starty);
  80
  81                 BasicBlockWidget *tmp, *nextl(NULL), *nextr(NULL);
  82                 BasicBlock * tmpblock;
  83                 if (block->getNextBlock(0) != 0) {
  84                         int xshift = 0;
  85                         if (block->getNextBlock(1) != 0)
  86                                 xshift = 1;
  87                         tmpblock = manager->getBasicBlock(block->getNextBlock(0));
  88                         tmp = local__add_basic_block(tmpblock, mainwindow, manager,
  89                                                      known_blocks,
  90                                                      scene, starty, startx+xshift);
  91                         nextl = tmp;
  92                         tmp->addPrevious(widget);
  93                 }
  94                 if (block->getNextBlock(1) != 0) {
  95                         tmpblock = manager->getBasicBlock(block->getNextBlock(1));
  96                         tmp = local__add_basic_block(tmpblock, mainwindow, manager,
  97                                                      known_blocks,
  98                                                      scene, starty, startx-1);
  99                         nextr = tmp;
 100                         tmp->addPrevious(widget);
 101                 }
 102                 widget->addNext(nextl, nextr);
 103                 return widget;
 104         }
 105 }