inspectorJ -- JavaTM Profiler

sf project site browse source checkout source

src/inspectorj/agent/agentsocket.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/agent/agentsocket.h"
        
using inspectorj::jdwp::JDWPReply;
using inspectorj::jdwp::JDWP_HEADER_LENGTH;
        
namespace inspectorj {
namespace agent {
        
AgentSocket::AgentSocket(tcp::socket* socket, boost::system::error_code* error)
    : socket(socket), error(error)
{       
}

bool AgentSocket::read(jint &i)
{    
    size_t bytesRead = 0;
    char *p = (char *)(&i);
    char b[sizeof(jint)];
    bytesRead = boost::asio::read(*socket, 
                                    boost::asio::buffer(b,sizeof(jint)),
                                    boost::asio::transfer_all(), 
                                    *error);
    int sz = (int) sizeof(jint);
    sz--;

    while (sz > 0) {
        *p++ = b[sz];
        sz--;
    }
    *p   = b[0];
        
    return (bytesRead == sizeof(jint) && !(*error));       
}

bool AgentSocket::read(jbyte &j)
{
    size_t bytesRead = 0;
    char *p = (char *)(&j);
    char b[1];
    bytesRead = boost::asio::read(*socket, 
                                    boost::asio::buffer(b,1),
                                    boost::asio::transfer_all(), 
                                    *error);
    *p = b[0];    
    
    return (bytesRead == 1 && !(*error));
}    

bool AgentSocket::read(char *ch, int len)
{
    size_t bytesRead = 0;
    bytesRead = boost::asio::read(*socket, 
                                    boost::asio::buffer(ch, len),
                                    boost::asio::transfer_all(), *error); 
    
    return (bytesRead == len && !(*error));
}

bool AgentSocket::write(jint i)
{    
    size_t bytesWritten = 0;
    char *p = (char *)(&i);    
    char b[sizeof(jint)];

    int sz = (int) sizeof(jint);
    sz--;

    while (sz > 0) {
        b[sz] = *p++;
        sz--;
    }
    b[0] = *p;

    bytesWritten = boost::asio::write(*socket, 
                                    boost::asio::buffer(b, sizeof(jint)),
                                    boost::asio::transfer_all(), 
                                    *error);
       
    return (bytesWritten == sizeof(jint) && !(*error));       
}

bool AgentSocket::write(short s)
{    
    size_t bytesWritten = 0;
    char *p = (char *)(&s);    
    char b[2];
    b[1] = *p++;
    b[0] = *p;    
    bytesWritten = boost::asio::write(*socket, 
                                    boost::asio::buffer(b,2),
                                    boost::asio::transfer_all(), 
                                    *error);
       
    return (bytesWritten == 2 && !(*error));       
}

bool AgentSocket::write(jbyte j)
{
    size_t bytesWritten = 0;
    char *p = (char *)(&j);
    char b[1];
    b[0] = *p;
    bytesWritten = boost::asio::write(*socket, 
                                    boost::asio::buffer(b,1),
                                    boost::asio::transfer_all(), 
                                    *error);
    
    return (bytesWritten == 1 && !(*error));
}    


bool AgentSocket::writeReply(JDWPReply *reply)
{
    if (reply) {
        // send the reply           
        size_t bytesWritten = 0; 
        write(reply->getLength());
        write(reply->getId());
        write(reply->getPacketType());
        write(reply->getErrorCode());
    
        bytesWritten = boost::asio::write(*socket, 
                                        boost::asio::buffer(reply->getDataBytes(),
                                                            reply->getLength() - JDWP_HEADER_LENGTH),
                                        boost::asio::transfer_all(), 
                                        *error);    
        delete reply;
        reply = 0;
        return (*error); 
    }
    
    return false;
   
}
  
} // end namespace agent
} // end namespace inspectorj

---