inspectorJ -- JavaTM Profiler

sf project site browse source checkout source

src/inspectorj/jdwp/jdwppacket.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 "jdwppacket.h"
#include <iostream>
#include <QIODevice>
#include <QDataStream>

using namespace inspectorj::jdwp;
        
// initialize the static data type sizes
int JDWPPacket::FIELD_ID_SIZE = 0;
int JDWPPacket::OBJECT_ID_SIZE = 0;
int JDWPPacket::METHOD_ID_SIZE = 0;
int JDWPPacket::REFERENCE_TYPE_ID_SIZE = 0;
int JDWPPacket::FRAME_ID_SIZE = 0;
                
JDWPPacket::JDWPPacket()
{
}

JDWPPacket::~JDWPPacket()
{
    delete data;
    data = 0;
}

void JDWPPacket::setReadMode()
{
    if (! data->device()->isReadable()) {
        data->device()->open(QIODevice::ReadOnly);
    }    
}

void JDWPPacket::setWriteMode()
{
    if (! data->device()->isWritable()) {
        data->device()->open(QIODevice::WriteOnly);
    }    
}

void JDWPPacket::initIDSizes(JDWPPacket &packet)
{
    packet >> JDWPPacket::FIELD_ID_SIZE;
    packet >> JDWPPacket::METHOD_ID_SIZE;
    packet >> JDWPPacket::OBJECT_ID_SIZE;
    packet >> JDWPPacket::REFERENCE_TYPE_ID_SIZE;
    packet >> JDWPPacket::FRAME_ID_SIZE;
}

short JDWPPacket::getErrorCode()
{
    return ERR_NONE;
}

jint JDWPPacket::getLength()
{
    return bytes.size() + JDWP_HEADER_LENGTH;
}

char* JDWPPacket::getDataBytes()
{
    return this->bytes.data();
}


/* Write operators */

JDWPPacket& JDWPPacket::operator<<(jbyte &b)
{
    setWriteMode();
    *data << (jbyte) b;
    return *this;
}

JDWPPacket& JDWPPacket::operator<<(bool &b)
{
    setWriteMode();
    if (b) {
        *data << (jbyte) 1;
    } else {
        *data << (jbyte) 0;
    }
    return *this;
}

JDWPPacket& JDWPPacket::operator<<(jint &i)
{
    setWriteMode();
    if (sizeof(jint) == sizeof(qint32)) {
        *data << (qint32)i;
    } else if (sizeof(jint) == sizeof(qint64)) {
        *data << (qint64)i;
    }
    return *this;
}

#if defined (_WIN32)

JDWPPacket& JDWPPacket::operator<<(qint32 &i)
{
    setWriteMode();
    *data << i;
    return *this;
}
#endif

JDWPPacket& JDWPPacket::operator<<(qint64 &i)
{
    setWriteMode();
    *data << i;
    return *this;
}

JDWPPacket& JDWPPacket::operator<<(QString &s)
{
    setWriteMode();
    
    // add the string size
    *data << (jint) s.size();

    // add the string data (char*)
    data->writeRawData(s.toAscii().data(), s.size());
    return *this;
}

JDWPPacket& JDWPPacket::operator<<(ObjectID &oid)
{
    if (sizeof(qint64) == OBJECT_ID_SIZE) {
        *this << oid.value;
    } 
    else if (sizeof(qint32) == OBJECT_ID_SIZE) {
        *this << (qint32&)(oid.value);
    }
    
    return *this;
}

JDWPPacket& JDWPPacket::operator<<(FieldID &fid)
{
    if (sizeof(qint64) == FIELD_ID_SIZE) {
        *this << fid.value;
    } 
    else if (sizeof(qint32) == FIELD_ID_SIZE) {
        *this << (qint32&)(fid.value);
    }
    
    return *this;
}


JDWPPacket& JDWPPacket::operator<<(MethodID &mid)
{
    if (sizeof(qint64) == METHOD_ID_SIZE) {
        *this << mid.value;
    } 
    else if (sizeof(qint32) == METHOD_ID_SIZE) {
        *this << (qint32&)(mid.value);
    }
    
    return *this;
}
        
JDWPPacket& JDWPPacket::operator<<(ReferenceTypeID &rid)
{
    if (sizeof(qint64) == REFERENCE_TYPE_ID_SIZE) {
        *this << rid.value;
    } 
    else if (sizeof(qint32) == REFERENCE_TYPE_ID_SIZE) {
        *this << (qint32&)(rid.value);
    }
    
    return *this;
}

JDWPPacket& JDWPPacket::operator<<(FrameID &fid)
{
    if (sizeof(qint64) == FRAME_ID_SIZE) {
        *this << fid.value;
    } 
    else if (sizeof(qint32) == FRAME_ID_SIZE) {
        *this << (qint32&)(fid.value);
    }
    
    return *this;
}

/*  Read operators */

JDWPPacket& JDWPPacket::operator>>(jbyte &b)
{
    setReadMode();
    *data >> b;
    return *this;
}

JDWPPacket& JDWPPacket::operator>>(bool &b)
{
    setReadMode();
    jbyte jb;
    *data >> jb;
    
    b = (jb != 0 );

    return *this;
}

JDWPPacket& JDWPPacket::operator>>(qint32 &i)
{
    setReadMode();
    *data >> i;
    return *this;
}

JDWPPacket& JDWPPacket::operator>>(qint64 &i)
{
    setReadMode();
    *data >> i;
    return *this;
}

JDWPPacket& JDWPPacket::operator>>(QString &s)
{
    setReadMode();
    
    // get the length of the data string
    int len;
    *data >> len;
    
    // create a char array and read the data from the socket
    char *tmp = new char[len + 1];
    data->readRawData(tmp, len);
    tmp[len] = 0;
    
    // create a QString from the data string and assign it to s
    s = QString(tmp);
    delete[] tmp;
    tmp = NULL;
    
    return *this;
}

JDWPPacket& JDWPPacket::operator>>(ObjectID &oid)
{
    if (sizeof(qint64) == OBJECT_ID_SIZE) {
        *this >> oid.value;
    } 
    else if (sizeof(qint32) == OBJECT_ID_SIZE) {
        qint32 tmp;
        *this >> tmp;
        oid.value = qint64(tmp);
    }
    return *this;
}

JDWPPacket& JDWPPacket::operator>>(FieldID &fid)
{
    if (sizeof(qint64) == FIELD_ID_SIZE) {
        *this >> fid.value;
    } 
    else if (sizeof(qint32) == FIELD_ID_SIZE) {
        qint32 tmp;
        *this >> tmp;
        fid.value = qint64(tmp);
    }
    return *this;
}

JDWPPacket& JDWPPacket::operator>>(MethodID &mid)
{
    if (sizeof(qint64) == METHOD_ID_SIZE) {
        *this >> mid.value;
    } 
    else if (sizeof(qint32) == METHOD_ID_SIZE) {
        qint32 tmp;
        *this >> tmp;
        mid.value = qint64(tmp);
    }
    return *this;
}

JDWPPacket& JDWPPacket::operator>>(ReferenceTypeID &rid)
{
    if (sizeof(qint64) == REFERENCE_TYPE_ID_SIZE) {
        *this >> rid.value;
    } 
    else if (sizeof(qint32) == REFERENCE_TYPE_ID_SIZE) {
        qint32 tmp;
        *this >> tmp;
        rid.value = qint64(tmp);
    }
    return *this;
}

JDWPPacket& JDWPPacket::operator>>(FrameID &fid)
{
    if (sizeof(qint64) == FRAME_ID_SIZE) {
        *this >> fid.value;
    } 
    else if (sizeof(qint32) == FRAME_ID_SIZE) {
        qint32 tmp;
        *this >> tmp;
        fid.value = qint64(tmp);
    }
    return *this;
}

---