NetOriginalMessageProtocol.cpp

Go to the documentation of this file.

/***************************** License **********************************

 Copyright (C) 2008 by Communicative Machines
 http://www.cmlabs.com   All rights reserved.

 This library is free software; you can redistribute it and/or
 modify it under the terms of the GNU Lesser General Public
 License as published by the Free Software Foundation; either
 version 2.1 of the License, or (at your option) any later version.

 This library 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
 Lesser General Public License for more details.

 You should have received a copy of the GNU Lesser General Public
 License along with this library; if not, write to the Free Software
 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA

************************************************************************/
// NetOriginalMessageProtocol.cpp: implementation of the NetOriginalMessageProtocol class.
//
//////////////////////////////////////////////////////////////////////

#include "NetOriginalMessageProtocol.h"

namespace cmlabs {

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

NetOriginalMessageProtocol::NetOriginalMessageProtocol():NetProtocol("Original Message")
{
        netTimeout = 60000;
}

NetOriginalMessageProtocol::~NetOriginalMessageProtocol()
{

}

Object* NetOriginalMessageProtocol::clone() const {
        return new NetOriginalMessageProtocol();
}

bool NetOriginalMessageProtocol::checkBufferForCompatibility(char* buffer, int length) {

        if (length < 8) return false;

        JString str = buffer;
        if (str.startsWith("<message"))
                return true;
        
        return false;
}

bool NetOriginalMessageProtocol::initializeConversation(JSocket* socket) {
        if (socket == NULL)
                return false;

        return true;
}

bool NetOriginalMessageProtocol::sendObject(JSocket* socket, Message* msg, bool isReply) {


        int res;
        JString str;

        if (!isReply) {
                Message* greeting = new Message(msg->getFrom(), msg->getTo(), "CONNECT_REQUEST");
                str = greeting->toXML();
                str = str.removeCRs();
                res = socket->writeln(str);
                delete(greeting);
                if (res < (int)str.length()) {
                        if (DEBUGLEVEL(STATUS)) {
                                printf("[NetOrigMsg] Failed to send CONNECT_REQUEST\n");
                        }
                        return false;
                }

                Message* hmsg = receiveObject(socket, 1000);
                if (hmsg == NULL) {
                        if (DEBUGLEVEL(STATUS)) {
                                printf("[NetOrigMsg] Failed to receive CONNECT_ACCEPT\n");
                        }
                        return false;
                }

                if (!hmsg->getType().equalsIgnoreCase("CONNECT_ACCEPT")) {
                        if (DEBUGLEVEL(STATUS)) {
                                printf("[NetOrigMsg] Received something else: %s\n", (char*) hmsg->getType());
                        }
                        delete(hmsg);
                        return false;
                }
                delete(hmsg);
        }

        str = msg->toXML();
        str = str.removeCRs();
        res = socket->writeln(str);
        if (res < (int)str.length())
                return false;
//      delete(msg);
        return true;
}

Message* NetOriginalMessageProtocol::receiveObject(JSocket* socket, int timeout) {

        Message* msg = NULL;
        JString str = socket->readln(timeout);
        if (str.length() > 0)
                str = str;
        if (!str.startsWith("<message"))
                return NULL;
        msg = new Message(str);

        if (msg->getType().equalsIgnoreCase("CONNECT_REQUEST")) {
                Message* greeting = new Message(msg->getTo(), msg->getFrom(), "CONNECT_ACCEPT");
                delete(msg);
                if (DEBUGLEVEL(KITCHENSINK)) {
                        printf("[NetOrigMsg] Sending CONNECT_ACCEPT...\n");
                }
                str = greeting->toXML();
                delete(greeting);
                str = str.removeCRs();
                int res = socket->writeln(str);
                if (res < (int)str.length()) {
                        return NULL;
                }

                str = socket->readln(timeout);
                if (!str.startsWith("<message")) {
                        return NULL;
                }
                msg = new Message(str);
        }
/*      else if (msg->getType().length() == 0) {
                delete(msg);
                return NULL;
        }
*/
        JString hostname = socket->getRemoteIPAddress();
        if (hostname.equals("localhost"))
                hostname = socket->getLocalIPAddress();

        if (msg->getOrigin().length() == 0)
                msg->setOrigin(hostname);
        msg->set("netprotocol", name);

        return msg;
}













} // namespace cmlabs

---