|

TCPAuthServer.hpp

Go to the documentation of this file.

#ifndef IONFLUX_TOOLS_TCPAUTHSERVER
#define IONFLUX_TOOLS_TCPAUTHSERVER
/* ==========================================================================
 * Ionflux Tools
 * Copyright (c) 2004 Joern P. Meier
 * mail@ionflux.org
 * --------------------------------------------------------------------------
 * TCPAuthServer.hpp                   Authenticated TCP server.
 * ==========================================================================
 * 
 * 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/TCPMessageServer.hpp"
#include "ionflux/AuthFile.hpp"

namespace Ionflux
{

namespace Tools
{

class TCPAuthServer
: public TCPMessageServer
{
    protected:
        Auth* auth;
        std::string authFileName;
        unsigned int currentMsgID;
        MTRand *randomizer;
        
        virtual void onConnect(TCPRemotePeer &client);
        
        virtual void onAuthMessage(TCPRemotePeer &peer);
        
        virtual void onAuthOK(TCPRemotePeer &peer);
        
        virtual void onAuthFailed(TCPRemotePeer &peer);
        
    public:
        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_HELLO;
        static const int PS_WAIT_AUTH;
        static const int PS_AUTH_OK;
        
        TCPAuthServer();
        
        TCPAuthServer(const std::string& initAuthFileName);
        
        virtual ~TCPAuthServer();
        
        virtual void onMessage(TCPRemotePeer &peer);
        
        virtual void onInvalidMessage(TCPRemotePeer &peer);
        
        virtual void broadcast(const std::string &bytes);
        
        virtual void setAuth(Auth* newAuth);
        
        virtual void setAuthFileName(const std::string& newAuthFileName);
        
        virtual void setCurrentMsgID(unsigned int newCurrentMsgID);
        
        virtual Auth* getAuth();
        
        virtual std::string getAuthFileName();
        
        virtual unsigned int getCurrentMsgID();
};

}

}

#endif

---