|

TCPAuthClient.hpp

Go to the documentation of this file.

#ifndef IONFLUX_TOOLS_TCPAUTHCLIENT
#define IONFLUX_TOOLS_TCPAUTHCLIENT
/* ==========================================================================
 * Ionflux Tools
 * Copyright (c) 2004 Joern P. Meier
 * mail@ionflux.org
 * --------------------------------------------------------------------------
 * TCPAuthClient.hpp                   Authenticated TCP client.
 * ==========================================================================
 * 
 * This file is part of Ionflux Tools.
 * 
 * Ionflux Tools 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.
 * 
 * Ionflux Tools 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 Ionflux Tools; if not, write to the Free Software Foundation, 
 * Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 * 
 * ========================================================================== */

#include <string>
#include <iostream>
#include <vector>
#include "ionflux/tools.hpp"
#include "ionflux/TCPMessageClient.hpp"

namespace Ionflux
{

namespace Tools
{

class TCPAuthClient
: public TCPMessageClient
{
    protected:
        std::string username;
        std::string password;
        unsigned int currentMsgID;
        
        virtual void onConnect(TCPRemotePeer &peer);
        
        virtual void onAuthMessage(TCPRemotePeer &peer);
        
        virtual void onAuthOK(TCPRemotePeer &peer);
        
        virtual void onAuthFailed(TCPRemotePeer &peer);
        
    public:
        using TCPClient::addConnection;
        
        static const std::string MAGIC_WORD;
        static const unsigned int AUTH_VERSION;
        static const unsigned int AUTH_VERSION_UPPER;
        static const unsigned int AUTH_VERSION_LOWER;
        static const int MSGID_HELLO;
        static const int MSGID_WELCOME;
        static const int MSGID_AUTH;
        static const int MSGID_AUTH_OK;
        static const int MSGID_AUTH_FAILED;
        static const int PS_WAIT_WELCOME;
        static const int PS_WAIT_AUTH_RESULT;
        static const int PS_AUTH_OK;
        
        TCPAuthClient();
        
        TCPAuthClient(bool initInteractive);
        
        TCPAuthClient(const std::string &initUsername, 
            const std::string &initPassword, bool initInteractive);
        
        virtual ~TCPAuthClient();
        
        virtual void onMessage(TCPRemotePeer &peer);
        
        virtual void onInvalidMessage(TCPRemotePeer &peer);
        
        virtual void broadcast(const std::string &bytes);
        
        virtual bool addConnection(const std::string &host, int port, 
            const std::string &connUser, const std::string &connPass);
        
        virtual void setUsername(const std::string& newUsername);
        
        virtual void setPassword(const std::string& newPassword);
        
        virtual void setCurrentMsgID(unsigned int newCurrentMsgID);
        
        virtual std::string getUsername();
        
        virtual void getPassword(std::string &target);
        
        virtual unsigned int getCurrentMsgID();
};

}

}

#endif

---