KjClient.cpp

Go to the documentation of this file.

/***************************************************************************
 *   Copyright (C) 2007 by Jan Koci   *
 *   honza.koci@email.cz   *
 *   http://kengine.sourceforge.net/tutorial/
 *                                                                         *
 *   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 "KjClient.h"



KjClient::KjClient()
{
        m_pCon = new KjSslConnection(KjSslConnection::KJ_SSL_CLIENT,KJ_TEST_CLIENT_CONN, KJ_TEST_CRT, KJ_TEST_KEY);
        m_pSignal = new KjSignal();
}

KjClient::~KjClient()
{
        if(m_pCon)
        {
                delete m_pCon;
                m_pCon = NULL;
        }
        if(m_pSignal)
        {
                delete m_pSignal;
                m_pSignal = NULL;
        }
}

int KjClient::Start()
{
        m_pSignal->InitSignals();
    if(!m_pCon)
        {
                cerr<< "Error:: int KjClient::Start() -- KjSslConnection can`t allocated" << endl;
                return 0;
        }
        if(m_pCon->GetIsValid())
        {
                string strClientMsg;
                string strServerMsg;
                char str[KJ_SSL_READ_CHUNK_SIZE];
                bool bCanWrite = true;
                // Check key CTRL-C or CTRL-\ */
                while(!m_pSignal->IsSigInt() && !m_pSignal->IsSigQuit()) 
                {
                        cout << "Please enter some message ... " << endl;
                        // If length of input string is correct 
                        if(cin.getline(str, KJ_SSL_READ_CHUNK_SIZE) != NULL)
                        {
                                strClientMsg += str;
                                // Delete end message if exists
                                DeleteEndMsg(strClientMsg);
                                //strClientMsg += KJ_END_MSG;
                                if(bCanWrite)
                                {
                                        bCanWrite = false;
                                        cout << "Client send message '" << strClientMsg << "'" << endl;
                                        m_pCon->Write(strClientMsg);
                                        // Wait timeout interval 
                                        Wait(500);
                                        // Read message from server, 
                                        bool bIsEndMsg = false;
                                        strServerMsg.clear();
                                        do
                                        {
                                                // Read some packet
                                                string strPacket;
                                                m_pCon->Read(strPacket);
                                                if(strPacket.empty())
                                                {
                                                        cout << "Server is disconnected ... '" << endl;
                                                        return 0;
                                                }
                                                strServerMsg += strPacket;
                                                size_t iFound = strPacket.find(KJ_END_MSG);
                                                if (iFound!=string::npos)
                                                {
                                                        bIsEndMsg = true;
                                                }
                                        }while (bIsEndMsg == false);
                                        // Delete end message if exists
                                        DeleteEndMsg(strServerMsg);
                                        cout << "Received server message is '" << strServerMsg << "'" << endl;
                                        // Clear client message
                                        strClientMsg.clear();
                                        bCanWrite = true;
                                }
                        }
                        else
                        {
                                cerr << "Error :: KjClient::Start() -- Input string is incorrect"  << endl;
                                cout << "Maximum length of input string must be "  << KJ_SSL_READ_CHUNK_SIZE << endl;
                                return 0;
                        }
                }
        }
        else
        {
                cerr<< "Error:: int KjClient::Start() --Invalid SSL connection !!! " << endl;
        }
        return 0;
        
}

void KjClient::DeleteEndMsg(string &_strMsg)
{
        size_t iFound = _strMsg.find(KJ_END_MSG);
        if (iFound!=string::npos)
        {
                string strEndMsg = KJ_END_MSG;
                _strMsg.erase(iFound , strEndMsg.size());
        }
}

---