inspectorJ -- JavaTM Profiler

sf project site browse source checkout source

src/inspectorj/toolset/classtoolset.cpp

Go to the documentation of this file.

/***************************************************************************
 *   inspectorJ - java profiler                                            *
 *   Copyright (C) 2007 by James May
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/

#include "inspectorj/toolset/classtoolset.h"
#include <iostream>

namespace inspectorj {
namespace toolset {

using inspectorj::model::ClassTableModel;
using inspectorj::model::ClassTreeModel;
using inspectorj::model::JavaClass;
using inspectorj::jdwp::JDWPCommand;
using inspectorj::client::ConnectionManager;

ClassToolSet::ClassToolSet(ConnectionManager& connectionManager, SessionProfile** prof)
    : connection(connectionManager), profile(prof)
{
    classTreeModel = new ClassTreeModel(this);    
    classTableModel = new ClassTableModel(classes, profile);
    createActions();
    setupConnections();
}

void ClassToolSet::createActions()
{
    tableModelAction = new JDWPAction(this);
    byteCodesAction = new JDWPAction(this);
    updateClassFieldsAction = new JDWPAction(this);
    updateClassMethodsAction = new JDWPAction(this);    
}

void ClassToolSet::setupConnections()
{
    connect(tableModelAction, SIGNAL(triggered(JDWPPacket&)), classTableModel, SLOT (initModel(JDWPPacket&)));
    connect(tableModelAction, SIGNAL(finished()), this, SIGNAL(tableModelReady()));
    connect(byteCodesAction, SIGNAL (triggered(JDWPPacket&)), this, SLOT (getBytecodeString(JDWPPacket&)));    
}

ClassToolSet::~ClassToolSet()
{       
    delete classTableModel;
    classTableModel = 0;
    
    delete tableModelAction;
    tableModelAction = 0;
}

void ClassToolSet::getAllLoadedClasses()
{
    qDeleteAll(classes);
    classes.clear();
    classMap.clear();
    JDWPCommand *cmd = JDWPCommand::newCommand(VM_CMD_ALL_CLASSES);
    connection.sendRequest(cmd, tableModelAction);
}

void ClassToolSet::getClassDetails(const QModelIndex& index)
{   
    classTreeModel->setJavaClass(classes[index.row()]);
    
    //Store a pointer to the class in a map with the refTypeId as the key.
    classMap.insert(classes[index.row()]->getRefTypeId().value, classes[index.row()]);
                                
    //Disconnect any previous signal / slot connections
    disconnect(updateClassFieldsAction, SIGNAL (triggered(JDWPPacket&)), 0, 0);
    disconnect(updateClassMethodsAction, SIGNAL (triggered(JDWPPacket&)), 0, 0);
    
    connect(updateClassMethodsAction, SIGNAL (triggered(JDWPPacket&)),
            classes[index.row()] , SLOT(setMethods(JDWPPacket&)));
    connect(updateClassFieldsAction,  SIGNAL (triggered(JDWPPacket&)),
            classes[index.row()] , SLOT(setFields(JDWPPacket&)));
    connect(classes[index.row()], SIGNAL (initialized()), this, SIGNAL (treeModelReady()));
        
    // If the class is already initialized, emit treeModelReady signal and return.
    if (classes[index.row()]->isInitialized()) {
        emit treeModelReady();
        return;
    }
    
    // Else we need to retrieve the fields and/or methods
    if (! classes[index.row()]->isFieldsSet()) {
        JDWPCommand *fieldsCmd = JDWPCommand::newCommand(REF_TYPE_CMD_FIELDS);
        *fieldsCmd << classes[index.row()]->getRefTypeId();
        connection.sendRequest(fieldsCmd, updateClassFieldsAction);
    }
    
    if (! classes[index.row()]->isMethodsSet()) {    
        JDWPCommand *methodsCmd = JDWPCommand::newCommand(REF_TYPE_CMD_METHODS);
        *methodsCmd << classes[index.row()]->getRefTypeId();
        connection.sendRequest(methodsCmd, updateClassMethodsAction);
    }
}

ClassTableModel* ClassToolSet::getClassTableModel()
{
    return classTableModel;
}

ClassTreeModel* ClassToolSet::getClassTreeModel()
{
    return classTreeModel;
}

void ClassToolSet::disassembleBytecodes(const QModelIndex &index, bool disassemble,
                                bool lineNumbers, bool fld, bool internal, bool verbose)
{
    // clear any residual data
    javapOutput.clear();

    // get the referenceType ID for the class to disassemble
    ReferenceTypeID refid = classes[index.row()]->getRefTypeId();
    
    // get the disassembly options
    QString options;
    if (disassemble)
        options.append("d");
    if (lineNumbers)
        options.append("l");
    if (fld)
        options.append("f");
    if (internal)
        options.append("i");
    if (verbose)
        options.append("v");
    
    if (options.length() < 5) {
        options.resize(5);
    }
    
    if(connection.isAttached()) {
        JDWPCommand *cmd = JDWPCommand::newCommand(BC_TYPE_CMD_DISASSEMBLE); 
        *cmd << refid << options;           
        connection.sendRequest(cmd,  byteCodesAction);
    }
}

void ClassToolSet::prepareBytecodeOutput(QTextDocument *doc)
{
    doc->setPlainText(javapOutput);
}

void ClassToolSet::getBytecodeString(JDWPPacket& packet)
{
    if (packet.getErrorCode() != ERR_NONE) {
        QString err;
        connection.getJDWPError(packet.getErrorCode(), err);
       javapOutput.append("Bytecode generation failed:").append("\n");
       javapOutput.append(err).append("\n");
    } else {
        packet >> javapOutput;
    }
    emit bytecodesReady();
}


} // end namespace toolset
} // end namespace inspectorj

---