inspectorJ -- JavaTM Profiler

sf project site browse source checkout source

src/inspectorj/agent/agentpackethandler.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 "agentpackethandler.h"

using namespace inspectorj::agent::commandset;            
using namespace inspectorj::jdwp;
        
namespace inspectorj {
namespace agent {
    
AgentPacketHandler::AgentPacketHandler(JavaVM* vm, JNIEnv *env, jvmtiEnv *jvmti, AgentSocket *socket)
    : vm(vm), env(env), jvmti(jvmti), socket(socket)
{   
    // initialize the command sets    
    createCommandSets(); 
}

AgentPacketHandler::~AgentPacketHandler()
{
    if (vmCmdSet) {
        delete vmCmdSet;
       // delete refTypeCmdSet;
    }
}

void AgentPacketHandler::createCommandSets()
{
    vmCmdSet = new VirtualMachineCmdSet(vm, env, jvmti);
    refTypeCmdSet = new ReferenceTypeCmdSet(vm, env, jvmti);
    bcCmdSet = new BytecodeCmdSet(vm, env, jvmti);
}

void AgentPacketHandler::attachCurrentThreadToVM()
{
    vm->AttachCurrentThread((void**)&env, NULL);    
}


void AgentPacketHandler::writeReply(JDWPReply* reply)
{
    if (reply) {
        socket->writeReply(reply);
    }
}


void AgentPacketHandler::processCommand(JDWPCommand *cmd)
{
    JDWPCommandSet *cmdSet = getCommandSet(cmd->getCmdSet());
    
    // Check to see if the command is supported
    if (!cmdSet) {
        jdwpReplyPacket replyPkt;
        replyPkt.len = (jint)11;
        replyPkt.id = cmd->getId();
        replyPkt.flags = (jbyte)JDWPTRANSPORT_FLAGS_REPLY;
        replyPkt.errorCode = (jshort)ERR_NOT_IMPLEMENTED;
        
        JDWPReply *reply = new JDWPReply(replyPkt);
        writeReply(reply);
    } else {
        cmdSet->process(cmd, boost::bind(&AgentPacketHandler::writeReply, this, _1));  
    } 
    
    delete cmd;
}

void AgentPacketHandler::setAgentSocket(AgentSocket *& socket)
{
    this->socket = socket;
}

JDWPCommandSet* AgentPacketHandler::getCommandSet(jbyte cmd)
{
    switch (cmd) {
        /*
    CLASS_TYPE_COMMAND_SET = 3,
    ARRAY_TYPE_COMMAND_SET = 4,
    IF_COMMAND_SET = 5,
    METHOD_COMMAND_SET = 6,
    OBJECT_REF_COMMAND_SET = 9,
    STRING_REF_COMMAND_SET = 10,
    THREAD_REF_COMMAND_SET = 11,
    THREAD_GROUP_REF_COMMAND_SET = 12,
    ARRAY_REF_COMMAND_SET = 13,
    CLASS_LDR_REF_COMMAND_SET = 14,
    EVENT_REQUEST_COMMAND_SET = 15,
    STACK_FRAME_COMMAND_SET = 16,
    CLASS_OBJECT_REF_COMMAND_SET = 17,
    EVENT_COMMAND_SET = 64,        
        
        */
        case BYTECODE_COMMAND_SET : return bcCmdSet;
        case VM_COMMAND_SET : return vmCmdSet;
        case REF_TYPE_COMMAND_SET : return refTypeCmdSet;
        default: return (JDWPCommandSet*)0;
        
    }
    return vmCmdSet;
}

} // end namespace agent
} // end namespace inspectorj

---