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, 3*start_address, 100);
  42
  43         QGraphicsView * view = new QGraphicsView(scene);
  44         view->setDragMode(QGraphicsView::ScrollHandDrag);
  45         addTab(view, "CFG");
  46
  47         // Listing
  48         QTableWidget * t = new QTableWidget();
  49         t->setColumnCount(3);
  50         t->horizontalHeader()->setSectionResizeMode(QHeaderView::ResizeToContents);
  51
  52         addTab(t, "Listing");
  53 }
  54
  55
  56
  57
  58 namespace {
  59         BasicBlockWidget *
  60         local__add_basic_block(BasicBlock * block,
  61                                Mainwindow * mainwindow, InformationManager * manager,
  62                                std::map<uint64_t, BasicBlockWidget*>& known_blocks,
  63                                CFGScene * scene, uint64_t starty, uint64_t startx) {
  64
  65                 decltype(known_blocks.begin()) old;
  66                 if (!block) return NULL;
  67                 if ((old = known_blocks.find(block->getStartAddress())) != known_blocks.end())
  68                         return old->second;
  69
  70                 std::stringstream s;
  71                 s << "BLOCK_" << std::hex << block->getStartAddress()
  72                   << "_" << block->getEndAddress();
  73                 BasicBlockWidget * widget = new BasicBlockWidget(s.str().c_str(),
  74                                                                  block, mainwindow);
  75
  76                 known_blocks.insert(std::make_pair(block->getStartAddress(), widget));
  77
  78                 scene->addItem(widget);
  79                 widget->setFlag(QGraphicsItem::ItemIsMovable, true);
  80                 widget->moveBy(100*startx, 3*block->getStartAddress() - starty);
  81
  82                 BasicBlockWidget *tmp, *nextl(NULL), *nextr(NULL);
  83                 BasicBlock * tmpblock;
  84                 if (block->getNextBlock(0) != 0) {
  85                         int xshift = 0;
  86                         if (block->getNextBlock(1) != 0)
  87                                 xshift = 1;
  88                         tmpblock = manager->getBasicBlock(block->getNextBlock(0));
  89                         tmp = local__add_basic_block(tmpblock, mainwindow, manager,
  90                                                      known_blocks,
  91                                                      scene, starty, startx+xshift);
  92                         nextl = tmp;
  93                         tmp->addPrevious(widget);
  94                 }
  95                 if (block->getNextBlock(1) != 0) {
  96                         tmpblock = manager->getBasicBlock(block->getNextBlock(1));
  97                         tmp = local__add_basic_block(tmpblock, mainwindow, manager,
  98                                                      known_blocks,
  99                                                      scene, starty, startx-1);
 100                         nextr = tmp;
 101                         tmp->addPrevious(widget);
 102                 }
 103                 widget->addNext(nextl, nextr);
 104                 return widget;
 105         }
 106 }